mirror of
https://github.com/elliotnunn/supermario.git
synced 2024-11-26 01:49:19 +00:00
91 lines
2.9 KiB
Python
Executable File
91 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
from os import path, getcwd, makedirs, listdir, remove, rename
|
|
from subprocess import run, DEVNULL
|
|
|
|
|
|
def folder(default_search, user_input):
|
|
if path.sep in user_input:
|
|
return user_input
|
|
else:
|
|
the_path = path.dirname(path.dirname(path.abspath(__file__)))
|
|
the_path = path.join(the_path, default_search, user_input)
|
|
return the_path
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='''
|
|
Shrink a git changelog into a patchset (based on git format-patch)
|
|
''')
|
|
|
|
parser.add_argument('worktree', metavar='WORKTREE', nargs='?', action='store', default=getcwd(), type=lambda x: folder('worktree', x), help='Worktree (default is cwd, or assumed in ../worktree/)')
|
|
parser.add_argument('patchset', metavar='PATCHSET', action='store', type=lambda x: folder('patchset', x), help='Destination patchset (assumed in ../patchset/)')
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
args.patchset = path.abspath(args.patchset)
|
|
|
|
|
|
assert not path.exists(path.join(args.patchset, '.git')) # protect against argument swap
|
|
assert path.exists(path.join(args.worktree, '.git'))
|
|
|
|
|
|
makedirs(args.patchset, exist_ok=True)
|
|
|
|
for do_delete in listdir(args.patchset):
|
|
do_delete = path.join(args.patchset, do_delete)
|
|
if path.splitext(do_delete)[1].lower() == '.patch':
|
|
remove(do_delete)
|
|
|
|
|
|
run(['git', 'format-patch', '-o', args.patchset, '--anchored=;', 'master-patchset-base'], stdout=DEVNULL, cwd=args.worktree, check=True)
|
|
|
|
|
|
# Edit the patches to look a bit better (and be more reproducible)
|
|
patchnames = [x for x in sorted(listdir(args.patchset)) if path.splitext(x)[1].lower() == '.patch']
|
|
strip_chars = min(len(x) - len(x.lstrip('0')) for x in patchnames)
|
|
|
|
for do_edit in patchnames:
|
|
real_file = path.join(args.patchset, do_edit)
|
|
fake_file = path.join(args.patchset, do_edit + '~')
|
|
new_file = path.join(args.patchset, do_edit[strip_chars:].lower())
|
|
|
|
print(path.basename(new_file))
|
|
|
|
rename(real_file, fake_file)
|
|
|
|
with open(new_file, 'w') as o, open(fake_file, 'r') as i:
|
|
for hdr_line in i:
|
|
if hdr_line == '\n':
|
|
o.write(hdr_line)
|
|
break
|
|
elif hdr_line.startswith('Subject:'):
|
|
if '[' in hdr_line:
|
|
o.write('Subject:' + hdr_line.rpartition(']')[2])
|
|
else:
|
|
o.write(hdr_line)
|
|
elif hdr_line.startswith('From:'):
|
|
o.write('From: Horst Beepmanh <>\n')
|
|
|
|
for msg_line in i:
|
|
o.write(msg_line)
|
|
if msg_line == '---\n': break
|
|
|
|
for stat_line in i:
|
|
if stat_line.startswith('---'):
|
|
o.write(stat_line)
|
|
break
|
|
|
|
is_separated = False
|
|
|
|
for diff_line in i:
|
|
if diff_line[0].isalpha():
|
|
if not is_separated: o.write('\n')
|
|
is_separated = True
|
|
else:
|
|
o.write(diff_line)
|
|
is_separated = False
|
|
|
|
remove(fake_file)
|