Upload main.py

main.py

@@ -30,7 +30,6 @@ import io
 import string
 import openai
 from io import BytesIO
-import nest_asyncio
 from datetime import datetime as dt
 from dotenv import load_dotenv
 from bs4 import BeautifulSoup
@@ -144,6 +143,14 @@ Please modify your search terms and/or try again later thank you for your unders
 ~ @xtdevs Team
 """
 
+class FedBans(BaseModel):
+    user_id: int
+    hashtag: str
+    reason: str
+
+class GetsaFedBans(BaseModel):
+    user_id: int
+
 def get_all_api_keys():
     user = collection.find({})
     api_keys = []
@@ -153,6 +160,16 @@ def get_all_api_keys():
             api_keys.append(api_key)
     return api_keys
 
+def get_all_fedbans_api_keys():
+    user = collection.find({})
+    api_keys = []
+    for x in user:
+        api_key = x.get("fedbans_api_key")
+        if api_key:
+            api_keys.append(api_key)
+    return api_keys
+
+
 def validate_api_key(api_key: str = Header(...)):
     USERS_API_KEYS = get_all_api_keys()
     if api_key not in USERS_API_KEYS:
@@ -162,6 +179,11 @@ def validate_api_key_only_devs(api_key: str = Header(...)):
     if api_key not in ONLY_DEVELOPER_API_KEYS:
         raise HTTPException(status_code=401, detail="Invalid API key")
 
+def validate_api_key_fedbans(api_key: str = Header(...)):
+    USERS_API_KEYS = get_all_fedbans_api_keys()
+    if api_key not in USERS_API_KEYS:
+        raise HTTPException(status_code=401, detail="Invalid API key")
+
 RAMDOM_STATUS = [
     "civilian",
     "wanted",
@@ -169,9 +191,11 @@ RAMDOM_STATUS = [
     "rogue_agent",
     "innocent",
     "fugitive",
-    "covert_operator"
+    "covert_operator",
+    "spammer",
 ]
 
+
 def remove_sibyl_system_banned(user_id):
     update_doc = {
         "sibyl_ban": None,
@@ -182,6 +206,22 @@ def remove_sibyl_system_banned(user_id):
     }
     return collection.update_one({"user_id": user_id}, {"$unset": update_doc}, upsert=True)
 
+def new_user_spammers(
+    user_id,
+    hashtag,
+    reason,
+    is_banned: bool,
+    date_joined,
+):
+    update_doc = {
+        "user_id": user_id,
+        "hashtag": hashtag,
+        "reason": reason,
+        "date_joined": date_joined,
+        "is_banned": is_banned
+    }
+    return collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
+
 def new_sibyl_system_banned(user_id, name, reason, date_joined):
     update_doc = {
         "sibyl_ban": name,
@@ -208,6 +248,24 @@ def get_sibyl_system_banned(user_id):
     else:
         return None
 
+def get_fedbans_(user_id):
+    user = collection.find_one({"user_id": user_id})
+    if user:
+        user_id = user.get("user_id")
+        hashtag = user.get("hashtag")
+        reason = user.get("reason")
+        is_banned = user.get("is_banned")
+        date_joined = user.get("date_joined")
+        return [
+            user_id,
+            hashtag,
+            reason,
+            is_banned,
+            date_joined
+        ]
+    else:
+        return None
+
 def get_all_banned():
     banned_users = []
 
@@ -438,13 +496,93 @@ def sibyl_system_delete(
     except Exception as e:
         return SuccessResponse(status="False", randydev={"message": "Internal server error."})
 
+@app.get("/user/get-fedban", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+def getfedbans_(payload: GetsaFedBans, api_key: str = Depends(validate_api_key_fedbans)):
+    users_bans = get_fedbans_(payload.user_id)
+    try:
+        if users_bans[3] == True:
+            return SuccessResponse(
+                status="True",
+                randydev={
+                    "user_id": users_bans[0],
+                    "hashtag": users_bans[1],
+                    "reason": users_bans[2],
+                    "is_banned": users_bans[3],
+                    "date_joined": users_bans[4],
+                    "message": f"#GBANNED Successfully banned {payload.user_id} from the fedban list."
+                }
+            )
+        else:
+            return SuccessResponse(
+                status="False",
+                randydev={
+                    "user_id": None,
+                    "hashtag": None,
+                    "reason": None,
+                    "is_banned": False,
+                    "date_joined": None,
+                    "message": f"Not successful"
+                }
+            )
+    except Exception as e:
+        return SuccessResponse(
+            status="False",
+            randydev={
+                "message": "Not successful",
+                "error": str(e)
+            }
+        ) 
+                
+@app.post("/user/fedban", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+def fedbans_(payload: FedBans, api_key: str = Depends(validate_api_key_fedbans)):
+    if payload.user_id == 1191668125:
+        return SuccessResponse(status="False", randydev={"message": "Only Developer."})
+
+    date_joined = str(dt.now())
+    if not payload.hashtag.startswith("#"):
+        return SuccessResponse(status="False", randydev={"message": "Invalid hashtag."})
+
+    try:
+        new_user_spammers(
+            user_id=payload.user_id,
+            hashtag=payload.hashtag,
+            reason=payload.reason,
+            is_banned=True,
+            date_joined=date_joined
+        )
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "user_id": payload.user_id,
+                "hashtag": payload.hashtag,
+                "reason": payload.reason,
+                "is_banned": True,
+                "date_joined": date_joined,
+                "message": f"#GBANNED Successfully banned {payload.user_id} from the fedban list."
+            }
+        )
+    except Exception as e:
+        return SuccessResponse(
+            status="False",
+            randydev={
+                "user_id": None,
+                "hashtag": None,
+                "reason": None,
+                "is_banned": False,
+                "date_joined": None,
+                "message": "Not successful",
+                "error": str(e)
+            }
+        )
+
+
 @app.post("/ryuzaki/sibylban", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
 def sibyl_system_ban(
     item: SibylSystemBan,
     api_key: None = Depends(validate_api_key)
 ):
     if item.user_id == 1191668125:
-        return {"status": "false", "message": "Only Developer"}
+        return SuccessResponse(status="False", randydev={"message": "Only Developer."})
 
     try:
         date_joined = str(dt.now())
@@ -453,7 +591,6 @@ def sibyl_system_ban(
 
         if sibyl_user_id is not None and is_banned:
             return SuccessResponse(status="False", randydev={"message": "User is already banned."})
-
         new_sibyl_system_banned(item.user_id, sibyl_ban, item.reason, date_joined)
         return SuccessResponse(
             status="True",
@@ -895,19 +1032,16 @@ async def chatgpt_4o(
     item: ChatgptCustom,
     api_key: None = Depends(validate_api_key)
 ):
-    try:
-        client = Client()
-        response = client.chat.completions.create(
-            model="gpt-4o",
-            messages=[{"role": "user", "content": item.query}]
-        )
-        answer = response.choices[0].message.content
-        return SuccessResponse(
-            status="True",
-            randydev={"message": answer}
-        )
-    except Exception as e:
-        return SuccessResponse(status="False", randydev={"message": str(e)})
+    client = Client()
+    response = client.chat.completions.create(
+        model="gpt-4o",
+        messages=[{"role": "user", "content": item.query}],
+    )
+    answer = response.choices[0].message.content
+    return SuccessResponse(
+        status="True",
+        randydev={"message": answer}
+    )
 
 @app.post("/ryuzaki/chatgpt-custom", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 async def chatgpt_customs(
@@ -1578,4 +1712,5 @@ def custom_openapi():
     return app.openapi_schema
 
 app.openapi = custom_openapi
-app.add_exception_handler(HTTPException, custom_exception_handler)
+
+app.add_exception_handler(HTTPException, custom_exception_handler)