import logging import sys import random MOOD_KAOMOJI = { "<_mood.happy>": ["(ノ◕ヮ◕)ノ*:・゚✧", "(^▽^)", "( ˘⌣˘)♡(˘⌣˘ )"], "<_mood.excited>": ["(๑˃ᴗ˂)ﻭ"], "<_mood.sad>": ["(。•́︿•̀。)", "(╯︵╰,)", "(ಥ﹏ಥ)", "(︶︹︺)"], "<_mood.anxious>": ["(ノдヽ)", "(◎_◎;)", "(・_・;)", "(゚Д゚;)"], } LEVEL_KAOMOJI = { logging.DEBUG: ["(=^・^=)", "(=・ェ・=)", "(=^-ω-^=)", "(=^‥^=)"], logging.INFO: ["(^• ω •^)", "(=^・ω・^)y=", "(≚ᄌ≚)", "(。♥‿♥。)"], logging.WARNING: ["(; ̄Д ̄)", "(¬_¬;)", "(・ัω・ั)", "(・_・ヾ"], logging.ERROR: ["(╥﹏╥)", "(≧Д≦)", "(;′⌒`)", "(T▽T)"], logging.CRITICAL: ["(╯°□°)╯︵ ┻━┻"], } LEVEL_ANSI_STYLES = { logging.DEBUG: "\033[95m", # light magenta logging.INFO: "\033[96m", # light cyan logging.WARNING: "\033[93m", # light yellow logging.ERROR: "\033[91m", # light red logging.CRITICAL: "\033[35m", # magenta } RESET_ANSI = "\033[0m" class KawaiiFormatter(logging.Formatter): def format(self, record: logging.LogRecord): message = record.msg.strip() ts = self.formatTime(record, "%Y-%m-%d %H:%M:%S") lvl = record.levelname filename = record.filename lineno = str(record.lineno) split_last_word = message.rsplit(" ", 1) last_word = split_last_word[-1] if split_last_word else "" emote_pool = LEVEL_KAOMOJI.get(record.levelno, []) if last_word in MOOD_KAOMOJI: message = split_last_word[0] if len(split_last_word) > 1 else "" emote_pool = MOOD_KAOMOJI.get(last_word, []) emote = random.choice(emote_pool) if emote_pool else "(・ω・)" color = LEVEL_ANSI_STYLES.get(record.levelno, "") formatted = f"[{ts}] {color}[{lvl} {filename}:{lineno}]{RESET_ANSI} {color}{message}{RESET_ANSI} {emote}" return f"{formatted}{RESET_ANSI}" def setup_logger(verbose: bool = False, logger_name: str = "") -> logging.Logger: logger = logging.getLogger(logger_name) logger.setLevel(logging.DEBUG if verbose else logging.INFO) handler = logging.StreamHandler(sys.stdout) handler.setFormatter(KawaiiFormatter()) logger.handlers.clear() logger.addHandler(handler) logger.propagate = False return logger