#coding:utf-8
import requests
import browsercookie
from urllib.parse import urlencode
import pandas as pd
from datetime import datetime

class BaiduIndex(object):
    def __init__(self,kw,startDate,endDate):
        self.kw = kw
        self.startDate = startDate
        self.endDate = endDate
        self.headers = {'User-Agent':'Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/45.0.2454.101 Safari/537.36'}
        self.cookies = browsercookie.chrome()
        self.baidu_indexs = self.get_baidu_indexs()
        self.categories = {'all':'total_index','pc':'pc_index','wise':'mobile_index'}

    def get_baidu_indexs(self):
        request_args = {
                    'word': self.kw,
                    'startDate': self.startDate,
                    'endDate': self.endDate,
                    'area': 0,
                }
        url = 'http://index.baidu.com/api/SearchApi/index?{}'.format(urlencode(request_args))
        # print(url)
        html = requests.get(url,headers = self.headers,cookies = self.cookies).json()
        # print(html)
        return html

    #获取关键词某段时间内的平均指数
    def avg_baidu_index(self):
        datas = self.baidu_indexs
        if datas.get('data'):
            datas = datas['data']['generalRatio'][0]
            # datas = html['data']['generalRatio'][0]
            total_index = datas['all']['avg']
            pc_index = datas['pc']['avg']
            mobile_index = datas['wise']['avg']
            return {'total_index':total_index,'mobile_index':mobile_index,'pc_index':pc_index}
        else:
            return {'total_index':0,'mobile_index':0,'pc_index':0}

    # 获取关键词某段时间内每天的指数
    def daily_baidu_index(self):
        datelist = self.datelist()
        datas = self.baidu_indexs
        if datas.get('data'):
            uniqid = datas['data']['uniqid']
            datas = datas['data']['userIndexes'][0]
            key = self.get_key(uniqid)
            d = {}
            for category,value in self.categories.items():
                data = datas[category]['data']
                # print(key,data)
                index = self.decrypt(key,data)
                num = len(index)
                # print(len(datelist),num,len(datelist[-num:]))
                indexs = {date:index for date,index in zip(datelist[-num:],index)}
                d[value] = indexs
            return d
        else:
            d = {}
            for category, value in self.categories.items():
                indexs = {date:0 for date in datelist}
                d[value] = indexs
            return d

    def get_key(self,uniqid):
        url = 'http://index.baidu.com/Interface/api/ptbk?uniqid={}'.format(uniqid)
        html = requests.get(url,headers = self.headers,cookies=self.cookies).json()
        return html['data']

    # 根据起始日期生成每日日期
    def datelist(self):
        datelist =[datetime.strftime(x,'%Y-%m-%d') for x in list(pd.date_range(start=self.startDate, end=self.endDate))]
        return datelist

    # 解密百度指数
    def decrypt(self,key,data):
        num = {}
        datas = []
        for line in range(len(key) // 2):
            num[key[line]] = key[len(key) // 2 + line]
        for dt in range(len(data)):
            datas.append(num[data[dt]])
        return ''.join(datas).split(',')

if __name__ == "__main__":
    b = BaiduIndex('周杰伦','2020-03-15','2020-03-17')
    daily_index = b.daily_baidu_index()   #某个时间段每一天的指数
    print(daily_index)
    print(daily_index.get('total_index').get('2020-03-16'))  #2020年3月16日的整体指数
    print(daily_index.get('pc_index').get('2020-03-16'))  # 2020年3月16日的PC指数
    print(daily_index.get('mobile_index').get('2020-03-16'))  # 2020年3月16日的移动指数
    avg_index = b.avg_baidu_index()     #某个时间段平均指数
    print(avg_index)

输出结果如下所示：

{'total_index': {'2020-03-15': '18201', '2020-03-16': '18934', '2020-03-17': '19095'}, 'pc_index': {'2020-03-15': '3288', '2020-03-16': '3912', '2020-03-17': '3969'}, 'mobile_index': {'2020-03-15': '14913', '2020-03-16': '15022', '2020-03-17': '15126'}}
18934
3912
15022
{'total_index': 18743, 'mobile_index': 15020, 'pc_index': 3723}

可以看到，可查询到某段时间内平均指数以及每天的指数，包括总指数，PC指数以及移动端指数。

如果想查某一天的指数，设置startDate和endDate是同一日期即可。如下所示：

if __name__ == "__main__":
    b = BaiduIndex('周杰伦', '2020-03-16', '2020-03-16')
    avg_index = b.avg_baidu_index()
    print(avg_index)

输出：

{'total_index': 18934, 'mobile_index': 15022, 'pc_index': 3912}

这里仅以单个关键词查询作示例，至于改成批量查询及将查询结果写入文件，有Python基础的话不是什么难事儿，这里不再示例。

没有大量查询测试，但很可能会遇到封账号问题，可尝试换IP、多账号查询等方式。

老规矩，关注公众号【Python加SEO做增长】，后台回复关键词【百度指数】即可获得这份代码，不会使用可私信我。