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