Changes pubsub condition to use REDIS_ENDPOINT and adds more explicit…

… traceback method.
canvas-medical · May 13, 2024 · 15a239c · 15a239c
1 parent 14941bb
commit 15a239c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/logger/__init__.py b/logger/__init__.py
@@ -1,2 +1,2 @@
 from logger.logger import PluginLogger
-log = PluginLogger()
+log = PluginLogger()
diff --git a/logger/logger.py b/logger/logger.py
@@ -3,8 +3,6 @@
 
 from pubsub.pubsub import Publisher
 
-ENV = os.getenv("ENV")
-
 class PubSubLogHandler(logging.Handler):
     def __init__(self)-> None:
         self.publisher = Publisher()
@@ -25,7 +23,7 @@ def __init__(self) -> None:
         streaming_handler.setFormatter(formatter)
         self.logger.addHandler(streaming_handler)
 
-        if ENV == "production":
+        if os.getenv("REDIS_ENDPOINT"):
             pubsub_handler = PubSubLogHandler()
             pubsub_handler.setFormatter(formatter)
             self.logger.addHandler(pubsub_handler)

diff --git a/plugin_runner/plugin_runner.py b/plugin_runner/plugin_runner.py
@@ -60,7 +60,7 @@ async def HandleEvent(self, request: Event, context):
             try:
                 effects = protocol_class(request).compute()
             except Exception as e:
-                log.error(traceback.format_exc())
+                log.error(traceback.format_exception(e))
                 continue
             effect_list += effects