Update main.py

main.py

@@ -153,39 +153,35 @@ def sibyl_system_delete(
     except Exception as e:
         return ErrorStatus(status="false", message=f"Internal server error: {str(e)}")
 
-@app.post("/UFoP/banner")
+@app.post("/UFoP/banner", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def sibyl_system_ban(
-    user_id: int = Query(..., description="User ID in query parameter"),
-    reason: str = Query(..., description="Reason in query parameter"),
-    api_key: None = Depends(validate_api_key_only_devs)
+    item: SibylSystemBan,
+    api_key: None = Depends(validate_api_key)
 ):
-    if user_id == int(DEVELOPER_ID):
+    if item.user_id == DEVELOPER_ID:
         return {"status": "false", "message": "Only Developer"}
 
     try:
         date_joined = str(dt.now())
         sibyl_ban = random.choice(db.RAMDOM_STATUS)
-        ban_data = db.get_sibyl_system_banned(user_id)
-        
-        if ban_data is not None:
-            _, _, is_banned, _, sibyl_user_id = ban_data
-            return {"status": "false", "message": "User is already banned"}
+        _, _, is_banned, _, sibyl_user_id = db.get_sibyl_system_banned(item.user_id)
 
-        db.new_sibyl_system_banned(user_id, sibyl_ban, reason, date_joined)
+        if sibyl_user_id and is_banned:
+            return {"status": "false", "message": "User is already banned"}
 
-        return {
-            "status": "true",
-            "sunkuna": {
-                "user_id": user_id,
+        db.new_sibyl_system_banned(item.user_id, sibyl_ban, item.reason, date_joined)
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "user_id": item.user_id,
                 "sibyl_name": sibyl_ban,
-                "reason": reason,
+                "reason": item.reason,
                 "date_joined": date_joined,
-                "message": f"Successfully banned {user_id} from the Sibyl ban list."
+                "message": f"Successfully banned {item.user_id} from the Sibyl ban list."
             }
-        }
+        )
     except Exception as e:
-        logging.error(f"Error in sibyl_system_ban: {e}")
-        return {"status": "false", "message": "Internal server error"}
+        return ErrorStatus(status="false", message=f"Internal server error: {str(e)}")
 
 @app.get("/UFoP/bans")
 def sibyl_system(