|
| 1 | +#! python3 # noqa: E265 |
| 2 | + |
| 3 | +"""Helper to configure logging depending on CLI options.""" |
| 4 | + |
| 5 | +# ############################################################################ |
| 6 | +# ########## IMPORTS ############# |
| 7 | +# ################################ |
| 8 | + |
| 9 | +# standard library |
| 10 | +import logging |
| 11 | +from getpass import getuser |
| 12 | +from logging.handlers import RotatingFileHandler |
| 13 | +from os import environ, getenv |
| 14 | +from os.path import expanduser, expandvars |
| 15 | +from pathlib import Path |
| 16 | +from platform import architecture, platform |
| 17 | +from socket import gethostname |
| 18 | +from typing import Optional |
| 19 | + |
| 20 | +# package |
| 21 | +from geotribu_cli.__about__ import __title__, __version__ |
| 22 | +from geotribu_cli.constants import GeotribuDefaults |
| 23 | +from geotribu_cli.utils.check_path import check_path |
| 24 | +from geotribu_cli.utils.proxies import get_proxy_settings |
| 25 | +from geotribu_cli.utils.str2bool import str2bool |
| 26 | + |
| 27 | +# ############################################################################ |
| 28 | +# ########## GLOBALS ############# |
| 29 | +# ################################ |
| 30 | + |
| 31 | +# logs |
| 32 | +logger = logging.getLogger(__name__) |
| 33 | +default_settings = GeotribuDefaults() |
| 34 | + |
| 35 | +# ############################################################################ |
| 36 | +# ########## FUNCTIONS ########### |
| 37 | +# ################################ |
| 38 | + |
| 39 | + |
| 40 | +def configure_logger(verbosity: int = 1, logfile: Optional[Path] = None): |
| 41 | + """Configure logging according to verbosity from CLI. |
| 42 | +
|
| 43 | + Args: |
| 44 | + verbosity (int): verbosity level |
| 45 | + logfile (Path, optional): file where to store log. Defaults to None. |
| 46 | + """ |
| 47 | + # handle log level overridden by environment variable |
| 48 | + verbosity = getenv("GEOTRIBU_LOGS_LEVEL", verbosity) |
| 49 | + try: |
| 50 | + verbosity = int(verbosity) |
| 51 | + except ValueError as err: |
| 52 | + logger.error(f"Bad verbosity value type: {err}. Fallback to 1.") |
| 53 | + verbosity = 1 |
| 54 | + |
| 55 | + # set log level depending on verbosity argument |
| 56 | + if 0 < verbosity < 4: |
| 57 | + verbosity = 40 - (10 * verbosity) |
| 58 | + elif verbosity >= 4: |
| 59 | + # debug is the limit |
| 60 | + verbosity = 40 - (10 * 3) |
| 61 | + else: |
| 62 | + verbosity = 0 |
| 63 | + |
| 64 | + # set console handler |
| 65 | + log_console_handler = logging.StreamHandler() |
| 66 | + log_console_handler.setLevel(verbosity) |
| 67 | + |
| 68 | + # set log file |
| 69 | + if not logfile: |
| 70 | + logging.basicConfig( |
| 71 | + level=verbosity, |
| 72 | + format="%(asctime)s||%(levelname)s||%(module)s||%(funcName)s||%(lineno)d||%(message)s", |
| 73 | + datefmt="%Y-%m-%d %H:%M:%S", |
| 74 | + handlers=[log_console_handler], |
| 75 | + ) |
| 76 | + |
| 77 | + else: |
| 78 | + if getenv("GEOTRIBU_LOGS_DIR") and check_path( |
| 79 | + input_path=Path(expandvars(expanduser(getenv("GEOTRIBU_LOGS_DIR")))), |
| 80 | + must_be_a_file=False, |
| 81 | + must_be_a_folder=True, |
| 82 | + must_be_writable=True, |
| 83 | + raise_error=False, |
| 84 | + ): |
| 85 | + logs_folder = Path(expandvars(expanduser(getenv("GEOTRIBU_LOGS_DIR")))) |
| 86 | + logger.debug( |
| 87 | + f"Logs folder set with GEOTRIBU_LOGS_DIR environment variable: {logs_folder}" |
| 88 | + ) |
| 89 | + else: |
| 90 | + logs_folder: Path = default_settings.geotribu_working_folder.joinpath( |
| 91 | + "logs" |
| 92 | + ) |
| 93 | + logger.debug( |
| 94 | + "Logs folder specified in GEOTRIBU_LOGS_DIR environment variable " |
| 95 | + f"{getenv('GEOTRIBU_LOGS_DIR')} can't be used (see logs above). Fallback on " |
| 96 | + f"default folder: {logs_folder}" |
| 97 | + ) |
| 98 | + |
| 99 | + # make sure folder exists |
| 100 | + logs_folder.mkdir(exist_ok=True, parents=True) |
| 101 | + logs_filepath = Path(logs_folder, logfile) |
| 102 | + |
| 103 | + log_file_handler = RotatingFileHandler( |
| 104 | + backupCount=10, |
| 105 | + delay=True, |
| 106 | + encoding="UTF-8", |
| 107 | + filename=logs_filepath, |
| 108 | + maxBytes=3000000, |
| 109 | + mode="a", |
| 110 | + ) |
| 111 | + # force new file by execution |
| 112 | + if logs_filepath.is_file(): |
| 113 | + log_file_handler.doRollover() |
| 114 | + |
| 115 | + logging.basicConfig( |
| 116 | + level=verbosity, |
| 117 | + format="%(asctime)s||%(levelname)s||%(module)s||%(funcName)s||%(lineno)d||%(message)s", |
| 118 | + datefmt="%Y-%m-%d %H:%M:%S", |
| 119 | + handlers=[log_console_handler, log_file_handler], |
| 120 | + ) |
| 121 | + |
| 122 | + logger.info(f"Log file: {logs_filepath}") |
| 123 | + |
| 124 | + headers() |
| 125 | + |
| 126 | + |
| 127 | +def headers(): |
| 128 | + """Basic information to log before other message.""" |
| 129 | + # initialize the log |
| 130 | + logger.info(f"{'='*10} {__title__} - {__version__} {'='*10}") |
| 131 | + logger.debug(f"Operating System: {platform()}") |
| 132 | + logger.debug(f"Architecture: {architecture()[0]}") |
| 133 | + logger.debug(f"Computer: {gethostname()}") |
| 134 | + logger.debug(f"Launched by user: {getuser()}") |
| 135 | + |
| 136 | + if getenv("userdomain"): |
| 137 | + logger.debug(f"OS Domain: {getenv('userdomain')}") |
| 138 | + |
| 139 | + if get_proxy_settings(): |
| 140 | + logger.debug(f"Network proxies detected: {get_proxy_settings()}") |
| 141 | + else: |
| 142 | + logger.debug("No network proxies detected") |
| 143 | + |
| 144 | + if str2bool(getenv("QDT_SSL_USE_SYSTEM_STORES", False)): |
| 145 | + logger.debug("Option to use native system certificates stores is enabled.") |
| 146 | + if "REQUESTS_CA_BUNDLE" in environ: |
| 147 | + environ.pop("REQUESTS_CA_BUNDLE") |
| 148 | + logger.debug( |
| 149 | + "Custom path to CA Bundle (REQUESTS_CA_BUNDLE) has been removed from " |
| 150 | + "environment variables." |
| 151 | + ) |
| 152 | + if "CURL_CA_BUNDLE" in environ: |
| 153 | + environ.pop("CURL_CA_BUNDLE") |
| 154 | + logger.debug( |
| 155 | + "Custom path to CA Bundle (CURL_CA_BUNDLE) has been removed from " |
| 156 | + "environment variables." |
| 157 | + ) |
| 158 | + |
| 159 | + |
| 160 | +def get_logger_filepath() -> Optional[Path]: |
| 161 | + """Retrieve log filepath within logger handlers. |
| 162 | +
|
| 163 | + Returns: |
| 164 | + Path | None: path to the logfile or None if no handler has baseFilename attr. |
| 165 | + """ |
| 166 | + if logger.root.hasHandlers(): |
| 167 | + for handler in logger.root.handlers: |
| 168 | + if hasattr(handler, "baseFilename"): |
| 169 | + return Path(handler.baseFilename) |
| 170 | + |
| 171 | + logger.warning("No file found in ay log handlers.") |
| 172 | + return None |
0 commit comments