mirror of
https://github.com/GaMeNu/HFCNotificator.git
synced 2024-11-16 15:24:51 +02:00
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
import __main__
|
|
import logging
|
|
import logging.handlers
|
|
from pathlib import Path
|
|
|
|
import discord
|
|
|
|
MAIN_DIR = Path(__main__.__file__).parent
|
|
BOTDATA_DIR = MAIN_DIR.joinpath('botdata')
|
|
LOGGING_DIR = BOTDATA_DIR.joinpath('logs')
|
|
|
|
|
|
class ColorFormatter(discord.utils._ColourFormatter):
|
|
"""
|
|
Custom formatter based on Discord.py's color formatter, while changing the color for the datetime
|
|
"""
|
|
|
|
LEVEL_COLOURS = [
|
|
(logging.DEBUG, '\x1b[40;1m', '\x1b[97m'),
|
|
(logging.INFO, '\x1b[34;1m', '\x1b[97;1m'),
|
|
(logging.WARNING, '\x1b[33;1m', '\x1b[93m'),
|
|
(logging.ERROR, '\x1b[31m', '\x1b[91m'),
|
|
(logging.CRITICAL, '\x1b[41m', '\x1b[41;97;1m'),
|
|
]
|
|
FORMATS = {
|
|
level: logging.Formatter(
|
|
f'\x1b[37;1m%(asctime)s\x1b[0m {type_color}%(levelname)-8s\x1b[0m \x1b[35m%(name)s\x1b[0m {msg_color}%(message)s\x1b[0m',
|
|
'%Y-%m-%d %H:%M:%S',
|
|
)
|
|
for level, type_color, msg_color in LEVEL_COLOURS
|
|
}
|
|
|
|
|
|
class DefaultFileHandler(logging.FileHandler):
|
|
def __init__(self, filename: str):
|
|
super().__init__(Path(LOGGING_DIR, filename))
|
|
self.setFormatter(logging.Formatter(
|
|
"[%(asctime)s] [%(levelname)s] %(name)s: %(message)s",
|
|
'%Y-%m-%d %H:%M:%S',
|
|
))
|
|
self.setLevel(logging.INFO)
|
|
|
|
|
|
def generate_logging_folder():
|
|
if not BOTDATA_DIR.is_dir():
|
|
BOTDATA_DIR.mkdir()
|
|
|
|
if not LOGGING_DIR.is_dir():
|
|
LOGGING_DIR.mkdir()
|