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

Question

假设：

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>

我需要将基本版本和目标版本值按上述指定放置（请注意，在第二个和第三个条目上，基本版本和目标版本是相反的）。

我尝试按照以下步骤进行操作，但是它不起作用：

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仅替换捕获的组？

Answer 1

1. 您正在使用打开文件a > - append 。 因此，您的更改应在文件末尾。 您应该创建一个新文件，并在脚本末尾替换old_one。
2. 我只有一种方法知道是否要替换几个匹配的组：首先，您使用regexp查找单词并将其替换为不带regexp的字符串。

Answer 2

感谢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()