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

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()