omnata-plugin-runtime 0.3.23a69__py3-none-any.whl → 0.3.24a70__py3-none-any.whl

omnata_plugin_runtime/omnata_plugin.py

@@ -1827,26 +1827,35 @@ def managed_outbound_processing(concurrency: int, batch_size: int):
                 raise ValueError(
                     "To use the managed_outbound_processing decorator, you must attach a sync request to the plugin instance (via the _sync_request property)"
                 )
-            # if self._sync_request.api_limits is None:
-            #    raise ValueError('To use the managed_outbound_processing decorator, API constraints must be defined. These can be provided in the response to the connect method')
             logger.info(f"Batch size: {batch_size}. Concurrency: {concurrency}")
-            if len(method_args) == 0:
-                raise ValueError(
-                    "You must provide at least one method argument, and the first argument must be a DataFrame or DataFrame generator (from outbound_sync_request.get_records)"
-                )
-            first_arg = method_args[0]
-            #logger.info(first_arg.__class__.__name__)
-            if first_arg.__class__.__name__ == "DataFrame":
-                logger.info("managed_outbound_processing received a DataFrame")
-            elif hasattr(first_arg, "__next__"):
-                logger.info("managed_outbound_processing received an iterator function")
-            else:
-                raise ValueError(
-                    f"The first argument to a @managed_outbound_processing method must be a DataFrame or DataFrame generator (from outbound_sync_request.get_records). Instead, a {first_arg.__class__.__name__} was provided."
-                )
+
+            dataframe_arg = None
+            if 'dataframe' in method_kwargs:
+                dataframe_arg = method_kwargs['dataframe']
+                del method_kwargs['dataframe']
+                if dataframe_arg.__class__.__name__ != "DataFrame":
+                    raise ValueError(
+                        f"The 'dataframe' named argument to the @managed_outbound_processing must be a DataFrame. Instead, a {dataframe_arg.__class__.__name__} was provided."
+                    )
+                
+            elif 'dataframe_generator' in method_kwargs:
+                dataframe_arg = method_kwargs['dataframe_generator']
+                del method_kwargs['dataframe_generator']
+                if not hasattr(dataframe_arg, "__next__"):
+                    raise ValueError(
+                        f"The 'dataframe_generator' named argument to the @managed_outbound_processing must be an iterator function. Instead, a {dataframe_arg.__class__.__name__} was provided."
+                    )
+            # if the dataframe was provided as the first argument, we'll use that
+            if dataframe_arg is None and len(method_args) > 0:
+                dataframe_arg = method_args[0]
+                if dataframe_arg.__class__.__name__ != "DataFrame" and not hasattr(dataframe_arg, "__next__"):
+                    raise ValueError(
+                        f"The first argument to a @managed_outbound_processing method must be a DataFrame or DataFrame generator (from outbound_sync_request.get_records). Instead, a {first_arg.__class__.__name__} was provided. Alternatively, you can provide these via the 'dataframe' or 'dataframe_generator' named arguments."
+                    )
+                method_args = method_args[1:]
 
             # put the record iterator on the queue, ready for the first task to read it
-            fixed_size_generator = FixedSizeGenerator(first_arg, batch_size=batch_size)
+            fixed_size_generator = FixedSizeGenerator(dataframe_arg, batch_size=batch_size)
             tasks:List[threading.Thread] = []
             logger.info(f"Creating {concurrency} worker(s) for applying records")
             # just in case
@@ -1862,7 +1871,7 @@ def managed_outbound_processing(concurrency: int, batch_size: int):
                         i,
                         fixed_size_generator,
                         self._sync_request._thread_cancellation_token,
-                        method_args[1:],
+                        method_args,
                         method_kwargs,
                     ),
                 )
@@ -1987,22 +1996,21 @@ def managed_inbound_processing(concurrency: int):
                 concurrency_to_use = 1 # disable concurrency when running in development mode, it interferes with pyvcr
             else:
                 concurrency_to_use = concurrency
