Upload main.py

main.py

@@ -86,7 +86,6 @@ SOURCE_ALPHA_URL = os.environ["SOURCE_ALPHA_URL"]
 SOURCE_WAIFU_URL = os.environ["SOURCE_WAIFU_URL"]
 SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
-SOURCE_CALLI_GRAPHY_URL = os.environ["SOURCE_CALLI_GRAPHY_URL"]
 SOURCE_WHAT_GAY_URL = os.environ["SOURCE_WHAT_GAY_URL"]
 SOURCE_ASSISTANT_GOOGLE_AI = os.environ["SOURCE_ASSISTANT_GOOGLE_AI"]
 SOURCE_MONITOR_URL = os.environ["SOURCE_MONITOR_URL"]
@@ -241,6 +240,13 @@ class RyuzakiAi(BaseModel):
 class SibylSystem(BaseModel):
     user_id: int
 
+class SibylSystemResponse(BaseModel):
+    sibyl_name: str
+    reason: str
+    is_banned: bool
+    date_joined: str
+    sibyl_user_id: int
+
 class GoogleReverse(BaseModel):
     engine: str="google_reverse_image"
     image_url: str
@@ -314,6 +320,9 @@ class TiktokDownloader(BaseModel):
 class GetMonitorLogs(BaseModel):
     logs: int
 
