from discord import InputTextStyle, Interaction, Member from discord.ui import Modal, InputText from Christmas.Module import check_curse from Christmas.UI.Embed import Mail_Embed from Christmas.UI.Buttons import Mail_Confirm_Button class Send_Mail_Modal(Modal): def __init__(self, reciveuser: Member, *args, **kwargs): self.reciveuser = reciveuser super().__init__(timeout=None, *args, **kwargs) self.add_item(InputText("제목", placeholder="제목을 입력해주세요.", style=InputTextStyle.short, required=True, custom_id="mail_title")) self.add_item(InputText("내용", placeholder="내용을 입력해주세요.", style=InputTextStyle.long, required=True, custom_id="mail_content")) async def callback(self, interaction: Interaction): await interaction.response.defer() await interaction.edit_original_message(embed=Mail_Embed.mail_confirm(self.items[0].value, self.items[1].value), view=Mail_Confirm_Button(self.reciveuser))