omnata-plugin-runtime 0.5.1a114__py3-none-any.whl → 0.5.2a117__py3-none-any.whl

omnata_plugin_runtime/configuration.py

@@ -8,7 +8,7 @@ from typing import Any, List, Dict, Literal, Union, Optional
 from enum import Enum
 
 from abc import ABC
-from pydantic import BaseModel, Field, PrivateAttr, validator  # pylint: disable=no-name-in-module
+from pydantic import BaseModel, Field, PrivateAttr, SerializationInfo, model_serializer, validator  # pylint: disable=no-name-in-module
 
 if tuple(sys.version_info[:2]) >= (3, 9):
     # Python 3.9 and above
@@ -110,16 +110,18 @@ class OutboundSyncAction(SubscriptableBaseModel, ABC):
                     **STANDARD_OUTBOUND_SYNC_ACTIONS[data["action_name"]]().__dict__,
                 }
         super().__init__(**data)
-
-    def dict(self, *args, trim: bool = True, **kwargs) -> Dict[str, any]:
-        """
-        OutboundSyncStrategy objects can be serialized much smaller if the instance is of a Standard strategy that resides in the
-        Omnata plugin runtime. This method trims out the unneeded attributes in these scenarios.
-        This method also trims the actions for this strategy if it's a custom one.
-
-        """
-        excluded_fields = {} if not trim or self.custom_action else {"description"}
-        return super().dict(exclude=excluded_fields)
+    
+    @model_serializer(mode='wrap')
+    def ser_model(self,handler,info:SerializationInfo) -> Dict[str, Any]:
+        serialized:Dict[str,Any] = handler(self)
+        if not self.custom_action and (info.exclude_none is None or info.exclude_none == False):
+            return {k:v for k,v in serialized.items() if k not in [
+                "description"]}
+        return serialized
+    
+    def model_dump_no_trim(self) -> Dict[str, Any]:
+        # we use our own special include value to signal not to trim
+        return self.model_dump(exclude_none=True)
 
 
 class CreateSyncAction(OutboundSyncAction):
@@ -227,7 +229,7 @@ class OutboundSyncStrategy(SubscriptableBaseModel, ABC):
     action_on_record_unchanged: Optional[OutboundSyncAction] = None
     custom_strategy: bool = True
 
-    def __eq__(self, other):
+    def __eq__(self, other:OutboundSyncStrategy):
         if hasattr(other, 'custom_strategy') and hasattr(other, 'name'):
             return (
                 self.custom_strategy == other.custom_strategy
@@ -251,27 +253,23 @@ class OutboundSyncStrategy(SubscriptableBaseModel, ABC):
                 }
         super().__init__(**data)
 
