fixed streaks issue

streaksManagement.py

@@ -1,6 +1,6 @@
 from pymongo import MongoClient
 from datetime import datetime, timedelta
-from typing import Dict, Union
+from typing import Dict, List, Union
 
 def get_current_date() -> str:
     # Get the current date and return it in YYYY-MM-DD format
@@ -17,7 +17,45 @@ def check_date_streak(start_date: str, end_date: str) -> bool:
     end_date = datetime.fromisoformat(end_date)
     
     # Compare the dates to check if they are consecutive (1 day apart)
-    return end_date - start_date == timedelta(days=1)
+    if end_date - start_date == timedelta(days=0) or end_date - start_date == timedelta(days=1):
+        return True
+    else:
+        return False
+
+def save_in_streaks_history(db_uri,document:dict={"user_id":"67c9b68678fbf39f4ed94e01","streaks_date":["2025-03-07"]}):
+    client = MongoClient(db_uri)
+    document= document.pop("_id",None)
+    try:
+        db = client["crayonics"]
+        collection = db["StreaksHistory"]
+        # Check for existing user
+        found_user = collection.find_one({"user_id": document.get('user_id')})
+        current_date = get_current_date()
+        if found_user is None:
+            # New user case
+            print("added new streak reset record")
+            document["streaks_records"] = {"date_of_streaks_reset":current_date,"streaks_dates":document["streaks_date"]}
+            document.pop("streaks_date")
+            collection.insert_one(document)
+            return True
+        else:
+            print("added another streak reset record")
+            current_record={"date_of_streaks_reset":current_date,"streaks_dates":document["streaks_date"]}
+            
+            dates = found_user["streaks_records"] + [current_record]
+            collection.update_one(
+                    {"user_id": document.get("user_id")},
+                    {"$set": {"streaks_records": dates}}
+                )
+            return True
+        
+    except Exception as e:
+        print(f"Error in creating a streaks reset record: {str(e)}")
+        return False
+    finally:
+        client.close()
+    
+   
 
 def streaks_manager(db_uri: str, document: Dict) -> Union[bool, str]:
     """
@@ -38,6 +76,7 @@ def streaks_manager(db_uri: str, document: Dict) -> Union[bool, str]:
         # Check for existing user
         found_user = collection.find_one({"user_id": document.get('user_id')})
         current_date = get_current_date()
+        
         document['streak_dates'] = [current_date]
 
         if found_user is None:
@@ -52,18 +91,23 @@ def streaks_manager(db_uri: str, document: Dict) -> Union[bool, str]:
             if is_a_streak:
                 print("its a streak guys")
                 # Extend existing streak
-                dates = found_user["streak_dates"] + [current_date]
+                if not (current_date== found_user["streak_dates"][-1]):
+                    dates = found_user["streak_dates"] + [current_date]
+                else:
+                    dates=found_user["streak_dates"]
                 collection.update_one(
                     {"user_id": document.get("user_id")},
                     {"$set": {"streak_dates": dates}}
                 )
                 return True
             else:
-                # Reset streak if not consecutive
-                collection.find_one_and_replace(
-                    {"user_id": document.get('user_id')},
-                    document
-                )
+                print(found_user,type(found_user))
+                # save_in_streaks_history(db_uri=db_uri,document=found_user)
+                # # Reset streak if not consecutive
+                # collection.find_one_and_replace(
+                #     {"user_id": document.get('user_id')},
+                #     document
+                # )
                 return "User Already Exists"
                 
     except Exception as e:

testes.py

@@ -0,0 +1,6 @@
+from streaksManagement import streaks_manager
+
+
+streaks_manager(db_uri="mongodb+srv://groupcresearchseminar:[email protected]/?retryWrites=true&w=majority&appName=Cluster0",document={"user_id":"67c9b68678fbf39f4ed94e01"})
+
+