forked from lsbnbdz/damai_tickets
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main_cable.py
327 lines (285 loc) · 13.9 KB
/
main_cable.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
# coding: utf-8
from json import loads
from time import sleep, time
from pickle import dump, load
from os.path import exists
from selenium import webdriver
from selenium.webdriver.remote.webelement import WebElement
from selenium.common import exceptions
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
class Concert(object):
def __init__(self, date, session, price, real_name, nick_name, ticket_num, damai_url, target_url, driver_path):
self.date = date # 日期序号
self.session = session # 场次序号优先级
self.price = price # 票价序号优先级
self.real_name = real_name # 实名者序号
self.status = 0 # 状态标记
self.time_start = time() # 开始时间
self.time_end = 0 # 结束时间
self.num = 0 # 尝试次数
self.ticket_num = ticket_num # 购买票数
self.nick_name = nick_name # 用户昵称
self.damai_url = damai_url # 大麦网官网网址
self.target_url = target_url # 目标购票网址
self.driver_path = driver_path # 浏览器驱动地址
self.driver = None
def isClassPresent(self, item, name, ret=False):
try:
result = item.find_element_by_class_name(name)
if ret:
return result
else:
return True
except:
return False
# 获取账号的cookie信息
def get_cookie(self):
self.driver.get(self.damai_url)
print(u"###请点击登录###")
self.driver.find_element(by=By.CLASS_NAME, value='login-user').click()
while self.driver.title.find('大麦网-全球演出赛事官方购票平台') != -1: # 等待网页加载完成
sleep(0.1)
print(u"###请扫码登录###")
while self.driver.title == '大麦登录': # 等待扫码完成
sleep(0.1)
dump(self.driver.get_cookies(), open("cookies.pkl", "wb"))
print(u"###Cookie保存成功###")
def set_cookie(self):
try:
cookies = load(open("cookies.pkl", "rb")) # 载入cookie
for cookie in cookies:
cookie_dict = {
'domain': '.damai.cn', # 必须有,不然就是假登录
'name': cookie.get('name'),
'value': cookie.get('value'),
"expires": "",
'path': '/',
'httpOnly': False,
'HostOnly': False,
'Secure': False}
self.driver.add_cookie(cookie_dict)
print(u'###载入Cookie###')
except Exception as e:
print(e)
def login(self):
print(u'###开始登录###')
self.driver.get(self.target_url)
WebDriverWait(self.driver, 10, 0.05).until(EC.title_contains('商品详情'))
self.set_cookie()
def enter_concert(self):
print(u'###打开浏览器,进入大麦网###')
if not exists('cookies.pkl'): # 如果不存在cookie.pkl,就获取一下
self.driver = webdriver.Chrome(executable_path=self.driver_path)
self.get_cookie()
print(u'###成功获取Cookie,重启浏览器###')
self.driver.quit()
options = webdriver.ChromeOptions()
# 禁止图片、js、css加载
prefs = {"profile.managed_default_content_settings.images": 2,
"profile.managed_default_content_settings.javascript": 1,
'permissions.default.stylesheet': 2}
mobile_emulation = {"deviceName": "Nexus 6"}
options.add_experimental_option("prefs", prefs)
options.add_experimental_option("mobileEmulation", mobile_emulation)
options.add_argument("--disable-blink-features=AutomationControlled")
# 更换等待策略为不等待浏览器加载完全就进行下一步操作
capa = DesiredCapabilities.CHROME
capa["pageLoadStrategy"] = "none"
self.driver = webdriver.Chrome(
executable_path=self.driver_path, options=options, desired_capabilities=capa)
# 登录到具体抢购页面
self.login()
self.driver.refresh()
# try:
# # 等待nickname出现
# locator = (By.XPATH, "/html/body/div[1]/div/div[3]/div[1]/a[2]/div")
# WebDriverWait(self.driver, 5, 0.3).until(EC.text_to_be_present_in_element(locator, self.nick_name))
# self.status = 1
# print(u"###登录成功###")
# self.time_start = time()
# except:
# self.status = 0
# self.driver.quit()
# raise Exception(u"***错误:登录失败,请删除cookie后重试***")
def click_util(self, btn, locator):
while True:
btn.click()
try:
return WebDriverWait(self.driver, 1, 0.05).until(EC.presence_of_element_located(locator))
except:
continue
# 实现购买函数
def choose_ticket(self):
print(u"###进入抢票界面###")
# 如果跳转到了确认界面就算这步成功了,否则继续执行此步
while self.driver.title.find('订单确认') == -1:
self.num += 1 # 尝试次数加1
if con.driver.current_url.find("buy.damai.cn") != -1:
break
# 确认页面刷新成功
try:
box = WebDriverWait(self.driver, 1, 0.05).until(
EC.presence_of_element_located((By.ID, 'app')))
except:
raise Exception(u"***Error: 页面刷新出错***")
try:
sleep(1)
realname_popup = box.find_elements(
by=By.XPATH, value="//div[@class='realname-popup']") # 寻找实名身份遮罩
if len(realname_popup) != 0:
known_button = realname_popup[0].find_element(
by=By.XPATH, value="//div[@class='operate']//div[@class='button']")
known_button[0].click()
except:
raise Exception(u"***Error: 实名制遮罩关闭失败***")
try:
# sleep(1)
buybutton = box.find_element(by=By.CLASS_NAME, value='buy')
buybutton_text: str = buybutton.text
except Exception as e:
raise Exception(f"***Error: buybutton 位置找不到***: {e}")
if "即将开抢" in buybutton_text or "提醒" in buybutton_text:
self.status = 2
raise Exception(u"---尚未开售,刷新等待---")
if "缺货" in buybutton_text:
raise Exception("---已经缺货了---")
sleep(0.05)
buybutton.click()
box = WebDriverWait(self.driver, 1, 0.05).until(
EC.presence_of_element_located((By.CSS_SELECTOR, '.sku-pop-wrapper')))
try:
session = WebDriverWait(self.driver, 1, 0.05).until(
EC.presence_of_element_located((By.CLASS_NAME, 'sku-times-card'))) # 日期、场次和票档进行定位
date = None # 暂不支持日期
toBeClicks = []
if date is not None:
date_list = date.find_element(
by=By.XPATH, value="//div[@class='wh_content_item']//div[starts-with(@class,'wh_item_date')]") # 选定日期
# print('可选日期数量为:{}'.format(len(date_list)))
for i in self.date:
j: WebElement = date_list[i-1]
toBeClicks.append(j)
break
# 选定场次('select_right_list_item')#选定场次
session_list = session.find_elements(
by=By.CLASS_NAME, value='bui-dm-sku-card-item')
print('可选场次数量为:{}'.format(len(session_list)))
for i in self.session: # 根据优先级选择一个可行场次
j: WebElement = session_list[i-1]
# TODO 不确定已满的场次带的是什么Tag
k = self.isClassPresent(j, 'item-text-tag', True)
if k: # 如果找到了带presell的类
if k.text == '无票':
continue
elif k.text == '预售':
toBeClicks.append(j)
break
elif k.text == '惠':
toBeClicks.append(j)
break
else:
# toBeClicks.append(j)
j.click()
break
price = WebDriverWait(self.driver, 1, 0.05).until(
EC.presence_of_element_located((By.CLASS_NAME, 'sku-tickets-card'))) # 日期、场次和票档进行定位
price_list = price.find_elements(
by=By.CLASS_NAME, value='bui-dm-sku-card-item') # 选定票档
print('可选票档数量为:{}'.format(len(price_list)))
for i in self.price:
j = price_list[i-1]
k = self.isClassPresent(j, 'item-tag')
if k: # 存在notticket代表存在缺货登记,跳过
continue
else:
#toBeClicks.append(j)
j.click()
break
buybutton = box.find_element(
by=By.CLASS_NAME, value='sku-footer-buy-button')
buybutton_text = buybutton.text
print("提示:{}".format(buybutton.text))
# sleep(1)
# for i in toBeClicks:
# i.click()
# sleep(0.05)
# WebDriverWait(self.driver, 1, 0.05).until(
# EC.presence_of_element_located((By.CLASS_NAME, 'bui-dm-sku-counter')))
except Exception as e:
raise Exception(f"***Error: 选择日期or场次or票档不成功***: {e}")
try:
ticket_num_up = box.find_element(
by=By.CLASS_NAME, value='plus-enable')
except:
if buybutton_text == "选座购买" or buybutton_text == "去选座": # 选座购买没有增减票数键
buybutton.click()
sleep(100)
self.status = 5
print(u"###请自行选择位置和票价###")
break
elif buybutton_text == "提交缺货登记":
raise Exception(u'###票已被抢完,持续捡漏中...或请关闭程序并手动提交缺货登记###')
else:
raise Exception(u"***Error: ticket_num_up 位置找不到***")
if buybutton_text == "立即预订" or buybutton_text == "立即购买" or buybutton_text == '确定':
for i in range(self.ticket_num-1): # 设置增加票数
ticket_num_up.click()
sleep(0.1)
buybutton.click()
sleep(0.5)
self.status = 4
WebDriverWait(self.driver, 1, 0.05).until(
EC.title_contains("确认"))
break
else:
raise Exception(f"未定义按钮:{buybutton_text}")
def check_order(self):
if self.status in [3, 4, 5]:
people = WebDriverWait(self.driver, 1, 0.05)\
.until(EC.presence_of_element_located((By.CLASS_NAME, 'viewer')))
people_list = people.find_elements(
by=By.CLASS_NAME, value='icondanxuan-weixuan_')
sleep(0.5)
for i in self.real_name:
j = people_list[i-1]
j.click()
comfirmBtn = self.driver.find_element(
By.XPATH, '//*[@id="dmOrderSubmitBlock_DmOrderSubmitBlock"]/div[2]/div/div[2]/div[3]/div[2]')
#print("提交按钮:"+comfirmBtn.text)
comfirmBtn.click()
# 判断title是不是支付宝
print(u"###等待跳转到--付款界面--,可自行刷新,若长期不跳转可选择-- CRTL+C --重新抢票###")
try:
WebDriverWait(self.driver, 3600, 0.05).until(
EC.title_contains('支付宝'))
except:
raise Exception(u'***Error: 长期跳转不到付款界面***')
self.status = 6
print(u'###成功提交订单,请手动支付###')
self.time_end = time()
if __name__ == '__main__':
try:
with open('./config.json', 'r', encoding='utf-8') as f:
config = loads(f.read())
# params: 场次优先级,票价优先级,实名者序号, 用户昵称, 购买票数, 官网网址, 目标网址, 浏览器驱动地址
con = Concert(config['date'], config['sess'], config['price'], config['real_name'], config['nick_name'],
config['ticket_num'], config['damai_url'], config['target_url'], config['driver_path'])
con.enter_concert() # 进入到具体抢购页面
except Exception as e:
print(e)
exit(1)
while True:
try:
con.choose_ticket()
con.check_order()
except Exception as e:
con.driver.get(con.target_url)
print(e)
continue
if con.status == 6:
print(u"###经过%d轮奋斗,共耗时%.1f秒,抢票成功!请确认订单信息###" %
(con.num, round(con.time_end-con.time_start, 3)))
break