get_discord_emoji.py: also get banner and icon
This commit is contained in:
parent
5d7349062d
commit
73d0a52df4
|
@ -16,10 +16,13 @@ def download_file(url, path):
|
||||||
print("Already downloaded, skipping...")
|
print("Already downloaded, skipping...")
|
||||||
return
|
return
|
||||||
out_file = open(path, "wb")
|
out_file = open(path, "wb")
|
||||||
req = urllib.request.urlopen(
|
try:
|
||||||
urllib.request.Request(url, headers={"User-Agent": "curl/8.6.0"})
|
req = urllib.request.urlopen(
|
||||||
)
|
urllib.request.Request(url, headers={"User-Agent": "curl/8.6.0"})
|
||||||
shutil.copyfileobj(req, out_file)
|
)
|
||||||
|
shutil.copyfileobj(req, out_file)
|
||||||
|
except urllib.error.HTTPError:
|
||||||
|
print(f"Failed to download {url}")
|
||||||
time.sleep(0.1)
|
time.sleep(0.1)
|
||||||
|
|
||||||
|
|
||||||
|
@ -65,8 +68,32 @@ def main(*args):
|
||||||
emoji_objs = guild["emojis"]
|
emoji_objs = guild["emojis"]
|
||||||
|
|
||||||
dir_name = safe_name(name)
|
dir_name = safe_name(name)
|
||||||
sticker_dir = base_dir / "stickers" / dir_name
|
guild_dir = base_dir / dir_name
|
||||||
emoji_dir = base_dir / "emoji" / dir_name
|
sticker_dir = guild_dir / "stickers"
|
||||||
|
emoji_dir = guild_dir / "emoji"
|
||||||
|
|
||||||
|
guild_dir.mkdir(exist_ok=True, parents=True)
|
||||||
|
|
||||||
|
properties = guild["properties"]
|
||||||
|
guild_id = properties["id"]
|
||||||
|
subset_properties = {
|
||||||
|
"id": properties["id"],
|
||||||
|
"name": properties["name"],
|
||||||
|
"description": properties["description"],
|
||||||
|
}
|
||||||
|
with open(guild_dir / "properties.json", "w") as properties_file:
|
||||||
|
json.dump(subset_properties, properties_file)
|
||||||
|
|
||||||
|
if properties["icon"] is not None:
|
||||||
|
download_file(
|
||||||
|
f"https://cdn.discordapp.com/icons/{guild_id}/{properties['icon']}.png?size=4096",
|
||||||
|
guild_dir / "icon.png",
|
||||||
|
)
|
||||||
|
if properties["banner"] is not None:
|
||||||
|
download_file(
|
||||||
|
f"https://cdn.discordapp.com/banners/{guild_id}/{properties['banner']}.png?size=4096",
|
||||||
|
guild_dir / "banner.png",
|
||||||
|
)
|
||||||
|
|
||||||
if len(sticker_objs) > 0:
|
if len(sticker_objs) > 0:
|
||||||
sticker_dir.mkdir(exist_ok=True, parents=True)
|
sticker_dir.mkdir(exist_ok=True, parents=True)
|
||||||
|
|
Loading…
Reference in a new issue