velocity-python 0.0.138__py3-none-any.whl → 0.0.140__py3-none-any.whl

velocity/__init__.py

@@ -1,4 +1,4 @@
-__version__ = version = "0.0.138"
+__version__ = version = "0.0.140"
 
 from . import aws
 from . import db

velocity/aws/handlers/mixins/activity_tracker.py

@@ -9,33 +9,34 @@ import copy
 import json
 import os
 import time
-from abc import ABC, abstractmethod
-from typing import Dict, Any, Optional
+from abc import ABC
+from datetime import date, datetime
+from typing import Dict, Any
 
 
 class ActivityTracker(ABC):
     """
     Mixin class providing standardized activity tracking for Lambda handlers.
-    
+
     Tracks API calls to the aws_api_activity table with consistent data structure
     and automatic duration calculation.
     """
-    
+
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.activity_log_key = None
         self.start_time = None
         self.end_time = None
         self.activity_data = {}
-    
+
     def track_activity_start(self, tx, context):
         """Start tracking activity for the current request"""
         self.start_time = time.time()
-        
+
         # Gather common activity data
         postdata = context.postdata()
         payload = context.payload()
-        
+
         self.activity_data = {
             "action": context.action(),
             "args": json.dumps(context.args()),
@@ -45,21 +46,34 @@ class ActivityTracker(ABC):
             "user_branch": os.environ.get("USER_BRANCH", "Unknown"),
             "start_timestamp": self.start_time,
         }
-        
+
         # Add user information if available
         user_info = self._extract_user_info(payload)
         if user_info:
             self.activity_data.update(user_info)
-            
+
         # Add session information
         session_data = context.session()
         if session_data:
-            self.activity_data.update({k: v for k, v in session_data.items() 
-                                     if k not in ['cognito_user']})
-        
+            self.activity_data.update(self._sanitize_session_data(session_data))
+
+        # Ensure all values are serializable before persisting
+        self.activity_data = {
+            key: self._normalize_activity_value(value)
+            for key, value in self.activity_data.items()
+            if value is not None
+        }
+
         # Create the activity record
-        self.activity_log_key = tx.table("aws_api_activity").new(self.activity_data).pk
-        
+        try:
+            self.activity_log_key = tx.table("aws_api_activity").new(self.activity_data).pk
+        except Exception as exc:
+            context.log(
+                f"ActivityTracker.track_activity_start failed: {exc}; keys={list(self.activity_data.keys())}",
+                "ActivityTracker.track_activity_start",
+            )
+            raise
+
         return self.activity_log_key
     
     def track_activity_success(self, tx, context):
@@ -140,3 +154,28 @@ class ActivityTracker(ABC):
                 pass
                 
         return user_info
+
+    def _sanitize_session_data(self, session: Dict[str, Any]) -> Dict[str, Any]:
+        """Remove sensitive session keys and normalize value types"""
+        sanitized = {}
+
+        for key, value in session.items():
+            if key == "cognito_user":
+                continue
+
+            sanitized[key] = self._normalize_activity_value(value)
+
+        return sanitized
+
+    def _normalize_activity_value(self, value: Any) -> Any:
+        """Convert activity data values into types acceptable by psycopg2"""
+        if isinstance(value, (dict, list, tuple, set)):
+            try:
+                return json.dumps(value)
+            except (TypeError, ValueError):
+                return str(value)
+        if isinstance(value, (datetime, date)):
+            return value.isoformat()
+        if isinstance(value, (bytes, bytearray)):
+            return value.decode("utf-8", errors="ignore")
+        return value

