201 lines
7.6 KiB
Python
201 lines
7.6 KiB
Python
# -*- coding:utf-8 -*-
|
||
import datetime
|
||
import json
|
||
import sys
|
||
from os import path
|
||
|
||
from py12306.config import Config
|
||
from py12306.cluster.cluster import Cluster
|
||
from py12306.log.base import BaseLog
|
||
from py12306.helpers.func import *
|
||
|
||
|
||
@singleton
|
||
class QueryLog(BaseLog):
|
||
# 这里如果不声明，会出现重复打印，目前不知道什么原因
|
||
logs = []
|
||
thread_logs = {}
|
||
quick_log = []
|
||
|
||
data = {
|
||
'query_count': 0,
|
||
'last_time': '',
|
||
}
|
||
data_path = None
|
||
|
||
LOG_INIT_JOBS = ''
|
||
|
||
MESSAGE_GIVE_UP_CHANCE_CAUSE_TICKET_NUM_LESS_THAN_SPECIFIED = '余票数小于乘车人数，放弃此次提交机会'
|
||
MESSAGE_QUERY_LOG_OF_EVERY_TRAIN = '{}'
|
||
MESSAGE_QUERY_LOG_OF_TRAIN_INFO = '{} {}'
|
||
MESSAGE_QUERY_START_BY_DATE = '出发日期 {}: {} - {}'
|
||
|
||
MESSAGE_JOBS_DID_CHANGED = '任务已更新，正在重新加载...\n'
|
||
|
||
MESSAGE_SKIP_ORDER = '跳过本次请求，节点 {} 用户 {} 正在处理该订单\n'
|
||
|
||
MESSAGE_QUERY_JOB_BEING_DESTROY = '查询任务 {} 已结束\n'
|
||
|
||
MESSAGE_INIT_PASSENGERS_SUCCESS = '初始化乘客成功'
|
||
MESSAGE_CHECK_PASSENGERS = '查询任务 {} 正在验证乘客信息'
|
||
|
||
MESSAGE_USER_IS_EMPTY_WHEN_DO_ORDER = '未配置自动下单账号，{} 秒后继续查询\n'
|
||
MESSAGE_ORDER_USER_IS_EMPTY = '未找到下单账号，{} 秒后继续查询'
|
||
|
||
cluster = None
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.data_path = Config().QUERY_DATA_DIR + 'status.json'
|
||
self.cluster = Cluster()
|
||
|
||
@classmethod
|
||
def init_data(cls):
|
||
self = cls()
|
||
# 获取上次记录
|
||
result = False
|
||
if not Config.is_cluster_enabled() and path.exists(self.data_path):
|
||
with open(self.data_path, encoding='utf-8') as f:
|
||
result = f.read()
|
||
try:
|
||
result = json.loads(result)
|
||
except json.JSONDecodeError as e:
|
||
result = {}
|
||
# self.add_quick_log('加载status.json失败, 文件内容为: {}.'.format(repr(result)))
|
||
# self.flush() # 这里可以用不用提示
|
||
|
||
if Config.is_cluster_enabled():
|
||
result = self.get_data_from_cluster()
|
||
|
||
if result:
|
||
self.data = {**self.data, **result}
|
||
self.print_data_restored()
|
||
|
||
def get_data_from_cluster(self):
|
||
query_count = self.cluster.session.get(Cluster.KEY_QUERY_COUNT, 0)
|
||
last_time = self.cluster.session.get(Cluster.KEY_QUERY_LAST_TIME, '')
|
||
if query_count and last_time:
|
||
return {'query_count': query_count, 'last_time': last_time}
|
||
return False
|
||
|
||
def refresh_data_of_cluster(self):
|
||
return {
|
||
'query_count': self.cluster.session.incr(Cluster.KEY_QUERY_COUNT),
|
||
'last_time': self.cluster.session.set(Cluster.KEY_QUERY_LAST_TIME, time_now()),
|
||
}
|
||
|
||
@classmethod
|
||
def print_init_jobs(cls, jobs):
|
||
"""
|
||
输出初始化信息
|
||
:return:
|
||
"""
|
||
self = cls()
|
||
self.add_log('# 发现 {} 个任务 #'.format(len(jobs)))
|
||
index = 1
|
||
for job in jobs:
|
||
self.add_log('================== 任务 {} =================='.format(index))
|
||
for station in job.stations:
|
||
self.add_log('出发站：{} 到达站：{}'.format(station.get('left'), station.get('arrive')))
|
||
|
||
self.add_log('乘车日期：{}'.format(job.left_dates))
|
||
self.add_log('坐席：{}'.format('，'.join(job.allow_seats)))
|
||
self.add_log('乘车人：{}'.format('，'.join(job.members)))
|
||
if job.except_train_numbers:
|
||
train_number_message = '排除 ' + '，'.join(job.allow_train_numbers)
|
||
else:
|
||
train_number_message = '，'.join(job.allow_train_numbers if job.allow_train_numbers else ['不筛选'])
|
||
self.add_log('筛选车次：{}'.format(train_number_message))
|
||
self.add_log('任务名称：{}'.format(job.job_name))
|
||
# 乘车日期：['2019-01-24', '2019-01-25', '2019-01-26', '2019-01-27']
|
||
self.add_log('')
|
||
index += 1
|
||
|
||
self.flush()
|
||
return self
|
||
|
||
@classmethod
|
||
def print_ticket_num_less_than_specified(cls, rest_num, job):
|
||
self = cls()
|
||
self.add_quick_log(
|
||
'余票数小于乘车人数，当前余票数: {rest_num}, 实际人数 {actual_num}, 删减人车人数到: {take_num}'.format(rest_num=rest_num,
|
||
actual_num=job.member_num,
|
||
take_num=job.member_num_take))
|
||
self.flush()
|
||
return self
|
||
|
||
@classmethod
|
||
def print_ticket_seat_available(cls, left_date, train_number, seat_type, rest_num):
|
||
self = cls()
|
||
self.add_quick_log(
|
||
'[ 查询到座位可用 出发时间 {left_date} 车次 {train_number} 座位类型 {seat_type} 余票数量 {rest_num} ]'.format(
|
||
left_date=left_date,
|
||
train_number=train_number,
|
||
seat_type=seat_type,
|
||
rest_num=rest_num))
|
||
self.flush()
|
||
return self
|
||
|
||
@classmethod
|
||
def print_ticket_available(cls, left_date, train_number, rest_num):
|
||
self = cls()
|
||
self.add_quick_log('检查完成 开始提交订单 '.format())
|
||
self.notification('查询到可用车票', '时间 {left_date} 车次 {train_number} 余票数量 {rest_num}'.format(left_date=left_date,
|
||
train_number=train_number,
|
||
rest_num=rest_num))
|
||
self.flush()
|
||
return self
|
||
|
||
@classmethod
|
||
def print_query_error(cls, reason, code=None):
|
||
self = cls()
|
||
self.add_quick_log('查询余票请求失败')
|
||
if code:
|
||
self.add_quick_log('状态码 {} '.format(code))
|
||
if reason:
|
||
self.add_quick_log('错误原因 {} '.format(reason))
|
||
self.flush(sep='\t')
|
||
return self
|
||
|
||
@classmethod
|
||
def print_job_start(cls, job_name):
|
||
self = cls()
|
||
message = '>> 第 {query_count} 次查询 {job_name} {time}'.format(
|
||
query_count=int(self.data.get('query_count', 0)) + 1,
|
||
job_name=job_name, time=time_now().strftime("%Y-%m-%d %H:%M:%S"))
|
||
self.add_log(message)
|
||
self.refresh_data()
|
||
if is_main_thread():
|
||
self.flush(publish=False)
|
||
return self
|
||
|
||
@classmethod
|
||
def add_query_time_log(cls, time, is_cdn):
|
||
return cls().add_log(('*' if is_cdn else '') + '耗时 %.2f' % time)
|
||
|
||
@classmethod
|
||
def add_stay_log(cls, second):
|
||
self = cls()
|
||
self.add_log('停留 {}'.format(second))
|
||
return self
|
||
|
||
def print_data_restored(self):
|
||
self.add_quick_log('============================================================')
|
||
self.add_quick_log('|=== 查询记录恢复成功 上次查询 {last_date} ===|'.format(last_date=self.data.get('last_time')))
|
||
self.add_quick_log('============================================================')
|
||
self.add_quick_log('')
|
||
self.flush(publish=False)
|
||
return self
|
||
|
||
def refresh_data(self):
|
||
if Config.is_cluster_enabled():
|
||
self.data = {**self.data, **self.refresh_data_of_cluster()}
|
||
else:
|
||
self.data['query_count'] += 1
|
||
self.data['last_time'] = str(datetime.datetime.now())
|
||
self.save_data()
|
||
|
||
def save_data(self):
|
||
with open(self.data_path, 'w') as file:
|
||
file.write(json.dumps(self.data))
|