github/StreamingCommunity
1
1
Fork 0
mirror of https://github.com/Arrowar/StreamingCommunity.git synced 2025-06-07 20:15:24 +00:00
Code Issues 2 Packages Projects Releases Wiki Activity

start implementing episode info for series

Browse Source
This commit is contained in:
Francesco Grazioso 2024-04-18 15:41:25 +02:00
parent 12e225ee53
commit 96bc7ffb8a
1 changed files with 29 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
api/endpoints/views.py
View File

@ -1,18 +1,19 @@
import json import json
from rest_framework import viewsets from rest_framework import viewsets
from rest_framework.decorators import action
from rest_framework.response import Response from rest_framework.response import Response
from Src.Api import search, get_version_and_domain, download_film, anime_download_film from Src.Api import search, get_version_and_domain, download_film, anime_download_film
from Src.Api.anime import EpisodeDownloader
from Src.Api.site import media_search_manager, anime_search from Src.Api.site import media_search_manager, anime_search
class SearchView(viewsets.ViewSet): class SearchView(viewsets.ViewSet):
def get_queryset(self): def list(self, request):
# Questa funzione viene chiamata prima di list e retrieve self.search_query = request.query_params.get("search_terms")
self.search_query = self.request.query_params.get("search_terms") self.type_search = request.query_params.get("type")
self.type_search = self.request.query_params.get("type")
media_search_manager.media_list = [] media_search_manager.media_list = []
self.site_version, self.domain = get_version_and_domain() self.site_version, self.domain = get_version_and_domain()
@ -22,11 +23,7 @@ class SearchView(viewsets.ViewSet):
elif self.type_search == "anime": elif self.type_search == "anime":
self.len_database = anime_search(self.search_query) self.len_database = anime_search(self.search_query)
return media_search_manager.media_list media_list = media_search_manager.media_list
def list(self, request):
# Ottieni il queryset dalle impostazioni comuni
media_list = self.get_queryset()
if self.len_database != 0: if self.len_database != 0:
data_to_return = [] data_to_return = []
@ -39,6 +36,29 @@ class SearchView(viewsets.ViewSet):
return Response({"error": "No media found with that search query"}) return Response({"error": "No media found with that search query"})
@action(detail=False, methods=['get'])
def get_episodes_info(self, request):
self.media_id = request.query_params.get("media_id")
self.media_slug = request.data.get("media_slug")
self.type_media = request.query_params.get("type_media")
self.site_version, self.domain = get_version_and_domain()
match self.type_media.upper():
case "TV":
pass
case "TV_ANIME":
episodes = []
episodes_downloader = EpisodeDownloader(self.media_id, self.media_slug)
episoded_count = episodes_downloader.get_count_episodes()
for i in range(0, episoded_count - 210): # TODO: need to implement pagination, more than 5/6 episodes is slow
print(f"Getting info for episode {i}")
episode_info = episodes_downloader.get_info_episode(index_ep=i)
episodes.append(episode_info)
return Response({"episodes_count": episodes})
return Response({"error": "No media found with that search query"})
class DownloadView(viewsets.ViewSet): class DownloadView(viewsets.ViewSet):