From 1bb74ed8361470fe13660baf53aee361e5040b60 Mon Sep 17 00:00:00 2001
From: Normann <github@koldrack.com>
Date: Mon, 27 Jan 2025 21:18:15 +0100
Subject: [PATCH] replacing import logging (#425)

---
 src/akkudoktoreos/devices/heatpump.py     | 5 +++--
 src/akkudoktoreos/optimization/genetic.py | 3 +--
 src/akkudoktoreos/utils/visualize.py      | 7 +++----
 3 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/src/akkudoktoreos/devices/heatpump.py b/src/akkudoktoreos/devices/heatpump.py
index a4d8424..5df98d5 100644
--- a/src/akkudoktoreos/devices/heatpump.py
+++ b/src/akkudoktoreos/devices/heatpump.py
@@ -1,6 +1,7 @@
-import logging
 from typing import List, Sequence
 
+from akkudoktoreos.core.logging import get_logger
+
 
 class Heatpump:
     MAX_HEAT_OUTPUT = 5000
@@ -21,7 +22,7 @@ class Heatpump:
     def __init__(self, max_heat_output: int, hours: int):
         self.max_heat_output = max_heat_output
         self.hours = hours
-        self.log = logging.getLogger(__name__)
+        self.log = get_logger(__name__)
 
     def __check_outside_temperature_range__(self, temp_celsius: float) -> bool:
         """Check if temperature is in valid range between -100 and 100 degree Celsius.
diff --git a/src/akkudoktoreos/optimization/genetic.py b/src/akkudoktoreos/optimization/genetic.py
index a8118e0..451e715 100644
--- a/src/akkudoktoreos/optimization/genetic.py
+++ b/src/akkudoktoreos/optimization/genetic.py
@@ -1,4 +1,3 @@
-import logging
 import random
 import time
 from typing import Any, Optional
@@ -121,7 +120,7 @@ class optimization_problem(ConfigMixin, DevicesMixin, EnergyManagementSystemMixi
         # Set a fixed seed for random operations if provided or in debug mode
         if self.fix_seed is not None:
             random.seed(self.fix_seed)
-        elif logger.level == logging.DEBUG:
+        elif logger.level == "DEBUG":
             self.fix_seed = random.randint(1, 100000000000)
             random.seed(self.fix_seed)
 
diff --git a/src/akkudoktoreos/utils/visualize.py b/src/akkudoktoreos/utils/visualize.py
index 6e7225a..75c8cf0 100644
--- a/src/akkudoktoreos/utils/visualize.py
+++ b/src/akkudoktoreos/utils/visualize.py
@@ -1,5 +1,4 @@
 import json
-import logging
 import os
 import textwrap
 from collections.abc import Sequence
@@ -615,7 +614,7 @@ def prepare_visualize(
 
     if filtered_balance.size > 0 or filtered_losses.size > 0:
         report.finalize_group()
-    if logger.level == logging.DEBUG or results["fixed_seed"]:
+    if logger.level == "DEBUG" or results["fixed_seed"]:
         report.create_line_chart(
             0,
             [
@@ -711,9 +710,9 @@ def generate_example_report(filename: str = "example_report.pdf") -> None:
 
     report.finalize_group()  # Finalize the third group of charts
 
-    logger.setLevel(logging.DEBUG)  # set level for example report
+    logger.setLevel("DEBUG")  # set level for example report
 
-    if logger.level == logging.DEBUG:
+    if logger.level == "DEBUG":
         report.create_line_chart(
             x_hours,
             [np.array([0.2, 0.25, 0.3, 0.35])],