From 5bb3fa9240dbcafdea78fbd501f24801ea90c144 Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Sun, 3 Jul 2022 13:50:11 +0800
Subject: [PATCH] =?UTF-8?q?sticker=5Ftransfer=20=E8=B4=B4=E7=BA=B8?=
 =?UTF-8?q?=E8=BF=81=E7=A7=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 list.json                | 10 +++++
 sticker_transfer/main.py | 84 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 94 insertions(+)
 create mode 100644 sticker_transfer/main.py

diff --git a/list.json b/list.json
index fad0660..3e7cd3f 100644
--- a/list.json
+++ b/list.json
@@ -259,6 +259,16 @@
             "supported": true,
             "des-short": "历史上的今天",
             "des": "历史上的今天。\n指令：,today_in_history"
+        },
+        {
+            "name": "sticker_transfer",
+            "version": "1.0",
+            "section": "chat",
+            "maintainer": "xtaodada",
+            "size": "3.24 kb",
+            "supported": true,
+            "des-short": "贴纸迁移",
+            "des": "贴纸迁移。\n指令：,sticker_transfer"
         }
     ]
 }
diff --git a/sticker_transfer/main.py b/sticker_transfer/main.py
new file mode 100644
index 0000000..f52e4d6
--- /dev/null
+++ b/sticker_transfer/main.py
@@ -0,0 +1,84 @@
+""" PagerMaid module that 贴纸迁移 """
+
+import csv
+
+from asyncio import sleep
+
+from pyrogram.errors import StickersetInvalid, FloodWait
+from pyrogram.raw.functions.messages import GetAllStickers
+from pyrogram.raw.functions.messages import InstallStickerSet
+from pyrogram.raw.types import InputStickerSetShortName
+
+from pagermaid import bot
+from pagermaid.listener import listener
+from pagermaid.single_utils import Message, safe_remove
+
+
+async def export_sticker_to_csv():
+    stickers = await bot.invoke(GetAllStickers(hash=0))
+    if not stickers.sets:
+        return False
+    with open("stickers.csv", "w", newline="", encoding="utf-8-sig") as f:
+        writer = csv.writer(f)
+        writer.writerow(["name", "short_name", "is_masks", "is_animated", "is_video"])
+        for sticker_set in stickers.sets:
+            writer.writerow([sticker_set.title,
+                             sticker_set.short_name,
+                             sticker_set.archived if hasattr(sticker_set, "archived") else False,
+                             sticker_set.animated if hasattr(sticker_set, "animated") else False,
+                             sticker_set.videos if hasattr(sticker_set, "videos") else False, ])
+    return True
+
+
+async def import_sticker(short_name):
+    await bot.invoke(InstallStickerSet(
+        stickerset=InputStickerSetShortName(short_name=short_name),
+        archived=False,
+    ))
+
+
+async def import_sticker_from_csv(file_name):
+    success = 0
+    failed = 0
+    with open(file_name, "r", newline="", encoding="utf-8-sig") as f:
+        reader = csv.reader(f)
+        for row in reader:
+            if row[0] == "name":
+                continue
+            try:
+                await import_sticker(row[1])
+                success += 1
+            except StickersetInvalid:
+                failed += 1
+            except FloodWait as e:
+                await sleep(e.value)
+                await import_sticker(row[1])
+                success += 1
+    return success, failed
+
+
+@listener(command="sticker_transfer",
+          need_admin=True,
+          parameters="导出/导入",
+          description="导出、导入已安装的贴纸")
+async def sticker_transfer(message: Message):
+    if message.arguments == "导出":
+        await export_sticker_to_csv()
+        await bot.send_document(message.chat.id, "stickers.csv", caption="贴纸导出文件")
+        safe_remove("stickers.csv")
+        await message.safe_delete()
+    elif message.arguments == "导入":
+        reply = message.reply_to_message
+        if not reply:
+            return await message.edit("❌ 请回复贴纸导出文件")
+        if not reply.document:
+            return await message.edit("❌ 请回复贴纸导出文件")
+        if not reply.document.file_name.endswith(".csv"):
+            return await message.edit("❌ 请回复贴纸导出文件")
+        message = await message.edit("导入中")
+        file_name = await reply.download()
+        success, failed = await import_sticker_from_csv(file_name)
+        safe_remove(file_name)
+        await message.edit(f"导入成功 {success} 个贴纸，失败 {failed} 个贴纸")
+    else:
+        await message.edit("❌ 参数错误，请选择 `导出` 或 `导入` ")