最近想用Python爬虫搞搞百度贴吧的操作，所以我得把原来申请的小号找出来用。有一个小号我忘了具体ID，只记得其中几个字母以及某个加入的贴吧。所以今天就用爬虫来获取C语言贴吧的所有成员。

计划很简单，爬百度贴吧的会员页面，把结果存到MySQL数据库中，等到所有会员都爬完之后。我就可以使用简单的SQL语句查询账号名了。由于C语言贴吧会员有50多万，所以我还需要在合适的时候(例如插入数据库失败)把错误信息打印到日志文件中。由于我是Python新手，所以就不弄什么多线程得了，直接一个脚本用到黑。

看着很简单，实际也很简单。写完了我看了一下，用到的知识只有最基础的SQL操作、BeautifulSoup解析。

首先第一步就是看一下这个吧的信息页有多少页，关键代码如下。踩了两天坑，总算感觉对BeautifulSoup熟悉了一点。代码也很简单，按照class名查找到总页数这个标签，然后用正则表达式匹配到页数数字。这里要说一下，正则表达式的分组真好用。以前偷懒只学了一点正则表达式，发现没啥作用，只有配合分组才能比较精确的查找字符。

html = request.urlopen(base_url).read().decode(encoding)

soup = BeautifulSoup(html, 'lxml')

page_span = soup.find('span', class_='tbui_total_page')

p = re.compile(r'共(\d+)页')

result = p.match(page_span.string)

global total_pages

total_pages = int(result.group(1))

logger.info(f'会员共{total_pages}页')

有了总页数，我们就可以遍历页面了，代码如下。写的虽然比较脏，但是能用就行了，大家嫌难看就难看吧。这里做的事情就很简单了，从第一页开始遍历，一直遍历到最后一页。把每一页的用户名字提取出来，然后用_insert_table(connection, name)函数存到MySQL中。

因为我为了省事，直接把百度用户名当做主键了。但是保不齐贴吧有什么bug，导致用户名重复之类的问题，导致插入失败。所以我用try把保存这一块包起来。有异常的话就打印到日志中，方便排查。日志分成两种级别的，INFO级别输出到控制台，ERROR级别输出到文件。

def _find_all_users():

global connection

for i in range(start_page, total_pages + 1):

target_url = f'{base_url}&pn={i}'

logger.info(f'正在分析第{i}页')

html = request.urlopen(target_url).read().decode(encoding)

soup = BeautifulSoup(html, 'lxml')

outer_div = soup.find('div', class_='forum_info_section member_wrap clearfix bawu-info')

inner_spans = outer_div.find_all('span', class_='member')

for index, span in enumerate(inner_spans):

name_link = span.find('a', class_='user_name')

name = name_link.string

logger.info(f'已找到 {name}')

try:

_insert_table(connection, name)

except:

logger.error(f'第{i}页{index}第个用户 {name} 发生异常')

完整的代码见下。

"""

Python写的百度贴吧工具

"""

import pymysql

host = 'localhost'

db_name = 'tieba'

username = 'root'

password = '12345678'

def _get_connection(host, username, password, db_name):

return pymysql.connect(host=host,

user=username,

password=password,

charset='utf8mb4',

db=db_name)

def _create_table(connection):

create_table_sql = """

CREATE TABLE tieba_member(

username CHAR(255) PRIMARY KEY

)

"""

with connection.cursor() as cursor:

cursor.execute(create_table_sql)

connection.commit()

def _insert_table(connection, username):

insert_table_sql = """

INSERT INTO tieba_member

VALUES(%s)"""

with connection.cursor() as cursor:

cursor.execute(insert_table_sql, (username,))

connection.commit()

import urllib.request as request

from bs4 import BeautifulSoup

import re

import tieba.log_config

import logging

logger = logging.getLogger()

encoding = 'GBK'

base_url = 'http://tieba.baidu.com/bawu2/platform/listMemberInfo?word=c%D3%EF%D1%D4'

# base_url = 'http://tieba.baidu.com/bawu2/platform/listMemberInfo?word=%B9%FD%C1%CB%BC%B4%CA%C7%BF%CD'

start_page = 1

total_pages = None

connection = _get_connection(host, username, password, db_name)

def _get_total_pages():

html = request.urlopen(base_url).read().decode(encoding)

soup = BeautifulSoup(html, 'lxml')

page_span = soup.find('span', class_='tbui_total_page')

p = re.compile(r'共(\d+)页')

result = p.match(page_span.string)

global total_pages

total_pages = int(result.group(1))

logger.info(f'会员共{total_pages}页')

def _find_all_users():

global connection

