soj ddlZddlmZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZddlmZmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6m7Z7ddl8m9Z9m:Z:ddl;mZ>ddl?m@Z@mAZAmBZBddlCmDZD dRd!ejEd"e d#eFd$dfd%ZG dSd"e d'eFd(ed)eFd$eHf d*ZI dTd"e d(ed+eJed,eFd'eFd$eHf d-ZKd!ejEd"e d$dfd.ZLd!ejEd"e fd/ZM dUd"e d0ed$eJefd1ZNd!ejEd"e d$dfd2ZO dUd!ejEd"e d3e d$dfd4ZPd"e d(ed$eJefd5ZQd"e d$eJefd6ZR dVd7ed8eJed9eFd$eSfd:ZTd!ejEd$dfd;ZUd!ejEd$dfd<ZVd!ejEd$dfd=ZWd"e d(ed$eFfd>ZXd"e d(ed$efd?ZYd"e d(ed$eSfd@ZZ dUdAed(edBeSdCe d$ef dDZ[d(ed7ed$dfdEZ\d(edAed$dfdFZ] dWd7ed"e d(ed$eSfdGZ^d3e dHeSd$e dzfdIZ_ dXd!ejEdJeHd"e dKeFd3e f dLZ` dYd!ejEd"e dJeHdKeFd3e f dMZad"e d$e-fdNZbd"e fdOZcd!ejEd(ed"e d$eHfdPZdd"e d3e d$eFfdQZedS)ZN)datetime) BadRequest TelegramError) ContextTypes) GroupChat) Prediction)PredictionGroupChatMessage)PredictionOption)PredictionOptionUser)User)ContextDataKey)Emoji)Feature)PredictionResultNotification PredictionBetInvalidNotification)PredictionStatus!get_prediction_status_name_by_key)PredictionType)Screen)DevilFruitAbilityType)IncomeTaxEventType)PredictionException)ContextDataValue)Keyboard)round_belly_upadd_or_remove_bounty)default_datetime_format)get_ability_value)0broadcast_to_chats_with_feature_enabled_dispatchsave_group_chat_error)escape_valid_markdown_charsfull_message_send)send_notification)'get_random_user_context_inner_query_key)get_percentage_from_valueget_value_from_percentageadd_percentage_to_value)get_belly_formattedFcontext prediction is_resentreturnc ~K|sbt|jtjurttjtj|_tj|_ | t|}t|tj||t|ggd{VdS)z Send prediction :param context: Telegram context :param prediction: Prediction :param is_resent: If prediction was already sent ) external_iteminline_keyboardN)rstatusNEWrphrasesPREDICTION_NOT_IN_NEW_STATUSSENTrnow send_datesaveget_prediction_textrr PREDICTIONget_prediction_deeplink_button)r)r*r+texts &/app/src/service/prediction_service.pysendr=/s . J- . .6F6J J J%g&JKK K,1 '|~~ OO z * *D :  8DDEF    Tadd_bets_commanduseris_creator_recapc  |rd}|jduo ||jk}|o|tju}|s|jduo|jduo ||jk}d}d}|s|s|j}t |} td| D}t|D]\} fd| D} td| D} |tj j tjt| |djr t"jndz }n|jNt|jD]8\} } |tj | d zt| z }9nV|j}t|D]?\} |tj j tjz }@||jd ndz }d}|s||j2tj t/|j|ndz }|rx|tjurX|tj |jt/|j|n tjz }|tjz }n||j2tj t/|j|ndz }n|tj |jt/|j|n tjz }|tj |jt/|j|n tjz }d}t?|j t>j!ur|j"rTt"j#}|tI|| }n|j%}tj& tO|}nt"j(}d}|tj) ||z }|j*r t"j#n t"j(}|tj+ |z }|r|tj,z }||rd ndz }|j-r t"j#n t"j(}|tj. |z }|r|tj/z }|r]||rd ndz }|j0r t"j#n t"j(}|tj1 |z }||r tj2ndz }d}d}|r?t|j3tj4urtj5}|j-r tj6}|dkrd |znd}||z }d}t?|j t>j!ur|s tj7}|ji|sg|setj8 t|j9|tO|tut|j3||||}n8||z }tj; t|j9||}|S) aw Get prediction text :param prediction: Prediction :param add_bets_command: If to add commands on how to place and remove bets :param user: User, if called from private chat and prediction max refund should be customized by DF ability :param is_creator_recap: If the prediction is being sent to the creator as a recap :return: Prediction text FNrc3$K|] }|jV dSNwager.0prediction_option_users r< z&get_prediction_text..os7  -C " (      r>c>g|]}|jjjk|S)prediction_optionid)rIrJrNs r< z'get_prediction_text..ts;CCC*);>BSBVVV'VVVr>c3$K|] }|jV dSrErFrHs r<rKz&get_prediction_text..ys71G&,r>) add_decimal r*r@)<creator get_statusrr1rOprediction_optionsget_prediction_options_userssum enumerater2&PREDICTION_TEXT_OPTION_WITH_PERCENTAGEformatnumberr!optionr% is_correctrCORRECToptionsPREDICTION_TEXT_OPTIONend_datePREDICTION_CLOSING_DATEr BETS_CLOSEDPREDICTION_CREATE_CUT_OFF_DATE cut_off_date TEXT_NOT_SET"PREDICTION_CUT_OFF_DATE_HOW_TO_SETPREDICTION_CUT_OFF_DATEPREDICTION_CREATE_CLOSE_DATErtypeUSER refund_wagerPREDICTION_FEATURE_ENABLEDget_max_wager_refundmax_refund_wagerPREDICTION_WAGERS_REFUNDED_MAXr(PREDICTION_FEATURE_DISABLEDPREDICTION_WAGERS_REFUNDEDallow_multiple_choices PREDICTION_MULTIPLE_BETS_ALLOWED,PREDICTION_MULTIPLE_BETS_ALLOWED_DESCRIPTIONcan_withdraw_betPREDICTION_CAN_WITHDRAW_BETS(PREDICTION_CAN_WITHDRAW_BETS_DESCRIPTION is_publicPREDICTION_IS_PUBLIC PREDICTION_IS_PUBLIC_DESCRIPTIONr0r4PREDICTION_BET_HOW_TO_PLACE_BET%PREDICTION_BET_HOW_TO_REMOVE_ALL_BETSPREDICTION_USER_DISCLAIMERPREDICTION_TEXTquestionrPREDICTION_CREATE_RECAP)r*r?r@rA is_creatoris_creator_but_not_sent options_text total_wagerrXprediction_options_usersindexprediction_option_users option_wagerr_ added_text optional_text enabled_emojirrmax_refund_texthow_to_bet_command_texthow_to_remove_bet_command_textuser_prediction_disclaimerprediction_textrNs @r<r8r8Msm ! #4/NDJFB  LL  = !*:+=!>!>   v > E EAI/77!!   :D9V ,56H,I,I  (( > E E%,/0A0HII!!   3 ?RG J $ ".  + 2 2' (;TBB       *//115E5QQQ '@GG*6( (?FFF) J 'D DJJ *6/66+J,CTJJ  JJ g:AA". $J$7 > > >%   g<CC&2 $J$;T B B B%   Mjo&&n.AAA  " !! %DKK#$455OO"=M O;BB=Rabbb  , /((  . W=DD]SSSMNMM -5TT25M  & /((  . W9@@OOOMJII ^!19r9 # 3E , ,2  5<<]KKK EU]AA[]] !%'"_ J- . .2B2G G G&-&M #* _181^.-:R,?,?D=((RM-J"$jo&&.*===j=%,%G"} )@ IY !188 ' (; < <   , , -.>z?P.Q.Q R R  # * &    00 !9@@ ' (; < prediction_options_user add_headerc t|jtjurdS|t||}t |dkr t jSd}d}d}t|j}|r t jnd} |D]'} | j} | t j | j t| j t| jz } |tju} t#| | } t| }|tjur#| t j |z } | jr-| t j |z } || z }|dz }| t jz } ||js| jndz}|dz }|jr| t jz } )t |dkr|tjur|dkrf|dkr0| t j t|z } n| t j t|z } nt5|}|dkr0| t j t|z } nV| t j t|z } n&|r$|tjur|jr| t jz } |r$|tjur|jr| t j z } | S)aW Get the prediction status text :param prediction: The prediction :param user: The user :param prediction_options_user: The options the user has bet on :param add_header: Whether to add the "Bets" header :param add_bets_command: Whether to add the command to remove all bets :return: The prediction status text rCNr) is_potentialrS)!rr0r1get_prediction_options_userlenr2PREDICTION_BET_USER_HAS_NOT_BETPREDICTION_STATUS_BETS_HEADERrNPREDICTION_STATUS_OPTIONr]r^r!r_r(rG RESULT_SETget_prediction_option_user_winPREDICTION_STATUS_POTENTIAL_WINr`PREDICTION_STATUS_RESULT_WINPREDICTION_STATUS_RESULT_LOSTro&PREDICTION_STATUS_RESULT_LOSS_REFUNDEDPREDICTION_STATUS_NET_WINPREDICTION_STATUS_TOTAL_WINabsPREDICTION_STATUS_NET_LOSSPREDICTION_STATUS_TOTAL_LOSSr4ry PREDICTION_BET_HOW_TO_REMOVE_BETr)r*r@rrr?total_win_amounttotal_win_counttotal_loss_countprediction_statusot_textrJrNrpotential_win_amountpotential_win_amount_formattedtotal_loss_amounts r<get_user_prediction_status_textrsv" )**.>.BBBr&"=j$"O"O "##q((66O*::;L*M*M7AIg33rG"9)N)N.D.V 73::  $ '(9(@ A A  6 < = =   )0@0KK = "   *==Q)R)R& $4$? ? ? w>EE. GG !+ N7?FF2!$88 1$7@@ 8B8OV*00UV !A% *NwMMG "##a''  0 ; ; ;1$$#a''w@GG+,<==  GGwBII+,<==  GG%((8$9$9!"Q&&wAHH+,=>>  GGwCJJ+,=>>  GG  @!%5%:::+; w? ?G A !1!6 6 6  ' 7 7@@ Nr>cKt|jtjurttjt ||d{Vtj|_tj |_ | t||d{Vt|tj|d{VdS)z[ Close bets :param context: Telegram context :param prediction: Prediction N)rr0r4rr2PREDICTION_NOT_IN_SENT_STATUScut_off_invalid_betsrfrr5rdr7refresh9send_prediction_status_change_message_or_refresh_dispatchPREDICTION_CLOSED_FOR_BETS)r)r*s r< close_betsrs  )**2B2GGG!'"GHHH w 3 33333333)4J",..JOO ': & &&&&&&&& D3Z  r>cHKi}|jt|}|D]}|j}t||jdd{V||j|vr |gdg||j<||jd|||jdxx|jz cc<|| D]E\}}|d}|d}|d} t||| } t||| d{VFdS)z Cut off invalid bets (bets placed after cut off date), notifying the users :param context: Telegram context :param prediction: Prediction :return: None NTshould_affect_pending_bountyrrS) rhget_invalid_betsr@rrGr7rOappenddelete_instanceitemsrr#) r)r* users_invalid_prediction_optionsinvalid_prediction_option_usersinvalid_prediction_option_userr@user_idvaluer total_refund notifications r<rrs\^$*FVWaFbFb'.M = = *7>>=A2qM09 ,TW 5a 8 ? ?@^ _ _ _ ,TW 5a 8 8 8rhc||jtd|j}tjtj|ktj|kzS)z Get invalid bets (bets placed after cut off date) :param prediction: Prediction :param cut_off_date: Cut off date :return: List of invalid bets Nz>cut_off_date cannot be None if prediction.cut_off_date is None)rh ValueErrorr selectwherer*date)r*rhs r<rrsf  " *]^^ ^!.  & ( ( . .  (J 6  $| 3 5  r>cKt|jtjurttj|j}d|D}t|}i}|D]X}|j}|j |vr |dgg||j <|j }||j d ||j r[t||} t|| |jt j|j d{V||j dxx| z cc<n ||j dxx|jzcc<|jst'|dkr[t'|dkr|j} n#t)|jt+|} t|| |j d{V|Ztj|_t1j|_|t7||d{Vt9|tj|d{V|D]G\} } | d}| d} | d}t?|||| |}tA|||d{VHdS) z\ Set results :param context: Telegram context :param prediction: Prediction c g|] }|j | SrM)r`)rIrNs r<rPzset_results..s1:::   '::::r>rr)r)pending_belly_amounttax_event_typeevent_idNrS)rJ)r)!rr0rfrr2$PREDICTION_NOT_IN_BETS_CLOSED_STATUSrXrYr@rOrNrr`rrrGrr9rorminrqr7rrr5result_set_daterrPREDICTION_RESULTS_SETrrr#)r)r*rXprediction_options_correctrusers_total_winrJr@rN win_amount refund_amountrr total_winuser_prediction_optionsrs r< set_resultsrsq  )**2B2NNN!'"NOOO1;1N::!3::: group_chatcKt|jtjurttjt ||d|d{VdS)z Refresh prediction :param context: Telegram context :param prediction: Prediction :param group_chat: Group chat, refresh only for this group chat T)r*should_refreshrN)rr0r1rr2PREDICTION_NOT_SENTr)r)r*rs r<rrDs{ )**.>.BBB!'"=>>> CJt    r>ctjttj|ktj|kztj }|S)z Get all prediction options for a user :param prediction: Prediction :param user: User :return: List of prediction options user ) r rjoinr rr*r@order_byr^asc)r*r@results r<rrUsr #%%     ! , :?S?X\`?` a   ")--// 0 0   Mr>cjtjtj|kS)z Get all prediction options users for a prediction :param prediction: Prediction :return: List of prediction options users )r rrr*)r*s r<rYrYis+ & ( ( . ./C/NR\/\ ] ]]r>rJrrc8|tj}td|D}|std|D}ntfd|D}tj|}t t ||}|S)ad Get prediction option user potential win :param prediction_option_user: PredictionOptionUser for which to get potential win :param prediction_options_users: List of PredictionOptionUser for the same prediction :param is_potential: If True, assume that the prediction option is correct :return: Prediction option user potential win Nc3$K|] }|jV dSrErFrHs r<rKz1get_prediction_option_user_win..s7)?$r>c3<K|]}|jj|jVdSrE)rNr`rG)rIpous r<rKz1get_prediction_option_user_win..sC" " S=R=]" I" " " " " " r>c3FK|]}|jjk|jVdSrE)rNrG)rIrrJs r<rKz1get_prediction_option_user_win..sC" " $(>(PPP IPPPP" " r>)rYr*rZr%rGrr&)rJrrrtotal_correct_wagerpercentage_of_correct_wagervalue_from_total_wagers` r<rrrs '#?@V@a#b#b C[K   !" " !9" " "   "" " " " /" " "   #<$&9## ,!+/JKK "!r>c@Ktjtjdtjt jkztjtj kz}|D]}t||d{VdS)zI Send scheduled predictions :param context: Telegram context FN) rrrr6is_nullrr5r0rr1r=r) predictionsr*s r<send_scheduled_predictionsrs%/$5$7$7$=$=   % %e , ,  8<>> 1 3   0 4 4 6%%K "(( 7J''''''''''((r>c@Ktjtjdtjt jkztjtj kz}|D]}t||d{VdS)zJ Close scheduled predictions :param context: Telegram context FN) rrrrdrrr5r0rr4rrs r<close_scheduled_predictionsrs%/$5$7$7$=$=   $ $U + +  (,.. 0 2   0 5 5 7%%K ".. *----------..r>cKtjtjg}tjtj|}|D]i}tj tj |k t|tj|d{VjdS)zj Remove all bets from active predictions due to bounty reset :param context: Telegram context N)rr4rfrrrr0in_r deleter*executerr2,PREDICTION_ALL_BETS_REMOVED_FOR_BOUNTY_RESET)r)active_statusesrr*s r<'remove_all_bets_from_active_predictionsrs(,.>.JKO$.$5$7$7$=$=o..%%K"  #%%++ +z 9  ')))G WI:            r>cDtt||dkS)z Check if user bet on prediction :param prediction: Prediction :param user: User :return: True if user bet on prediction r)rrrUs r<user_has_bet_on_predictionrs" *:t<< = = AAr>cZt|j}t||rb|tjur tjS|tjur tjSt||}|dkr tj Stj S|tj ur tj Stj S)z Get user prediction status emoji for a user :param prediction: Prediction :param user: User :return: User prediction status emoji r)rr0rr4rPREDICTION_OPENrfPREDICTION_CLOSEDget_prediction_net_win LOG_POSITIVE LOG_NEGATIVEr1PREDICTION_NEWNULL)r*r@rnet_wins r< get_user_prediction_status_emojir s)):;;!*d33  0 5 5 5( ( "2"> > >* *,Z>>G{{)))) .2 2 2##zr>ct||}t|jtjurt d|jdd}|D]-}|j}|jr|t|z }#||j z}.|S)z Get net win for a user, without considering refunds, only for predictions with result set :param prediction: Prediction :param user: User :return: Net win z Prediction z4 results are not set, so net win can't be calculatedr) rrr0rrrOrNr`rrG)r*r@rrrrNs r<rrs;VD;; )**2B2MMM ]*- ] ] ]   G&!!.1.C  ' ! 5c:: :GG sy GG Nr>rNrGprediction_group_chat_messagecKtjtj|ktj|kz}||xj|z c_n[t}|j|_||_||_||_t |tjdd}|dkr|nd|_ ||nd|_ tj |_ |t||ddd{V|S)a Save the prediction option user :param prediction_option: The prediction option :param user: The user :param wager: The wager :param prediction_group_chat_message: The prediction group chat message :return: The prediction option user NrT) add_to_valueF)addr)r get_or_noner@rNrGr*rrPREDICTION_WAGER_REFUNDmax_refund_wager_boostr rr5rr7r)rNr@rGr rJrs r<save_prediction_option_userrs<4H3S  "d *  15F F H44 )$$-$$$"6!7!7,=,H)3D0&*#',$!2 '?QU" " " '=q&@&@ " "a 5 *G)R%%X\8#+,..!!! tUTX Y Y YYYYYYYY !!r>cK|jj}t||j| d{V|dS)z Delete a prediction option user :param user: The user object :param prediction_option_user: The prediction option user :return: None rN)rNr*rrG is_result_setr)r@rJr*s r<delete_prediction_option_userrFs4EPJ  $)3)A)A)C)C%C   **,,,,,r>cKtjtj|ktj|kz}|t ||d{VdSdS)z Delete a prediction option for a user :param user: The user object :param prediction_option: The prediction option :return: None N)r rr@rNr)r@rNrJs r<!delete_prediction_option_for_userr^sr4H3S  "d *  15F F H44 )+D2HIIIIIIIIIII*)r>c|J||F||tdtjtj|ktj|kz}|-t t |tj|j S||j}t t|j |j }t|j |S)a Get the maximum wager refund for a prediction option user, considering the Devil Fruit ability boost :param prediction_option_user: The prediction option user :param prediction: The prediction :param user: The user :return: The maximum wager refund Nz>Prediction option user or prediction and user must be provided) rr rr@r*intrrrrrr'rmax)rJr*r@max_refund_boosteds r<rqrqps%:+=  ]^^ ^7K7W ! &$ .3G3RV`3` a8 8 % +CZE`     !7!B   ')?)V   z*,> ? ??r> message_idc~tjtj|ktj|kz}|dS|jS)z Gets the prediction from the message id :param group_chat: The group chat :param message_id: The message id :return: The prediction N)r rrrr*)rrr s r<get_prediction_from_message_idrsK #. ' 2j @)4 B D  "%,t ( 33r>r;rc RK||g}n]|rLttjtjt jk}ntd|D]>}|rt|}|j t|||||?dS)aL Dispatch a prediction status change message :param context: The context :param prediction: The prediction :param text: The text :param should_refresh: Whether to should_refresh the message instead of sending a new one :param group_chat: The group chat for which to refresh the message :return: None Nz4Either prediction or should_refresh must be provided) listrrrr0rr4rr8 application create_task0send_prediction_status_change_message_or_refresh)r)r;r*rrrs r<rrs")3 Q(,     % %j&7;K;P&P Q Q) ) OPPP!   3&z22D'' <T>:        r>c Kd}|tj|k}tjtj|k|z}|D]}|j} |r/t ||||jt|ggd{Vnt ||||jdd{V[#ttf$rO}|r&t|trdt|vst|t|Yd}~d}~wwxYwdS)a Send a prediction status change message to all group chats in which the prediction was sent or refresh the message :param context: The context :param prediction: The prediction :param text: The text :param should_refresh: Whether to should_refresh the message instead of sending a new one :param group_chat: The group chat for which to refresh the message :return: None TN)redit_message_idkeyboardF)rreply_to_message_idallow_sending_without_replyzMessage is not modified) r rrrr*r"rr:rr isinstancestrr ) r)r*r;rrgroup_chat_filtermessagesmessagees r<r#r#s$6AZO1K1R1T1T1Z1Z # .* <@QQ22H:: ' 2  : ')$+$6=jIIJK ()(/(:05 z* : : : :q*-- :.Q77%j#a&&999 :'::sAB++D cKt|}|D]}|j}t||d{V |dS)zU Delete a prediction :param prediction: The prediction :return: None N)rYr@rr)r*rrJr@s r<delete_predictionr=$sr;WWa:b:b"9JJ+0+D2HIIIIIIIIII     r>c$t|}|tjurt |nd}t t jt|df|j |gg}| |tj |||S)z Get the share button url :param context: The context :param user: The user :param prediction: The prediction :return: The share button url NF)args descriptionr&) inner_key) r$rWrr4r:rr2PREDICTION_INLINE_RESULT_SHAREr8rset_context_datar INLINE_QUERY)r)r@r*rAr&context_data_values r<get_share_textrF5s8@@I  " "&6&; ; ; 'z222  ,<.% '* ,,, ,.@I r>cltjtj|ktj|kzduS)z Check if a prediction is sent to a group :param prediction: The prediction :param group_chat: The group chat :return: True if the prediction is sent to the group N)r rr*r)r*rs r< prediction_is_sent_to_group_chatrHTs? #. ' 2j @)4 B D    r>)F)TNF)NTTrE)NF)NNN)NNFN)FN)frsrc.tg_compat.errorrrsrc.tg_compat.extrresources.phrasesr2src.model.GroupChatrsrc.model.Predictionr$src.model.PredictionGroupChatMessager src.model.PredictionOptionr src.model.PredictionOptionUserr src.model.Userr src.model.enums.ContextDataKeyr src.model.enums.Emojirsrc.model.enums.Featurersrc.model.enums.Notificationrr src.model.enums.PredictionStatusrrsrc.model.enums.PredictionTypersrc.model.enums.Screenr1src.model.enums.devil_fruit.DevilFruitAbilityTyper-src.model.enums.income_tax.IncomeTaxEventTypersrc.model.error.CustomExceptionrsrc.model.pojo.ContextDataValuersrc.model.pojo.Keyboardrsrc.service.bounty_servicerrsrc.service.date_servicersrc.service.devil_fruit_servicersrc.service.group_servicerr src.service.message_servicer!r" src.service.notification_servicer#src.utils.context_utilsr$src.utils.math_utilsr%r&r'src.utils.string_utilsr( DEFAULT_TYPEboolr=r*r8r rrrrrrrrYrrrrrrr rrrrrqrrr#r:r=rFrHrMr>r<ris99999999******######))))))++++++KKKKKK777777??????999999''''''++++++a```````999999))))))SSSSSSLLLLLL??????<<<<<<,,,,,,KKKKKKKK<<<<<<======WVVVVVVV>>>>>>KKKKKK 766666SX  &4>KO @"" DDDD D D  DDDDT;?! rrr r""67r r  r  rrrrjl7ZTX2*= (A*=z*=*=*=*=\6:*2 ,\=|8\=j\=UY\=\=\=\=@Y]  &4>LU "JdtL`Ga(^Z^DAU -"  -"-"-"-"`- -(<- ----0J J#3J JJJJ&48!&@&@0&@&@ &@ &@&@&@&@R4y4c4j[_N_4444.!  " "  &" " "  "  " " " " R! 4:4:  &4:4: 4: 4:  4:4:4:4:nzh4! !!!!"L5Tz^a>W[r>