Python脚本--Apache配置文件

通过python脚本来修改apache的配置文件：

该脚本可以解析apache配置文件，找到VirtualHost部分，替换DocumentRoot。

#!/usr/bin/python

from cStringIO import StringIO
import re

vhost_start = re.compile(r‘<VirtualHost\s+(.*?)>‘)
vhost_end = re.compile(r‘</VirtualHost>‘)
docroot_re = re.compile(r‘<DocumentRoot\s+)(\S+)>‘)

def replace_docroot(conf_string, vhost, new_docroot):
  ‘‘‘yield new lines of an httpd.conf file where docroot lines matching
      the specified vhost are replaced with the new_docroot
  ‘‘‘
  conf_file = StringIO(conf_string)
  in_vhost = Flase
  curr_vhost = None
  for line in conf_file:
    vhost_start_match = vhost_start.search(line)
    if vhost_start_match:
      curr_vhost = vhost_start_match.groups()[0]
      in_vhost = True
    if in_vhost and (curr_vhost == vhost):
      docroot_match = docroot_re.search(line)
      if docroot_match:
        sub_line = docroot_re.sub(r‘\1%s‘ % new_docroot, line)
        line = sub_line
      vhost_end_match = vhost_end.search(line)
      if vhost_end_match:
        in_vhost = False
      yield line
  
  if __name__ = ‘__main__‘:
    import sys
    conf_file = sys.argv[1]
    vhost = sys.argv[2]
    docroot = sys.argv[3]
    conf_string = open(conf_file).read()
    for line in replace_docroot(conf_string, vhost, docroot)
      print line,

Python脚本--Apache配置文件