From 691fdecff8520434516c63296b349617aac7a2a2 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Fri, 21 Jun 2024 22:20:09 +0100 Subject: [PATCH] Add logging util --- src/editor/utils/__init__.py | 1 + src/editor/utils/set_up_logging.py | 20 ++++++++++++++++++++ 2 files changed, 21 insertions(+) create mode 100644 src/editor/utils/set_up_logging.py diff --git a/src/editor/utils/__init__.py b/src/editor/utils/__init__.py index 4ca51e5..12f8fb1 100644 --- a/src/editor/utils/__init__.py +++ b/src/editor/utils/__init__.py @@ -3,3 +3,4 @@ from .compute_histogram import compute_histogram from .generate_rotation_matrices import generate_rotation_matrices from .get_next_run_name import get_next_run_name from .kldiv import kldiv +from .set_up_logging import set_up_logging diff --git a/src/editor/utils/set_up_logging.py b/src/editor/utils/set_up_logging.py new file mode 100644 index 0000000..defd212 --- /dev/null +++ b/src/editor/utils/set_up_logging.py @@ -0,0 +1,20 @@ +import logging +from datetime import datetime +from typing import Optional + + +def set_up_logging(logs_path: Optional[str] = None): + logging.basicConfig( + level=logging.INFO, + format="%(asctime)s - %(levelname)s - %(message)s", + handlers=[ + logging.StreamHandler(), + ( + logging.FileHandler( + logs_path / f"{datetime.now().isoformat(timespec='minutes')}.log" + ) + if logs_path + else logging.NullHandler() + ), + ], + )