|
|
import os
|
|
|
import time
|
|
|
import pickle # 实现读取cookie
|
|
|
|
|
|
from selenium import webdriver
|
|
|
|
|
|
# 大麦网首页
|
|
|
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=779820312992'
|
|
|
|
|
|
|
|
|
class Concert:
|
|
|
# 初始化加载
|
|
|
def __init__(self):
|
|
|
self.status = 0 # 状态，表示当前操作执行到那一步了
|
|
|
self.login_method = 1 # {0: 模拟登录， 1：cookie登录}
|
|
|
self.driver = webdriver.Edge() # 初始化浏览器
|
|
|
|
|
|
# cookies： 登录网站的时候出现的，记录用户信息
|
|
|
def set_cookies(self):
|
|
|
self.driver.get(login_url) # 帮助我们打开一个网页
|
|
|
print("###请扫码登录###")
|
|
|
time.sleep(20)
|
|
|
print("###登录成功###")
|
|
|
# cookie.pkl
|
|
|
pickle.dump(self.driver.get_cookies(), open('cookies.pkl', 'wb'))
|
|
|
print("###cookie保存成功###")
|
|
|
# 抢票
|
|
|
self.driver.get(target_url)
|
|
|
|
|
|
# 如果当前我已经有了cookie.pkl
|
|
|
def get_cookies(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):
|
|
|
# 如果为0 模拟登陆一下
|
|
|
if self.login_method == 0:
|
|
|
self.driver.get(login_url)
|
|
|
|
|
|
elif self.login_method == 1:
|
|
|
# 如果当前目录下 没有这个cookie.pkl这个文件
|
|
|
if not os.path.exists('cookies.pkl'):
|
|
|
# 登录一下 登录信息记录
|
|
|
self.set_cookies()
|
|
|
|
|
|
else:
|
|
|
self.driver.get(target_url)
|
|
|
# 登录一下 通过selenium传入一些用户信息
|
|
|
self.get_cookies()
|
|
|
|
|
|
# 打开浏览器
|
|
|
def enter_concert(self):
|
|
|
print("###打开浏览器，进入大麦网###")
|
|
|
# 调用login
|
|
|
self.login()
|
|
|
self.driver.refresh()
|
|
|
self.status = 2
|
|
|
print("###登录成功###")
|
|
|
|
|
|
# 二、抢票且下单
|
|
|
# 选票操作
|
|
|
def choose_ticket(self):
|
|
|
if self.status == 2:
|
|
|
print('=' * 30)
|
|
|
print('###请开始日期选择以及票价选择###')
|
|
|
print(self.driver.title.find("确认订单") == 1)
|
|
|
while self.driver.title.find("确认订单") == -1:
|
|
|
|
|
|
# 下单按钮
|
|
|
buybutton = self.driver.find_element_by_class_name("buybtn").text
|
|
|
|
|
|
if buybutton == '提交缺货登记':
|
|
|
self.driver.refresh()
|
|
|
elif buybutton == '立即购买':
|
|
|
self.driver.find_element_by_class_name("buybtn").click()
|
|
|
self.status = 5
|
|
|
elif buybutton == '选座购买':
|
|
|
self.driver.find_element_by_class_name("buybtn").click()
|
|
|
self.status = 4
|
|
|
else:
|
|
|
self.status = 100
|
|
|
title = self.driver.title
|
|
|
if title == '选座购买':
|
|
|
|
|
|
# 执行选座操作
|
|
|
self.status = 10
|
|
|
elif title == '确认订单':
|
|
|
|
|
|
# 实现下单的逻辑
|
|
|
while True:
|
|
|
print('###正在加载###')
|
|
|
self.check_order()
|
|
|
break
|
|
|
|
|
|
# 下单操作
|
|
|
def check_order(self):
|
|
|
if self.status in [4, 5]:
|
|
|
print("###开始确认订单###")
|
|
|
try:
|
|
|
self.driver.find_element_by_xpath('//*[@id="container"]/div/div[2]/div[2]/div[1]/div/label').click()
|
|
|
except Exception as e:
|
|
|
print("###购票人信息选中失败，自行查看元素位置###")
|
|
|
print(e)
|
|
|
time.sleep(0.5)
|
|
|
self.driver.find_element_by_xpath('//*[@id="container"]/div/div[9]/button').click()
|
|
|
|
|
|
|
|
|
if '__main__' == __name__:
|
|
|
con = Concert()
|
|
|
con.enter_concert()
|
|
|
time.sleep(1)
|
|
|
con.choose_ticket()
|
