Add get_boosts method

This commit is contained in:
KurimuzonAkuma 2023-10-29 10:40:28 +03:00
parent 6be799d21a
commit cdc1fd9d12
2 changed files with 59 additions and 0 deletions

View File

@ -17,8 +17,10 @@
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
from .apply_boost import ApplyBoost
from .get_boosts import GetBoosts
class Premium(
ApplyBoost,
GetBoosts
):
pass

View File

@ -0,0 +1,57 @@
# Pyrogram - Telegram MTProto API Client Library for Python
# Copyright (C) 2017-present Dan <https://github.com/delivrance>
#
# 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 <http://www.gnu.org/licenses/>.
from typing import Union
import pyrogram
from pyrogram import raw
from pyrogram import types
class GetBoosts:
async def get_boosts(
self: "pyrogram.Client",
) -> bool:
"""Get your boosts list
.. include:: /_includes/usable-by/users-bots.rst
Returns:
List of :obj:`~pyrogram.types.MyBoost`: On success.
Example:
.. code-block:: python
# get boosts list
app.get_boosts()
"""
r = await self.invoke(
raw.functions.premium.GetMyBoosts()
)
users = {i.id: i for i in r.users}
chats = {i.id: i for i in r.chats}
return types.List(
types.MyBoost._parse(
self,
boost,
users,
chats,
) for boost in r.my_boosts
)