diff --git a/ait/core/log.py b/ait/core/log.py index 56672ee5..7bea5a87 100644 --- a/ait/core/log.py +++ b/ait/core/log.py @@ -23,6 +23,7 @@ import socket import datetime import time +from typing import Callable, Optional import logging import logging.handlers @@ -38,6 +39,13 @@ logging.addLevelName(COMMAND, "COMMAND") logging.addLevelName(PROGRAM, "PROGRAM") +logger = None +crit: Optional[Callable[[str], str]] = None +debug: Optional[Callable[[str], str]] = None +error: Optional[Callable[[str], str]] = None +info: Optional[Callable[[str], str]] = None +warn: Optional[Callable[[str], str]] = None + class LogFormatter(logging.Formatter): """LogFormatter @@ -312,14 +320,4 @@ def notice(*args, **kwargs): logger.log(NOTICE, *args, **kwargs) -# These are "guaranteed" at runtime to be not-None. Marking this as -# type `Any` addresses mypy issues where log calls are marked as -# "None" being not callable. -logger = None -crit = None -debug = None -error = None -info = None -warn = None - init()