这段代码问题出在哪里？

tianer

#! /usr/bin/env python
#coding=utf-8

from cStringIO import StringIO
import re

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

def replace_docroot(conf_string,vhost,new_docroot):
    confile=StringIO(conf_string)
    in_host=False
    curr_vhost=None
    for line in confile:
        vhost_start_match=vhost_start.search(line)
        if vhost_start_match:
            curr_vhost=vhost_start_match.groups()[0] #所以group() 和group(0)返回，匹配的整个表达式的字符串
            in_host=True
        if in_host and (curr_vhost==vhost):
            docroot_match=docroot.search(line)
            if docroot_match:
                sub_line=docroot.sub(r'\1%s' % new_docroot,line)
                line=sub_line
        vhost_end_match=vhost_end.search(line)
        if vhost_end_match:
            in_host=False
        yield line

if __name__=='__main__':
    import sys
    confile=sys.argv[1]
    vhost=sys.argv[2]
    docroot=sys.argv[3]
    conf_string=open(confile).read()
    for line  in replace_docroot(conf_string, vhost, docroot):
        print line,


执行结果是

python 13.py  httpd.conf *:80  /home/html/support.xxx.com

Traceback (most recent call last):
File "13.py", line 36, in ?
    for line  in replace_docroot(conf_string, vhost, docroot):
  File "13.py", line 21, in replace_docroot
    docroot_match=docroot.search(line)
AttributeError: 'str' object has no attribute 'search'

哪位大大能帮我看看

jeppeter

回复 1# tianer

在函数的开始加global docroot
   

tianer

回复 2# jeppeter


    你好，加了还是有报错

#! /usr/bin/env python
#coding=utf-8

from cStringIO import StringIO
import re

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

def replace_docroot(conf_string,vhost,new_docroot):
    confile=StringIO(conf_string)
    in_host=False
    curr_vhost=None
    for line in confile:
        vhost_start_match=vhost_start.search(line)
        if vhost_start_match:
            curr_vhost=vhost_start_match.groups()[0] #所以group() 和group(0)返回，匹配的整个表达式的字符串
            in_host=True
        if in_host and (curr_vhost==vhost):
            docroot_match=docroot.search(line)
            if docroot_match:
                sub_line=docroot.sub(r'\1%s' % new_docroot,line)
                line=sub_line
        vhost_end_match=vhost_end.search(line)
        if vhost_end_match:
            in_host=False
        yield line

if __name__=='__main__':
    import sys
    confile=sys.argv[1]
    vhost=sys.argv[2]
    docroot=sys.argv[3]
    conf_string=open(confile).read()
    for line  in replace_docroot(conf_string, vhost, docroot):
        print line,

python 13.py  httpd.conf *:80  /home/html/support.xxx.com
Traceback (most recent call last):
  File "13.py", line 37, in ?
    for line  in replace_docroot(conf_string, vhost, docroot):
  File "13.py", line 22, in replace_docroot
    docroot_match=docroot.search(line)
AttributeError: 'str' object has no attribute 'search'

ts32767

有两个问题：

1. Docroot定义了两次：

第一次：

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

第二次，main函数中：

if __name__=='__main__':
    import sys
    confile=sys.argv[1]
    vhost=sys.argv[2]
    docroot=sys.argv[3]
    conf_string=open(confile).read()
    for line  in replace_docroot(conf_string, vhost, docroot):
        print line,

从报错的地方看，显然函数 replace_docroot 传入的参数docroot只是个字符串，不是re模块compile出来的正则表达式集合。 所以字符串没有search方法。

你自己看看replace_docroot函数中用的docroot到底应该是哪一个？ 最好的办法，将下面的代码：
vhost_start=re.compile(r'<VirtualHost\s+(.*?)>')
vhost_end=re.compile(r'</VirtualHost>')
docroot=re.compile(r'(DocumentRoot\s+)(\S+)')
直接移到函数replace_docroot中。

tianer

回复 4# ts32767


    果然是这样，谢谢了。终于解决了