sticker_transfer 支持清空已安装的贴纸包

list.json

@@ -262,10 +262,10 @@
         },
         {
             "name": "sticker_transfer",
-            "version": "1.01",
+            "version": "1.1",
             "section": "chat",
             "maintainer": "xtaodada",
-            "size": "3.5 kb",
+            "size": "4.9 kb",
             "supported": true,
             "des-short": "贴纸迁移",
             "des": "贴纸迁移。\n指令：,sticker_transfer"

sticker_transfer/main.py

@@ -6,7 +6,7 @@ from asyncio import sleep
 from os import sep
 
 from pyrogram.errors import StickersetInvalid, FloodWait
-from pyrogram.raw.functions.messages import GetAllStickers
+from pyrogram.raw.functions.messages import GetAllStickers, UninstallStickerSet
 from pyrogram.raw.functions.messages import InstallStickerSet
 from pyrogram.raw.types import InputStickerSetShortName
 
@@ -15,10 +15,14 @@ from pagermaid.listener import listener
 from pagermaid.single_utils import Message, safe_remove
 
 
+class NoStickerSetError(Exception):
+    pass
+
+
 async def export_sticker_to_csv():
     stickers = await bot.invoke(GetAllStickers(hash=0))
     if not stickers.sets:
-        return False
+        raise NoStickerSetError
     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"])
@@ -38,16 +42,25 @@ async def import_sticker(short_name):
     ))
 
 
+async def remove_sticker(short_name):
+    await bot.invoke(UninstallStickerSet(
+        stickerset=InputStickerSetShortName(short_name=short_name),
+    ))
+
+
 async def import_sticker_from_csv(file_name):
     success = 0
     failed = 0
+    need_import = []
     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
+            need_import.insert(0, row[1])
+    for i in need_import:
         try:
-                await import_sticker(row[1])
+            await import_sticker(i)
             success += 1
         except StickersetInvalid:
             failed += 1
@@ -55,16 +68,40 @@ async def import_sticker_from_csv(file_name):
             await sleep(e.value)
             await import_sticker(row[1])
             success += 1
+        except Exception:
+            failed += 1
+    return success, failed
+
+
+async def clear_sets():
+    stickers = await bot.invoke(GetAllStickers(hash=0))
+    if not stickers.sets:
+        raise NoStickerSetError
+    success = 0
+    failed = 0
+    for sticker_set in stickers.sets:
+        try:
+            await remove_sticker(sticker_set.short_name)
+            success += 1
+        except FloodWait as e:
+            await sleep(e.value)
+            await remove_sticker(sticker_set.short_name)
+            failed += 1
+        except Exception:
+            failed += 1
     return success, failed
 
 
 @listener(command="sticker_transfer",
           need_admin=True,
-          parameters="导出/导入",
-          description="导出、导入已安装的贴纸包")
+          parameters="导出/导入/清空",
+          description="导出、导入、清空已安装的贴纸包")
 async def sticker_transfer(message: Message):
     if message.arguments == "导出":
+        try:
             num = await export_sticker_to_csv()
+        except NoStickerSetError:
+            return await message.edit("没有贴纸包可以导出")
         if num:
             await bot.send_document(message.chat.id,
                                     "stickers.csv",
@@ -87,5 +124,12 @@ async def sticker_transfer(message: Message):
         success, failed = await import_sticker_from_csv(file_name)
         safe_remove(file_name)
         await message.edit(f"导入成功 {success} 个贴纸包，失败 {failed} 个贴纸包")
+    elif message.arguments == "清空":
+        message = await message.edit("清空中")
+        try:
+            success, failed = await clear_sets()
+        except NoStickerSetError:
+            return await message.edit("没有贴纸包可以清空")
+        await message.edit(f"清空成功 {success} 个贴纸包，失败 {failed} 个贴纸包")
     else:
-        await message.edit("❌ 参数错误，请选择 `导出` 或 `导入` ")
+        await message.edit("❌ 参数错误，请选择 `导出` 或 `导入` 或 `清空`")