omnata-plugin-runtime 0.2.99__py3-none-any.whl → 0.3.1__py3-none-any.whl

omnata_plugin_runtime/configuration.py

@@ -745,7 +745,7 @@ class SyncScheduleSnowflakeTask(SubscriptableBaseModel):
 
     """
 
-    mode: Literal["snowflake_task"]
+    mode: Literal["snowflake_task"] = "snowflake_task"
     sync_frequency: str
     sync_frequency_name: Literal["1 min", "5 mins", "15 mins", "Hourly", "Daily","Custom"]
     warehouse: str = None
@@ -770,7 +770,7 @@ class SyncScheduleDbt(SubscriptableBaseModel):
         is_dbt_cloud (bool): If true, dbt cloud is in use
     """
 
-    mode: Literal["dbt"]
+    mode: Literal["dbt"] = "dbt"
     dbt_prod_target_name: str = "prod"
     task_warehouse_dbt_defined: bool = True
     warehouse: str
@@ -792,7 +792,7 @@ class SyncScheduleDependant(SubscriptableBaseModel):
         selected_sync (int): The sync ID of the sync to depend on
     """
 
-    mode: Literal["dependent"]
+    mode: Literal["dependent"] = "dependent"
     run_when: Literal[
         "after_parent_completes", "at_same_time_as"
     ] = "after_parent_completes"
@@ -810,7 +810,7 @@ class SyncScheduleManual(SubscriptableBaseModel):
         time_limit_mins (int): The maximum time the sync can run for before being cancelled
     """
 
-    mode: Literal["manual"]
+    mode: Literal["manual"] = "manual"
     warehouse: str
     time_limit_mins: int = 60 * 4
 

omnata_plugin_runtime/omnata_plugin.py

@@ -169,6 +169,7 @@ class SyncRequest(ABC):
         :return: nothing
         """
         logger.info(f"Initiating SyncRequest for sync run {run_id}, run deadline: {run_deadline}")
+        self.deadline_reached:bool = False
         self._run_deadline = run_deadline
         self.plugin_instance = plugin_instance
         self._source_app_name = source_app_name
@@ -273,6 +274,7 @@ class SyncRequest(ABC):
             logger.info("cancel checking worker checking for cancellation")
             if (datetime.datetime.now(datetime.timezone.utc) > self._run_deadline):  # pylint: disable=protected-access
                 # if we've reached the deadline for the run, end it
+                self.deadline_reached = True
                 self.apply_deadline_reached()  # pylint: disable=protected-access
                 return
 

omnata_plugin_runtime/plugin_entrypoints.py

@@ -30,6 +30,7 @@ from .omnata_plugin import (
     InboundSyncRequest,
     OmnataPlugin,
     OutboundSyncRequest,
+    DeadlineReachedException,
 )
 from .rate_limiting import ApiLimits, RateLimitState
 
@@ -147,6 +148,10 @@ class PluginEntrypoint:
                     self._plugin_instance.sync_outbound(parameters, outbound_sync_request)
                     outbound_sync_request.apply_results_queue()
                 outbound_sync_request.apply_rate_limit_state()
+                if outbound_sync_request.deadline_reached:
+                    # if we actually hit the deadline, this is flagged by the cancellation checking worker and the cancellation
+                    # token is set. We throw it here as an error since that's currently how it flows back to the engine with a DELAYED state
+                    raise DeadlineReachedException()
             finally:
                 try:
                     # cancel the thread so we don't leave anything hanging around and cop a nasty error
@@ -198,6 +203,10 @@ class PluginEntrypoint:
                 logger.info("Calling apply_results_queue")
                 inbound_sync_request.apply_results_queue()
                 inbound_sync_request.apply_rate_limit_state()
+                if inbound_sync_request.deadline_reached:
+                    # if we actually hit the deadline, this is flagged by the cancellation checking worker and the cancellation
+                    # token is set. We throw it here as an error since that's currently how it flows back to the engine with a DELAYED state
+                    raise DeadlineReachedException()
             finally:
                 # cancel the thread so we don't leave anything hanging around and cop a nasty error
                 try:

{omnata_plugin_runtime-0.2.99.dist-info → omnata_plugin_runtime-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.2.99
+Version: 0.3.1
 Summary: Classes and common runtime components for building and running Omnata Plugins
 Author: James Weakley
 Author-email: james.weakley@omnata.com

omnata_plugin_runtime-0.3.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+omnata_plugin_runtime/__init__.py,sha256=w63LVME5nY-hQ4BBzfacy9kvTunwqHGs8iiSPGAX2ns,1214
+omnata_plugin_runtime/api.py,sha256=_N5ok5LN7GDO4J9n3yduXp3tpjmhpySY__U2baiygrs,6217
+omnata_plugin_runtime/configuration.py,sha256=at29ExowF_T4_2U9gY0BF4IVdwC-vDytmNRHL7UCWh8,34742
+omnata_plugin_runtime/forms.py,sha256=30CJB24TqfLYNnkplZdUbeqA-P9rUIBujVKXw_S-wKY,18371
+omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
+omnata_plugin_runtime/omnata_plugin.py,sha256=yqzFoAHbJ7zumfG27o9wVXTum0v1sYcYDxHPdYsjYLU,86572
+omnata_plugin_runtime/plugin_entrypoints.py,sha256=4qpTzUrfL_gGMDYhGnEwHgCRUJII1yU2AB4WgCko_8Q,26650
+omnata_plugin_runtime/rate_limiting.py,sha256=se6MftQI5NrVHaLb1hByPCgAESPQhkAgIG7KIU1clDU,16562
+omnata_plugin_runtime-0.3.1.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.3.1.dist-info/METADATA,sha256=4n8-0Crp_6hG-Pwi36DbS0Vfw5JLVOdqZGjmrb_yeSc,1604
+omnata_plugin_runtime-0.3.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+omnata_plugin_runtime-0.3.1.dist-info/RECORD,,

omnata_plugin_runtime-0.2.99.dist-info/RECORD

@@ -1,12 +0,0 @@
-omnata_plugin_runtime/__init__.py,sha256=w63LVME5nY-hQ4BBzfacy9kvTunwqHGs8iiSPGAX2ns,1214
-omnata_plugin_runtime/api.py,sha256=_N5ok5LN7GDO4J9n3yduXp3tpjmhpySY__U2baiygrs,6217
-omnata_plugin_runtime/configuration.py,sha256=Glj4Xd9lz1QAMkpSkIaLeEcwWs0ns_WgS9U8BODgbA8,34690
-omnata_plugin_runtime/forms.py,sha256=30CJB24TqfLYNnkplZdUbeqA-P9rUIBujVKXw_S-wKY,18371
-omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
-omnata_plugin_runtime/omnata_plugin.py,sha256=Gfy77vmLY-Y-rrmB9ytmh-KFfAEFIXyUU_4_K1a1EYM,86484
-omnata_plugin_runtime/plugin_entrypoints.py,sha256=WWgWFuZ24w2J2a6swW-F8NTYZ7FdN8mBDjxtIMh9sZ4,25859
-omnata_plugin_runtime/rate_limiting.py,sha256=se6MftQI5NrVHaLb1hByPCgAESPQhkAgIG7KIU1clDU,16562
-omnata_plugin_runtime-0.2.99.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.2.99.dist-info/METADATA,sha256=rVxLnjPiwTigT54fLH9eBlcl00ZKE3ZecUMov6PVryw,1605
-omnata_plugin_runtime-0.2.99.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-omnata_plugin_runtime-0.2.99.dist-info/RECORD,,