From b44c60fa32823dfa39887fd3655f0e05416f45fd Mon Sep 17 00:00:00 2001
From: Paul Kronenwetter <kronenpj@gmail.com>
Date: Tue, 31 May 2022 20:34:22 -0400
Subject: [PATCH] Add flexibility to log_handler.

---
 tnc/log_handler.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/tnc/log_handler.py b/tnc/log_handler.py
index bda36caf..514c2086 100644
--- a/tnc/log_handler.py
+++ b/tnc/log_handler.py
@@ -4,13 +4,13 @@ import structlog
 
 
 # https://www.structlog.org/en/stable/standard-library.html
-def setup_logging(filename):
+def setup_logging(filename: str = "", level: str = "DEBUG"):
     """
 
     Args:
       filename:
-
-    Returns:
+      level:str: Log level to output, possible values are:
+        "CRITICAL", "FATAL", "ERROR", "WARNING", "WARN", "INFO", "DEBUG"
 
     """
 
@@ -40,12 +40,12 @@ def setup_logging(filename):
             },
             "handlers": {
                 "default": {
-                    "level": "DEBUG",
+                    "level": level,
                     "class": "logging.StreamHandler",
                     "formatter": "colored",
                 },
                 "file": {
-                    "level": "DEBUG",
+                    "level": level,
                     "class": "logging.handlers.WatchedFileHandler",
                     "filename": f"{filename}.log",
                     "formatter": "plain",
@@ -53,8 +53,8 @@ def setup_logging(filename):
             },
             "loggers": {
                 "": {
-                    "handlers": ["default", "file"],
-                    "level": "DEBUG",
+                    "handlers": ["default", "file"] if filename else ["default"],
+                    "level": level,
                     "propagate": True,
                 },
             },