Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/python3
- import re
- import sys
- #import pdb
- def main():
- #pdb.set_trace()
- for num_file, file_name in enumerate(sys.argv[1:], 1):
- print("Starting the change for file: %s" % file_name)
- line_list = []
- with open(file_name, "r+") as fhandle:
- for line in fhandle:
- if "ALTER_ME_IN_THE NEW_FILE" in line:
- new_line_to_insert = re.sub(",.*", ");", line)
- newlines = (
- "#ifdef F1\n",
- line,
- "#else\n",
- new_line_to_insert,
- "#endif\n",
- )
- line_list.extend(newlines)
- else:
- line_list.append(line)
- fhandle.seek(0)
- fhandle.writelines(line_list)
- print("Finished the change for file: %s" % file_name)
- print("DONE for %d number of files" % num_file)
- if __name__ == '__main__':
- main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement