本次目标
知识点
- 面向对象编程
- selenium 操作浏览器
- pickle 保存和读取Cookie实现免登陆
- time 做延时操作
- os 创建文件,判断文件是否存在
第三方库
selenium >>> pip install selenium==3.4.1
开发环境
- 版 本:anaconda(python3.8.8)
- 编辑器:pycharm
点击领取【代码】【相关教程、资料】,或者对于本篇文章有疑问的同学可以私信我
先导入所需模块
from selenium import webdriver # 操作谷歌浏览器 需要额外安装的 并且现在安装这个模块得指定版本 3.4 from time import sleep import 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会发生变化
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')) # 保存cookie print('###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 = 2 print('###登陆成功###')
执行代码
if __name__ == '__main__': con = Concert() con.enter_concert()