Use list comprehension rather explicitly mention countries on AniBrain

nattadasu · Dec 4, 2023 · 715ef35 · 715ef35
1 parent 55d525c
commit 715ef35
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/extensions/anime.py b/extensions/anime.py
@@ -257,12 +257,8 @@ async def anime_random(
         )
         try:
             async with AniBrainAI() as ai:
-                countries = [
-                    ai.CountryOfOrigin.CHINA,
-                    ai.CountryOfOrigin.JAPAN,
-                    ai.CountryOfOrigin.KOREA,
-                    ai.CountryOfOrigin.TAIWAN,
-                ] if country == "any" else [ai.CountryOfOrigin(country)]
+                countries = [i for i in ai.CountryOfOrigin] if country == "any" else [
+                    ai.CountryOfOrigin(country)]
                 if media_type != "any":
                     target_media_type = [ai.AnimeMediaType(media_type.lower())]
                 else:

diff --git a/extensions/manga.py b/extensions/manga.py
@@ -261,10 +261,7 @@ async def random_manga(
         try:
             async with AniBrainAI() as anibrain:
                 countries = [
-                    anibrain.CountryOfOrigin.JAPAN,
-                    anibrain.CountryOfOrigin.KOREA,
-                    anibrain.CountryOfOrigin.CHINA,
-                    anibrain.CountryOfOrigin.TAIWAN,
+                    i for i in anibrain.CountryOfOrigin
                 ] if country == "any" else [anibrain.CountryOfOrigin(country)]
                 match media_type:
                     case "manga":