mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
b8dd807c31
* Use `collections.defaultdict` instead of a regular dict, so we don't have to initialise the dict keys * Add edits to `edits_per_file` directly, rather than via an intermediate set * Use tuple unpacking in the final loop so it's easier to see the similarity between the unpacked variables and the tuple added on line 47
76 lines
2.3 KiB
Python
Executable File
76 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# utils/apply-fixit-edits.py - Apply edits from .remap files -*- python -*-
|
|
#
|
|
# This source file is part of the Swift.org open source project
|
|
#
|
|
# Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
|
|
# Licensed under Apache License v2.0 with Runtime Library Exception
|
|
#
|
|
# See http://swift.org/LICENSE.txt for license information
|
|
# See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
|
|
from __future__ import print_function
|
|
|
|
import argparse
|
|
import collections
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
|
|
def find_remap_files(path):
|
|
for root, dirs, files in os.walk(path):
|
|
for filename in files:
|
|
if not filename.endswith(".remap"):
|
|
continue
|
|
yield os.path.join(root, filename)
|
|
|
|
|
|
def apply_edits(path):
|
|
remap_files = find_remap_files(path)
|
|
if not remap_files:
|
|
print("No remap files found")
|
|
return 1
|
|
|
|
edits_per_file = collections.defaultdict(list)
|
|
for remap_file in remap_files:
|
|
with open(remap_file) as f:
|
|
json_data = f.read()
|
|
json_data = json_data.replace(",\n }", "\n }")
|
|
json_data = json_data.replace(",\n]", "\n]")
|
|
curr_edits = json.loads(json_data)
|
|
for ed in curr_edits:
|
|
fname = ed["file"]
|
|
offset = ed["offset"]
|
|
length = ed.get("remove", 0)
|
|
text = ed.get("text", "")
|
|
edits_per_file[fname].append((offset, length, text))
|
|
|
|
for fname, edits in edits_per_file.iteritems():
|
|
print('Updating', fname)
|
|
edits.sort(reverse=True)
|
|
with open(fname) as f:
|
|
file_data = f.read()
|
|
for ed in edits:
|
|
offset, length, text = ed
|
|
file_data = file_data[:offset] + str(text) + \
|
|
file_data[offset + length:]
|
|
with open(fname, 'w') as f:
|
|
f.write(file_data)
|
|
return 0
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
description="""Finds all .remap files in a directory and applies their
|
|
edits to the source files.""")
|
|
parser.add_argument("build_dir_path",
|
|
help="path to index info")
|
|
|
|
args = parser.parse_args()
|
|
return apply_edits(args.build_dir_path)
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|