Update app/database.py

app/database.py

@@ -171,6 +171,8 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
     # Ensure today is at midnight
     today_start = today.replace(hour=0, minute=0, second=0, microsecond=0)
     
+    logger.info("Updating daily count for user %s, today: %s", str(user_object_id), today_start.isoformat())
+    
     # Check if today's date already exists
     user_doc = collection.find_one({"userId": user_object_id})
     
@@ -194,6 +196,7 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
     
     # Get existing daily counts
     existing_counts = user_doc.get("ai_edit_daily_count", [])
+    logger.debug("Existing daily counts: %d entries", len(existing_counts))
     
     # Check if today's date already exists
     today_exists = False
@@ -203,8 +206,10 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
             # Normalize to midnight for comparison
             if isinstance(entry_date, datetime):
                 normalized_date = entry_date.replace(hour=0, minute=0, second=0, microsecond=0)
+                logger.debug("Comparing entry date %s with today %s", normalized_date.isoformat(), today_start.isoformat())
                 if normalized_date == today_start:
                     today_exists = True
+                    logger.info("Today's date already exists in daily count, leaving unchanged: %s", today_start.isoformat())
                     break
     
     # If today exists, do nothing (leave it as is)
@@ -213,6 +218,8 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
         return
     
     # Today doesn't exist - need to add it and fill missing dates
+    logger.info("Today's date does not exist in daily count, adding new entry")
+    
     # Find the latest date in existing counts
     last_date = None
     if existing_counts:
@@ -225,6 +232,11 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
                     dates.append(entry_date.replace(hour=0, minute=0, second=0, microsecond=0))
         if dates:
             last_date = max(dates)
+            logger.debug("Last date found: %s", last_date.isoformat())
+        else:
+            logger.debug("No valid dates found in existing counts")
+    else:
+        logger.debug("No existing counts, will add today as first entry")
     
     # Generate missing dates between last_date and today
     dates_to_add = []
@@ -239,12 +251,14 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
                 "count": 0
             })
             current_date += timedelta(days=1)
+        logger.debug("Filled %d gap dates between %s and %s", len(dates_to_add), last_date.isoformat(), today_start.isoformat())
     
     # Add today's entry with count 1
     dates_to_add.append({
         "date": today_start,
         "count": 1
     })
+    logger.info("Adding %d new date entries (including today with count 1)", len(dates_to_add))
     
     # Merge existing entries with the new ones, sort by date (oldest first),
     # and keep only the most recent 32 dates (drop the oldest beyond 32).
@@ -259,16 +273,20 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
 
         all_entries.sort(key=_entry_sort_key)
         if len(all_entries) > 32:
+            removed = len(all_entries) - 32
             all_entries = all_entries[-32:]
+            logger.debug("Removed %d oldest entries to maintain 32-entry limit", removed)
 
-        collection.update_one(
+        result = collection.update_one(
             {"userId": user_object_id},
             {"$set": {"ai_edit_daily_count": all_entries}},
         )
-        logger.debug(
-            "Updated ai_edit_daily_count with %d entries (oldest first, max 32)",
-            len(all_entries),
+        logger.info(
+            "Updated ai_edit_daily_count with %d entries (oldest first, max 32). Matched: %d, Modified: %d",
+            len(all_entries), result.matched_count, result.modified_count,
         )
+    else:
+        logger.warning("No dates to add - this should not happen!")
 
 def log_api_call(
     endpoint: str,
@@ -568,7 +586,6 @@ def close_connection():
 
 
 
-
 # """ 
 # MongoDB database connection and logging utilities, including admin media click logging. 
 # """ 
@@ -817,28 +834,29 @@ def close_connection():
 #         "count": 1
 #     })
     
-#     # Push all missing dates (including today)
+#     # Merge existing entries with the new ones, sort by date (oldest first),
+#     # and keep only the most recent 32 dates (drop the oldest beyond 32).
 #     if dates_to_add:
-#         # Ensure ai_edit_daily_count field exists
-#         if "ai_edit_daily_count" not in user_doc:
-#             collection.update_one(
-#                 {"userId": user_object_id},
-#                 {"$set": {"ai_edit_daily_count": []}}
-#             )
-        
+#         all_entries = list(existing_counts) + dates_to_add
+
+#         def _entry_sort_key(entry: Dict[str, Any]) -> datetime:
+#             dt = entry.get("date")
+#             if isinstance(dt, datetime):
+#                 return dt.replace(hour=0, minute=0, second=0, microsecond=0)
+#             return datetime.min
+
+#         all_entries.sort(key=_entry_sort_key)
+#         if len(all_entries) > 32:
+#             all_entries = all_entries[-32:]
+
 #         collection.update_one(
 #             {"userId": user_object_id},
-#             {
-#                 # Append new dates and keep only the most recent 32 entries
-#                 "$push": {
-#                     "ai_edit_daily_count": {
-#                         "$each": dates_to_add,
-#                         "$slice": -32,
-#                     }
-#                 }
-#             }
+#             {"$set": {"ai_edit_daily_count": all_entries}},
+#         )
+#         logger.debug(
+#             "Updated ai_edit_daily_count with %d entries (oldest first, max 32)",
+#             len(all_entries),
 #         )
-#         logger.debug("Added %d daily count entries (including today with count 1)", len(dates_to_add)) 
 
 # def log_api_call(
 #     endpoint: str,