From 343e169e8b527ab534b54e7d465636fa8cb70f68 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Tue, 6 Apr 2021 10:03:43 +0200
Subject: [PATCH] add some logging statements

---
 mlair/data_handler/abstract_data_handler.py       | 2 ++
 mlair/data_handler/data_handler_single_station.py | 1 +
 mlair/data_handler/data_handler_wrf_chem.py       | 2 ++
 mlair/data_handler/default_data_handler.py        | 1 +
 4 files changed, 6 insertions(+)

diff --git a/mlair/data_handler/abstract_data_handler.py b/mlair/data_handler/abstract_data_handler.py
index 419db059..53d45970 100644
--- a/mlair/data_handler/abstract_data_handler.py
+++ b/mlair/data_handler/abstract_data_handler.py
@@ -6,6 +6,7 @@ import inspect
 from typing import Union, Dict
 
 from mlair.helpers import remove_items
+import logging
 
 
 class AbstractDataHandler:
@@ -13,6 +14,7 @@ class AbstractDataHandler:
     _requirements = []
 
     def __init__(self, *args, **kwargs):
+        logging.info("Initialise AbstractDataHandler")
         pass
 
     @classmethod
diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py
index 9fdba43c..163a0a77 100644
--- a/mlair/data_handler/data_handler_single_station.py
+++ b/mlair/data_handler/data_handler_single_station.py
@@ -62,6 +62,7 @@ class DataHandlerSingleStation(AbstractDataHandler):
                  overwrite_local_data: bool = False, transformation=None, store_data_locally: bool = True,
                  min_length: int = 0, start=None, end=None, variables=None, data_origin: Dict = None,
                  lazy_preprocessing: bool = False, aggregation_dim=None, **kwargs):
+        logging.info(f"Initialise DataHandlerSingleStation")
         super().__init__()
         self.station = helpers.to_list(station)
         self.path = self.setup_data_path(data_path, sampling)
diff --git a/mlair/data_handler/data_handler_wrf_chem.py b/mlair/data_handler/data_handler_wrf_chem.py
index 839d1652..69d2cb7f 100644
--- a/mlair/data_handler/data_handler_wrf_chem.py
+++ b/mlair/data_handler/data_handler_wrf_chem.py
@@ -396,6 +396,7 @@ class DataHandlerSingleGridColumn(DataHandlerSingleStation):
     def __init__(self, *args, external_coords_file=None,
                  var_logical_z_coord_selector=None,
                  targetvar_logical_z_coord_selector=None, **kwargs):
+        logging.info("Initialise DataHandlerSingleGridColumn")
         self.external_coords_file = external_coords_file
         self.var_logical_z_coord_selector = self._ret_z_coord_select_if_valid(var_logical_z_coord_selector,
                                                                               as_input=True)
@@ -607,6 +608,7 @@ class DataHandlerSectorGrid(DataHandlerSingleGridColumn):
     _requirements = DataHandlerWRF.requirements()
 
     def __init__(self, *args, radius=None, sectors=None, wind_sector_edge_dim_name=None,  **kwargs):
+        logging.info("Initialise DataHandlerSectorGrid")
         if radius is None:
             radius = 100  # km
         self.radius = radius
diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py
index 15271c80..5a5e8d14 100644
--- a/mlair/data_handler/default_data_handler.py
+++ b/mlair/data_handler/default_data_handler.py
@@ -60,6 +60,7 @@ class DefaultDataHandler(AbstractDataHandler):
 
     @classmethod
     def build(cls, station: str, **kwargs):
+        logging.info(f"build dh: DefaultDataHandler")
         sp_keys = {k: copy.deepcopy(kwargs[k]) for k in cls._requirements if k in kwargs}
         sp = cls.data_handler(station, **sp_keys)
         dp_args = {k: copy.deepcopy(kwargs[k]) for k in cls.own_args("id_class") if k in kwargs}
-- 
GitLab