canvas 0.22.0__py3-none-any.whl → 0.22.1__py3-none-any.whl

{canvas-0.22.0.dist-info → canvas-0.22.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: canvas
-Version: 0.22.0
+Version: 0.22.1
 Summary: SDK to customize event-driven actions in your Canvas instance
 Author-email: Canvas Team <engineering@canvasmedical.com>
 License-Expression: MIT

{canvas-0.22.0.dist-info → canvas-0.22.1.dist-info}/RECORD

@@ -150,9 +150,9 @@ canvas_sdk/commands/commands/vitals.py,sha256=jidkqVNpEqVewMBHztY0p7xkfk9xdBNvGS
 canvas_sdk/commands/commands/questionnaire/__init__.py,sha256=huAj5B93JW_tqur-s1w-mzZia_QPIYvbBngHgCyVFDk,3635
 canvas_sdk/commands/commands/questionnaire/question.py,sha256=h0AyCfMqpIMUMFZ-4xCL-vEbrhkgTWnOGVj5MTOKZjw,4334
 canvas_sdk/commands/tests/test_base_command.py,sha256=k5I4psjNX4VgfC4XWHuG0sJdgUa58DJP7WQQ3QncMtQ,2667
-canvas_sdk/commands/tests/test_utils.py,sha256=Jrkbz2xltrjUVYRdIyNaKdq5tCtGQ4vygic7hpBp6Ek,12157
+canvas_sdk/commands/tests/test_utils.py,sha256=4XBQLJiWqs3frqgofA9I9o2wvi8Og2378QkcS4kgYpI,12275
 canvas_sdk/commands/tests/protocol/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-canvas_sdk/commands/tests/protocol/tests.py,sha256=e0fiJUKsfi0aYSlaTdrMHybZntC7oBdaHks9PSrMuhU,2168
+canvas_sdk/commands/tests/protocol/tests.py,sha256=ApG6R66e7-PyMTAPcVHVujvdEmCOEFVlcORtYs4KD7c,2651
 canvas_sdk/commands/tests/schema/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 canvas_sdk/commands/tests/schema/tests.py,sha256=rSu146RouMrPLv0G-E6QtgQhG2vkUCCk1M3Baa6fXBg,3701
 canvas_sdk/commands/tests/unit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -167,7 +167,7 @@ canvas_sdk/effects/show_button.py,sha256=JnW9nM8S_GUXIOufs-uef3pg0HPDxSbF0l51Wh1
 canvas_sdk/effects/banner_alert/__init__.py,sha256=PcMmOjLHP_MOZiP8157JkTdcO4mZTn-kFcRylOB9AK0,209
 canvas_sdk/effects/banner_alert/add_banner_alert.py,sha256=W4Fv9IHGCWvVtFiH3s68JBRpUb8b3xsncGt2WfgMFc0,1681
 canvas_sdk/effects/banner_alert/remove_banner_alert.py,sha256=PK_LzXBQ-rOH3fX182ISyutu4W69xyGgMX-1G4g4XhY,613
-canvas_sdk/effects/banner_alert/tests.py,sha256=DoCQ9ZfpcUBne-Le-zCCOTjzF_WHyuiay-FCex4aHJ0,9665
+canvas_sdk/effects/banner_alert/tests.py,sha256=ddSIiyD1sIzWbZmJEHcSN8cXn2R-0-CqD7jyDzqZOzE,10179
 canvas_sdk/effects/billing_line_item/__init__.py,sha256=IiuClemwdeMNku_uyphQ8OIE7xLiNg5sgPpTvdU3uJw,393
 canvas_sdk/effects/billing_line_item/add_billing_line_item.py,sha256=TJp3KKhpDK48T9kxvD7rOttFx6OLb6wfS5Y2Pmg9sGk,1019
 canvas_sdk/effects/billing_line_item/remove_billing_line_item.py,sha256=tUCv9NVTVeDixkJM_TNN9TcjUvefR0GA3IOzVVDdD6E,585
@@ -266,8 +266,8 @@ plugin_runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plugin_runner/authentication.py,sha256=SDPso2AogtLAV_H0LuMDp99IMZuF3oTq-Q_AXAvJ8uc,1116
 plugin_runner/aws_headers.py,sha256=DenX_nAMVhXMJZw88PLZbqJsi5_XriNtr3jE-eJqHY4,2773
 plugin_runner/exceptions.py,sha256=YnRZiQVzbU3HrVlmEXLje_np99009YnhTRVHHyBCtqc,433
-plugin_runner/installation.py,sha256=ix-GvllnnQMSwbs9r0FxgWiyd0GsTsaOIrdgYytfxBo,7596
-plugin_runner/plugin_runner.py,sha256=XfNbZLvp04GoVkOCj1SYUuuOmqWV18GJslFoshRUd5E,17173
+plugin_runner/installation.py,sha256=-TntCAveju5vWrKRLnIxy9xn3pnU3goo5dT4tGs-85s,7537
+plugin_runner/plugin_runner.py,sha256=8f0wJp1QirzKQjwmIZzVgTwtx0OxVKY4-xTiBxhFJHQ,18023
 plugin_runner/sandbox.py,sha256=P64SKpKZdoJwk2NsirGRKq7xjygBtvdipIAVSD5roY8,14509
 plugin_runner/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plugin_runner/tests/test_application.py,sha256=e1R2YagMRD96gZALx-Zra-e-sR3SiP7cIpI6pheZnUc,2427
@@ -340,7 +340,7 @@ protobufs/canvas_generated/messages/plugins.proto,sha256=oNainUPWFYQjgCX7bJEPI9_
 protobufs/canvas_generated/services/plugin_runner.proto,sha256=doadBKn5k4xAtOgR-q_pEvW4yzxpUaHNOowMG6CL5GY,304
 pubsub/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pubsub/pubsub.py,sha256=pyTW0JU8mtaqiAV6g6xjZwel1CVy2EonPMU-_vkmhUM,1044
-canvas-0.22.0.dist-info/METADATA,sha256=q6j_FoE3wxCTZldQsWQolKZmxqi23_NU90Evpdqcebs,4375
-canvas-0.22.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-canvas-0.22.0.dist-info/entry_points.txt,sha256=0Vs_9GmTVUNniH6eDBlRPgofmADMV4BES6Ao26M4AbM,47
-canvas-0.22.0.dist-info/RECORD,,
+canvas-0.22.1.dist-info/METADATA,sha256=5XB3DdvBg3cfMCwFEHZE6vhAZPhk0u93rWhD0SVhd0E,4375
+canvas-0.22.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+canvas-0.22.1.dist-info/entry_points.txt,sha256=0Vs_9GmTVUNniH6eDBlRPgofmADMV4BES6Ao26M4AbM,47
+canvas-0.22.1.dist-info/RECORD,,

canvas_sdk/commands/tests/protocol/tests.py

@@ -49,7 +49,16 @@ def write_and_install_protocol_and_clean_up(
         f"Loading plugin '{plugin_name}",
     )
     install_plugin(plugin_name, token)
-    message_received_event.wait(timeout=5.0)
+    message_received_event.wait(timeout=15.0)
+
+    # unfortunately sometimes the log websocket just doesn't return any
+    # messages, so asserting on the state of the timeout here causes failures
+    # even though the delay itself will cause the waiting test to pass (because
+    # the plugin has been loaded).
+    # timeout_not_hit = message_received_event.wait(timeout=15.0)
+    # if not timeout_not_hit:
+    #     ws.close()
+    # assert timeout_not_hit, f"plugin loading message timeout hit: Loading plugin '{plugin_name}"
 
     yield
 

canvas_sdk/commands/tests/test_utils.py

@@ -367,6 +367,9 @@ def wait_for_log(
     thread = threading.Thread(target=ws.run_forever)
     thread.start()
 
-    connected_event.wait(timeout=5.0)
+    timeout_not_hit = connected_event.wait(timeout=5.0)
+    if not timeout_not_hit:
+        ws.close()
+    assert timeout_not_hit, "connection timeout hit"
 
     return message_received_event, thread, ws

canvas_sdk/effects/banner_alert/tests.py

@@ -107,8 +107,16 @@ class Protocol(BaseProtocol):
         )
         response.raise_for_status()
 
-        message_received_event.wait(timeout=5.0)
-
+        message_received_event.wait(timeout=15.0)
+
+        # unfortunately sometimes the log websocket just doesn't return any
+        # messages, so asserting on the state of the timeout here causes failures
+        # even though the delay itself will cause the waiting test to pass (because
+        # the plugin has been loaded).
+        # timeout_not_hit = message_received_event.wait(timeout=15.0)
+        # if not timeout_not_hit:
+        #     ws.close()
+        # assert timeout_not_hit, f"plugin loading message timeout hit: Loading plugin '{plugin_name}"
     yield
 
     ws.close()

plugin_runner/installation.py

@@ -137,7 +137,7 @@ def download_plugin(plugin_package: str) -> Generator[Path, None, None]:
 def install_plugin(plugin_name: str, attributes: PluginAttributes) -> None:
     """Install the given Plugin's package into the runtime."""
     try:
-        log.info(f'Installing plugin "{plugin_name}"')
+        log.info(f'Installing plugin "{plugin_name}", version {attributes["version"]}')
 
         plugin_installation_path = Path(PLUGIN_DIRECTORY) / plugin_name
 
@@ -224,7 +224,6 @@ def install_plugins() -> None:
 
     for plugin_name, attributes in enabled_plugins().items():
         try:
-            log.info(f'Installing plugin "{plugin_name}", version {attributes["version"]}')
             install_plugin(plugin_name, attributes)
         except PluginInstallationError:
             disable_plugin(plugin_name)

plugin_runner/plugin_runner.py

@@ -13,6 +13,8 @@ from typing import Any, TypedDict
 
 import grpc
 import redis.asyncio as redis
+from asgiref.sync import sync_to_async
+from django.db import connections
 
 from canvas_generated.messages.effects_pb2 import EffectType
 from canvas_generated.messages.plugins_pb2 import ReloadPluginsRequest, ReloadPluginsResponse
@@ -30,6 +32,7 @@ from plugin_runner.installation import install_plugins
 from plugin_runner.sandbox import Sandbox
 from settings import (
     CHANNEL_NAME,
+    IS_TESTING,
     MANIFEST_FILE_NAME,
     PLUGIN_DIRECTORY,
     REDIS_ENDPOINT,
@@ -103,6 +106,25 @@ class PluginManifest(TypedDict):
     readme: str
 
 
+@sync_to_async
+def reconnect_if_needed() -> None:
+    """
+    Reconnect to the database if the connection has been closed.
+
+    NOTE: Django database functions are not async-safe and must be called
+    via e.g. sync_to_async. They will silently fail and block the handler
+    if called in an async context directly.
+    """
+    if IS_TESTING:
+        return
+
+    connection = connections["default"]
+
+    if not connection.is_usable():
+        log.debug("Connection was unusable, reconnecting...")
+        connection.connect()
+
+
 class PluginRunner(PluginRunnerServicer):
     """This process runs provided plugins that register interest in incoming events."""
 
@@ -118,6 +140,11 @@ class PluginRunner(PluginRunnerServicer):
         event_name = event.name
         relevant_plugins = EVENT_HANDLER_MAP[event_name]
 
+        log.debug(f"Processing {relevant_plugins} for {event_name}")
+
+        if relevant_plugins:
+            await reconnect_if_needed()
+
         if event_type in [EventType.PLUGIN_CREATED, EventType.PLUGIN_UPDATED]:
             plugin_name = event.target.id
             # filter only for the plugin(s) that were created/updated
@@ -126,6 +153,8 @@ class PluginRunner(PluginRunnerServicer):
         effect_list = []
 
         for plugin_name in relevant_plugins:
+            log.debug(f"Processing {plugin_name}")
+
             plugin = LOADED_PLUGINS[plugin_name]
             handler_class = plugin["class"]
             base_plugin_name = plugin_name.split(":")[0]
@@ -142,7 +171,7 @@ class PluginRunner(PluginRunnerServicer):
                 )
 
                 compute_start_time = time.time()
-                _effects = await asyncio.get_running_loop().run_in_executor(None, handler.compute)
+                _effects = await sync_to_async(handler.compute)()
                 effects = [
                     Effect(
                         type=effect.type,
@@ -166,9 +195,12 @@ class PluginRunner(PluginRunnerServicer):
                     tags={"plugin": plugin_name},
                 )
             except Exception as e:
+                log.error(f"Encountered exception in plugin {plugin_name}:")
+
                 for error_line_with_newlines in traceback.format_exception(e):
                     for error_line in error_line_with_newlines.split("\n"):
                         log.error(error_line)
+
                 continue
 
             effect_list += effects
@@ -204,7 +236,7 @@ async def synchronize_plugins(run_once: bool = False) -> None:
     """
     log.info(f'synchronize_plugins: listening for messages on pubsub channel "{CHANNEL_NAME}"')
 
-    client, pubsub = get_client()
+    _, pubsub = get_client()
     await pubsub.psubscribe(CHANNEL_NAME)
 
     while True: