Upload files to 'anime_scripts'

anime_scripts/seasons.py

@@ -0,0 +1,115 @@
+from requests import post
+from sys import argv
+
+BASE_URL = "https://graphql.anilist.co"
+
+class SortableAnime:
+    def __init__(self, id, year, reltype, title, frmt):
+        self.id = id
+        self.year = year if year else 9999
+        self.type = reltype
+        self.title = title
+        self.frmt = frmt
+
+    def __eq__(self, other):
+        return self.id == other.id
+
+    def __hash__(self):
+        return self.id
+    
+    def __str__(self):
+        return str(self.title)
+
+
+def search(query):
+    response = post(BASE_URL, json={
+        'query': """
+            query ($q: String) {
+                Media (search: $q) {
+                    id
+                }
+            }
+        """,
+        'variables': {"q": query}
+    })
+    if response.ok:
+        return response.json()["data"]["Media"]["id"]
+    raise ValueError("Show does not exist")
+
+def get_show(id):
+    response = post(BASE_URL, json={
+        'query': """
+            query ($id: Int) {
+                Media (id: $id) {
+                    id
+                    title {
+                        english
+                        romaji
+                    }
+                    startDate {
+                        year
+                    }
+                    format
+                    relations {
+                        nodes {
+                            id
+                            format
+                            startDate {
+                                year
+                            }
+                            title {
+                                english
+                                romaji
+                            }
+                        }
+                        edges{
+                            relationType
+                        }
+                    }
+                }
+            }
+        """,
+        'variables': {"id": id}
+    })
+    if response.ok:
+        return response.json()
+    raise ValueError("Bad show")
+
+items = []
+
+def get_base_show(res):
+    base = res["data"]["Media"]
+    return SortableAnime(base["id"], base["startDate"]["year"], "BASE", base["title"], base["format"])
+
+def process_shows(res):
+    ls = []
+    ls.append(get_base_show(res))
+    for i,v in enumerate(res["data"]["Media"]["relations"]["nodes"]):
+        ls.append(SortableAnime(v["id"], v["startDate"]["year"], res["data"]["Media"]["relations"]["edges"][i]["relationType"], v["title"], v["format"]))
+        pass
+    return ls
+
+show_id = search(argv[1])
+res = get_show(show_id)
+items.extend(process_shows(res))
+base_show = get_base_show(res)
+
+if "PREQUEL" not in [i.type for i in items]:
+    season = 1
+else:
+    ignore = []
+    while True:
+        f = False
+        for i in items:
+            if i.type == "PREQUEL" and i not in ignore:
+                fi = [i for i in process_shows(get_show(i.id)) if i not in items]
+                items.extend(fi)
+                ignore.append(i)
+                f = True
+        if not f:
+            break
+    items = [i for i in items if i.frmt == "TV"]
+    items.sort(key=lambda i: i.year)
+    season = items.index(base_show)+1
+
+print("Season {}".format(season))