Niansuh commited on
Commit
7c578c0
·
verified ·
1 Parent(s): 6eca54d

Update api/logger.py

Browse files
Files changed (1) hide show
  1. api/logger.py +7 -1
api/logger.py CHANGED
@@ -1,9 +1,15 @@
 
 
1
  import logging
 
2
 
3
  def setup_logger(name):
4
  logger = logging.getLogger(name)
5
  if not logger.handlers:
6
- logger.setLevel(logging.INFO)
 
 
 
7
  formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
8
 
9
  # Console handler
 
1
+ # api/logger.py
2
+
3
  import logging
4
+ import os
5
 
6
  def setup_logger(name):
7
  logger = logging.getLogger(name)
8
  if not logger.handlers:
9
+ # Set logging level based on environment
10
+ log_level = os.getenv("LOG_LEVEL", "INFO").upper()
11
+ logger.setLevel(getattr(logging, log_level, logging.INFO))
12
+
13
  formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
14
 
15
  # Console handler