Upload main.py

main.py

@@ -49,7 +49,7 @@ from telegraph import Telegraph, upload_file
 from pathlib import Path
 from serpapi import GoogleSearch
 
-from fastapi import FastAPI, UploadFile, File, Response
+from fastapi import FastAPI, UploadFile, File, Response, Request
 from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
 from fastapi import Depends, HTTPException, status
 from fastapi.openapi.utils import get_openapi
@@ -246,11 +246,13 @@ def get_profile_clone(user_id):
 def new_verify_otp(
     user_id,
     otp_code,
+    ip_addres,
     auth_status
 ):
     update_doc = {
         "user_id": user_id,
         "otp_code": otp_code,
+        "ip_addres": ip_addres,
         "auth_status": auth_status,
     }
     collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
@@ -272,14 +274,16 @@ class OTPCallbackPayload(BaseModel):
     user_id: int
 
 @app.get("/api/get-verify", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
-def get_verify_otp(user_id: int):
+def get_verify_otp(user_id: int, request: Request):
     try:
+        client_host = request.client.host
         get_otp_code = get_stored_otp(user_id)
         if get_otp_code:
             return SuccessResponse(
                 status="True",
                 randydev={
                     "otp_code": get_otp_code,
+                    "ip_addres": client_host,
                     "auth_status": "verified",
                 }
             )
@@ -288,6 +292,7 @@ def get_verify_otp(user_id: int):
                 status="False",
                 randydev={
                     "otp_code": None,
+                    "ip_addres": client_host,
                     "auth_status": "unverified",
                 }
             )
@@ -298,12 +303,14 @@ def get_verify_otp(user_id: int):
         )
         
 @app.post("/api/verify", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
-def verify_otp(payload: OTPCallbackPayload):
+def verify_otp(payload: OTPCallbackPayload, request: Request):
     try:
+        client_host = request.client.host
         otp = generate_otp()
         new_verify_otp(
             payload.user_id,
             otp,
+            ip_addres=client_host,
             auth_status="verified"
         )
         return SuccessResponse(
@@ -311,6 +318,7 @@ def verify_otp(payload: OTPCallbackPayload):
             randydev={
                 "message": "successfully verified",
                 "otp_code": otp,
+                "ip_addres": client_host,
                 "auth_status": "verified",
             }
         )
@@ -320,6 +328,7 @@ def verify_otp(payload: OTPCallbackPayload):
             randydev={
                 "message": "Failed unverified",
                 "otp_code": None,
+                "ip_addres": client_host,
                 "auth_status": "unverified",
             }
         )