基于python的大麦网自动抢票工具

# 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.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, session, price, real_name, nick_name, ticket_num, damai_url, target_url): self.session = session # 场次序号优先级 self.price = price # 票价序号优先级 self.real_name = real_name # 实名者序号 self.status = 0 # 状态标记 self.time_start = 0 # 开始时间 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 = 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 def get_cookie(self): self.driver.get(self.damai_url) print(u"###请点击登录###") while self.driver.title.find('大麦网-全球演出赛事官方购票平台') != -1: # 等待网页加载完成 sleep(1) print(u"###请扫码登录###") while self.driver.title == '大麦登录': # 等待扫码完成 sleep(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.1).until(EC.title_contains('大麦网')) self.set_cookie() def enter_concert(self): print(u'###打开浏览器，进入大麦网###') if not exists('cookies.pkl'): # 如果不存在cookie.pkl,就获取一下 self.driver = webdriver.Chrome() 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} options.add_experimental_option("prefs", prefs) # 更换等待策略为不等待浏览器加载完全就进行下一步操作 capa = DesiredCapabilities.CHROME capa["pageLoadStrategy"] = "none" self.driver = webdriver.Chrome(options=options, desired_capabilities=capa) self.login() self.driver.refresh() try: 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 choose_ticket(self): print(u"###进入抢票界面###") while self.driver.title.find('确认订单') == -1: # 如果跳转到了确认界面就算这步成功了，否则继续执行此步 self.num += 1 if con.driver.current_url.find("buy.damai.cn") != -1: break # 确认页面刷新成功 try: #box = self.driver.find_element_by_class_name('perform__order__box') box = WebDriverWait(self.driver, 1, 0.1).until(EC.presence_of_element_located((By.CLASS_NAME, 'perform__order__box'))) except: raise Exception(u"***Error: 页面刷新出错***") try: buybutton = box.find_element_by_class_name('buybtn') buybutton_text = buybutton.text except: raise Exception(u"***Error: buybutton 位置找不到***") if buybutton_text == "即将开抢" or buybutton_text == "即将开售": self.status = 2 raise Exception(u"---尚未开售，刷新等待---") try: selects = box.find_elements_by_class_name('perform__order__select') for item in selects: if item.find_element_by_class_name('select_left').text == '场次': session = item # print('\t场次定位成功') elif item.find_element_by_class_name('select_left').text == '票档': price = item # print('\t票档定位成功') session_list = session.find_elements_by_class_name('select_right_list_item') # print('可选场次数量为：{}'.format(len(session_list))) for i in self.session: # 根据优先级选择一个可行场次 j = session_list[i-1] k = self.isClassPresent(j, 'presell', True) if k: # 如果找到了带presell的类 if k.text == '无票': continue elif k.text == '预售': j.click() break else: j.click() break price_list = price.find_elements_by_class_name('select_right_list_item') # print('可选票档数量为：{}'.format(len(price_list))) for i in self.price: j = price_list[i-1] k = self.isClassPresent(j, 'notticket') if k: # 存在notticket代表存在缺货登记，跳过 continue else: j.click() break except: raise Exception(u"***Error: 选择场次or票档不成功***") try: ticket_num_up = box.find_element_by_class_name('cafe-c-input-number-handler-up') except: if buybutton_text == "选座购买": # 选座购买没有增减票数键 buybutton.click() self.status = 5 print(u"###请自行选择位置和票价###") break elif buybutton_text == "提交缺货登记": raise Exception(u'###票已被抢完，持续捡漏中...或请关闭程序并手动提交缺货登记###') else: raise Exception(u"***Error: ticket_num_up 位置找不到***") if buybutton_text == "立即预订": for i in range(self.ticket_num-1): # 设置增加票数 ticket_num_up.click()