soj^S!bddlZddlZddlZddlmZddlmZddlmZddl m Z ddl Z ddl mZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd l m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;mm?Z@ddlAmBZBddlCmDZDmEZEm?Z?mFZFddlGmHZHddlImJZJdd lKmLZLmMZMdd!lNmOZOdd"lPmQZQmRZRmSZSd#e jTd$dfd%ZU dMd(ed)eVd#e jTd*ed+eWd,eVd-eWd.eWd/eWd0eWd1e3d2eVd3eWd4ed5eWd$df d6ZXd)eYd(ed$eVfd7ZZ dNd*ed#e jTd(ed8eYeVzd9eVd:eWd;e9deWd?eWd@eWd$eWfdAZ]dBe^d$eVfdCZ_d$ejfdDZ`d$eYfdEZad$ejfdFZbdGed$eWfdHZcdIedJedKe^d$e^fdLZddS)ON)ceil)Case)Update) ContextTypes) BountyGift) BountyLoan)Crew)CrewMemberChestContribution) DavyBackFight)IncomeTaxEvent)User)BossType)BountyGiftStatus)BountyLoanStatus)get_first_new_world)Screen)DevilFruitAbilityType)IncomeTaxBracket)IncomeTaxBreakdown)IncomeTaxContributionType)IncomeTaxEventType)CommonChatException)BellyValidationException)Keyboard) get_next_runget_previous_runget_remaining_duration)add_contribution)get_ability_value)get_tax_amountget_tax_deductionsruser_has_complete_tax_deduction)reset_location)"full_message_or_media_send_or_edit) get_boss_type user_is_boss)subtract_percentage_from_value)get_unit_value_from_stringobject_to_json_stringget_belly_formattedcontextreturncKddlm}ddlm}ddlm}|||d{Vt jt jt j zd t jdz tj ktj ft jdz tjkdfg}t!d|t jdz }t j| t#t jd  t%jt$jt,jk t1jt0jt5j t jd  t jd  t9j t:jtjktjfg}t!d|t:j}t;j||d  tAj!t@j"t j!t j"t j#t@j$kk}tAjt@j#%d|D t;j&t;jd  |j'(||dS)z; Resets the bounty for all users :return: None r)force_end_all_active)'remove_all_bets_from_active_predictions)disband_inactive_crewsN)bountypending_bounty)r1T)can_create_crew)bounty_gift_tax)total_gained_bounty) chest_amountfrom_reset_chest_amounttotal_gained_chest_amountcg|] }|j S)id).0 contributions "/app/src/service/bounty_service.py z reset_bounty..sIII\III)can_promote_captain))src.service.game_servicer.src.service.prediction_servicer/src.service.crew_servicer0r updater1r2executerrequired_bountyEnvDAILY_REWARD_BONUS_BASE_AMOUNTget_intrr#rdeletewherestatusrAWAITING_CONFIRMATIONrin_rget_not_confirmed_statusesr r r7CREW_MAX_CHEST_AMOUNT_RESETr selectcrewr<usernot_in reset_level application create_task)r+force_end_all_active_gamesremove_all_betsr0 conditions case_stmtvalid_contributionss r? reset_bountyr_5s \[[[[[@?????    /' " """""""" Kt{T%88KKKSSUUU[1_ 3 5 5 E E  ! ! 1  +/S?GGII I1M *JT:t{Q77IKy!!!))+++ K%%%--///-CC giii.IKKLL giii K"""**,,, KA&&&..000##%%%   ? G G I I I  + 3 3 5 5 *J T:t'899IK ]^ giii6<>>DD#( ;ty ! ! ' '3N3S(S T T U  &((.. ' * 1 1II5HIII   giii  KD)))11333 ##$:$:7$C$CDDDDDrAFTrUamountrFshould_update_locationpending_belly_amountadd should_saveshould_affect_pending_bountycheck_for_loantax_event_typeevent_idraise_error_if_negative_bountyopponent should_taxc Kddlm}||td| | | | td| |dkrd}n|r|}ddlm}|}|5t j|j}|j |_ |j |_ |j |_ |j |_ |j }|s|xj |zc_ |r|xj ||n|z c_ |j dkr| rtjd|jd |j d |d ||nd d dt#j t't(jt/||r| ddddS|r|xj ||n|zc_ |j dkrh|dkrbtjd|jd|d||n|d|j d||nd dt#j d|_ |r||dkr|s ddddS|r ddddS|dkr:|||ndz }|}|}t7j|j |}|rt;j|}|dkrt?|stA||}||z }||z}d}| tC}||_"| j#|_$| |_%|j |_&||_'tQ||_)tQtU||_+||,rt[t\j/||||xj |z c_ |xj |z c_ | r|0}|}|D]v}tc|tdj34}|5tm|}|7||d{V||z}||z}w|xj |z c_ |r|| tqj9vr)|j:;ty||| dddn #1swxYwY|r&|td||||d{VdSdS)aF Adds a bounty to a user :param context: Telegram context :param user: The user to add the bounty to :param amount: The amount to add to the bounty :param update: Telegram update :param should_update_location: Whether to update the user's location :param pending_belly_amount: How much of the amount is from pending belly, so not newly acquired. Will be used to calculate eventual taxes :param add: Whether to add or remove the bounty :param should_save: Whether to save the user :param should_affect_pending_bounty: Whether to affect the pending bounty :param check_for_loan: Whether to check for an expired bounty loan when adding bounty :param tax_event_type: The tax event type :param event_id: The event id :param raise_error_if_negative_bounty: Whether to raise an error if the user has negative :param opponent: The opponent from which the bounty is being taken contribution if there is an active challenge bounty after the update :param should_tax: Whether to tax the bounty :return: The updated user r)update_locationNz0Amount or pending belly amount must be specifiedz6Tax event type and event id must be specified togetherT)initzUser z has negative bounty: z after removing z bounty in event None u3⚠️ NEGATIVE PENDING BOUNTY CLAMPED ⚠️ User z had pending bounty z, attempted to remove z, result would be z. Clamped to 0. Event: z Stack trace: ) tax_eventrUz.Context is required when updating the location)=src.service.location_servicerm ValueErrorsrc.chat.manage_messagernatomicr get_by_idr<r1r2r6total_gained_bounty_unmodifiedloggingwarningto_dictjoin traceback format_stackrphrasesACTION_INSUFFICIENT_BOUNTYformatr*saveerroris_arrested_permanentrget_tax_breakdownrget_amount_from_listr"r r rUvalue event_typerhstarting_amountr`r)breakdown_listr!deduction_listis_crew_memberrr CREW_CHESTget_expired_bounty_loansr'rIBOUNTY_LOAN_GARNISH_PERCENTAGE get_floatget_maximum_payable_amountintpayr get_contribution_eventsrXrYadd_dbf_contribution)rUr`r+rFrarbrcrdrerfrgrhrirjrkrmrndbrefreshed_userprevious_pending_bountynet_amount_without_pendingnet_amount_after_tax amount_to_add tax_breakdown tax_amountrr expired_loansamount_for_loansloanamount_for_repays r?add_or_remove_bountyrsKP=<<<<< ~.6KLLL"h&:("2QRRR',@1,D,D'+$$ %&%,,,,,, B CC#~dg66$+ ,;#1#E .<.[+"&"5  KK6 !KK+ ##2:FF@T##{Q#A=DG==4;====+1+=v~~'''6==!7!9!9::==*6==>QRX>Y>YZZ  ?CCCCCCCCB (    -A-I66Oc c  "Q&&+Ba+G+G @ G@@9P@@5I5Q66Wk@@(,':@@392Dfnn...& @@ &/%;%=%= @@'(#  Q;;5; gCCCCCCCCl  % % ' '  oCCCCCCCCr A::)/(<(H$$a* &$> "M7G6X(*D77M /D]SS >>*I$*O*O>!/8""J,F +R(!Z/M8>))#r22::3CCDDD PPP)&!2NOOOOOPsAA$A?>A? wager_strrequired_bellyadd_delete_buttoninbound_keyboardprevious_screens"previous_screen_list_keyboard_infoshould_validate_user_has_amountsend_error_message raise_belly_validation_exceptionc JK t|tr|} n@ t||} n.#ttf$rt t jwxYw| rD|j| kr9t t j t| |?| |kr9t t j t|n?#t $r2} | r!t|| j|||||d{V| r| Yd} ~ dSd} ~ wwxYwdS)a Validates the wager. Checks if the wager is a valid number, the user has enough belly, and if the wager is higher than the required belly :param update: Telegram update :param context: Telegram context :param user: The user to validate the wager for :param wager_str: The wager string :param required_belly: The required belly :param add_delete_button: Whether to add a delete button to the keyboard :param inbound_keyboard: The inbound keyboard :param previous_screens: The previous screens, for the back button if in private chat :param previous_screen_list_keyboard_info: The previous screen list keyboard info, for the back button if in private :param should_validate_user_has_amount: Whether to validate that the user has the amount :param send_error_message: Whether to send an error message if the wager is invalid :param raise_belly_validation_exception: Whether to raise a BellyValidationException if the wager is invalid :return: Whether the wager is valid N)rFrrrrFT) isinstancerrAttributeErrorrtrrACTION_INVALID_WAGER_AMOUNTr1rrr*ACTION_WAGER_LESS_THAN_MINr$message)rFr+rUrrrrrrrrrwageres r?validate_amountrrsF# i % % TEE T3ItDD"J/ T T T.w/RSSS T + t{U/B/B*299:Me:T:TUU   %%.*@*@*299:Mn:]:]^^  $  4 "3!1!13U        , Guuuuu" 4s,C$.C$+AB C$$ D .'DD bellyct|tjz tjzS)z Rounds the belly up to the nearest given defined amount :param belly: The belly to round up :return: The rounded belly )rrIBELLY_UPPER_ROUND_AMOUNTrK)rs r?round_belly_uprs? US199;; ;<< & . . 0 0 1rActjtjj} t t j|}t|r|S|tj dz}W)zP Get the next bounty reset time :return: The next bounty reset time Tstart_datetimeseconds) datetimenowtimezoneutcrrICRON_SEND_LEADERBOARDgetshould_reset_bounty timedelta)r next_run_times r?get_next_bounty_reset_timers &**8+<+@AAN G$  % ) ) + +N    } - - ! &);A)F)F)FF GrAc8ttS)zz Get the remaining duration to the next bounty reset :return: The remaining duration to the next bounty reset )rrr;rAr?!get_duration_to_next_bounty_resetrs ""<">"> ? ??rActjtjj} t t j|}t|r|S|tj dz }W)zX Get the previous bounty reset time :return: The previous bounty reset time Trrr) rrrrrrIrrrr)rprevious_run_times r?get_previous_bounty_reset_timers &**8+<+@AAN K,  % ) ) + +N    0 1 1 %$ $*X-?-J-J-JJ KrArun_timecttj|t jdz}|j|jkr|jdz dzdkp|jdko|jdz dzdkS)z Checks if the bounty should be reset given the run time. Bounties are reset on the last leaderboard of every even month. :param run_time: The run time :return: Whether the bounty should be reset r) millisecondsrr3r)rrIrrrrmonthday)rrs r?rrs! !%%''("4!"D"D"DDM Nm1 1 Sx~7IQ6NRS6S  ;x~1Q6!;rAsenderreceiverbase_taxcB|}t|s|t|rt|rt|nt|}|tjtjfvrdS|tjtjfvr;t|r,t|tj }|A| |r,t|tj }t|tj|}|S)z Get the transaction tax :param base_tax: The base tax :param sender: The sender :param receiver: The receiver :return: The transaction tax Nr)r&r%rADMINLEGENDARY_PIRATE PIRATE_KINGWARLORDr'rI$PIRATE_KING_TRANSACTION_TAX_DISCOUNTr in_same_crewCREW_TRANSACTION_TAX_DISCOUNTrr GIFT_LOAN_TAX)rrrtax boss_types r?get_transaction_taxrs  CF  4h9O9O 4%1&%9%9 VM& ! ! !}X?V?V  )BC C C1 8/1AB B B|TZG[G[ B0#BLLNNC  3 3H = =,S#2S2]2]2_2_`` F$9$G M MC JrA)NNNFNTFFTNNTNT)NTNNNTTF)erryr}mathrpeeweer src.tg_compatrsrc.tg_compat.extr constantsrresources.Environment EnvironmentrIresources.phrasesrsrc.model.BountyGiftrsrc.model.BountyLoanrsrc.model.Crewr %src.model.CrewMemberChestContributionr src.model.DavyBackFightr src.model.IncomeTaxEventr src.model.Userr src.model.enums.BossTyper src.model.enums.BountyGiftStatusr src.model.enums.BountyLoanStatusrsrc.model.enums.Locationrsrc.model.enums.Screenr1src.model.enums.devil_fruit.DevilFruitAbilityTyper+src.model.enums.income_tax.IncomeTaxBracketr-src.model.enums.income_tax.IncomeTaxBreakdownr0src.model.enums.income_tax.IncomeTaxContributionr-src.model.enums.income_tax.IncomeTaxEventTypersrc.model.error.CommonChatErrorrsrc.model.error.CustomExceptionrsrc.model.pojo.Keyboardrsrc.service.date_servicerrr#src.service.davy_back_fight_servicerrsrc.service.devil_fruit_servicersrc.service.income_tax_servicer r!r"rsr#src.service.message_servicer$src.service.user_servicer%r&src.utils.math_utilsr'src.utils.string_utilsr(r)r* DEFAULT_TYPEr_rboolrstrrlistdictrfloatrrrrrrr;rAr?rs ******############++++++++++++MMMMMM111111333333------============888888))))))SSSSSSHHHHHHLLLLLLVVVVVVLLLLLL??????DDDDDD,,,,,,[[[[[[[[[[XXXXXX====== 877777JJJJJJ@@@@@@@@??????aE 9aEdaEaEaEaEL)-#( $).)-+/D5D5 D5 D5 &D5  D5 ! D5  D5 D5D5#'D5D5'D5D5%)D5D5D5  !D5D5D5D5NP3PdPsPPPP."!%%)/3,0#-2HH H  &H HSy H  H  HH6lH)-H&*HH'+H HHHHV % C    GH$5GGGG&@3@@@@K(9KKKK&(t,    %      rA