soj1pddlmZddlmZddlmZddlmZddlm cm cm Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZdd lmZmZmZm Z ddl!m"Z"GddeZ#GddeZ$dZ% d)dedej&de j de dede ddfdZ' d*dedej&de j de de de(ddfdZ) d*dedej&de j de de(ddf dZ*dedej&de ddfd Z+dedej&deddfd!Z,de de-fd"Z.de/e0e0ffd#Z1d$e0de-fd%Z2de j de-fd&Z3d'e-de dzfd(Z4dS)+)StrEnum)fn)Update) ContextTypesN)User) CommandName) MessageSource)ReservedKeyboardKeys)Screen)Keyboard)add_or_remove_bountyget_amount_from_stringvalidate_amount)"full_message_or_media_send_or_editget_message_sourceget_yes_no_keyboardmention_markdown_user)get_belly_formattedceZdZdZdZdS)OwnerBountyReservedKeysz4 Reserved keys for owner bounty management. aN)__name__ __module__ __qualname____doc__ACTION3/app/src/chat/common/screens/screen_owner_bounty.pyrrsFFFrrceZdZdZdZdS)OwnerBountyActionz Owner bounty actions. rpallN)rrrrREVERT_PENDING_ALLrrrr!r!$s"!rr!z-eupdatecontextcommanduserinbound_keyboard target_userreturncfK|t|||d{VdSt|tju}|jt jur|rdSt|||d{VdS|jt jurt|||||d{VdSt||||||d{VdS)aS Manage owner-only bounty adjustment commands. :param update: The update :param context: The context :param command: The command :param user: The owner user :param inbound_keyboard: Inbound keyboard for confirmation callbacks :param target_user: Target user from replied message in groups :return: None N) manage_keyboardrr GROUPnamerREVERT_PENDING_BOUNTY_ALL$send_revert_pending_all_confirmationREVERT_PENDING_BOUNTYmanage_revert_pendingmanage_add_or_take)r$r%r&r'r(r)is_groups rmanager53s&#fg/?@@@@@@@@@!&))]-@@H|{<<<   F267DIIIIIIIII|{888#FGWk8TTTTTTTTT VWgt[( S SSSSSSSSSSrFr4c rK|}t|jv}d|jD}|ra|t|d|dd{VdSt|dkr(t|t ||dd{VdS|d} nt|dkrt|d|d{VdS|d} t|d krt|d |d{VdSt |d }|t|d |d{VdSt |||| |d d{VsdSt| |} tj |j }|j tj urt|| ||d||d{V|r1dt!| t#|} ndt!| t#|} n|j| krLd|_|dt#|t!| } nT|xj| zc_|dt!| t#|} t|| ||d{VdS)a= Manage owner-only add/take bounty commands. :param update: The update :param context: The context :param command: The command :param user: The owner user :param target_user: Target user from replied message in groups :param is_group: Whether this is a group message :return: None c(g|]}|tk |Sr) EXEMPT_FLAG).0ps r z&manage_add_or_take..ps"DDD1 3C3C!3C3C3CrNz)rrFrErHrevert_pending_bounty)r$r%r&r)r4rQs rr2r2sx"  4U"&         F  w! " "a ' '4@        F%g&8&;<<  4@        F  - -D , "    rcKt|tjtjt jig}t|d||d{VdS)z Send pending bounty revert all confirmation. :param update: The update :param context: The context :param user: The owner :return: None )screen extra_keyszDAre you sure you want to revert the pending bounty for every player?)r$keyboardN)rr PVT_OWNER_BOUNTYrrr!r#r)r$r%r'rWs rr0r0s  *'.0A0T   H -N    rcK|tjtjkrdS|t jst|d|d{VdSt|d|d{Vt\}}t|d |||d{VdS)z Manage owner bounty keyboard actions. :param update: The update :param context: The context :param inbound_keyboard: Inbound keyboard :return: None Nz Pending bounty revert cancelled.r>z4Reverting pending bounty for all eligible players...uPending bounty revert completed successfully. *Statistics* • Players processed: *{}* • Players skipped \(no pending bounty\): *{}*) getrrr!r#get_boolr CONFIRMrrevert_pending_bounty_allrL)r$r%r( processedskippeds rr,r,*s? 4;<<  / 0 0   $ $%9%A B B0  .           ,>    344Iw , <=CF9g$' " "D(-AN>EEF[\`FaFabbb   # # % %.. ;/    %47" # #GGIII ............... Z ` `d##%?%O%O  s.ACCCc|d}tjtjdtjdkz}|D]}|jpd}tjj5tj tj |zdtj |j k dddn #1swxYwY|dz }tj |z }||fS)zu Revert pending bounty for every user without applying income tax. :return: Processed and skipped counts rFraNr@)rselectrfrbis_nulliteratorrcrdrer$rMrJrgcount)r^users_with_pendingr'rbr_s rr]r]jsS I,,  $ $U + +0Cq0HI#++--,1 Z ' ' ) ) 2 2 K{^3    eDGtw&''  2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 Q kmm!!##i/G g sADD D rPcv|dkrdnd}d|tt|S)zd Format a signed bounty amount. :param amount: The amount :return: Formatted amount r-u{}฿{})rLrabs)rPsigns rrhrhs:1**33"D   D"5c&kk"B"B C CCrc0|jtjurdSdS)zv Get missing amount text for group commands. :param command: The command :return: Missing amount text z(Please specify the bounty amount to add.z+Please specify the bounty amount to remove.)r.rrK)r&s rrGrGs |{---99 8 8r identifiercx|}|dr |dd}|r&tjtj|k}||S|dkrdStjt jtj| kS)z Resolve a user by Telegram user id or username. :param identifier: Telegram user id or username :return: User, if found @r@Nrq) strip startswith isnumericr get_or_none tg_user_idrLOWER tg_usernamelower)rur's rrHrHs!!##JS!!$^ : =>>  KRt  BHT%566*:J:J:L:LL M MMr)NN)NF)5enumrpeeweer src.tg_compatrsrc.tg_compat.extrsrc.model.enums.CommandmodelenumsCommandsrc.model.Userrsrc.model.enums.CommandNamersrc.model.enums.MessageSourcer $src.model.enums.ReservedKeyboardKeysr src.model.enums.Screenr src.model.pojo.Keyboardr src.service.bounty_servicer rrsrc.service.message_servicerrrrsrc.utils.string_utilsrrr!r8 DEFAULT_TYPEr5boolr3r2r0r,strrStupleintr]rhrGrHrrrrs; ******))))))))))))333333777777EEEEEE)))))),,,,,,  766666g!!!!!!!! "& $T$T $T  &$T_$T  $T  $T  $T $T$T$T$TX || |  &|_|  |  |  | ||||F 22 2  &2_2 2  2  2222j )6>B 8& &)6&JR& &&&&R.5c?0DsDsDDDD 97? 9s 9 9 9 9NNt NNNNNNr