Merge pull request #138 from meisnate12/develop

v1.6.0

README.md

@@ -1,5 +1,5 @@
 # Plex Meta Manager
-#### Version 1.5.1
+#### Version 1.6.0
 
 The original concept for Plex Meta Manager is [Plex Auto Collections](https://github.com/mza921/Plex-Auto-Collections), but this is rewritten from the ground up to be able to include a scheduler, metadata edits, multiple libraries, and logging. Plex Meta Manager is a Python 3 script that can be continuously run using YAML configuration files to update on a schedule the metadata of the movies, shows, and collections in your libraries as well as automatically build collections based on various methods all detailed in the wiki. Some collection examples that the script can automatically build and update daily include Plex Based Searches like actor, genre, or studio collections or Collections based on TMDb, IMDb, Trakt, TVDb, AniDB, or MyAnimeList lists and various other services.
 
@@ -17,7 +17,8 @@ The script is designed to work with most Metadata agents including the new Plex
 
 ## Support
 
-* If you're getting an Error or have an Enhancement post in the [Issues](https://github.com/meisnate12/Plex-Meta-Manager/issues)
+* Before posting on Github about an enhancement, error, or configuration question please visit the [Plex Meta Manager Discord Server](https://discord.gg/NfH6mGFuAB) 
+* If you're getting an error or have an enhancement post in the [Issues](https://github.com/meisnate12/Plex-Meta-Manager/issues)
 * If you have a configuration question visit the [Discussions](https://github.com/meisnate12/Plex-Meta-Manager/discussions)
 * To see user submitted Metadata configuration files and you could even add your own go to the [Plex Meta Manager Configs](https://github.com/meisnate12/Plex-Meta-Manager-Configs)
 * Pull Request are welcome but please submit them to the develop branch

modules/anilist.py

@@ -9,6 +9,13 @@ class AniListAPI:
     def __init__(self, config):
         self.config = config
         self.url = "https://graphql.anilist.co"
+        self.tags = {}
+        self.genres = {}
+
+        for tag in self.send_request("query{MediaTagCollection {name}}", {})["data"]["MediaTagCollection"]:
+            self.tags[tag["name"].lower()] = tag["name"]
+        for genre in self.send_request("query{GenreCollection}", {})["data"]["GenreCollection"]:
+            self.genres[genre.lower()] = genre
 
     @retry(stop_max_attempt_number=6, wait_fixed=10000)
     def post(self, query, variables):
@@ -56,7 +63,6 @@ def get_pagenation(self, query, limit=0, variables=None):
                         break
             if 0 < limit == count:
                 break
-
         return mal_ids
 
     def top_rated(self, limit):
@@ -93,6 +99,30 @@ def season(self, season, year, sort, limit):
         variables = {"season": season.upper(), "year": year, "sort": "SCORE_DESC" if sort == "score" else "POPULARITY_DESC"}
         return self.get_pagenation(query, limit=limit, variables=variables)
 
+    def genre(self, genre, sort, limit):
+        query = """
+            query ($page: Int, $genre: String, $sort: [MediaSort]) {
+              Page(page: $page){
+                pageInfo {hasNextPage}
+                media(genre: $genre, sort: $sort){idMal}
+              }
+            }
+        """
+        variables = {"genre": genre, "sort": "SCORE_DESC" if sort == "score" else "POPULARITY_DESC"}
+        return self.get_pagenation(query, limit=limit, variables=variables)
+
+    def tag(self, tag, sort, limit):
+        query = """
+            query ($page: Int, $tag: String, $sort: [MediaSort]) {
+              Page(page: $page){
+                pageInfo {hasNextPage}
+                media(tag: $tag, sort: $sort){idMal}
+              }
+            }
+        """
+        variables = {"tag": tag, "sort": "SCORE_DESC" if sort == "score" else "POPULARITY_DESC"}
+        return self.get_pagenation(query, limit=limit, variables=variables)
+
     def studio(self, studio_id):
         query = """
             query ($page: Int, $id: Int) {
@@ -154,6 +184,16 @@ def relations(self, anilist_id, ignore_ids=None):
 
         return mal_ids, ignore_ids, name
 
+    def validate_genre(self, genre):
+        if genre.lower() in self.genres:
+            return self.genres[genre.lower()]
+        raise Failed(f"AniList Error: Genre: {genre} does not exist")
+
+    def validate_tag(self, tag):
+        if tag.lower() in self.tags:
+            return self.tags[tag.lower()]
+        raise Failed(f"AniList Error: Tag: {tag} does not exist")
+
     def validate_anilist_ids(self, anilist_ids, studio=False):
         anilist_values = []
         for anilist_id in anilist_ids:
@@ -183,7 +223,15 @@ def get_items(self, method, data, status_message=True):
         elif method == "anilist_season":
             mal_ids = self.season(data["season"], data["year"], data["sort_by"], data["limit"])
             if status_message:
-                logger.info(f"Processing {pretty}: {data['limit']} Anime from {util.pretty_seasons[data['season']]} {data['year']} sorted by {util.anilist_pretty[data['sort_by']]}")
+                logger.info(f"Processing {pretty}: {data['limit'] if data['limit'] > 0 else 'All'} Anime from {util.pretty_seasons[data['season']]} {data['year']} sorted by {util.anilist_pretty[data['sort_by']]}")
+        elif method == "anilist_genre":
+            mal_ids = self.genre(data["genre"], data["sort_by"], data["limit"])
+            if status_message:
+                logger.info(f"Processing {pretty}: {data['limit'] if data['limit'] > 0 else 'All'} Anime from the Genre: {data['genre']} sorted by {util.anilist_pretty[data['sort_by']]}")
+        elif method == "anilist_tag":
+            mal_ids = self.tag(data["tag"], data["sort_by"], data["limit"])
+            if status_message:
+                logger.info(f"Processing {pretty}: {data['limit'] if data['limit'] > 0 else 'All'} Anime from the Tag: {data['tag']} sorted by {util.anilist_pretty[data['sort_by']]}")
         elif method in ["anilist_studio", "anilist_relations"]:
             if method == "anilist_studio":          mal_ids, name = self.studio(data)
             else:                                   mal_ids, _, name = self.relations(data)