利用Selenium添加cookie实现自动登录的示例代码(fofa)

2024-04-02 19:04:59 827人浏览八月长安

Python 官方文档：入门教程 => 点击学习

摘要

介绍 selenium可以模拟浏览器进行自动化操作，但一些网站需要进行登录才能进行一些操作，比起输入账号密码，cookie是更加方便的。而且fofa首先登录邮箱账号时获得的cooki

介绍

selenium可以模拟浏览器进行自动化操作，但一些网站需要进行登录才能进行一些操作，比起输入账号密码，cookie是更加方便的。而且fofa首先登录邮箱账号时获得的cookie并不是fofa的cookie，因此我们直接选择利用fofa的cookie进行自动登录。但是selenium需要先打开一个网站才会加载进去cookies，因此我们需要将cookies写在代码中，加载进去

扩展

get_cookies()：获得所有cookie信息。
get_cookie(name)：返回字典的key为“name”的cookie信息。
add_cookie(cookie_dict)：添加cookie。“cookie_dict”指字典对象，必须有name 和value 值。
delete_cookie(name,optionsString)：删除cookie信息。“name”是要删除的cookie的名称，“optionsString”是该cookie的选项，目前支持的选项包括“路径”，“域”。
delete_all_cookies()：删除所有cookie信息。

步骤

（分为两部分，先1后2）
以fofa为例：

1.模拟登录并保存此次登录的cookies

代码如下：


import JSON
from selenium import WEBdriver

dirver = webdriver.Chrome(executable_path=r'C:\Program Files (x86)\Google\Chrome\Application\chromedriver.exe')
dirver.get('https://classic.fofa.so/')
dictCookies = dirver.get_cookies()  #获得所有cookie信息(返回是字典)
print(dictCookies)
print(type(dictCookies))
jsonCookies = json.dumps(dictCookies)  #dumps是将dict转化成str格式
print(jsonCookies)
print(type(jsonCookies))
# 登录完成后,将cookies保存到本地文件
with open("cookies_fofa.json", "w") as fp:
    fp.write(jsonCookies)

在这里插入图片描述

可以看到生成cookies_fofa.json

在这里插入图片描述

也就是转化成str的cookies被保存在该文件中

在这里插入图片描述

第一部分完成，获得了页面的cookies，这样获得好处是能知道cookies的格式。

2.添加我们登录账户的cookies实现自动登录

我们可以先登录我们的会员号，然后查看其cookie的名和值，然后添加至下面代码，即可实现自动登录

在这里插入图片描述

代码如下：


import json
from selenium import webdriver

browser = webdriver.Chrome()
browser.get("Https://classic.fofa.so/")
browser.delete_all_cookies()  #删除所有cookie信息
with open('cookies_fofa.json', 'r', encoding='utf-8') as f:
    listCookies = json.loads(f.read())     #loads是将str转化成dict格式

for cookie in listCookies:
    browser.add_cookie({     #此处模仿之前生成的cookies_fofa.json的格式写即可，要用单引号，因为已经转成dict格式了
        'domain': '.fofa.so',
        'name': 'cookie的名',
        'value': 'cookie的值',
        'path': '/',
        'expires': None
    })

browser.get("https://classic.fofa.so/")

在这里插入图片描述