用python自动去除python中的注释

二话不说，上代码

import sys
path=sys.argv
with open(path[1],"r") as f:
    list0=f.readlines()
for i in range(len(list0)):
    j=list0[i].find("#")
    if j!=-1:
        list0[i]=list0[i][0:j]
    

text=" ".join(list0+[" "])
result=text[0:3]
flag=1


for i in range(3,len(text)):
    
    if text[i-3] != "=" and text[i-2:i+1]=="\"\"\"":
        flag=not flag
        if not flag:
            result=result[0:-2]
            #print(flag,result)
        continue
    if flag:
        result+=text[i]
result=result.replace("\n ","\n")   
with open("result.py","w") as f:
    f.write(result)

将从命令行输入的文件去掉#注释和"""注释，保存在result.py中