soj!$ddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZm Z ddl!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2dedej3dede"dedede ddfdZ4 d&dedej3ded eded!e de5fd"Z6dedej3ded edede ddfd#Z7 d&ded eded!e de8e9e9e9e9ff d$Z:dedej3dede"ddf d%Z;dS)')UpdateMessage) ContextTypesN) BountyGift) GroupChat)User)BountyGiftStatus)Command)BountyGiftReceivedNotification)ReservedKeyboardKeys)Screen)DevilFruitAbilityType)IncomeTaxEventType)GroupChatExceptionGroupChatError)Keyboard)get_amount_from_stringvalidate_amountget_transaction_taxadd_or_remove_bounty)get_ability_value)full_message_sendget_yes_no_keyboard)send_notification)get_value_from_percentage)get_belly_formattedupdatecontextuserinbound_keyboard target_usercommand group_chatreturncvK|t||||||d{VdSt||||d{VdS)aX Manage the Bounty gift screen :param update: The update object :param context: The context object :param user: The user object :param inbound_keyboard: The keyboard object :param target_user: The target user in case of a reply :param command: The command :param group_chat: The group chat :return: None N) send_requestkeyboard_interaction)rrrr r!r"r#s 1/app/src/chat/group/screens/screen_bounty_gift.pymanager)sj,67D+w SSSSSSSSS vw6F G GGGGGGGGGGsenderreceiver bounty_giftc K|t|jdkr%t|tj|dd{VdSt ||||jdt jd{VsdSt||||d{V\}}}} |j | krt|j d|dz zz } tj t|j t|t||t| t| } t|| |dd{VdS|9|t jurt%t&jdS)a Validate the bounty gift request :param update: The update object :param context: The context object :param sender: The user that wants to send a bounty gift :param receiver: The user that wants to receive a bounty gift :param command: The command :param bounty_gift: The bounty gift object :return: True if the request is valid, False otherwise NrTradd_delete_buttonFd)len parametersrphrasesBOUNTY_GIFT_NO_AMOUNTrEnvBOUNTY_GIFT_MIN_AMOUNTget_int get_amountsbountyintBOUNTY_GIFT_NOT_ENOUGH_BOUNTYformatr get_statusr CONFIRMEDrrITEM_IN_WRONG_STATUS) rrr+r,r"r-amounttax_percentage tax_amount total_amount max_amountot_texts r(validaterH<s( w! " "a ' '#6vY]       5% GVW%7%:CI';>>8888884FNJ }|##!~/C*DEFF 7>>  . .  ' '  + +   - -  + +    SWXXXXXXXXXXu;#9#9#;#;?O?Y#Y#Y !DEEE 4r*c pKt|||||d{VsdSt|||d{V\}}}} tjt ||t ||t | } t} || _|| _ || _ || _ || _ | t|tj| jg} t%|| || d{V} | j| _| dS)aF Send request to send a bounty gift :param update: The update object :param context: The context object :param sender: The user that wants to send a bounty gift :param receiver: The user to send the bounty gift to :param command: The command :param group_chat: The group chat :return: None N)screen primary_key)rkeyboard)rHr:r5BOUNTY_GIFT_REQUESTr>rget_markdown_mentionrr+r,rBrCr#saverr GRP_BOUNTY_GIFTidr message_id)rrr+r,r"r#rBrCrDrErGr-inline_keyboardmessages r(r&r&xsz&&'68WEE E E E E E E>IQY[b=c=c7c7c7c7c7c7c4FNJ )00F##%%''J''L)) G)llKK#KK!/K'K F6+A{~^^^-O//G%/Kr*cnK|t|jd|}n|j}t|||j}|dkrt |t j|}t| rt|}tt||}||z}||||fS)z Get the amounts for a bounty gift :param sender: The sender :param receiver: The receiver :param command: The command :param bounty_gift: The bounty gift object :return: The amount, tax percentage, tax amount, total amount Nr) rr4rBrbounty_gift_taxrr GIFT_LOAN_TAXfloat is_integerr<r)r+r,r"r-rBrCrDrEs r(r:r:s'(:1(=vFF#(6;QRRN* )7    ^''))-^,,.v~FFGGJJ&L >:| ;;r*c 2Ktj|tj}|jtjs9|t|tj |dd{VdS|j }t|||||d{Vs|dSt|||d{V\}}}} ||_||_t j|_||xjt*jz c_t1|| d|dd{Vt1|||t2j|jdd{Vtjt=||t=||t=| } t|| |dd{VtA||} tC||| d{VdS)z Keyboard interaction :param update: The update object :param context: The context object :param sender: The user that wants to send a bounty gift :param inbound_keyboard: The inbound keyboard :return: None Tr/N)r-F)addr should_save)rtax_event_typeevent_idr\)"r get_by_idgetr DEFAULT_PRIMARY_KEYinfoCONFIRMdelete_instancerr5BOUNTY_GIFT_CANCELLEDr,rHr:rBrCr r@statusrOrVr7BOUNTY_GIFT_TAX_INCREASEr9rr BOUNTY_GIFTrQBOUNTY_GIFT_CONFIRMEDr>rrNr r) rrr+r r-r,rBrCrDrErG notifications r(r'r's)21EFFK  !5!= >##%%% W26UY            )H&'68UUU U U U U U U##%%%>Ik>>>8888884FNJ  K!/K)3K c:BBDDD v|v[_ ` ` ```````` )5    +22F##%%''J''L)) G GWVt T T TTTTTTTT2&&AAL GX| < <<<<<<<<<rs))))))))******############++++++))))))======++++++GGGGGGEEEEEE))))))SSSSSSLLLLLLMMMMMMMM,,,,,, >=====NNNNNNNN>>>>>>::::::666666H H  &H H H  H  HH HHHHD" 99 9  &9 9 9  9  9 9999x4 4  &4 4 4  4  4 4444pVZ < <  <  <+2