-    def dict(self, *args, trim: bool = True, **kwargs) -> Dict[str, any]:
-        """
-        OutboundSyncStrategy objects can be serialized much smaller if the instance is of a Standard strategy that resides in the
-        Omnata plugin runtime. This method trims out the unneeded attributes in these scenarios.
-        This method also trims the actions for this strategy if it's a custom one.
-
-        """
-        excluded_fields = (
-            {}
-            if not trim or self.custom_strategy
-            else {
+    @model_serializer(mode='wrap')
+    def ser_model(self,handler,info:SerializationInfo) -> Dict[str, Any]:
+        serialized:Dict[str,Any] = handler(self)
+        if not self.custom_strategy and (info.exclude_none is None or info.exclude_none == False):
+            return {k:v for k,v in serialized.items() if k not in [
                 "description",
                 "icon_source",
                 "action_on_record_create",
                 "action_on_record_update",
                 "action_on_record_delete",
-                "action_on_record_unchanged",
-            }
-        )
-        return super().dict(exclude=excluded_fields)
-
+                "action_on_record_unchanged"]}
+        return serialized
+    
+    def model_dump_no_trim(self) -> Dict[str, Any]:
+        # we use our own special include value to signal not to trim
+        return self.model_dump(exclude_none=True)
+    
 
 class CreateSyncStrategy(OutboundSyncStrategy):
     """

omnata_plugin_runtime/plugin_entrypoints.py

@@ -32,6 +32,7 @@ from .omnata_plugin import (
     OutboundSyncRequest,
     DeadlineReachedException,
 )
+from pydantic import TypeAdapter
 from .rate_limiting import ApiLimits, RateLimitState
 
 # set the logger class to our custom logger so that pydantic errors are handled correctly
@@ -233,7 +234,7 @@ class PluginEntrypoint:
                     stream_errors=omnata_log_handler.stream_global_errors,
                     total_records_estimate=inbound_sync_request._total_records_estimate
                 )
-                return_dict["final_progress_update"] = final_progress_update.dict()
+                return_dict["final_progress_update"] = final_progress_update.model_dump()
                 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
@@ -305,10 +306,10 @@ class PluginEntrypoint:
         )
         script_result = the_function(parameters)
         if isinstance(script_result, BaseModel):
-            script_result = script_result.dict()
+            script_result = script_result.model_dump()
         elif isinstance(script_result, List):
             if len(script_result) > 0 and isinstance(script_result[0], BaseModel):
-                script_result = [r.dict() for r in script_result]
+                script_result = [r.model_dump() for r in script_result]
         return script_result
         
     def inbound_list_streams(
@@ -341,10 +342,10 @@ class PluginEntrypoint:
 
         script_result = self._plugin_instance.inbound_stream_list(parameters)
         if isinstance(script_result, BaseModel):
-            script_result = script_result.dict()
+            script_result = script_result.model_dump()
         elif isinstance(script_result, List):
             if len(script_result) > 0 and isinstance(script_result[0], BaseModel):
-                script_result = [r.dict() for r in script_result]
+                script_result = [r.model_dump() for r in script_result]
         return script_result
 
 
@@ -365,17 +366,17 @@ class PluginEntrypoint:
             script_result = the_function(stored_value)
             if not isinstance(script_result, FormOption):
                 raise ValueError(f"Expected a FormOption from function {function_name}, got {type(script_result)}")
-            results.append(script_result.dict())
+            results.append(script_result.model_dump())
         return results
 
     def connection_form(self):
         logger.info("Entered connection_form method")
         form: List[ConnectionMethod] = self._plugin_instance.connection_form()
-        return [f.dict() for f in form]
+        return [f.model_dump() for f in form]
 
     def create_billing_events(self, session, event_request: Dict):
         logger.info("Entered create_billing_events method")
-        request = BillingEventRequest.model_validate(event_request)
+        request = TypeAdapter(BillingEventRequest).validate_python(event_request)
         events: List[SnowflakeBillingEvent] = self._plugin_instance.create_billing_events(
             request
         )
@@ -408,7 +409,7 @@ class PluginEntrypoint:
                     logger.warn('Billing event creation failed due to running internally to Omnata')
                 else:
                     raise e
-        return [e.dict() for e in events]
+        return [e.model_dump() for e in events]
 
     def get_secrets(
         self, oauth_secret_name: Optional[str], other_secrets_name: Optional[str]
@@ -461,7 +462,7 @@ class PluginEntrypoint:
         script_result = the_function(parameters)
         if isinstance(script_result, List):
             if len(script_result) > 0 and isinstance(script_result[0], BaseModel):
-                script_result = [r.dict() for r in script_result]
+                script_result = [r.model_dump() for r in script_result]
         else:
             raise ValueError(f"Expected a List from function {function_name}, got {type(script_result)}")
         return script_result
@@ -531,7 +532,7 @@ class PluginEntrypoint:
                 f"alter network rule {network_rule_name} set value_list = ({rule_values_string})"
             ).collect()
 
-        return connect_response.dict()
+        return connect_response.model_dump()
 
     def api_limits(self,
                    method:str,
@@ -564,7 +565,7 @@ class PluginEntrypoint:
         # There's a bit of parsing here that could possibly be done outside of the handler function, but this shouldn't be too expensive
         sync_parameters: Dict[str, StoredConfigurationValue] = TypeAdapter(
             Dict[str, StoredConfigurationValue]).validate_python(sync_parameters)
-        field_mappings: StoredMappingValue = StoredMappingValue.model_validate(field_mappings)
+        field_mappings: StoredMappingValue = TypeAdapter(StoredMappingValue).validate_python(field_mappings)
         return self._plugin_instance.outbound_record_validator(
             sync_parameters, field_mappings, transformed_record, source_types
         )

{omnata_plugin_runtime-0.5.1a114.dist-info → omnata_plugin_runtime-0.5.2a117.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.5.1a114
+Version: 0.5.2a117
 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.5.1a114.dist-info → omnata_plugin_runtime-0.5.2a117.dist-info}/RECORD

@@ -1,12 +1,12 @@
 omnata_plugin_runtime/__init__.py,sha256=MS9d1whnfT_B3-ThqZ7l63QeC_8OEKTuaYV5wTwRpBA,1576
 omnata_plugin_runtime/api.py,sha256=FxzTqri4no8ClkOm7vZADG8aD47jcGBCTTQDEORmOJM,6326
-omnata_plugin_runtime/configuration.py,sha256=LvQVIKwo8XlfHf-9Jkj1GDH0U6sOKDKi3IMDyAwAnHE,35781
+omnata_plugin_runtime/configuration.py,sha256=GI7RBerWjymVtXRFaAhbmgmoct-m943wta0TX0m6_z8,35790
 omnata_plugin_runtime/forms.py,sha256=hV6jVaizex20Pb9NxPx11TBPK-Yy8pREAnTtCxHo4Qo,18409
 omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
 omnata_plugin_runtime/omnata_plugin.py,sha256=zIk8dS5m4pXsVy_2im-Cd-t1uQV9hf1seosb1T2jGGs,110880
-omnata_plugin_runtime/plugin_entrypoints.py,sha256=xlPTI25N5G3l0bRFSkga98TIHM44NRaN2g7T5jSsqX0,29181
+omnata_plugin_runtime/plugin_entrypoints.py,sha256=JirYUbPBaN0UMh9t_uAHDdhaQZ7NUhdMJ11eHRKOoNY,29302
 omnata_plugin_runtime/rate_limiting.py,sha256=DVQ_bc-mVLBkrU1PTns1MWXhHiLpSB5HkWCcdePtJ2A,25611
-omnata_plugin_runtime-0.5.1a114.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.5.1a114.dist-info/METADATA,sha256=yf8uYGHInQ_N6bUEZyTgvvVodiYL0mxHtj3jy94BIhU,1888
-omnata_plugin_runtime-0.5.1a114.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.5.1a114.dist-info/RECORD,,
+omnata_plugin_runtime-0.5.2a117.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.5.2a117.dist-info/METADATA,sha256=5moARb8FGRx-6ulORy_zkGj7a_kc77Jd_GsDrZIu15E,1888
+omnata_plugin_runtime-0.5.2a117.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.5.2a117.dist-info/RECORD,,