mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-22 08:27:39 +00:00
110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
from typing import List
|
|
|
|
from pyrogram.raw.functions.messages import GetStickerSet
|
|
from pyrogram.raw.functions.stickers import CreateStickerSet
|
|
from pyrogram.raw.types import (
|
|
InputStickerSetShortName,
|
|
InputStickerSetItem,
|
|
InputDocument,
|
|
)
|
|
from pyrogram.raw.types.messages import StickerSet
|
|
|
|
from pagermaid.listener import listener
|
|
from pagermaid.services import bot
|
|
from pagermaid.enums import Message
|
|
|
|
|
|
class NoStickerSetNameError(Exception):
|
|
"""
|
|
Occurs when no username is provided
|
|
"""
|
|
|
|
def __init__(self, string: str = "贴纸包不存在"):
|
|
super().__init__(string)
|
|
|
|
|
|
async def get_pack(name: str):
|
|
try:
|
|
return await bot.invoke(
|
|
GetStickerSet(stickerset=InputStickerSetShortName(short_name=name), hash=0)
|
|
)
|
|
except Exception as e: # noqa
|
|
raise NoStickerSetNameError() from e
|
|
|
|
|
|
async def create_sticker_set(
|
|
sticker_set: str, title: str, is_animated: bool, is_video: bool, stickers
|
|
):
|
|
try:
|
|
await bot.invoke(
|
|
CreateStickerSet(
|
|
user_id=await bot.resolve_peer((await bot.get_me()).id),
|
|
title=title,
|
|
short_name=sticker_set,
|
|
stickers=stickers,
|
|
animated=is_animated,
|
|
videos=is_video,
|
|
software="pagermaid-pyro",
|
|
)
|
|
)
|
|
except Exception as e:
|
|
raise NoStickerSetNameError("贴纸包名称或者链接非法或者已被占用，请换一个") from e
|
|
|
|
|
|
async def process_old_sticker_set(sticker_sets: List[str]):
|
|
is_animated = False
|
|
is_video = False
|
|
stickers = []
|
|
for idx, sticker_set in enumerate(sticker_sets):
|
|
pack: StickerSet = await get_pack(sticker_set)
|
|
if idx == 0:
|
|
is_animated = pack.set.animated
|
|
is_video = pack.set.videos
|
|
else:
|
|
if pack.set.animated != is_animated:
|
|
raise NoStickerSetNameError("贴纸包类型不一致")
|
|
if pack.set.videos != is_video:
|
|
raise NoStickerSetNameError("贴纸包类型不一致")
|
|
hash_map = {}
|
|
for i in pack.packs:
|
|
for j in i.documents:
|
|
hash_map[j] = i.emoticon
|
|
_stickers = [
|
|
InputStickerSetItem(
|
|
document=InputDocument(
|
|
id=i.id,
|
|
access_hash=i.access_hash,
|
|
file_reference=i.file_reference,
|
|
),
|
|
emoji=hash_map.get(i.id, "👀"),
|
|
)
|
|
for i in pack.documents
|
|
]
|
|
if len(stickers) + len(_stickers) > 120:
|
|
raise NoStickerSetNameError("贴纸包过多")
|
|
stickers.extend(_stickers)
|
|
return stickers, is_animated, is_video
|
|
|
|
|
|
@listener(
|
|
command="copy_sticker_set",
|
|
parameters="旧的贴纸包用户名1,旧的贴纸包用户名2 贴纸包用户名 贴纸包名称",
|
|
description="复制某个贴纸包",
|
|
)
|
|
async def copy_sticker_set(message: Message):
|
|
if len(message.parameter) < 3:
|
|
return await message.edit(
|
|
"请指定贴纸包链接和贴纸包名称，例如 <code>xxx xxxx_sticker xxxx 的贴纸包</code>"
|
|
)
|
|
old_set_names = message.parameter[0].split(",")
|
|
set_name = message.parameter[1]
|
|
name = " ".join(message.parameter[2:])
|
|
try:
|
|
stickers, is_animated, is_video = await process_old_sticker_set(old_set_names)
|
|
await create_sticker_set(set_name, name, is_animated, is_video, stickers)
|
|
except Exception as e:
|
|
return await message.edit(f"复制贴纸包失败：{e}")
|
|
await message.edit(
|
|
f'复制贴纸包成功 <a href="https://t.me/addstickers/{set_name}">{name}</a>'
|
|
)
|