Selenium cookies for requests

发布时间：2019-11-28 14:50:58编辑：Run阅读（4005）

使用selenium模拟登录,保存cookies

代码示例:

import json
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time

def selenium_login_apple(username=None, password=None):
    if not username:
        username = 'your username'
    if not password:
        password = 'your password'
   
    url = "URL"
    option = webdriver.ChromeOptions()
    option.add_argument('--headless')
    option.add_argument('--no-sandbox')
    option.add_argument('--disable-gpu')
    option.add_argument('--disable-infobars')
    option.add_argument('--start-maximized')
    # option.add_argument('--hide-scrollbars')
    # option.add_argument('blink-settings=imagesEnabled=false')
    # 开发者模式打开,防止被检测
    # option.add_argument("--incognito")
    # option.add_argument("--disable-plugins-discovery")
    # option.add_argument('--profile-directory=Default')
    # option.add_argument('--disable-extensions')
    option.add_argument(
        'user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.100 Safari/537.36')
    
    browser = webdriver.Chrome(chrome_options=option)
    browser.delete_all_cookies()  # 清除cookies
    try:
        browser.get(url)
        # 隐式等待
        browser.implicitly_wait(60)
        # 切换到iframe里面,如果没有不需要,下面每一步都等待零点几秒是防止上一步还没完成,就执行了下一步
        browser.switch_to.frame(browser.find_element_by_xpath("//iframe"))
        # 输入用户名
        browser.find_element_by_xpath("//input[@id='account_name_text_field']").send_keys(username)
        time.sleep(0.2)
        # Keys.ENTER 输入回车键
        browser.find_element_by_xpath("//input[@id='account_name_text_field']").send_keys(Keys.ENTER)
        time.sleep(0.3)
        # 输入密码
        browser.find_element_by_xpath("//input[@id='password_text_field']").send_keys(password)
        time.sleep(0.2)
        browser.find_element_by_xpath("//input[@id='password_text_field']").send_keys(Keys.ENTER)
        time.sleep(0.3)
        # # 等待跳转到提问页面
        browser.implicitly_wait(60)
        time.sleep(3)
        # 获取cookies
        login_rear_cookie = browser.get_cookies()
        tmp_dict = {}
        if login_rear_cookie:
            for item in login_rear_cookie:
                tmp_dict.setdefault(item["name"], item["value"])
            # 把cookie信息用json序列化后写入cookie.txt文件
            print(tmp_dict)
            with open('cookie.txt', 'w') as f:
                f.write(json.dumps(login_rear_cookie))
    except Exception as e:
        print(e)
    finally:
        browser.quit()




执行上面的代码会生成一个cookie.txt文件,下面重点来了，requests怎么使用selenium的cookie

示例代码:

import requests
import json

# file_path为cookie.txt的路径
def search_apple_id(file_path):
    s = requests.session()
    s.keep_alive = False
    c = requests.cookies.RequestsCookieJar()
    with open(file_path, encoding='utf-8', mode='r') as f:
        cookies = json.loads(f.read())
        for item in cookies:
            c.set(item["name"], item["value"])
            # xsrf-token为登录是生成的验证码,有的是放在headers里面返回,一般是登录后保存在cookies信息里面的
            if item["name"] == 'XSRF-TOKEN-CM':
                xsrf_token = item["value"]
        s.cookies.update(c)
    result_url = "接口URL地址"
    headers = {
        "Accept": "application/json, text/plain, */*",
        "Content-Type": "application/json;charset=UTF-8",
        "Accept-Encoding": "gzip, deflate, br",
        # 如果有xsrf-token请求接口时,必须要在headers里面带上,不然接口请求会失败
        "X-XSRF-TOKEN-CM": xsrf_token,
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.100 Safari/537.36",
    }
    body = {"storefronts":["AL","AR","AU","AT","AZ","BE","KH","CA","CL","CO","HR","CZ","DK","EC","EG","FI","FR","DE","GR","HK","HU","IN","ID","IE","IL","IT","JO","KZ","KR","KW","LB","MO","MY","MX","NL","NZ","NO","OM","PK","PE","PH","PL","PT","QA","RO","SA","SG","ZA","ES","SE","CH","TW","TH","UA","AE","GB","US","VN"]}
    # allow_redirects 禁止重定向
    # verify=False 不验证https
    r1 = s.post(url=result_url, headers=headers, json=body, timeout=60, allow_redirects=False, verify=False)
    if r1.status_code == 200:
        ret1 = json.loads(r1.text)
        print(ret1)

关键字：