ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

带有界面的12306!无限自动查询并购票的脚本!年关买票了吗

2020-12-24 14:32:26  阅读:252  来源: 互联网

标签:12306 self driver 购票 车次 station str 买票 id


分享记录一个带有GUI界面的12306(默认二等座)无限自动查询并购票的脚本(购票成功发送邮件)

from tkinter import * #编写GUI界面
import threading  #引入线程,解决GUI堵塞
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 #后面的until()必须元组形式,所以导入By
#导入csv模块来读取站点代号
import  csv
#导入表单下滑选项操作的库
from selenium.webdriver.support.ui import Select
#导入可能出现的异常
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import ElementNotVisibleException
from selenium.common.exceptions import StaleElementReferenceException
from selenium.common.exceptions import ElementNotInteractableException
#导入时间模块进行等待
from time import sleep
#导入发送邮件模块
import yagmail


"""将driver放在外面的原因是:
   如果放在里面,那么driver将会随着对象的销毁而销毁
   而我们的类TrainSpider的实例对象是放在main()函数中执行的
   只要main()函数运行完成后,里面所有的变量都会被销毁
   也就说spider类实例对象也会被销毁
"""

#初始化GUI界面
root = Tk()
root.title('Carson的12306购票器')
root.geometry('400x350')
#初始化基本GUI界面的组件
lb1 = Label(root,text = '欢迎使用Carson的12306二等座购票器',font=('Arial', 14))
lb1.place(relx = 0.08,rely=0.02)
lb2 = Label(root,text = '乘车人员:',font=('Arial', 12))
lb2.place(x = 45,y = 45)
lb3 = Label(root,text = '出发日期:',font=('Arial', 12))
lb3.place(x = 45,y = 78)
lb4 = Label(root,text = '出发车站:',font=('Arial', 12))
lb4.place(x = 45,y = 111)
lb5 = Label(root,text = '终点车站:',font=('Arial', 12))
lb5.place(x = 45,y = 144)
lb6 = Label(root,text = '购买车次:',font=('Arial', 12))
lb6.place(x = 45,y = 177)
lb7 = Label(root,text = '购票信息如下:',font=('Arial', 12))
lb7.place(x = 0,y = 205)
text = Text(root,height = 5,width=56)
text.place(x = 0,y= 232)
entry1_str = StringVar()
entry1_str.set('输入乘车人的姓名,如:张三')
entry1 = Entry(root,textvariable = entry1_str,)
entry1.place(x = 120,y = 46,height=28,width=160)
entry2_str = StringVar()
entry2_str.set('输入出发日期,格式如:2021-01-16')
entry2 = Entry(root,textvariable = entry2_str,)
entry2.place(x = 120,y = 79,height=28,width=190)
entry3_str = StringVar()
entry3_str.set('输入起始站,如:深圳北')
entry3 = Entry(root,textvariable = entry3_str)
entry3.place(x = 120,y = 112,height=28,width=160)
entry4_str = StringVar()
entry4_str.set('输入终点站,如:潮阳')
entry4 = Entry(root,textvariable = entry4_str)
entry4.place(x = 120,y = 145,height=28,width=160)
entry5_str = StringVar()
entry5_str.set('输入车次,格式如:G6006 D1234')
entry5 = Entry(root,textvariable = entry5_str)
entry5.place(x = 120,y = 178,height=28,width=180)
class TrainSpider:  
    #将属性放类里面定义为类属性
    login_url = 'https://kyfw.12306.cn/otn/resources/login.html' #二维码登陆界面url
    personal_url = 'https://kyfw.12306.cn/otn/view/index.html'  #登陆后进入的个人中心url
    left_ticket_url = 'https://kyfw.12306.cn/otn/leftTicket/init?linktypeid=dc' #查询车次和余票的url
    confirm_passenger_url = 'https://kyfw.12306.cn/otn/confirmPassenger/initDc' #确认乘客信息的url
    
    def __init__(self,from_station,to_station,train_date,trains,passengers,driver):
        """
        :param from_station:  起始车站
        :param to_station: 目的地车站
        :param train_date: 出发日期
        :param trains: 需要购买的车次。需要字典形式,形式如:{“G529”:["O","M"],"G403":["O","M"]}多车次就多个键值对
        :param passengers: 需要买票的乘车人,需要为一个列表
        """
        self.driver = driver
        self.from_station = from_station
        self.to_station = to_station
        self.train_date = train_date
        self.trains = trains
        self.passengers = passengers
        self.current_number = None #定义一下变量保存下当前预定的车次序号信息
        self.current_seat = None #定义一下变量保存下当前选中的车次的选中的席位信息
        #为了方便根据站名文字来取得车站代号,需要创建字典存储代号数据
        #且空集合的创建要在函数外,不然执行完函数集合数据就没有了
        self.station_codes = {}
        #初始化站点代号数据
        self.get_station_codes()


    #获取车站代码
    #这里需要本地有一个station.csv的车站对应代码的文件
    def get_station_codes(self):
        #读取数据并存放到空字典中
        with open('stations.csv', 'r', encoding='utf-8') as fp:
            reader = csv.DictReader(fp)
            for line in reader:
                name = line['name']
                code = line['code']
                self.station_codes[name] = code


    #实现登陆功能
    def login(self):
        self.driver.maximize_window() #最大化窗口
        # 将属性放类里面定义为类属性,调用时需要加self进行调用
        self.driver.get(self.login_url)
        # 进行显式等待(有条件)设置100秒,且用来判断是否登陆成功
        # 即后面判断条件是url是否变化成个人中心的url
        WebDriverWait(self.driver, 100).until(
            EC.url_to_be(self.personal_url)  # 注意类中变量调用加self
            #或者EC.url_contains(self.personal_url)
        )
        print('登陆成功!')
        print('开始刷票!')


    #查询车次余票
    def search_left_tickets(self):
        self.driver.get(self.left_ticket_url)
        """起始站的代号设置"""
        from_station_input = self.driver.find_element_by_id('fromStation')
        #利用用户输入文字获取起始站的代号
        from_station_code = self.station_codes[self.from_station]
        #通过js代码修改隐藏标签的value值来达到输入起始站的目的
        self.driver.execute_script("arguments[0].value = '%s'"%from_station_code,from_station_input)
        """终点站的代号设置"""
        to_station_input = self.driver.find_element_by_id('toStation')
        to_station_code = self.station_codes[self.to_station]
        self.driver.execute_script("arguments[0].value = '%s'"%to_station_code,to_station_input)
        """日期设置"""
        #这里没有前面两个复杂,没有被隐含,理论上标签send_keys即可
        #但可能也像前面两个输入框一样被处理过,故输入时间也才用执行js代码的方式
        train_date_input = self.driver.find_element_by_xpath('//*[@id="train_date"]') #xpath*表任意
        self.driver.execute_script("arguments[0].value = '%s'" % self.train_date, train_date_input)
        """执行查询操作"""
        search_button = self.driver.find_element_by_id('query_ticket').click()
        print("第1次查询中...")
        # 因为点击查询按钮后需等待一下才会返回列车车次数据
        # 所以在解析具体的车次信息前需要设置等待,采用显示等待(条件即加载出tbody下的tr标签)
        """注意,在until(EC.presence_of_element_located())中
        验证元素是否出现,传入的参数必须都是元组类型的locator,如(By.ID, ‘kw’),
        不能传入webelement对象,即driver.find_elemet_by_id()的写法不行会报错
        """
        # 设置1000秒显式等待有各个列车信息的tr标签出现(一般开售前5分种即5*60=300秒足够了)
        WebDriverWait(self.driver, 1000).until(
            # 条件判断是某元素即tr标签是否出现,注意..located()里面是元组类型的locator,必须如下写法
            EC.presence_of_element_located((By.XPATH, '//*[@id="queryLeftTable"]/tr'))
        )
        # 注意有许多车次对应许多的tr标签,注意用elements返回列表
        # 且注意对第二个tr标签利用xpath里面的not(@属性名)过滤掉
        trains = self.driver.find_elements_by_xpath('//tbody[@id="queryLeftTable"]/tr[not(@datatran)]')
        #添加一个布尔标志,用于判断所选车次是否有票再去查询
        is_searched = False
        n =1
        #添加死循环,直到数据符合条件才退出
        while True:
            for train in trains:
                # 利用text打印出标签里面关于车次信息的文本即可
                # 由于刚打印出来的数据之间都是换行的,现在将其替换空格放成同一行
                # 调用split(),以空格进行分割,分割上面替换后的字符串,会返回列表形式的车次的所有信息
                infos = train.text.replace("\n", ' ').split(' ')
                # 从返回的车次列表信息中提取出车次序号数据
                number = infos[0]
                # 判断提取的车次序号数据(number)有没有在用户要的车次字典的里面
                # 是的话再判断有无席位的信息再去预定,
                if number in self.trains:  # 注意self.trains我们已定义是字典,{“G529":["O"."M"]}
                    seat_types = self.trains[number]  # 根据numer的键取得定义字典的座席类型
                    # 取得的座位类型是列表,需要for循环遍历
                    for seat_type in seat_types:
                        # 当座位席位是二等座时,且二等座对应infos[9]
                        if seat_type == "O":
                            count = infos[9]
                            # 当count是数字或者是有 时代表有座
                            # 用.isdigtit()方法说明是数字
                            if count.isdigit() or count == '有':
                                is_searched = True
                                break  # 找到一个座位类型就可以退出自己想要的座位类型列表了
                        # 当座位席位是一等座时,且一等座对应infos[8]
                        elif seat_type == "M":
                            count = infos[8]
                            if count.isdigit() or count == '有':
                                is_searched = True
                                break  # 找到一个座位类型就可以退出自己想要的座位类型列表了
                # 当有票即布尔标志为True时执行预定按钮
                if is_searched:
                    self.current_number = number  # 保存下当前选择的车次序号信息
                    # 从train即第一个有数据的tr标签里面用xpath去找预定按钮执行预定
                    order_button = train.find_element_by_xpath('.//a[@class="btn72"]')
                    order_button.click()
                    print(str(number)+"车次有票,当前购买的车次是"+str(number))
                    # 当有票且执行预定了的话,买到票了,就可以退出最外层的对车次解析的循环了
                    return#不能用break只能退出for,return才能退出死循环

            # 当标志为False时,不断执行查询操作
            if is_searched==False:
                try:
                    search_button = self.driver.find_element_by_id('query_ticket')
                    search_button.click()
                    n += 1
                    #trains也要在每次查询点击后再重新查找一下,即更新trains元素
                    trains = self.driver.find_elements_by_xpath('//tbody[@id="queryLeftTable"]/tr[not(@datatran)]')
                    print("第%d次查询中..."%n)
                    #设置等待,让其监控余票
                    #这里可以不设置sleep,或者更慢,自动查询的速度更块
                    sleep(3)

                except StaleElementReferenceException: #俘获异常则pass
                    pass


    def confirm_passengers(self):
        #需要显示等待下,确认下url是否已经变化到确认乘客信息
        WebDriverWait(self.driver,100).until(
            #EC.url_to_be(self.confirm_passenger_url)
            EC.url_contains(self.confirm_passenger_url)
        )
        #需要再显示等待下,确认下乘车人的横栏信息是否加载出来了
        WebDriverWait(self.driver,100).until(
            EC.presence_of_element_located((By.XPATH,'//ul[@id="normal_passenger_id"]/li/label'))
        )
        """确认需要购买的乘客"""
        #需要找到多个li标签下的label,需要elements且返回列表
        passenger_lables = self.driver.find_elements_by_xpath('//ul[@id="normal_passenger_id"]/li/label')
        for passenger_lable in passenger_lables:
            name = passenger_lable.text
            #判断这个获取的name在不在所要买票的人的列表里
            if name in self.passengers:  #注意是列表形式的数据才能用in
                passenger_lable.click() #勾选起来即可

        """确认需要购买的座位类型"""
        #先用Select()包装下
        seat_select = Select(self.driver.find_element_by_id('seatType_1'))
        #下面选择席位,需要根据用户能够接受的席位来选择
        #这步的话有个细节,前面需要保存下之前预定按钮选择的车次序号,以便根据序号来看看对应用户需要的座位类型
        seat_types = self.trains[self.current_number]  # 使用相应的key值查找对应车次序号的座位类型列表
        for seat_type in seat_types:
            #注意细节,假如第一个选择的席位没有票了,选择不到,会抛出异常
            try:
                self.current_seat = seat_type #保存一下当前选泽的席位信息
                seat_select.select_by_value(seat_type)
            except NoSuchElementException:
                continue
            else:
                break  #假如第一个有票就直接选择然后退出循环

        #等待提交按钮可以被点击
        WebDriverWait(self.driver,100).until(
            #即等待某个元素可以被点击
            EC.element_to_be_clickable((By.ID,'submitOrder_id'))
        )
        sumit_button = self.driver.find_element_by_id('submitOrder_id')
        sumit_button.click()

        #判断模态对话框即购票信息对话框出现并确认按钮可以点击了
        WebDriverWait(self.driver,100).until(
            EC.presence_of_element_located((By.CLASS_NAME,'dhtmlx_window_active'))
        )
        WebDriverWait(self.driver,100).until(
            EC.element_to_be_clickable((By.ID,'qr_submit_id'))
        )
        sumit_button = self.driver.find_element_by_id('qr_submit_id')
        #注意这里的细节,由于Seenium自身的Bug,可能会导致确认点击操作无法正确执行
        #故需俘获异常,且需加入无限循环操作,点击之后再获取再点击
        try:
            while sumit_button:
                try:
                    sumit_button.click()
                    sumit_button = self.driver.find_element_by_id('qr_submit_id')
                except (ElementNotVisibleException,ElementNotInteractableException): #当在此页面见不到此元素,代表已进入付款页面
                    break
            print("恭喜鲁!%s车次%s席位抢票成功"%(self.current_number,self.current_seat))
        except:
            pass

    def send_mail(self):
        """发送邮件"""
        # 连接服务器,提供用户名,授权码,服务器地址
        #这里需要您的QQ邮箱开启smtp服务才行。
        yag_server = yagmail.SMTP(user='你的qq邮箱账号', password='你的smtp授权码', host='smtp.qq.com'
        # 填写发送对象,邮件主题和内容
        email_to = ['你的接收通知信息的邮箱账号', ]
        email_title = "恭喜你!%s的%s车次二等座购票成功"%(self.passengers,self.current_number)
        #由于self.passengers是列表的形式,要转为str进行拼接然后加[0]提取内容
        email_content = "乘车人:"+str(self.passengers[0])+"\n"+"出发日期:"+str(self.train_date)+"\n"+"所买车次:"+str(self.current_number)+"\n"+"所买路线:"+str(self.from_station)+"------>>"+str(self.to_station)
          
        # 发送邮件
        yag_server.send(email_to, email_title, email_content)
        print('已发送邮件!')
        # 关闭服务
        yag_server.close()


    """写个run方法,将步骤封装在一起,让使用起来更方便,即不用理里面的细节"""
    def run(self):
        #先登陆
        self.login()
        #查车次余票
        self.search_left_tickets()
        #确认乘客和车次信息
        self.confirm_passengers()
        #购票后发送邮件通知
        self.send_mail()

#引入线程,target是main函数,防止GUI堵塞
def run():
    t = threading.Thread(target=main)
    t.start() #启动线程
    
#输出信息函数
def printlog():
    #提取输入的数据
    name = entry1_str.get()
    date = entry2_str.get() 
    start_station = entry3_str.get()
    stop_station = entry4_str.get()
    train_s = entry5_str.get()
    #打印个人信息
    info='乘车人:'+name+"\n"+"出发日期:"+date+"\n"+"路线:"+start_station+"---->>"+stop_station+"\n"+"所选车次:"+train_s
    text.insert(END,info)

#运行函数
def main():
    #由12306座位类型的代号设置如下
    #9:商务座 M:一等座 O:二等座 3:硬卧 4:软卧 1:硬座      注意:G6006车次7点27出发,9点16到是【复兴号】
    #初始化chromedriver
    driver = webdriver.Chrome(executable_path='chromedriver.exe')
    name = entry1_str.get()
    date = entry2_str.get() 
    start_station = entry3_str.get()
    stop_station = entry4_str.get()
    train_s = entry5_str.get()
    train_infos = train_s.split(' ')#以空格符为分界形成车次信息列表
    trains = {} #创建空字典存储车次信息
    for train_info in train_infos:
        trains[train_info] = ["O"]  #默认都选为O二等座,然后加入空字典
    spider = TrainSpider(start_station,stop_station,date,trains,[name,],driver)
    spider.run()

#输出所填的信息确认
bt2 = Button(root,text = '输出购票信息',font=('Arial', 14),command=printlog)
bt2.place(x= 20,y= 307)
#按钮事件执行时间过长,引入线程,
bt2 = Button(root,text = '确认无误,开始刷票',font=('Arial', 14),command=run)
bt2.place(x= 200,y= 307)

root.mainloop() #加载GUI界面输入信息后再执行购票程序

#if __name__ == '__main__':
    #main()

其中有一个stations.csv文件需要通过爬虫从12306爬取下来,获得其车站和其对应的编码。由于这里上传不了文件,就只给出GUI脚本制作的代码。

GUI界面

 

后记

近期有很多朋友通过私信咨询有关Python学习问题。为便于交流,点击蓝色自己加入讨论解答资源基地

 

标签:12306,self,driver,购票,车次,station,str,买票,id
来源: https://blog.csdn.net/weixin_43881394/article/details/111628013

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有