Add a dedicated function to create a logger

scripts/_helpers.py

@@ -27,11 +27,6 @@
 
 
 def handle_exception(exc_type, exc_value, exc_traceback):
-    logger = logging.getLogger(__name__)
-    logger.setLevel(logging.INFO)
-    handler = logging.StreamHandler(stream=sys.stdout)
-    logger.addHandler(handler)
-
     tb = exc_traceback
     while tb.tb_next:
         tb = tb.tb_next
@@ -55,6 +50,14 @@ def handle_exception(exc_type, exc_value, exc_traceback):
         )
 
 
+def create_logger(log_name, level=logging.INFO):
+    logger = logging.getLogger(log_name)
+    logger.setLevel(level)
+    handler = logging.StreamHandler(stream=sys.stdout)
+    logger.addHandler(handler)
+    sys.excepthook = handle_exception
+
+
 def read_osm_config(*args):
     """
     Read values from the osm_config.yaml file based on provided key arguments.

scripts/build_shapes.py

@@ -37,7 +37,7 @@
 from shapely.validation import make_valid
 from tqdm import tqdm
 
-sys.excepthook = handle_exception
+create_logger(__name___, level=logging.INFO)
 
 sets_path_to_root("pypsa-earth")