wah.su/stickerbridge
1
0
Fork 0
mirror of https://github.com/wah-su/stickerbridge.git synced 2025-04-05 15:54:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: allow uploading sticker pack to default room set

Browse source 
fix: cannot open pack if imported by url

fix: File Access Error on windows
This commit is contained in:
Kentai Radiquum 2024-12-23 16:36:51 +05:00
parent b9461d75be
commit ab6da62e15
Signed by: Radiquum
GPG key ID: 858E8EE696525EED
2 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
stickerbridge/chat_functions.py
View file

@ -49,8 +49,8 @@ async def is_stickerpack_existing(client: AsyncClient, room_id: str, pack_name:
return not response.content == {}
async def upload_stickerpack(client: AsyncClient, room_id: str, stickerset: MatrixStickerset):
return await client.room_put_state(room_id, 'im.ponies.room_emotes', stickerset.json(), state_key=stickerset.name())
async def upload_stickerpack(client: AsyncClient, room_id: str, stickerset: MatrixStickerset, name):
return await client.room_put_state(room_id, 'im.ponies.room_emotes', stickerset.json(), state_key=name)
async def upload_image(client: AsyncClient, image: str):

16
stickerbridge/matrix_reuploader.py
View file

@ -1,4 +1,5 @@
import tempfile
import os
from nio import MatrixRoom, AsyncClient
@ -46,9 +47,13 @@ class MatrixReuploader:
converted_stickerset = await self.exporter.get_stickerset(pack_name)
yield self.STATUS_UPLOADING
for sticker in converted_stickerset:
with tempfile.NamedTemporaryFile('w+b') as file:
with tempfile.NamedTemporaryFile('w+b', delete=False) as file:
file.write(sticker.image_data)
sticker_mxc = await upload_image(self.client, file.name)
file.close()
os.unlink(file.name)
stickerset.add_sticker(sticker_mxc, sticker.alt_text)
if not stickerset.count():
@ -56,6 +61,13 @@ class MatrixReuploader:
return
yield self.STATUS_UPDATING_ROOM_STATE
await upload_stickerpack(self.client, self.room.room_id, stickerset)
name = stickerset.name()
if import_name.lower() == "default":
name = ""
elif import_name.startswith("http"):
name = import_name.split("/")[-1]
await upload_stickerpack(self.client, self.room.room_id, stickerset, name)
yield self.STATUS_OK