python爬虫爬百度云盘的资源



最近百度云盘不知道为啥不提供资源检索，正好最近看了一下python，正好来练练手，写歌爬虫爬一下百度云盘的资源。

分析了一下百度云盘的网友源码和js文件，里面有大量ajax的东西，利用json传输数据，前端显示。话说，这样数据爬去就方便多了，也不要用scrapy啥的，直接解析json数据就好。

分析js文件提炼了下面三个链接：

URL_SHARE = 'http://yun.baidu.com/pcloud/feed/getsharelist?auth_type=1&start={start}&limit=20&query_uk={uk}&urlid={id}'URL_FOLLOW = 'http://yun.baidu.com/pcloud/friend/getfollowlist?query_uk={uk}&limit=20&start={start}&urlid={id}'#http://yun.baidu.com/pcloud/friend/getfanslist?query_uk=1327787586&limit=25&start=0URL_FANS = 'http://yun.baidu.com/pcloud/friend/getfanslist?query_uk={uk}&limit=20&start={start}&urlid={id}'

整个数据爬取流程起到很重要的作用。

爬虫分三步，一个是urlids 保存要爬取的网址，一个是user存放用户uk，另一个是share存放user分享的数据，包含任何你想要的数据。

下面提供三个核心函数代码：

def response_worker():global news,totalsdbconn = mdb.connect(DB_HOST, DB_USER, DB_PASS, 'baiduyun', charset='utf8')dbcurr = dbconn.cursor()dbcurr.execute('SET NAMES utf8')dbcurr.execute('set global wait_timeout=60000')while True:print "function response_worker",hc_r.qsize()# if hc_r.qsize()==0:# print "continue"# continuemetadata, effective_url = hc_r.get()print "response_worker:", effective_urltry:tnow = datetime.datetime.utcnow()date = (tnow + datetime.timedelta(hours=8))date = datetime.datetime(date.year, date.month, date.day)if news>=100:try:dbcurr.execute('INSERT INTO spider_statusreport(date,new_hashes,total_requests)  VALUES(%s,%s,%s) ON DUPLICATE KEY UPDATE ' +'total_requests=total_requests+%s,new_hashes=new_hashes+%s',(date, news,totals,totals,news))except Exception as ex:print "E10", str(ex)news=0id = re_urlid.findall(effective_url)[0]start = re_start.findall(effective_url)[0]if True:if 'getfollowlist' in effective_url: #type = 1follows = json.loads(metadata)print "-------------------------------------follows-------------------------------\n"uid = re_uid.findall(effective_url)[0]if "total_count" in follows.keys() and follows["total_count"]>0 and str(start) == "0":for i in range((follows["total_count"]-1)/ONEPAGE):try:dbcurr.execute('INSERT INTO urlids(uk, start, limited, type, status) VALUES(%s, %s, %s, 1, 0)' % (uid, str(ONEPAGE*(i+1)), str(ONEPAGE)))except Exception as ex:print "E1", str(ex)passif "follow_list" in follows.keys():for item in follows["follow_list"]:if item['pubshare_count']==0:print "---------------------count ==0-------------------------------------------\n"#continuey = dbcurr.execute('SELECT id FROM user WHERE userid=%s', (item['follow_uk'],))y = dbcurr.fetchone()print "user uk",item['follow_uk']if not y:try:dbcurr.execute('INSERT INTO user(userid, username, files, status, downloaded, lastaccess,avatar_url,fans_count,follow_count,album_count) VALUES(%s, "%s", %s, 0, 0, "%s","%s",%s,%s,%s)' % (item['follow_uk'], item['follow_uname'],item['pubshare_count'],tnow,item['avatar_url'],item['fans_count'],item['follow_count'],item['album_count']))except Exception as ex:print "E13", str(ex)passelse:print "-----------------userid exists---------------------------------\n"else:print "delete 1", uid, startdbcurr.execute('delete from urlids where uk=%s and type=1 and start>%s' % (uid, start))elif 'getfanslist' in effective_url: #type = 2fans = json.loads(metadata)print "----------------------------------------fans----------------------------------\n"uid = re_uid.findall(effective_url)[0]if "total_count" in fans.keys() and fans["total_count"]>0 and str(start) == "0":for i in range((fans["total_count"]-1)/ONEPAGE):try:dbcurr.execute('INSERT INTO urlids(uk, start, limited, type, status) VALUES(%s, %s, %s, 2, 0)' % (uid, str(ONEPAGE*(i+1)), str(ONEPAGE)))except Exception as ex:print "E2", str(ex)passif "fans_list" in fans.keys():for item in fans["fans_list"]:if item['pubshare_count']==0:print "---------------------count ==0-------------------------------------------\n"#continuey = dbcurr.execute('SELECT id FROM user WHERE userid=%s', (item['fans_uk'],))y = dbcurr.fetchone()print "user uk",item['fans_uk']if not y:try:dbcurr.execute('INSERT INTO user(userid, username, files, status, downloaded, lastaccess,avatar_url,fans_count,follow_count,album_count) VALUES(%s, "%s", %s, 0, 0, "%s","%s",%s,%s,%s)' % (item['fans_uk'], item['fans_uname'],item['pubshare_count'],tnow,item['avatar_url'],item['fans_count'],item['follow_count'],item['album_count']))except Exception as ex:print "E23", str(ex)passelse:print "-----------------userid exists---------------------------------\n"else:print "delete 2", uid, startdbcurr.execute('delete from urlids where uk=%s and type=2 and start>%s' % (uid, start))else:shares = json.loads(metadata)print "shares"uid = re_uid.findall(effective_url)[0]totals+=1if "total_count" in shares.keys() and shares["total_count"]>0 and str(start) == "0":for i in range((shares["total_count"]-1)/ONESHAREPAGE):try:dbcurr.execute('INSERT INTO urlids(uk, start, limited, type, status) VALUES(%s, %s, %s, 0, 0)' % (uid, str(ONESHAREPAGE*(i+1)), str(ONESHAREPAGE)))except Exception as ex:print "E3", str(ex)passif "records" in shares.keys():for item in shares["records"]:print "-------------------------------------filename------------------ ",item['title']print "---------------------------------------------------------------\n"try:stamp_t=int(item["feed_time"])/1000t= time.localtime(int(stamp_t))share_time=time.strftime("%Y-%m-%d %H:%M:%S",t)urls=""if "shorturl" in item.keys():urls=item['shorturl']news+=1length=""if "filelist" in item.keys():length=str(item['filelist'][0]['size'])dbcurr.execute('INSERT INTO share(fid,userid, filename, shareid, status,filetype,share_time,create_time,urls,down,length) VALUES("%s",%s, "%s", %s, 0,"%s","%s","%s","%s",0,"%s")' % (sid(int(item['shareid'])),uid, item['title'], item['shareid'],get_category(get_ext(item['title'])),share_time,tnow,urls,length))# time.sleep(10)except Exception as ex:print "\n>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>E33\n", str(ex)print "item ---------------------------------------------\n"# time.sleep(10)passelse:print "delete 0", uid, startdbcurr.execute('delete from urlids where uk=%s and type=0 and start>%s' % (uid, str(start)))dbcurr.execute('delete from urlids where id=%s' % (id, ))dbconn.commit()except Exception as ex:print "E5", str(ex), idpid = re_pptt.findall(effective_url)if pid:print "pid>>>", pidppid = int(pid[0])PROXY_LIST[ppid][6] -= 1dbcurr.close()dbconn.close()def worker(k):global success, faileddbconn = mdb.connect(DB_HOST, DB_USER, DB_PASS, 'baiduyun', charset='utf8')dbcurr = dbconn.cursor()dbcurr.execute('SET NAMES utf8')dbcurr.execute('set global wait_timeout=60000')while True:#dbcurr.execute('select * from urlids where status=0 order by type limit 1')dbcurr.execute('select * from urlids where status=0 limit %s,1'%(str(k),))d = dbcurr.fetchall()#print dif d:id = d[0][0]uk = d[0][1]start = d[0][2]limit = d[0][3]type = d[0][4]dbcurr.execute('update urlids set status=1 where id=%s' % (str(id),))url = ""if type == 0:url = URL_SHARE.format(uk=uk, start=start, id=id).encode('utf-8')elif  type == 1:url = URL_FOLLOW.format(uk=uk, start=start, id=id).encode('utf-8')elif type == 2:url = URL_FANS.format(uk=uk, start=start, id=id).encode('utf-8')if url:hc_q.put((type, url))if len(d)==0:print "\ndata user uk\n "dbcurr.execute('select * from user where status=0 limit %s,100'%(str(k*100),))print "user "d = dbcurr.fetchall()#print "uk",dif d:for item in d:try:print "update user",item[1]dbcurr.execute('insert into urlids(uk, start, limited, type, status) values("%s", 0, %s, 0, 0)' % (item[1], str(ONESHAREPAGE)))dbcurr.execute('insert into urlids(uk, start, limited, type, status) values("%s", 0, %s, 1, 0)' % (item[1], str(ONEPAGE)))dbcurr.execute('insert into urlids(uk, start, limited, type, status) values("%s", 0, %s, 2, 0)' % (item[1], str(ONEPAGE)))dbcurr.execute('update user set status=1 where userid=%s and id=%s' % (item[1],item[6]))except Exception as ex:print "E6", str(ex)else:time.sleep(1)dbconn.commit()dbcurr.close()dbconn.close()def req_worker(inx):s = requests.Session()while True:time.sleep(1)req_item = hc_q.get()req_type = req_item[0]url = req_item[1]try:r = s.get(url)hc_r.put((r.text, url))except:passfor item in range(3):t = threading.Thread(target = req_worker, args = (item,))t.setDaemon(True)t.start()for item in range(2): s = threading.Thread(target = worker, args = (item,))s.setDaemon(True)s.start()for item in range(2):t = threading.Thread(target = response_worker, args = ())t.setDaemon(True)t.start()while 1:pass