runabot/Christmas/UI/Modal.py
2023-11-26 02:28:48 +09:00

20 lines
1.0 KiB
Python

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))