使用更新的功能搜索和替换现有的Python脚本

Question

我需要编写一个python脚本来更新另一个python脚本，以便将较旧的代码替换为较新的函数。基本上，只要看到"模型... read((或模型...写(x(之类的东西，它应该用更新的字符串替换它们。并替换为" maport.read(model.cb1_ov_en(.value"。有很多变量可以更新，并且手工做到这一点会很乏味，所以我很好奇是否有一个很好的方法可以自动化。我有字符串替换功能完成了，但我不确定如何自动化.py文件中的变量。

def main():
    textfile = "C:\Desktop\Trip_Logic.py"
    with open(textfile, 'r') as file :
        filedata = file.read()
    readString     = "Model....Read()"
    if readString in filedata:
        print readString[2:7]
    filedata = filedata.replace(readString, convertReadHILAPI(readString))
    with open(textfile, 'w') as file:
        file.write(filedata)
    #filedata = filedata.replace(string, convertWriteHILAPI(writeString))
    ##############################################################
    #Example:
    readString1     = "Model.CB1_OV_En.Read()"
    newReadString1  = "MAPort.Read(Model.CB1_OV_En).Value"
    readString2     = "Model.CB2_OV_En.Read()"
    newReadString2  = "MAPort.Read(Model.CB2_OV_En).Value"
    ##############################################################
    writeString1    = "Model.CB1_OV_En.Write(1)"
    newWriteString1 = "MAPort.Write(Model.CB1_OV_En, Float(1))"
    writeString2    = "Model.CB2_OV_En.Write(1)"
    newWriteString2 = "MAPort.Write(Model.CB2_OV_En, Float(1))"
    ##############################################################
def convertReadHILAPI(string):
    variableStartIndex = 6
    variableStopIndex  = string.find(".Read")
    varName = string[variableStartIndex:variableStopIndex]
    cmd = 'newVarName = "MAPort.Read(Model.%s).Value" % (varName)'
    exec cmd
    return newVarName
def convertWriteHILAPI(string):
    variableStartIndex = 6
    variableStopIndex  = string.find(".Write")
    valueStartIndex = string.find("(")
    valueStopIndex = string.find(")")
    varName = string[variableStartIndex:variableStopIndex]
    value = string[valueStartIndex+1:valueStopIndex]
    cmd = 'newVarName = "MAPort.Write(Model.%s, Float(%s)) " % (varName, value)'
    exec cmd
    return newVarName
if __name__ == '__main__':
    main()

Answer 1

您可以使用RE模块。这是实现您想要的东西的更多Pythonic方法：

import re

def main():
    fname = r'C:DesktopTrip_Logic.py'
    with open(fname) as reader:
        data = reader.read()
    # Model.XXX.Read() => MAPort.Read(Model.XXX).Value
    data = re.sub(r'(Model.(.+).Read())',
                  r'MAPort.Read(Model.2).Value',
                  data)
    # Model.XXX.Write(N) => MAPort.Write(Model.XXX, Float(N))
    data = re.sub(r'(Model.(.+).Write((d+)))',
                  r'MAPort.Write(Model.2, Float(3))',
                  data)
    with open(fname, 'w') as writer:
        writer.write(data)