fix log
Browse files- app/core/logger.py +11 -8
app/core/logger.py
CHANGED
@@ -26,9 +26,13 @@ class Logger:
|
|
26 |
def __init__(self, json_logs: bool = False, log_level: str = "INFO"):
|
27 |
self.json_logs = json_logs
|
28 |
self.log_level = log_level.upper()
|
|
|
29 |
|
30 |
-
|
31 |
-
self.
|
|
|
|
|
|
|
32 |
|
33 |
def _get_default_log_file_path(self) -> str:
|
34 |
"""Get the default log file path."""
|
@@ -132,12 +136,11 @@ class Logger:
|
|
132 |
stream_handler.setFormatter(formatter)
|
133 |
root_logger.addHandler(stream_handler)
|
134 |
|
135 |
-
# Only add file logging if
|
136 |
-
|
137 |
-
if log_file_path and os.getenv("ENVIRONMENT") != "HUGGINGFACE":
|
138 |
try:
|
139 |
file_handler = TimedRotatingFileHandler(
|
140 |
-
filename=log_file_path,
|
141 |
when="midnight",
|
142 |
interval=1,
|
143 |
backupCount=7,
|
@@ -146,10 +149,10 @@ class Logger:
|
|
146 |
file_handler.setFormatter(formatter)
|
147 |
root_logger.addHandler(file_handler)
|
148 |
except PermissionError:
|
149 |
-
# If file logging fails, just
|
150 |
pass
|
151 |
|
152 |
-
root_logger.setLevel(self.log_level
|
153 |
return root_logger
|
154 |
|
155 |
def _configure(self):
|
|
|
26 |
def __init__(self, json_logs: bool = False, log_level: str = "INFO"):
|
27 |
self.json_logs = json_logs
|
28 |
self.log_level = log_level.upper()
|
29 |
+
self.environment = os.getenv("ENVIRONMENT", "PROD").upper()
|
30 |
|
31 |
+
# Skip file logging in Hugging Face environment
|
32 |
+
if self.environment == "HUGGINGFACE":
|
33 |
+
self.log_file_path = None
|
34 |
+
else:
|
35 |
+
self.log_file_path = os.getenv("LOG_FILE_PATH", self._get_default_log_file_path())
|
36 |
|
37 |
def _get_default_log_file_path(self) -> str:
|
38 |
"""Get the default log file path."""
|
|
|
136 |
stream_handler.setFormatter(formatter)
|
137 |
root_logger.addHandler(stream_handler)
|
138 |
|
139 |
+
# Only add file logging if not in Hugging Face environment and log_file_path is set
|
140 |
+
if self.environment != "HUGGINGFACE" and self.log_file_path:
|
|
|
141 |
try:
|
142 |
file_handler = TimedRotatingFileHandler(
|
143 |
+
filename=self.log_file_path,
|
144 |
when="midnight",
|
145 |
interval=1,
|
146 |
backupCount=7,
|
|
|
149 |
file_handler.setFormatter(formatter)
|
150 |
root_logger.addHandler(file_handler)
|
151 |
except PermissionError:
|
152 |
+
# If file logging fails, just continue with console logging
|
153 |
pass
|
154 |
|
155 |
+
root_logger.setLevel(self.log_level)
|
156 |
return root_logger
|
157 |
|
158 |
def _configure(self):
|