soj,"ddlZddlZddlmZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZd e jd ed eezdedededdfdZd e jd ed eezdedededdfdZd e jd ed eezdedededdfdZd e jd ed eezdedededdfdZdZdedefdZ d e jdeezd eezdededzde!edzeff dZ"d e jdeezdedefdZ#d e jd edededededed eddfd!Z$dS)"N)Message) Forbidden BadRequest TelegramError RetryAfter) ContextTypes)phrases)Group) GroupChat)User)save_group_chat_error)full_message_sendcontextowner from_chat_id message_id should_pinprogress_message_idreturnc bK|jt||||||dS)a Dispatches the group broadcast (gcast) to a background task :param context: The context :param owner: The owner, used to send the final report :param from_chat_id: The chat id of the message to broadcast :param message_id: The id of the message to broadcast :param should_pin: True if the broadcast message should be pinned :param progress_message_id: The id of the progress message to edit with the final report :return: None N) application create_taskbroadcast_to_groupsrrrrrrs %/app/src/service/broadcast_service.pybroadcast_to_groups_dispatchrsH& ## UL*jBU  c Kd}d}d}d} d} tjtjdk} | D]} t | j} | D]}|dz }t|| j|||j d{V\}}|:|dz }t|d|r"| jrd| _| i|dz }|sqt|| j|j d{Vr| dz } | dz } t|dt|||tjtj||||| | d{VdS)a Broadcasts a message to every group/topic the Bot is an active member of :param context: The context :param owner: The owner, used to send the final report :param from_chat_id: The chat id of the message to broadcast :param message_id: The id of the message to broadcast :param should_pin: True if the broadcast message should be pinned :param progress_message_id: The id of the progress message to edit with the final report :return: None rTNz#Failed to broadcast message (gcast)Fz'Failed to pin broadcast message (gcast))r selectwhere is_activeiteratorlist group_chats _copy_message tg_group_id tg_topic_idr save _pin_messager_send_final_reportr BROADCAST_GROUPS_COMPLETEDBROADCAST_GROUPS_STATISTICSformat)rrrrrrtotal successfulfailedpinned pin_failedgroupsgroupr% group_chatcopied_messageis_permanent_failures rrr*s& EJ F FJ,....u$/FGGF""]]'+E,='>'> % ] ]J QJE9F*L*jF\::444444 0N0%! %j2WXXX'!EO!',EOJJLLL !OJ !'5+WXXXXXXXX ]! a %j2[\\\\7 ]:  *+225*fMM           rc bK|jt||||||dS)a Dispatches the player broadcast (pcast) to a background task :param context: The context :param owner: The owner, used to send the final report :param from_chat_id: The chat id of the message to broadcast :param message_id: The id of the message to broadcast :param should_pin: True if the broadcast message should be pinned :param progress_message_id: The id of the progress message to edit with the final report :return: None N)rrbroadcast_to_playersrs rbroadcast_to_players_dispatchr;psH& ## UL*jBU  rc JKd}d}d}d} d} tjtjdk} | D]} |dz }t || j||dd{V\} }| *|dz }|r"| jrd| _| T|dz }|s\t|| j| j d{Vr| dz } | dz } t|||tj tj ||||| | d{VdS)a Broadcasts a message to every registered player's private chat :param context: The context :param owner: The owner, used to send the final report :param from_chat_id: The chat id of the message to broadcast :param message_id: The id of the message to broadcast :param should_pin: True if the broadcast message should be pinned :param progress_message_id: The id of the progress message to edit with the final report :return: None rTrNF)r r r!r"r#r& tg_user_idr)r*rr+r BROADCAST_PLAYERS_COMPLETEDBROADCAST_PLAYERS_STATISTICSr.)rrrrrrr/r0r1r2r3playersplayerr7r8s rr:r:s& EJ F FJ+----dn.DEEG""$$  5B V& j$6 6 0 0 0 0 0 0 ,,  ! aKF# (8 $)  a   gv'8.:STT T T T T T T  aKFF !OJJ  +,33E:vNN           r)zchat not foundzuser not foundzuser is deactivatedpeer_id_invaliderrorcttrdSttr"tfdtDrdSdS)az Determines whether an error means the destination is permanently unreachable (Bot kicked/ blocked, group or account deleted), as opposed to a transient error that might succeed if retried in a future broadcast. :param error: The error raised while broadcasting to a destination :return: True if the destination should be excluded from future broadcasts Tc3^K|]'}|tvV(dS)N)strlower).0snippetrCs r z(_is_permanent_failure..sK--*13u::##%%%------rF) isinstancerrany_PERMANENT_BAD_REQUEST_SNIPPETS)rCs`r_is_permanent_failurerNsq%##t%$$----5T---**t 5rchat_idmessage_thread_idcKtdD]} |j||||dd{V}|dfcS#t$rh}|dkr ! ! !   !|| Yg Y Y YZZZmBN333333333 OTgTTPRTT U U U;;;;;; 3 3 3.r222 2 2 2 2 2 2 2 2 3 ;s/(A C =B- B-- C:C CCcKtdD]} |j||dd{VdS#t$rP}|dkr$t j|jd{VYd}~]tjd|d|Yd}~dSd}~wt$r&}tjd|d |Yd}~dSd}~wwxYwdS) a Pins a message in a chat. Retries once if rate limited. A failure to pin must not interrupt the broadcast. :param context: The context :param chat_id: The chat id :param message_id: The id of the message to pin :return: True if the message was pinned rRT)rOrrSNrz#Failed to pin broadcast message in rTFz: ) rUrVpin_chat_messagerrZr[r\rXrYr)rrOrr]r_r`s rr*r*sD88 +..JT/       44   !||mBN333333333 O`'``\^`` a a a555555    OQ'QQRQQ R R R555555  5s'#; C%B/B CB>>Ccompleted_phrasestatistics_textr2r3cK|r tj||nd}t|||||j|d{VdS)a2 Edits the progress message with the final broadcast report :param context: The context :param owner: The owner :param progress_message_id: The id of the progress message to edit :param completed_phrase: The completed phrase template, expecting statistics and pin stats :param statistics_text: The broadcast statistics text :param should_pin: True if pin statistics should be included :param pinned: The number of successfully pinned messages :param pin_failed: The number of messages that failed to pin :return: None )rOedit_message_idN)r BROADCAST_PIN_STATISTICSr.rr=) rrrrcrdrr2r3pin_statistics_texts rr+r+5s0HRY(// CCCWY 1DEE +    r)%rZrX src.tg_compatrsrc.tg_compat.errorrrrrsrc.tg_compat.extr resourcesr src.model.Groupr src.model.GroupChatr src.model.Userr src.service.group_servicer src.service.message_servicer DEFAULT_TYPEintrFboolrrr;r:rMrNtupler&r*r+rrrxs!!!!!!PPPPPPPPPPPP******!!!!!!))))));;;;;;999999  & )      4C  &C C)C C  C  C CCCCL  & )      4>  &> >)> >  >  > >>>>F#4*)  &) 3Y))) ) Tz )  7T>4  ))))X  &14sHK >   &                   r