velocity-python 0.0.138__tar.gz → 0.0.140__tar.gz

{velocity_python-0.0.138 → velocity_python-0.0.140}/PKG-INFO

@@ -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 → velocity_python-0.0.140}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "velocity-python"
-version = "0.0.138"
+version = "0.0.140"
 authors = [
   { name="Velocity Team", email="info@codeclubs.org" },
 ]

{velocity_python-0.0.138 → velocity_python-0.0.140}/src/velocity/__init__.py

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

{velocity_python-0.0.138 → velocity_python-0.0.140}/src/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 → velocity_python-0.0.140}/src/velocity_python.egg-info/PKG-INFO

@@ -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