78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
import os
|
|
import sys
|
|
import io
|
|
from contextlib import redirect_stdout
|
|
|
|
from py12306.config import Config
|
|
from py12306.helpers.func import *
|
|
|
|
|
|
class BaseLog:
|
|
logs = []
|
|
thread_logs = {}
|
|
quick_log = []
|
|
|
|
@classmethod
|
|
def add_log(cls, content=''):
|
|
self = cls()
|
|
# print('添加 Log 主进程{} 进程ID{}'.format(is_main_thread(), current_thread_id()))
|
|
if is_main_thread():
|
|
self.logs.append(content)
|
|
else:
|
|
tmp_log = self.thread_logs.get(current_thread_id(), [])
|
|
tmp_log.append(content)
|
|
self.thread_logs[current_thread_id()] = tmp_log
|
|
return self
|
|
|
|
@classmethod
|
|
def flush(cls, sep='\n', end='\n', file=None, exit=False, publish=True):
|
|
from py12306.cluster.cluster import Cluster
|
|
self = cls()
|
|
logs = self.get_logs()
|
|
# 输出到文件
|
|
if file == None and Config().OUT_PUT_LOG_TO_FILE_ENABLED and not Const.IS_TEST: # TODO 文件无法写入友好提示
|
|
file = open(Config().OUT_PUT_LOG_TO_FILE_PATH, 'a', encoding='utf-8')
|
|
if not file: file = None
|
|
# 输出日志到各个节点
|
|
if publish and self.quick_log and Config().is_cluster_enabled() and Cluster().is_ready: #
|
|
f = io.StringIO()
|
|
with redirect_stdout(f):
|
|
print(*logs, sep=sep, end='' if end == '\n' else end)
|
|
out = f.getvalue()
|
|
Cluster().publish_log_message(out)
|
|
else:
|
|
print(*logs, sep=sep, end=end, file=file)
|
|
self.empty_logs(logs)
|
|
if exit: sys.exit()
|
|
|
|
def get_logs(self):
|
|
if self.quick_log:
|
|
logs = self.quick_log
|
|
else:
|
|
if is_main_thread():
|
|
logs = self.logs
|
|
else:
|
|
logs = self.thread_logs.get(current_thread_id())
|
|
return logs
|
|
|
|
def empty_logs(self, logs=None):
|
|
if self.quick_log:
|
|
self.quick_log = []
|
|
else:
|
|
if is_main_thread():
|
|
self.logs = []
|
|
else:
|
|
if logs and self.thread_logs.get(current_thread_id()): del self.thread_logs[current_thread_id()]
|
|
|
|
@classmethod
|
|
def add_quick_log(cls, content=''):
|
|
self = cls()
|
|
self.quick_log.append(content)
|
|
return self
|
|
|
|
def notification(self, title, content=''):
|
|
# if sys.platform == 'darwin': # 不太友好 先关闭，之前没考虑到 mac 下会请求权限
|
|
# os.system( 'osascript -e \'tell app "System Events" to display notification "{content}" with title "{title}"\''.format(
|
|
# title=title, content=content))
|
|
pass
|