diff options
Diffstat (limited to 'utils/opt-viewer/opt-diff.py')
| -rwxr-xr-x | utils/opt-viewer/opt-diff.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/utils/opt-viewer/opt-diff.py b/utils/opt-viewer/opt-diff.py index 2b5d1bbfabbec..9e921f8488d36 100755 --- a/utils/opt-viewer/opt-diff.py +++ b/utils/opt-viewer/opt-diff.py @@ -44,20 +44,21 @@ if __name__ == '__main__': default=cpu_count(), type=int, help='Max job count (defaults to %(default)s, the current CPU count)') + parser.add_argument( + '--no-progress-indicator', + '-n', + action='store_true', + default=False, + help='Do not display any indicator of how many YAML files were read.') parser.add_argument('--output', '-o', default='diff.opt.yaml') args = parser.parse_args() - if args.jobs == 1: - pmap = map - else: - pool = Pool(processes=args.jobs) - pmap = pool.map - files1 = find_files(args.yaml_dir_or_file_1) files2 = find_files(args.yaml_dir_or_file_2) - all_remarks1, _, _ = optrecord.gather_results(pmap, files1) - all_remarks2, _, _ = optrecord.gather_results(pmap, files2) + print_progress = not args.no_progress_indicator + all_remarks1, _, _ = optrecord.gather_results(files1, args.jobs, print_progress) + all_remarks2, _, _ = optrecord.gather_results(files2, args.jobs, print_progress) added = set(all_remarks2.values()) - set(all_remarks1.values()) removed = set(all_remarks1.values()) - set(all_remarks2.values()) @@ -66,5 +67,5 @@ if __name__ == '__main__': r.Added = True for r in removed: r.Added = False - stream = file(args.output, 'w') - yaml.dump_all(added | removed, stream) + with open(args.output, 'w') as stream: + yaml.dump_all(added | removed, stream) |