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

Python
Python, Spider

一、前文

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

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

改为

1
2
3
4
5
6
<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()，这三个函数，第一个是当读取到;时，执行的函数，第一个是当读取到;时，执行的函数,第三个是当遇到标签内的内容，就会调用这个函数，更详细一点可以参考
这篇博文

变量说明：

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

方法说明：

1、通过SGMLParser读取;;中的内容，调用solve()函数，利用正则匹配的方法将原来不符合规范的空格替换为’\t’，返回需要替换的内容到comment变量
2、利用正则匹配找到定位到;;，替换为Parse类对象的comment值

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

完整代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from sgmllib import SGMLParser
import re,os
rootdir = '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()

结果：

image_mark

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

Comments