refactor: 등록/탈퇴 명령어 별도 파일로 분리

functions/account.py

@@ -0,0 +1,107 @@
+import discord
+from discord.ext import commands
+from utils.commands import slash_command
+from discord.commands import ApplicationContext, Option
+import os
+import pickle
+
+class Account(commands.Cog):
+
+    @slash_command(name="등록", description="유저 정보를 등록합니다.")
+    async def register(self, ctx):
+        if os.path.isfile(f"{ctx.user.id}.pkl") == False:
+            embed = discord.Embed(
+                title="등록", description="봇의 서비스에 등록합니다. 동의하지 않을 경우 서비스 이용에 제한이 있을 수 있습니다."
+            )
+            embed.add_field(name="수집하는 정보", value="유저의 디스코드 id", inline=False)
+
+            class Button(discord.ui.View):
+                @discord.ui.button(label="⭕ 동의", style=discord.ButtonStyle.primary)
+                async def primary(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    UserData = 1
+                    if interaction.user.id == ctx.user.id:
+                        with open(f"{interaction.user.id}.pkl", "wb") as f:
+                            pickle.dump(UserData, f)
+                        new_embed = discord.Embed(
+                            title="등록 완료", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"<@!{interaction.user.id}>님, 등록을 완료하였습니다. 아이디는 {interaction.user.id}입니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+
+                @discord.ui.button(label="❌ 취소", style=discord.ButtonStyle.danger)
+                async def danger(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    if interaction.user.id == ctx.user.id:
+                        new_embed = discord.Embed(
+                            title="등록 취소", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"등록이 취소되었습니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+            await ctx.respond(embed=embed, view=Button())
+
+        else:
+            embed = discord.Embed(title="등록된 유저", description="")
+            embed.add_field(
+                name="", value="이미 등록되었습니다. `/탈퇴` 명령어를 통해 탈퇴할 수 있습니다.", inline=False
+            )
+            await ctx.respond(embed=embed)
+
+
+    @slash_command(name="탈퇴", description="봇 서비스를 탈퇴합니다.")
+    async def delete(self, ctx):
+        if os.path.isfile(f"{ctx.user.id}.pkl"):
+            embed = discord.Embed(
+                title="탈퇴", description="봇의 서비스를 탈퇴합니다. 탈퇴할 경우 모든 데이터가 파기되며, 복구하기 어렵습니다."
+            )
+            embed.add_field(
+                name="파기하는 정보", value="유저의 디스코드 id, 서비스 이용 중 가챠 기록", inline=False
+            )
+
+            class Button(discord.ui.View):
+                @discord.ui.button(label="⭕ 동의", style=discord.ButtonStyle.primary)
+                async def primary(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    if interaction.user.id == ctx.user.id:
+                        os.remove(f"{interaction.user.id}.pkl")
+                        new_embed = discord.Embed(
+                            title="탈퇴 완료", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"<@!{interaction.user.id}>님, 탈퇴를 완료하였습니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+
+                @discord.ui.button(label="❌ 취소", style=discord.ButtonStyle.danger)
+                async def danger(
+                    self, button: discord.ui.Button, interaction: discord.Interaction
+                ):
+                    if interaction.user.id == ctx.user.id:
+                        new_embed = discord.Embed(
+                            title="탈퇴 취소", description=""
+                        )
+                        new_embed.add_field(
+                            name="", value=f"탈퇴가 취소되었습니다."
+                        )
+                        self.disable_all_items()
+                        await interaction.response.edit_message(view=self, embed=new_embed)
+
+            await ctx.respond(embed=embed, view=Button())
+
+        else:
+            embed = discord.Embed(title="등록하지 않은 유저", description="")
+            embed.add_field(name="", value="`/등록`을 통해 가입해주세요.", inline=False)
+            await ctx.respond(embed=embed)
+
+def setup(bot):
+    bot.add_cog(Account())