python - 将正则表达式与文件输入一起使用

Question

我正在尝试使用正则表达式替换存储在另一个文件中的变量。我试过的代码是:

r = re.compile(r"self\.uid\s*=\s*('\w{12})'")
for line in fileinput.input(['file.py'], inplace=True): 
    print line.replace(r.match(line), sys.argv[1]), 

文件中变量的格式为:

self.uid = '027FC8EBC2D1'

我试图以这种格式传入一个参数，并使用正则表达式来验证 sys.argv[1] 的格式是否正确，并找到存储在此文件中的变量并替换它使用新变量。

谁能帮忙。感谢您的帮助。

Answer 1

您可以使用 re.sub 来匹配正则表达式并一次性完成替换:

r = re.compile(r"(self\.uid\s*=\s*)'\w{12}'")
for line in fileinput.input(['file.py'], inplace=True):
    print r.sub(r"\1'%s'" %sys.argv[1],line),