fix setting environment name for Mongo logging

Change-Id: Iaa8509743ef924e6cbd2016a875dbf8f44815819
Signed-off-by: Yaron Yogev <yaronyogev@gmail.com>

1 files changed, 3 insertions, 7 deletions

diff --git a/app/utils/logging/full_logger.py b/app/utils/logging/full_logger.py
index a88f00e..411eceb 100644
--- a/app/utils/logging/full_logger.py
+++ b/app/utils/logging/full_logger.py
@@ -35,13 +35,9 @@ class FullLogger(Logger):
     def set_env(self, env):
         super().set_env(env)
 
-        defined_handler = next(
-            filter(
-                lambda handler: handler.__class__ == MongoLoggingHandler.__class__,
-                self.log.handlers
-            ), None)
-
+        defined_handler = [h for h in self.log.handlers
+                           if isinstance(h, MongoLoggingHandler)]
         if defined_handler:
-            defined_handler.env = env
+            defined_handler[0].env = env
         else:
             self.add_handler(MongoLoggingHandler(env, self.level))