网页Cookies获取与保存工具详解

场景与背景

在网络爬虫、自动化测试等场景中，获取和管理Cookies是常见的需求。Cookies可以用于模拟登录状态、保持会话等。实现一个简单易用的Cookies获取与保存工具，对提高工作效率具有重要意义。

工具实现

以下是一个使用Python的tkinter和selenium库实现的网页Cookies获取与保存工具的代码：

Python

import tkinter
import time
import json
import os
import selenium
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By

driver = None  # 全局变量存储 driver

def 获取cookie():
    """
    获取网页 cookie 并保存到本地文件中
    """
    global driver  # 声明使用全局变量 driver
    url = 链接输入框.get()
    name = 名称输入框.get()
    path_begin = 地点输入框.get()
    path = os.path.join(path_begin, f'{name}.json')
    os.makedirs(os.path.dirname(path), exist_ok=True)
    service = Service()
    service.path = r'E:\py项目文件\pythonProject\2\项目学习\chrome_win64\chromedriver_win64.exe'
    options = Options()
    options.add_experimental_option('detach', True)
    driver = selenium.webdriver.Chrome(service=service, options=options)
    driver.get(url)
    cookies = driver.get_cookies()
    try:
        with open(path, 'w') as f:
            json.dump(cookies, f)
        print('写入成功')
    except Exception as e:
        print(f'写入失败: {e}')

def 关闭():
    """
    关闭 Selenium WebDriver 对象
    """
    global driver  # 声明使用全局变量 driver
    if driver:
        driver.quit()

root = tkinter.Tk()
root.title('获取cookies')
root.geometry('300x200')
root.resizable(width=False, height=False)

链接标签 = tkinter.Label(root, text='链接：', font='宋体 12')
链接标签.grid(row=0, column=0)
链接输入框 = tkinter.Entry(root, width=30)
链接输入框.grid(row=0, column=1)

名称标签 = tkinter.Label(root, text='名称：', font='宋体 12')
名称标签.grid(row=1, column=0)
名称输入框 = tkinter.Entry(root, width=30)
名称输入框.grid(row=1, column=1)

地点标签 = tkinter.Label(root, text='地点：', font='宋体 12')
地点标签.grid(row=2, column=0)
地点输入框 = tkinter.Entry(root, width=30)
地点输入框.grid(row=2, column=1)

begin = tkinter.Button(root, text='开始', width=10, command=获取cookie)
begin.place(relx=0.2, rely=0.4)

out = tkinter.Button(root, text='结束', width=10, command=关闭)
out.place(relx=0.5, rely=0.4)

root.mainloop()

代码解析

1. 整体架构

该工具由主程序和两个核心函数组成：

• 主程序：负责构建图形界面（GUI），接收用户输入。

• 核心函数：

  • 获取cookie()：用于获取指定网页的Cookies并保存为JSON文件。

  • 关闭()：用于关闭浏览器实例，释放资源。

2. 功能实现

(1) 获取Cookies

Python

def 获取cookie():
    global driver
    url = 链接输入框.get()
    name = 名称输入框.get()
    path_begin = 地点输入框.get()
    path = os.path.join(path_begin, f'{name}.json')
    os.makedirs(os.path.dirname(path), exist_ok=True)
    service = Service()
    service.path = r'E:\py项目文件\pythonProject\2\项目学习\chrome_win64\chromedriver_win64.exe'
    options = Options()
    options.add_experimental_option('detach', True)
    driver = selenium.webdriver.Chrome(service=service, options=options)
    driver.get(url)
    cookies = driver.get_cookies()
    try:
        with open(path, 'w') as f:
            json.dump(cookies, f)
        print('写入成功')
    except Exception as e:
        print(f'写入失败: {e}')

• 功能描述：

  • 从GUI输入框获取用户指定的URL、文件名和保存路径。

  • 构造完整的保存路径，并创建必要的目录（如果不存在）。

  • 启动Chrome浏览器，访问指定URL。

  • 获取当前页面的所有Cookies，并以JSON格式保存到指定文件。

• 关键代码解析：

  Python

  os.makedirs(os.path.dirname(path), exist_ok=True)

  • 确保保存目录存在，避免因路径不存在导致的写入错误。

  Python

  driver = selenium.webdriver.Chrome(service=service, options=options)

  • 创建并启动Chrome浏览器实例，设置浏览器行为（如保持打开状态）。

  Python

  cookies = driver.get_cookies()

  • 获取当前页面的所有Cookies。

  Python

  with open(path, 'w') as f:
      json.dump(cookies, f)

  • 将Cookies保存为JSON文件。

(2) 关闭浏览器

Python

def 关闭():
    global driver
    if driver:
        driver.quit()

• 功能描述：

  • 安全关闭浏览器实例，释放系统资源。

3. 图形界面

Python

root = tkinter.Tk()
root.title('获取cookies')
root.geometry('300x200')
root.resizable(width=False, height=False)

链接标签 = tkinter.Label(root, text='链接：', font='宋体 12')
链接标签.grid(row=0, column=0)
链接输入框 = tkinter.Entry(root, width=30)
链接输入框.grid(row=0, column=1)

名称标签 = tkinter.Label(root, text='名称：', font='宋体 12')
名称标签.grid(row=1, column=0)
名称输入框 = tkinter.Entry(root, width=30)
名称输入框.grid(row=1, column=1)

地点标签 = tkinter.Label(root, text='地点：', font='宋体 12')
地点标签.grid(row=2, column=0)
地点输入框 = tkinter.Entry(root, width=30)
地点输入框.grid(row=2, column=1)

begin = tkinter.Button(root, text='开始', width=10, command=获取cookie)
begin.place(relx=0.2, rely=0.4)

out = tkinter.Button(root, text='结束', width=10, command=关闭)
out.place(relx=0.5, rely=0.4)

root.mainloop()

• 功能描述：

  • 创建主窗口，设置窗口标题和大小。

  • 添加标签和输入框，用于接收用户输入。

  • 添加按钮，绑定点击事件（调用核心函数）。

4. 优点与局限性

优点

1. 易用性：

   • 提供直观的图形界面，用户无需编写代码即可操作。

   • 允许用户自定义URL、文件名和保存路径。

2. 灵活性：

   • 支持任意网页的Cookies获取。

   • Cookies以JSON格式保存，便于后续使用。

3. 健壮性：

   • 在保存路径不存在时自动创建目录，避免文件写入错误。

局限性

1. 依赖性：

   • 需要正确安装和配置Chrome浏览器及ChromeDriver。

   • 仅支持Chrome浏览器，其他浏览器需额外修改代码。

2. 手动操作：

   • 当前实现假定用户会手动完成登录过程，缺乏自动化登录支持。

3. 异常处理：

   • 缺乏完善的错误处理机制，如网络异常、文件权限错误等。

总结

该工具通过tkinter构建图形界面，结合selenium的强大网页自动化能力，实现了一个简单实用的网页Cookies获取与保存功能。尽管存在一些局限性，但其设计理念和实现方式为网络爬虫、自动化测试等场景提供了良好的基础。

通过进一步改进和扩展，该工具可以更好地满足复杂多变的实际需求，成为开发者在处理网页Cookies相关任务时的有力助手。