--- embedaddon/rsync/support/git-set-file-times 2013/10/14 07:51:15 1.1.1.2 +++ embedaddon/rsync/support/git-set-file-times 2021/03/17 00:32:36 1.1.1.3 @@ -1,37 +1,93 @@ -#!/usr/bin/perl -use strict; -use warnings; +#!/usr/bin/env python3 -# Sets mtime and atime of files to the latest commit time in git. -# -# This is useful after the first clone of the rsync repository BEFORE you -# do any building. It is also safe if you have done a "make distclean". +import os, re, argparse, subprocess +from datetime import datetime -my %ls; -my $commit_time; -my $prefix = @ARGV && $ARGV[0] =~ s/^--prefix=// ? shift : ''; +NULL_COMMIT_RE = re.compile(r'\0\0commit [a-f0-9]{40}$|\0$') -$/ = "\0"; -open FH, 'git ls-files -z|' or die $!; -while () { - chomp; - $ls{$_} = $_; -} -close FH; +def main(): + if not args.git_dir: + cmd = 'git rev-parse --show-toplevel 2>/dev/null || echo .' + top_dir = subprocess.check_output(cmd, shell=True, encoding='utf-8').strip() + args.git_dir = os.path.join(top_dir, '.git') + if not args.prefix: + os.chdir(top_dir) -$/ = "\n"; -open FH, "git log -r --name-only --no-color --pretty=raw -z @ARGV |" or die $!; -while () { - chomp; - if (/^committer .*? (\d+) (?:[\-\+]\d+)$/) { - $commit_time = $1; - } elsif (s/\0\0commit [a-f0-9]{40}$// or s/\0$//) { - my @files = delete @ls{split(/\0/, $_)}; - @files = grep { defined $_ } @files; - next unless @files; - map { s/^/$prefix/ } @files; - utime $commit_time, $commit_time, @files; - } - last unless %ls; -} -close FH; + git = [ 'git', '--git-dir=' + args.git_dir ] + + if args.tree: + cmd = git + 'ls-tree -z -r --name-only'.split() + [ args.tree ] + else: + cmd = git + 'ls-files -z'.split() + + proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, encoding='utf-8') + out = proc.communicate()[0] + ls = set(out.split('\0')) + ls.discard('') + + if not args.tree: + # All modified files keep their current mtime. + proc = subprocess.Popen(git + 'ls-files -m -z'.split(), stdout=subprocess.PIPE, encoding='utf-8') + out = proc.communicate()[0] + for fn in out.split('\0'): + if fn == '': + continue + if args.list: + mtime = os.lstat(fn).st_mtime + print_line(fn, mtime, mtime) + ls.discard(fn) + + cmd = git + 'log -r --name-only --no-color --pretty=raw --no-renames -z'.split() + if args.tree: + cmd.append(args.tree) + cmd += ['--'] + args.files + + proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, encoding='utf-8') + for line in proc.stdout: + line = line.strip() + m = re.match(r'^committer .*? (\d+) [-+]\d+$', line) + if m: + commit_time = int(m[1]) + elif NULL_COMMIT_RE.search(line): + line = NULL_COMMIT_RE.sub('', line) + files = set(fn for fn in line.split('\0') if fn in ls) + if not files: + continue + for fn in files: + if args.prefix: + fn = args.prefix + fn + mtime = os.lstat(fn).st_mtime + if args.list: + print_line(fn, mtime, commit_time) + elif mtime != commit_time: + if not args.quiet: + print(f"Setting {fn}") + os.utime(fn, (commit_time, commit_time), follow_symlinks = False) + ls -= files + if not ls: + break + proc.communicate() + + +def print_line(fn, mtime, commit_time): + if args.list > 1: + ts = str(commit_time).rjust(10) + else: + ts = datetime.utcfromtimestamp(commit_time).strftime("%Y-%m-%d %H:%M:%S") + chg = '.' if mtime == commit_time else '*' + print(chg, ts, fn) + + +if __name__ == '__main__': + parser = argparse.ArgumentParser(description="Set the times of the files in the current git checkout to their last-changed time.", add_help=False) + parser.add_argument('--git-dir', metavar='GIT_DIR', help="The git dir to query (defaults to affecting the current git checkout).") + parser.add_argument('--tree', metavar='TREE-ISH', help="The tree-ish to query (defaults to the current branch).") + parser.add_argument('--prefix', metavar='PREFIX_STR', help="Prepend the PREFIX_STR to each filename we tweak (defaults to the top of current checkout).") + parser.add_argument('--quiet', '-q', action='store_true', help="Don't output the changed-file information.") + parser.add_argument('--list', '-l', action='count', help="List files & times instead of changing them. Repeat for Unix timestamp instead of human readable.") + parser.add_argument('files', metavar='FILE', nargs='*', help="Specify a subset of checked-out files to tweak.") + parser.add_argument("--help", "-h", action="help", help="Output this help message and exit.") + args = parser.parse_args() + main() + +# vim: sw=4 et