Update src/supabase_integration.py

src/supabase_integration.py

@@ -28,6 +28,9 @@ class AdvancedSupabaseIntegration:
         self.connection_timeout = 30
         self.max_retries = 3
         self.retry_delay = 1
+        self._cache_hits = 0
+        self._cache_misses = 0
+        self._response_times = []
         self.setup_logging()
         
     def setup_logging(self):
@@ -56,8 +59,10 @@ class AdvancedSupabaseIntegration:
         cache_key = f"context_{hash(query)}_{user_id}"
         cached = self.get_cached(cache_key)
         if cached:
+            self._cache_hits += 1
             return cached
         
+        self._cache_misses += 1
         try:
             context = {
                 "tracks": [],
@@ -133,6 +138,7 @@ class AdvancedSupabaseIntegration:
             
             for table_name, stat_key in tables_to_check:
                 try:
+                    start_time = time.time()
                     response = requests.get(
                         f"{self.supabase_url}/rest/v1/{table_name}",
                         headers=self.headers,
@@ -142,13 +148,20 @@ class AdvancedSupabaseIntegration:
                         },
                         timeout=10
                     )
+                    response_time = time.time() - start_time
+                    self._response_times.append(response_time)
                     
                     if response.status_code == 200:
                         content_range = response.headers.get('content-range')
                         if content_range and '/' in content_range:
-                            count = int(content_range.split('/')[-1])
-                            stats[stat_key] = count
-                            self.logger.debug(f"Retrieved {stat_key}: {count}")
+                            count_str = content_range.split('/')[-1]
+                            try:
+                                count = int(count_str)
+                                stats[stat_key] = count
+                                self.logger.debug(f"Retrieved {stat_key}: {count}")
+                            except (ValueError, TypeError):
+                                self.logger.warning(f"Unexpected count value for {table_name}: {count_str}")
+                                continue
                     
                 except Exception as e:
                     self.logger.warning(f"Could not get count for {table_name}: {e}")
@@ -188,28 +201,33 @@ class AdvancedSupabaseIntegration:
         }
         
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/profiles?id=eq.{user_id}",
                 headers=self.headers,
                 timeout=10
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200 and response.json():
                 profile_data = response.json()[0]
                 user_context["is_premium"] = profile_data.get("subscription_tier") in ["premium", "pro", "enterprise"]
                 user_context["account_created"] = profile_data.get("created_at")
             
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/user_preferences?user_id=eq.{user_id}",
                 headers=self.headers,
                 timeout=10
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200 and response.json():
                 preferences = response.json()[0]
                 if preferences.get("favorite_genres"):
                     user_context["favorite_genres"] = preferences["favorite_genres"][:5]
             
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/user_activity?user_id=eq.{user_id}",
                 headers=self.headers,
@@ -220,6 +238,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=10
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200 and response.json():
                 activities = response.json()
@@ -228,11 +247,13 @@ class AdvancedSupabaseIntegration:
                     for act in activities
                 ]
             
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/learning_progress?user_id=eq.{user_id}",
                 headers=self.headers,
                 timeout=10
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200 and response.json():
                 progress_data = response.json()[0]
@@ -251,6 +272,7 @@ class AdvancedSupabaseIntegration:
     def get_popular_tracks(self, limit: int = 5) -> List[Dict[str, Any]]:
         """Get popular tracks from tracks table"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/tracks",
                 headers=self.headers,
@@ -261,6 +283,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 tracks = response.json()
@@ -288,6 +311,7 @@ class AdvancedSupabaseIntegration:
     def get_popular_artists(self, limit: int = 5) -> List[Dict[str, Any]]:
         """Get popular artists from artists table"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/artists",
                 headers=self.headers,
@@ -298,6 +322,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 artists = response.json()
@@ -324,6 +349,7 @@ class AdvancedSupabaseIntegration:
     def get_recent_courses(self, limit: int = 5) -> List[Dict[str, Any]]:
         """Get recent courses from courses table"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/courses",
                 headers=self.headers,
@@ -334,6 +360,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 courses = response.json()
@@ -361,6 +388,7 @@ class AdvancedSupabaseIntegration:
     def get_featured_artists(self, limit: int = 5) -> List[Dict[str, Any]]:
         """Get featured artists (verified artists)"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/artists",
                 headers=self.headers,
