soj ddlTddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZmZGd d eZedS) )*N) BaseModel) BountyLoan) GroupChat)User)BountyLoanSource) GameStatus)get_elapsed_hoursget_remaining_durationget_datetime_in_future_hoursceZdZUdZeedddZeezed<eedddZeezed<e Z e e zed<e e j Zee zed <eed d d dZeezed<ed Zeezed<ed Zeezed<ed Zedd dd d ZdZGddZdede fdZeded e defdZeded e defdZeded e ddfdZ ededeeffdZ!ededdfdZ"dedefd Z#de fd!Z$de%dzfd"Z&d&d#Z'd'dedefd$Z(de)fd%Z*dS)(Plunderz Plunder class plunder_challengersCASCADE)backref on_delete on_update challengerplunder_opponentsopponentwin_probability)defaultstatusTplunder_groups_chatsRESTRICT)nullrrr group_chat)r message_idbellyselfrevenge_attack_plundersNceZdZdZdS) Plunder.MetaplunderN)__name__ __module__ __qualname__db_table/app/src/model/Plunder.pyMetar#4sr*r,userreturnc|j|jjkr!tt|jSd|jz S)zz Get the win probability given a user :param user: The user :return: The win probability d)idrfloatstrrr r-s r+get_win_probabilityzPlunder.get_win_probability7s= 7do( ( (T12233 3-- -r*cttj|ktj|kzttj|ktj|kzzS)z Get the total amount of wins or losses a user has param user: The user param status: The status (won or lost) return: The total amount of wins or losses )rselectwhererrcountrget_opposite_statusr-rs r+get_total_win_or_losszPlunder.get_total_win_or_lossDs II VXX UG&$.7>V3KL M M UWWii VXX UG$,6C]C]C_C_1_` a a UWW  r*c(tttjtj|ktj|kztttjtj |ktj| kzzS)z Get the total amount of belly a user has won or lost param user: The user param status: The status (won or lost) return: The total amount of belly ) rr7fnSUMrr8rrscalarrr:r;s r+get_total_belly_won_or_lostz#Plunder.get_total_belly_won_or_lostXs NN266'-00 1 1 7 7#t+&0HI   &(( NN266'-00 1 1 7 7!T)gn@Z@Z@\@\.\]   &(( r*cttj|ktj|kztj}ttj |ktj| kztj}t||dS)z Get the plunder with the max belly won or lost param user: The user param status: The status (won or lost) return: The plunder c|jSN)r)fs r+z-Plunder.get_max_won_or_lost..s UVU\r*)key) rr7r8rrorder_byrdescfirstrr:max)r-rmax_plunder_as_challengermax_plunder_as_opponents r+get_max_won_or_lostzPlunder.get_max_won_or_lostks II VXX UG&$.7>V3KL M M Xgm((** + + UWW " II VXX UG$,6C]C]C_C_1_` a a Xgm((** + + UWW ,.EK\K\]]]]r*cttjttjdtj|ktj td  }|j }ttjttjdtj|ktj td  }|j }||kr |j|fS|j|fS)z Get the most plundered user and the amount of plunders param user: The user return: The most plundered user and the amount of plunders r9)rr7rr>COUNTaliasr8rgroup_byrHSQLrIrJr9)r-!most_plundered_user_as_challengeramount_as_challengermost_plundered_user_as_opponentamount_as_opponents r+get_most_plundered_userzPlunder.get_most_plundered_usersW II VG$bhhw/?&@&@&F&Fw&O&O P P U7%- . . Xg& ' ' Xc'll'')) * * UWW * AF II VG&1C(D(D(J(J7(S(S T T U7#t+ , , Xg( ) ) Xc'll'')) * * UWW (=B  4 4 42=?QQ Q09;OOOr*c<ttj|ktjt jkztj  S)zp Get the plunder with the max sentence param user: The user return: The plunder ) rr7r8rrr LOSTrHsentence_durationrIrJ)r-s r+get_max_sentencezPlunder.get_max_sentencesa II VXX UW'4/GNjo4UV X X Xg/4466 7 7 UWW  r*c4|j|kr|jn|jS)zb Get the other opponent param user: The user return: The opponent )rrr4s r+ get_opponentzPlunder.get_opponents#'/T"9"9tt}Lr*c*t|jS)zT Get the status of the game :return: The status of the game )r rr s r+ get_statuszPlunder.get_statuss $+&&&r*ctjtjtjktj|jkS)z Get the loan :return: The loan, or None if no loan was created for this plunder (e.g. the loser had combat-loss immunity, so nothing was owed) )r get_or_nonesourcerPLUNDER external_idr1r`s r+get_loanzPlunder.get_loans6%  !1!9 9:;QUYU\;\   r*cRttj|kS)zN Get the revenge plunder :return: The revenge plunder )rrcin_revenge_to_plunderr`s r+get_revenge_plunderzPlunder.get_revenge_plunders ""7#@D#HIIIr*c|j|krdSt|jtjkrdS|jdS|dSdS)a Check if a plunder can be revenged. User can revenge a plunder if they were attacked less than x time ago and have not already revenged that plunder :param user: The user :return: True if the user can revenge the plunder, False otherwise FNT)rr dateEnv$FIGHT_PLUNDER_REVENGE_DURATION_HOURSget_intrirjr4s r+ can_revengezPlunder.can_revengeso ?d " "5 TY ' '#*R*Z*Z*\*\ \ \5  % 15  # # % % 15tr*c~tttj|jS)zd Get the remaining duration for the revenge :return: The remaining duration )r r rmrnrorlr`s r+get_revenge_remaining_durationz&Plunder.get_revenge_remaining_durations9 & (8@@BBDI     r*)r.rrD)+r%r&r'__doc__ForeignKeyFieldrr__annotations__r FloatFieldrr2SmallIntegerFieldr IN_PROGRESSrboolrr IntegerFieldrintBigIntegerFieldrr[rir!r,r5 staticmethodr<rArNrXr\r^rarrgrjrpr3rrr)r*r+rrs*9 +yI***J&(7 )Y)(((Hd_$+5*,,OUZ'666'8'8AW'X'X'XFD$ $XXX.=o & ///J O+&2\t%<%<%rs######))))))++++++))))))======111111u u u u u iu u u pr*