You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

75 lines
2.3 KiB
Plaintext

#!/usr/bin/env python3
"""show dirty git repos"""
from __future__ import division, print_function
from colorama import Fore
from pathlib import Path
import contextlib
import os
import re
import subprocess
import click
# TODO config file (and defaults here)
IGNORES = [
r"\.sync",
r"\.git/modules", # XXX check this again
r"\.local/share/containers/storage",
]
def git_directories(startdir) -> Path:
for dirpath, dirnames, _ in os.walk(startdir):
if any(re.search(ignore, dirpath) for ignore in IGNORES):
continue
if set(['info', 'objects', 'refs']).issubset(set(dirnames)):
yield Path(dirpath)
@contextlib.contextmanager
def working_directory(directory: Path):
saved_cwd = os.getcwd()
os.chdir(directory)
yield
os.chdir(saved_cwd)
@click.command
@click.argument('topdirs', nargs=-1, type=click.Path(exists=True, file_okay=False,
path_type=Path))
@click.option('-l', '--list', 'list_', is_flag=True, default=False,
help='Just list the dirty directories')
def search_dirty(topdirs, list_):
"""Search for dirty git working directories in TOPDIRS"""
if len(topdirs) == 0:
topdirs = [Path(".")]
for topdir in topdirs:
for git_directory in git_directories(topdir):
# technically,we could have a different GIT_DIR than ".git", but this script
# assumes ".git".
if git_directory.parts[-1] != ".git":
continue
work_tree_directory = git_directory.parent
with working_directory(work_tree_directory):
try:
out = subprocess.check_output(['git', 'status', '-s'], stderr=subprocess.STDOUT)
if len(out) > 0:
if not list_:
print('== {}\n{}'.format(work_tree_directory, out.decode('utf-8')))
else:
print(work_tree_directory)
except subprocess.CalledProcessError as e:
print((Fore.RED + 'git status is unhappy with {}' + Fore.RESET)
.format(work_tree_directory))
print(e.output)
if __name__ == '__main__':
search_dirty()