Python3 使用selenium库登陆知乎并保存cookie为本地文件

Python3 使用selenium库登陆知乎并保存cookie为本地文件

学习使用selenium库模拟登陆知乎，并将cookie保存为本地文件，然后供以后（requests模块）使用，用selenium模拟登陆时，比较顺利，没有碰到需要验证码的情况，代码放在一个名为cookiesload.py模块中：

# -*- coding: utf-8 -*-

from selenium.webdriver import Chrome
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.common.exceptions import TimeoutException
from settings import *
import time
import os
import json


def zhihu_login():
    # 知乎登陆url
    login_url = 'https://www.zhihu.com/signup?next=%2F'
    brower = Chrome()
    # 设置浏览器窗口最大化
    brower.maximize_window()
    # 响应等待时间
    wait = WebDriverWait(brower, 5)
    brower.get(login_url)
    try:
        # 模拟点击最下方那个登陆链接书，跳转到账号密码登陆界面
        login_switch = wait.until(EC.element_to_be_clickable((By.XPATH, '//*[@id="root"]/div/main/div/div/div/div[2]/div[2]/span')))
        login_switch.click()
        # 模拟输入用户名
        username_input = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="root"]/div/main/div/div/div/div[2]/div[1]/form/div[1]/div[2]/div[1]/input')))
        username_input.send_keys(USERNAME)
        # 模拟输入密码
        password_input = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="root"]/div/main/div/div/div/div[2]/div[1]/form/div[2]/div/div[1]/input')))
        password_input.send_keys(PASSWORD)
        # 模拟点击登陆按钮
        login_button = wait.until(EC.element_to_be_clickable((By.XPATH, '//*[@id="root"]/div/main/div/div/div/div[2]/div[1]/form/button')))
        login_button.click()
        time.sleep(5)
        # 获取cookies
        cookies = brower.get_cookies()
        # 保存cookies为本地文件
        savecookies(cookies)
    except TimeoutException:
        # 如果捕获到超时异常则递归调用自己
        zhihu_login()


def savecookies(cookies):
    path = os.getcwd() + '/cookies/'
    if not os.path.exists(path):
        os.makedirs(path)
    with open(path + 'cookies.txt', 'w') as f:
        cookie = json.dumps(cookies)
        f.write(cookie)


if __name__ == '__main__':
    zhihu_login()

由于个人习惯，用了一个settings.py配置模块：

# 用户名
USERNAME = '173866016XX'

# 密码
PASSWORD = 'XXX'

执行完毕后，会在当前工作目录生成一个名为cookies的文件夹，cookies.txt文件在该文件夹中。在cookie没有过期的情况下，以后要访问知乎就可以不用再登陆了，直接加载该cookie文件即可。

上面使用的Chrome浏览器的有界面模式，如果不想让浏览器直接在后台运行，不弹出界面，可以这样：

from selenium.webdriver.chrome.options import Options

options = Options()
options.add_argument('--headless')
brower = Chrome(options=options)

更详细的写法可以参考：https://www.cnblogs.com/z-x-y/p/9026226.html