lib.itmens/catalog/api.py

from .models import *
from .common import *
from .sites import *
from ninja import Schema
from django.http import Http404
from common.api import api, Result
from .search.views import enqueue_fetch


class SearchResult(Schema):
    items: list[ItemSchema]


@api.post("/catalog/search", response={200: SearchResult, 400: Result})
def search_item(request, query: str, category: ItemCategory | None = None):
    query = query.strip()
    if not query:
        return 200, {"message": "Invalid query"}
    result = Indexer.search(query, page=1, category=category)
    return 200, {"items": result.items}


@api.post("/catalog/fetch", response={200: ItemSchema, 202: Result})
def fetch_item(request, url: str):
    site = SiteManager.get_site_by_url(url)
    if not site:
        raise Http404(url)
    item = site.get_item()
    if item:
        return 200, item
    enqueue_fetch(url, False)
    return 202, {"message": "Fetch in progress"}


@api.get("/book/{uuid}/", response=EditionSchema)
def get_edition(request, uuid: str):
    item = Edition.get_by_url(uuid)
    if not item:
        raise Http404(uuid)
    return item


@api.get("/movie/{uuid}/", response=MovieSchema)
def get_movie(request, uuid: str):
    item = Movie.get_by_url(uuid)
    if not item:
        raise Http404(uuid)
    return item


@api.get("/tv/{uuid}/", response=TVShowSchema)
def get_tvshow(request, uuid: str):
    item = TVShow.get_by_url(uuid)
    if not item:
        raise Http404(uuid)
    return item


@api.get("/tvseason/{uuid}/", response=TVSeasonSchema)
def get_tvseason(request, uuid: str):
    item = TVSeason.get_by_url(uuid)
    if not item:
        raise Http404(uuid)
    return item


@api.get("/podcast/{uuid}/", response=PodcastSchema)
def get_podcast(request, uuid: str):
    item = Podcast.get_by_url(uuid)
    if not item:
        raise Http404(uuid)
    return item


@api.get("/album/{uuid}/", response=AlbumSchema)
def get_album(request, uuid: str):
    item = Album.get_by_url(uuid)
    if not item:
        raise Http404(uuid)
    return item


@api.get("/game/{uuid}/", response=GameSchema)
def get_game(request, uuid: str):
    item = Game.get_by_url(uuid)
    if not item:
        raise Http404(uuid)
    return item


# @api.get("/book", response=List[EditionSchema])
# def list_editions(request):
#     qs = Edition.objects.all()
#     return qs


# @api.post("/book/")
# def create_edition(request, payload: EditionInSchema):
#     edition = Edition.objects.create(**payload.dict())
#     return {"id": edition.uuid}


# @api.put("/book/{uuid}/")
# def update_edition(request, uuid: str, payload: EditionInSchema):
#     edition = get_object_or_404(Item, uid=base62.decode(uuid))
#     for attr, value in payload.dict().items():
#         setattr(edition, attr, value)
#     edition.save()
#     return {"success": True}


# @api.delete("/book/{uuid}/")
# def delete_edition(request, uuid: str):
#     edition = get_object_or_404(Edition, uid=base62.decode(uuid))
#     edition.delete()
#     return {"success": True}