updated logger filters

niall-turbitt · Jul 25, 2022 · 47f6cd8 · 47f6cd8
1 parent 39c97fd
commit 47f6cd8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/telco_churn/utils/logger_utils.py b/telco_churn/utils/logger_utils.py
@@ -3,12 +3,14 @@
 
 class NoReceivedCommandFilter(logging.Filter):
     def filter(self, record):
-        return not record.getMessage().startswith('Received command c')
+        if 'Received command c' not in record.getMessage():
+            return record.getMessage()
 
 
 class NoPythonDotEnvFilter(logging.Filter):
     def filter(self, record):
-        return not record.getMessage().startswith('Python-dotenv')
+        if 'Python-dotenv' not in record.getMessage():
+            return record.getMessage()
 
 
 def get_logger():