insightconnect-plugin-runtime 6.3.2__py3-none-any.whl → 6.3.4__py3-none-any.whl

insightconnect_plugin_runtime/telemetry.py

@@ -1,7 +1,8 @@
-import functools
+import contextvars
+from functools import wraps, partial
 from typing import Any, Callable
-from flask.app import Flask
 
+from flask.app import Flask
 from opentelemetry import trace
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.instrumentation.flask import FlaskInstrumentor
@@ -53,7 +54,7 @@ def auto_instrument(func: Callable) -> Callable:
     :return:
     """
 
-    @functools.wraps(func)
+    @wraps(func)
     def wrapper(*args, **kwargs):
         tracer = trace.get_tracer(__name__)
         with tracer.start_as_current_span(func.__name__):
@@ -71,3 +72,23 @@ def create_post_fork(app_getter: Callable, plugin_getter: Callable, config_gette
             init_tracing(app, plugin, endpoint)
 
     return post_fork
+
+
+def with_context(context: contextvars.Context, function: Callable) -> Callable:
+    """
+    Creates a wrapper function that executes the target function with the specified context.
+
+    :param context: The Context object to apply when executing the function
+    :type context: contextvars.Context
+
+    :param function: The function to wrap with the specified context
+    :type function: Callable
+
+    :return: A wrapper function that applies the context when called
+    :rtype: Callable
+    """
+
+    def _wrapper(context_: contextvars.Context, function_: Callable, *args, **kwargs):
+        return context_.copy().run(function_, *args, **kwargs)
+
+    return partial(_wrapper, context, function)

{insightconnect_plugin_runtime-6.3.2.dist-info → insightconnect_plugin_runtime-6.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: insightconnect-plugin-runtime
-Version: 6.3.2
+Version: 6.3.4
 Summary: InsightConnect Plugin Runtime
 Home-page: https://github.com/rapid7/komand-plugin-sdk-python
 Author: Rapid7 Integrations Alliance
@@ -16,7 +16,7 @@ Requires-Dist: requests==2.32.3
 Requires-Dist: python_jsonschema_objects==0.5.2
 Requires-Dist: jsonschema==4.22.0
 Requires-Dist: certifi==2024.12.14
-Requires-Dist: Flask==3.1.0
+Requires-Dist: Flask==3.1.1
 Requires-Dist: gunicorn==23.0.0
 Requires-Dist: greenlet==3.1.1
 Requires-Dist: gevent==24.11.1
@@ -224,6 +224,8 @@ contributed. Black is installed as a test dependency and the hook can be initial
 after cloning this repository.
 
 ## Changelog
+* 6.3.4 - Addressed vulnerabilities within the slim and non-slim Python images (bumping packages)
+* 6.3.3 - Add helper func for tracing context to preserve parent span in threadpools
 * 6.3.2 - Raise `APIException` from within the `response_handler` to easily access the status code within the plugin
 * 6.3.1 - Improved filtering for `custom_config` parameters for plugin tasks
 * 6.3.0 - Add Tracing Instrumentation

{insightconnect_plugin_runtime-6.3.2.dist-info → insightconnect_plugin_runtime-6.3.4.dist-info}/RECORD

@@ -11,7 +11,7 @@ insightconnect_plugin_runtime/schema.py,sha256=6MVw5hqGATU1VLgwfOWfPsP3hy1OnsugC
 insightconnect_plugin_runtime/server.py,sha256=DHooHBQa1M2z7aETTWK6u9B2jChi8vONzqK4n_c94f4,13138
 insightconnect_plugin_runtime/step.py,sha256=KdERg-789-s99IEKN61DR08naz-YPxyinPT0C_T81C4,855
 insightconnect_plugin_runtime/task.py,sha256=d-H1EAzVnmSdDEJtXyIK5JySprxpF9cetVoFGtWlHrg,123
-insightconnect_plugin_runtime/telemetry.py,sha256=6JgZOyhB0qcDYKPlQZ47M2ZPfDpfAE_zuAhBtw4GMvo,2586
+insightconnect_plugin_runtime/telemetry.py,sha256=qJJ6N7hWhOQ-I40AJIAAZjdOvRmXMdJo0e3jwXQLOfs,3312
 insightconnect_plugin_runtime/trigger.py,sha256=Zq3cy68N3QxAGbNZKCID6CZF05Zi7YD2sdy_qbedUY8,874
 insightconnect_plugin_runtime/util.py,sha256=8cle29INhnshEcL2LWpaC0ZGqevjq8pW8TE0MFEiYYw,8475
 insightconnect_plugin_runtime/variables.py,sha256=7FjJGnU7KUR7m9o-_tRq7Q3KiaB1Pp0Apj1NGgOwrJk,3056
@@ -79,7 +79,7 @@ tests/unit/test_server_spec.py,sha256=je97BaktgK0Fiz3AwFPkcmHzYtOJJNqJV_Fw5hrvqX
 tests/unit/test_trigger.py,sha256=E53mAUoVyponWu_4IQZ0IC1gQ9lakBnTn_9vKN2IZfg,1692
 tests/unit/test_variables.py,sha256=OUEOqGYZA3Nd5oKk5GVY3hcrWKHpZpxysBJcO_v5gzs,291
 tests/unit/utils.py,sha256=hcY0A2H_DMgCDXUTvDtCXMdMvRjLQgTaGcTpATb8YG0,2236
-insightconnect_plugin_runtime-6.3.2.dist-info/METADATA,sha256=vgKGS09fSB4nQCCckH633xSzDf4UO6Waaae_XHnVivE,16419
-insightconnect_plugin_runtime-6.3.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-insightconnect_plugin_runtime-6.3.2.dist-info/top_level.txt,sha256=AJtyJOpiFzHxsbHUICTcUKXyrGQ3tZxhrEHsPjJBvEA,36
-insightconnect_plugin_runtime-6.3.2.dist-info/RECORD,,
+insightconnect_plugin_runtime-6.3.4.dist-info/METADATA,sha256=5S-gdJ376YDQ9HoRZbaBE2D2mg8PdRGvRx52MsdIHoM,16602
+insightconnect_plugin_runtime-6.3.4.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+insightconnect_plugin_runtime-6.3.4.dist-info/top_level.txt,sha256=AJtyJOpiFzHxsbHUICTcUKXyrGQ3tZxhrEHsPjJBvEA,36
+insightconnect_plugin_runtime-6.3.4.dist-info/RECORD,,

{insightconnect_plugin_runtime-6.3.2.dist-info → insightconnect_plugin_runtime-6.3.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any