Python爬虫：爬取斗鱼直播信息（实例）

以爬取斗鱼直播页面的部分信息为例：

"""
https://www.douyu.com/gapi/rkc/directory/0_0/2
https://www.douyu.com/gapi/rkc/directory/0_0/3
"""

import requests

#请求网页
def page_def(url,ua):
    resp = requests.get(url,headers = ua)
    json_text = resp.json()
    return json_text


#解析网页
def info_def(json_text):
    douyu_list=[]
    for i in range(len(json_text['data']['rl'])):
        data_list = json_text['data']['rl'][i]
        #print(data_list)
        temp_dict={
            'rid':data_list['rid'], #主播ID
            'rn':data_list['rn'], #主题
            'nn':data_list['nn'], #昵称
            'ol':data_list['ol'], #点击数
            'c2name':data_list['c2name'], #分类
            'url':'https://www.douyu.com/'+str(data_list['rid'])
        }
        douyu_list.append(temp_dict)
    return douyu_list

#写入csv
def csv_def(douyu_list):
    import csv
    with open(r'C:\Users\HP\Desktop\douyu.csv','a',encoding='utf-8-sig',newline='') as cf:
        w = csv.DictWriter(cf,fieldnames = ['rid','rn','nn','ol','c2name','url'])
        w.writeheader()
        w.writerows(douyu_list)
        print("爬取完成！")

#主函数
def main_def():
    ua = {'User-Agent':'User-Agent:Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1'}
    for i in range(1,11):
        url = 'https://www.douyu.com/gapi/rkc/directory/0_0/%d'%(i)
        json_text = page_def(url,ua)
        douyu_list = info_def(json_text)
        print("正在爬取第%d"%(i)+"页")
        csv_def(douyu_list)
    
if __name__ == '__main__':
    main_def()