def set_cookie(self):

self.driver.get(damai_url)

print("###请点击登录###")

while self.driver.title.find('大麦网-全球演出赛事官方购票平台') != -1:

sleep(1)

print('###请扫码登录###')

while self.driver.title != '大麦网-全球演出赛事官方购票平台-100%正品、先付先抢、在线选座！':

sleep(1)

print("###扫码成功###")

pickle.dump(self.driver.get_cookies(), open("cookies.pkl", "wb"))

print("###Cookie保存成功###")

self.driver.get(target_url)

获取cookie

def get_cookie(self):

try:

cookies = pickle.load(open("cookies.pkl", "rb")) # 载入cookie

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###')

except Exception as e:

print(e)

登录

def login(self):

if self.login_method==0:

self.driver.get(login_url)

# 载入登录界面

print('###开始登录###')

elif self.login_method==1:

if not os.path.exists('cookies.pkl'):

# 如果不存在cookie.pkl,就获取一下

self.set_cookie()

else:

self.driver.get(target_url)

self.get_cookie()

打开浏览器

def enter_concert(self):

"""打开浏览器"""

print('###打开浏览器，进入大麦网###')

# self.driver.maximize_window() # 最大化窗口

# 调用登陆

self.login() # 先登录再说

self.driver.refresh() # 刷新页面

self.status = 2 # 登录成功标识

print("###登录成功###")

# 后续德云社可以讲

if self.isElementExist('/html/body/div[2]/div[2]/div/div/div[3]/div[2]'):

self.driver.find_element_by_xpath('/html/body/div[2]/div[2]/div/div/div[3]/div[2]').click()

第二步，抢票并下单

判断元素是否存在

def isElementExist(self, element):

flag = True

browser = self.driver

try:

browser.find_element_by_xpath(element)

return flag

except:

flag = False

return flag

选票操作

def choose_ticket(self):

if self.status == 2: #登录成功入口

print("="*30)

print("###开始进行日期及票价选择###")

while self.driver.title.find('确认订单') == -1: # 如果跳转到了订单结算界面就算这步成功了，否则继续执行此步

try:

buybutton = self.driver.find_element_by_class_name('buybtn').text

if buybutton == "提交缺货登记":

# 改变现有状态

self.status=2

self.driver.get(target_url)

print('###抢票未开始，刷新等待开始###')

continue

elif buybutton == "立即预定":

self.driver.find_element_by_class_name('buybtn').click()

# 改变现有状态

self.status = 3

elif buybutton == "立即购买":

self.driver.find_element_by_class_name('buybtn').click()

# 改变现有状态

self.status = 4

# 选座购买暂时无法完成自动化

elif buybutton == "选座购买":

self.driver.find_element_by_class_name('buybtn').click()

self.status = 5

except:

print('###未跳转到订单结算界面###')

title = self.driver.title

if title == '选座购买':

# 实现选座位购买的逻辑

self.choice_seats()

elif title == '确认订单':

while True:

# 如果标题为确认订单

print('waiting ......')

if self.isElementExist('//*[@id="container"]/div/div[9]/button'):code>

self.check_order()

break

选择座位