refactor: make logging level configurable

Signed-off-by: bossenti <[email protected]>
bossenti · Oct 19, 2023 · 043e43a · 043e43a
1 parent 8c14442
commit 043e43a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/vistafetch/client.py b/vistafetch/client.py
@@ -24,8 +24,8 @@ class VistaFetchClient:
         client_headers: additional headers to be sent with every request, optional
     """
 
-    def __init__(self, client_headers: Optional[Dict[str, str]] = None):
-        set_up_logging()
+    def __init__(self, client_headers: Optional[Dict[str, str]] = None, logging_level: Optional[int] = None):
+        set_up_logging(logging_level=logging_level)
 
         http_headers = {"Application": "application/json; charset=utf-8"}
 

diff --git a/vistafetch/logs.py b/vistafetch/logs.py
@@ -8,11 +8,11 @@
 ]
 
 
-def set_up_logging() -> None:
+def set_up_logging(logging_level: int = logging.INFO) -> None:
     """Configure logging."""
     log_format = "%(message)s"
     logging.basicConfig(
-        level="NOTSET", format=log_format, datefmt="[%X]", handlers=[RichHandler()]
+        level=logging_level, format=log_format, datefmt="[%X]", handlers=[RichHandler()]
     )
 
     logging.debug("Logging set up successfully.")