52 lines
2.9 KiB
Python
52 lines
2.9 KiB
Python
import random
|
|
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, Arcade_Embed
|
|
from Christmas.Database import database
|
|
class Mail_Confirm_Button(View):
|
|
def __init__(self, editmode: bool, recive_user: Member, title: str, description: str, *args, **kwargs):
|
|
self.editmode = editmode
|
|
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:
|
|
if self.editmode == True:
|
|
await database.update_inserted_mail(self.recive_user.id, interaction.user.id, self.title, self.description)
|
|
await interaction.edit_original_response(embed=Mail_Embed.mail_edit_sended(receive_user=self.recive_user), view=None)
|
|
else:
|
|
await database.insert_mail(interaction.user.id, self.recive_user.id, interaction.user.name, 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)
|
|
|
|
class Arcade_Register_Agree(View):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs, timeout=None)
|
|
|
|
@button(label="동의", style=ButtonStyle.green, custom_id="arcade_register_agree")
|
|
async def agree(self, button, interaction: Interaction):
|
|
await interaction.response.defer()
|
|
try:
|
|
await database.register_arcade(interaction.guild.id, interaction.guild.name, random.choice(["south", "north"]))
|
|
except Exception as e:
|
|
await interaction.edit_original_response(embed=Arcade_Embed.register_failed(), view=None)
|
|
return
|
|
await interaction.edit_original_response(embed=Arcade_Embed.register_sucessfull(), view=None)
|
|
|
|
@button(label="거절", style=ButtonStyle.red, custom_id="arcade_register_disagree")
|
|
async def disagree(self, button, interaction: Interaction):
|
|
await interaction.response.defer()
|
|
await interaction.edit_original_response(embed=Arcade_Embed.register_cancel(), view=None)
|
|
|