디스코드 봇 만들기#13역할·채널 자동화로 서버 관리하기! (파이썬 예제)

pip install -U discord.py

import discord
from discord.ext import commands

intents = discord.Intents.default()
intents.members = True          # 멤버 입장 감지 (프리빌리지드 인텐트)
intents.message_content = True  # 접두사 명령어용

bot = commands.Bot(command_prefix="!", intents=intents)

AUTO_ROLE_NAME = "새싹"   # 입장 시 자동으로 줄 역할 이름

@bot.event
async def on_member_join(member: discord.Member):
    role = discord.utils.get(member.guild.roles, name=AUTO_ROLE_NAME)
    if role is not None:
        await member.add_roles(role)
        print(f"👋 {member.name} 님에게 '{role.name}' 역할을 줬어요!")

# 버튼 역할 목록 (custom_id, 역할이름, 버튼라벨, 이모지)
BUTTON_ROLES = [
    ("role_notice", "공지알림", "공지 알림", "🔔"),
    ("role_game",   "게임메이트", "게임 메이트", "🎮"),
    ("role_study",  "스터디원",  "스터디원",   "📚"),
]

class RoleButton(discord.ui.Button):
    def __init__(self, custom_id, role_name, label, emoji):
        super().__init__(
            label=label, emoji=emoji,
            style=discord.ButtonStyle.secondary,
            custom_id=custom_id,        # 영구 View 는 custom_id 필수!
        )
        self.role_name = role_name

    async def callback(self, interaction: discord.Interaction):
        role = discord.utils.get(interaction.guild.roles, name=self.role_name)
        if role is None:
            await interaction.response.send_message(
                f"❌ '{self.role_name}' 역할이 없어요. 먼저 서버에 만들어 주세요!",
                ephemeral=True)
            return
        if role in interaction.user.roles:
            await interaction.user.remove_roles(role)
            msg = f"'{role.name}' 역할을 뺐어요. 👋"
        else:
            await interaction.user.add_roles(role)
            msg = f"'{role.name}' 역할을 드렸어요! 🎉"
        await interaction.response.send_message(msg, ephemeral=True)

class RoleView(discord.ui.View):
    def __init__(self):
        super().__init__(timeout=None)   # 영구 View 는 timeout=None!
        for custom_id, role_name, label, emoji in BUTTON_ROLES:
            self.add_item(RoleButton(custom_id, role_name, label, emoji))

class MyBot(commands.Bot):
    def __init__(self):
        super().__init__(command_prefix="!", intents=intents)

    async def setup_hook(self):
        self.add_view(RoleView())   # 재시작 후 버튼 되살리기

bot = MyBot()

@bot.command(name="역할판")
@commands.has_permissions(administrator=True)
async def role_panel(ctx: commands.Context):
    embed = discord.Embed(
        title="🎭 역할 받기",
        description="버튼을 누르면 역할을 드려요! 한 번 더 누르면 역할이 빠져요.",
        color=0x5865F2,
    )
    await ctx.send(embed=embed, view=RoleView())

CREATOR_CHANNEL_ID = 123456789012345678   # '➕ 채널 만들기' 음성 채널 ID
temp_channels = set()                     # 우리가 만든 임시 채널 ID 보관

@bot.event
async def on_voice_state_update(member, before, after):
    # (생성) '만들기' 채널에 들어오면 개인 채널을 만들고 이동
    if after.channel is not None and after.channel.id == CREATOR_CHANNEL_ID:
        new_channel = await member.guild.create_voice_channel(
            name=f"🔊 {member.display_name}의 방",
            category=after.channel.category,
        )
        temp_channels.add(new_channel.id)
        await member.move_to(new_channel)

    # (삭제) 우리가 만든 임시 채널이 비면 제거
    if before.channel is not None and before.channel.id in temp_channels:
        if len(before.channel.members) == 0:
            await before.channel.delete()
            temp_channels.discard(before.channel.id)

# discord.py 2.x (2.7.1 기준) / Python 3.9+
import discord
from discord.ext import commands

