From f38d90cc0ad8d97b4c78e0e5ecbcd38d41033f0e Mon Sep 17 00:00:00 2001 From: wulan17 Date: Mon, 10 Apr 2023 21:37:28 +0700 Subject: [PATCH] PyroFork: Add AddStickerToSet Methods Signed-off-by: wulan17 --- pyrogram/methods/stickers/__init__.py | 2 + .../methods/stickers/add_sticker_to_set.py | 85 +++++++++++++++++++ 2 files changed, 87 insertions(+) create mode 100644 pyrogram/methods/stickers/add_sticker_to_set.py diff --git a/pyrogram/methods/stickers/__init__.py b/pyrogram/methods/stickers/__init__.py index 117d202e..6376f1e8 100644 --- a/pyrogram/methods/stickers/__init__.py +++ b/pyrogram/methods/stickers/__init__.py @@ -16,10 +16,12 @@ # You should have received a copy of the GNU Lesser General Public License # along with Pyrogram. If not, see . +from .add_sticker_to_set import AddStickerToSet from .create_sticker_set import CreateStickerSet from .get_sticker_set import GetStickerSet class Stickers( + AddStickerToSet, CreateStickerSet, GetStickerSet ): diff --git a/pyrogram/methods/stickers/add_sticker_to_set.py b/pyrogram/methods/stickers/add_sticker_to_set.py new file mode 100644 index 00000000..84fec552 --- /dev/null +++ b/pyrogram/methods/stickers/add_sticker_to_set.py @@ -0,0 +1,85 @@ +# Pyrogram - Telegram MTProto API Client Library for Python +# Copyright (C) 2017-present Dan +# +# This file is part of Pyrogram. +# +# Pyrogram is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License as published +# by the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Pyrogram is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public License +# along with Pyrogram. If not, see . + +import os +import re +from typing import Union + +import pyrogram +from pyrogram import raw +from pyrogram import types + +class AddStickerToSet: + async def add_sticker_to_set( + self: "pyrogram.Client", + set_short_name: str, + sticker: str, + emoji: str = "🤔", + ) -> "types.StickerSet": + """Get info about a stickerset. + + .. include:: /_includes/usable-by/bot.rst + + Parameters: + set_short_name (``str``): + Stickerset shortname. + + sticker (``str``): + sticker to add. + Pass a file_id as string to send a file that exists on the Telegram servers. + + emoji (``str``, *optional*): + Associated emoji. + default to "🤔" + + Returns: + :obj:`~pyrogram.types.StickerSet`: On success, the StickerSet information is returned. + + Example: + .. code-block:: python + + await app.add_sticker_to_set("mypack1", "AsJiasp") + """ + file = None + + if isinstance(sticker, str): + if os.path.isfile(sticker) or re.match("^https?://", sticker): + raise ValueError(f"file_id is invalid!") + else: + decoded = FileId.decode(sticker) + media = raw.types.InputDocument( + id=decoded.media_id, + access_hash=decoded.access_hash, + file_reference=decoded.file_reference + ) + else: + raise ValueError(f"file_id is invalid!") + + r = await self.invoke( + raw.functions.stickers.AddStickerToSet( + stickerset=raw.types.InputStickerSetShortName(short_name=set_short_name), + sticker=[ + raw.types.InputStickerSetItem( + document=media, + emoji=emoji + ) + ] + ) + ) + + return types.StickerSet._parse(r.set)