fetch_anilist_ids.py

#!/usr/bin/env python3

import requests
import time
import fileinput


def fetch(id: int):
    time.sleep(1)
    query = (
        """query($id: Int, $type: MediaType){Media(idMal: $id, type: $type){siteUrl}}"""
    )

    variables = {"id": id, "type": "ANIME"}

    url = "https://graphql.anilist.co"

    response = requests.post(url, json={"query": query, "variables": variables})
    data = response.json().get("data", {})
    if not data:
        return None

    return data.get("Media", {}).get("siteUrl", "")


def main() -> None:
    # pipe mal ids to STDIN/pass in a file as the first argument to this script
    # prints anilist URLs if they exist
    for mal_id in fileinput.input():
        line = int(mal_id)
        resp = fetch(line)
        print(f"{line} -> {resp}")


if __name__ == "__main__":
    main()