# ── 설정 ──────────────────────────────
TOKEN = "여기에_봇_토큰"
AUTO_ROLE_NAME = "새싹"                       # 입장 시 자동으로 줄 역할 이름
CREATOR_CHANNEL_ID = 123456789012345678       # '➕ 채널 만들기' 음성 채널 ID

# 버튼 역할 목록 (custom_id, 역할이름, 버튼라벨, 이모지)
BUTTON_ROLES = [
    ("role_notice", "공지알림", "공지 알림", "🔔"),
    ("role_game",   "게임메이트", "게임 메이트", "🎮"),
    ("role_study",  "스터디원",  "스터디원",   "📚"),
]

intents = discord.Intents.default()
intents.members = True          # 멤버 입장 감지 (프리빌리지드 인텐트)
intents.message_content = True  # 접두사 명령어용
# voice_states 는 default() 에 이미 포함돼 있어요.

# ── 버튼 역할 View ────────────────────
class RoleButton(discord.ui.Button):
    def __init__(self, custom_id, role_name, label, emoji):
        super().__init__(
            label=label, emoji=emoji,
            style=discord.ButtonStyle.secondary,
            custom_id=custom_id,        # 영구 View 는 custom_id 필수!
        )
        self.role_name = role_name

    async def callback(self, interaction: discord.Interaction):
        role = discord.utils.get(interaction.guild.roles, name=self.role_name)
        if role is None:
            await interaction.response.send_message(
                f"❌ '{self.role_name}' 역할이 없어요. 먼저 서버에 만들어 주세요!",
                ephemeral=True)
            return
        if role in interaction.user.roles:
            await interaction.user.remove_roles(role)
            msg = f"'{role.name}' 역할을 뺐어요. 👋"
        else:
            await interaction.user.add_roles(role)
            msg = f"'{role.name}' 역할을 드렸어요! 🎉"
        await interaction.response.send_message(msg, ephemeral=True)

class RoleView(discord.ui.View):
    def __init__(self):
        super().__init__(timeout=None)   # 영구 View 는 timeout=None!
        for custom_id, role_name, label, emoji in BUTTON_ROLES:
            self.add_item(RoleButton(custom_id, role_name, label, emoji))

# ── 봇 본체 ───────────────────────────
class MyBot(commands.Bot):
    def __init__(self):
        super().__init__(command_prefix="!", intents=intents)

    async def setup_hook(self):
        # 재시작 후에도 버튼이 동작하도록 영구 View 재등록
        self.add_view(RoleView())

bot = MyBot()
temp_channels = set()   # 우리가 만든 임시 음성 채널 ID 보관

@bot.event
async def on_ready():
    print(f"✅ {bot.user} 로 로그인했어요!")

# 1) 입장하면 자동으로 역할 부여
@bot.event
async def on_member_join(member: discord.Member):
    role = discord.utils.get(member.guild.roles, name=AUTO_ROLE_NAME)
    if role is not None:
        await member.add_roles(role)
        print(f"👋 {member.name} → '{role.name}' 역할 부여 완료")

# 2) 역할 버튼 패널 띄우기 (관리자만)
@bot.command(name="역할판")
@commands.has_permissions(administrator=True)
async def role_panel(ctx: commands.Context):
    embed = discord.Embed(
        title="🎭 역할 받기",
        description="버튼을 누르면 역할을 드려요! 한 번 더 누르면 역할이 빠져요.",
        color=0x5865F2,
    )
    await ctx.send(embed=embed, view=RoleView())

# 3) 지정 채널 입장 시 개인 음성 채널 자동 생성 + 비면 삭제
@bot.event
async def on_voice_state_update(member, before, after):
    # (생성) '만들기' 채널에 들어오면 개인 채널을 만들고 이동
    if after.channel is not None and after.channel.id == CREATOR_CHANNEL_ID:
        new_channel = await member.guild.create_voice_channel(
            name=f"🔊 {member.display_name}의 방",
            category=after.channel.category,
        )
        temp_channels.add(new_channel.id)
        await member.move_to(new_channel)

    # (삭제) 우리가 만든 임시 채널이 비면 제거
    if before.channel is not None and before.channel.id in temp_channels:
        if len(before.channel.members) == 0:
            await before.channel.delete()
            temp_channels.discard(before.channel.id)

bot.run(TOKEN)