使用PyQt开发 火车票查询软件

开发环境：Python3.3.3，PyQt5.1.1
注意：Python版本

1. # -*- coding: utf-8 -*-
   
2. # Python 3.3.3
   
3. # PyQt 5.1.1
   
4. import sys,time,re,urllib.parse,urllib.request,http.cookiejar,json
   
5. from PyQt5.QtCore import *
   
6. from PyQt5.QtGui import *
   
7. from PyQt5.QtWidgets import *
   
8. 
   
9. """cookie"""
   
10. cookie=http.cookiejar.LWPCookieJar()
    
11. #cookie.load('f:/cookie.txt',True,True)
    
12. chandle=urllib.request.HTTPCookieProcessor(cookie)
    
13. 
    
14. """获取数据"""
    
15. def getData(url):
    
16.     r=urllib.request.Request(url)
    
17.     opener=urllib.request.build_opener(chandle)
    
18.     u=opener.open(r)
    
19.     #chandle.cookiejar.save('f:/cookie.txt',True,True)
    
20.     data=u.read()
    
21.     try:
    
22.         data=data.decode('utf-8')
    
23.     except:
    
24.         data=data.decode('gbk','ignore')
    
25.     return data
    
26. def postData(url,data):
    
27.     data=urllib.parse.urlencode(data);data=bytes(data,'utf-8')
    
28.     r=urllib.request.Request(url,data)
    
29.     opener=urllib.request.build_opener(chandle)
    
30.     u=opener.open(r)
    
31.     #chandle.cookiejar.save('f:/cookie.txt',True,True)
    
32.     data=u.read()
    
33.     try:
    
34.         data=data.decode('utf-8')
    
35.     except:
    
36.         data=data.decode('gbk','ignore')
    
37.     return data
    
38. """火车票"""
    
39. class Ticket:
    
40.     def init(self,s,e,d):
    
41.         self.li=[]
    
42.         cont=getData('https://kyfw.12306.cn/otn/resources/js/framework/station_name.js')
    
43.         s=re.findall('%s\|([^|]+)' % s,cont)[0]
    
44.         e=re.findall('%s\|([^|]+)' % e,cont)[0]
    
45.         url='https://kyfw.12306.cn/otn/lcxxcx/query?purpose_codes=0X00&queryDate=%s&from_station=%s&to_station=%s' % (d,s,e)
    
46.         cont=json.loads(getData(url))["data"]["datas"]
    
47.         name=[
    
48.             "station_train_code",
    
49.             "from_station_name",
    
50.             "to_station_name",
    
51.             "lishi",
    
52.             "swz_num",
    
53.             "tz_num",
    
54.             "zy_num",
    
55.             "ze_num",
    
56.             "gr_num",
    
57.             "rw_num",
    
58.             "yw_num",
    
59.             "rz_num",
    
60.             "yz_num",
    
61.             "wz_num",
    
62.             "qt_num"
    
63.         ]
    
64.         for x in cont:
    
65.             tmp=[]
    
66.             for y in name:
    
67.                 if y=="from_station_name":
    
68.                     s=x[y]+'\n'+x["start_time"]
    
69.                     tmp.append(s)
    
70.                 elif y=="to_station_name":
    
71.                     s=x[y]+'\n'+x["arrive_time"]
    
72.                     tmp.append(s)
    
73.                 else:
    
74.                     tmp.append(x[y])
    
75.             self.li.append(tmp)
    
76. """ui"""
    
77. class Dialog(QDialog):
    
78.     ticket=Ticket()
    
79.     def __init__(self):
    
80.         super().__init__()
    
81.         self.resize(750,350)
    
82.         #布局管理器
    
83.         self.layout=[QVBoxLayout(self),QHBoxLayout()]
    
84.         self.layout[1].setContentsMargins(0,0,0,0)
    
85.         self.layout[1].setSpacing(0)
    
86.         self.layout[0].setContentsMargins(0,0,0,0)
    
87.         self.layout[0].setSpacing(0)
    
88.         self.layout[0].addLayout(self.layout[1])
    
89.         #按钮
    
90.         btn=QPushButton("ok")
    
91.         btn.clicked.connect(self.submit)
    
92.         #输入选项
    
