q792602257/Douyin_Web_Live
Archived
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

由outputManager管理整个输出类及解析消息、分流消息功能，debug部分配置对接、xml导出部分配置对接

Browse Source
This commit is contained in:
Jerry Yan
2022-06-05 14:48:59 +08:00
parent 9fe1384b5d
commit 49a0715191
7 changed files with 153 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
output/debug.py
View File

@ -1,26 +1,33 @@
import os
import time
import traceback
from config.helper import config
from output.IOutput import IOutput
class DebugWriter(IOutput):
def __init__(self):
# 获取对应配置文件
self.unknown_output_dir = config()['output']['debug']['save_path']['unknown']
if not os.path.isdir(self.unknown_output_dir):
os.makedirs(self.unknown_output_dir)
self.error_output_dir = config()['output']['debug']['save_path']['error']
if not os.path.isdir(self.error_output_dir):
os.makedirs(self.error_output_dir)
def other_output(self, message_type: str, message_raw: bytes):
if not os.path.isdir(os.path.join("", "debug")):
os.makedirs(os.path.join("", "debug"))
if not os.path.isdir(os.path.join("", "debug", message_type)):
os.makedirs(os.path.join("", "debug", message_type))
with open(os.path.join("", "debug", message_type, str(time.time())), "wb") as f:
if not os.path.isdir(os.path.join(self.unknown_output_dir, message_type)):
os.makedirs(os.path.join(self.unknown_output_dir, message_type))
with open(os.path.join(self.unknown_output_dir, message_type, str(time.time())), "wb") as f:
f.write(message_raw)
def error_output(self, message_type: str, message_raw: bytes, exception: Exception):
if not os.path.isdir(os.path.join("", "error")):
os.makedirs(os.path.join("", "error"))
if not os.path.isdir(os.path.join("", "error", message_type)):
os.makedirs(os.path.join("", "error", message_type))
if not os.path.isdir(os.path.join(self.error_output_dir, message_type)):
os.makedirs(os.path.join(self.error_output_dir, message_type))
ts = time.time()
with open(os.path.join("", "error", message_type, str(ts)), "wb") as f:
with open(os.path.join(self.error_output_dir, message_type, str(ts)), "wb") as f:
f.write(message_raw)
traceback.print_exc(file=open(os.path.join("", "error", message_type, str(ts)) + ".exc", "w", encoding="UTF-8"))
traceback.print_exc(file=open(os.path.join(self.error_output_dir, message_type, str(ts)) + ".exc", "w", encoding="UTF-8"))