#!/usr/bin/python3
import re
import sys
#import pdb
def main():
#pdb.set_trace()
pattern = re.compile(r"\s*((ALTER_ME_IN_THE_NEW_FILE.*),.*\);)");
replacement = r"#ifdef F1\n" \
"\1\n" \
"#else\n" \
"\2);\n" \
"#endif"
for num_file, file_name in enumerate(sys.argv[1:], 1):
print("Starting the change for file: %s" % file_name)
with open(file_name, "r+") as fhandle:
orig = fhandle.read()
repl, changed = re.subn(pattern, replacement, orig)
if changed:
fhandle.seek(0)
fhandle.write(repl)
print("Finished the change for file: %s" % file_name)
print("DONE for %d number of files" % num_file)
if __name__ == '__main__':
main()
Comments