mirror of
https://github.com/GaMeNu/HFCNotificator.git
synced 2024-11-16 07:14:52 +02:00
106 lines
2.4 KiB
Python
106 lines
2.4 KiB
Python
import datetime
|
|
import logging
|
|
import os
|
|
import traceback
|
|
|
|
import __main__
|
|
from pathlib import Path
|
|
|
|
main_dir = Path(__main__.__file__).parent
|
|
botdata_dir = main_dir.joinpath('botdata')
|
|
errlog_dir = botdata_dir.joinpath('errlogs')
|
|
|
|
|
|
class ErrLogger:
|
|
def __init__(self, handler: logging.Handler = None):
|
|
if handler is None:
|
|
handler = logging.StreamHandler()
|
|
|
|
self.log = logging.Logger('ErrLogger')
|
|
self.log.addHandler(handler)
|
|
|
|
def new_errlog(self, err: BaseException):
|
|
self.log.error('An error has occurred! Check the latest ERRLOG file for more info.')
|
|
e: BaseException = err
|
|
time = datetime.datetime.now()
|
|
path = os.path.join(errlog_dir,
|
|
f'ERRLOG_{time.strftime("%Y-%m-%d_%H-%M-%S")}.txt')
|
|
|
|
context_ls = list()
|
|
context_ls.append(e)
|
|
|
|
ctx = e.__context__
|
|
|
|
if ctx is not None:
|
|
context_ls.append(ctx)
|
|
|
|
context_str = '\n'.join([context.__str__() for context in context_ls])
|
|
|
|
tb_str = '\n\n'.join(
|
|
['\n'.join(traceback.format_tb(exc.__traceback__)) for exc in context_ls]
|
|
)
|
|
|
|
data = f"""An error has occurred! Don't worry, I saved an automatic log for ya :)
|
|
----------------------------------------------------------------------
|
|
Rough DateTime: {time.strftime("%Y-%m-%d %H:%M:%S")}
|
|
|
|
Error Info:
|
|
-----------
|
|
{type(e).__name__}: {e}
|
|
|
|
Context:
|
|
{context_str}
|
|
|
|
Caused by:
|
|
{e.__cause__}
|
|
|
|
Full Traceback:
|
|
---------------
|
|
{tb_str}
|
|
"""
|
|
|
|
with open(path, 'w') as f:
|
|
f.write(data)
|
|
|
|
def errlog(self, func):
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
res = func(*args, **kwargs)
|
|
except Exception as e:
|
|
self.new_errlog(e)
|
|
else:
|
|
return res
|
|
|
|
return wrapper
|
|
|
|
def async_errlog(self, func):
|
|
async def wrapper(*args, **kwargs):
|
|
try:
|
|
res = await func(*args, **kwargs)
|
|
except Exception as e:
|
|
self.new_errlog(e)
|
|
else:
|
|
return res
|
|
|
|
return wrapper
|
|
|
|
|
|
def generate_errlog_folder():
|
|
if not botdata_dir.is_dir():
|
|
botdata_dir.mkdir()
|
|
|
|
if not errlog_dir.is_dir():
|
|
errlog_dir.mkdir()
|
|
|
|
|
|
def new_errlog(err: BaseException):
|
|
ErrLogger().new_errlog(err)
|
|
|
|
|
|
def errlog(func):
|
|
return ErrLogger().errlog(func)
|
|
|
|
|
|
def async_errlog(func):
|
|
return ErrLogger().async_errlog(func)
|