python+selenium实现自动抢票

使用说明

程序运行开始，需要输入出发地，目的地，出发时间，乘客信息，车次；乘客信息和车次可以输入多个
刚刚开始学习爬虫，selenium仅仅是解放了双手，运行效率不是很高；
程序运行时会打开chrome浏览器，因为使用的是chrome的浏览器驱动；

# encoding: utf-8

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
import time


class Qiangpiao(object):

    def __init__(self):
        driver_path = "E:\py_pachong\chromedriver.exe"
        # 浏览器驱动
        self.drive = webdriver.Chrome(executable_path=driver_path)
        # 登陆链接
        self.login_url = "https://kyfw.12306.cn/otn/login/init"
        # 登陆后的链接
        self.initmy_url = "https://kyfw.12306.cn/otn/index/initMy12306"
        # 查询页面
        self.search_url = "https://kyfw.12306.cn/otn/leftTicket/init"
        # 乘车人页面
        self.passenger_url = "https://kyfw.12306.cn/otn/confirmPassenger/initDc"

    def wait_input(self):
        """
        主要用于页面审核；
        当输入出发地，目的地，乘车时间后，会自动出发查询submit
        :return:
        """
        self.from_station = input("出发地：")
        self.to_station = input("目的地：")
        self.depart_time = input("出发时间：")
        self.passengers = input("乘客姓名：").split(",")
        self.trains = input("车次：").split(",")

    def _login(self):
        """
        登陆验证，登陆成功后会跳转到 self.initmy_url
        :return:
        """
        self.drive.get(self.login_url)
        WebDriverWait(self.drive, 1000).until(
            EC.url_to_be(self.initmy_url)
        )
        print('登录成功')

    def _order_ticket(self):
        """
        只实现了在无票的情况下去刷票
        最关键的是等待验证  WebDriverWait
        :return:
        """
        self.drive.get(self.search_url)
        WebDriverWait(self.drive, 1000).until(
            EC.text_to_be_present_in_element_value((By.ID, "fromStationText"), self.from_station)
        )

        WebDriverWait(self.drive, 1000).until(
            EC.text_to_be_present_in_element_value((By.ID, "toStationText"), self.to_station)
        )

        WebDriverWait(self.drive, 1000).until(
            EC.text_to_be_present_in_element_value((By.ID, "train_date"), self.depart_time)
        )

        WebDriverWait(self.drive, 10000).until(
            EC.element_to_be_clickable((By.ID, "query_ticket"))
        )

        searchBtn = self.drive.find_element_by_id("query_ticket")
        searchBtn.click()

        WebDriverWait(self.drive, 1000).until(
            EC.presence_of_element_located((By.XPATH, ".//tbody[@id='queryLeftTable']/tr"))
        )

        # find_elements_by_xpath 返回的是一个列表
        # find_element_by_xpath 返回的是一个元素
        tr_list = self.drive.find_elements_by_xpath(".//tbody[@id='queryLeftTable']/tr[not(@datatran)]")

        for tr in tr_list:
            train_num = tr.find_element_by_class_name("number").text
            # print(train_num)
            if train_num in self.trains:
                left_ticket_td = tr.find_element_by_xpath(".//td[4]").text
                num = 1
                while left_ticket_td == "无":
                    print("暂时无票，正在刷新")
                    time.sleep(2)
                    searchBtn.click()
                    num += 1
                    print("抢票%s次" % num)

                    if left_ticket_td != "无":
                        print(train_num + "有票")
                        oderBtn = tr.find_element_by_xpath(".//td[13]/a")
                        oderBtn.click()

                        WebDriverWait(self.drive, 1000).until(
                            EC.url_to_be(self.passenger_url)
                        )

                        WebDriverWait(self.drive, 1000).until(
                            EC.presence_of_element_located((By.XPATH, ".//ul[@id='normal_passenger_id']/li"))
                        )

                        passenger_labels = self.drive.find_elements_by_xpath(
                            ".//ul[@id='normal_passenger_id']/li/label")
                        for passenger_label in passenger_labels:
                            name = passenger_label.text
                            if name in self.passengers:
                                passenger_label.click()

                        submitBtn = self.drive.find_element_by_id("submitOrder_id")
                        submitBtn.click()

                        WebDriverWait(self.drive, 1000).until(
                            EC.presence_of_element_located((By.CLASS_NAME, "dhtmlx_wins_body_outer"))
                        )

                        WebDriverWait(self.drive, 1000).until(
                            EC.presence_of_element_located((By.ID, "qr_submit_id"))
                        )

                        qr_submit = self.drive.find_element_by_id("qr_submit_id")
                        qr_submit.click()

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


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

