File size: 2,214 Bytes
1eb6087 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from fastapi import APIRouter
from fastapi import status
from typing import Optional
from helper.is_site_available import check_if_site_available
from helper.error_messages import error_handler
router = APIRouter(tags=["Category Torrents Route"])
@router.get("/get_category")
async def get_category(
site: str,
query: str,
category: str,
limit: Optional[int] = 0,
page: Optional[int] = 1,
):
all_sites = check_if_site_available(site)
site = site.lower()
query = query.lower()
category = category.lower()
if all_sites:
limit = (
all_sites[site]["limit"]
if limit == 0 or limit > all_sites[site]["limit"]
else limit
)
if all_sites[site]["search_by_category"]:
if category not in all_sites[site]["categories"]:
return error_handler(
status_code=status.HTTP_404_NOT_FOUND,
json_message={
"error": "Selected category not available.",
"available_categories": all_sites[site]["categories"],
},
)
resp = await all_sites[site]["website"]().search_by_category(
query, category, page, limit
)
if resp is None:
return error_handler(
status_code=status.HTTP_403_FORBIDDEN,
json_message={
"error": "Website Blocked Change IP or Website Domain."
},
)
elif len(resp["data"]) > 0:
return resp
else:
return error_handler(
status_code=status.HTTP_404_NOT_FOUND,
json_message={"error": "Result not found."},
)
else:
return error_handler(
status_code=status.HTTP_404_NOT_FOUND,
json_message={
"error": "Category search not availabe for {}.".format(site)
},
)
return error_handler(
status_code=status.HTTP_404_NOT_FOUND,
json_message={"error": "Selected Site Not Available"},
) |