whatap-python 1.8.11__tar.gz → 1.8.11.post0__tar.gz

{whatap_python-1.8.11 → whatap_python-1.8.11.post0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: whatap-python
-Version: 1.8.11
+Version: 1.8.11.post0
 Summary: Monitoring and Profiling Service
 Home-page: https://www.whatap.io
 Author: whatap

whatap_python-1.8.11.post0/whatap/build.py

@@ -0,0 +1,4 @@
+app = 'Python'
+name = 'whatap-python'
+version = '1.8.11.post0'
+release_date = '20250903'

{whatap_python-1.8.11 → whatap_python-1.8.11.post0}/whatap/trace/mod/application/wsgi.py

@@ -77,6 +77,36 @@ def trace_handler(fn, start=False, preload=None):
 
     return handler
 
+def async_trace_handler(fn, start=False, preload=None):
+    def handler(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            if preload:
+                preload(*args, **kwargs)
+
+            ctx = TraceContextManager.getLocalContext()
+
+            if not start and not ctx:
+                return await fn(*args, **kwargs)
+
+            try:
+                callback = await func(*args, **kwargs)
+            except Exception as e:
+                if ctx and ctx.error_step == e:
+                    ctx.error_step = None
+                    raise e
+                raise
+            else:
+                if ctx and ctx.error_step:
+                    e = ctx.error_step
+                    ctx.error_step = None
+                    raise e
+                return callback
+
+        return wrapper
+
+    return handler
+
 
 def start_interceptor(ctx):
     if conf.dev:

{whatap_python-1.8.11 → whatap_python-1.8.11.post0}/whatap/trace/mod/database/psycopg3.py

@@ -1,6 +1,6 @@
 import inspect
 
-from whatap.trace.mod.application.wsgi import trace_handler
+from whatap.trace.mod.application.wsgi import trace_handler , async_trace_handler
 from whatap.trace.mod.database.util import (
     interceptor_db_con, interceptor_db_execute, interceptor_db_close,
     async_interceptor_db_con, async_interceptor_db_execute, async_interceptor_db_close,
@@ -236,7 +236,7 @@ def _sync_wrapper(fn):
 
 
 def _async_wrapper(fn):
-    @trace_handler(fn)
+    @async_trace_handler(fn)
     async def wrapper(*args, **kwargs):
         if _is_called_by_pool():
             return await fn(*args, **kwargs)
@@ -278,7 +278,6 @@ def _get_conn_info(connection):
 
 
 def _pool_getconn_wrapper(original_getconn):
-    @trace_handler(original_getconn)
     def wrapper(self, *args, **kwargs):
         connection = original_getconn(self, *args, **kwargs)
         conn_info = _get_conn_info(connection)
@@ -292,7 +291,6 @@ def _pool_getconn_wrapper(original_getconn):
 
 
 def _async_pool_getconn_wrapper(original_getconn):
-    @trace_handler(original_getconn)
     async def wrapper(self, *args, **kwargs):
         connection = await original_getconn(self, *args, **kwargs)
         conn_info = _get_conn_info(connection)

{whatap_python-1.8.11 → whatap_python-1.8.11.post0}/whatap_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: whatap-python
-Version: 1.8.11
+Version: 1.8.11.post0
 Summary: Monitoring and Profiling Service
 Home-page: https://www.whatap.io
 Author: whatap

whatap_python-1.8.11/whatap/build.py

@@ -1,4 +0,0 @@
-app = 'Python'
-name = 'whatap-python'
-version = '1.8.11'
-release_date = '20250901'