logging_revamp: replace modules' logger name

lib/apt_news.py

@@ -8,7 +8,7 @@
 from uaclient.config import UAConfig
 from uaclient.daemon import setup_logging
 
-uaclient_logger = logging.getLogger("uaclient")
+upro_logger = logging.getLogger("ubuntu_pro")
 
 
 def main(cfg: UAConfig):
@@ -28,13 +28,13 @@ def main(cfg: UAConfig):
         logging.INFO,
         logging.DEBUG,
         defaults.CONFIG_DEFAULTS["log_file"],
-        logger=uaclient_logger,
+        logger=upro_logger,
     )
     cfg = UAConfig()
     setup_logging(
         logging.INFO,
         logging.DEBUG,
         log_file=cfg.log_file,
-        logger=uaclient_logger,
+        logger=upro_logger,
     )
     main(cfg)

lib/auto_attach.py

@@ -31,8 +31,8 @@
 )
 from uaclient.files import state_files
 
-uaclient_logger = logging.getLogger("uaclient")
-LOG = logging.getLogger("uaclient.lib.auto_attach")
+upro_logger = logging.getLogger("ubuntu_pro")
+LOG = logging.getLogger("ubuntu_pro.lib.auto_attach")
 
 try:
     import cloudinit.stages as ci_stages  # type: ignore
@@ -108,13 +108,13 @@ def main(cfg: UAConfig):
         logging.INFO,
         logging.DEBUG,
         defaults.CONFIG_DEFAULTS["log_file"],
-        logger=uaclient_logger,
+        logger=upro_logger,
     )
     cfg = UAConfig()
     setup_logging(
         logging.INFO,
         logging.DEBUG,
         log_file=cfg.log_file,
-        logger=uaclient_logger,
+        logger=upro_logger,
     )
     sys.exit(main(cfg))

lib/daemon.py

@@ -12,8 +12,8 @@
     setup_logging,
 )
 
-LOG = logging.getLogger("uaclient.lib.daemon")
-uaclient_logger = logging.getLogger("uaclient")
+LOG = logging.getLogger("ubuntu_pro.lib.daemon")
+upro_logger = logging.getLogger("ubuntu_pro")
 
 
 def main() -> int:
@@ -36,7 +36,7 @@ def main() -> int:
         logging.CRITICAL,
         logging.ERROR,
         log_file=cfg.daemon_log_file,
-        logger=uaclient_logger,
+        logger=upro_logger,
     )
 
     LOG.debug("daemon starting")

lib/esm_cache.py

@@ -7,8 +7,8 @@
 from uaclient.config import UAConfig
 from uaclient.daemon import setup_logging
 
-LOG = logging.getLogger("uaclient.lib.esm_cache")
-uaclient_logger = logging.getLogger("uaclient")
+LOG = logging.getLogger("ubuntu_pro.lib.esm_cache")
+upro_logger = logging.getLogger("ubuntu_pro")
 
 
 def main(cfg: UAConfig) -> None:
@@ -24,13 +24,13 @@ def main(cfg: UAConfig) -> None:
         logging.INFO,
         logging.DEBUG,
         defaults.CONFIG_DEFAULTS["log_file"],
-        logger=uaclient_logger,
+        logger=upro_logger,
     )
     cfg = UAConfig()
     setup_logging(
         logging.INFO,
         logging.DEBUG,
         log_file=cfg.log_file,
-        logger=uaclient_logger,
+        logger=upro_logger,
     )
     main(cfg)

lib/patch_status_json.py

@@ -21,7 +21,7 @@
 from uaclient import system, util
 from uaclient.cli import setup_logging
 
-LOG = logging.getLogger("uaclient.lib.patch_status_json")
+LOG = logging.getLogger("ubuntu_pro.lib.patch_status_json")
 
 
 def patch_status_json_schema_0_1(status_file: str):

lib/reboot_cmds.py

@@ -31,7 +31,7 @@
 from uaclient.entitlements.fips import FIPSEntitlement
 from uaclient.files import notices, state_files
 
-LOG = logging.getLogger("uaclient.lib.reboot_cmds")
+LOG = logging.getLogger("ubuntu_pro.lib.reboot_cmds")
 
 
 def fix_pro_pkg_holds(cfg: config.UAConfig):

lib/timer.py

@@ -19,7 +19,7 @@
 from uaclient.timer.update_contract_info import update_contract_info
 from uaclient.timer.update_messaging import update_motd_messages
 
-LOG = logging.getLogger("uaclient.lib.timer")
+LOG = logging.getLogger("ubuntu_pro.lib.timer")
 UPDATE_MESSAGING_INTERVAL = 21600  # 6 hours
 METERING_INTERVAL = 14400  # 4 hours
 UPDATE_CONTRACT_INFO_INTERVAL = 86400  # 24 hours

