Logging - Disable flask info logs in production

1 files changed, 4 insertions, 4 deletions

diff --git a/src/blog_node5_net.py b/src/blog_node5_net.py
index ee6185b..16e2042 100644
--- a/src/blog_node5_net.py
+++ b/src/blog_node5_net.py
@@ -40,14 +40,14 @@ class CustomFormatter(logging.Formatter):
 
 
 logger = logging.getLogger(__name__)
-if os.environ.get('FLASK_DEBUG') == '1':  # Local development
-    logger.root.setLevel(logging.DEBUG)
-else:  # Production
-    logger.root.setLevel(logging.INFO)
+debug = os.environ.get('FLASK_DEBUG') == '1'
+logger.setLevel(logging.DEBUG if debug else logging.WARNING)
+logging.getLogger('werkzeug').setLevel(logging.DEBUG if debug else logging.WARNING)
 
 stream_handler = logging.StreamHandler()
 stream_handler.setFormatter(CustomFormatter())
 logger.root.addHandler(stream_handler)
+logger.info(f"Debug: {'en' if debug else 'dis'}abled")
 
 with open(os.path.join('configs', 'config.yml'), 'r') as file:
     config = yaml.safe_load(file.read())