+class DownloadLink(BaseModel):
+    link: str
+
 @app.post("/ryuzaki/profile-clone", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def profile_clone(
     item: ProfileClone,
@@ -444,25 +453,28 @@ def sibyl_system_ban(
     except Exception as e:
         return ErrorStatus(status="false", message=f"Internal server error: {str(e)}")
 
-@app.get("/ryuzaki/sibyl", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+@app.get("/ryuzaki/sibyl", response_model=SibylSystemResponse, responses={422: {"model": SibylSystemResponse}})
 def sibyl_system(
     item: SibylSystem,
     api_key: None = Depends(validate_api_key)
 ):
     sibyl_name, reason, is_banned, date_joined, sibyl_user_id = get_sibyl_system_banned(item.user_id)
     if sibyl_name and reason and is_banned and date_joined and sibyl_user_id:
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "sibyl_name": sibyl_name,
-                "reason": reason,
-                "is_banned": is_banned,
-                "date_joined": date_joined,
-                "sibyl_user_id": sibyl_user_id
-            }
+        return SibylSystemResponse(
+            sibyl_name=sibyl_name,
+            reason=reason,
+            is_banned=is_banned,
+            date_joined=date_joined,
+            sibyl_user_id=sibyl_user_id
         )
     else:
-        return ErrorStatus(status="false", message="Not found user")
+        return SibylSystemResponse(
+            sibyl_name="",
+            reason="",
+            is_banned=False,
+            date_joined="",
+            sibyl_user_id=0
+        )
 
 @app.get("/ryuzaki/ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def ryuzaki_ai(
@@ -1086,9 +1098,9 @@ def tiktok_downloader(item: TiktokBeta):
         return {"status": "false", "message": "Invalid Link"}
 
 @app.get("/ryuzaki/mediafire")
-def mediafire(link: Union[str, None] = None):
+def mediafire(item: DownloadLink):
     try:
-        down_link = str(link)
+        down_link = str(item.link)
         mid = down_link.split('/', 5)
         if mid[3] == "view":
             mid[3] = "file"
@@ -1113,7 +1125,7 @@ def mediafire(link: Union[str, None] = None):
             status="True",
             randydev={
                 "directDownload": a,
-                "original": link,
+                "original": item.link,
                 "id": id,
                 "name": name,
                 "readable": byte,
@@ -1125,140 +1137,111 @@ def mediafire(link: Union[str, None] = None):
         return {'status': 'false', 'message': 'Invalid Link'}
 
 @app.get("/ryuzaki/gdrive")
-def gdrive(link: Union[str, None] = None):
-  try:
-    down = link.split('/', 6)
-    url = f'https://drive.google.com/uc?export=download&id={down[5]}'
-    session = requests.Session()
-
-    response = session.get(url, stream=True)
-    headers = response.headers
-    content_disp = headers.get('content-disposition')
-    filename = None
-    if content_disp:
-      match = re.search(r'filename="(.+)"', content_disp)
-      if match:
-        filename = match.group(1)
-
-    content_length = headers.get('content-length')
-    last_modified = headers.get('last-modified')
-    content_type = headers.get('content-type')
-
-    return {
-      "status": "true",
-      "data": {
-        "file": {
-          "url": {
-            'directDownload': url,
-            "original": link,
-          },
-          "metadata": {
-            "id":
-            down[5],
-            "name":
-            filename if filename else 'No filename provided by the server.',
-            "size": {
-              "readable":
-              f'{round(int(content_length) / (1024 * 1024), 2)} MB' if
-              content_length else 'No content length provided by the server.',
-              "type":
-              content_type
-              if content_type else 'No content type provided by the server.'
-            },
-            "DateAndTime":
-            last_modified if last_modified else
-            'No last modified date provided by the server.',
-          }
-        }
-      }
-    }
-  except:
-    return "{'status': 'false', 'message': 'Invalid Link'}"
+def gdrive(item: DownloadLink):
+    try:
+        down = link.split('/', 6)
+        url = f'https://drive.google.com/uc?export=download&id={down[5]}'
+        session = requests.Session()
+        response = session.get(url, stream=True)
+        headers = response.headers
+        content_disp = headers.get('content-disposition')
+        filename = None
+        if content_disp:
+            match = re.search(r'filename="(.+)"', content_disp)
+            if match:
+                filename = match.group(1)
+        content_length = headers.get('content-length')
+        last_modified = headers.get('last-modified')
+        content_type = headers.get('content-type')
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "directDownload": url,
+                "original": item.link,
+                "id": down[5],
+                "name": filename if filename else "No filename provided by the server.",
+                "readable": f"{round(int(content_length) / (1024 * 1024), 2)} MB" if content_length else "No content length provided by the server.",
+                "type": content_type if content_type else "No content type provided by the server.",
+                "DateAndTime": last_modified if last_modified else "No last modified date provided by the server."
+            }
+        )
+    except:
+        return {'status': 'false', 'message': 'Invalid Link'}
 
 @app.get("/ryuzaki/anonfiles")
-def anonfiles(link: Union[str, None] = None):
-  try:
-    r = requests.get(link)
-    soup = BeautifulSoup(r.content, "html.parser")
-    a_href = soup.find("a", {"id": "download-url"}).get("href")
-    a = str(a_href)
-    id = link.split('/', 4)[3]
-    jsondata = requests.get(
-      f'https://api.anonfiles.com/v2/file/{id}/info').json()
-    jsondata['data']['file']['url']['directDownload'] = a
-    del jsondata['data']['file']['url']['full']
-
-    return jsondata
-  except:
-    return "{'status': 'false', 'message': 'Invalid Link'}"
+def anonfiles(item: DownloadLink):
+    try:
+        r = requests.get(item.link)
+        soup = BeautifulSoup(r.content, "html.parser")
+        a_href = soup.find("a", {"id": "download-url"}).get("href")
+        a = str(a_href)
+        id = link.split('/', 4)[3]
+        jsondata = requests.get(f'https://api.anonfiles.com/v2/file/{id}/info').json()
+        jsondata['data']['file']['url']['directDownload'] = a
+        del jsondata['data']['file']['url']['full']
+        return jsondata
+    except:
+        return "{'status': 'false', 'message': 'Invalid Link'}"
 
 @app.get("/ryuzaki/filechan")
-def filechan(link: Union[str, None] = None):
-  try:
-    r = requests.get(link)
-    soup = BeautifulSoup(r.content, "html.parser")
-    a_href = soup.find("a", {"id": "download-url"}).get("href")
-    a = str(a_href)
-    id = link.split('/', 4)[3]
-    jsondata = requests.get(
-      f'https://api.filechan.org/v2/file/{id}/info').json()
-    jsondata['data']['file']['url']['directDownload'] = a
-    del jsondata['data']['file']['url']['full']
-
-    return jsondata
-  except:
-    return "{'status': 'false', 'message': 'Invalid Link'}"
+def filechan(item: DownloadLink):
+    try:
+        r = requests.get(item.link)
+        soup = BeautifulSoup(r.content, "html.parser")
+        a_href = soup.find("a", {"id": "download-url"}).get("href")
+        a = str(a_href)
+        id = link.split('/', 4)[3]
+        jsondata = requests.get(f'https://api.filechan.org/v2/file/{id}/info').json()
+        jsondata['data']['file']['url']['directDownload'] = a
+        del jsondata['data']['file']['url']['full']
+        return jsondata
+    except:
+        return {'status': 'false', 'message': 'Invalid Link'}
 
 @app.get("/ryuzaki/letsupload")
-def letsupload(link: Union[str, None] = None):
-  try:
-    r = requests.get(link)
-    soup = BeautifulSoup(r.content, "html.parser")
-    a_href = soup.find("a", {"id": "download-url"}).get("href")
-    a = str(a_href)
-    id = link.split('/', 4)[3]
-    jsondata = requests.get(
-      f'https://api.letsupload.cc/v2/file/{id}/info').json()
-    jsondata['data']['file']['url']['directDownload'] = a
-    del jsondata['data']['file']['url']['full']
-
-    return jsondata
-  except:
-    return "{'status': 'false', 'message': 'Invalid Link'}"
+def letsupload(item: DownloadLink):
+    try:
+        r = requests.get(item.link)
+        soup = BeautifulSoup(r.content, "html.parser")
+        a_href = soup.find("a", {"id": "download-url"}).get("href")
+        a = str(a_href)
+        id = link.split('/', 4)[3]
+        jsondata = requests.get(f'https://api.letsupload.cc/v2/file/{id}/info').json()
+        jsondata['data']['file']['url']['directDownload'] = a
+        del jsondata['data']['file']['url']['full']
+        return jsondata
+    except:
+        return {'status': 'false', 'message': 'Invalid Link'}
 
 @app.get("/ryuzaki/megaupload")
-def megaupload(link: Union[str, None] = None):
-  try:
-    r = requests.get(link)
-    soup = BeautifulSoup(r.content, "html.parser")
-    a_href = soup.find("a", {"id": "download-url"}).get("href")
-    a = str(a_href)
-    id = link.split('/', 4)[3]
-    jsondata = requests.get(
-      f'https://api.megaupload.nz/v2/file/{id}/info').json()
-    jsondata['data']['file']['url']['directDownload'] = a
-    del jsondata['data']['file']['url']['full']
-
-    return jsondata
-  except:
-    return "{'status': 'false', 'message': 'Invalid Link'}"
+def megaupload(item: DownloadLink):
+    try:
+        r = requests.get(item.link)
+        soup = BeautifulSoup(r.content, "html.parser")
+        a_href = soup.find("a", {"id": "download-url"}).get("href")
+        a = str(a_href)
+        id = link.split('/', 4)[3]
+        jsondata = requests.get(f'https://api.megaupload.nz/v2/file/{id}/info').json()
+        jsondata['data']['file']['url']['directDownload'] = a
+        del jsondata['data']['file']['url']['full']
+        return jsondata
+    except:
+        return {'status': 'false', 'message': 'Invalid Link'}
 
 @app.get("/ryuzaki/myfile")
-def myfile(link: Union[str, None] = None):
-  try:
-    r = requests.get(link)
-    soup = BeautifulSoup(r.content, "html.parser")
-    a_href = soup.find("a", {"id": "download-url"}).get("href")
-    a = str(a_href)
-    id = link.split('/', 4)[3]
-    jsondata = requests.get(
-      f'https://api.myfile.is/v2/file/{id}/info').json()
-    jsondata['data']['file']['url']['directDownload'] = a
-    del jsondata['data']['file']['url']['full']
-
-    return jsondata
-  except:
-    return "{'status': 'false', 'message': 'Invalid Link'}"
+def myfile(item: DownloadLink):
+    try:
+        r = requests.get(item.link)
+        soup = BeautifulSoup(r.content, "html.parser")
+        a_href = soup.find("a", {"id": "download-url"}).get("href")
+        a = str(a_href)
+        id = link.split('/', 4)[3]
+        jsondata = requests.get(f'https://api.myfile.is/v2/file/{id}/info').json()
+        jsondata['data']['file']['url']['directDownload'] = a
+        del jsondata['data']['file']['url']['full']
+        return jsondata
+    except:
+        return {'status': 'false', 'message': 'Invalid Link'}
 
 description = """
 - Ryuzaki Library: [Library Here](https://github.com/TeamKillerX/RyuzakiLib)