From 19a5185a0a25c2526e6ec1e7dfe5bf72ab667665 Mon Sep 17 00:00:00 2001
From: Benex254 <benextempest@gmail.com>
Date: Mon, 5 Aug 2024 09:47:03 +0300
Subject: [PATCH] feat(allanime): change typing from generator to iterator

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

diff --git a/fastanime/libs/anime_provider/allanime/api.py b/fastanime/libs/anime_provider/allanime/api.py
index 59907d2..cf38082 100644
--- a/fastanime/libs/anime_provider/allanime/api.py
+++ b/fastanime/libs/anime_provider/allanime/api.py
@@ -1,6 +1,6 @@
 import json
 import logging
-from typing import Generator
+from typing import Iterator
 
 import requests
 from requests.exceptions import Timeout
@@ -55,7 +55,12 @@ class AllAnimeAPI:
             return {}
 
     def search_for_anime(
-        self, user_query: str, translation_type: str = "sub", nsfw=True, unknown=True
+        self,
+        user_query: str,
+        translation_type: str = "sub",
+        nsfw=True,
+        unknown=True,
+        **kwargs,
     ):
         search = {"allowAdult": nsfw, "allowUnknown": unknown, "query": user_query}
         limit = 40
@@ -106,14 +111,7 @@ class AllAnimeAPI:
 
     def get_episode_streams(
         self, anime: Anime, episode_number: str, translation_type="sub"
-    ) -> (
-        Generator[
-            Server,
-            Server,
-            Server,
-        ]
-        | None
-    ):
+    ) -> Iterator[Server] | None:
         anime_id = anime["id"]
         allanime_episode = self.get_anime_episode(
             anime_id, episode_number, translation_type