Skip to content

Commit aa79531

Browse files
authored
Update broadcast.py
1 parent e466f51 commit aa79531

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

Thunder/utils/broadcast.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -107,13 +107,12 @@ async def do_broadcast():
107107
success = True
108108
break
109109
except FloodWait as e:
110-
if attempt < 2: # Don't sleep on last attempt
110+
if attempt < 2:
111111
await asyncio.sleep(e.value)
112112
else:
113-
raise
114-
115-
if not success:
116-
stats["failed"] += 1
113+
logger.warning(f"FloodWait persisted for user {user_id} after 3 attempts, last wait: {e.value}s")
114+
stats["failed"] += 1
115+
break
117116

118117
except (UserDeactivated, UserIsBlocked, PeerIdInvalid, ChatWriteForbidden, ChannelInvalid, InputUserDeactivated) as e:
119118
if isinstance(e, ChannelInvalid):
@@ -144,6 +143,8 @@ async def do_broadcast():
144143
if not is_authorized:
145144
await db.delete_user(user_id)
146145
stats["deleted"] += 1
146+
else:
147+
stats["failed"] += 1
147148

148149
except Exception as e:
149150
logger.error(f"Error copying message to user {user_id}: {e}", exc_info=True)

0 commit comments

Comments
 (0)