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

时间：2022-12-21 14:03:35

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

相关文章