Update main.py

main.py

@@ -30,7 +30,6 @@ import json
 import io
 import g4f
 import functions as code
-import database as db
 import base64
 
 from typing_extensions import Annotated
@@ -138,6 +137,15 @@ Please modify your search terms and/or try again later thank you for your unders
 ~ 🌀ʊʄ⊕ք🌀 Team
 """
 
+def get_all_api_keys():
+    user = collection.find({})
+    api_keys = []
+    for x in user:
+        api_key = x.get("ryuzaki_api_key")
+        if api_key:
+            api_keys.append(api_key)
+    return api_keys
+
 def validate_api_key(api_key: str = Header(...)):
     USERS_API_KEYS = db.get_all_api_keys()
     if api_key not in USERS_API_KEYS:
@@ -147,10 +155,63 @@ 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")
 
+RAMDOM_STATUS = [
+    "Spammer",
+    "wanted",
+    "scammer",
+    "rogue_agent",
+    "pornbot prolly",
+    "fugitive",
+    "simp"
+]
+
+def remove_sibyl_system_banned(user_id):
+    update_doc = {
+        "sibyl_ban": None,
+        "reason_sibyl": None,
+        "is_banned_sibly": None,
+        "date_joined_sib": None,
+        "sibyl_userid": None
+    }
+    return collection.update_one({"user_id": user_id}, {"$unset": update_doc}, upsert=True)
+
+def new_sibyl_system_banned(user_id, name, reason, date_joined):
+    update_doc = {
+        "sibyl_ban": name,
+        "reason_sibyl": reason,
+        "is_banned_sibly": True,
+        "date_joined_sib": date_joined,
+        "sibyl_userid": user_id
+    }
+    return collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
+
 def cybersecuritydb(user_id, mongo_url):
     update_doc = {"mongodb": mongo_url}
     return collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
 
+def get_sibyl_system_banned(user_id):
+    user = collection.find_one({"user_id": user_id})
+    if user:
+        sibyl_name = user.get("sibyl_ban")
+        reason = user.get("reason_sibyl")
+        is_banned = user.get("is_banned_sibly")
+        date_joined = user.get("date_joined_sib")
+        sibyl_user_id = user.get("sibyl_userid")
+        return sibyl_name, reason, is_banned, date_joined, sibyl_user_id
+    else:
+        return None, None, False, None, None
+
+def get_all_banned():
+    banned_users = []
+
+    users = collection.find({})
+
+    for user_id in users:
+        reason = user_id.get("reason_sibyl")
+        user_id = user_id.get("sibyl_userid")
+        banned_users.append({"user_id": user_id, "reason": reason})
+    return banned_users
+
 @app.get("/UFoP/blacklist-words")
 def blacklist_words():
     try: