soj*g RddlZddlZddlmZddlmZddlmZmZddl m cm cm Z ddl mZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$ddl m%Z%ddl&m'Z'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:m;Z;mde?fdZ@dejAddfdZBdejAdeCddfd ZD d:dedeCd!ed"ededzf d#ZEd$e>de?fd%ZFded!edzdeGefd&ZHdedeGefd'ZIded(e d)e>d*e j def d+ZJ d;d$e>d!ed,ededzfd-ZK d:d(e d$e>d,ededzfd.ZL d9d(e d,ededzfd/ZM d9d(e d,ede j fd0ZNd(e ddfd1ZOdejAddfd2ZPd3ed(e d4ejddfd5ZQd3NV//TV2335NNN   contextc Kddlm}m}tjtjj}||}t||d{V|j tdtj d tj dd tjtjj}|r$|j ||ns|tt j|tjd z r'|j t)||j t+||j t-|dS) z` Sends the weekly leaderboard to the group chat :param context: Context of callback r)should_reset_bounty reset_bountyNT)reset_previous_leaderboard) can_join_crew)can_accept_new_memberscan_promote_first_mater-) milliseconds)start_datetime)r3rwrxdatetimenowtimezoneutcmanage_leaderboard application create_taskrrupdateexecuter rrNCRON_SEND_LEADERBOARDrP timedeltarrr)rurwrxrr`s rrsend_leaderboardrs MLLLLLLL%.2283D3HIIC))#..O Wo 6 66666666 ##$=Y]$^$^$^___ Kd###++--- KtDIIIQQSSS&.2283D3HIIC I'' W(=(=>>>>    % ) ) + +!3!C!C!CC     I ''(>w(G(GHHH ##$J7$S$STTT ##$Nw$W$WXXXXXrtr`c ZKt||d{V}tjtjdktj}|D],}|j t||||-dS)z Manages the leaderboard :param context: Context of callback :param is_bounty_reset: Whether the bounty is reset :return: None NT) create_and_send_leaderboardr selectwhere is_activeorder_byrBascrr)rur`global_leaderboardgroupsr7s rrrrs ;7OTTTTTTTT ,....u$/FGGPPQVQYQ]Q]Q_Q_``F  '' '%I[ \ \      rtr7rcK||tdt||nd|ot|t jkrdSd|jD}td|DrdSt}||j }|j }nltj d}tj d}tjtj |ktj |ktj|ktj d|_ tj d|_ ||_||_|t-||t/|t jrt5|||jnd}|&t9|t:j|||gd{Vn t?||t j !d{V} | j|_|nC#tD$r6tGj$d t j !YnwxYw|S) a0 Creates a leaderboard list and sends it to the group chat :param context: Context of callback :param is_bounty_reset: Whether the bounty is reset :param group: The group to create the leaderboard for :param global_leaderboard: The global leaderboard :return: The leaderboard NzMIn order to create a local leaderboard, a global leaderboard must be providedc g|] }|j | S)r.0gcs rr z/create_and_send_leaderboard..s'W'W'Wr",'W'W'W'Wrtc3LK|]}t|tj V dSN)rrr]rs rr z.create_and_send_leaderboard..s3UU2%b'*=>>>UUUUUUrtrr-) external_itemfilter_by_groups)r/z%Failed to send global leaderboard to )% ValueErrorr get_languagerUget_active_usersrNLEADERBOARD_MIN_ACTIVE_USERSget_int group_chatsallr r_r^rr isocalendardeleterr7rr`savecreate_leaderboard_userscreate_leaderboard_crewsSEND_MESSAGE_LEADERBOARDget_boolrsr.rrr]r!rOrPrlogging exception) rur`r7rrr$r_r^ot_textmessages rrrrsl /7 [   1B++---MMM  u%%'' ( (3+K+S+S+U+U U U F'W'WU5F'W'W'W UUUUU U U  F--K%!&!& $$&&2244Q7 $$&&2244Q7D +"2d":K|vr8t3||t4j} || |dz }n?d} | D]`t;fd|DsCt3||t4j} || |dz }| dz } | dkrnad} | D]`t;fd |DsCt3||t4j} || |dz }| dz } | dkrnattjtjtAjktjztjdkztj|ztj|ztjdkztj} d} | D]`t;fd |DsCt3||t4j!} || |dz }| dz } | d krna|tEtj#D]>\} }t3||j$| dzt4j%} || ?tEtj#D]K\} }|j$j st3||j$| dzt4j&} || L|SdS) z Creates a leaderboard list :param leaderboard: The leaderboard to create the users for :param group: The group to create the users for :return: The leaderboard users r-NTcPg|]#}|jtjjk|j$Sr)r;rEMPERORrgr:)rrhs rrrz,create_leaderboard_users..fs9%%%   &/*A*G G G  G G GrtFrc32K|]}|jk |VdSrr:rlur:s rrrz+create_leaderboard_users..)EE"RW__2____EErtc32K|]}|jk |VdSrrrs rrrz+create_leaderboard_users..rrtc32K|]}|jk |VdSrrrs rrrz+create_leaderboard_users..rrt )'listrrrBr.is_exempt_from_global_leaderboard_requirementsrextendrget_active_user_idsrget_leaderboardr9rin_get_active_users_idslocation_levelrlevel'get_is_not_arrested_statement_conditionis_adminnot_inbountyrdescsave_leaderboard_userr PIRATE_KINGappendanyr FIRST_MATEr SUPERNOVAr8get_active_order_by_bountyr:r<rC)r$r7r9rFexcluded_user_idsprevious_leaderboardprevious_leaderboard_userseligible_leaderboard_user_idseligible_pk_usersnew_world_usersrhadded_users_countparadise_usersrgrirjr:s @rrrr7s02H$& }'+ K U4F$N O O WYY( (    !!>???   !D!F!FGGG)85(A(A#%'""%9%K" }37;tw3G3G3M3M Nd "4 4 %%48;tw3G3G3M3M GKK2244 5 54 4 % %% :%%%#'   $7$9$9$? ?;== ?}% 'w{{899 ;w~~/00  2 {Q     $+""$$ % % # #O  $ $ $0ET8_-H11   $ $%5 6 6 6 MH E %  EEEE 1EEEEE 0ET8_-D11   $ $%5 6 6 6 MH  "  A%%  EEEE 1EEEEE 0ET8_-G11   $ $%5 6 6 6 MH  "  A%%"&   $5$7$7$= =;== ?}% 'w{{899 ;w~~/00  2 {Q     $+""$$ % % " "N  EEEE 1EEEEE 0ET8_-F11   $ $%5 6 6 6 MH  "  B&& }'(J(L(LMM 7 7NE70EW\519o6M11   $ $%5 6 6 6 6'01[1]1]'^'^ 7 7 #E##(2 0E % 0 11   $ $%5 6 6 6 6  '}rtcg}tjtjdktjdkztjtj tj }t|D]\}}t}||_||_||_|dz|_|j|_|j|_||||S)z Creates a leaderboard list :param leaderboard: The leaderboard to create the crews for :return: The leaderboard crews Tr-)r rrrallow_view_in_searchrrrtotal_gained_chest_amountlimitrNLEADERBOARD_CREW_LIMITrr8r r$rX get_captainrZrFtotal_chest_amountrr)r$rVcrewsrgrXrqs rrrrs%02 $&4+D+LM N N $*//##T%C%H%H%J%J K K s)1133 4 4  !'' 3 3 t,;,=,='2$ $#'#3#3#5#5 $)AI!!%.2.L+  !12222 rtr:rFrankct}||_||_||_|j|_|j|_||S)z Saves a leaderboard user :param leaderboard: The leaderboard :param user: The user :param position: The position :param rank: The rank )rr$r:rFrgr;rr)r$r:rFrrhs rrrrsY'((#.   ("&*"k rt group_chatc||j}tjtj|ktjtjd | }|S)a Gets the current leaderboard :param index: The index of the leaderboard to get. Higher the index, older the leaderboard. :param group: The group to get the leaderboard for :param group_chat: The group chat to get the leaderboard for :return: The leaderboard Nr-) r7r rrrr_rr^roffsetfirst)rgr7rr$s rrrrs   { E) * * +"''));+;+@+@+B+B C C q    rtct||}||t|}|dS|jtj|k}|S)a1 Gets the leaderboard user for the user :param user: The user to get the leaderboard user for :param index: The index of the leaderboard to get. Higher the index, older the leaderboard :param group_chat: The group chat to get the leaderboard user for :return: The leaderboard user rN)rr9rrr:r)r:rgrr$rhs rrget_leaderboard_userrsp"%J???Kz5%e,, t(3(E(K(K$)) egg rtc&t|d|S)z Gets the current leaderboard user for the user :param user: The user to get the leaderboard user for :param group_chat: The group chat to get the leaderboard user for :return: The leaderboard user r)rgr)r)r:rs rrget_current_leaderboard_userr6s A* E E EErtcLt||}tj|S)z Gets the current leaderboard rank for the user :param user: The user to get the leaderboard rank for :param group_chat: The group chat to get the leaderboard rank for :return: The leaderboard rank r)rrget_rank_by_leaderboard_user)r:rrhs rrget_current_leaderboard_rankrCs+)ETV`(a(a(a  78H I IIrtc*t|}|dS|jtjko| }|jtjko| }|s|r'tj|_|dSdS)a A Warlord or Legendary Pirate who appears in the (global) leaderboard with their special rank, and later loses that status (duration expiring or being manually revoked), would otherwise keep showing that rank until the next leaderboard is generated and overwrites it. This checks the user's current global leaderboard entry and, if it's stale, immediately resets it to Rookie instead of waiting for the next leaderboard :param user: The user to check :return: None N) rr;rr< is_warlordrCis_legendary_pirateROOKIEr)r:rhlost_warlord_statuslost_legendary_pirate_statuss rr#reset_expired_special_rank_for_userrQs)ET(J(J #';'CC]DOOL]L]H] #';'LL +((** *!  : &:&A#  rtcdKtj|jj}t |dS)a Job callback, scheduled to run at the exact moment a Warlord/Legendary Pirate's duration ends, that resets their leaderboard rank if it's gone stale :param context: Telegram context, contains the target user's id in job.data :return: None N)r get_by_idjobdatar)rur:s rrjob_reset_expired_special_rankrms0 011D'-----rt job_queueend_datec t|tjz d}|t ||jd|jd|dS)a Schedules a one-off job to reset the user's leaderboard rank at the exact moment their Warlord/Legendary Pirate duration ends, rather than only catching it on the next leaderboard. Computed as a delay relative to now (rather than as an absolute datetime) so it lines up with however the bot's own clock already evaluates "is this user still a Warlord/Legendary Pirate", regardless of timezone configuration :param job_queue: The job queue to schedule the job on :param user: The user whose rank should be reset once their special status expires :param end_date: The datetime the user's special status ends :return: None rreset_special_rank__)whenrnameN)maxrr total_secondsrun_oncerrB timestamp)rr:r delay_secondss rr schedule_special_rank_expiry_jobrysH$5$9$9$;$;;JJLLaPPM &  W C47 C CX-?-?-A-A C C rtcKtjD]"}t|j|j|j#t jD])}|j t|j|j|j*dS)a On startup, (re)schedules the expiry job for every currently active Warlord/Legendary Pirate with a set end date. Needed because one-off jobs scheduled via schedule_special_rank_expiry_job live only in memory and are lost on every restart, so this re-creates them for any appointment that's still active and hasn't expired yet :param application: The bot application, used to access the job queue :return: None N)r get_activerrr:rr)rrirjs rr)schedule_pending_special_rank_expiry_jobsrs%''``()> gN^____+688  $ 0 ,%'7'<>N>W   rtct||}tj|}tj|}|||r|S|S)a Get the highest active rank for a given user If the user has a special rank, the special rank is returned if it's higher than the user's leaderboard rank :param user: User :param group_chat: Group chat :return: Leaderboard rank r)rrrget_special_rank is_higher)r:rrhr special_ranks rrget_highest_active_rankrsi)ETV`(a(a(a)8(T)) %4$DT$J$JL/99,GG rtcXttjS)z Gets the remaining time to the next leaderboard description :return: The remaining time to the next leaderboard description )rrNrrPrrtrr&get_remaining_time_to_next_leaderboardrs! -S-F-J-J-L-L M MMrtr)NN)rNN)r&N)Urr src.tg_compatrsrc.tg_compat.errorrsrc.tg_compat.extrrsrc.model.enums.LeaderboardRankmodelenumsr resourcesrrrNsrc.model.Crewr src.model.Groupr src.model.GroupChatr src.model.Leaderboardr src.model.LeaderboardCrewr src.model.LeaderboardUserrsrc.model.LegendaryPiratersrc.model.Userrsrc.model.Warlordrsrc.model.enums.Featurerrsrc.model.enums.Locationrr!src.service.bounty_poster_servicersrc.service.language_servicersrc.service.crew_servicerr4rrrsrc.service.devil_fruit_servicerrsrc.service.group_servicerrsrc.service.message_servicer r!r"r#intstrrs DEFAULT_TYPErboolrrrGrrrrrrrrrrrrrrrrtrrr5s!!!!!!------44444444999999999999<<<<<<<<!!!!!!))))))------555555555555555555%%%%%%++++++@@@@@@KKKKKKKKGGGGGG======;;;;;; DHaaa=@aaaaaH*YL$=*Y$*Y*Y*Y*YZ l&? RV [_    0&* VV VV V$ V 4 VVVVr999999V!V!%*T\V! /V!V!V!V!r+$:O@$(47?N?^,BF  5>44<@ /8t2)- F F  F% Ft F F F F)- J J  J% J$ J J J J d t    8 .,2K .PT . . . .#/7/@ 0($I/4NNNNNNNrt