sojOddlZddlZddlZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZddlZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?ddl@mAZAddlBmCZCddlDmEZEdd lFmGZGdd!lHmIZIdd"lJmKZKdd#lLmMZMmNZNdd$lOmPZPmQZQmRZRdd%lSmTZTdd&lUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\dd'l]m^Z^dd(l_m`Z`maZambZbdd)lcmdZddd*lemfZfddlgmhcmiZidd+ljmkZkdd,llmmZndd-lUmoZodd.lpmqZqmrZrdd/lsmtZtdd0lumvZvdd1lgmwZwmxZxmyZymzZzdd2l{m|Z|Gd3d4e|Z} dNd6e d7ej~d8ed9eEd:ed;edZd:ed8ed=e5e5e5e5gfd?Zd7ej~d:ed@ekd6e d8ed=eeeEf dAZd:ed=ekfdBZd:ed=ekfdCZdDed=efdEZd:edFekd8edGekd=ef dHZd6e d=e-fdIZd6e d7ej~d:edJedKedLed=dfdMZdS)ON)datetime timedelta)TupleCallable) DoesNotExist)Update) RetryAfter BadRequest) ContextTypes)phrases Environment)Game) GroupChat)User)ContextDataKey)Emoji) GameStatus)GameTurnNotificationGameOutcomeNotification)ReservedKeyboardKeys) SavedMedia)SavedMediaType)Screen)DevilFruitAbilityType)IncomeTaxEventType)CommonChatException) GameBoard) GameOutcome)GameTurn)GameType) PunkRecords)Shambles)WhosWho)ContextDataValue)Keyboard) Character) SupabaseRest) Terminology)add_or_remove_bountyvalidate_amount)convert_seconds_to_durationget_remaining_durationget_elapsed_duration)get_ability_adjusted_datetime)mention_markdown_userdelete_messagefull_media_sendget_message_urlfull_message_send"full_message_or_media_send_or_edit get_deeplink)send_notification)get_bot_context_dataset_bot_context_data'get_random_user_context_inner_query_key)get_outcome_text)get_belly_formatted)RussianRoulette)RussianRouletteChamberStatus)get_message_source)get_current_languageset_current_language) MessageSource)SavedMediaName)edit_other_player_messageenqueue_auto_moveget_global_share_keyboardend_global_game_player)StrEnumceZdZdZdZdZdS)GameRRReservedKeysz+ The reserved keys for this screen abN)__name__ __module__ __qualname____doc__GAME_IDPOSITION./app/src/chat/common/screens/screen_game_rr.pyrIrIRs$GHHHrSrIFupdatecontextuserinbound_keyboardgame is_auto_moveedit_message_idreturncR Ktj||||d{V}|dSt||d{V}|d}|d} |d} |d} tj|_|jt|kr|s<| ||s&t|tj |ddd{VdS|j tj\} } |j| | t"jkr&t|tj|ddd{VdS|r&t|tj|ddd{VdSt"j|j| | <|s8t|t-||dd{V||||r+t3|||| d{V|| r|||| }tj|||||s|r||nd d{Vd |_|} t?|tA|||| |tC||||||"d|dd d{V}|r|j#}n=#tHtJf$r)}tMj'd|j#d|Yd}~nd}~wwxYw|r|j(||}tS}tU|j+,tA|| |j+| }tU|j(,tA|| |j(| }tU||j-.t_||||||tC||| ||dS|jt|kr(|s|0t?|tA|||| |tC||||||"tbj2|dd{V}|r|j(tS}tU|j+,tA|| |j+| }tU|j(,tA|| |j(| }tU||j-.t_||||r|j#nd||tC||| ||||||3|s!tj4|||j5d{V|sM|rM|j-.tm||||7|tp|j#dSdSdS)a Manage the Russian Roulette screen :param update: The update object :param context: The context object :param user: The user object :param inbound_keyboard: The inbound keyboard :param game: The game object :param is_auto_move: If the interaction is from auto move :param edit_message_id: Message to be edited. Useful for when a user does not interact with the private game before auto-move is triggered, since the message tied with the update would be the start command :return: None NrT)rUanswer_callback show_alert)rUra) is_challenger other_board)rUsend_outcome_to_userF)captionrUkeyboardauthorized_usersedit_only_caption_and_keyboardr[ignore_bad_request_exceptionshould_log_ignored_exceptionz1Failed to send auto-move result message for game z: )rgrUrhrisaved_media_namer[rk)9 game_service validate_gameget_all_boardsrnowlast_interaction_datescreen get_screen is_global is_user_turnr3r GAME_NOT_YOUR_TURNinforIrQcylinderRRChamberStatusFIRED+RUSSIAN_ROULETTE_GAME_CHAMBER_ALREADY_FIRED is_finishedGAME_GLOBAL_WAIT_FOR_OPPONENTget_choice_textbullet_is_fired set_boardrFrc get_outcomeend_gameget_other_playershould_update_modelr1get_specific_textget_outbound_keyboard get_playersidr r loggingwarningopponentr?r@ challenger get_language application create_taskrCset_turnrBGAME_RUSSIAN_ROULETTEsavenotify_game_turn game_turnrD get_user_turn auto_move)rUrVrWrXrYrZr[ all_boardsboardrechallenger_boardopponent_boardxy game_outcomesent_message_idmessagee other_playerplayer_languagechallenger_text opponent_texts rTmanager[s.+FG=MtTT T T T T T TD |.! Q ?#8 8 8#C $         F      #5 $         F.4q! # 5 5 7 788 $         ud###      `($dFXFXY]F^F^___ _ _ _ _ _ _ _ [11@$)$5$5d6H6H6N6NP[$\$\ #   040@0@[L[%%d+++W[           $) * %4)$t[II.weVTRR!%!1!1!3!3/3 /-1-1 & & &      G -")*J'    ORDGRRqRR          >>    90066L244O !=!=!?!? @ @ @/&O !!;!;!=!= > > >-ndm5EM ! 1 1 1   + +)##!)'4fl[[    *V"4"444T^^=M=M4 -!$t[AA&weVTJJ))++'='%)         G ~~ DM5.00T_99;;<<<+D2BDOUcddT]7799:::)$ O_`` _---'' %%/ 4%gt[&$OO    NN5$IIKKK >>  L+GT5?KKKKKKKKK      7 '' u':':4'@'@)WZ           s3ALMMMcfK|st|}d}nf||rt|nt|}||rt|nt|}||r|}|}n|}|}||||fS)z Get all boards :param game: The game object :param user: The user object :return: Current board, other board, challenger board, opponent board N)ruget_board_challengerrcget_board_opponent)rYrWrrerrs rTrprp+s >>    $T** +/*<*>00 3(* &/nn c c "M711)]KK c$$Xe.B%C%CDDDD'.&/)]1K !1!:::%)9)BBBAJJLLC"6EE 999!3EE)1Y>>,5FF % : % 9$$Xe+jY_N`N`%a%a%abbbb  ////  ' ' ) ) .2.@.@.F.F !'4000  rSc|j=t}||_||St j|j}tdi|S)zZ Get the challenger board :param game: The game object :return: The board NrR)rr<get_as_json_stringrjsonloadsrYrrussian_roulette_dicts rTrrse z*,,%88::   Jtz22  3 32 3 33rSc|jIttj}||_||St j|j}tdi|S)zX Get the opponent board :param game: The game object :return: The board N)rrR)rr<rOPPONENTrrrrrs rTrrsp "*X5FGGG.AACC  Jt':;;  3 32 3 33rS bullet_shotc6|r tjStjS)zb Get the choice text :param bullet_shot: If the bullet was shot :return: The text )r !RUSSIAN_ROULETTE_GAME_BULLET_SHOT%RUSSIAN_ROULETTE_GAME_BULLET_NOT_SHOT)rs rTrrs988  88rSrrec ||}||}d}|rL|tjtj||j|j zdz z }| s| r|tj z }|j |dtj| |||j|j zdz zz }| s| r|tj z }| r|dz }|r| p|o|duo| }|r| p|o|duo| }tj|| p| |||||s||n||||S)z Get the text :param game: The game object :param board: The russian roulette board :param user: The user :param other_board: The other board :return: The text r_N z rd)text_to_add_before_footerropponent_has_finished)rc is_opponentrur GAME_POINTSformat TEXT_YOURget_fired_correct_countrowscolumnsr}GAME_POINTS_FINISHEDrrget_markdown_mentionrnget_textrr) rYrrWrercr extra_textrrs rTrrs&&t,,M""4((KJ ~~%g)00  u<<>>emA[_`@`         7!2!2!4!4 7 '6 6J = $ $!4!;!;%%d++@@BB3355em+q0"" J !! ;[%<%<%>%> ;g:: !! %f$ ,D1B1B1D1DM 4/MK4K4K4M4M)@U->->-@-@O+T1Ok6M6M6O6O    1e//11 $,,T22 LL)-)9)9CD!!!t", 73   rScht|tjur tjStjS)zU Get the screen :param update: The update object :return: The screen )r>rAGROUPrGRP_RUSSIAN_ROULETTE_GAMEPVT_RUSSIAN_ROULETTE_GAME)rUs rTrtrts,&!!]%888//  ++rSplayer message_id previous_gamec K|r||rt|}t|}nt|}t|}||krdSt jtj t j |j kt j |j kz }|dkrdStdtj|j tj|j|jfit'|} t)j|j }t-|||| |d|d{VdS)a> Auto-move for the given player :param update: The update :param context: The context object :param game: The game object :param player: The player object :param message_id: Identifier of the message to be edited :param previous_game: Game object state during enqueue :return: None N)rrrrrT)rZr[)rurcrrrrrUrrqwhererrrexecuter%rIrPrQrrrtr get_by_idr) rUrVrYrrrrprevious_board claimed_rows mock_keyboards rTrrs& >>  ;t11&99;!5d!;!;-m<<!3D!9!9+M:: !!^%F%F%H%HHH (,..999 47"t'ATE_'_` a a  q'  &  '%.%.)I &!! M^FI & &F 4Yc   rS)NFN)asynciorrrrtypingrrpeeweer src.tg_compatrsrc.tg_compat.errorr r src.tg_compat.extr constantsc resourcesr r rsrc.model.Gamersrc.model.GroupChatrsrc.model.Userrsrc.model.enums.ContextDataKeyrsrc.model.enums.Emojirsrc.model.enums.GameStatusrsrc.model.enums.Notificationrr$src.model.enums.ReservedKeyboardKeysrsrc.model.enums.SavedMediarsrc.model.enums.SavedMediaTypersrc.model.enums.Screenr1src.model.enums.devil_fruit.DevilFruitAbilityTyper-src.model.enums.income_tax.IncomeTaxEventTypersrc.model.error.CommonChatErrorrsrc.model.game.GameBoardrsrc.model.game.GameOutcomersrc.model.game.GameTurnrsrc.model.game.GameTyper &src.model.game.punkrecords.PunkRecordsr! src.model.game.shambles.Shamblesr"src.model.game.whoswho.WhosWhor#src.model.pojo.ContextDataValuer$src.model.pojo.Keyboardr%src.model.wiki.Characterr&src.model.wiki.SupabaseRestr'src.model.wiki.Terminologyr(src.service.bounty_servicer)r*src.service.date_servicer+r,r-src.service.devil_fruit_servicer.src.service.message_servicer/r0r1r2r3r4r5 src.service.notification_servicer6src.utils.context_utilsr7r8r9src.utils.phrase_utilsr:src.utils.string_utilsr;src.service.game_serviceservicern.src.model.game.russianroulette.RussianRouletter<;src.model.game.russianroulette.RussianRouletteChamberStatusr=rzr>src.service.language_servicer?r@src.model.enums.MessageSourcerAsrc.model.enums.SavedMediaNamerBrCrDrErFenumrGrI DEFAULT_TYPEboolintrrplistrrrstrrrrtrrRrSrTr%s  (((((((("""""""" 66666666******<<<<<<<<))))))999999''''''111111VVVVVVVVEEEEEE111111999999))))))SSSSSSLLLLLL??????......222222,,,,,,,,,,,,>>>>>>555555222222<<<<<<,,,,,,......444444222222LLLLLLLL JIIIII?>>>>> 433333666666/////////JJJJJJ;:::::SSSSSSSS777777999999 M M M  &M  M  M  M  M M  M M M M `t4Y 9V_4`@:  &: :&:  :  :  $x. ::::z4t44444$4T4o4444$ 9 9# 9 9 9 93 3&3.23AP33333l ,v ,& , , , ,; ;  &; ;  ;  ;  ; ;;;;;;rS