discord-rich-presence-plex/utils/logging.py

24 lines
914 B
Python

from typing import Any, Callable
import logging
logger = logging.getLogger("discord-rich-presence-plex")
logger.setLevel(logging.INFO)
formatter = logging.Formatter("[%(asctime)s] [%(levelname)s] %(message)s", datefmt = "%d-%m-%Y %I:%M:%S %p")
streamHandler = logging.StreamHandler()
streamHandler.setFormatter(formatter)
logger.addHandler(streamHandler)
class LoggerWithPrefix:
def __init__(self, prefix: str) -> None:
self.prefix = prefix
self.info = self.wrapLoggerFunc(logger.info)
self.warning = self.wrapLoggerFunc(logger.warning)
self.error = self.wrapLoggerFunc(logger.error)
self.exception = self.wrapLoggerFunc(logger.exception)
self.debug = self.wrapLoggerFunc(logger.debug)
def wrapLoggerFunc(self, func: Callable[..., None]) -> Callable[..., None]:
def wrappedFunc(obj: Any, *args: Any, **kwargs: Any) -> None:
func(self.prefix + str(obj), *args, **kwargs)
return wrappedFunc