This commit is contained in:
EDM115 2025-05-18 06:25:22 -07:00 committed by GitHub
commit 0b0845a746
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 ( from pyrogram.errors import (
SessionPasswordNeeded, SessionPasswordNeeded,
VolumeLocNotFound, ChannelPrivate, VolumeLocNotFound, ChannelPrivate,
BadRequest, ChannelInvalid, PersistentTimestampInvalid, PersistentTimestampOutdated BadRequest, ChannelInvalid, PersistentTimestampInvalid, PersistentTimestampOutdated,
FloodWait, FloodPremiumWait
) )
from pyrogram.handlers.handler import Handler from pyrogram.handlers.handler import Handler
from pyrogram.methods import Methods from pyrogram.methods import Methods
@ -1129,7 +1130,7 @@ class Client(Methods):
if isinstance(e, asyncio.CancelledError): if isinstance(e, asyncio.CancelledError):
raise e raise e
if isinstance(e, pyrogram.errors.FloodWait): if isinstance(e, (FloodWait, FloodPremiumWait)):
raise e raise e
return None return None
@ -1354,7 +1355,7 @@ class Client(Methods):
await cdn_session.stop() await cdn_session.stop()
except pyrogram.StopTransmission: except pyrogram.StopTransmission:
raise raise
except pyrogram.errors.FloodWait: except (FloodWait, FloodPremiumWait):
raise raise
except Exception as e: except Exception as e:
log.exception(e) log.exception(e)