Remove all comments from a .py file

clean.py

# use python clean.py -f <filename>
# not meant to be very smart (can't distinguish "a = "#test"" for example...)
import os
import argparse
import re


parser = argparse.ArgumentParser()
parser.add_argument('-f', '--file', type=str)
args = parser.parse_args()

with open(args.file, 'r') as f:
    lines = f.readlines()
lines = ''.join(lines)

lines = re.sub(r'\"\"\"[^\"]*\"\"\"', '', lines)
lines = re.sub(r'#.*', '', lines)


with open(args.file, 'w') as f:
    f.write(lines)