# -*- 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))