python进行文档抽取与解析的简单实现

一、前文

之前被叫去做网络爬虫，爬取新浪新闻的url，标题，内容和评论，不过在需求上有点改变，主要是评论的间隔被要求有‘\t’的分割，比如将

<comment>
2014-12-10 18:53:20    1004400533   遗弃亲生骨肉猪狗不如，难道就不怕受到良心谴责？你能睡得安稳？
2014-12-10 17:17:07    3294923134   这父亲是人吗？
</comment>

改为

<comment>
2014-12-10 18:53:20 1004400533遗弃亲生骨肉猪狗不如，难道就不怕受到良心谴责？你能睡得安稳？
2014-12-10 17:17:07 3294923134这父亲是人吗？
</comment>

试过重新爬取，不过新浪在2015年后，做了些改变，一是评论api下的内容有了些修改，从原本的中文字符变为了'\u'，解决办法可以参考另一篇博文三张图告诉你python爬虫时转换\u中文字符的“坑”，二是在爬虫上需要做好浏览器模拟的工作，不然很容易被封杀，而且，在做大规模爬取的时候，需要利用多线程或者分布式的手段进行，多线程稍不注意，到了系统线程的上限，这是要注意的。

二、抽取和解析

因为我想爬取这个页面（social）一年的内容，已经超过2万了（只算那些有评论的新闻），所以经常第二天回来发现中途error。分布式计算的东西现在还没去学习，所以还是复习一下python。

在这个简短的程序中，用到了SGMLParser类进行解析，这个类封装了可以实现对具有标签的字符串的处理函数，当然没有标签也是可以用的，不需要冠以html文本种类的要求，网上很多例子可能会拿HTMLParser与SGMLParser进行说明，通过解析X.html进行说明。这里我只采用单线程的方法。

SGMLParser:

1、通过利用开关变量的方法进行标签间内容的抽取和解析，在此之前，需要进行变量的初始化，可以利用__init__()或者__reset__()，一般用__reset__()。

2、start_x(),end_x(),handle_data()，这三个函数，第一个是当读取到<x>时，执行的函数，第一个是当读取到</x>时，执行的函数,第三个是当遇到标签内的内容，就会调用这个函数，更详细一点可以参考这篇博文

变量说明：

rootdir：原始文件目录

rootdir：需要存放转换好的文件目录

方法说明：

1、通过SGMLParser读取<comment></comment>中的内容，调用solve()函数，利用正则匹配的方法将原来不符合规范的空格替换为'\t'，返回需要替换的内容到comment变量

2、利用正则匹配找到定位到<comment></comment>，替换为Parse类对象的comment值

（正则表达式的方法参考以下注释）

完整代码：

#!/usr/bin/env python# -*- coding: utf-8 -*-from sgmllib import SGMLParserimport re,osrootdir = 'F:\\Python27\\pythonproject\\fuck\\file1\\'rootdir2 = 'F:\\Python27\\pythonproject\\fuck\\file2\\'class Parse(SGMLParser):    def __init__(self,filename):        self.filename = filename          self.comment = ''         SGMLParser.__init__(self)           def reset(self):        self.found_comment = False        SGMLParser.reset(self)    def start_comment(self, attrs):        self.found_comment = True    def end_comment(self):        self.found_comment = False    def handle_data(self, text):        if self.found_comment == True:            self.comment=solve(self.filename,text)def solve(filename,text):       #找到3个或3个以上空格的地方，替换为'\t'    strinfo = re.compile('(\s\s\s*)')    str_result = strinfo.sub('\t',text)    return str_result    if __name__ == '__main__':            global rootdir,rootdir2    for parent,dirnames,filenames in os.walk(rootdir):    #三个参数：分别返回1.父目录 2.所有文件夹名字（不含路径） 3.所有文件名字        for filename in filenames:            #print filename                        filesource = open(rootdir+filename, 'a+')                s = filesource.read()                        filesource.close()            #初始化实例p,feed()函数为添加处理字符串            p = Parse(filename)                        p.feed(s)                        filedes = open(rootdir2+filename, 'a+')            #因为原来原数据中存在不以</comment>结尾的内容，因此需要先做处理            pattern1='</comment>'            if re.findall(pattern1,s) == []:                s=s+'\n</comment>'            <span style="white-space:pre"></span>    #定位到comment标签位置，([\s\S]*)能够匹配所有字符，包括换行符，然后替换            strinfo = re.compile('<comment>([\s\S]*)</comment>')            str_result = strinfo.sub('<comment>'+p.comment+'</comment>',s)                        filedes.write(str_result)            filedes.close()

结果：

注意：win7系统一般文件夹下的文件总数（包括文件夹）不能超过21845个，不然会提示错误。