python火车余票查询（s10）

港控/mmm° 2022-05-31 14:48 170阅读 0赞

from tkinter import *
    from tkinter import ttk
    from tkinter.ttk import Treeview,Checkbutton
    import tkinter.messagebox as messagebox
    from stationapi import *
    from datetime import date,timedelta
    import re
    from ticketapi import query,analyze,get_price
    
    class Ticket(Frame):
        def __init__(self):
            Frame.__init__(self)
            self.pack(expand = YES, fill = BOTH)
            self.master.title("火车票查询V1.3")
            self.master.geometry('890x600')
            self.master.resizable(width=False, height=True)        
            self.createWidgets()
            
        def createWidgets(self):
            self.style1 = ttk.Style()
            self.style1.configure("Treeview", background='light yellow')
            self.var_from = StringVar()
            self.var_to = StringVar()
            self.var_date = StringVar()
            self.var_days = StringVar()
            self.onlyG = IntVar()
            self.onlyD = IntVar()
            self.onlyZ = IntVar()
            self.onlyT = IntVar()
            self.onlyK = IntVar()
            self.other = IntVar()
            self.tickets = []
            self.prices = []
            
            self.frame1 = Frame(self,bg='light blue')
            self.frame2 = Frame(self,bg='light blue')
            self.frame3 = Frame(self)
            self.frame4 = Frame(self)
    
            self.label1 = Label(self.frame1,text='出发站:',font=('微软雅黑', 11),padx=10,bg='light blue')
            self.entry1 = Entry(self.frame1, textvariable=self.var_from,font=('微软雅黑', 11),width=10)
            self.label2 = Label(self.frame1,text='到达站:',font=('微软雅黑', 11),padx=10,bg='light blue')
            self.entry2 = Entry(self.frame1, textvariable=self.var_to,font=('微软雅黑', 11),width=10)
            self.label3 = Label(self.frame1,text='出发日期:',font=('微软雅黑', 11),padx=10,bg='light blue')
            self.entry3 = Entry(self.frame1, textvariable=self.var_date,font=('微软雅黑', 11),width=10)
            self.label4 = Label(self.frame1,text='查询天数:',font=('微软雅黑', 11),padx=10,bg='light blue')
            self.entry4 = Entry(self.frame1, textvariable=self.var_days,font=('微软雅黑', 11),width=3)
            self.label1.pack(side=LEFT)
            self.entry1.pack(side=LEFT)
            self.label2.pack(side=LEFT)
            self.entry2.pack(side=LEFT)
            self.label3.pack(side=LEFT)
            self.entry3.pack(side=LEFT)
            self.label4.pack(side=LEFT)
            self.entry4.pack(side=LEFT)
            self.btn1 = Button(self.frame1,text='查 询',command=self.search,font=('微软雅黑', 11),width=9,padx=10,bg='light cyan')
            self.btn2 = Button(self.frame1,text='清 除',command=self.reset,font=('微软雅黑', 11),width=9,padx=10,bg='light cyan')
            self.btn1.pack(side=LEFT,padx=10)
            self.btn2.pack(side=RIGHT,padx=5)
            
            self.label5 = Label(self.frame2,text='输入示例：出发站：西安（或xian）  到达站：北京（或beijing）  出发日期：2018-01-03（月和日不足2位前面补0）',font=('微软雅黑', 11),bg='light blue',padx=15)
            self.label5.pack(side=LEFT)
            
            self.label6 = Label(self.frame3,text='车次类型:',padx=5,font=('微软雅黑', 11))
            self.label6.pack(side=LEFT)
            self.check1 = Checkbutton(self.frame3,text='GC-高铁/城际',command=self.select,variable=self.onlyG)
            self.check2 = Checkbutton(self.frame3,text='D-动车',command=self.select,variable=self.onlyD)
            self.check3 = Checkbutton(self.frame3,text='Z-直达',command=self.select,variable=self.onlyZ)
            self.check4 = Checkbutton(self.frame3,text='T-特快',command=self.select,variable=self.onlyT)
            self.check5 = Checkbutton(self.frame3,text='K-快速',command=self.select,variable=self.onlyK)
            self.check6 = Checkbutton(self.frame3,text='其他',command=self.select,variable=self.other)
            self.check1.pack(side=LEFT,padx=20)
            self.check2.pack(side=LEFT,padx=20)
            self.check3.pack(side=LEFT,padx=20)
            self.check4.pack(side=LEFT,padx=20)
            self.check5.pack(side=LEFT,padx=20)
            self.check6.pack(side=LEFT,padx=20)
    
            self.scrollBar = Scrollbar(self.frame4)
            self.scrollBar.pack(side=RIGHT, fill=Y)
            self.tree = Treeview(self.frame4, 
                                 columns=('c0','c1','c2','c3','c4','c5','c6',
                                          'c7','c8','c9','c10','c11','c12'),
                                 show="headings",
                                 height=30,
                                 yscrollcommand=self.scrollBar.set)
            self.tree.column('c0', width=85, anchor='center')
            self.tree.column('c1', width=45, anchor='center')
            self.tree.column('c2', width=110, anchor='center')
            self.tree.column('c3', width=90, anchor='center')
            self.tree.column('c4', width=60, anchor='center')
            self.tree.column('c5', width=60, anchor='center')
            self.tree.column('c6', width=60, anchor='center')
            self.tree.column('c7', width=60, anchor='center')
            self.tree.column('c8', width=60, anchor='center')
            self.tree.column('c9', width=60, anchor='center')
            self.tree.column('c10', width=60, anchor='center')
            self.tree.column('c11', width=60, anchor='center')
            self.tree.column('c12', width=60, anchor='center')
            #显示日期、车次、出发/到达站、 出发/到达时间、历时、商务座、一等座、二等座、软卧、动卧、硬卧、硬座、无座
            self.tree.heading('c0', text='日期')
            self.tree.heading('c1', text='车次')
            self.tree.heading('c2', text='出发/到达站')
            self.tree.heading('c3', text='出发/到达时间')
            self.tree.heading('c4', text='历时')
            self.tree.heading('c5', text='商务座')
            self.tree.heading('c6', text='一等座')
            self.tree.heading('c7', text='二等座')
            self.tree.heading('c8', text='软卧')
            self.tree.heading('c9', text='动卧')
            self.tree.heading('c10', text='硬卧')
            self.tree.heading('c11', text='硬座')
            self.tree.heading('c12', text='无座')
            self.tree.pack(side=LEFT, fill=BOTH)
            #Treeview组件与垂直滚动条结合
            self.scrollBar.config(command=self.tree.yview)
            
            self.frame1.pack(fill=X,ipady=5)
            self.frame2.pack(fill=X)
            self.frame3.pack(fill=X,ipady=5)
            self.frame4.pack(fill=X)
    
        def reset(self):
            self.tickets.clear()
            self.onlyG.set(0)
            self.onlyD.set(0)
            self.onlyZ.set(0)
            self.onlyT.set(0)
            self.onlyK.set(0)
            self.other.set(0)
            if len(self.tree.get_children())>0:
                items = self.tree.get_children()
                for item in items:
                    self.tree.delete(item)
                
        def select(self):
            allow = []
            if self.onlyG.get():
                allow.append('G')
                allow.append('C')
            if self.onlyD.get():
                allow.append('D')
            if self.onlyZ.get():
                allow.append('Z')
            if self.onlyT.get():
                allow.append('T')
            if self.onlyK.get():
                allow.append('K')
            if self.other.get():
                allow.extend(['1','2','3','4','5','6','7','8','9','S','Y'])
                
            if len(self.tree.get_children())>0:
                items = self.tree.get_children()
                for item in items:
                    self.tree.delete(item)       
            length = len(self.tickets)
            #print(length,allow)
            if length > 0:     
                if allow:
                    for i in range(length):
                        if (self.tickets[i][1][0] in allow):                
                            self.tree.insert('',i*2,values=self.tickets[i][:13])
                            self.tree.insert('',i*2+1,values=self.prices[i])
                else:
                    for i in range(length):
                        self.tree.insert('',i*2,values=self.tickets[i][:13])
                        self.tree.insert('',i*2+1,values=self.prices[i])
                items = self.tree.get_children()
                for i in range(len(items)): 
                    if i%2==1:
                        self.tree.item(items[i], tags=('oddrow'))
                self.tree.tag_configure('oddrow', background='#6633ff', foreground='#ffff66')
                
        def search(self):
            #清除车票信息和表格
            self.tickets.clear()
            self.prices.clear()
            if len(self.tree.get_children())>0:
                items = self.tree.get_children()
                for item in items:
                    self.tree.delete(item)
            #检查输入条件合法性        
            from_station = self.var_from.get()
            to_station = self.var_to.get()
            from_date = self.var_date.get()
            if not station_valid(from_station,stations):
                messagebox.showinfo('输入条件有误', '请检查后重新输入出发车站')
            elif not station_valid(to_station,stations):
                messagebox.showinfo('输入条件有误', '请检查后重新输入到达车站')
            else:
                from_station = prase_station(from_station,stations)
                to_station = prase_station(to_station,stations)
                rdate = re.match(r'^(\d{4})-(\d{2})-(\d{2})$', from_date)
                con = self.var_days.get()
                if from_station == to_station:
                    messagebox.showinfo('输入条件有误', '车站重复，请重新输入到达车站')
                elif rdate:
                    year = int(rdate.group(1))
                    month = int(rdate.group(2))
                    day = int(rdate.group(3))
                    try:
                        qday = date(year,month,day)
                    except ValueError:
                        messagebox.showinfo('输入条件有误', '请检查后重新输入出发日期')
                    today = date.today()
                    lastday = today + timedelta(days=15)
                    if today > qday or qday > lastday:
                        messagebox.showinfo('输入条件有误', '请检查后重新输入出发日期')
                    elif con == '':
                        condition = [from_date,from_station,to_station]
                        ticketinfo_list = query(condition)
                        self.tickets.extend(analyze(ticketinfo_list,stations))
                        for t in self.tickets:
                            t.insert(0,from_date)
                            price = get_price(t)
                            self.prices.append(price)
                        
                        self.select()
                    else:
                        try:
                            dd = int(con)
                        except ValueError:
                            messagebox.showinfo('输入条件有误', '请检查后重新输入查询天数')
                        if qday > lastday - timedelta(days=dd):
                            messagebox.showinfo('输入条件有误', '请检查后重新输入查询天数')
                        else:
                            for i in range(dd):    
                                ndate = qday + timedelta(days=i)
                                condition = [str(ndate),from_station,to_station]
                                ticketinfo_list = query(condition)
                                tt = analyze(ticketinfo_list,stations)
                                for t in tt:
                                    t.insert(0,str(ndate))
                                    price = get_price(t)
                                    self.prices.append(price)
                                self.tickets.extend(tt)
                                print(len(self.prices))
                            self.select()
                else:
                    messagebox.showinfo('输入条件有误', '请检查后重新输入出发日期')
        
    if __name__ == '__main__':
        Ticket().mainloop()

![20180227160807836][]

[20180227160807836]: /images/20220531/2291cdca84b04d68871a6f18098b8893.png