sojAY ddlZddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1m2Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?ddl@mAZAmBZBmCZCmDZDmEZEmFZFmGZGddlHmIZImJZJmKZKdd lLmMZMdd!lNmOZOdd"lPmQZQmRZRmSZSmTZTmUZUmVZV dLd#ed$e!d%e7d&eWd'ed(df d)ZXd#ed(eYefd*ZZ dMd#ed,e[d(eWfd-Z\dNd.e5d(eYefd/Z]d(e[fd0Z^d1ej_d(dfd2Z`d3ed1ej_d4e!d5ed(df d6Zad#ed(dfd7ZbdOd#ed9e[d(dfd:Zcd1ej_d(dfd;Zd dOd4e!de[d(eef d?Zfd4e!ds" V,<,ACSCX+YYY!'+ _;4466 _ _UM] _ _   !''''+ U   !K)3K+ 'B  + 3 3 5 5' ' #'/lnn # -<->->$/!!)%%4>"*,..)+66666cjtjtj|kS)zj Get the devil fruit abilities :param devil_fruit: The devil fruit :return: The abilities )r selectwherer5r5s rQget_devil_fruit_abilitiesrX|s*  # % % + +,=,I[,X Y YYrST add_headerct|}|r tjnd}|jr|tjz }n|s|jrr|D]n}t |j}|tj tj | | |jz }on|tjz }|S)a Get devil fruit abilities text :param devil_fruit: The devil fruit :param add_header: Whether to add the header :param always_show_abilities: Whether to always show the abilities, even if the Devil Fruit has never been eaten :return: The text )rXrDEVIL_FRUIT_ABILITY_TEXT is_defective#DEVIL_FRUIT_ABILITY_DEFECTIVE_SMILEshould_show_abilitiesr ability_typeDEVIL_FRUIT_ABILITY_TEXT_LINEformatr LOG_POSITIVEget_descriptionget_signvalueDEVIL_FRUIT_ABILITY_UNKNOWN)r5rYalways_show_abilities abilitiesabilities_textabilityr`s rQget_devil_fruit_abilities_textrls*C;)O)OI>HP'::bN >'EE  >+"C >   G2GH\2]2]L gCJJ",,..%%''  NN  '== rScategoryctjtjt j}|r#|tj|k}t|S)zx Get all devil fruits in circulation :param category: The category (optional) :return: The devil fruits ) r rUrVrDin_rget_released_statusesrmlist)rmquerys rQget_devil_fruits_in_circulationrsso     % %.DFFGG  E= J/8;<< ;;rScdS)z Checks if a Devil Fruit should be released based on the number of Devil Fruits in circulation and the number of active users :return: Whether a Devil Fruit should be released TrurSrQshould_release_devil_fruitrvs  4rScontextcPKtsdStjtjt jktjtj kztjtj kz tj }|sKtjd}t%||d{Vt'j|dSt+|ddS)zF Schedule devil fruit release :param context: The context Nz"Any (non-Mythical Zoan, non-Smile)Tis_new_release)rvr rUrVrDrENABLEDrmr MYTHICAL_ZOANSMILEorder_byidasc get_or_nonerNO_DEVIL_FRUIT_TO_SCHEDULErbr*loggingerrorset_devil_fruit_release_date)rwr5ot_texts rQschedule_devil_fruit_releasers  & ' '    "2": :  $6$D D F  $6$< < >  *-##%% & &  4;;<`aa))))))))) g TBBBBBBrSupdateuser group_chatc Kddlm}ddlm}|}t |jsdS|5}tj tj tj k tj}|s ddddSt%|j} d| z dz} n#t&$rd} YnwxYwt)| s>t+t,j|_ ddddSt2j|t;|t?|j !tE|d} tG|gg} ||_$tK||tLj'||_tQ|tRj*| | |d d d{V} | j+|_,|-nQ#t\t^f$r=}|0tcj2d |j3d |Yd}~nd}~wwxYwddddS#1swxYwYdS) z Release a Devil Fruit to a user :param update: The update :param context: The context :param user: The user :param group_chat: The group chat :return: None r)init)get_next_bounty_reset_timeNdFrhT)saved_media_namecaptionkeyboardr new_messagequote_if_groupz!Error giving devil fruit to user z: )4src.chat.manage_messagersrc.service.bounty_servicerr((devil_fruit_collection_cooldown_end_dateatomicr rUrVrDr SCHEDULED for_updater~ release_daterrr'ZeroDivisionErrorr1r&rG%DEVIL_FRUIT_COLLECT_COOLDOWN_DURATIONrIr DEVIL_FRUIT_RELEASE_MESSAGE_INFOrbget_markdown_mentionr+r@rrmrdrlget_manage_deeplink_keyboardrelease_group_chatrRrrEr,rDEVIL_FRUIT_NEW message_idrelease_message_idrMrr!rollbackrrrA)rrwrrrrdb transactionr5minutes_to_release probabilitytextinline_keyboardmessagees rQrelease_devil_fruit_to_userrs-,,,,,EEEEEE B dK L L ?V      U:$(8(BB C C Z\\ Xj-1133 4 4 []]   ?V?V?V?V?V?V?V?V#@ @X"Y"Y "#&8"8C!?KK    KKK k** >  % % ' ' ' (A(A(C(C D D {3 4 4 D D F F *;e T T T   3O{2[2[1\0] V .8K * $[$8H8L M M M}|j }t|t|}t!|||d{V?dS)zO Respawn Devil Fruit :param context: The context :return: None N)r rUrVrDrrCEATENrmrr}rJrrKr=rrr-)rw devil_fruitsr5r= notifications rQrespawn_devil_fruitrts&0%6%8%8%>%>  "2"< <"&6&<<&*<*BBD   % 7 9 & &L$>> !'$[1115[AA  ==========>>rSr`rf add_to_valuecg}tj||}||||r/|j}|||t|dkr|S|}ttd|D}|tj kr|r||zSt||S|r||z St||S)a  Given a value, gets the updated value if user has eaten a Devil Fruit that modifies it :param user: The user :param ability_type: The ability type :param value: The value :param add_to_value: Whether to add to the value :return: The value Nrcg|] }|j Sru)rf).0rks rQ z%get_ability_value..s&N&N&Nw}&N&N&NrS)r get_user_abilityappendis_crew_membercrewextendget_active_abilitylenrer3r2DevilFruitAbilityTypeSignPOSITIVEr/r0) rr`rfrridevil_fruit_abilityrability_type_sign ability_values rQget_ability_valuers&*,I.?-OPTVb-c-c&,--- @Y00>>??? 9~~ 3?3H3H3J3J+%&N&NI&N&N&NOOM 5>>>  A=( (*5-@@ @%}$$ )% ? ??rShourscBt|||}t|S)z Given a value, get the updated datetime if user has eaten a Devil Fruit that modifies it :param user: The user :param ability_type: The ability type :param hours: The hours :return: The value )rr&)rr`r new_hourss rQget_ability_adjusted_datetimers#)|UCCI ' 2 22rSctjtj|ktjt jkzS)z{ Check if user has eaten a Devil Fruit :param user: The user :return: Whether user has eaten a Devil Fruit )r rUrVr=rDrrexists)rs rQuser_has_eaten_devil_fruitrsD    D(Z->BRBX-XY Z Z rSuserscKttjdz }|D]9}| |3|j|vr*t ||jt |d{V:dS)z Warn inactive users with eaten Devil Fruits :param context: The context object :param users: The users to warn. If provided, it will only warn these users, else it will warn all inactive users :return: None rNrW)*get_inactive_users_with_eaten_devil_fruitsrG6DEVIL_FRUIT_MAINTAIN_MIN_LATEST_LEADERBOARD_APPEARANCErIr=r-r)rwrinactive_users_devil_fruitsr5s rQ*warn_inactive_users_with_eaten_devil_fruitrs#M BJJLLqP##3 =U.;3D3M3M#!3 LLL       rSlatest_leaderboard_appearancecLtj|}tjt tttj tj ktj |z}t|}tjt tj tj kt jdkzt jdkzt j t'jzt j t+jztj d|Dz}t|S)z Find Devil Fruits eaten by users that have not appeared in the latest N leaderboards :param latest_leaderboard_appearance: The latest leaderboard appearance :return: The inactive captains )nFcg|] }|j Sru)r)rdfs rQrz>get_inactive_users_with_eaten_devil_fruits..!s$O$O$OrRU$O$O$OrS)r get_latest_nr rUdistinctjoinrrrVrDrrrroexecuterqis_admin.is_exempt_from_global_leaderboard_requirementsnot_inrget_active_user_idsr)rlatest_leaderboardsrreaten_active_devil_fruitsinactive_devil_fruitss rQrrs.9-E '...   d o   k     "2"8 8~!!"566 8    37u++   d   "2"8 8}% 'BeK Mw~~oACCDD Fw~~g9;;<<  > }##$O$O5N$O$O$OPP  R    % & &&rScKttj}|D]=}|j}t |t ||t|d{V>dS)zk Revoke Devil Fruits from inactive users :param context: The context object :return: None rWN)rrGrrIr=rr-r)rwrr5r=s rQ&revoke_devil_fruit_from_inactive_usersr)s#M BJJLL##3  !'$[111 U8[QQQ            rScbt}tj|_t t j|_d|_| t}||_ tj |_tdd|_| |S)z1 Generate a SMILE :return: The SMILE Trr)r rr}rmr.rANIMALSnamer_rMr r5rget_random_abilityr`r4rf)r5rks rQ create_smiler@s )llK-3K/ 0ABBK(,K%"3!4!4G%G0CEEG"1c**GM LLNNN rSadd_sell_commandc6d}d}d}|tjus |tjur|jetj t|jd}|tjur|tj z }d}ntj d|j|tjur|r|dz }|tjz }|r,|tjur tj}t%|d}tj |t |j|||S) z Get the recap text :param devil_fruit: The devil fruit :param add_sell_command: Whether to add the sell command :return: The text r[FNT)add_remaining_timez9Devil Fruit %s in collected status has no expiration date r)rrrCrrr}rJr*DEVIL_FRUIT_ITEM_DETAIL_TEXT_EXPIRING_DATErbr)"DEVIL_FRUIT_EXPIRATION_EXPLANATIONrrr(DEVIL_FRUIT_SMILE_EXPIRATION_EXPLANATIONr)DEVIL_FRUIT_ITEM_DETAIL_TEXT_SELL_COMMANDrlDEVIL_FRUIT_ITEM_DETAIL_TEXTr@rmrd)r5rexpiring_date_textsell_command_text&has_standard_expiring_date_explanationrjs rQget_recap_textrVs-2*   $4$>>>  # # % %);)A A A  & 2!(!S!Z!Z' (CX\]]]"" %%''+;+EEE"g&PP"9=6 MK[^    !!%7%=== 1 ' $ & gNNNK2244r/s))))))))------******######++++++999999555555))))))------555555555555%%%%%%//////'''''' FEEEEE999999))))))NMMMMMIIIIIIIIIIIISSSSSSIIIIII,,,,,,`_________>>>>>>999999)- ;7;7;7;7 ;7  ;7 ' ;7  ;7;7;7;7|Z:Z$?P:QZZZZMQ)-D  .@ DQ[L\     DC 0ICdCCCC>SV SV)6SV>BSVPYSV SVSVSVSVlCJC4CCCC(&&j&$&[_&&&&R>|'@>T>>>>:Y^+@+@ +@3+@