93.         label=[QLabel("发站："),QLabel("到站："),QLabel("日期：")]
    
94.         self.line=[QLineEdit(),QLineEdit(),QComboBox()]
    
95.         y=int(time.strftime("%Y",time.localtime()))
    
96.         m=int(time.strftime("%m",time.localtime()))
    
97.         d=int(time.strftime("%d",time.localtime()))
    
98.         i=0
    
99.         yy=y
    
100.         mm=m
     
101.         dd=d
     
102.         while i<20:
     
103.             if m in (1,3,5,7,8,10,12):
     
104.                 if d+i>31:
     
105.                     dd=d+i-31
     
106.                     mm=m+1
     
107.                     if mm>12:
     
108.                         yy=y+1
     
109.                         mm=mm-12
     
110.                 else:
     
111.                     dd=d+i
     
112.             elif m in (4,6,9,11):
     
113.                 if d+i>30:
     
114.                     dd=d+i-30
     
115.                     mm=m+1
     
116.                     if mm>12:
     
117.                         yy=y+1
     
118.                         mm=mm-12
     
119.                 else:
     
120.                     dd=d+i
     
121.             else:
     
122.                 if (m%400==0) or ((m%4==0) and (m%100!=0)):
     
123.                     if d+i>29:
     
124.                         dd=d+i-29
     
125.                         mm=m+1
     
126.                         if mm>12:
     
127.                             yy=y+1
     
128.                             mm=mm-12
     
129.                     else:
     
130.                         dd=d+i
     
131.                 else:
     
132.                     if d+i>28:
     
133.                         dd=d+i-28
     
134.                         mm=m+1
     
135.                         if mm>12:
     
136.                             yy=y+1
     
137.                             mm=mm-12
     
138.                     else:
     
139.                         dd=d+i
     
140.             s='%d-%02d-%02d' % (yy,mm,dd)
     
141.             self.line[2].addItem(s)
     
142.             i+=1
     
143.         i=0
     
144.         while i<3:
     
145.             self.line[i].setMaximumWidth(100)
     
146.             label[i].setMaximumWidth(50)
     
147.             label[i].setAlignment(Qt.AlignRight|Qt.AlignVCenter)
     
148.             self.layout[1].addWidget(label[i],Qt.AlignRight)
     
149.             self.layout[1].addWidget(self.line[i],Qt.AlignLeft)
     
150.             i+=1
     
151.         self.layout[1].addWidget(btn)
     
152.         #表格
     
153.         head=['车次','发站','到站','历时','商务座','特等座','一等座','二等座','高级软卧','软卧','硬卧','软座','硬座','无座','其他']
     
154.         self.table=QTableWidget()
     
155.         self.table.setEditTriggers(QAbstractItemView.NoEditTriggers)
     
156.         self.table.setColumnCount(15)
     
157.         self.table.setHorizontalHeaderLabels(head)
     
158.         self.layout[0].addWidget(self.table)
     
159.         self.show()
     
160.     def submit(self):
     
161.         self.table.clearContents()
     
162.         s=self.line[0].text()
     
163.         e=self.line[1].text()
     
164.         d=self.line[2].currentText()
     
165.         self.ticket.init(s,e,d)
     
166.         self.table.setRowCount(len(self.ticket.li))
     
167.         i=0
     
168.         for x in self.ticket.li:
     
169.             j=0
     
170.             for y in x:
     
171.                 if j==1 or j==2:
     
172.                     item=QLabel(y)
     
173.                     item.setAlignment(Qt.AlignCenter)
     
174.                     self.table.setCellWidget(i,j,item)
     
175.                 else:
     
176.                     item=QTableWidgetItem(y)
     
177.                     item.setTextAlignment(Qt.AlignCenter)
     
178.                     if not re.search('\D',y):
     
179.                         item.setForeground(QBrush(Qt.red))
     
180.                     self.table.setItem(i,j,item)
     
181.                 if j>2 or j==0:
     
182.                     self.table.resizeColumnToContents(j)
     
183.                 j+=1
     
184.             i+=1
     
185. if __name__=="__main__":
     
186.     app=QApplication(sys.argv)
     
187.     dialog=Dialog()
     
188.     sys.exit(app.exec_())

复制代码