Advertisement
homer512

py re whole file

Apr 21st, 2015
236
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. #!/usr/bin/python3
  2.  
  3. import re
  4. import sys
  5. #import pdb
  6.  
  7.  
  8. def main():
  9.     #pdb.set_trace()
  10.     pattern = re.compile(r"\s*((ALTER_ME_IN_THE_NEW_FILE.*),.*\);)");
  11.     replacement = r"#ifdef F1\n" \
  12.                   "\1\n" \
  13.                   "#else\n" \
  14.                   "\2);\n" \
  15.                   "#endif"
  16.     for num_file, file_name in enumerate(sys.argv[1:], 1):
  17.         print("Starting the change for file: %s" % file_name)
  18.         with open(file_name, "r+") as fhandle:
  19.             orig = fhandle.read()
  20.             repl, changed = re.subn(pattern, replacement, orig)
  21.             if changed:
  22.                 fhandle.seek(0)
  23.                 fhandle.write(repl)
  24.         print("Finished the change for file: %s" % file_name)
  25.         print("DONE for %d number of files" % num_file)
  26.  
  27.  
  28. if __name__ == '__main__':
  29.     main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement