Upload main.py

main.py

@@ -167,6 +167,46 @@ class YouTubeSearch(BaseModel):
     query: str
     limit: int = 7
 
+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 = []
+    for x in user:
+        api_key = x.get("ryuzaki_api_key")
+        if api_key:
+            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:
+        raise HTTPException(status_code=401, detail="Invalid API key")
+
+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")
+
 def secs_to_mins(secs: int) -> str:
     mins, secs = divmod(secs, 60)
     return f"{mins}:{secs}"
@@ -432,47 +472,6 @@ async def serve_file(filename: str):
         content={"error": "File not found"}
     )
 
-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 = []
-    for x in user:
-        api_key = x.get("ryuzaki_api_key")
-        if api_key:
-            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:
-        raise HTTPException(status_code=401, detail="Invalid API key")
-
-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",