build the basic frame

.gitignore

 .idea
 .DS_Store
 venv
+__pycache__
+env.py
\ No newline at end of file

env.py.example

+# encoding=utf8
+
+# 12306 账号
+USER_ACCOUNTS = [
+    {
+        'user_name': '',
+        'password': ''
+    }
+]
+
+# 查询任务
+QUERY_JOBS = [
+    {
+        'left_dates': [  # 出发日期 :Array
+            "2019-01-24",
+            "2019-01-25",
+            "2019-01-26",
+            "2019-01-27",
+            "2019-01-28"
+        ],
+        'stations': {  # 车站 :Dict
+            'left': '广州',
+            'arrive': '达州',
+        },
+        'members': [
+            '1'
+        ],
+        'allow_less_member': 0,
+        'seats': [  # 筛选座位  有先后顺序 :Array
+            '硬卧',
+            '硬座'
+        ],
+        'train_numbers': [  # 筛选车次
+            "K1096",
+            "K814",
+            "K356",
+            "K1172",
+            "K4184"
+        ]
+
+    },
+
+    {
+        'left_dates': [  # 出发日期 :Array
+            "2019-01-24",
+            "2019-01-25",
+            "2019-01-26",
+            "2019-01-27",
+            "2019-01-28"
+        ],
+        'stations': {  # 车站 :Dict
+            'left': '广州',
+            'arrive': '达州',
+        },
+        'members': [
+            '1'
+        ],
+        'allow_less_member': 0,
+        'seats': [  # 筛选座位  有先后顺序 :Array
+            '硬卧',
+            '硬座'
+        ],
+        'train_numbers': [  # 筛选车次
+            "K1096",
+            "K814",
+            "K356",
+            "K1172",
+            "K4184"
+        ]
+
+    }
+]

main.py

+# encoding=utf8
+
+from threading import Thread
+
+from py12306.log.query_log import QueryLog
+from py12306.query.query import Query
+
+
+def main():
+    # Thread(target=Query.run).start()  # 余票查询
+    QueryLog.add_log('init')
+    Query.run()
+    pass
+
+
+if __name__ == '__main__':
+    main()

py12306/config.py

+from os import path
+
+# from py12306.helpers.config import Config
+
+# 12306 账号
+USER_ACCOUNTS = []
+
+# 查询任务
+QUERY_JOBS = []
+
+SEAT_TYPES = {
+    '商务座': 32,
+    '一等座': 31,
+    '二等座': 30,
+    '特等座': 25,
+    '软卧': 23,
+    '硬卧': 28,
+    '硬座': 29,
+    '无座': 26,
+}
+
+# Query
+QUERY_DATA_DIR = 'runtime/query'
+
+
+STATION_FILE = 'data/stations.txt'
+CONFIG_FILE = 'env.py'
+
+if path.exists(CONFIG_FILE):
+    exec(open(CONFIG_FILE, encoding='utf8').read())
+
+
+def get(key, default=None):
+    return eval(key)

py12306/helpers/config.py

+# from py12306.config import get_value_by_key
+from py12306.config import get_value_by_key
+
+
+class BaseConfig:
+    AA = 'USER_ACCOUNTS'
+
+
+class Config(BaseConfig):
+
+    @classmethod
+    def get(cls, key, default=None):
+        self = cls()
+        return get_value_by_key(key);

py12306/helpers/func.py

+from py12306 import config
+import functools
+
+
+def singleton(cls):
+    """
+    将一个类作为单例
+    来自 https://wiki.python.org/moin/PythonDecoratorLibrary#Singleton
+    """
+
+    cls.__new_original__ = cls.__new__
+
+    @functools.wraps(cls.__new__)
+    def singleton_new(cls, *args, **kw):
+        it = cls.__dict__.get('__it__')
+        if it is not None:
+            return it
+
+        cls.__it__ = it = cls.__new_original__(cls, *args, **kw)
+        it.__init_original__(*args, **kw)
+        return it
+
+    cls.__new__ = singleton_new
+    cls.__init_original__ = cls.__init__
+    cls.__init__ = object.__init__
+
+    return cls
+
+
+# 座位
+def get_seat_number_by_name(name):
+    return config.SEAT_TYPES[name]
+
+
+def get_seat_name_by_number(number):
+    return [k for k, v in config.SEAT_TYPES.items() if v == number].pop()
+
+
+# def test:

py12306/helpers/station.py

+from os import path
+from py12306.helpers.func import *
+
+
+@singleton
+class Station:
+    stations = []
+
+    def __init__(self):
+        print('Station 初始化')
+        if path.exists(config.STATION_FILE):
+            result = open(config.STATION_FILE, encoding='utf-8').read()
+            result = result.lstrip('@').split('@')
+            for i in result:
+                tmp_info = i.split('|')
+                self.stations.append({
+                    'key': tmp_info[2],
+                    'name': tmp_info[1],
+                    'pinyin': tmp_info[3],
+                    'id': tmp_info[5]
+                })
+
+    @classmethod
+    def get_station_by_name(cls, name):
+        self = cls()
+        for station in self.stations:
+            if station.get('name') == name:
+                return station
+        return None
+
+    @classmethod
+    def get_station_key_by_name(cls, name):
+        return cls.get_station_by_name(name).get('key')
+

py12306/log/base.py

+from py12306.helpers.func import *
+
+
+class BaseLog:
+    logs = []
+
+    @classmethod
+    def add_log(cls, content):
+        self = cls()
+        self.logs.append(content)
+        return self
+
+    @classmethod
+    def flush(cls):
+        self = cls()
+        for i in self.logs:
+            print(i)
+        # print(self.logs)

py12306/log/query_log.py

+import json
+from os import path
+from py12306.log.base import BaseLog
+from py12306.helpers.func import *
+
+
+@singleton
+class QueryLog(BaseLog):
+    data = {
+        'query_count': 0,
+        'last_time': '',
+    }
+
+    LOG_INIT_JOBS = ''
+
+    def __init__(self):
+        super().__init__()
+        self.init_data()
+
+    def init_data(self):
+        # 获取上次记录
+        print('Query Log 初始化')
+        file_path = config.QUERY_DATA_DIR + '/status.json'
+        if path.exists(file_path):
+            result = open(file_path, encoding='utf-8').read()
+            if result:
+                result = json.loads(result)
+                self.data = {**self.data, **result}
+
+    @classmethod
+    def print_init_jobs(cls, jobs):
+        self = cls()
+        """
+        输出初始化信息
+        :return:
+        """
+        self.add_log('# 发现任务 {} 条 #'.format(len(jobs)))
+        index = 1
+        for job in jobs:
+            self.add_log('================== 任务 {} =================='.format(index))
+            self.add_log('出发站：{}    到达站：{}'.format(job.left_station, job.arrive_station))
+            self.add_log('乘车日期：{}'.format(job.left_dates))
+            self.add_log('坐席：{}'.format('，'.join(job.allow_seats)))
+            self.add_log('乘车人：{}'.format('，'.join(job.members)))
+            self.add_log('筛选车次：{}'.format('，'.join(job.allow_train_numbers)))
+            # 乘车日期：['2019-01-24', '2019-01-25', '2019-01-26', '2019-01-27']
+            index += 1
+
+        return self

py12306/query/job.py

+from py12306.helpers.station import Station
+
+
+class Job:
+    """
+    查询任务
+    """
+
+    left_dates = []
+    left_station = ''
+    arrive_station = ''
+    left_station_code = ''
+    arrive_station_code = ''
+
+    allow_seats = []
+    allow_train_numbers = []
+    members = []
+    member_num = []
+
+
+    def __init__(self, info):
+        self.left_dates = info.get('left_dates')
+        self.left_station = info.get('stations').get('left')
+        self.arrive_station = info.get('stations').get('arrive')
+        self.left_station_code = Station.get_station_key_by_name(self.left_station)
+        self.arrive_station_code = Station.get_station_key_by_name(self.arrive_station)
+
+        self.allow_seats = info.get('seats')
+        self.allow_train_numbers = info.get('train_numbers')
+        self.members = info.get('members')
+        self.member_num = len(self.members)

py12306/query/query.py

+from requests_html import HTMLSession
+
+import py12306.config as config
+from py12306.helpers.api import LEFT_TICKETS
+from py12306.helpers.func import *
+from py12306.helpers.station import Station
+from py12306.log.query_log import QueryLog
+from py12306.query.job import Job
+
+
+class Query:
+    """
+    余票查询
+
+    """
+    jobs = []
+    session = {}
+
+    def __init__(self):
+        self.session = HTMLSession()
+
+    @classmethod
+    def run(cls):
+        self = cls()
+        self.start()
+        pass
+
+    def start(self):
+        self.init_jobs()
+        QueryLog.print_init_jobs(jobs=self.jobs).flush()
+        for job in self.jobs:
+            self.handle_single_job(job)
+
+    def init_jobs(self):
+        jobs = config.QUERY_JOBS
+        for job in jobs:
+            job = Job(info=job)
+            self.jobs.append(job)
+
+    def handle_single_job(self, job):
+        """
+        处理单个任务
+        根据日期循环查询
+
+        展示处理时间
+        :param job:
+        :return:
+        """
+        for date in job.left_dates:
+            result = self.query_by_job_and_date(job, date)
+            self.handle_single_result(result)
+
+        station = Station.get_station_by_name('广州')
+        print(station)
+        pass
+
+    def query_by_job_and_date(self, job, date):
+        """
+        通过日期进行查询
+        :return:
+        """
+        QueryLog.add_log('正在查询 {}, {} - {}'.format(date, job.left_station, job.arrive_station))
+        url = LEFT_TICKETS.get('url').format(left_date=date, left_station=job.left_station_code,
+                                             arrive_station=job.arrive_station_code, type='leftTicket/queryZ')
+
+        return self.session.get(url)
+
+    def handle_single_result(self, result):
+        """
+        错误判断
+        余票判断
+        小黑屋判断
+        座位判断
+        乘车人判断
+        :param result:
+        :return:
+        """
+        pass