Question

我正在尝试使用Python修改配置文件。如何以这种格式执行等效的多个sed命令：

sed -ci 's/ServerTokens OS/ServerTokens Prod/' /etc/httpd/conf/httpd.conf

在Python中效率最高？这就是我现在正在做的事情：

with open("httpd.conf", "r+") as file:
    tmp = []
    for i in file:
        if '#ServerName' in i:
            tmp.append(i.replace('#ServerName www.example.com', 'ServerName %s' % server_name , 1))
        elif 'ServerAdmin' in i:
            tmp.append(i.replace('root@localhost', webmaster_email, 1))
        elif 'ServerTokens' in i:
            tmp.append(i.replace('OS', 'Prod', 1))
        elif 'ServerSignature' in i:
            tmp.append(i.replace('On', 'Off', 1))
        elif 'KeepAlive' in i:
            tmp.append(i.replace('Off', 'On', 1))
        elif 'Options' in i:
            tmp.append(i.replace('Indexes FollowSymLinks', 'FollowSymLinks', 1))
        elif 'DirectoryIndex' in i:
            tmp.append(i.replace('index.html index.html.var', 'index.php index.html', 1))
        else:
            tmp.append(i)
    file.seek(0)
    for i in tmp:
        file.write(i)

它不必要地复杂，因为我可以使用subprocess和sed代替。有什么建议？

Answer 1

您可以在Python中使用正则表达式，就像在sed中执行此操作一样。只需使用Python regular expressions library即可。您可能对re.sub()方法感兴趣，它与您的示例中使用的sed的s命令相同。

如果您想有效地执行此操作，您可能必须每行只运行一个替换命令，如果它已更改则跳过它，与您在示例代码中执行此操作的方式类似。为了实现这一目标，您可以使用re.subn代替re.sub或re.match以及匹配的组。

以下是一个例子：

import re

server_name = 'blah'
webmaster_email = 'blah@blah.com'

SUBS = ( (r'^#ServerName www.example.com', 'ServerName %s' % server_name),
        (r'^ServerAdmin root@localhost', 'ServerAdmin %s' % webmaster_email),
        (r'KeepAlive On', 'KeepAlive Off')
       )

with open("httpd.conf", "r+") as file:
    tmp=[]
    for i in file:
        for s in SUBS:
            ret=re.subn(s[0], s[1], i)
            if ret[1]>0:
                tmp.append(ret[0])
                break
        else:
            tmp.append(i)
    for i in tmp:
        print i,

如何使用Python修改配置文件

1 个答案: