From d96fbe56faa22f7860b8a1e82d5d7e840967d2db Mon Sep 17 00:00:00 2001
From: Benex254 <benextempest@gmail.com>
Date: Mon, 5 Aug 2024 09:47:02 +0300
Subject: [PATCH] feat(allanime-provider): add episode number to avoid
 confusion when streaming downloaded content

---
 fastanime/libs/anime_provider/allanime/api.py | 28 +++++++++++++------
 1 file changed, 20 insertions(+), 8 deletions(-)

diff --git a/fastanime/libs/anime_provider/allanime/api.py b/fastanime/libs/anime_provider/allanime/api.py
index 0d0a0e7..59907d2 100644
--- a/fastanime/libs/anime_provider/allanime/api.py
+++ b/fastanime/libs/anime_provider/allanime/api.py
@@ -162,8 +162,11 @@ class AllAnimeAPI:
                                     print("[yellow]GogoAnime Fetched")
                                     yield {
                                         "server": "gogoanime",
-                                        "episode_title": allanime_episode["notes"]
-                                        or f'{anime["title"]}: Episode {episode_number}',
+                                        "episode_title": (
+                                            allanime_episode["notes"]
+                                            or f'{anime["title"]}'
+                                        )
+                                        + f"; Episode {episode_number}",
                                         "links": resp.json()["links"],
                                     }  # pyright:ignore
                                 case "Kir":
@@ -173,8 +176,11 @@ class AllAnimeAPI:
                                     print("[yellow]WeTransfer Fetched")
                                     yield {
                                         "server": "wetransfer",
-                                        "episode_title": allanime_episode["notes"]
-                                        or f'{anime["title"]}: Episode {episode_number}',
+                                        "episode_title": (
+                                            allanime_episode["notes"]
+                                            or f'{anime["title"]}'
+                                        )
+                                        + f"; Episode {episode_number}",
                                         "links": resp.json()["links"],
                                     }  # pyright:ignore
                                 case "S-mp4":
@@ -184,8 +190,11 @@ class AllAnimeAPI:
                                     print("[yellow]Sharepoint Fetched")
                                     yield {
                                         "server": "sharepoint",
-                                        "episode_title": allanime_episode["notes"]
-                                        or f'{anime["title"]}: Episode {episode_number}',
+                                        "episode_title": (
+                                            allanime_episode["notes"]
+                                            or f'{anime["title"]}'
+                                        )
+                                        + f"; Episode {episode_number}",
                                         "links": resp.json()["links"],
                                     }  # pyright:ignore
                                 case "Sak":
@@ -193,8 +202,11 @@ class AllAnimeAPI:
                                     print("[yellow]Dropbox Fetched")
                                     yield {
                                         "server": "dropbox",
-                                        "episode_title": allanime_episode["notes"]
-                                        or f'{anime["title"]}: Episode {episode_number}',
+                                        "episode_title": (
+                                            allanime_episode["notes"]
+                                            or f'{anime["title"]}'
+                                        )
+                                        + f"; Episode {episode_number}",
                                         "links": resp.json()["links"],
                                     }  # pyright:ignore
                     except Timeout: