import time
import datetime
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
class Qiangpiao():
? ? def __init__(self,from_station,to_station,depart_time,train_num,passenger):
? ? ? ? self.login_url = 'https://kyfw.12306.cn/otn/resources/login.html'
? ? ? ? self.init_my_url = 'https://kyfw.12306.cn/otn/view/index.html'
? ? ? ? self.order_url = 'https://kyfw.12306.cn/otn/confirmPassenger/initDc'
? ? ? ? # input("出發(fā)地：")
? ? ? ? self.from_station = from_station
? ? ? ? # input("目的地：")
? ? ? ? self.to_station = to_station
? ? ? ? # 時間格式必須是M-d的方式
? ? ? ? # input("出發(fā)時間(格式必須是M-d的方式)：")
? ? ? ? self.depart_time = depart_time
? ? ? ? # input("列車號：")
? ? ? ? self.train_num = train_num
? ? ? ? self.passenger = passenger
? ? ? ? #獲取當(dāng)前月份
? ? ? ? self.now_month = datetime.date.today().month

? ? ? ? self.leave_month = int(self.depart_time.split('-')[0])
? ? ? ? self.leave_day = int(self.depart_time.split('-')[1])

? ? ? ? self.driver = webdriver.Chrome()

? ? def _login(self):
? ? ? ? self.driver.get(self.login_url)
? ? ? ? # 窗口最大化
? ? ? ? #self.driver.maximize_window()
? ? ? ? # 設(shè)置窗口大小
? ? ? ? self.driver.set_window_size(1300,800)
? ? ? ? #print('調(diào)整前尺寸:', self.driver.get_window_size())
? ? ? ? #顯式等待
? ? ? ? #這里進(jìn)行手動登錄，可以掃碼，也可以輸入賬號密碼點(diǎn)擊登錄
? ? ? ? WebDriverWait(self.driver,1000).until(EC.url_to_be(self.init_my_url))
? ? ? ? print('登錄成功！')

? ? def _pop_window(self):
? ? ? ? time.sleep(1)
? ? ? ? self.driver.find_element_by_xpath('//*[@class="dzp-confirm"]/div[2]/div[3]/a').click()

? ? def _enter_order_ticket(self):
? ? ? ? action = ActionChains(self.driver) ? # 實(shí)例化一個動作鏈對象
? ? ? ? element = self.driver.find_element_by_link_text('車票')
? ? ? ? # 鼠標(biāo)移動到 '車票' 元素上的中心點(diǎn)
? ? ? ? action.move_to_element(element).perform()
? ? ? ? # 點(diǎn)擊'單程'
? ? ? ? self.driver.find_element_by_xpath('//*[@id="J-chepiao"]/div/div[1]/ul/li[1]/a').click()
? ? ? ? # 消除第二次彈窗
? ? ? ? self.driver.find_element_by_link_text('確認(rèn)').click()

? ? def _search_ticket(self):
? ? ? ? #出發(fā)地輸入
? ? ? ? self.driver.find_element_by_id("fromStationText").click()
? ? ? ? self.driver.find_element_by_id("fromStationText").send_keys(self.from_station)
? ? ? ? self.driver.find_element_by_id("fromStationText").send_keys(Keys.ENTER)
? ? ? ? #目的地輸入
? ? ? ? self.driver.find_element_by_id("toStationText").click()
? ? ? ? self.driver.find_element_by_id("toStationText").send_keys(self.to_station)
? ? ? ? self.driver.find_element_by_id("toStationText").send_keys(Keys.ENTER)
? ? ? ? #出發(fā)日期輸入
? ? ? ? self.driver.find_element_by_id("date_icon_1").click()
? ? ? ? if self.leave_month == self.now_month:
? ? ? ? ? ? xpath_str = f"/html/body/div[37]/div[1]/div[2]/div[{self.leave_day}]"
? ? ? ? ? ? if EC.element_to_be_clickable((By.XPATH, xpath_str)):
? ? ? ? ? ? ? ? self.driver.find_element_by_xpath(xpath_str).click()
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? print("當(dāng)前日期未到或已過售票日期，無法購票！")
? ? ? ? elif self.leave_month == self.now_month + 1:
? ? ? ? ? ? xpath_str = f"/html/body/div[37]/div[2]/div[2]/div[{self.leave_day}]"
? ? ? ? ? ? if EC.element_to_be_clickable((By.XPATH, xpath_str)):
? ? ? ? ? ? ? ? self.driver.find_element_by_xpath(xpath_str).click()
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? print("當(dāng)前日期未到或已過售票日期，無法購票！")
? ? ? ? else:
? ? ? ? ? ? print("月份超前一個月以上，無法購票！")
? ? ? ? #等待查詢按鈕是否可用
? ? ? ? WebDriverWait(self.driver,1000).until(EC.element_to_be_clickable((By.ID,"query_ticket")))
? ? ? ? #執(zhí)行點(diǎn)擊事件
? ? ? ? search_btn = self.driver.find_element_by_id("query_ticket")
? ? ? ? search_btn.click()
? ? ? ? #等待查票信息加載
? ? ? ? WebDriverWait(self.driver, 1000).until(EC.presence_of_element_located((By.XPATH, '//*[@id="queryLeftTable"]/tr')))

