added streaks feature

app.py

@@ -336,7 +336,6 @@ def protected_route(authorization: str = Header(...)):
     # Here, you would validate the token (e.g., check with a JWT library)
     decoded_user_id,decoded_access_token = decode_jwt(token) 
     is_valid = verify_access_token(db_uri=MONGO_URI, user_id=decoded_user_id, access_token=decoded_access_token)
-    print(decoded_user_id,decoded_access_token)
     if is_valid != True:  # Example check
         raise HTTPException(status_code=401, detail="Invalid token")
     

streaksManagement.py

@@ -0,0 +1,72 @@
+
+from pymongo import MongoClient
+from datetime import datetime,timedelta
+
+def get_current_date():
+    # Get the current date and return it in YYYY-MM-DD format
+    return datetime.now().date().isoformat()
+
+def get_another_date(day:int):
+    # Get the current date and add one day to it
+    tomorrow = datetime.now() + timedelta(days=day)
+    return tomorrow.date().isoformat()
+
+def check_date_streak(start_date: str, end_date: str) -> bool:
+    # Convert ISO strings to datetime objects
+    start_date = datetime.fromisoformat(start_date)
+    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)
+
+def streaks_manager(db_uri: str,  document: dict) -> str:
+    """
+    Inserts a new document into the specified MongoDB collection.
+
+    Parameters:
+        db_uri (str): MongoDB connection URI.
+        db_name (str): Name of the database.
+        collection_name (str): Name of the collection.
+        document (dict): The document to insert.
+
+    Returns:
+        str: The ID of the inserted document.
+    """
+    # Connect to MongoDB
+    client = MongoClient(db_uri)
+    db = client["crayonics"]
+    collection = db["Streaks"]
+    
+    # Insert the document
+    foundUser = collection.find_one({"user_id":document.get('user_id')})
+    streak_dates=get_current_date()
+
+    document['streak_dates']= [streak_dates]
+    if foundUser==None:
+        result = collection.insert_one(document)
+        client.close()
+        return True
+    else:
+        print("user has a streak record")
+        is_a_streak=check_date_streak(start_date=foundUser['streak_dates'][-1],end_date=get_current_date())
+        if is_a_streak:
+            print("its a streak guys")
+            dates = []
+            for d in foundUser["streak_dates"]:
+                dates.append(d)
+            dates.append(get_current_date())
+            collection.update_one(filter={"user_id":document.get("user_id")},update={
+                "$set":{"streak_dates":dates}
+            }
+                                  )
+            client.close()
+            return True
+        else:
+            collection.find_one_and_replace(filter={"user_id":document.get('user_id')},replacement=document)
+        return "User Already Exists"
+    
+    # Close the connection
+    
+
+
+# streaks_manager(db_uri="mongodb+srv://groupcresearchseminar:[email protected]/?retryWrites=true&w=majority&appName=Cluster0",document={"user_id":"67c9600773ef7350189a2d90"})

tokenManagement.py

@@ -3,8 +3,10 @@
 import datetime
 from bson import ObjectId
 
+from streaksManagement import streaks_manager
 
-def is_current_date_greater_than_previous(previous_date):
+
+def isexpired(previous_date):
     # Get the current date and time
     current_date =datetime.datetime.now()
 
@@ -42,6 +44,9 @@ def create_accessToken(db_uri: str, user_id: str, refresh_token: str) -> str:
     collection.find_one_and_delete({"refresh_token":refresh_token})
     # Insert the document
     result = collection.insert_one({"user_id":user_id,"refresh_token":refresh_token,"current_time":current_time,"expire_at":expire_at})
+    streaks_doc={}
+    streaks_doc['user_id'] = user_id
+    streaks_manager(db_uri=db_uri,document=streaks_doc)
     client.close()
     return str(result.inserted_id)
 
@@ -73,6 +78,9 @@ def create_refreshToken(db_uri: str, user_id: str) -> str:
     collection = db["RefreshToken"]
     # Insert the document
     result = collection.insert_one({"user_id":user_id,"current_time":current_time,"expire_at":expire_at,"previous_access_token":"None"})
+    streaks_doc={}
+    streaks_doc['user_id'] = user_id
+    streaks_manager(db_uri=db_uri,document=streaks_doc)
     client.close()
     return str(result.inserted_id)
     
@@ -123,12 +131,20 @@ def verify_access_token(db_uri: str, user_id: str, access_token: str) -> bool:
         return False
     else:
         if str(doc['_id']) == access_token:
-            if is_current_date_greater_than_previous(doc['expire_at']):
+            if isexpired(doc['expire_at']):
+                streaks_doc={}
+                streaks_doc['user_id'] = user_id
+                streaks_manager(db_uri=db_uri,document=streaks_doc)
                 return False
             else:
+                streaks_doc={}
+                streaks_doc['user_id'] = user_id
+                streaks_manager(db_uri=db_uri,document=streaks_doc)
                 return True
         else:
-            print("doc exists",str(doc['_id']),access_token,str(doc['_id']) == access_token )
+            streaks_doc={}
+            streaks_doc['user_id'] = user_id
+            streaks_manager(db_uri=db_uri,document=streaks_doc)
             return False
     
 

utils.py

@@ -1,6 +1,7 @@
 import requests
 from pymongo import MongoClient
 from password import *
+from streaksManagement import streaks_manager
 
 def google_search(query, api_key, cx):
     url = f"https://www.googleapis.com/customsearch/v1?q={query}&key={api_key}&cx={cx}"
@@ -130,7 +131,6 @@ def create_user(db_uri: str, db_name: str, collection_name: str, document: dict)
     client = MongoClient(db_uri)
     db = client[db_name]
     collection = db[collection_name]
-    
     # Insert the document
     s = collection.find_one({"email":document.get('email')})
     password = hash_password(document.get('password'))
@@ -138,6 +138,9 @@ def create_user(db_uri: str, db_name: str, collection_name: str, document: dict)
     if s==None:
         result = collection.insert_one(document)
         client.close()
+        streaks_doc={}
+        streaks_doc['user_id'] = result.inserted_id
+        streaks_manager(db_uri=db_uri,document=streaks_doc)
         return str(result.inserted_id)
     else:
         client.close()
@@ -167,6 +170,7 @@ def create_questionaire(db_uri: str, db_name: str, collection_name: str, documen
     # Insert the document
     result = collection.insert_one(document)
     client.close()
+
     return str(result.inserted_id)
 
     
@@ -178,6 +182,7 @@ def create_questionaire(db_uri: str, db_name: str, collection_name: str, documen
 
 
 def login_user(db_uri: str, db_name: str, collection_name: str, document: dict) -> str:
+    streaks_doc={}
     """
     Inserts a new document into the specified MongoDB collection.
 
@@ -204,6 +209,8 @@ def login_user(db_uri: str, db_name: str, collection_name: str, document: dict)
     else:
 
         if check_password(password=document['password'],hashed_password=s['password']):
+            streaks_doc['user_id'] = s.get("_id")
+            streaks_manager(db_uri=db_uri,document=streaks_doc)
             client.close()
             return str(s['_id'])
         else: