add get_discord_emoji script because that's fun

scripts/get_discord_emoji.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env nix-shell
+#!nix-shell -i python3 -p python3
+import json
+import zlib
+import sys
+import pathlib
+import time
+import urllib.request
+import shutil
+
+# PNG, APNG, Lottie, GIF
+STICKER_EXTENSIONS = ["", "png", "png", "json", "gif"]
+
+
+def download_file(url, path):
+    if path.exists():
+        print("Already downloaded, skipping...")
+        return
+    out_file = open(path, "wb")
+    req = urllib.request.urlopen(
+        urllib.request.Request(url, headers={"User-Agent": "curl/8.6.0"})
+    )
+    shutil.copyfileobj(req, out_file)
+    time.sleep(0.1)
+
+
+def get_data(filename):
+    bindata = open(filename, "rb").read()
+    decompress = zlib.decompressobj()
+    jsondata = decompress.decompress(bindata)
+    while len(jsondata) > 0:
+        try:
+            obj = json.loads(jsondata)
+            if obj.get("t", "") == "READY":
+                return obj
+        except json.JSONDecodeError as jde:
+            obj = json.loads(jsondata[0 : jde.pos])
+            if obj.get("t", "") == "READY":
+                return obj
+            jsondata = jsondata[jde.pos :]
+
+
+def main(*args):
+    data = get_data(args[1])
+
+    base_dir = pathlib.Path("guilds")
+
+    for guild in data["d"]["guilds"]:
+        name = guild["properties"]["name"]
+        print(f"Processing guild '{name}'")
+
+        guild_dir = base_dir / name.replace("/", " ")
+        sticker_dir = guild_dir / "stickers"
+        sticker_dir.mkdir(parents=True, exist_ok=True)
+        emoji_dir = guild_dir / "emoji"
+        emoji_dir.mkdir(exist_ok=True)
+
+        for sticker in guild["stickers"]:
+            break
+            print(f"Found sticker '{sticker['name']}'")
+            extension = STICKER_EXTENSIONS[sticker["format_type"]]
+            url = f"https://media.discordapp.net/stickers/{sticker['id']}.{extension}?size=4096"
+            # Set passthrough=false for APNGs if you don't want animated
+            filename = sticker_dir / f"{sticker['name']}.{extension}"
+            print(f"Downloading {url} to {filename}")
+            download_file(url, filename)
+
+        for emoji in guild["emojis"]:
+            print(f"Found emoji '{emoji['name']}'")
+            extension = "gif" if emoji["animated"] else "png"
+            url = f"https://cdn.discordapp.com/emojis/{emoji['id']}.{extension}"
+            filename = emoji_dir / f"{emoji['name']}.{extension}"
+            print(f"Downloading {url} to {filename}")
+            download_file(url, filename)
+
+
+if __name__ == "__main__":
+    main(*sys.argv)