Add comprehensive debug logging for expense checking in check_user_has_category_data

app/smart_recommendation.py

@@ -285,32 +285,100 @@ class SmartBudgetRecommender:
             import traceback
             traceback.print_exc()
         
-        # Check expenses collection as fallback
+        # Check expenses collection as fallback - more comprehensive search
+        print(f"🔍 Checking expenses collection for user {user_id} with category_id {category_id}")
         try:
             try:
                 category_objid = ObjectId(category_id)
-                expense_query = {
-                    "$or": [
-                        {"user_id": ObjectId(user_id), "category": category_objid},
-                        {"user_id": ObjectId(user_id), "category": category_id},
-                        {"user_id": user_id, "category": category_objid},
-                        {"user_id": user_id, "category": category_id},
-                    ]
-                }
+                # Try multiple expense query patterns
+                expense_queries = [
+                    {"user_id": ObjectId(user_id), "category": category_objid},
+                    {"user_id": ObjectId(user_id), "category": category_id},
+                    {"user_id": ObjectId(user_id), "categoryId": category_objid},
+                    {"user_id": ObjectId(user_id), "categoryId": category_id},
+                    {"user_id": ObjectId(user_id), "headCategory": category_objid},
+                    {"user_id": ObjectId(user_id), "headCategory": category_id},
+                    {"user_id": user_id, "category": category_objid},
+                    {"user_id": user_id, "category": category_id},
+                    {"user_id": user_id, "categoryId": category_objid},
+                    {"user_id": user_id, "categoryId": category_id},
+                    {"user_id": user_id, "headCategory": category_objid},
+                    {"user_id": user_id, "headCategory": category_id},
+                ]
             except (ValueError, TypeError):
-                expense_query = {
-                    "$or": [
-                        {"user_id": ObjectId(user_id), "category": category_id},
-                        {"user_id": user_id, "category": category_id},
+                expense_queries = [
+                    {"user_id": ObjectId(user_id), "category": category_id},
+                    {"user_id": ObjectId(user_id), "categoryId": category_id},
+                    {"user_id": ObjectId(user_id), "headCategory": category_id},
+                    {"user_id": user_id, "category": category_id},
+                    {"user_id": user_id, "categoryId": category_id},
+                    {"user_id": user_id, "headCategory": category_id},
+                ]
+            
+            # Try each query pattern
+            for i, expense_query in enumerate(expense_queries, 1):
+                try:
+                    expense_count = self.db.expenses.count_documents(expense_query)
+                    if expense_count > 0:
+                        print(f"✅ Found {expense_count} expense(s) for user {user_id} with category_id {category_id}")
+                        print(f"   Expense query that matched (pattern {i}): {expense_query}")
+                        return True
+                except Exception as e:
+                    print(f"⚠️ Error with expense query pattern {i}: {e}")
+            
+            # Also try comprehensive query
+            try:
+                comprehensive_expense_query = {
+                    "$and": [
+                        {"$or": [
+                            {"user_id": ObjectId(user_id)},
+                            {"user_id": user_id}
+                        ]},
+                        {"$or": [
+                            {"category": category_objid if 'category_objid' in locals() else category_id},
+                            {"category": category_id},
+                            {"categoryId": category_objid if 'category_objid' in locals() else category_id},
+                            {"categoryId": category_id},
+                            {"headCategory": category_objid if 'category_objid' in locals() else category_id},
+                            {"headCategory": category_id},
+                        ]}
                     ]
                 }
+                expense_count = self.db.expenses.count_documents(comprehensive_expense_query)
+                if expense_count > 0:
+                    print(f"✅ Found {expense_count} expense(s) using comprehensive query for user {user_id} with category_id {category_id}")
+                    return True
+            except Exception as e:
+                print(f"⚠️ Error with comprehensive expense query: {e}")
             
-            expense_count = self.db.expenses.count_documents(expense_query)
-            if expense_count > 0:
-                print(f"✅ Found {expense_count} expense(s) for user {user_id} with category_id {category_id}")
-                return True
+            # Debug: Check total expenses for this user
+            try:
+                total_user_expenses = self.db.expenses.count_documents({
+                    "$or": [
+                        {"user_id": ObjectId(user_id)},
+                        {"user_id": user_id}
+                    ]
+                })
+                print(f"🔍 Debug: User {user_id} has {total_user_expenses} total expenses")
+                
+                # Check if category exists in any expense (sample check)
+                sample_expenses = list(self.db.expenses.find({
+                    "$or": [
+                        {"user_id": ObjectId(user_id)},
+                        {"user_id": user_id}
+                    ]
+                }).limit(5))
+                if sample_expenses:
+                    print(f"🔍 Debug: Sample expense category fields:")
+                    for exp in sample_expenses:
+                        print(f"   - category: {exp.get('category')}, categoryId: {exp.get('categoryId')}, headCategory: {exp.get('headCategory')}")
+            except Exception as e:
+                print(f"⚠️ Error in expense debug check: {e}")
+                
         except Exception as e:
-            print(f"Error checking expenses: {e}")
+            print(f"❌ Error checking expenses: {e}")
+            import traceback
+            traceback.print_exc()
         
         print(f"❌ No previous data found for user {user_id} with category_id {category_id}")
         return False