python3获取请求cookie

电玩女神 2022-10-29 15:20 231阅读 0赞

[GitHub链接][GitHub]  
`https://github.com/WindSnowLi/My-python-tools/tree/main/PythonGetCookie`

# -*- coding: UTF-8 -*-
    
    from http import cookiejar
    from urllib import request
    from pip._vendor import requests
    from selenium.webdriver import Firefox
    from selenium.webdriver.firefox.options import Options
    from selenium.webdriver.firefox.service import Service
    import time
    import json
    
    
    # 使用python库直接获取cookie
    def get_cookie(url):
        """ :param url: 请求连接应保证服务器有set-cookie写入操作 :return: """
        # 请求负载
        data = { }
        # 请求头
        headers = { }
        try:
            # 声明一个CookieJar对象实例来保存cookie
            cookie = cookiejar.CookieJar()
            # 利用urllib.request库的HTTPCookieProcessor对象来创建cookie处理器,也就CookieHandler
            handler = request.HTTPCookieProcessor(cookie)
            # 通过CookieHandler创建opener
            opener = request.build_opener(handler)
            my_request = request.Request(
                url, headers=headers, data=json.dumps(data).encode("utf-8"))
            # 此处的open方法打开网页
            response = opener.open(my_request)
            # 打印cookie信息
            return requests.utils.dict_from_cookiejar(cookie)
        except Exception as e:
            print(e)
            return ''
    
    
    # 需要安装火狐浏览器，并下载geckodriver驱动
    # 通过火狐浏览器获取cookie
    # 通过正常的浏览器请求，优点是在js中的请求也会发起，因为不是所有的请求都会有set-cookie写入
    # 缺点是加载慢
    def firefox_get_cookies(url):
        """ :param url: 请求连接 :return: """
        c_service = Service('geckodriver')
        driver = any
        try:
            c_service.command_line_args()
            c_service.start()
            firefox_options = Options()
            # 不启动界面显示- linux下命令行模式必须启用
            firefox_options.add_argument('-headless')
            driver = Firefox(options=firefox_options)
            driver.get(url)
            # 第一次请求浏览器一般无法显示cookie
            # 等待第一次加载完成
            time.sleep(2)
            # 刷新
            driver.refresh()
            # 等待第二次加载完成
            time.sleep(2)
            return driver.get_cookies()
        except Exception as e:
            print(e)
        finally:
            driver.quit()
            c_service.stop()
    
    
    if __name__ == '__main__':
        # 只会获取当前连接写入的cookie
        print(get_cookie('https://www.baidu.com'))
        # 附带的请求写入的cookie也可以查询得到
        print(firefox_get_cookies('https://www.baidu.com'))

[GitHub]: https://github.com/WindSnowLi/My-python-tools/tree/main/PythonGetCookie