fix kodik (step 1/?), fix tests

moe_parsers/classes.py

@@ -231,6 +231,7 @@ class Anime(object):
     def __init__(self, *args, **kwargs):
         self.orig_title: str = None
         self.title: str = None
+        self.all_titles: List[str] = None
         self.anime_id: int | str = None
         self.id_type: str = None
         self.url: str = None
@@ -243,6 +244,7 @@ def __init__(self, *args, **kwargs):
         self.data: dict = None
         self.language: str = None
         self.status: str = self.Status.UNKNOWN
+        self.description: str = None
         self.args = args
         for kwarg in kwargs:
             setattr(self, kwarg, kwargs[kwarg])

moe_parsers/providers/aniboom.py

@@ -44,7 +44,9 @@ async def get_episodes(self) -> List[AniboomEpisode]:
         if self.episodes:
             return self.episodes
         self.episodes: List[AniboomEpisode] = await self.parser.get_episodes(self.url)
-        self.total_episodes = int(self.episodes[-1].get("episode_num", 0)) or len(self.episodes)
+        self.total_episodes = int(self.episodes[-1].get("episode_num", 0)) or len(
+            self.episodes
+        )
         return self.episodes
 
     async def get_translations(self) -> dict:
@@ -98,7 +100,11 @@ async def get_videos(self) -> List[dict]:
         Returns:
             List[dict]: List of videos for each episode
         """
-        for episode in self.episodes if (self.episodes and isinstance(self.episodes[0], AniboomEpisode)) else await self.get_episodes():
+        for episode in (
+            self.episodes
+            if (self.episodes and isinstance(self.episodes[0], AniboomEpisode))
+            else await self.get_episodes()
+        ):
             try:
                 await episode.get_videos()
             except Exception:
@@ -332,7 +338,7 @@ async def get_info(self, link: str) -> dict:
         anime_data = {}
         response = await self.get(link)
         soup = await self.soup(response)
-        
+
         anime_data["link"] = link
         anime_data["animego_id"] = link[link.rfind("-") + 1 :]
         anime_data["title"] = (

moe_parsers/providers/kodik.py

@@ -1,7 +1,7 @@
 from typing import List, Literal
 from json import loads
 from base64 import b64decode
-from ..classes import Anime, Parser, ParserParams, Exceptions, M3U8Playlist
+from ..classes import Anime, Parser, ParserParams, Exceptions, M3U8Playlist, Media
 
 
 class KodikVideo:
@@ -28,6 +28,14 @@ async def get_m3u8(self) -> str:
         return await self.parser.get_m3u8(self.cloud_url)
 
 
+class KodikIframe(Media):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.parser: KodikParser = (
+            kwargs["parser"] if "parser" in kwargs else KodikParser()
+        )
+
+
 class KodikEpisode(Anime.Episode):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -75,12 +83,32 @@ async def convert2anime(self, **kwargs) -> KodikAnime:
         anime = KodikAnime(
             orig_title=kwargs["title_orig"],
             title=kwargs["title"],
+            all_titles=[]
+            + [kwargs.get("other_title", [])]
+            + kwargs.get("other_titles_en", [])
+            + kwargs.get("other_titles_jp", []),
             anime_id=kwargs["shikimori_id"],
             url="https:" + kwargs["link"],
             parser=self,
             id_type="shikimori",
             language=self.language,
             data=kwargs,
+            status=Anime.Status.COMPLETED
+            if kwargs.get("all_status", "") == "released"
+            else (
+                Anime.Status.ONGOING
+                if kwargs.get("all_status", "") == "ongoing"
+                else (Anime.Status.UNKNOWN)
+            ),
+            year=kwargs.get("year", None),
+            description=kwargs.get("description", None),
+            type=Anime.Type.TV
+            if kwargs.get("anime_kind", "") == "tv"
+            else (
+                Anime.Type.MOVIE
+                if kwargs.get("anime_kind", "") == "movie"
+                else (Anime.Type.UNKNOWN)
+            ),
         )
         return anime
 
@@ -142,6 +170,10 @@ async def search(
                         "imdb_id": result.get("imdb_id"),
                         "worldart_link": result.get("worldart_link"),
                         "link": result.get("link"),
+                        "all_status": result.get("all_status"),
+                        "description": result.get("description"),
+                        "other_titles_en": result.get("other_titles_en"),
+                        "other_titles_jp": result.get("other_titles_jp"),
                     }
                 )
                 added_titles.add(result["title"])

requirements.txt

@@ -1,5 +1,6 @@
 pytest==7.2.2
 pytest-cov==4.0.0
+pytest-asyncio
 ruff==v0.0.260
 coverage-badge~=1.1.0
 aiohttp

tests/unit/test_aniboom.py

@@ -17,8 +17,8 @@ async def test_aniboom_get_info():
         "https://animego.org/anime/atri-moi-dorogie-momenty-2595"
     )
     assert data["animego_id"] == "2595"
-    assert data["status"] == 'Вышел'
-    assert len(data['translations']) >= 1
+    assert data["status"] == "Вышел"
+    assert len(data["translations"]) >= 1
 
 
 @pytest.mark.asyncio
@@ -29,8 +29,8 @@ async def test_aniboom_get_episodes():
     )
     assert len(episodes) == 13
     assert isinstance(episodes[0], AniboomEpisode)
-    
-    
+
+
 @pytest.mark.asyncio
 async def test_aniboom_get_episode_videos():
     parser = AniboomParser()
@@ -51,7 +51,7 @@ async def test_aniboom_get_translations():
 @pytest.mark.asyncio
 async def test_aniboom_get_playlist():
     parser = AniboomParser()
-    playlist = await parser.get_mpd_content("2318", 1, '1')
+    playlist = await parser.get_mpd_content("2318", 1, "1")
     assert len(playlist.content) >= 256
 
 
@@ -61,5 +61,5 @@ async def test_aniboom_get_shikimori_id():
         await AniboomAnime().get_shikimori_id(
             "https://animego.org/anime/plastikovye-vospominaniya-2318"
         )
-        == "    "
+        == "27775"
     )

tests/unit/test_animego.py

@@ -35,6 +35,7 @@ async def test_animego_get_translations():
     translations = await parser.get_translations("2595")
     assert len(translations) >= 1
 
+
 @pytest.mark.asyncio
 async def test_animego_get_episode_videos():
     parser = AnimegoParser()
@@ -45,6 +46,7 @@ async def test_animego_get_episode_videos():
         assert len(await episode.get_videos()) > 0
         break
 
+
 @pytest.mark.asyncio
 async def test_animego_get_videos():
     parser = AnimegoParser()