28 lines
1.5 KiB
Python
28 lines
1.5 KiB
Python
from discord.ui import View, button
|
|
from discord import Interaction, ButtonStyle, Embed, Color, TextChannel, Forbidden, Message, Member, User, File, HTTPException
|
|
|
|
from Christmas.UI.Embed import Mail_Embed
|
|
from Christmas.Database import database
|
|
|
|
class Mail_Confirm_Button(View):
|
|
def __init__(self, recive_user: Member, title: str, description: str, *args, **kwargs):
|
|
self.recive_user = recive_user
|
|
self.title = title
|
|
self.description = description
|
|
super().__init__(timeout=None)
|
|
|
|
@button(label="전송", style=ButtonStyle.green, custom_id="mail_send")
|
|
async def send(self, button, interaction: Interaction):
|
|
await interaction.response.defer()
|
|
if await database.get_guild(interaction.guild.id) == None:
|
|
await interaction.edit_original_response(embed=Embed(title="❌ 에러!", description="서버가 가입되어있지 않아요! 서버에 가입해주세요!"), view=None)
|
|
return
|
|
else:
|
|
await database.insert_mail(interaction.user.id, self.recive_user.id, self.title, self.description)
|
|
await interaction.edit_original_response(embed=Mail_Embed.mail_sended(receive_user=self.recive_user), view=None)
|
|
|
|
@button(label="취소", style=ButtonStyle.red, custom_id="mail_cancel")
|
|
async def cancel(self, button, interaction: Interaction):
|
|
await interaction.response.defer()
|
|
await interaction.edit_original_response(embed=Embed(title="✅ 전송취소!", description="편지 전송을 취소했어요!"), view=None)
|
|
|