Python 用Redis簡單實現(xiàn)分布式爬蟲的方法
Redis通常被認為是一種持久化的存儲器關(guān)鍵字-值型存儲,可以用于幾臺機子之間的數(shù)據(jù)共享平臺。
連接數(shù)據(jù)庫
注意:假設(shè)現(xiàn)有幾臺在同一局域網(wǎng)內(nèi)的機器分別為Master和幾個Slaver
Master連接時host為localhost即本機的ip
_db = redis.Reds(host='localhost', port=6379, db=0)
Slaver連接時的host也為Master的ip,端口port和數(shù)據(jù)庫db不寫時為默認值6379、0
_db = redis.Redis(host='192.168.235.80')
Redis含列表、集合,字符串等幾種數(shù)據(jù)結(jié)構(gòu),具體詳細的幾種數(shù)據(jù)結(jié)構(gòu)的操作命令可查看Redis官網(wǎng)http://redis.io/commands,下面是對集合進行操作的一些命令
_db.sadd(key, *values) # 插入指定values到集合中并返回新插入的的值的數(shù)量 _db.spop(self.key) # 隨機從集合中的得到一個元素將其從集合中刪除并作為返回值返回 _db.smembers(self.key) # 返回集合的所有元素 _db.delete(key) # 刪除整個集合, 成功返回1, 失敗返回0 _db.srandmember(self.key) # 從集合中隨機選取一個元素作為返回 _db.sismember(self.key, value) # 判斷指定value是否在本集合中,成功返回1,失敗返回0
例如,Master可利用redis將url上傳到數(shù)據(jù)庫
for i in range(20): # 將需爬取的糗事百科前20頁的url并存入urls集合
url = 'http://www.qiushibaike.com/hot/page/%d/' % (i + 1)
_db.sadd('urls', url) # 將url插入關(guān)鍵字urls集合中,若url已存在則不再插入
進而Master和Slaver可從數(shù)據(jù)庫里獲取url
url = _db.spop('urls') # 隨機從urls集合中取出一個url返回并將其刪去
同理,Master可利用上面的方法將其他一些共用的資源上傳到數(shù)據(jù)庫,Slaver就可從數(shù)據(jù)庫下載需要的東西。下面是一個完整的例子
# coding=utf-8
import urllib2
import re
import time
import redis
headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Ubuntu Chromium/49.0.2623.108 Chrome/49.0.2623.108 Safari/537.36'}
job_redis = redis.Redis(host='192.168.235.80') # host為主機的IP,port和db為默認值
class Clawer(object):
identity = 'master' # 或slaver
def __init__(self):
if self.identity == 'master':
for i in range(20): # 將需爬取的糗事百科前20頁的url并存入urls集合
url = 'http://www.qiushibaike.com/hot/page/%d/' % (i + 1)
job_redis.sadd('urls', url)
self.main()
def get_content(self):
"""
從糗事百科中獲取故事
:return: 故事列表
"""
stories = []
content_pattern = re.compile('<div class="content">([\w\W]*?)</div>([\w\W]*?)class="stats"') # 匹配故事內(nèi)容(第一空)和是否含有圖片(第二空)的模板
pattern = re.compile('<.*?>') # 匹配包括括號及括號內(nèi)無關(guān)內(nèi)容的模板
url = job_redis.spop('urls')
while url: # 當數(shù)據(jù)庫還存在網(wǎng)頁url,取出一個并爬取
try:
request = urllib2.Request(url, headers=headers)
response = urllib2.urlopen(request)
text = response.read()
except urllib2.URLError, e: # 若出現(xiàn)網(wǎng)頁讀取錯誤捕獲并輸出
if hasattr(e, "reason"):
print e.reason
content = re.findall(content_pattern, text) # 獲取含模板內(nèi)容的列表
for x in content:
if "img" not in x[1]: # 過濾含圖片的故事
x = re.sub(pattern, '', x[0])
x = re.sub('\n', '', x)
stories.append(x)
url = job_redis.spop('urls')
time.sleep(3)
return stories
def main(self):
self.get_content()
if __name__ == '__main__':
Clawer()
將此代碼在幾臺機器同時運行,調(diào)節(jié)其身份為master或slaver,做到簡單的分布式爬蟲
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。

