logger - copy from blog because it's not logging in prod

1 files changed, 3 insertions, 3 deletions

diff --git a/src/ssh_node5_net.py b/src/ssh_node5_net.py
index 496a631..0fc6bdb 100644
--- a/src/ssh_node5_net.py
+++ b/src/ssh_node5_net.py
@@ -33,14 +33,14 @@ class CustomFormatter(logging.Formatter):
         formatter = logging.Formatter(log_fmt)
         return formatter.format(record)
 
-
 logger = logging.getLogger(__name__)
-
-logger.root.setLevel(logging.DEBUG)
+debug = 'PROD' not in os.environ
+logger.setLevel(logging.DEBUG if debug else logging.WARNING)
 
 stream_handler = logging.StreamHandler(sys.stdout)
 stream_handler.setFormatter(CustomFormatter())
 logger.root.addHandler(stream_handler)
+logger.info(f"Debug: {'en' if debug else 'dis'}abled")
 
 app = flask.Flask(__name__, template_folder='templates', static_folder='static', static_url_path='')