這段代碼問題出在哪里？

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)返回，匹配的整個(gè)表達(dá)式的字符串
            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,


執(zhí)行結(jié)果是

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

回復(fù) 1# tianer

在函數(shù)的開始加global docroot
   

tianer

回復(fù) 2# jeppeter


    你好，加了還是有報(bào)錯(cuò)

#! /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)返回，匹配的整個(gè)表達(dá)式的字符串
            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

有兩個(gè)問題：

1. Docroot定義了兩次：

第一次：

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

第二次，main函數(shù)中：

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,

從報(bào)錯(cuò)的地方看，顯然函數(shù) replace_docroot 傳入的參數(shù)docroot只是個(gè)字符串，不是re模塊compile出來的正則表達(dá)式集合。 所以字符串沒有search方法。

你自己看看replace_docroot函數(shù)中用的docroot到底應(yīng)該是哪一個(gè)？ 最好的辦法，將下面的代碼：
vhost_start=re.compile(r'<VirtualHost\s+(.*?)>')
vhost_end=re.compile(r'</VirtualHost>')
docroot=re.compile(r'(DocumentRoot\s+)(\S+)')
直接移到函數(shù)replace_docroot中。

tianer

回復(fù) 4# ts32767


    果然是這樣，謝謝了。終于解決了