python用什么数据库 Python用selenium实现模拟登录，各种抢票脚本的第一步

本次目标

文章插图
知识点

面向对象编程
selenium 操作浏览器
pickle 保存和读取Cookie实现免登陆
time 做延时操作
os 创建文件，判断文件是否存在

第三方库selenium >>> pip install selenium==3.4.1
开发环境

版本：anaconda（python3.8.8）
编辑器：pycharm

点击领取【代码】【相关教程、资料】，或者对于本篇文章有疑问的同学可以私信我先导入所需模块from selenium import webdriver# 操作谷歌浏览器需要额外安装的并且现在安装这个模块得指定版本 3.4from time import sleepimport pickle# 保存和读取cookie实现免登录的工具import os# 操作文件的模块定义变量# 大麦网主页damai_url = 'https://www.damai.cn/'# 登陆login_url = 'https://passport.damai.cn/login?ru=https%3A%2F%2Fwww.damai.cn%2F'# 抢票目标页面target_url = 'https://detail.damai.cn/item.htm?id=610820299671'

文章插图

文章插图
在这个类里面完成所有的登陆功能class Concert:# 完成一个初始化def __init__(self):self.status = 0# 状态码, 表示当前操作执行到了哪个步骤self.login_method = 1# {0: 模拟登陆, 1: 免登录}self.driver = webdriver.Chrome(executable_path='chromedriver.exe')登陆网站的时候要用的方法没有登陆的情况下第一次登陆的时候，会帮助我记录我们的登陆信息
登陆成功之后 cookie会发生变化
【python用什么数据库 Python用selenium实现模拟登录，各种抢票脚本的第一步】def set_cookies(self):self.driver.get(damai_url)# 打开大麦网主页print("###请点击登陆###")# 如果说我一直没有点击登陆? 我是不是这个程序就停在这里了while self.driver.title.find('大麦网-全球演出赛事官方购票平台') != -1:sleep(1)# 程序休眠print("###扫码成功###")pickle.dump(self.driver.get_cookies(), open('cookies.pkl', 'wb'))#保存cookieprint('###cookie保存成功###')self.driver.get(target_url)直接拿本地的登陆信息假如我现在已经登陆过了，那么直接拿本地的登陆信息，直接登陆就可以了
def get_cookie(self):cookies = pickle.load(open('cookies.pkl', 'rb'))for cookie in cookies:cookie_dict = {'domain': '.damai.cn','name': cookie.get('name'),'value': cookie.get('value')}self.driver.add_cookie(cookie_dict)print('###载入cookie###')实现登录def login(self):if self.login_method == 0:self.driver.get(login_url)# 跳转到登陆界面登陆一下print('###开始登陆###')elif self.login_method == 1:# 如果我本地没有cookies.pkl文件if not os.path.exists('cookies.pkl'):self.set_cookies()# 没有文件的情况下, 再次登陆一下else:self.driver.get(target_url)# 跳转到抢票页面self.get_cookie()# 登陆信息载入浏览器打开浏览器功能，调用所以所有函数调用登陆，并且刷新页面，改变登陆成功的标识
def enter_concert(self):print('###打开浏览器, 进入大麦网###')# 调用登陆self.login()self.driver.refresh()self.status = 2print('###登陆成功###')执行代码if __name__ == '__main__':con = Concert()con.enter_concert()