如何基于正则表达式替换文件文本中的字符串?

时间:2015-09-24 15:47:46

标签: python regex python-2.7

假设:

self.base_version = 1000
self.target_version = 2000

我有一个文件如下:

some text...
<tsr_args> \"upgrade_test test_mode=upgrade       base_sw=1000      target_sw=2000      system_profile=eth\"</tsr_args>

some text...

<tsr_args> \"upgrade_test test_mode=rollback      base_sw=2000      target_sw=1000      system_profile=eth      manufacture_type=no-manufacture\"</tsr_args>

some text...

<tsr_args> \"upgrade_test test_mode=downgrade     base_sw=2000      target_sw=1000      system_profile=eth      no_boot_next_enable_flag=True\"</tsr_args>

我需要按照上面的指定放置基本和目标版本值(注意,在第2和第3个条目上,基数和目标相反)。

我尝试按照以下方式执行此操作,但它不起作用:

base_regex   = re.compile('.*test_mode.*base_sw=(.*)')
target_regex = re.compile('.*test_mode.*target_sw=(.*)')

o = open(file,'a')

for line in open(file):
    if      'test_mode' in line:
        if     'upgrade' in line:
                new_line = (re.sub(base_regex, self.base_version, line))
                new_line = (re.sub(target_regex, self.target_version, line))
                o.write(new_line)

        elif   'rollback' in line or 'downgrade' in line):
               new_line = (re.sub(base_regex, self.target_version, line))
               new_line = (re.sub(target_regex, self.base_version, line))
               o.write(new_line)
o.close()

假设上述代码正常运行,没有任何语法错误。

  1. 文件根本没有修改。

  2. 修改完整的行而不仅仅是捕获的组。如何使re.sub仅替换捕获的组?

2 个答案:

答案 0 :(得分:2)

  1. 您正在使用a - &gt;打开文件append。因此,您的更改应该在文件末尾。您应该创建一个新文件并在脚本末尾替换old_one。
  2. 如果你想要替换几个匹配的组,我只知道一种方法:首先你使用regexp找到单词并将其替换为没有正则表达式的字符串。

答案 1 :(得分:0)

感谢Jimilan的发言。我修复了我的代码,现在它正在工作:

        base_regex   = re.compile(.*test_mode.*base_sw=(\S*))
        target_regex = re.compile(.*test_mode.*target_sw=(\S*))

        for file in self.upgrade_cases_files_list:

            file_handle = open(file, 'r')
            file_string = file_handle.read()
            file_handle.close()

            base_version_result   = base_regex.search(file_string)
            target_version_result = target_regex.search(file_string)

            if base_version_result is not None:
                current_base_version = base_version_result.group(1)
            else:
                raise Exception("Could not detect base version in the following file: -> %s \n" % (file))

            if target_version_result is not None:
                current_target_version = target_version_result.group(1)
            else:
                raise Exception("Could not detect target version in the following file: -> %s \n" % (file))

            file_string = file_string.replace(current_base_version, self.base_version)
            file_string = file_string.replace(current_target_version, self.target_version)

            file_handle = open(file, 'w')
            file_handle.write(file_string)
            file_handle.close()