{velocity_python-0.0.138.dist-info → velocity_python-0.0.140.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: velocity-python
-Version: 0.0.138
+Version: 0.0.140
 Summary: A rapid application development library for interfacing with data storage
 Author-email: Velocity Team <info@codeclubs.org>
 License-Expression: MIT

{velocity_python-0.0.138.dist-info → velocity_python-0.0.140.dist-info}/RECORD

@@ -1,4 +1,4 @@
-velocity/__init__.py,sha256=4cK9msUzcPYo3tFiZ2-Ww6__hxYNmQ_B876ClZvAKwY,147
+velocity/__init__.py,sha256=4tuIcCyAF_BNn2pB2swolVQBfQPmUTvsDhNr9YkR7zM,147
 velocity/app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 velocity/app/invoices.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 velocity/app/orders.py,sha256=fr1oTBjSFfyeMBUXRG06LV4jgwrlwYNL5mbEBleFwf0,6328
@@ -18,7 +18,7 @@ velocity/aws/handlers/lambda_handler.py,sha256=0wa_CHyJOaI5RsHqB0Ae83-B-SwlKR0qk
 velocity/aws/handlers/response.py,sha256=s2Kw7yv5zAir1mEmfv6yBVIvRcRQ__xyryf1SrvtiRc,9317
 velocity/aws/handlers/sqs_handler.py,sha256=azuV8DrFOh0hM13EnPzyYVBS-3fLe2fn9OPc4ho7sGc,3375
 velocity/aws/handlers/mixins/__init__.py,sha256=_zyEpsnKikF7D7X-F0GA4cyIrQ6wBq7k5j6Vhp17vaQ,623
-velocity/aws/handlers/mixins/activity_tracker.py,sha256=b2Cu46FM0fLS2-FkWXXAPw6yvMV_Viu5-IlpGkVAlmk,5254
+velocity/aws/handlers/mixins/activity_tracker.py,sha256=vyQ_8kpSprjzLoALDv7g2rVkfstn89Tbsg6Zb9GmVOk,6579
 velocity/aws/handlers/mixins/error_handler.py,sha256=uN2YF9v-3LzS3o_HdVpO-XMcPy3sS7SHjUg_LfbsG7Q,6803
 velocity/aws/handlers/mixins/legacy_mixin.py,sha256=_YhiPU-zzXQjGNSAKhoUwfTFlnczmU-3BkwNFqr0hYE,2117
 velocity/aws/handlers/mixins/standard_mixin.py,sha256=-wBX0PFlZAnxQsaMDEWr-xmU8TcRbQ4BZD3wmAKR2d0,2489
@@ -121,8 +121,8 @@ velocity/misc/tests/test_merge.py,sha256=Vm5_jY5cVczw0hZF-3TYzmxFw81heJOJB-dvhCg
 velocity/misc/tests/test_oconv.py,sha256=fy4DwWGn_v486r2d_3ACpuBD-K1oOngNq1HJCGH7X-M,4694
 velocity/misc/tests/test_original_error.py,sha256=iWSd18tckOA54LoPQOGV5j9LAz2W-3_ZOwmyZ8-4YQc,1742
 velocity/misc/tests/test_timer.py,sha256=l9nrF84kHaFofvQYKInJmfoqC01wBhsUB18lVBgXCoo,2758
-velocity_python-0.0.138.dist-info/licenses/LICENSE,sha256=aoN245GG8s9oRUU89KNiGTU4_4OtnNmVi4hQeChg6rM,1076
-velocity_python-0.0.138.dist-info/METADATA,sha256=Rt_XjAFeisvsIMrNAx1j4whaIjqHk8aHcFKg7IOGIxk,34262
-velocity_python-0.0.138.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-velocity_python-0.0.138.dist-info/top_level.txt,sha256=JW2vJPmodgdgSz7H6yoZvnxF8S3fTMIv-YJWCT1sNW0,9
-velocity_python-0.0.138.dist-info/RECORD,,
+velocity_python-0.0.140.dist-info/licenses/LICENSE,sha256=aoN245GG8s9oRUU89KNiGTU4_4OtnNmVi4hQeChg6rM,1076
+velocity_python-0.0.140.dist-info/METADATA,sha256=1ILF_tgyA45S-ax-Me6sC9D7Dqcb8zohfPe60SYyHEQ,34262
+velocity_python-0.0.140.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+velocity_python-0.0.140.dist-info/top_level.txt,sha256=JW2vJPmodgdgSz7H6yoZvnxF8S3fTMIv-YJWCT1sNW0,9
+velocity_python-0.0.140.dist-info/RECORD,,