基于selenium的12306抢票

喜欢ヅ旅行 2022-05-15 07:11 241阅读 0赞

from selenium import webdriver
    from selenium.webdriver.common.by import By
    from selenium.webdriver.support.ui import WebDriverWait
    from selenium.webdriver.support import expected_conditions as EC
    
    
    class Qiangpiao(object):
        def __init__(self):
            # 登录的url
            self.login_url = "https://kyfw.12306.cn/otn/login/init"
            # 登录成功后的url
            self.init_my_url = "https://kyfw.12306.cn/otn/index/initMy12306"
            # 查询余票界面
            self.search_url = "https://kyfw.12306.cn/otn/leftTicket/init"
            self.driver = webdriver.Chrome()
    
    
        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.init_my_url)
            )
            print("登陆成功")
    
        def _order_ticket(self):
            # 1.跳转到查余票的界面
            self.driver.get(self.search_url)
            # 2.等待出发地是否输入正确
            WebDriverWait(self.driver, 1000).until(
                EC.text_to_be_present_in_element_value((By.ID, "fromStationText"), self.from_station)
            )
            # 3.等待目的地是否输入正确
            WebDriverWait(self.driver, 1000).until(
                EC.text_to_be_present_in_element_value((By.ID, "toStationText"), self.to_station)
            )
            # 4.等待出发日期是否输入正确
            WebDriverWait(self.driver, 1000).until(
                EC.text_to_be_present_in_element_value((By.ID, "train_date"), self.depart_time)
            )
            # 5.等待查询日期是否可用
            WebDriverWait(self.driver, 1000).until(
                EC.element_to_be_clickable((By.ID, "query_ticket"))
            )
            # 6.如果能够被点击了，那么就找到这个查询按钮，执行点击事件
            searchBtn = self.driver.find_element_by_id("query_ticket")
            searchBtn.click()
    
        def run(self):
            self.wait_input()
            self._login()
            self._order_ticket()
    
    
    if __name__ == '__main__':
        spider = Qiangpiao()
        spider.run()