encoding: utf-8

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
import time

class Qiangpiao(object):

def __init__(self):
    driver_path = "E:\py_pachong\chromedriver.exe"
    # 浏览器驱动
    self.drive = webdriver.Chrome(executable_path=driver_path)
    # 登陆链接
    self.login_url = "https://kyfw.12306.cn/otn/login/init"
    # 登陆后的链接
    self.initmy_url = "https://kyfw.12306.cn/otn/index/initMy12306"
    # 查询页面
    self.search_url = "https://kyfw.12306.cn/otn/leftTicket/init"
    # 乘车人页面
    self.passenger_url = "https://kyfw.12306.cn/otn/confirmPassenger/initDc"

def wait_input(self):
    """
    主要用于页面审核；
    当输入出发地，目的地，乘车时间后，会自动出发查询submit
    :return:
    """
    self.from_station = input("出发地：")
    self.to_station = input("目的地：")
    self.depart_time = input("出发时间：")
    self.passengers = input("乘客姓名：").split(",")
    self.trains = input("车次：").split(",")

def _login(self):
    """
    登陆验证，登陆成功后会跳转到 self.initmy_url
    :return:
    """
    self.drive.get(self.login_url)
    WebDriverWait(self.drive, 1000).until(
        EC.url_to_be(self.initmy_url)
    )
    print('登录成功')

def _order_ticket(self):
    """
    只实现了在无票的情况下去刷票
    最关键的是等待验证  WebDriverWait
    :return:
    """
    self.drive.get(self.search_url)
    WebDriverWait(self.drive, 1000).until(
        EC.text_to_be_present_in_element_value((By.ID, "fromStationText"), self.from_station)
    )

    WebDriverWait(self.drive, 1000).until(
        EC.text_to_be_present_in_element_value((By.ID, "toStationText"), self.to_station)
    )

    WebDriverWait(self.drive, 1000).until(
        EC.text_to_be_present_in_element_value((By.ID, "train_date"), self.depart_time)
    )

    WebDriverWait(self.drive, 10000).until(
        EC.element_to_be_clickable((By.ID, "query_ticket"))
    )

    searchBtn = self.drive.find_element_by_id("query_ticket")
    searchBtn.click()

    WebDriverWait(self.drive, 1000).until(
        EC.presence_of_element_located((By.XPATH, ".//tbody[@id='queryLeftTable']/tr"))
    )

    # find_elements_by_xpath 返回的是一个列表
    # find_element_by_xpath 返回的是一个元素
    tr_list = self.drive.find_elements_by_xpath(".//tbody[@id='queryLeftTable']/tr[not(@datatran)]")

    for tr in tr_list:
        train_num = tr.find_element_by_class_name("number").text
        # print(train_num)
        if train_num in self.trains:
            left_ticket_td = tr.find_element_by_xpath(".//td[4]").text
            num = 1
            while left_ticket_td == "无":
                print("暂时无票，正在刷新")
                time.sleep(2)
                searchBtn.click()
                num += 1
                print("抢票%s次" % num)

                if left_ticket_td != "无":
                    print(train_num + "有票")
                    oderBtn = tr.find_element_by_xpath(".//td[13]/a")
                    oderBtn.click()

                    WebDriverWait(self.drive, 1000).until(
                        EC.url_to_be(self.passenger_url)
                    )

                    WebDriverWait(self.drive, 1000).until(
                        EC.presence_of_element_located((By.XPATH, ".//ul[@id='normal_passenger_id']/li"))
                    )

                    passenger_labels = self.drive.find_elements_by_xpath(
                        ".//ul[@id='normal_passenger_id']/li/label")
                    for passenger_label in passenger_labels:
                        name = passenger_label.text
                        if name in self.passengers:
                            passenger_label.click()

                    submitBtn = self.drive.find_element_by_id("submitOrder_id")
                    submitBtn.click()

                    WebDriverWait(self.drive, 1000).until(
                        EC.presence_of_element_located((By.CLASS_NAME, "dhtmlx_wins_body_outer"))
                    )

                    WebDriverWait(self.drive, 1000).until(
                        EC.presence_of_element_located((By.ID, "qr_submit_id"))
                    )

                    qr_submit = self.drive.find_element_by_id("qr_submit_id")
                    qr_submit.click()

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

if name == 'main':
spider = Qiangpiao()
spider.run()