opentf-toolkit-nightly 0.56.0.dev985__py3-none-any.whl → 0.56.0.dev988__py3-none-any.whl

opentf/toolkit/__init__.py

@@ -296,6 +296,17 @@ def _dispatch_executioncommand(_, handler: Handler, body: Dict[str, Any]):
         return make_status_response('InternalError', msg)
 
 
+def _dispatch_generatorcommand(_, handler: Handler, body: Dict[str, Any]):
+    """Generator plugin dispatcher."""
+    try:
+        inputs: Dict[str, Any] = body.get('with', {})
+        core.publish_generatorresult(handler(inputs))
+    except Exception as err:
+        msg = f'Unexpected execution error: {err}.'
+        core.publish_error(msg)
+        return make_status_response('InternalError', msg)
+
+
 ########################################################################
 #  Watchdog
 
@@ -476,6 +487,37 @@ def run_plugin(plugin):
             context[SUBSCRIPTION_KEY].append(
                 subscribe(kind=EXECUTIONCOMMAND, target='inbox', app=plugin)
             )
+        elif context[KIND_KEY] == GENERATORCOMMAND:
+            for manifest in plugin.config['DESCRIPTOR']:
+                metadata = manifest.get('metadata', {})
+                if metadata.get('name', '').lower() != plugin.name.lower():
+                    continue
+                if 'action' not in metadata:
+                    continue
+                for event in manifest.get('events', []):
+                    cat_prefix = event.get('categoryPrefix')
+                    cat = event.get('category')
+                    if cat or cat_prefix:
+                        cat_version = event.get('categoryVersion')
+                        labels = {}
+                        if cat is not None:
+                            labels['opentestfactory.org/category'] = cat
+                        if cat_prefix is not None:
+                            labels['opentestfactory.org/categoryPrefix'] = cat_prefix
+                        if cat_version is not None:
+                            labels['opentestfactory.org/categoryVersion'] = cat_version
+                        context[SUBSCRIPTION_KEY].append(
+                            subscribe(
+                                kind=GENERATORCOMMAND,
+                                target='inbox',
+                                labels=labels,
+                                app=plugin,
+                            )
+                        )
+                    else:
+                        plugin.logger.warning(
+                            "At least one of 'category', 'categoryPrefix' required, ignoring."
+                        )
         run_app(plugin)
     finally:
         for subscription_id in plugin.config['CONTEXT'][SUBSCRIPTION_KEY]:
@@ -558,6 +600,8 @@ def make_plugin(
             _dispatch_providercommand(plugin, provider, body)
         elif channel:
             return _dispatch_executioncommand(plugin, channel, body)
+        elif generator:
+            _dispatch_generatorcommand(plugin, generator, body)
         else:
             return make_status_response('BadRequest', 'Not implemented yet.')
 

opentf/toolkit/core.py

@@ -24,6 +24,7 @@ import posixpath
 import sys
 
 from opentf.commons import (
+    GENERATORRESULT,
     PROVIDERRESULT,
     EXECUTIONERROR,
     publish,
@@ -140,6 +141,16 @@ def publish_providerresult(steps: Iterable) -> None:
     publish_event(command)
 
 
+def publish_generatorresult(jobs: Dict[str, Any]) -> None:
+    """Publish GeneratorResult event."""
+    command = make_event(
+        GENERATORRESULT,
+        metadata=_getbody()['metadata'],
+        jobs={k: v.copy() for k, v in jobs.items()},
+    )
+    publish_event(command)
+
+
 ########################################################################
 ## Toolkit helpers
 

{opentf_toolkit_nightly-0.56.0.dev985.dist-info → opentf_toolkit_nightly-0.56.0.dev988.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentf-toolkit-nightly
-Version: 0.56.0.dev985
+Version: 0.56.0.dev988
 Summary: OpenTestFactory Orchestrator Toolkit
 Home-page: https://gitlab.com/henixdevelopment/open-source/opentestfactory/python-toolkit
 Author: Martin Lafaix

{opentf_toolkit_nightly-0.56.0.dev985.dist-info → opentf_toolkit_nightly-0.56.0.dev988.dist-info}/RECORD

@@ -45,11 +45,11 @@ opentf/schemas/opentestfactory.org/v1beta1/Workflow.json,sha256=QZ8mM9PhzsI9gTmw
 opentf/schemas/opentestfactory.org/v1beta2/ServiceConfig.json,sha256=rEvK2YWL5lG94_qYgR_GnLWNsaQhaQ-2kuZdWJr5NnY,3517
 opentf/scripts/launch_java_service.sh,sha256=S0jAaCuv2sZy0Gf2NGBuPX-eD531rcM-b0fNyhmzSjw,2423
 opentf/scripts/startup.py,sha256=Da2zo93pBWbdRmj-wgekgLcF94rpNc3ZkbvR8R0w8XY,21279
-opentf/toolkit/__init__.py,sha256=g3DiTZlSvvzZWKgM8qU47muLqjQrpWZ6M6PWZ-sBsvQ,19610
+opentf/toolkit/__init__.py,sha256=icNo_jz2IfhwddunQpfq6F-0xeKS70S-0Or5zJliEZs,21692
 opentf/toolkit/channels.py,sha256=Cng3b4LUsxvCHUbp_skys9CFcKZMfcKhA_ODg_EAlIE,17156
-opentf/toolkit/core.py,sha256=L1fT4YzwZjqE7PUXhJL6jSVQge3ohBQv5UBb9DAC6oo,9320
-opentf_toolkit_nightly-0.56.0.dev985.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-opentf_toolkit_nightly-0.56.0.dev985.dist-info/METADATA,sha256=imWnMdMeEY6slHQO74Bc2ILL3b3RTasTfgUzWU0hnPY,1945
-opentf_toolkit_nightly-0.56.0.dev985.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-opentf_toolkit_nightly-0.56.0.dev985.dist-info/top_level.txt,sha256=_gPuE6GTT6UNXy1DjtmQSfCcZb_qYA2vWmjg7a30AGk,7
-opentf_toolkit_nightly-0.56.0.dev985.dist-info/RECORD,,
+opentf/toolkit/core.py,sha256=MvvWUrS4dvo3Qrd_Yl9uND3CllL0qtMi-y1S-Na1uQs,9621
+opentf_toolkit_nightly-0.56.0.dev988.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+opentf_toolkit_nightly-0.56.0.dev988.dist-info/METADATA,sha256=ojgsewIQOmWuo06J0f6c113DhiU3VErtAhlQh2G8c5I,1945
+opentf_toolkit_nightly-0.56.0.dev988.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+opentf_toolkit_nightly-0.56.0.dev988.dist-info/top_level.txt,sha256=_gPuE6GTT6UNXy1DjtmQSfCcZb_qYA2vWmjg7a30AGk,7
+opentf_toolkit_nightly-0.56.0.dev988.dist-info/RECORD,,