fix errors

src/fillname/utils/logging.py

@@ -1,10 +1,20 @@
 """
 Setup project wide loggers.
+
+This is a thin wrapper around Python's logging module. It supports colored
+logging.
 """
 
 import logging
 from typing import TextIO
 
+NOTSET = logging.NOTSET
+DEBUG = logging.DEBUG
+INFO = logging.INFO
+WARNING = logging.WARNING
+ERROR = logging.ERROR
+CRITICAL = logging.CRITICAL
+
 COLORS = {
     "GREY": "\033[90m",
     "BLUE": "\033[94m",
@@ -41,7 +51,7 @@ def configure_logging(stream: TextIO, level: int) -> None:
     """
     log_message_str = "{}%(levelname)s:{}  - %(message)s{}"
 
-    def make_handler(level: int, color: str) -> logging.StreamHandler:
+    def make_handler(level: int, color: str) -> "logging.StreamHandler[TextIO]":
         handler = logging.StreamHandler(stream)
         handler.addFilter(SingleLevelFilter(level, False))
         handler.setLevel(level)
@@ -58,7 +68,7 @@ def make_handler(level: int, color: str) -> logging.StreamHandler:
     logging.basicConfig(handlers=handlers, level=level)
 
 
-def get_logger(name):
+def get_logger(name: str) -> logging.Logger:
     """
     Get a logger with the given name.
     """

src/fillname/utils/parser.py

@@ -2,12 +2,13 @@
 The command line parser for the project.
 """
 
-import logging
 import sys
 from argparse import ArgumentParser
 from textwrap import dedent
 from typing import Any, Optional, cast
 
+from . import logging
+
 __all__ = ["get_parser"]
 
 if sys.version_info[1] < 8:
@@ -31,7 +32,6 @@ def get_parser() -> ArgumentParser:
             """
         ),
     )
-
     levels = [
         ("error", logging.ERROR),
         ("warning", logging.WARNING),

tests/test_main.py

@@ -2,11 +2,11 @@
 Test cases for main application functionality.
 """
 
-import logging
 from io import StringIO
 from unittest import TestCase
 
-from fillname.utils.logger import setup_logger
+from fillname.utils import logging
+from fillname.utils.logging import configure_logging, get_logger
 from fillname.utils.parser import get_parser
 
 
@@ -19,11 +19,9 @@ def test_logger(self) -> None:
         """
         Test the logger.
         """
-        log = setup_logger("global", logging.INFO)
         sio = StringIO()
-        for handler in log.handlers:
-            assert isinstance(handler, logging.StreamHandler)
-            handler.setStream(sio)
+        configure_logging(sio, logging.INFO)
+        log = get_logger("main")
         log.info("test123")
         self.assertRegex(sio.getvalue(), "test123")