✨ add logger

arpes/endstations/fits_utils.py

@@ -5,6 +5,7 @@
 import warnings
 from ast import literal_eval
 from collections.abc import Callable, Iterable
+from logging import DEBUG, Formatter, StreamHandler, getLogger
 from typing import TYPE_CHECKING, Any, TypeAlias
 
 import numpy as np
@@ -22,6 +23,18 @@
     "find_clean_coords",
 )
 
+LOGLEVEL = DEBUG
+logger = getLogger(__name__)
+fmt = "%(asctime)s %(levelname)s %(name)s :%(message)s"
+formatter = Formatter(fmt)
+handler = StreamHandler()
+handler.setLevel(LOGLEVEL)
+logger.setLevel(LOGLEVEL)
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+logger.propagate = False
+
+
 DEFAULT_DIMENSION_RENAMINGS: dict[str, str] = {
     "Beta": "beta",
     "Theta": "theta",

arpes/fits/fit_models/x_model_mixin.py

@@ -3,6 +3,7 @@
 
 import operator
 import warnings
+from logging import DEBUG, Formatter, StreamHandler, getLogger
 from typing import TYPE_CHECKING
 
 import lmfit as lf
@@ -20,6 +21,18 @@
 __all__ = ["XModelMixin", "gaussian_convolve"]
 
 
+LOGLEVEL = DEBUG
+logger = getLogger(__name__)
+fmt = "%(asctime)s %(levelname)s %(name)s :%(message)s"
+formatter = Formatter(fmt)
+handler = StreamHandler()
+handler.setLevel(LOGLEVEL)
+logger.setLevel(LOGLEVEL)
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+logger.propagate = False
+
+
 def dict_to_parameters(dict_of_parameters: dict[str, Any]) -> lf.Parameters:
     """[TODO:summary].
 

arpes/plotting/utils.py

@@ -14,6 +14,7 @@
 from collections import Counter
 from collections.abc import Sequence
 from datetime import UTC
+from logging import DEBUG, Formatter, StreamHandler, getLogger
 from pathlib import Path
 from typing import TYPE_CHECKING, Literal, Unpack
 
@@ -107,6 +108,18 @@
     "h_gradient_fill",
 )
 
+LOGLEVEL = DEBUG
+logger = getLogger(__name__)
+fmt = "%(asctime)s %(levelname)s %(name)s :%(message)s"
+formatter = Formatter(fmt)
+handler = StreamHandler()
+handler.setLevel(LOGLEVEL)
+logger.setLevel(LOGLEVEL)
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+logger.propagate = False
+
+
 TwoDimensional = 2
 
 

arpes/xarray_extensions.py

@@ -44,6 +44,7 @@
 import itertools
 import warnings
 from collections import OrderedDict
+from logging import DEBUG, Formatter, StreamHandler, getLogger
 from typing import TYPE_CHECKING, Any, Literal, TypeAlias
 
 import lmfit
@@ -95,6 +96,17 @@
 
 ANGLE_VARS = ("alpha", "beta", "chi", "psi", "phi", "theta")
 
+LOGLEVEL = DEBUG
+logger = getLogger(__name__)
+fmt = "%(asctime)s %(levelname)s %(name)s :%(message)s"
+formatter = Formatter(fmt)
+handler = StreamHandler()
+handler.setLevel(LOGLEVEL)
+logger.setLevel(LOGLEVEL)
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+logger.propagate = False
+
 
 def _iter_groups(grouped: dict[str, Any]) -> Generator:
     """Iterates through a flattened sequence.