sticker 增加错误提示

This commit is contained in:
xtaodada 2022-12-05 21:49:48 +08:00
parent 51d12fc23c
commit 8a6178ad85
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659

View File

@ -3,6 +3,7 @@ import contextlib
from asyncio import sleep from asyncio import sleep
from typing import Optional from typing import Optional
from pyrogram.errors import PeerIdInvalid
from pyrogram.raw.functions.messages import GetStickerSet from pyrogram.raw.functions.messages import GetStickerSet
from pyrogram.raw.functions.stickers import CreateStickerSet from pyrogram.raw.functions.stickers import CreateStickerSet
from pyrogram.raw.types import InputStickerSetShortName, InputDocument, InputStickerSetItem from pyrogram.raw.types import InputStickerSetShortName, InputDocument, InputStickerSetItem
@ -117,7 +118,7 @@ class Sticker:
self.sticker_set += "_animated" self.sticker_set += "_animated"
try: try:
await self.check_pack_full() await self.check_pack_full()
except NoStickerSetNameError as e: except NoStickerSetNameError:
self.should_create = True self.should_create = True
except StickerSetFullError: except StickerSetFullError:
await self.generate_sticker_set(time + 1) await self.generate_sticker_set(time + 1)
@ -275,6 +276,8 @@ async def sticker(message: Message):
try: try:
await one_sticker.process_sticker() await one_sticker.process_sticker()
await one_sticker.to_sticker_set() await one_sticker.to_sticker_set()
except PeerIdInvalid:
return await message.edit("请先私聊一次 @Stickers 机器人")
except Exception as e: except Exception as e:
return await message.edit(f"收藏到贴纸包失败:{e}") return await message.edit(f"收藏到贴纸包失败:{e}")
await message.edit(f"收藏到贴纸包 {one_sticker.mention()} 成功") await message.edit(f"收藏到贴纸包 {one_sticker.mention()} 成功")