98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
import logging
|
|
import os
|
|
import platform
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
import emoji
|
|
|
|
from TwitchChannelPointsMiner.utils import remove_emoji
|
|
|
|
|
|
class EmojiFormatter(logging.Formatter):
|
|
def __init__(self, *, fmt, datefmt=None, print_emoji=True):
|
|
self.print_emoji = print_emoji
|
|
logging.Formatter.__init__(self, fmt=fmt, datefmt=datefmt)
|
|
|
|
def format(self, record):
|
|
record.emoji_is_present = (
|
|
record.emoji_is_present if hasattr(record, "emoji_is_present") else False
|
|
)
|
|
if (
|
|
hasattr(record, "emoji")
|
|
and self.print_emoji is True
|
|
and record.emoji_is_present is False
|
|
):
|
|
record.msg = emoji.emojize(
|
|
f"{record.emoji} {record.msg.strip()}", use_aliases=True
|
|
)
|
|
record.emoji_is_present = True
|
|
|
|
if self.print_emoji is False:
|
|
if "\u2192" in record.msg:
|
|
record.msg = record.msg.replace("\u2192", "-->")
|
|
|
|
# With the update of Stream class It's possible that the Stream Title contains emoji
|
|
# Full remove using a method from utils.
|
|
record.msg = remove_emoji(record.msg)
|
|
|
|
return super().format(record)
|
|
|
|
|
|
class LoggerSettings:
|
|
def __init__(
|
|
self,
|
|
save: bool = True,
|
|
less: bool = False,
|
|
console_level: int = logging.INFO,
|
|
file_level: int = logging.DEBUG,
|
|
emoji: bool = platform.system() != "Windows",
|
|
):
|
|
self.save = save
|
|
self.less = less
|
|
self.console_level = console_level
|
|
self.file_level = file_level
|
|
self.emoji = emoji
|
|
|
|
|
|
def configure_loggers(username, settings):
|
|
root_logger = logging.getLogger()
|
|
root_logger.setLevel(logging.DEBUG)
|
|
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(settings.console_level)
|
|
console_handler.setFormatter(
|
|
EmojiFormatter(
|
|
fmt=(
|
|
"%(asctime)s - %(levelname)s - [%(funcName)s]: %(message)s"
|
|
if settings.less is False
|
|
else "%(asctime)s - %(message)s"
|
|
),
|
|
datefmt=(
|
|
"%d/%m/%y %H:%M:%S" if settings.less is False else "%d/%m %H:%M:%S"
|
|
),
|
|
print_emoji=settings.emoji,
|
|
)
|
|
)
|
|
root_logger.addHandler(console_handler)
|
|
|
|
if settings.save is True:
|
|
logs_path = os.path.join(Path().absolute(), "logs")
|
|
Path(logs_path).mkdir(parents=True, exist_ok=True)
|
|
logs_file = os.path.join(
|
|
logs_path,
|
|
f"{username}.{datetime.now().strftime('%d%m%Y-%H%M%S')}.log",
|
|
)
|
|
file_handler = logging.FileHandler(logs_file, "w", "utf-8")
|
|
file_handler.setFormatter(
|
|
EmojiFormatter(
|
|
fmt="%(asctime)s - %(levelname)s - %(name)s - [%(funcName)s]: %(message)s",
|
|
datefmt="%d/%m/%y %H:%M:%S",
|
|
print_emoji=settings.emoji,
|
|
)
|
|
)
|
|
file_handler.setLevel(settings.file_level)
|
|
root_logger.addHandler(file_handler)
|
|
return logs_file
|
|
return None
|