for i in range(start_page, total_pages + 1):

target_url = f'{base_url}&pn={i}'

logger.info(f'正在分析第{i}页')

html = request.urlopen(target_url).read().decode(encoding)

soup = BeautifulSoup(html, 'lxml')

outer_div = soup.find('div', class_='forum_info_section member_wrap clearfix bawu-info')

inner_spans = outer_div.find_all('span', class_='member')

for index, span in enumerate(inner_spans):

name_link = span.find('a', class_='user_name')

name = name_link.string

logger.info(f'已找到 {name}')

try:

_insert_table(connection, name)

except:

logger.error(f'第{i}页{index}第个用户 {name} 发生异常')

import datetime

if __name__ == '__main__':

_get_total_pages()

_find_all_users()

还有另一个文件用来配置日志的。你也可以把这两个文件合在一起，只不过看着可能更乱了。

import logging

# 创建Logger

logger = logging.getLogger()

logger.setLevel(logging.DEBUG)

# 创建Handler

# 终端Handler

consoleHandler = logging.StreamHandler()

consoleHandler.setLevel(logging.DEBUG)

# 文件Handler

fileHandler = logging.FileHandler('log.log', mode='a', encoding='UTF-8')

fileHandler.setLevel(logging.ERROR)

# Formatter

formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

consoleHandler.setFormatter(formatter)

fileHandler.setFormatter(formatter)

# 添加到Logger中

logger.addHandler(consoleHandler)

logger.addHandler(fileHandler)

性能测试

当然由于要爬的数据量比较大，我们还要计算一下可能的运行时间。首先不考虑爬虫被百度封了的情况。我把代码稍作修改，设定只爬前100页。

import datetime

if __name__ == '__main__':

# _get_total_pages()

total_pages = 100

time1 = datetime.datetime.today()

_find_all_users()

time2 = datetime.datetime.today()

print(time2)

print(time1)

print(time2 - time1)

结果如下，用时将近两分钟。做了简单计算得出结论，要爬完c语言贴吧的52万个会员，需要将近7个小时。所以程序还需要改进。

2017-04-04 23:57:59.197993

2017-04-04 23:56:10.064666

0:01:49.133327

首先先从数据库方面考虑一下。Windows下MySQL默认的数据库引擎是Innodb，特点是支持事务管理、外键、行级锁，但是相应的速度比较慢。我把表重新建为MyISAM类型的。然后重新运行一下测试，看看这次速度会不会有变化。

CREATE TABLE tieba_member (

username CHAR(255) PRIMARY KEY

)

ENGINE = MyISAM

这次性能提升的有点快，速度足足提高了76%。可见默认的并不一定是最好的。

2017-04-05 00:15:19.989766

2017-04-05 00:14:53.407476

0:00:26.582290

既然都开始测试了，不妨干脆点。MySQL还有一种引擎是Memory，直接把数据放到内存中。速度肯定会更快！不过测试结果很遗憾，还是26秒。可见数据库这方面的优化到头了。

CREATE TABLE tieba_member (

username CHAR(255) PRIMARY KEY

)

ENGINE = MEMORY

不过性能确实提高了很多。经过计算，这次只需要一个半小时即可爬完52万个用户。如果在开多个进程，相信速度还会更快。所以这篇文章就差不多完成了。等明天爬完之后，我把结果更新一下，任务就真正完成了！

不过结果很遗憾，爬虫失败了。为了速度更快我开了4个进程，分别爬1-5000页，5001-10000页，10001-15000页，以及15000-到最后4部分。

但是日志输出显示出现很多重复的用户名，5000页之后的用户名竟然和第一页相同。我百思不得其解，在使用浏览器测试发现，不知道是百度的防爬虫机制还是bug之类的，浏览器只能显示到450多页，在往后就会显示为空页面，如果页数更大，就一直返回第一页的内容。因此依赖于这个页面的贴吧爬虫宣布失败。

虽然失败了，但是还是学习到了不少经验。我测试了一下爬前450页，仅用时44秒。说明爬虫速度倒是还星还行。

import datetime

from multiprocessing import Process

if __name__ == '__main__':

total_pages = _get_total_pages()

processes = []

processes.append(Process(target=_find_all_users, args=(1, 150)))

processes.append(Process(target=_find_all_users, args=(151, 300)))

processes.append(Process(target=_find_all_users, args=(301, 450)))

time1 = datetime.datetime.today()

for process in processes:

process.start()

for process in processes:

process.join()

time2 = datetime.datetime.today()

print(f'开始时间{time1}')

print(f'结束时间{time2}')

print(f'用时{time2 - time1}')