如何用Python修改指定define service段内的contact_groups参数值

需求说明 #

批量处理多台主机的Nagios配置文件，仅定位到service_description为NRPE - NTP_TIME的define service段，将该段内contact_groups字段中的ops1替换为ops2，避免重复手动修改。

配置文件示例 #

define service{
        use                             generic-service

        service_description             NRPE
        is_volatile                     
        check_period                    
        max_check_attempts              
        check_interval           
        retry_interval            
        contact_groups                  xyz,xyz,ops1,xyz

        notification_period             24x7
        check_command                   xyz
        }
define service{
        use                             generic-service

        service_description             NRPE
        is_volatile                     
        check_period                    
        max_check_attempts              
        check_interval           
        retry_interval            
        contact_groups                  xyz,xyz,ops1,xyz

        notification_period             24x7
        check_command                   xyz
define service{
        use                             generic-service

        service_description             NRPE - NTP_TIME
        is_volatile                     
        check_period                    
        max_check_attempts              
        check_interval           
        retry_interval            
        contact_groups                  xyz,xyz,ops1,xyz

        notification_period             24x7
        check_command                   xyz

原代码问题分析 #

原代码的正则逻辑存在两处关键错误：

1. 使用贪婪模式(\n.*)*匹配内容，会跨多个define service块，无法精准锁定目标服务段内的contact_groups。
2. 替换字符串中直接写入正则表达式语法（如service_description.*NTP_TIME），而非实际需要的配置文本，导致替换后内容出现无效字符串。

正确实现方案 #

方法1：回调函数精准替换 #

通过正则匹配所有define service块，对每个块单独判断是否为目标段，再执行替换：

import re

def replace_target_contact_group(match):
    # 获取当前service块的内容
    block_content = match.group(1)
    # 检查是否是目标服务段
    if 'service_description             NRPE - NTP_TIME' in block_content:
        # 替换该块内contact_groups中的ops1为ops2
        updated_block = re.sub(r'(contact_groups\s+.*?)ops1(.*)', r'\1ops2\2', block_content)
        return f'define service{{{updated_block}}}'
    # 非目标段直接返回原内容
    return match.group(0)

# 读取配置文件内容
with open("plik.txt", "r") as file:
    contents = file.read()

# 执行替换：匹配所有define service块，使用DOTALL让.匹配换行符
updated_contents = re.sub(r'define service\{(.*?)\}', replace_target_contact_group, contents, flags=re.DOTALL)

# 写入修改后的内容
with open("plik.txt", "w") as file:
    file.write(updated_contents)

方法2：分块遍历处理 #

先拆分所有define service块，逐个处理后重新拼接：

import re

# 读取配置文件
with open("plik.txt", "r") as file:
    contents = file.read()

# 提取所有define service块
service_blocks = re.findall(r'define service\{(.*?)\}', contents, re.DOTALL)
updated_blocks = []

for block in service_blocks:
    if 'service_description             NRPE - NTP_TIME' in block:
        # 替换目标块内的ops1
        updated_block = re.sub(r'contact_groups\s+(.*?)ops1(.*?)', r'contact_groups \1ops2\2', block)
        updated_blocks.append(f'define service{{{updated_block}}}')
    else:
        updated_blocks.append(f'define service{{{block}}}')

# 拼接修改后的内容并写入
with open("plik.txt", "w") as file:
    file.write('\n'.join(updated_blocks))

代码说明 #

• 使用re.DOTALL标志让正则表达式中的.匹配换行符，确保完整捕获每个define service块的内容。
• 仅对包含service_description NRPE - NTP_TIME的块执行替换操作，不影响其他服务段。
• 替换contact_groups时，精准匹配字段后的内容，只替换ops1为ops2，保留其他组名不变。

内容的提问来源于stack exchange，提问作者Linuxuser