Multiple search result support.

scrobble-moe · Aug 16, 2019 · f17222b · f17222b
1 parent 659cc71
commit f17222b
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 17 deletions.
diff --git a/Contents/Code/anilist.py b/Contents/Code/anilist.py
@@ -86,6 +86,7 @@ def get_anime(id):
         return request.content
     except:
         Log.Error('Error getting anime info')
+    return
 
 def get_anime_kitsu(id):
     headers = {
@@ -135,4 +136,4 @@ def get_episodes_kitsu(id):
         return anime_episodes
     except:
         Log.Error('Error anime episodes info')
-    return    
+    return
diff --git a/Contents/Code/search.py b/Contents/Code/search.py
@@ -1,18 +1,21 @@
 def search_anime(type, results, media, lang):
     query = media.show if type == 'tv' else media.name
     query = String.Quote(query)
-    # if media.year is not None:
-    #     query += ' (' + media.year + ')'
 
     query = '''
     query {
-        Media (search: "''' + query + '''", type: ANIME) {
-            id
-            title {
-                romaji
+        anime: Page (perPage: 8) {
+            pageInfo {
+                total
             }
-            startDate {
-                year
+            results: media (type: ANIME, search: "''' + query + '''") {
+                id
+                title {
+                    romaji
+                }
+                startDate {
+                    year
+                }
             }
         }
     }
@@ -30,12 +33,16 @@ def search_anime(type, results, media, lang):
     except:
         Log.Error('Error searching AniList - Anime: ' + query)
         return
-    result = JSON.ObjectFromString(request.content)
 
-    results.Append(MetadataSearchResult(
-        id = str(result['data']['Media']['id']),
-        name = result['data']['Media']['title']['romaji'],
-        year = result['data']['Media']['startDate']['year'],
-        score = 100,
-        lang = lang
-    ))
+    s = 100
+    for result in JSON.ObjectFromString(request.content)['data']['anime']['results']:
+        results.Append(MetadataSearchResult(
+            id = str(result['id']),
+            name = result['title']['romaji'],
+            year = result['startDate']['year'],
+            score = s,
+            lang = lang
+        ))
+        s = s - 1
+    return
+