# -*- coding: utf-8 -*-

from splinter.browser import Browser

from time import sleep

import traceback

#初始化信息

# 用户名，密码

username = u"用户名"

passwd = u"密码"

# cookies值得自己去找

starts = u"杭州,HZH"

ends = u"黄石,HSN"

# 时间格式2016-03-22

dtime = u"2016-03-22"

# 车次，选择第几趟，0则从上之下依次点击

order = 0

###乘客名,如果是自己，则上面有个链接是自己的，会跳转到个人中心,用index[1]

#如果乘客不是自己，则用index[0]

pa = u"名字"

"""网址"""

ticket_url = "https://kyfw.12306.cn/otn/leftTicket/init"

login_url = "https://kyfw.12306.cn/otn/login/init"

initmy_url = "https://kyfw.12306.cn/otn/index/initMy12306"

def login():

#点击登录，从购票页面调到了登录页面

b.find_by_text(u"登录").click()

sleep(3)

#给name为loginUserDTO.user_name附上用户名

#也可以用Id查找的方法b.find_by_id('username').fill(username)

b.fill("loginUserDTO.user_name", username)

sleep(1)

#给name为userDTO.password值

b.fill("userDTO.password", passwd)

sleep(1)

#验证码必须手动输入

print u"等待验证码，自行输入..."

#检测页面是否跳转(用户是否输入了验证码并且提交)

while True:

if b.url != initmy_url:

sleep(1)

else:

break

def huoche():

global b

b = Browser(driver_name="chrome")

#访问12306

b.visit(ticket_url)

#查找带登录的字符串，u是避免中文报错

while b.is_text_present(u"登录"):

sleep(1)

#如果存在调用登录方法

login()

if b.url == initmy_url:

break

#当执行完login方法后，登录成功，继续向下执行

try:

print u"购票页面..."

# 跳回购票页面

b.visit(ticket_url)

# 加载查询信息

b.cookies.add({"_jc_save_fromStation": starts})

b.cookies.add({"_jc_save_toStation": ends})

b.cookies.add({"_jc_save_fromDate": dtime})

b.reload()

sleep(2)

count = 0

# 循环点击预订

#order=0则从第一班车开始全部依次预订

if order != 0:

#如果页面没有跳转到，说明没有抢到票，继续抢票

while b.url == ticket_url:

#点击查询进行刷新

b.find_by_text(u"查询").click()

count +=1

print u"循环点击查询... 第 %s 次" % count

sleep(1)

try:

#点击预订的那班车

b.find_by_text(u"预订")[order - 1].click()

except:

print u"还没开始预订"

continue

else:

while b.url == ticket_url:

b.find_by_text(u"查询").click()

count += 1

print u"循环点击查询... 第 %s 次" % count

sleep(1)

try:

#从第一个开始依次预订

for i in b.find_by_text(u"预订"):

i.click()

except:

print u"还没开始预订"

continue

sleep(1)

#若循环结束了，则选择联系人

b.find_by_text(pa)[0].click()

print u"能做的都做了.....不再对浏览器进行任何操作"

except Exception as e:

print(traceback.print_exc())

if __name__ == "__main__":

huoche()