This commit is contained in:
EDM115 2025-06-27 14:05:59 +00:00 committed by GitHub
commit c2917447e4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -46,7 +46,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
@ -1130,7 +1131,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
@ -1355,7 +1356,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)