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 Giveaway

Browse source 
Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
wulan17 2025-03-25 22:50:04 +07:00
parent 3622b13a9f
commit bf3dba50ee
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
3 changed files with 4 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pyrogram/types/messages_and_media/external_reply_info.py
View file

@ -218,7 +218,7 @@ class ExternalReplyInfo(Object):
game = types.Game._parse(client, media) game = types.Game._parse(client, media)
media_type = enums.MessageMediaType.GAME media_type = enums.MessageMediaType.GAME
elif isinstance(media, raw.types.MessageMediaGiveaway): elif isinstance(media, raw.types.MessageMediaGiveaway):
giveaway = types.Giveaway._parse(client, reply) giveaway = types.Giveaway._parse(client, reply, chats)
media_type = enums.MessageMediaType.GIVEAWAY media_type = enums.MessageMediaType.GIVEAWAY
elif isinstance(media, raw.types.MessageMediaGiveawayResults): elif isinstance(media, raw.types.MessageMediaGiveawayResults):
giveaway_winners = await types.GiveawayResult._parse(client, media) giveaway_winners = await types.GiveawayResult._parse(client, media)

22
pyrogram/types/messages_and_media/giveaway.py
View file

@ -73,7 +73,6 @@ class Giveaway(Object):
stars: int = None, stars: int = None,
additional_price: str = None, additional_price: str = None,
allowed_countries: List[str] = None, allowed_countries: List[str] = None,
private_channel_ids: List[int] = None,
is_winners_hidden: bool = None is_winners_hidden: bool = None
): ):
super().__init__(client) super().__init__(client)
@ -86,28 +85,12 @@ class Giveaway(Object):
self.new_subscribers = new_subscribers self.new_subscribers = new_subscribers
self.additional_price = additional_price self.additional_price = additional_price
self.allowed_countries = allowed_countries self.allowed_countries = allowed_countries
self.private_channel_ids = private_channel_ids
self.is_winners_hidden = is_winners_hidden self.is_winners_hidden = is_winners_hidden
@staticmethod @staticmethod
async def _parse(client, message: "raw.types.Message") -> "Giveaway": async def _parse(client, message: "raw.types.Message", chats: dict) -> "Giveaway":
giveaway: "raw.types.MessageMediaGiveaway" = message.media giveaway: "raw.types.MessageMediaGiveaway" = message.media
chats = [] chats = types.List([types.Chat._parse_channel_chat(client, chats.get(i)) for i in giveaway.channels])
private_ids = []
for raw_chat_id in giveaway.channels:
chat_id = utils.get_channel_id(raw_chat_id)
try:
chat = await client.invoke(
raw.functions.channels.GetChannels(
id=[await client.resolve_peer(chat_id)]
)
)
except FloodWait as e:
await asyncio.sleep(e.value)
except Exception:
private_ids.append(chat_id)
else:
chats.append(types.Chat._parse_chat(client, chat.chats[0]))
return Giveaway( return Giveaway(
chats=chats, chats=chats,
@ -118,7 +101,6 @@ class Giveaway(Object):
new_subscribers=giveaway.only_new_subscribers, new_subscribers=giveaway.only_new_subscribers,
additional_price=giveaway.prize_description, additional_price=giveaway.prize_description,
allowed_countries=giveaway.countries_iso2 if len(giveaway.countries_iso2) > 0 else None, allowed_countries=giveaway.countries_iso2 if len(giveaway.countries_iso2) > 0 else None,
private_channel_ids=private_ids if len(private_ids) > 0 else None,
is_winners_hidden=not giveaway.winners_are_visible, is_winners_hidden=not giveaway.winners_are_visible,
client=client client=client
) )

2
pyrogram/types/messages_and_media/message.py
View file

@ -1039,7 +1039,7 @@ class Message(Object, Update):
game = types.Game._parse(client, media) game = types.Game._parse(client, media)
media_type = enums.MessageMediaType.GAME media_type = enums.MessageMediaType.GAME
elif isinstance(media, raw.types.MessageMediaGiveaway): elif isinstance(media, raw.types.MessageMediaGiveaway):
giveaway = await types.Giveaway._parse(client, message) giveaway = await types.Giveaway._parse(client, message, chats)
media_type = enums.MessageMediaType.GIVEAWAY media_type = enums.MessageMediaType.GIVEAWAY
elif isinstance(media, raw.types.MessageMediaGiveawayResults): elif isinstance(media, raw.types.MessageMediaGiveawayResults):
giveaway_result = await types.GiveawayResult._parse(client, message.media) giveaway_result = await types.GiveawayResult._parse(client, message.media)