-            # if self._sync_request.api_limits is None:
-            #    raise ValueError('To use the managed_inbound_processing decorator, API constraints must be defined. These can be provided in the response to the connect method')
-            if len(method_args) == 0:
-                raise ValueError(
-                    "You must provide at least one method argument, and the first argument must be a list of StoredStreamConfigurations (from inbound_sync_request.streams)"
-                )
-            first_arg: List[StoredStreamConfiguration] = method_args[0]
-            logger.info(first_arg.__class__.__name__)
-            if first_arg.__class__.__name__ == "list":
-                logger.info("managed_inbound_processing received a list")
-            else:
-                raise ValueError(
-                    f"The first argument to a @managed_inbound_processing method must be a list of StoredStreamConfigurations. Instead, a {first_arg.__class__.__name__} was provided."
-                )
+            stream_list_arg: List[StoredStreamConfiguration] = None
+            if 'streams' in method_kwargs:
+                stream_list_arg = cast(List[StoredStreamConfiguration],method_kwargs['streams'])
+                del method_kwargs['streams']
+            if stream_list_arg is None and len(method_args) > 0:
+                stream_list_arg = cast(List[StoredStreamConfiguration],method_args[0])
+                if stream_list_arg.__class__.__name__ != "list":
+                    raise ValueError(
+                        f"The first argument to a @managed_inbound_processing method must be a list of StoredStreamConfigurations if the 'streams' named argument is not provided. Instead, a {stream_list_arg.__class__.__name__} was provided."
+                    )
+                method_args = method_args[1:]
+            if stream_list_arg is None:
+                raise ValueError("You must provide a list of StoredStreamConfiguration objects to the method, either as the first argument or as a named argument 'streams'")
 
-            streams_list: List[StoredStreamConfiguration] = first_arg
+            streams_list = stream_list_arg
             # create a queue full of all the streams to process
             streams_queue = queue.Queue()
             for stream in streams_list:
@@ -2022,7 +2030,7 @@ def managed_inbound_processing(concurrency: int):
                         i,
                         streams_queue,
                         self._sync_request._thread_cancellation_token,
-                        method_args[1:],
+                        method_args,
                         method_kwargs,
                     ),
                 )

{omnata_plugin_runtime-0.3.23a69.dist-info → omnata_plugin_runtime-0.3.24a70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omnata-plugin-runtime
-Version: 0.3.23a69
+Version: 0.3.24a70
 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.23a69.dist-info → omnata_plugin_runtime-0.3.24a70.dist-info}/RECORD

@@ -3,10 +3,10 @@ omnata_plugin_runtime/api.py,sha256=_N5ok5LN7GDO4J9n3yduXp3tpjmhpySY__U2baiygrs,
 omnata_plugin_runtime/configuration.py,sha256=7cMekoY8CeZAJHpASU6tCMidF55Hzfr7CD74jtebqIY,35742
 omnata_plugin_runtime/forms.py,sha256=pw_aKVsXSz47EP8PFBI3VDwdSN5IjvZxp8JTjO1V130,18421
 omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
-omnata_plugin_runtime/omnata_plugin.py,sha256=EaGIQdq_gcmH2loTluTa0T3YxuUhDNZTQIItzYXtZxE,102783
+omnata_plugin_runtime/omnata_plugin.py,sha256=N05RyAmud6a6BXXtzccFTvbPz4bjF1yFE92Vx6FiRnc,103346
 omnata_plugin_runtime/plugin_entrypoints.py,sha256=s-SrUnXaS6FaBq1igiJhcCB3Md_a6op-dp_g1H_55QU,27736
 omnata_plugin_runtime/rate_limiting.py,sha256=se6MftQI5NrVHaLb1hByPCgAESPQhkAgIG7KIU1clDU,16562
-omnata_plugin_runtime-0.3.23a69.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-omnata_plugin_runtime-0.3.23a69.dist-info/METADATA,sha256=u_HhdrNMZOCVn0A9Lv-u9q3KnOmxvRlJLDFAHrU8si0,1604
-omnata_plugin_runtime-0.3.23a69.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-omnata_plugin_runtime-0.3.23a69.dist-info/RECORD,,
+omnata_plugin_runtime-0.3.24a70.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+omnata_plugin_runtime-0.3.24a70.dist-info/METADATA,sha256=qWHXnGRPpqyDfKDnXmgSFu3chfHRa4evj6apwWoCWpA,1604
+omnata_plugin_runtime-0.3.24a70.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+omnata_plugin_runtime-0.3.24a70.dist-info/RECORD,,