以selenium模拟登陆12306

 初级模拟，速度有点慢，后期有时间再优化

# -*- coding:utf-8 -*-
# author:zxy
# date:2018-12-23

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from http.cookiejar import CookieJar
from urllib import request
from urllib import parse


class login_12306():
    def get_opener(self):
        # 1登录
        # 1.1 创建一个CookieJar对象
        self.cookiejar = CookieJar()
        # 1.2 使用cookiejar创建一个HTTPCookieProcessor()对象
        self.handler = request.HTTPCookieProcessor(self.cookiejar)
        # 1.3 使用上一步创建的handler创建一个opener
        opener = request.build_opener(self.handler)
        return opener

class ticketGradding():

    def __init__(self):
        self.driver_path = r"E:chromDriverchromedriver.exe"
        self.initMy_url = "https://kyfw.12306.cn/otn/view/index.html"
        self.driver = webdriver.Chrome(executable_path=self.driver_path)
        self.login_url = "https://kyfw.12306.cn/otn/login/init"
        self.ticket_url = "https://kyfw.12306.cn/otn/leftTicket/init"
        self.confirm_url = "https://kyfw.12306.cn/otn/confirmPassenger/initDc"

    def _wait_input(self):
        self.from_station = input("起始站")
        self.to_station = input("目的地")

        # 时间格式  yyyy-MM-dd
        self.depart_time = input("出发时间：")
        self.passengers = input("乘客姓名(如有多个乘客，用英文逗号隔开)：").split(",")
        self.trains = input("车次(如果多个车次，用英文逗号隔开)：").split(",")

    def _login(self):  # 函数名前加_，表示该函数只能在当前类使用
        self.driver.get(self.login_url)
        # 显示等待
        # 隐示等待
        WebDriverWait(self.driver, 1000).until(
            EC.url_to_be(self.initMy_url)
        )
        print("登陆成功")

    def _order_ticket(self):
        # 跳转到查余票的界面
        self.driver.get(self.ticket_url)
        self.tickets = []
        # 等待出发地是否输入正确
        WebDriverWait(self.driver, 1000).until(

            # 元组
            EC.text_to_be_present_in_element_value((By.ID, "fromStationText"), self.from_station)
        )

        # 等待目的地时候输入正确
        WebDriverWait(self.driver, 1000).until(
            EC.text_to_be_present_in_element_value((By.ID, "toStationText"), self.to_station)
        )

        # 等待出发时间是否输入正确
        WebDriverWait(self.driver, 1000).until(
            EC.text_to_be_present_in_element_value((By.ID, "train_date"), self.depart_time)
        )

        # 等待查询按钮是否可用
        WebDriverWait(self.driver, 1000).until(
            EC.element_to_be_clickable((By.ID, 'query_ticket'))
        )

        # 执行查询事件
        searchBtn = self.driver.find_element_by_id('query_ticket')
        searchBtn.click()

        # 等待显示车次信息
        WebDriverWait(self.driver, 1000).until(
            EC.presence_of_element_located((By.XPATH, ".//tbody[@id='queryLeftTable']/tr"))
        )

        tr_list = self.driver.find_element_by_xpath(".//tbody[@id='queryLeftTable']/tr[not (@datatrain)]")

        for tr in tr_list:
            train_number = tr.find_element_by_class_name("number").text
            if train_number in self.trains:
                best_ticket = tr.find_element_by_xpath(".//td[2]").text
                first_ticket = tr.find_element_by_xpath(".//td[3]").text
                second_ticket = tr.find_element_by_xpath(".//td[4]").text
                super_soft_ticket = tr.find_element_by_xpath(".//td[5]").text
                soft_lay_ticket = tr.find_element_by_xpath(".//td[6]").text
                move_lay_ticket = tr.find_element_by_xpath(".//td[7]").text
                hard_lay_ticket = tr.find_element_by_xpath(".//td[8]").text
                sort_sit_ticket = tr.find_element_by_xpath(".//td[9]").text
                hard_sit_ticket = tr.find_element_by_xpath(".//td[10]").text
                no_sit_ticket = tr.find_element_by_xpath(".//td[11]").text
                others = tr.find_element_by_xpath(".//td[12]").text
                # self.tickets.append(best_ticket,first_ticket,second_ticket,super_soft_ticket,soft_lay_ticket,
                #                     move_lay_ticket,hard_lay_ticket,sort_sit_ticket,hard_sit_ticket,
                #                     no_sit_ticket,others)
                orderBtn = tr.find_element_by_class_name("btn72")
                if best_ticket.isdigit or best_ticket == "有":
                    print(train_number + "商务座特等座有票,序号为0")
                    # orderBtn.click()
                elif first_ticket.isdigit or first_ticket == "有":
                    print(train_number + "一等座有票，序号为1")
                    # orderBtn.click()
                elif second_ticket.isdigit or second_ticket == "有":
                    print(train_number + "二等座有票，序号为2")
                    # orderBtn.click()
                elif super_soft_ticket.isdigit or super_soft_ticket == "有":
                    print(train_number + "三等座有票，序号为3")
                    # orderBtn.click()
                elif soft_lay_ticket.isdigit or soft_lay_ticket == "有":
                    print(train_number + "高级软卧有票，序号为4")
                elif move_lay_ticket.isdigit or move_lay_ticket == "有":
                    print(train_number + "软卧有票，序号为5")
                elif hard_lay_ticket.isdigit or hard_lay_ticket == "有":
                    print(train_number + "动卧有票，序号为6")
                elif sort_sit_ticket.isdigit or sort_sit_ticket == "有":
                    print(train_number + "软座有票，序号为7")
                elif hard_sit_ticket.isdigit or hard_sit_ticket == "有":
                    print(train_number + "硬座有票，序号为8")
                elif no_sit_ticket.isdigit or no_sit_ticket == "有":
                    print(train_number + "无座有票，序号9")
                elif others.isdigit or others == "有":
                    print(train_number + "其他有票，序号10")

                choice = input("请输入有票的且座位等级对应的编号：")
                if (choice):
                    orderBtn.click()

                    # 确认选票页面
                    WebDriverWait(self.driver, 1000).until(
                        EC.url_to_be(self.confirm_url)
                    )

    def run(self):
        self._wait_input()
        self._login()
        self._order_ticket()


if __name__ == '__main__':
    spider = ticketGradding()
    spider.run()