runabot/RUNA/UI/Buttons.py

import random
import os
from discord.ui import View, button
from discord import Interaction, ButtonStyle, Embed, Member
from dotenv import load_dotenv, find_dotenv

from RUNA.UI.Embed import Mail_Embed, Arcade_Embed
from RUNA.Database import database
from koreanbots import KoreanbotsRequester


load_dotenv(find_dotenv())

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):
        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(os.getenv("KOREANBOT_TOKEN")).get_bot_vote(int(interaction.user.id), interaction.guild.me.id)
        if data["data"]["voted"] == True:
            if database.
            await database.insert_guild_art_point(interaction.guild.id, 5)
            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=self)