omnata-plugin-runtime 0.8.3a207__py3-none-any.whl → 0.9.0a208__py3-none-any.whl

omnata_plugin_runtime/api.py

@@ -120,6 +120,7 @@ class OutboundSyncRequestPayload(BaseModel):
     results_table_name: str  # used to stage results back to the engine, resides in the main Omnata app database
     logging_level: str
     connection_method: str
+    target_type: Optional[str] = None
     connectivity_option: ConnectivityOption = Field(default=ConnectivityOption.DIRECT)
     connection_parameters: Dict[str, StoredConfigurationValue]
     oauth_secret_name: Optional[str] = None
@@ -166,6 +167,41 @@ SyncRequestPayload = Annotated[
     Field(discriminator="sync_direction"),
 ]
 
+class OutboundConfigurationFormPayload(BaseModel):
+    """
+    Encapsulates the payload that is sent to the plugin when it is invoked to provide a configuration form for an outbound sync.
+    """
+    connectivity_option: ConnectivityOption = Field(default=ConnectivityOption.DIRECT)
+    connection_method: str
+    connection_parameters: Dict[str, StoredConfigurationValue]
+    oauth_secret_name: Optional[str] = None
+    other_secrets_name: Optional[str] = None
+    sync_direction: Literal["outbound"] = "outbound"
+    target_type: Optional[str] = None
+    sync_strategy: OutboundSyncStrategy
+    function_name: str = "outbound_configuration_form"
+    sync_parameters: Dict[str, StoredConfigurationValue]
+    current_form_parameters: Optional[Dict[str, StoredConfigurationValue]]
+
+class InboundConfigurationFormPayload(BaseModel):
+    """
+    Encapsulates the payload that is sent to the plugin when it is invoked to provide a configuration form for an inbound sync.
+    """
+    connectivity_option: ConnectivityOption = Field(default=ConnectivityOption.DIRECT)
+    connection_method: str
+    connection_parameters: Dict[str, StoredConfigurationValue]
+    oauth_secret_name: Optional[str] = None
+    other_secrets_name: Optional[str] = None
+    sync_direction: Literal["inbound"] = "inbound"
+    function_name: str = "inbound_configuration_form"
+    sync_parameters: Dict[str, StoredConfigurationValue]
+    current_form_parameters: Optional[Dict[str, StoredConfigurationValue]]
+
+ConfigurationFormPayload = Annotated[
+    Union[OutboundConfigurationFormPayload, InboundConfigurationFormPayload],
+    Field(discriminator="sync_direction"),
+]
+
 
 def handle_proc_result(query_result: List[Row]) -> Dict:
     """

omnata_plugin_runtime/plugin_entrypoints.py

@@ -6,13 +6,13 @@ import os
 import sys
 import time
 import threading
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, cast
 
 from pydantic import BaseModel,TypeAdapter  # pylint: disable=no-name-in-module
 from pydantic_core import to_jsonable_python
 from snowflake.snowpark import Session
 
-from .api import PluginMessageStreamProgressUpdate, SyncRequestPayload, handle_proc_result
+from .api import PluginMessageStreamProgressUpdate, SyncRequestPayload, ConfigurationFormPayload
 from .configuration import (
     ConnectionConfigurationParameters,
     InboundSyncConfigurationParameters,
@@ -83,7 +83,7 @@ class PluginEntrypoint:
         
 
     def sync(self, sync_request: Dict):
-        request = TypeAdapter(SyncRequestPayload).validate_python(sync_request)
+        request:SyncRequestPayload = TypeAdapter(SyncRequestPayload).validate_python(sync_request)
         logger.add_extra('omnata.operation', 'sync')
         logger.add_extra('omnata.sync.id', request.sync_id)
         logger.add_extra('omnata.sync.direction', request.sync_direction)
@@ -160,6 +160,7 @@ class PluginEntrypoint:
                 connection_secrets=connection_secrets,
                 sync_parameters=request.sync_parameters,
                 current_form_parameters={},
+                target_type=request.target_type,
                 sync_strategy=request.sync_strategy,
                 field_mappings=request.field_mappings,
             )
@@ -289,67 +290,43 @@ class PluginEntrypoint:
         logger.info("Finished applying records")
         return return_dict
 
-    def configuration_form(
-        self,
-        connectivity_option:str,
-        connection_method: str,
-        connection_parameters: Dict,
-        oauth_secret_name: Optional[str],
-        other_secrets_name: Optional[str],
-        sync_direction: str,
-        sync_strategy: Dict,
-        function_name: str,
-        sync_parameters: Dict,
-        current_form_parameters: Optional[Dict],
-    ):
-        if function_name is None:
-            function_name = f"{sync_direction}_configuration_form"
+    def configuration_form(self, configuration_form_request: Dict):
+        request:ConfigurationFormPayload = TypeAdapter(ConfigurationFormPayload).validate_python(configuration_form_request)
         logger.add_extra('omnata.operation', 'configuration_form')
-        logger.add_extra('omnata.connection.connectivity_option', connectivity_option)
-        logger.add_extra('omnata.connection.connection_method', connection_method)
-        logger.add_extra('omnata.configuration_form.function_name', function_name)
-        logger.add_extra('omnata.sync.direction', sync_direction)
+        logger.add_extra('omnata.connection.connectivity_option', request.connectivity_option)
+        logger.add_extra('omnata.connection.connection_method', request.connection_method)
+        logger.add_extra('omnata.configuration_form.function_name', request.function_name)
+        logger.add_extra('omnata.sync.direction', request.sync_direction)
 
         logger.info("Entered configuration_form method")
-        sync_strategy = normalise_nulls(sync_strategy)
-        oauth_secret_name = normalise_nulls(oauth_secret_name)
-        other_secrets_name = normalise_nulls(other_secrets_name)
-        connection_secrets = get_secrets(oauth_secret_name, other_secrets_name)
-        connectivity_option = TypeAdapter(ConnectivityOption).validate_python(connectivity_option)
-        connection_parameters = TypeAdapter(
-            Dict[str, StoredConfigurationValue]).validate_python(connection_parameters)
-        sync_parameters = TypeAdapter(
-            Dict[str, StoredConfigurationValue]).validate_python(sync_parameters)
-        form_parameters = None
-        if current_form_parameters is not None:
-            form_parameters = TypeAdapter(Dict[str, StoredConfigurationValue]).validate_python(current_form_parameters)
-        if sync_direction == "outbound":
-            sync_strat = OutboundSyncStrategy.model_validate(sync_strategy) if sync_strategy is not None else None
+        connection_secrets = get_secrets(request.oauth_secret_name, request.other_secrets_name)
+        if request.sync_direction == "outbound":
             parameters = OutboundSyncConfigurationParameters(
-                connection_parameters=connection_parameters,
+                connection_parameters=request.connection_parameters,
                 connection_secrets=connection_secrets,
-                sync_strategy=sync_strat,
-                sync_parameters=sync_parameters,
-                connection_method=connection_method,
-                connectivity_option=connectivity_option,
-                current_form_parameters=form_parameters,
+                sync_strategy=request.sync_strategy,
+                sync_parameters=request.sync_parameters,
+                connection_method=request.connection_method,
+                connectivity_option=request.connectivity_option,
+                current_form_parameters=request.current_form_parameters,
+                target_type=request.target_type
             )
-        elif sync_direction == "inbound":
+        elif request.sync_direction == "inbound":
             parameters = InboundSyncConfigurationParameters(
-                connection_parameters=connection_parameters,
+                connection_parameters=request.connection_parameters,
                 connection_secrets=connection_secrets,
-                sync_parameters=sync_parameters,
-                connection_method=connection_method,
-                connectivity_option=connectivity_option,
-                current_form_parameters=form_parameters,
+                sync_parameters=request.sync_parameters,
+                connection_method=request.connection_method,
+                connectivity_option=request.connectivity_option,
+                current_form_parameters=request.current_form_parameters,
             )
         else:
-            raise ValueError(f"Unknown direction {sync_direction}")
-        if oauth_secret_name is not None:
-                parameters.access_token_secret_name = oauth_secret_name
+            raise ValueError(f"Unknown direction {request.sync_direction}")
+        if request.oauth_secret_name is not None:
+                parameters.access_token_secret_name = request.oauth_secret_name
         the_function = getattr(
             self._plugin_instance,
-            function_name
+            request.function_name
         )
         with tracer.start_as_current_span("invoke_plugin"):
             script_result = the_function(parameters)
@@ -357,6 +334,7 @@ class PluginEntrypoint:
             script_result = script_result.model_dump()
         elif isinstance(script_result, List):
             if len(script_result) > 0 and isinstance(script_result[0], BaseModel):
+                script_result = cast(List[BaseModel], script_result)
                 script_result = [r.model_dump() for r in script_result]
         return script_result
         

{omnata_plugin_runtime-0.8.3a207.dist-info → omnata_plugin_runtime-0.9.0a208.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.8.3a207
+Version: 0.9.0a208
 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.8.3a207.dist-info → omnata_plugin_runtime-0.9.0a208.dist-info}/RECORD

@@ -1,12 +1,12 @@
 omnata_plugin_runtime/__init__.py,sha256=MS9d1whnfT_B3-ThqZ7l63QeC_8OEKTuaYV5wTwRpBA,1576
-omnata_plugin_runtime/api.py,sha256=tVi4KLL0v5N3yz3Ie0kSyFemryu572gCbtSRfWN6wBU,6523
+omnata_plugin_runtime/api.py,sha256=baGraSMiD4Yvi3ZWrEv_TKh8Ktd1U8riBdOpe9j0Puw,8202
 omnata_plugin_runtime/configuration.py,sha256=g6A9sFHBBaCgZPjhTM4iYcyEoZDPjB1F24QMboLteyY,38505
 omnata_plugin_runtime/forms.py,sha256=ueodN2GIMS5N9fqebpY4uNGJnjEb9HcuaVQVfWH-cGg,19838
 omnata_plugin_runtime/logging.py,sha256=WBuZt8lF9E5oFWM4KYQbE8dDJ_HctJ1pN3BHwU6rcd0,4461
 omnata_plugin_runtime/omnata_plugin.py,sha256=WrttykPnWR5ckL2SHUbcU6yYpNb-_HTL3LzFB6cBMd8,134356
-omnata_plugin_runtime/plugin_entrypoints.py,sha256=sB_h6OBEMk7lTLIjdNrNo9Sthk8UE9PnK2AUcQJPe9I,32728
+omnata_plugin_runtime/plugin_entrypoints.py,sha256=iqGl8_nEEnPGKg3Aem4YLSQ6d5xS3ju5gq8MJbx6sCA,31968
 omnata_plugin_runtime/rate_limiting.py,sha256=eOWVRYWiqPlVeYzmB1exVXfXbrcpmYb7vtTi9B-4zkQ,25868
-omnata_plugin_runtime-0.8.3a207.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
-omnata_plugin_runtime-0.8.3a207.dist-info/METADATA,sha256=aU_EJv5HVn3CXBKKuJaD3JFhBANrOT_o883o2fvIdpA,2158
-omnata_plugin_runtime-0.8.3a207.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-omnata_plugin_runtime-0.8.3a207.dist-info/RECORD,,
+omnata_plugin_runtime-0.9.0a208.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
+omnata_plugin_runtime-0.9.0a208.dist-info/METADATA,sha256=BdfXNGwXPGhbypwhYNed8BmD6jJanyddwHg1Y8011SU,2158
+omnata_plugin_runtime-0.9.0a208.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+omnata_plugin_runtime-0.9.0a208.dist-info/RECORD,,