'''
實現(xiàn)自動搶火車票(基于Python3.6+splinter)
Created on 2018年1月15日
@author: liuyazhuang
'''
from splinter.browser import Browser
from time import sleep
import traceback
 
#實現(xiàn)自動購票的類
class Buy_Tickets(object):
  # 定義實例屬性，初始化
  def __init__(self, username, passwd, order, passengers, dtime, starts, ends):
    self.username = username
    self.passwd = passwd
    # 車次，0代表所有車次，依次從上到下，1代表所有車次，依次類推
    self.order = order
    # 乘客名
    self.passengers = passengers
    # 起始地和終點
    self.starts = starts
    self.ends = ends
    # 日期
    self.dtime = dtime
    # self.xb = xb
    # self.pz = pz
    self.login_url = 'https://kyfw.12306.cn/otn/login/init'
    self.initMy_url = 'https://kyfw.12306.cn/otn/index/initMy12306'
    self.ticket_url = 'https://kyfw.12306.cn/otn/leftTicket/init'
    self.driver_name = 'chrome'
    self.executable_path = 'C:\Python36\Scripts\chromedriver.exe'
  # 登錄功能實現(xiàn)
  def login(self):
    self.driver.visit(self.login_url)
    self.driver.fill('loginUserDTO.user_name', self.username)
    # sleep(1)
    self.driver.fill('userDTO.password', self.passwd)
    # sleep(1)
    print('請輸入驗證碼...')
    while True:
      if self.driver.url != self.initMy_url:
        sleep(1)
      else:
        break
  # 買票功能實現(xiàn)
  def start_buy(self):
    self.driver = Browser(driver_name=self.driver_name, executable_path=self.executable_path)
    #窗口大小的操作
    self.driver.driver.set_window_size(700, 500)
    self.login()
    self.driver.visit(self.ticket_url)
    try:
      print('開始購票...')
      # 加載查詢信息
      self.driver.cookies.add({"_jc_save_fromStation": self.starts})
      self.driver.cookies.add({"_jc_save_toStation": self.ends})
      self.driver.cookies.add({"_jc_save_fromDate": self.dtime})
      self.driver.reload()
      count = 0
      if self.order != 0:
        while self.driver.url == self.ticket_url:
          self.driver.find_by_text('查詢').click()
          count += 1
          print('第%d次點擊查詢...' % count)
          try:
            self.driver.find_by_text('預(yù)訂')[self.order-1].click()
            sleep(1.5)
          except Exception as e:
            print(e)
            print('預(yù)訂失敗...')
            continue
      else:
        while self.driver.url == self.ticket_url:
          self.driver.find_by_text('查詢').click()
          count += 1
          print('第%d次點擊查詢...' % count)
          try:
            for i in self.driver.find_by_text('預(yù)訂'):
              i.click()
              sleep(1)
          except Exception as e:
            print(e)
            print('預(yù)訂失敗...')
            continue
      print('開始預(yù)訂...')
      sleep(1)
      print('開始選擇用戶...')
      for p in self.passengers:
 
        self.driver.find_by_text(p).last.click()
        sleep(0.5)
        if p[-1] == ')':
          self.driver.find_by_id('dialog_xsertcj_ok').click()
      print('提交訂單...')
      # sleep(1)
      # self.driver.find_by_text(self.pz).click()
      # sleep(1)
      # self.driver.find_by_text(self.xb).click()
      # sleep(1)
      self.driver.find_by_id('submitOrder_id').click()
      sleep(2)
      print('確認選座...')
      self.driver.find_by_id('qr_submit_id').click()
      print('預(yù)訂成功...')
    except Exception as e:
      print(e)
 
if __name__ == '__main__':
  # 用戶名
  username = 'xxxx'
  # 密碼
  password = 'xxx'
  # 車次選擇，0代表所有車次
  order = 2
  # 乘客名，比如passengers = ['XXX', 'XXX']
  # 學(xué)生票需注明，注明方式為：passengers = ['XXX(學(xué)生)', 'XXX']
  passengers = ['劉亞壯']
  # 日期，格式為：'2018-01-20'
  dtime = '2018-01-20'
  # 出發(fā)地(需填寫cookie值)
  starts = '%u6210%u90FD%2CCDW' #成都
  # 目的地(需填寫cookie值)
  ends = '%%u5317%u4EAC%u897F%2CBXP' #北京
 
  # xb =['硬座座'] 
  # pz=['成人票']
 
Buy_Tickets(username, password, order, passengers, dtime, starts, ends).start_buy()