mirror of
https://github.com/Benexl/FastAnime.git
synced 2025-12-05 20:40:09 -08:00
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from typing import Generator
|
|
|
|
from Model.base_model import BaseScreenModel
|
|
from libs.anilist import AniList
|
|
from Utility.media_card_loader import MediaCardLoader
|
|
from View.components import MediaCard
|
|
from Utility import show_notification
|
|
|
|
|
|
class SearchScreenModel(BaseScreenModel):
|
|
data = {}
|
|
|
|
def get_trending_anime(self)->MediaCard|dict:
|
|
success,data = AniList.get_trending()
|
|
if success:
|
|
def _data_generator()->Generator[MediaCard,MediaCard,MediaCard]:
|
|
for anime_item in data["data"]["Page"]["media"]:
|
|
yield MediaCardLoader.media_card(anime_item)
|
|
return _data_generator()
|
|
else:
|
|
return data
|
|
|
|
def search_for_anime(self,anime_title,**kwargs):
|
|
success,self.data = AniList.search(query=anime_title,**kwargs)
|
|
if success:
|
|
return self.media_card_generator()
|
|
else:
|
|
show_notification(f"Failed to search for {anime_title}",self.data["Error"])
|
|
|
|
def media_card_generator(self):
|
|
for anime_item in self.data["data"]["Page"]["media"]:
|
|
yield MediaCardLoader.media_card(anime_item)
|
|
self.pagination_info = self.data["data"]["Page"]["pageInfo"]
|
|
|
|
# def extract_pagination_info(self):
|
|
# pagination_info = None
|
|
# return pagination_info
|