Upload main.py

main.py

@@ -27,6 +27,7 @@ import shutil
 import random
 import tempfile
 import io
+import string
 import openai
 from io import BytesIO
 from datetime import datetime as dt
@@ -242,6 +243,88 @@ def get_profile_clone(user_id):
     else:
         return None
 
+def new_verify_otp(
+    user_id,
+    otp_code
+    auth_status
+):
+    update_doc = {
+        "user_id": user_id,
+        "otp_code": otp_code,
+        "auth_status": auth_status,
+    }
+    collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
+
+def get_verify_otp(user_id):
+    user = collection.find_one({"user_id": user_id})
+    if user:
+        otp_code = user.get("otp_code")
+        return otp_code
+    else:
+        return None
+
+def generate_otp(length=6):
+    """Generate a random OTP"""
+    digits = string.digits
+    otp = ''.join(random.choice(digits) for i in range(length))
+    return otp
+
+class OTPCallbackPayload(BaseModel):
+    user_id: str
+
+@app.get("/api/verify", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+def get_verify_otp(user_id: int):
+    try:
+        get_otp_code = get_verify_otp(user_id)
+        if get_otp_code:
+            return SuccessResponse(
+                status="True",
+                randydev={
+                    "otp_code": get_otp_code,
+                    "auth_status": "verified",
+                }
+            )
+        else:
+            return SuccessResponse(
+                status="False",
+                randydev={
+                    "otp_code": None,
+                    "auth_status": "unverified",
+                }
+            )
+    except Exception:
+        return SuccessResponse(
+            status="False",
+            randydev={"message": "Failed unverified"}
+        )
+        
+@app.post("/api/verify", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+def verify_otp(payload: OTPCallbackPayload):
+    try:
+        otp = generate_otp()
+        new_verify_otp(
+            payload.user_id,
+            otp,
+            auth_status="verified"
+        )
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "message": "successfully verified",
+                "otp_code": otp,
+                "auth_status": "verified",
+            }
+        )
+    except Exception:
+        return SuccessResponse(
+            status="False",
+            randydev={
+                "message": "Failed unverified",
+                "otp_code": None,
+                "auth_status": "unverified",
+            }
+        )
+
 @app.post("/ryuzaki/profile-clone", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
 def profile_clone(
     item: ProfileClone,