美文网首页Python程序员大数据 爬虫Python AI Sql
12306抢票脚本开发(四)完善上节课的代码并面向对象

12306抢票脚本开发(四)完善上节课的代码并面向对象

作者: 王一航 | 来源:发表于2017-02-23 20:45 被阅读309次

文章地址 :

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")

这样并不符合易用性的原则 , 想一下让一个完全不知道火车站的编号的人来用的话 , 他根本不知道应该如何初始化我们的类 , 因此下节课我们要做一个将城市的中文名到编号进行映射的功能 , 让用户能更方便地使用 , (还有一个需要做的就是对日期的表示 , 如果能支持更多的日期类型就很好了)

相关文章

网友评论

    本文标题:12306抢票脚本开发(四)完善上节课的代码并面向对象

    本文链接:https://www.haomeiwen.com/subject/brnbwttx.html