Browse Source

Add namespacing for IDs

pull/26/head
Tulir Asokan 4 years ago
parent
commit
314f6c0ff5
  1. 10
      import.py

10
import.py

@ -142,7 +142,9 @@ def add_to_index(name: str) -> None:
with open(index_path) as index_file: with open(index_path) as index_file:
index_data = json.load(index_file) index_data = json.load(index_file)
except (FileNotFoundError, json.JSONDecodeError): except (FileNotFoundError, json.JSONDecodeError):
index_data = {"packs": [], "homeserver_url": homeserver_url}
index_data = {"packs": []}
if "homeserver_url" not in index_data:
index_data["homeserver_url"] = homeserver_url
if name not in index_data["packs"]: if name not in index_data["packs"]:
index_data["packs"].append(name) index_data["packs"].append(name)
with open(index_path, "w") as index_file: with open(index_path, "w") as index_file:
@ -154,7 +156,7 @@ def add_meta(document: Document, info: 'MatrixStickerInfo', pack: StickerSetFull
for attr in document.attributes: for attr in document.attributes:
if isinstance(attr, DocumentAttributeSticker): if isinstance(attr, DocumentAttributeSticker):
info["body"] = attr.alt info["body"] = attr.alt
info["id"] = str(document.id)
info["id"] = f"tg-{document.id}"
info["net.maunium.telegram.sticker"] = { info["net.maunium.telegram.sticker"] = {
"pack": { "pack": {
"id": str(pack.set.id), "id": str(pack.set.id),
@ -212,9 +214,11 @@ async def reupload_pack(client: TelegramClient, pack: StickerSetFull) -> None:
with open(pack_path, "w") as pack_file: with open(pack_path, "w") as pack_file:
json.dump({ json.dump({
"title": pack.set.title, "title": pack.set.title,
"id": f"tg-{pack.set.id}",
"net.maunium.telegram.pack": {
"short_name": pack.set.short_name, "short_name": pack.set.short_name,
"id": str(pack.set.id),
"hash": str(pack.set.hash), "hash": str(pack.set.hash),
},
"stickers": list(reuploaded_documents.values()), "stickers": list(reuploaded_documents.values()),
}, pack_file, ensure_ascii=False) }, pack_file, ensure_ascii=False)
print(f"Saved {pack.set.title} as {pack.set.short_name}.json") print(f"Saved {pack.set.title} as {pack.set.short_name}.json")

Loading…
Cancel
Save