yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2025-12-29 12:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pyrofork: refactor add_sticker_to_set and create_sticker_set

Browse source 
Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
wulan17 2024-08-25 15:39:16 +07:00
parent 8e95ffcd6c
commit 7a063e4a72
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
2 changed files with 68 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
pyrogram/methods/stickers/add_sticker_to_set.py
View file

@ -22,13 +22,17 @@ import re
import pyrogram
from pyrogram import raw
from pyrogram import types
from pyrogram.file_id import FileId
from pyrogram import utils
from pyrogram.file_id import FileId, FileType
from typing import Union
class AddStickerToSet:
async def add_sticker_to_set(
self: "pyrogram.Client",
set_short_name: str,
sticker: str,
user_id: Union[int, str] = None,
emoji: str = "🤔",
) -> "types.StickerSet":
"""Add a sticker to stickerset.
@ -41,7 +45,16 @@ class AddStickerToSet:
sticker (``str``):
sticker to add.
Pass a file_id as string to send a file that exists on the Telegram servers.
Pass a file_id as string to add a sticker that exists on the Telegram servers,
pass an HTTP URL as a string for Telegram to get a sticker from the Internet,
pass a file path as string to upload a new sticker that exists on your local machine, or
pass a binary file-like object with its attribute ".name" set for in-memory uploads.
user_id (``int`` | ``str``, *optional*):
Unique identifier (int) or username (str) of the Stickerset owner.
For you yourself you can simply use "me" or "self" (users only).
required for bots.
default to "me".
emoji (``str``, *optional*):
Associated emoji.
@ -55,11 +68,20 @@ class AddStickerToSet:
await app.add_sticker_to_set("mypack1", "AsJiasp")
"""
file = None
if isinstance(sticker, str):
if self.me.is_bot and user_id is None:
raise ValueError("user_id is required for bots")
if os.path.isfile(sticker) or re.match("^https?://", sticker):
raise ValueError(f"file_id is invalid!")
document = await self.send_document(
user_id or "me",
sticker,
force_document=True,
disable_notification=True
)
uploaded_media = utils.get_input_media_from_file_id(document.document.file_id, FileType.DOCUMENT)
media = uploaded_media.id
_ = await document.delete()
else:
decoded = FileId.decode(sticker)
media = raw.types.InputDocument(
@ -68,7 +90,17 @@ class AddStickerToSet:
file_reference=decoded.file_reference
)
else:
raise ValueError(f"file_id is invalid!")
if self.me.is_bot and user_id is None:
raise ValueError("user_id is required for bots")
document = await self.send_document(
user_id or "me",
sticker,
force_document=True,
disable_notification=True
)
uploaded_media = utils.get_input_media_from_file_id(document.document.file_id, FileType.DOCUMENT)
media = uploaded_media.id
_ = await document.delete()
r = await self.invoke(
raw.functions.stickers.AddStickerToSet(

42
pyrogram/methods/stickers/create_sticker_set.py
View file

@ -23,16 +23,17 @@ from typing import Union, Optional
import pyrogram
from pyrogram import raw
from pyrogram import types
from pyrogram.file_id import FileId
from pyrogram import utils
from pyrogram.file_id import FileId, FileType
class CreateStickerSet:
async def create_sticker_set(
self: "pyrogram.Client",
user_id: Union[int, str],
title: str,
short_name: str,
sticker: str,
user_id: Union[int, str] = None,
emoji: str = "🤔",
masks: bool = None
) -> Optional["types.Message"]:
@ -41,10 +42,6 @@ class CreateStickerSet:
.. include:: /_includes/usable-by/users-bots.rst
Parameters:
user_id (``int`` | ``str``):
Unique identifier (int) or username (str) of the Stickerset owner.
For you yourself you can simply use "me" or "self" (users only).
title (``str``):
Stickerset name, 1-64 chars
@ -58,6 +55,12 @@ class CreateStickerSet:
sticker to add.
Pass a file_id as string to send a file that exists on the Telegram servers.
user_id (``int`` | ``str``, *optional*):
Unique identifier (int) or username (str) of the Stickerset owner.
For you yourself you can simply use "me" or "self" (users only).
required for bots.
default to "me".
emoji (``str``, *optional*):
Associated emoji.
default to "🤔"
@ -71,14 +74,23 @@ class CreateStickerSet:
Example:
.. code-block:: python
# Send document by uploading from local file
await app.create_sticker_set("me", "My First Pack", "myfirstpack", "AAjjHjk")
"""
file = None
if self.me.is_bot and user_id is None:
raise ValueError("user_id is required for bots")
if isinstance(sticker, str):
if os.path.isfile(sticker) or re.match("^https?://", sticker):
raise ValueError(f"file_id is invalid!")
document = await self.send_document(
user_id or "me",
sticker,
force_document=True,
disable_notification=True
)
uploaded_media = utils.get_input_media_from_file_id(document.document.file_id, FileType.DOCUMENT)
media = uploaded_media.id
_ = await document.delete()
else:
decoded = FileId.decode(sticker)
media = raw.types.InputDocument(
@ -87,11 +99,19 @@ class CreateStickerSet:
file_reference=decoded.file_reference
)
else:
raise ValueError(f"file_id is invalid!")
document = await self.send_document(
user_id or "me",
sticker,
force_document=True,
disable_notification=True
)
uploaded_media = utils.get_input_media_from_file_id(document.document.file_id, FileType.DOCUMENT)
media = uploaded_media.id
_ = await document.delete()
r = await self.invoke(
raw.functions.stickers.CreateStickerSet(
user_id=await self.resolve_peer(user_id),
user_id=await self.resolve_peer(user_id or "me"),
title=title,
short_name=short_name,
stickers=[