? ? def _order_ticket(self):
? ? ? ? train_num_list = [] ?# 列車號列表
? ? ? ? train_num_ele_list = self.driver.find_elements_by_xpath('//tr/td[1]/div/div[1]/div/a') ?# 列車號元素列表
? ? ? ? for t in train_num_ele_list: ? ?# 遍歷列車號元素列表，并把列車號添加到列車號列表
? ? ? ? ? ? train_num_list.append(t.text)
? ? ? ? tr_list = self.driver.find_elements_by_xpath('//*[@id="queryLeftTable"]/tr[not(@datatran)]') ?#每一列列車整行信息列表，列車號元素是tr的子元素
? ? ? ? if self.train_num in train_num_list:
? ? ? ? ? ? for tr in tr_list:
? ? ? ? ? ? ? ? train_num = tr.find_element_by_xpath("./td[1]/div/div[1]/div/a").text ?#取出元素tr里的列車號
? ? ? ? ? ? ? ? if self.train_num == train_num:
? ? ? ? ? ? ? ? ? ? #動車二等座余票信息
? ? ? ? ? ? ? ? ? ? text_1 = tr.find_element_by_xpath("./td[4]").text
? ? ? ? ? ? ? ? ? ? # 火車二等座余票信息
? ? ? ? ? ? ? ? ? ? text_2 = tr.find_element_by_xpath("./td[8]").text
? ? ? ? ? ? ? ? ? ? if (text_1 == "有" or text_1.isdigit()) or (text_2 == "有" or text_2.isdigit()):
? ? ? ? ? ? ? ? ? ? ? ? #點(diǎn)擊預(yù)訂按鈕
? ? ? ? ? ? ? ? ? ? ? ? order_btn = tr.find_element_by_class_name("btn72")
? ? ? ? ? ? ? ? ? ? ? ? order_btn.click()
? ? ? ? ? ? ? ? ? ? ? ? #等待訂票頁面
? ? ? ? ? ? ? ? ? ? ? ? WebDriverWait(self.driver,1000).until(EC.url_to_be(self.order_url))
? ? ? ? ? ? ? ? ? ? ? ? # 選定乘車人
? ? ? ? ? ? ? ? ? ? ? ? self.driver.find_element_by_xpath(f'//*[@id="normal_passenger_id"]/li/label[contains(text(),"{self.passenger}")]').click()
? ? ? ? ? ? ? ? ? ? ? ? #如果乘客是學(xué)生，對提示點(diǎn)擊確定
? ? ? ? ? ? ? ? ? ? ? ? if EC.presence_of_element_located((By.XPATH, '//div[@id="dialog_xsertcj"]')):
? ? ? ? ? ? ? ? ? ? ? ? ? ? self.driver.find_element_by_id('dialog_xsertcj_ok').click()
? ? ? ? ? ? ? ? ? ? ? ? ? ? # 提交訂單
? ? ? ? ? ? ? ? ? ? ? ? ? ? self.driver.find_element_by_id('submitOrder_id').click()
? ? ? ? ? ? ? ? ? ? ? ? ? ? time.sleep(2)
? ? ? ? ? ? ? ? ? ? ? ? ? ? # 點(diǎn)擊確認(rèn)訂單
? ? ? ? ? ? ? ? ? ? ? ? ? ? self.driver.find_element_by_id('qr_submit_id').click()
? ? ? ? ? ? ? ? ? ? ? ? else:
? ? ? ? ? ? ? ? ? ? ? ? ? ? # 提交訂單
? ? ? ? ? ? ? ? ? ? ? ? ? ? self.driver.find_element_by_id('submitOrder_id').click()
? ? ? ? ? ? ? ? ? ? ? ? ? ? time.sleep(2)
? ? ? ? ? ? ? ? ? ? ? ? ? ? # 點(diǎn)擊確認(rèn)
? ? ? ? ? ? ? ? ? ? ? ? ? ? self.driver.find_element_by_id('qr_submit_id').click()
? ? ? ? ? ? ? ? ? ? ? ? ? ? print("購票成功！")
? ? ? ? ? ? ? ? ? ? ? ? ? ? break
? ? ? ? ? ? ? ? ? ? else:
? ? ? ? ? ? ? ? ? ? ? ? print("二等座無票！")
? ? ? ? else:
? ? ? ? ? ? print("無此列車！")

? ? def run(self):
? ? ? ? #登錄
? ? ? ? self._login()
? ? ? ? #消除登錄后（第一次）的彈窗
? ? ? ? self._pop_window()
? ? ? ? #進(jìn)入購票頁面
? ? ? ? self._enter_order_ticket()
? ? ? ? #查票
? ? ? ? self._search_ticket()
? ? ? ? #訂票
? ? ? ? self._order_ticket()
? ? ? ? #關(guān)閉瀏覽器
? ? ? ? time.sleep(6)
? ? ? ? self.driver.quit()

if __name__ == '__main__':
? ? qiangpiao = Qiangpiao("蘭州","烏魯木齊","8-6","D55","小紅")
? ? qiangpiao.run()