uaclient/actions.py

@@ -24,7 +24,7 @@
 )
 from uaclient.files.state_files import timer_jobs_state_file
 
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(pro_log.replace_top_level_logger_name(__name__))
 
 
 UA_SERVICES = (
@@ -231,7 +231,7 @@ def collect_logs(cfg: config.UAConfig, output_dir: str):
                 content,
             )
         except Exception as e:
-            logging.warning(
+            LOG.warning(
                 "Failed to collect user log file: %s\n%s", log_file, str(e)
             )
 

uaclient/api/__init__.py

@@ -1,4 +1,4 @@
 import logging
 
 # setup null handler for all API endpoints
-logging.getLogger("uaclient").addHandler(logging.NullHandler())
+logging.getLogger("ubuntu_pro").addHandler(logging.NullHandler())

uaclient/apt.py

@@ -14,7 +14,7 @@
 import apt  # type: ignore
 import apt_pkg  # type: ignore
 
-from uaclient import event_logger, exceptions, gpg, messages, system
+from uaclient import event_logger, exceptions, gpg, log, messages, system
 from uaclient.defaults import ESM_APT_ROOTDIR
 
 APT_HELPER_TIMEOUT = 60.0  # 60 second timeout used for apt-helper call
@@ -56,7 +56,7 @@
 APT_RETRIES = [1.0, 5.0, 10.0]
 
 event = event_logger.get_event_logger()
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 
 @enum.unique

uaclient/apt_news.py

@@ -7,7 +7,7 @@
 
 import apt_pkg
 
-from uaclient import defaults, messages, system, util
+from uaclient import defaults, log, messages, system, util
 from uaclient.api.u.pro.status.is_attached.v1 import _is_attached
 from uaclient.clouds.identity import get_cloud_type
 from uaclient.config import UAConfig
@@ -25,7 +25,7 @@
     get_contract_expiry_status,
 )
 
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 
 class AptNewsMessageSelectors(DataObject):

uaclient/cli.py

@@ -100,8 +100,8 @@
 )
 
 event = event_logger.get_event_logger()
-uaclient_logger = logging.getLogger("uaclient")
-LOG = logging.getLogger(__name__)
+upro_logger = logging.getLogger("ubuntu_pro")
+LOG = logging.getLogger(pro_log.replace_top_level_logger_name(__name__))
 
 
 class UAArgumentParser(argparse.ArgumentParser):
@@ -1876,7 +1876,7 @@ def _warn_about_output_redirection(cmd_args) -> None:
     ):
         if hasattr(cmd_args, "format") and cmd_args.format in ("json", "yaml"):
             return
