mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from __future__ import print_function
|
|
|
|
import os
|
|
|
|
from argparse import ArgumentParser
|
|
|
|
parser = ArgumentParser(description='Patch some Python code generation, so it builds')
|
|
parser.add_argument("-f", "--file", required=True)
|
|
parser.add_argument("-p", "--patches", required=True)
|
|
parser.add_argument("-v", "--verbose", default=False, action="store_true")
|
|
parser.add_argument("-m", "--module", required=True)
|
|
args = parser.parse_args()
|
|
|
|
if os.path.isfile(args.patches):
|
|
with open(args.file) as fin:
|
|
# strip newlines for easier matching
|
|
lines = map(str.rstrip, fin.readlines())
|
|
|
|
patches = {}
|
|
with open(args.patches) as fin:
|
|
patches = eval(fin.read())
|
|
|
|
all_lines = []
|
|
for l in lines:
|
|
for patch_kind, patch_data in patches.items():
|
|
if patch_kind == "repl_line":
|
|
for from_, to in patch_data:
|
|
if l == from_:
|
|
l = to
|
|
all_lines.append(l)
|
|
|
|
import tempfile
|
|
temp = tempfile.NamedTemporaryFile(mode="w", delete=False)
|
|
# since we had stripped newlines, need to add them back explicitly
|
|
temp.write("\n".join(all_lines))
|
|
temp.close()
|
|
|
|
import shutil
|
|
shutil.move(temp.name, args.file)
|