12306搶票腳本開發(四)完善上節課的代碼并面向對象


文章地址 :

12306搶票腳本開發(一)提綱
12306搶票腳本開發(二)解析火車站代號并分析查詢的HTTP請求
12306搶票腳本開發(三)實現一個簡單的查詢腳本
12306搶票腳本開發(四)完善上節課的代碼并面向對象
12306搶票腳本開發(五)更友好的使用方式
12306搶票腳本開發(六)更友好的時間輸入方式
12306搶票腳本開發(七)將前幾節課的成果結合起來實現一個完整的工具


簡介 :

根據上節課的代碼 , 我們可以將查詢余票的功能抽象成一個查詢類
這個類的構造函數就是傳入 :

leftTicketDTO.train_date : 出發日期
leftTicketDTO.from_station : 出發站的代號
leftTicketDTO.to_station : 到達站的代號
purpose_codes : ADULT 表示成人票 , 改變選項為學生票可以發現該參數的值變成了 : 0X00

這個類還會向外部暴露獲取查詢結果的各種數據的接口 , 比如說 getStartStationName() 等...
開始構建代碼吧~


代碼 :

querier.py

#!/usr/bin/env python
# encoding:utf-8

import json
import requests
import logging

logging.captureWarnings(True)

class querier(object):
    train_date = "2017-01-01"
    from_station = "BJ"
    to_station = "SHH"
    purpose_codes = "ADULT"
    url = ""
    resultJson = {}
    trainsJson = []
    length = 0
    def __init__(self, train_date, from_station, to_station, purpose_codes):
        self.train_date = train_date
        self.from_station = from_station
        self.to_station = to_station
        self.purpose_codes = purpose_codes
        self.url = self.__buildQueryUrl(self.train_date, self.from_station, self.to_station, self.purpose_codes)
        self.resultJson = json.loads(requests.get(self.url, verify=False).text)
        self.trainsJson = self.resultJson["data"]
        self.length = len(self.trainsJson)

    def __buildQueryUrl(self, train_date, from_station, to_station, purpose_codes):
        url = "https://kyfw.12306.cn/otn/leftTicket/query?"
        url += "leftTicketDTO.train_date=" + train_date
        url += "&leftTicketDTO.from_station=" + from_station
        url += "&leftTicketDTO.to_station=" + to_station
        url += "&purpose_codes=" + purpose_codes
        return url

    def getStartStationName(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["start_station_name"]

    def getEndStationName(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["end_station_name"]

    def getFromStationName(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["from_station_name"]

    def getToStationName(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["to_station_name"]

    def getStartTime(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["start_time"]

    def getArriveTime(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["arrive_time"]

    def getQitaNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["qt_num"]

    def getWuzuoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["wz_num"]

    def getYingzuoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["yz_num"]

    def getYingwoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["yw_num"]

    def getRuanzuoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["rz_num"]

    def getRuanwoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["rw_num"]

    def getGaojiruanwoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["gr_num"]

    def getErdengzuoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["ze_num"]

    def getYidengzuoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["zy_num"]

    def getTengdengzuoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["tz_num"]

    def getShangwuzuoNumber(self, index):
        return self.trainsJson[index]["queryLeftNewDTO"]["swz_num"]

main.py

#!/usr/bin/env python
# encoding:utf-8

import querier

result = querier.querier("2017-02-28", "BJP", "SHH", "ADULT")
length = result.length
for i in range(length):
    print "=" * 36
    print "首發站 : ", result.getStartStationName(i)
    print "終點站 : ", result.getEndStationName(i)
    print "起始站 : ", result.getFromStationName(i)
    print "到達站 : ", result.getToStationName(i)
    print "出發時間 : ", result.getStartTime(i)
    print "到達時間", result.getArriveTime(i)
    print "其他 : ", result.getQitaNumber(i)
    print "無座", result.getWuzuoNumber(i)
    print "硬座 : ", result.getYingzuoNumber(i)
    print "硬臥 : ", result.getYingwoNumber(i)
    print "軟座 : ", result.getRuanzuoNumber(i)
    print "軟臥 : ", result.getRuanwoNumber(i)
    print "高級軟臥 : ", result.getGaojiruanwoNumber(i)
    print "二等座 : ", result.getErdengzuoNumber(i)
    print "一等座 : ", result.getYidengzuoNumber(i)
    print "特等座 : ", result.getTengdengzuoNumber(i)
    print "商務座 : ", result.getShangwuzuoNumber(i)

運行效果 :

Paste_Image.png

總結 :

雖然運行效果看起來和上節課并沒有什么區別 , 但是我們的代碼已經發生了質的轉變
代碼的可讀性 , 可維護性都高了許多 , 這也是面向對象帶來的好處
好了這節課就到這里 , 感興趣的小伙伴可以把代碼復制下來運行嘗試一下 , git 倉庫目前還比較亂 , 并沒有好好整理代碼結構 , 大家現在去看可能有點亂


預告 :

下節課我們就要開始讓我們的代碼面向用戶 , 還記得嗎 ?
我們這里初始化 querier 類的時候傳遞的參數 :

result = querier.querier("2017-02-28", "BJP", "SHH", "ADULT")

這樣并不符合易用性的原則 , 想一下讓一個完全不知道火車站的編號的人來用的話 , 他根本不知道應該如何初始化我們的類 , 因此下節課我們要做一個將城市的中文名到編號進行映射的功能 , 讓用戶能更方便地使用 , (還有一個需要做的就是對日期的表示 , 如果能支持更多的日期類型就很好了)

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容