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

Merge f22b68b3fb into 3a632d4590

Browse source
This commit is contained in:
EDM115 2025-04-10 22:20:44 +00:00 committed by GitHub
commit b2f5611b53
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pyrogram/client.py
View file

@ -45,7 +45,8 @@ from pyrogram.errors import CDNFileHashMismatch
from pyrogram.errors import (
SessionPasswordNeeded,
VolumeLocNotFound, ChannelPrivate,
BadRequest, ChannelInvalid, PersistentTimestampInvalid, PersistentTimestampOutdated
BadRequest, ChannelInvalid, PersistentTimestampInvalid, PersistentTimestampOutdated,
FloodWait, FloodPremiumWait
)
from pyrogram.handlers.handler import Handler
from pyrogram.methods import Methods
@ -1121,7 +1122,7 @@ class Client(Methods):
if isinstance(e, asyncio.CancelledError):
raise e
if isinstance(e, pyrogram.errors.FloodWait):
if isinstance(e, (FloodWait, FloodPremiumWait)):
raise e
return None
@ -1346,7 +1347,7 @@ class Client(Methods):
await cdn_session.stop()
except pyrogram.StopTransmission:
raise
except pyrogram.errors.FloodWait:
except (FloodWait, FloodPremiumWait):
raise
except Exception as e:
log.exception(e)