From 9fa7e144593b50e6152ba4c3cf1c38c4d374ba20 Mon Sep 17 00:00:00 2001 From: tsmorz Date: Thu, 30 Oct 2025 10:57:42 +0100 Subject: [PATCH 1/2] set logger level from main as an argument --- .coverage | Bin 53248 -> 53248 bytes src/change_me/__main__.py | 9 +++++++-- src/change_me/definitions.py | 10 ++++++++-- 3 files changed, 15 insertions(+), 4 deletions(-) diff --git a/.coverage b/.coverage index b4c119fca9d21fe12b318eba9414cd95f0c9e6a5..7cad5edaf143ac8e6bff245c271c32dacc606f68 100644 GIT binary patch delta 46 zcmV+}0MY+|paX!Q1F!~w2;Kk>`48+5-m?)9&<~T}j}8qB0t5jG7y?@HRPHYXv-yt) EK#!>qnE(I) delta 45 zcmV+|0Mh?}paX!Q1F!~w2;Tq?`48<6-?I@A(GQc}j}8n90t5jG76MxFRPGbA`Hu%c Dfcp@5 diff --git a/src/change_me/__main__.py b/src/change_me/__main__.py index 60bef99..1304e39 100644 --- a/src/change_me/__main__.py +++ b/src/change_me/__main__.py @@ -4,7 +4,7 @@ from loguru import logger -from change_me.definitions import DEFAULT_LOG_LEVEL +from change_me.definitions import DEFAULT_LOG_LEVEL, LogLevel from change_me.utils import setup_logger @@ -16,7 +16,12 @@ def main(log_level: str) -> None: if __name__ == "__main__": parser = argparse.ArgumentParser() - parser.add_argument("--log-level", default=DEFAULT_LOG_LEVEL, type=str) + parser.add_argument( + "--log-level", + default=DEFAULT_LOG_LEVEL, + choices=list(LogLevel), + help="Log level.", + ) args = parser.parse_args() main(log_level=args.log_level) diff --git a/src/change_me/definitions.py b/src/change_me/definitions.py index 973e831..a961428 100644 --- a/src/change_me/definitions.py +++ b/src/change_me/definitions.py @@ -1,6 +1,6 @@ """Common definitions for this module.""" -from dataclasses import dataclass +from dataclasses import asdict, dataclass from pathlib import Path import numpy as np @@ -24,13 +24,19 @@ @dataclass class LogLevel: - """Log levels for loguru.""" + """Log level.""" + trace: str = "TRACE" debug: str = "DEBUG" info: str = "INFO" + success: str = "SUCCESS" warning: str = "WARNING" error: str = "ERROR" critical: str = "CRITICAL" + def __iter__(self): + """Iterate over log levels.""" + return iter(asdict(self).values()) + DEFAULT_LOG_LEVEL = LogLevel.info From f8df3a4b444a03834fd3c009a281863c6ca4d13b Mon Sep 17 00:00:00 2001 From: tsmorz Date: Thu, 30 Oct 2025 10:58:06 +0100 Subject: [PATCH 2/2] poetry update --- .coverage | Bin 53248 -> 53248 bytes poetry.lock | 488 +++++++++++++++++++++++++--------------------------- 2 files changed, 238 insertions(+), 250 deletions(-) diff --git a/.coverage b/.coverage index 7cad5edaf143ac8e6bff245c271c32dacc606f68..ade57272641b575e588ae1bec4368a66d469f76d 100644 GIT binary patch delta 16 XcmZozz}&EadBb0QW