Upload main.py

main.py

@@ -283,6 +283,37 @@ class MakeRayso(BaseModel):
     auto_translate: Optional[bool] = None
     ryuzaki_dark: bool=True
 
+class Webshot(BaseModel):
+    url: str
+    quality: str="1920x1080"
+    type_mine: str="JPEG"
+    pixels: str="1024"
+    cast: str="Z100"
+
+class GithubUsernames(BaseModel):
+    username: str
+
+class ChatBots(BaseModel):
+    query: str
+    user_id: Optional[int] = None
+    bot_name: Optional[str] = None
+    bot_username: Optional[str] = None
+
+class NewMonitor(BaseModel):
+    type: int=1
+    url: str
+    friendly_name: str
+
+class TiktokBeta(BaseModel):
+    tiktok_url: str
+    only_video: Optional[bool] = None
+
+class TiktokDownloader(BaseModel):
+    tiktok_url: str
+
+class GetMonitorLogs(BaseModel):
+    logs: int
+
 @app.post("/ryuzaki/profile-clone", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def profile_clone(
     item: ProfileClone,
@@ -456,7 +487,7 @@ def ryuzaki_ai(
     except Exception:
         return {"status": "false", "message": "Internal server error"}
 
-@app.get("/ryuzaki/unsplash", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+@app.get("/ryuzaki/unsplash")
 async def get_image_unsplash(item: GetImageUnsplash):
     url = SOURCE_UNSPLASH_URL
     image_url = f"{url}/?{item.query}/{item.size}"
@@ -695,16 +726,14 @@ def v1beta2_google_ai(
     except:
         return {"status": "false", "message": "Error response."}
 
-@app.post("/ryuzaki/new-monitor")
+@app.post("/ryuzaki/new-monitor", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def new_monitor(
-    type: int=1,
-    url: str=None,
-    friendly_name: str=None,
+    item: NewMonitor,
     api_key: None = Depends(validate_api_key)
 ):
-    url = SOURCE_MONITOR_URL
+    urls = SOURCE_MONITOR_URL
     token = MONITOR_API_KEYS
-    api_url = f"{url}/newMonitor"
+    api_url = f"{urls}/newMonitor"
     try:
         headers = {
             "content-type": "application/x-www-form-urlencoded",
@@ -713,9 +742,9 @@ def new_monitor(
         payload = {
             "api_key": token,
             "format": "json",
-            "type": type,
-            "url": url,
-            "friendly_name": friendly_name
+            "type": item.type,
+            "url": item.url,
+            "friendly_name": item.friendly_name
         }
         response = requests.post(api_url, data=payload, headers=headers)
         response_str = response.json()
@@ -733,9 +762,9 @@ def new_monitor(
     except:
         return {"status": "false", "message": "Error response."}
 
-@app.post("/ryuzaki/get-monitors")
+@app.post("/ryuzaki/get-monitors", response_model=SuccessResponse, responses={422: {"model": ErrorResponse}})
 def getMonitors(
-    logs: int=1,
+    item: GetMonitorLogs,
     api_key: None = Depends(validate_api_key)
 ):
     url = SOURCE_MONITOR_URL
@@ -749,7 +778,7 @@ def getMonitors(
         payload = {
             "api_key": token,
             "format": "json",
-            "logs": logs
+            "logs": item.logs
         }
         response = requests.post(api_url, data=payload, headers=headers)
         response_str = response.json()
@@ -796,10 +825,10 @@ async def get_data(username):
             base_msg += f"**An error occured while parsing the data!** \n\n**Traceback:** \n `{e}` \n\n`Make sure that you've sent the command with the correct username!`"
             return [base_msg, "https://telegra.ph//file/32f69c18190666ea96553.jpg"]
 
-@app.get("/ryuzaki/github")
-async def github(username: str=None):
+@app.get("/ryuzaki/github", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+async def github(item: GithubUsernames):
     try:
-        details = await get_data(username)
+        details = await get_data(item.username)
         return {
             "status": "true",
             "randydev":{
@@ -810,16 +839,10 @@ async def github(username: str=None):
     except:
         return {"status": "false", "message": "Error response."}
 
-@app.get("/ryuzaki/webshot")
-def webshot(
-    url: str=None,
-    quality: str="1920x1080",
-    type_mine: str="JPEG",
-    pixels: str="1024",
-    cast: str="Z100"
-):
+@app.get("/ryuzaki/webshot", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def webshot(item: Webshot):
     try:
-        required_url = f"https://mini.s-shot.ru/{quality}/{type_mine}/{pixels}/{cast}/?{url}"
+        required_url = f"https://mini.s-shot.ru/{item.quality}/{item.type_mine}/{item.pixels}/{item.cast}/?{item.url}"
         return {
             "status": "true",
             "randydev":{
@@ -829,19 +852,14 @@ def webshot(
     except:
         return {"status": "false", "message": "Error response."}
 
-@app.get("/ryuzaki/chatbot")
-def chatbot(
-    query: str=None,
-    user_id: int=None,
-    bot_name: str=None,
-    bot_username: str=None
-):
+@app.get("/ryuzaki/chatbot", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def chatbot(item: ChatBots):
     api_url = b64decode("aHR0cHM6Ly9hcGkuc2Fmb25lLmRldi9jaGF0Ym90").decode("utf-8")
     params = {
-        "query": query,
-        "user_id": user_id,
-        "bot_name": bot_name,
-        "bot_master": bot_username
+        "query": item.query,
+        "user_id": item.user_id,
+        "bot_name": item.bot_name,
+        "bot_master": item.bot_username
     }
     x = requests.get(f"{api_url}", params=params)
     if x.status_code != 200:
@@ -858,7 +876,7 @@ def chatbot(
     except:
         return {"status": "false", "message": "Error response."}
 
-@app.get("/ryuzaki/llama")
+@app.get("/ryuzaki/llama", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def get_llama(item: ChatgptCustom):
     api_url = SOURCE_WHAT_GAY_URL
     params = {"query": item.query}
@@ -877,7 +895,7 @@ def get_llama(item: ChatgptCustom):
     except:
         return {"status": "false", "message": "Error response."}
 
-@app.get("/ryuzaki/waifu")
+@app.get("/ryuzaki/waifu", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def waifu_pics(item: WaifuPics):
     waifu_api = f"{SOURCE_WAIFU_URL}/{item.types}"
     waifu_param = f"{waifu_api}/{item.category}"
@@ -909,7 +927,7 @@ def waifu_pics(item: WaifuPics):
     else:
         return {"status": "false", "message": "Error response."}
 
-@app.get("/ryuzaki/rayso")
+@app.get("/ryuzaki/rayso", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def make_rayso(item: MakeRayso):
     trans = SyncTranslator()
     api_url = b64decode("aHR0cHM6Ly9hcGkuc2Fmb25lLm1lL3JheXNv").decode("utf-8")
@@ -1013,9 +1031,9 @@ def whois_ip_address(ip_address: str=None):
     else:
         return {"status": "false", "message": "Invalid ip address"}
 
-@app.get("/ryuzaki/tiktok_douyin")
-def tiktok_douyin(tiktok_url: str=None):
-    response = requests.get(f"{SOURCE_TIKTOK_WTF_URL}={tiktok_url}")
+@app.get("/ryuzaki/tiktok_douyin", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def tiktok_douyin(item: TiktokDownloader):
+    response = requests.get(f"{SOURCE_TIKTOK_WTF_URL}={item.tiktok_url}")
     if response.status_code != 200:
         return "Internal server error"
     try:
@@ -1037,10 +1055,10 @@ def tiktok_douyin(tiktok_url: str=None):
     except:
         return {"status": "false", "message": "Error request"}
 
-@app.get("/ryuzaki/tiktok")
-def tiktok_downloader(tiktok_url: Union[str, None] = None, only_video: bool=None):
+@app.get("/ryuzaki/tiktok", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def tiktok_downloader(item: TiktokBeta)
     api_devs = SOURCE_TIKTOK_TECH_URL
-    parameter = f"tiktok?url={tiktok_url}"
+    parameter = f"tiktok?url={item.tiktok_url}"
     api_url = f"{api_devs}/{parameter}"
     response = requests.get(api_url)
 
@@ -1049,7 +1067,7 @@ def tiktok_downloader(tiktok_url: Union[str, None] = None, only_video: bool=None
     try:
         results = response.json()
         caption = results.get("result", {}).get("desc", "")
-        if only_video:
+        if item.only_video:
             video_url = results.get("result", {}).get("withoutWaterMarkVideo", "")
             if video_url:
                 return {
@@ -1075,7 +1093,6 @@ def mediafire(link: Union[str, None] = None):
     if mid[3] == "view":
       mid[3] = "file"
       down_link = '/'.join(mid)
-      print(down_link)
     r = requests.get(down_link)
     soup = BeautifulSoup(r.content, "html.parser")
     a_href = soup.find("a", {"class": "input popsok"}).get("href")
@@ -1092,29 +1109,18 @@ def mediafire(link: Union[str, None] = None):
     date = dat[0]
     byte = down.split("(", 1)[1].split(")", 1)[0]
     name = a_name.replace(" ", "").strip()
-    return {
-      "status": "true",
-      "data": {
-        "file": {
-          "url": {
-            'directDownload': a,
+    return SuccessResponse(
+        status="True",
+        randydev={
+            "directDownload": a,
             "original": link,
-          },
-          "metadata": {
             "id": id,
             "name": name,
-            "size": {
-              "readable": byte
-            },
-            "DateAndTime": {
-              "time": time,
-              "date": date
-            }
-          }
+            "readable": byte,
+            "time": byte,
+            "date": date
         }
-      }
-    }
-
+    )
   except:
     return "{'status': 'false', 'message': 'Invalid Link'}"
 
@@ -1167,7 +1173,6 @@ def gdrive(link: Union[str, None] = None):
         }
       }
     }
-
   except:
     return "{'status': 'false', 'message': 'Invalid Link'}"