79 lines
2.4 KiB
Python
Executable file
79 lines
2.4 KiB
Python
Executable file
#!/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)
|