Refactor formatter

This commit is contained in:
Regen
2020-02-29 19:45:36 +09:00
parent 66af586b2a
commit cc9877adbe

View File

@@ -91,11 +91,12 @@ def main(args: List[str] = None):
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument('lists', type=Path, nargs='*', help='CMake list files') parser.add_argument('lists', type=Path, nargs='*', help='CMake list files')
parser.add_argument('-i', group = parser.add_mutually_exclusive_group()
group.add_argument('-i',
'--inplace', '--inplace',
action='store_true', action='store_true',
help='inplace edit') help='inplace edit')
parser.add_argument('-d', '--diff', action='store_true', help='show diff') group.add_argument('-d', '--diff', action='store_true', help='show diff')
parser.add_argument('--version', parser.add_argument('--version',
action='version', action='version',
version=f'%(prog)s {__version__}') version=f'%(prog)s {__version__}')
@@ -114,14 +115,12 @@ def main(args: List[str] = None):
if not remain: if not remain:
with listpath.open('w') as fp: with listpath.open('w') as fp:
fp.write(formatted) fp.write(formatted)
else: elif args.diff:
if args.diff:
diff = unified_diff(content.splitlines(True), diff = unified_diff(content.splitlines(True),
formatted.splitlines(True), str(listpath), formatted.splitlines(True), str(listpath),
str(listpath), '(before formatting)', str(listpath), '(before formatting)',
'(after formatting)') '(after formatting)')
diffstr = ''.join(diff) diffstr = ''.join(diff)
if diffstr:
print(diffstr, end='') print(diffstr, end='')
else: else:
print(formatted, end='') print(formatted, end='')