@@ -372,6 +400,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 artists = response.json()
@@ -395,6 +424,7 @@ class AdvancedSupabaseIntegration:
     def get_featured_playlists(self, limit: int = 3) -> List[Dict[str, Any]]:
         """Get featured playlists"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/playlists",
                 headers=self.headers,
@@ -405,6 +435,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 playlists = response.json()
@@ -429,6 +460,7 @@ class AdvancedSupabaseIntegration:
     def get_top_genres(self, limit: int = 5) -> List[Dict[str, Any]]:
         """Get top genres"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/genres",
                 headers=self.headers,
@@ -439,6 +471,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 genres = response.json()
@@ -742,8 +775,9 @@ class AdvancedSupabaseIntegration:
                     timeout=10
                 )
                 response_time = time.time() - start_time
+                self._response_times.append(response_time)
                 
-                results["tables"][table] = {
+                table_result = {
                     "accessible": response.status_code == 200,
                     "status_code": response.status_code,
                     "response_time_ms": round(response_time * 1000, 2)
@@ -752,7 +786,14 @@ class AdvancedSupabaseIntegration:
                 if response.status_code == 200:
                     content_range = response.headers.get('content-range')
                     if content_range and '/' in content_range:
-                        results["tables"][table]["record_count"] = int(content_range.split('/')[-1])
+                        count_str = content_range.split('/')[-1]
+                        try:
+                            table_result["record_count"] = int(count_str)
+                        except (ValueError, TypeError):
+                            self.logger.warning(f"Unexpected count value in test_connection for {table}: {count_str}")
+                            table_result["record_count"] = 0
+                
+                results["tables"][table] = table_result
                 
             except Exception as e:
                 results["tables"][table] = {
@@ -786,6 +827,7 @@ class AdvancedSupabaseIntegration:
     def get_tracks_by_popularity(self) -> List[Dict[str, Any]]:
         """Get tracks grouped by popularity"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/tracks",
                 headers=self.headers,
@@ -796,6 +838,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 tracks = response.json()
@@ -817,6 +860,7 @@ class AdvancedSupabaseIntegration:
     def get_artists_by_followers(self) -> List[Dict[str, Any]]:
         """Get artists grouped by follower count"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/artists",
                 headers=self.headers,
@@ -827,6 +871,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 artists = response.json()
@@ -848,6 +893,7 @@ class AdvancedSupabaseIntegration:
     def get_courses_by_rating(self) -> List[Dict[str, Any]]:
         """Get courses grouped by rating"""
         try:
+            start_time = time.time()
             response = requests.get(
                 f"{self.supabase_url}/rest/v1/courses",
                 headers=self.headers,
@@ -858,6 +904,7 @@ class AdvancedSupabaseIntegration:
                 },
                 timeout=15
             )
+            self._response_times.append(time.time() - start_time)
             
             if response.status_code == 200:
                 courses = response.json()
@@ -879,19 +926,9 @@ class AdvancedSupabaseIntegration:
     
     def calculate_cache_hit_rate(self) -> float:
         """Calculate cache hit rate"""
-        if not hasattr(self, '_cache_access_stats'):
-            return 0.0
-        
-        hits = getattr(self, '_cache_hits', 0)
-        misses = getattr(self, '_cache_misses', 0)
-        total = hits + misses
-        
-        return (hits / total) * 100 if total > 0 else 0.0
+        total = self._cache_hits + self._cache_misses
+        return (self._cache_hits / total) * 100 if total > 0 else 0.0
     
     def calculate_average_response_time(self) -> float:
         """Calculate average response time for API calls"""
-        if not hasattr(self, '_response_times'):
-            return 0.0
-        
-        times = getattr(self, '_response_times', [])
-        return sum(times) / len(times) if times else 0.0
+        return sum(self._response_times) / len(self._response_times) if self._response_times else 0.0