#!/usr/bin/python3 # written by andrewtunsw.edu.au as a COMP(2041|9044) lecture example # For each file given as argument replace occurrences of Hermione # allowing for some misspellings with Harry and vice-versa. # Relies on Zaphod not occurring in the text. # modified text is stored in a list then file over-written import re, sys, os for filename in sys.argv[1:]: changed_lines = [] with open(filename) as f: for line in f: changed_line = re.sub(r"Herm[io]+ne", "Zaphod", line) changed_line = changed_line.replace("Harry", "Hermione") changed_line = changed_line.replace("Zaphod", "Harry") changed_lines.append(changed_line) with open(filename, "w") as g: g.write("".join(changed_lines))