added a new streaks feature

streaksManagement.py

@@ -22,38 +22,40 @@ def check_date_streak(start_date: str, end_date: str) -> bool:
     else:
         return False
 
-def save_in_streaks_history(db_uri,document:dict={"user_id":"67c9b68678fbf39f4ed94e01","streaks_date":["2025-03-07"]}):
+def save_in_streaks_history(db_uri,document):
     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
+    document.pop("_id",None)
+    print(document)
+    db = client["crayonics"]
+    collection = db["StreaksHistory"]
+    # Check for existing user
+    found_user = collection.find_one({"user_id": document['user_id']})
+    current_date = get_current_date()
+    # try:
+    if found_user is None:
+        # New user case
+        print("added new streak reset record")
+        document["streaks_records"] = [{"date_of_streak_reset":current_date,"streak_dates":document["streak_dates"]}]
+        document.pop("streak_dates")
+        collection.insert_one(document)
+        return True
+    else:
+        print("added another streak reset record")
+        current_record={"date_of_streaks_reset":current_date,"streak_dates":document["streak_dates"]}
         
-    except Exception as e:
-        print(f"Error in creating a streaks reset record: {str(e)}")
-        return False
-    finally:
-        client.close()
+        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)}")
+    #     print(document,found_user)
+    #     return False
+    # finally:
+    #     client.close()
     
    
 
@@ -101,13 +103,12 @@ def streaks_manager(db_uri: str, document: Dict) -> Union[bool, str]:
                 )
                 return True
             else:
-                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
-                # )
+                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

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