-        logging.warning(
+        LOG.warning(
             messages.WARNING_HUMAN_READABLE_OUTPUT.format(
                 command=cmd_args.command
             )
@@ -1901,18 +1901,18 @@ def setup_logging(console_level, log_level, log_file=None, logger=None):
         log_level = log_level.upper()
 
     console_formatter = util.LogFormatter()
-    uaclient_logger.setLevel(log_level)
-    uaclient_logger.addFilter(pro_log.RedactionFilter())
+    upro_logger.setLevel(log_level)
+    upro_logger.addFilter(pro_log.RedactionFilter())
 
     # Clear all handlers, so they are replaced for this logger
-    uaclient_logger.handlers = []
+    upro_logger.handlers = []
 
     # Setup console logging
     console_handler = logging.StreamHandler(sys.stderr)
     console_handler.setFormatter(console_formatter)
     console_handler.setLevel(console_level)
-    console_handler.set_name("ua-console")  # Used to disable console logging
-    uaclient_logger.addHandler(console_handler)
+    console_handler.set_name("upro-console")  # Used to disable console logging
+    upro_logger.addHandler(console_handler)
 
     log_file_path = pathlib.Path(log_file)
 
@@ -1924,8 +1924,8 @@ def setup_logging(console_level, log_level, log_file=None, logger=None):
     file_handler = logging.FileHandler(log_file)
     file_handler.setFormatter(JsonArrayFormatter())
     file_handler.setLevel(log_level)
-    file_handler.set_name("ua-file")
-    uaclient_logger.addHandler(file_handler)
+    file_handler.set_name("upro-file")
+    upro_logger.addHandler(file_handler)
 
 
 def set_event_mode(cmd_args):

uaclient/clouds/aws.py

@@ -1,7 +1,7 @@
 import logging
 from typing import Any, Dict
 
-from uaclient import exceptions, http, system, util
+from uaclient import exceptions, http, log, system, util
 from uaclient.clouds import AutoAttachCloudInstance
 
 IMDS_IPV4_ADDRESS = "169.254.169.254"
@@ -19,7 +19,7 @@
 AWS_TOKEN_PUT_HEADER = "X-aws-ec2-metadata-token"
 AWS_TOKEN_REQ_HEADER = AWS_TOKEN_PUT_HEADER + "-ttl-seconds"
 
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 
 class UAAutoAttachAWSInstance(AutoAttachCloudInstance):

uaclient/clouds/identity.py

@@ -3,10 +3,10 @@
 from functools import lru_cache
 from typing import Dict, Optional, Tuple, Type  # noqa: F401
 
-from uaclient import clouds, exceptions, system
+from uaclient import clouds, exceptions, log, system
 from uaclient.config import apply_config_settings_override
 
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 CLOUD_TYPE_TO_TITLE = {
     "aws": "AWS",

uaclient/config.py

@@ -12,6 +12,7 @@
     exceptions,
     files,
     http,
+    log,
     messages,
     snap,
     system,
@@ -31,7 +32,7 @@
 from uaclient.files.notices import Notice
 from uaclient.yaml import safe_load
 
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 PRIVATE_SUBDIR = "private"
 MERGE_ID_KEY_MAP = {

uaclient/conftest.py

@@ -108,7 +108,7 @@ def _func():
 
     except LookupError:
         # If the caplog fixture isn't available, shim something in ourselves
-        root = logging.getLogger("uaclient")
+        root = logging.getLogger("ubuntu_pro")
         root.propagate = False
         root.setLevel(log_level)
         handler = logging.StreamHandler(io.StringIO())
@@ -135,13 +135,13 @@ def clear_handlers():
 def logging_sandbox():
     # Monkeypatch a replacement root logger, so that our changes to logging
     # configuration don't persist outside of the test
-    uaclient_logger = logging.getLogger("uaclient")
-    uaclient_logger.setLevel(logging.WARNING)
+    upro_logger = logging.getLogger("ubuntu_pro")
+    upro_logger.setLevel(logging.WARNING)
 
-    with mock.patch.object(logging, "root", uaclient_logger):
-        with mock.patch.object(logging.Logger, "root", uaclient_logger):
+    with mock.patch.object(logging, "root", upro_logger):
+        with mock.patch.object(logging.Logger, "root", upro_logger):
             with mock.patch.object(
-                logging.Logger, "manager", logging.Manager(uaclient_logger)
+                logging.Logger, "manager", logging.Manager(upro_logger)
             ):
                 yield
 

uaclient/contract.py

@@ -8,6 +8,7 @@
     event_logger,
     exceptions,
     http,
+    log,
     messages,
     system,
     util,
@@ -48,7 +49,7 @@
 }
 
 event = event_logger.get_event_logger()
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 
 class UAContractClient(serviceclient.UAServiceClient):

uaclient/daemon/__init__.py

@@ -10,7 +10,7 @@
 from uaclient.defaults import DEFAULT_DATA_DIR
 from uaclient.log import JsonArrayFormatter
 
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(pro_log.replace_top_level_logger_name(__name__))
 
 AUTO_ATTACH_STATUS_MOTD_FILE = os.path.join(
     DEFAULT_DATA_DIR, "messages", "motd-auto-attach-status"
@@ -50,11 +50,11 @@ def setup_logging(console_level, log_level, log_file, logger):
     console_handler = logging.StreamHandler(sys.stderr)
     console_handler.setFormatter(logging.Formatter("%(message)s"))
     console_handler.setLevel(console_level)
-    console_handler.set_name("ua-console")
+    console_handler.set_name("upro-console")
     logger.addHandler(console_handler)
 
     file_handler = logging.FileHandler(log_file)
     file_handler.setFormatter(JsonArrayFormatter())
     file_handler.setLevel(log_level)
-    file_handler.set_name("ua-file")
+    file_handler.set_name("upro-file")
     logger.addHandler(file_handler)

uaclient/data_types.py

@@ -4,9 +4,9 @@
 from enum import Enum
 from typing import Any, List, Optional, Type, TypeVar, Union
 
-from uaclient import exceptions, messages, util
+from uaclient import exceptions, log, messages, util
 
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 
 class IncorrectTypeError(exceptions.UserFacingError):

uaclient/entitlements/base.py

@@ -6,7 +6,15 @@
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Set, Tuple, Type, Union
 
-from uaclient import config, contract, event_logger, messages, system, util
+from uaclient import (
+    config,
+    contract,
+    event_logger,
+    log,
+    messages,
+    system,
+    util,
+)
 from uaclient.api.u.pro.status.is_attached.v1 import _is_attached
 from uaclient.defaults import DEFAULT_HELP_FILE
 from uaclient.entitlements.entitlement_status import (
@@ -24,7 +32,7 @@
 from uaclient.yaml import safe_load
 
 event = event_logger.get_event_logger()
-LOG = logging.getLogger(__name__)
+LOG = logging.getLogger(log.replace_top_level_logger_name(__name__))
 
 
 class IncompatibleService: