sojDddlmZddlmZddlmZmZmZmZddlm Z ddl m Z ddl mcmcmZddlmcmcmZddlmZddlmZddlmZddlmZddlmZddlmZddl mZ!ddl"mZ#ddl$mZ%ddl&mZ'ddl(mZ)ddl*mZ+ddl,mZ-ddl.mZ/ddl0mZ1ddl2mZ3ddl4mZ5ddl6mZ7ddl8mZ9ddl:mZ;ddlmZ?ddl@mZAddlBmZCddlDmZEddlFmZGddlHmZIddlJmZKddlLmMZMddlNmOZOdd lPmQZQdd lRmSZSdd lTmUZUmVZVmWZWdd lXmYZYdd lZm[Z[ddl\m]Z]m^Z^ddl_m`Z`ddlambZbddlcmdZdmeZeddlfmgZgddlhmiZiddljmkZkdede jldejdeQde`deQdemdeOddfdZdede jldeQde`dejdeQdeOd emddfd!Zn d*dede jldeQdemdeOd#emdemfd$Zo d+dede jldeQd%epdeOd&eqd'ereqdemfd(Zsdeememffd)ZtdS),)datetime)Tuple)UpdateMessageOriginUserMessageOriginChatMessageOriginChannel) ContextTypesN)manage)Group) GroupChat)User)Feature) DeletedMessageArrestNotificationDeletedMessageMuteNotification"DeletedMessageLocationNotification)Screen)GroupMessageValidationException)GroupChatErrorGroupChatException)Keyboard)release_devil_fruit_to_user) is_main_groupfeature_is_enabled)delete_message)send_notification) user_is_mutedupdatecontextcommanduserkeyboard target_user is_callback group_chatreturnc K|j}t|\} } | rd|_|dS| r(d|_|tj}n"|jsd|_| |jjr|dSn#t$rYnwxYwt|rmt|tj rt|||||d{VsdSn7t|tjrt|||||dd{VsdS|tjurdSt!|||||||| d{VdS)a Main function for the group chat manager :param update: Telegram update :param context: Telegram context :param command: Command to execute :param user: User object :param keyboard: Keyboard :param target_user: The target user in case of a reply :param is_callback: True if the message is a callback, False otherwise :param group_chat: The group chat :return: None FNT)check_only_muted)group%get_added_or_removed_from_group_event is_activesaveCommandGRP_SETTINGS_FEATURESeffective_useris_botAttributeErrorrrrMESSAGE_FILTERvalidateSILENCENDdispatch_screens) rrrr r!r"r#r$r(added_to_groupremoved_from_groups )/app/src/chat/group/group_chat_manager.pyr r Fs.#E*Ov)V)V&N&   /_   ' DL F      Z   j'*@ A A !&'4jQQQQQQQQ    GO < < !{JQU '* x+z>  sB B&%B&inbound_keyboardr6c K|jxtjkrt||||||d{Vnxtjkrt |||||d{Vnxtjkrt|||||d{Vn^xtjkrt|||||d{Vn2xtj krt|||||d{Vnxtj krt||||d{Vnxtjkrt||||d{Vnxtjkrt#||||d{Vnxtjkrt'||||d{VnVxtjkrt+|||||d{Vn*xtjkrt/|||||d{Vnxtjkrt3||||d{Vnxtjkrt7|||||d{Vnxtjkrt;|||||d{Vn{xtjkrt?|||d{VnQxtj krtC|||d{Vn'xtj"krtG||||d{Vnxtj$krtK|||||||d{Vnxtj&krtO|||||d{Vnxtj(krtS|||||||d{Vntxtj*krtW|||||||d{VnFxtj,krt[|||||d{Vnxtj.krt_||||d{Vnxtj0krtc|||d{Vnxtj2krtg||||||d{Vnxtj4krtk||d{Vnsxtj6krto||||d{VnItj8krts||||d{Vn! |j:twtxj=|Zt}j?|_@|jA}t|tjDr#|jEt||||d{VdSdSdSdS)a Dispatches the different screens :param update: Telegram update :param context: Telegram context :param user: User object :param inbound_keyboard: Keyboard to send :param command: Command to execute :param target_user: The target user in case of a reply :param group_chat: The group chat :param added_to_group: If the Bot was just added to the group chat :return: None )r$r"N)r!r)r9)GscreenrGRP_USER_STATUSmanage_screen_show_statusGRP_DOC_Q_GAMEmanage_screen_doc_q_gameGRP_CHANGE_REGIONmanage_screen_change_region GRP_FIGHTmanage_screen_fightGRP_GAMEmanage_screen_gameGRP_GAME_SELECTIONmanage_screen_game_selectionGRP_GAME_OPPONENT_CONFIRMATION(manage_screen_game_opponent_confirmationGRP_ROCK_PAPER_SCISSORS_GAMEmanage_screen_game_rpsGRP_RUSSIAN_ROULETTE_GAMEmanage_screen_game_rrGRP_PREDICTION_BETmanage_screen_prediction_betGRP_PREDICTION_BET_REMOVE#manage_screen_prediction_bet_removeGRP_PREDICTION_BET_STATUS#manage_screen_prediction_bet_status GRP_CREW_JOINmanage_screen_crew_joinGRP_CREW_INVITEmanage_screen_crew_invite GRP_SILENCEmanage_screen_silenceGRP_SILENCE_ENDmanage_screen_silence_end GRP_SPEAKmanage_screen_speakGRP_BOUNTY_GIFTmanage_screen_bounty_giftr-manage_screen_featuresGRP_DEVIL_FRUIT_SELLmanage_screen_devil_fruit_sellGRP_BOUNTY_LOANmanage_screen_bounty_loan GRP_PLUNDERmanage_screen_plunderGRP_DAILY_REWARDmanage_screen_daily_rewardGRP_DAILY_REWARD_PRIZE manage_screen_daily_reward_prizeGRP_OWNER_BOUNTYmanage_screen_owner_bounty GRP_SETTINGSmanage_screen_settingsGRP_SETTINGS_AUTO_DELETE"manage_screen_settings_auto_deleteGRP_SETTINGS_LANGUAGEmanage_screen_settings_languagecallback_queryrrUNRECOGNIZED_SCREENrnowlast_system_interaction_dater(rrDEVIL_FRUIT_APPEARANCEtg_group_usernamer) rrr r9rr"r$r6r(s r8r5r5s!0 . #V # # # #+$:S^        #V " " " "*67DBRT^__ _ _ _ _ _ _ _ _ %V % % % %-0@'        V    %fgt=MzZZ Z Z Z Z Z Z Z Z V____$VWdGZPP P P P P P P P P &V & & & &.8H        3V 2 2 2 2:8H        1V 0 0 0 0($Qabbb b b b b b b b b -V - - - -'P`aaa a a a a a a a a &V & & & &.vwgzZZ Z Z Z Z Z Z Z Z -V - - - -5fgtWV`aa a a a a a a a a -V - - - -5fgtZXX X X X X X X X X !V ! ! ! !)&'4AQS^__ _ _ _ _ _ _ _ _ #V # # # #+FGTCSU`aa a a a a a a a a V    'DD D D D D D D D D #V # # # #+FGZHH H H H H H H H H V    %fg{JOO O O O O O O O O #V # # # #+'7gz        *V ) ) ) )(!1:~        )V ( ( ( (0'7gz        $V # # # #+'7gz         V    '?OQ[\\ \ \ \ \ \ \ \ \ $V $ $ $ $,VWdJOO O O O O O O O O *V * * * *267DTUU U U U U U U U U $V $ $ $ $,$0@+        !V (99 9 9 9 9 9 9 9 9 ,V , , , ,4VWFVXbcc c c c c c c c c  ) ) )1&'CSU_`` ` ` ` ` ` ` ` ` $0()KLLL#,4LNN)!' z7+I J J Q'3-fgtZPP P P P P P P P P P$#  Q Q33Fr'c K|s.t|||tjj|d{VsdS|rdS |jj7t|||t j|d{VsdSn#t$rYnwxYw |jj 7t|||t j |d{VsdSn#t$rYnwxYw d}t|jj tr|jj jj}nkt|jj t"r|jj jj}n5t|jj t&r|jj jj}t|||t j||t jd{VsdSn#t$rYnwxYw |jj7t|||t j|d{VsdSn#t$rYnwxYw |jjrt|||t j|t9|jjjt jd{VsdSn#t$rYnwxYwdS)a Validates the message, deleting it if it's not valid :param update: Telegram update :param context: Telegram context :param user: User object :param is_callback: True if the message is a callback, False otherwise :param group_chat: The group chat :param check_only_muted: True if it should validate only if the user is muted, False otherwise :return: True if valid, False otherwise NFT)self_chat_user_idallowed_chat_user_ids)validate_location_levelLocationr4levelmessagestickerEnv$REQUIRED_LOCATION_LEVEL_SEND_STICKERget_intr0 animation&REQUIRED_LOCATION_LEVEL_SEND_ANIMATION isinstanceforward_originr sender_useridr sender_chatrchat'REQUIRED_LOCATION_LEVEL_FORWARD_MESSAGEWHITELIST_FORWARD_MESSAGEget_listdice'REQUIRED_LOCATION_LEVEL_SEND_DICE_EMOJIvia_bot'REQUIRED_LOCATION_LEVEL_USE_INLINE_BOTSstrWHITELIST_INLINE_BOTS)rrr r#r$r' identifiers r8r2r2s& ,VWdHKDUWabbbbbbbb 5  4 > ! - # />U GT3#M#U#U#W#WYc? ? 9 9 9 9 9 9 /5        fn35F G G ?6BEJJ 57H I I ?6BEJJ 57K L L ?6;>J,     7 ? ? A A ("%"?"H"H"J"J         5         >  *9P     7 ? ? A A  : : 4 4 4 4 4 4 *5        > ! -Aallowed_identifiers must be not None if identifier is not NoneT) notification)should_forward_messagerF) ValueErrorr is_arrestedrrrrrrrrr)rrr rr$r{r|es r8r}r}}sy*$)>)FYZZZ  (S1B-C-CG\-\-\4      c.1"4"41?_?a?abbb b z * * a1?]?_?_``` `   / /1?nUU  0 + T1>$v          V$$$$$$$$$uuuuu  4sB8B B88 D4C<<Dc@d}d} |jjjtjkrd}n#t $rYnwxYw |jjdjtjkrd}n#t tf$rYnwxYw||fS)z Gets the added or removed from group_chat event :param update: Telegram update :return: Tuple of added and removed FTr) rleft_chat_memberrrBOT_IDrr0new_chat_members IndexError)r added_to_chatremoved_from_chats r8r)r)sM > * -1C1C1E1E E E $        > *1 - 0CJ4F4F4H4H H H M J '      + ++s!3: AA 9BBB)F)NN)urtypingr src.tg_compatrrrrsrc.tg_compat.extr resources.Environment Environmentrsrc.model.enums.Commandmodelenumsr,src.model.enums.Locationr~$src.chat.common.screen_change_regionr rA+src.chat.common.screens.screen_daily_rewardrh1src.chat.common.screens.screen_daily_reward_prizerj)src.chat.common.screens.screen_doc_q_gamer?#src.chat.common.screens.screen_gamerE'src.chat.common.screens.screen_game_rpsrK&src.chat.common.screens.screen_game_rrrM-src.chat.common.screens.screen_game_selectionrG+src.chat.common.screens.screen_owner_bountyrl%src.chat.common.screens.screen_statusr=)src.chat.group.screens.screen_bounty_giftr_)src.chat.group.screens.screen_bounty_loanrd)src.chat.group.screens.screen_crew_inviterW'src.chat.group.screens.screen_crew_joinrU.src.chat.group.screens.screen_devil_fruit_sellrb#src.chat.group.screens.screen_fightrC8src.chat.group.screens.screen_game_opponent_confirmationrI%src.chat.group.screens.screen_plunderrf,src.chat.group.screens.screen_prediction_betrO3src.chat.group.screens.screen_prediction_bet_removerQ3src.chat.group.screens.screen_prediction_bet_statusrS&src.chat.group.screens.screen_settingsrn2src.chat.group.screens.screen_settings_auto_deleterp/src.chat.group.screens.screen_settings_languagerr/src.chat.group.screens.screen_settings_featuresr`%src.chat.group.screens.screen_silencerY)src.chat.group.screens.screen_silence_endr[#src.chat.group.screens.screen_speakr]src.model.Groupr src.model.GroupChatr src.model.Userr src.model.enums.Featurersrc.model.enums.Notificationrrrsrc.model.enums.Screenrsrc.model.error.CustomExceptionrsrc.model.error.GroupChatErrorrrsrc.model.pojo.Keyboardrsrc.service.devil_fruit_servicersrc.service.group_servicerrsrc.service.message_servicer src.service.notification_servicersrc.service.user_servicer DEFAULT_TYPEboolr5r2intrlistr}r)ryr8rsI\\\\\\\\\\\\******######))))))))))))++++++++++++VVVVVV\\\\\\YXXXXXLLLLLLTTTTTTRRRRRR``````\\\\\\UUUUUUYYYYYYYYYYYYYYYYYYUUUUUUccccccMMMMMMRQQQQQ______TSSSSS]\\\\\QQQQQQYYYYYYMMMMMM!!!!!!))))))++++++ *)))))KKKKKKMMMMMMMM,,,,,,GGGGGGGGGGGGGG666666>>>>>>222222C C  &C_C  C  C  CCC CCCCLLQ LQ  &LQ LQ LQ _ LQ  LQLQLQ LQLQLQLQj# __ _  &_ _ _  _  _ ____P"'+.. .  &. . .  .  . 9. ....b,U4:5F,,,,,,ry