Python爬微博(ajax+mongo)

来源:互联网 发布:jsp页面添加java代码 编辑:程序博客网 时间:2024/05/20 21:19
import requestsfrom urllib.parse import urlencodefrom pyquery import PyQuery as pqfrom pymongo import MongoClientbase_url = 'https://m.weibo.cn/api/container/getIndex?'headers = {    'Host': 'm.weibo.cn',    'Referer': 'https://m.weibo.cn/u/2145291155',    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36',    'X-Requested-With': 'XMLHttpRequest',}client = MongoClient()db = client['weibo']collection = db['weibo']max_page = 14def get_page(page):    params = {        'type': 'uid',        'value': '2145291155',        'containerid': '1076032145291155',        'page': page    }    url = base_url + urlencode(params)    try:        response = requests.get(url, headers=headers)        if response.status_code == 200:            return response.json()    except requests.ConnectionError as e:        print('Error', e.args)def parse_page(json):    if json:        items = json.get('cards')        for item in items:            item = item.get('mblog')            weibo = {}  #字典            weibo['id'] = item.get('id')            weibo['text'] = pq(item.get('text')).text()            weibo['attitudes'] = item.get('attitudes_count')            weibo['comments'] = item.get('comments_count')            weibo['reposts'] = item.get('reposts_count')            yield weibodef save_to_mongo(result):    if collection.insert(result):        print('Saved to Mongo')if __name__ == '__main__':    for page in range(1, max_page + 1):        json = get_page(page)        results = parse_page(json)        for result in results:            print('\n'.result)            save_to_mongo(result)
原创粉丝点击