main.py

# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from threadpool import SimpleThreadPool
import configparser
import time
import json


class App:
    def __init__(self, dotakey, chromedriver):
        self.dotakey = dotakey
        self.chromedriver = chromedriver
        self.driver = webdriver.Chrome(self.chromedriver)
        # self.phone_num = phone_num
        # self.passwd = passwd


    def login(self):
        """登陆模块"""

        self.driver.get('https://passport.damai.cn/login')
        # WebDriverWait(self.driver, 10).until(
        #     EC.presence_of_element_located((By.ID, 'alibaba-login-box')))
        # self.driver.switch_to.frame('alibaba-login-box')
        # self.driver.find_element_by_xpath('//*[@id="fm-login-id"]').send_keys(self.phone_num)
        # self.driver.find_element_by_xpath('//*[@id="fm-login-password"]').send_keys(self.passwd)


        str=''
        with open('cookie.json','r',encoding='utf-8') as f:
            listCookies=json.loads(f.read())
        cookie = [item["name"] + "=" + item["value"] for item in listCookies]
        cookiestr = '; '.join(item for item in cookie)
        print(cookiestr)
        driver.refresh()

        WebDriverWait(self.driver, 3000).until(
            EC.presence_of_element_located((By.XPATH, '//a[@data-spm="duserinfo"]/div')))
        print('登陆成功')
        user_name = self.driver.find_element_by_xpath('//a[@data-spm="duserinfo"]/div').text
        print('账号：', user_name)
        
        cookie = driver.get_cookies()
        jsonCookies = json.dumps(cookie)
        with open('cookie.json', 'w') as f:
            f.write(jsonCookies)


    def detail_page_auto(self, ticket_number=3):
        """详情页自动"""

        self.driver.get('https://detail.damai.cn/item.htm?id=593089517773')
        WebDriverWait(self.driver, 10).until(
            EC.presence_of_element_located((By.XPATH, '//button[@data-spm="dconfirm"]')))

        dconfirm_button = self.driver.find_element_by_xpath('//button[@data-spm="dconfirm"]')
        
        while dconfirm_button.get_attribute('class') == 'privilege_sub disabled':
            print(dconfirm_button.get_attribute('class'), '确定按钮无法点击，刷新页面')
            self.driver.refresh()
            WebDriverWait(self.driver, 10).until(
                EC.presence_of_element_located((By.XPATH, '//button[@data-spm="dconfirm"]')))
            try:
                dconfirm_button = self.driver.find_element_by_xpath('//button[@data-spm="dconfirm"]')
            except Exception as e:
                print('寻找按钮失败', e)

        self.driver.find_element_by_css_selector("#privilege_val").send_keys(self.dotakey)
        dconfirm_button.click()
        try:
            self.driver.find_element_by_xpath(
                '//a[@class="cafe-c-input-number-handler cafe-c-input-number-handler-up"]').click()
        except Exception as e:
            print("未成功点击+号", e)
        try:
            self.driver.find_element_by_xpath('//div[@class="select_right_list"]/div[{ticket_number}]/span[2]'.format(ticket_number=str(ticket_number))).click()
        except Exception as e:
            print("选择场次失败", e)
        dbuy_button = self.driver.find_element_by_xpath('//div[@data-spm="dbuy"]')
        print('寻找按钮:', dbuy_button.text)
        dbuy_button.click()

    def confirm_auto(self, name, phone):
        """自动确认订单"""

        title = self.driver.title
        while True:
            try:
                while title != '确认订单':
                    title = self.driver.title
                try:
                    self.driver.find_element_by_xpath(
                        '//*[@id="confirmOrder_1"]/div[2]/div[2]/div[1]/div[1]/label/span[1]/input').click()
                    time.sleep(0.5)
                    self.driver.find_element_by_xpath(
                        '//*[@id="confirmOrder_1"]/div[2]/div[2]/div[1]/div[2]/label/span[1]/input').click()
                except Exception as e:
                    raise ('购票人选择出错')
                time.sleep(0.5)
                self.driver.find_element_by_xpath('//div[@class="submit-wrapper"]/button').click()
                break
            except Exception as e:
                print(e)
                continue


def get_config(section, key):
    config = configparser.ConfigParser()
    config.read('config.ini', encoding='UTF-8')
    return config.get(section, key)

def work(ticket_number, driver_path):

    dotakey = get_config('info', 'privilege_val') 
    myapp = App(dotakey, driver_path)
    myapp.login()
    myapp.detail_page_auto(ticket_number)

if __name__ == '__main__': 
    _max_thread = 10
    pool = SimpleThreadPool(_max_thread)
    pool.add_task(work, 3, r"driver/chromedriver.exe")
    pool.add_task(work, 3, r"driver/chromedriver.1.exe")
    pool.add_task(work, 3, r"driver/chromedriver.2.exe")
    pool.wait_completion()