Configure custom logging (and different for root vs. package)

Resolves #37. Resolves #32.

Author: jspahrsummers

blotter/__main__.py

@@ -1,22 +1,22 @@
 import asyncio
-import logging
 import random
 import signal
 import sys
 from argparse import ArgumentParser
 from datetime import timedelta
+from logging import getLogger
 from types import TracebackType
 from typing import Type
 
-import google.cloud.logging
 import ib_insync
 
 from blotter.error_handling import ErrorHandlerConfiguration
 from blotter.ib_helpers import IBError, IBThread, IBWarning
+from blotter.logging import configure_package_logger, configure_root_logger
 from blotter.server import Servicer
 from blotter.streaming import StreamingManager
 
-logger = logging.getLogger(__package__)
+logger = getLogger(__package__)
 
 parser = ArgumentParser(
     prog="blotter",
@@ -89,14 +89,9 @@ def _hook(
 def main() -> None:
     args = parser.parse_args()
 
-    google.cloud.logging.Client().setup_logging(
-        log_level=logging.DEBUG if args.verbose >= 2 else logging.INFO
-    )
-
-    if args.verbose:
-        logging.basicConfig(level=logging.DEBUG if args.verbose >= 2 else logging.INFO)
-
     error_handler = ErrorHandlerConfiguration(report_to_gcloud=args.report_errors)
+    configure_root_logger(args.verbose)
+    configure_package_logger(logger, args.verbose)
 
     ib = ib_insync.IB()
 

blotter/logging.py

@@ -0,0 +1,48 @@
+import logging
+import google.cloud.logging
+
+
+def install_gcloud_logging_handler(logger: logging.Logger, level: int) -> None:
+    handler = google.cloud.logging.handlers.CloudLoggingHandler(
+        google.cloud.logging.Client()
+    )
+
+    handler.setLevel(level)
+    logger.addHandler(handler)
+
+
+def install_console_logging_handler(logger: logging.Logger) -> None:
+    handler = logging.StreamHandler()
+    handler.setLevel(logging.DEBUG)
+
+    formatter = logging.Formatter("%(asctime)s [%(name)s][%(levelname)s] %(message)s")
+    handler.setFormatter(formatter)
+
+    logger.addHandler(handler)
+
+
+def configure_root_logger(verbosity: int) -> None:
+    logger = logging.getLogger()
+
+    if verbosity >= 3:
+        logger.setLevel(logging.DEBUG)
+    elif verbosity >= 2:
+        logger.setLevel(logging.INFO)
+    else:
+        logger.setLevel(logging.WARN)
+
+    install_console_logging_handler(logger)
+    install_gcloud_logging_handler(logger, logging.WARN)
+
+
+def configure_package_logger(logger: logging.Logger, verbosity: int) -> None:
+    if verbosity >= 2:
+        logger.setLevel(logging.DEBUG)
+    elif verbosity >= 1:
+        logger.setLevel(logging.INFO)
+    else:
+        logger.setLevel(logging.WARN)
+
+    install_console_logging_handler(logger)
+    install_gcloud_logging_handler(logger, logging.INFO)
+    logger.propagate = False