sojy*6ddlZddlmZddlmZmZddlmZmZm Z m Z m Z m Z ddlm ZmZddlmZddlmZmZmZddlmZddlZddlmZdd lmZdd lm Z dd l!m"Z"dd l#m Z dd l$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z- d%dedede.dzfdZ/d%de de de%dzfdZ0de de1fdZ2d%de de de1fdZ3de de de1fdZ4 d&de de dedej5de.de dzf dZ6 d'de dedej5de de.de1f dZ7 d'de dedej5de de.de1f d Z8d!ed"e de.fd#Z9d(d$Z:dS))N)Path)SequenceOptional)Update PhotoSizeUserProfilePhotosFile ChatMemberMessage)UserChatMemberAdministrator)ChatMemberStatus) BadRequest ChatMigratedTimedOut) ContextTypes)Group) GroupChat) UnmutedUser)r )BossType) PIRATE_KING)AnonymousAdminException)get_current_leaderboard_rank)generate_temp_file_pathupdate telegram_userreturncXK||jjdS|j}|dSd} |dd{V}|jd}|d}|d{V} |ttj d{V}nr#t$re |ttj d{V}n.#t$r!}tj d|Yd}~nd}~wwxYwYnwxYwn#ttf$rYnwxYw|S)z Gets the user's profile photo :param update: Telegram update :param telegram_user: Optional Telegram user to fetch the photo for :return: The path of the user's profile photo N)limitrz1Failed to download profile photo after retrying: )effective_message sender_chateffective_userget_profile_photosphotosget_filedownload_to_drivercTG_PROFILE_PHOTO_EXTENSIONrloggingwarningAttributeError IndexError)rr photo_pathuser_profile_photoslast_set_photoslast_set_photofiletes /app/src/service/user_service.pyget_user_profile_photor6s  # / ;4- t"&J 7D7W7W8X8 8 2 2 2 2 2 2 0C/I!/L$3B$7)2244444444 Z%)%;%;'(DEE&&      JJ Z Z Z Z)-)?)?+A,HII**$$$$$$  Z Z Z XTV X XYYYYYYYY Z Z J '      s`A D-2B D D+2CD D (D?DD  D DDDD'&D'user group_chatcB|jr tjS|r tjSt ||t kr tjS|r tjS|$t |t kr tjSdS)z Get the boss type of the user :param user: The user :param group_chat: The group chat :return: The boss type N) is_adminrADMINis_legendary_pirateLEGENDARY_PIRATErr is_warlordWARLORDr7r8s r5 get_boss_typerAHs }~ !!)((#D*55DD##   ' - - < <' ' 4c*|S)z Returns True if the user is immune from combat-related losses :param user: The user :return: True if the user is immune from combat-related losses )r<)r7s r5user_has_combat_loss_immunityrDds  # # % %%rBc&t||duS)z Returns True if the user is a boss :param user: The user :param group_chat: The group chat :return: True if the user is a boss N)rAr@s r5 user_is_bossrFns z * *$ 66rBc|dSt|rdS|jsdStjtj|ktj|kduS)z Returns True if the user is muted :param user: The user :param group_chat: The group chat :return: True if the user is muted NF)rFis_mutedr get_or_noner7r8r@s r5 user_is_mutedrJysduDu  u  0D 8+:PT^:^__  rBcontext tg_group_idcK|||td||tdddlm} |2|jt |jd{VS|?|j}|j |j t|jd{VS|3|j |t|jd{VStd#t$r}|'|j}|j |_ ||=|j |j t|jd{VcYd}~S|J| |j t|jd{VcYd}~SYd}~dSd}~w|t f$rYdSt"$rYdSwxYw)z Returns the chat member of the user :param user: The user :param group_chat: The group chat :param update: The update :param context: The context :param tg_group_id: The chat id :return: The chat member Nz5update or group_chat or tg_group_id must be specifiedz4context must be specified if group_chat is specifiedr) Forbiddenz&update or group_chat must be specified) ValueErrorsrc.tg_compat.errorrNeffective_chat get_memberint tg_user_idgroupbotget_chat_memberrLstrr new_chat_idsaveget_botrr)r7r8rrKrLrNrUes r5rWrWsC"~*,1DPQQQ'/OPPP------  .99#do:N:NOOOOOOOO O  #%+E 44U5FDOH\H\]]]]]]]] ]  $ 44[#doBVBVWWWWWWWW WEFF F      !%+E ! E  JJLLL   44Q]CDXDXYYYYYYYY Y Y Y Y Y Y  ))99!-T_I]I]^^^^^^^^ ^ ^ ^ ^ ^ ^ttttt z "tt ttsJ3C+&AC+'4C+C++ G5A"F/GAF/#G/G G Gc~Kt|||||d{V}|duo|jtjtjfvS)a Returns True if the user is a member of the chat :param user: The user :param update: The update :param context: The context :param group_chat: The group chat :param tg_group_id: The chat id :return: True if the user is a member of the chat N)rWstatusrLEFTBANNEDr7rrKr8rL chat_members r5user_is_chat_memberrcs^"%4D*fgWb$c$cccccccK d " {'9B(rBc~Kt|||||d{V}|duo|jtjtjfvS)a Returns True if the user is an admin of the chat :param user: The user :param update: The update :param context: The context :param group_chat: The group chat :param tg_group_id: The chat id :return: True if the user is an admin of the chat N)rWr^rOWNER ADMINISTRATORras r5user_is_chat_adminrgs]"%4D*fgWb$c$cccccccK d " {'9&>(rBr$r"cjKt|jSjt|jSjd{V}t |dkrt t |dkrt|djjSjt fd|D}t |dkrtdjdt |dkrt t|djjS)z Get the effective Telegram user ID, even if anonymous admin :param effective_user: The effective user :param effective_message: The effective message :return: The effective Telegram user ID Nrrc4g|]}|jjk|S) custom_titleauthor_signature).0ar"s r5 z,get_effective_tg_user_id..s0!.4E4V"V"V"V"V"VrBzAdmin with title z not found) rXidr#chatget_administratorslenrr7rlrO)r$r" chat_adminsadmins_with_same_titles ` r5get_effective_tg_user_idrvs_ >$%%%$,>$%%% 8I7M7`7`7b7b1b1b1b1b1b1bK ;1%''' ;1;q>&)***)1%'''  !""a''[->-O[[[\\\ !""Q&&%''' %a(-0 1 11rBct}tjD]Y} |t t|3#t$rtj d|YVwxYwt|dkrhtj dtj|tjdkzt|dkrjtj dtjdktj|zdStj dtjdkdS)a` Syncs each user's Admin (Veteran) status with the ADMIN_USER_IDS env var. Telegram user ids listed there (space separated) are granted Admin status; any user currently marked as Admin whose id is no longer listed has it revoked. To appoint someone, add their Telegram user id to the env var. To revoke, remove it :return: None z4Invalid Telegram user id in ADMIN_USER_IDS env var: rT)r:FN)setEnvADMIN_USER_IDSgetsplitaddrXrSrOr+r,rsr rwhererTin_r:executenot_in)admin_user_idsraw_ids r5sync_admin_usersr%s #uuN$((**0022]] ]   s3v;;// 0 0 0 0 ] ] ] O[SY[[ \ \ \ \ \ ] >Q T"""(( _  0 0T]e5K L  '))) >Q U###)) ]d "t'='=n'M'M N  '))))) U###))$-4*?@@HHJJJJJs/A22!BB)N)NNN)NNNN)rN);r+pathlibrtypingrr src.tg_compatrrrr r r r TelegramUserr src.tg_compat.constantsrrPrrrsrc.tg_compat.extr constantsr)resources.Environment Environmentrysrc.model.Grouprsrc.model.GroupChatrsrc.model.UnmutedUserrsrc.model.Usersrc.model.enums.BossTypersrc.model.enums.LeaderboardRankrsrc.model.error.CustomExceptionrsrc.service.leaderboard_servicersrc.utils.download_utilsrrXr6rAboolrDrFrJ DEFAULT_TYPErWrcrgrvrrjrBr5rs%%%%%%%%YYYYYYYYYYYYYYYYGGGGGGGG444444BBBBBBBBBB******######!!!!!!))))))------------777777CCCCCCHHHHHH<<<<<<37-- -#/-4Z----`)x$8&&&&&&77t77d7777)4)- 22 22 2 & 2  2 $ 2222n)-     &     4)-     &     0-2 -25<-2-2-2-2-2`KKKKKKrB