75 lines
4.1 KiB
Python
75 lines
4.1 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
|
|
from Christmas.SearchEngine.Snowball import Snowball
|
|
from Christmas.config import ChristmasConfig
|
|
|
|
from koreanbots import KoreanbotsRequester
|
|
|
|
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:
|
|
region = random.choice(["south", "north"])
|
|
await database.register_arcade(interaction.guild.id, interaction.guild.name, region)
|
|
await Snowball.register_guild(interaction.guild.id, interaction.guild.name, region)
|
|
except Exception as e:
|
|
print(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)
|
|
|
|
class Recommanded(View):
|
|
def __init__(self, *args, **kwargs):
|
|
self.config = ChristmasConfig()
|
|
super().__init__(*args, **kwargs, timeout=None)
|
|
|
|
@button(label="추천완료", style=ButtonStyle.green, custom_id="arcade_recommanded")
|
|
async def finished(self, button, interaction: Interaction):
|
|
await interaction.response.defer()
|
|
data = await KoreanbotsRequester(self.config.KOREANBOT_TOKEN).get_bot_vote(int(interaction.user.id), interaction.guild.me.id)
|
|
if data["voted"] == True:
|
|
await database.insert_arcade_point(interaction.guild.id, interaction.user.id)
|
|
await interaction.edit_original_response(embed=Arcade_Embed.recommend_sucess(), view=None)
|
|
else:
|
|
await interaction.edit_original_response(embed=Arcade_Embed.recommend_failed(), view=None)
|
|
|