Create combo.py

src/combo.py

@@ -0,0 +1,125 @@
+from fastapi import APIRouter, status
+from typing import Optional
+from helper.is_site_available import check_if_site_available
+import time
+import asyncio
+from helper.error_messages import error_handler
+
+router = APIRouter(tags=["Combo Routes"])
+
+@router.get("/combo-search")
+async def get_search_combo(query: str, limit: Optional[int] = 0):
+    start_time = time.time()
+    query = query.lower()
+    all_sites = check_if_site_available("1337x")
+    sites_list = list(all_sites.keys())
+    tasks = []
+    COMBO = {"data": []}
+    total_torrents_overall = 0
+    for site in sites_list:
+        limit = (
+            all_sites[site]["limit"]
+            if limit == 0 or limit > all_sites[site]["limit"]
+            else limit
+        )
+        tasks.append(
+            asyncio.create_task(
+                all_sites[site]["website"]().search(query, page=1, limit=limit)
+            )
+        )
+    results = await asyncio.gather(*tasks)
+    for res in results:
+        if res is not None and len(res["data"]) > 0:
+            for torrent in res["data"]:
+                COMBO["data"].append(torrent)
+            total_torrents_overall = total_torrents_overall + res["total"]
+    COMBO["time"] = time.time() - start_time
+    COMBO["total"] = total_torrents_overall
+    if total_torrents_overall == 0:
+        return error_handler(
+            status_code=status.HTTP_404_NOT_FOUND,
+            json_message={"error": "Result not found."},
+        )
+    return COMBO
+
+
+@router.get("/combo-trending")
+async def get_all_trending(limit: Optional[int] = 0):
+    start_time = time.time()
+    # * just getting all_sites dictionary
+    all_sites = check_if_site_available("1337x")
+    sites_list = [
+        site
+        for site in all_sites.keys()
+        if all_sites[site]["trending_available"] and all_sites[site]["website"]
+    ]
+    tasks = []
+    COMBO = {"data": []}
+    total_torrents_overall = 0
+    for site in sites_list:
+        limit = (
+            all_sites[site]["limit"]
+            if limit == 0 or limit > all_sites[site]["limit"]
+            else limit
+        )
+        tasks.append(
+            asyncio.create_task(
+                all_sites[site]["website"]().trending(
+                    category=None, page=1, limit=limit
+                )
+            )
+        )
+    results = await asyncio.gather(*tasks)
+    for res in results:
+        if res is not None and len(res["data"]) > 0:
+            for torrent in res["data"]:
+                COMBO["data"].append(torrent)
+            total_torrents_overall = total_torrents_overall + res["total"]
+    COMBO["time"] = time.time() - start_time
+    COMBO["total"] = total_torrents_overall
+    if total_torrents_overall == 0:
+        return error_handler(
+            status_code=status.HTTP_404_NOT_FOUND,
+            json_message={"error": "Result not found."},
+        )
+    return COMBO
+
+
+@router.get("/combo-recent")
+async def get_all_recent(limit: Optional[int] = 0):
+    start_time = time.time()
+    # just getting all_sites dictionary
+    all_sites = check_if_site_available("1337x")
+    sites_list = [
+        site
+        for site in all_sites.keys()
+        if all_sites[site]["recent_available"] and all_sites[site]["website"]
+    ]
+    tasks = []
+    COMBO = {"data": []}
+    total_torrents_overall = 0
+    for site in sites_list:
+        limit = (
+            all_sites[site]["limit"]
+            if limit == 0 or limit > all_sites[site]["limit"]
+            else limit
+        )
+        tasks.append(
+            asyncio.create_task(
+                all_sites[site]["website"]().recent(category=None, page=1, limit=limit)
+            )
+        )
+    results = await asyncio.gather(*tasks)
+    for res in results:
+        if res is not None and len(res["data"]) > 0:
+            for torrent in res["data"]:
+                COMBO["data"].append(torrent)
+            total_torrents_overall = total_torrents_overall + res["total"]
+    COMBO["time"] = time.time() - start_time
+    COMBO["total"] = total_torrents_overall
+    if total_torrents_overall == 0:
+        return error_handler(
+            status_code=status.HTTP_404_NOT_FOUND,
+            json_message={"error": "Result not found."},
+        )
+    return COMBO