Python用selenium实现模拟登录，各种抢票脚本的第一步

本次目标

知识点

• 面向对象编程
• 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()