디스코드 봇 만들기#19Cog로 코드 쪼개기! 1000줄 봇도 안 더러워지는 구조 (파이썬 예제)

pip install -U discord.py python-dotenv

python -c "import discord; print(discord.__version__)"

DISCORD_TOKEN=여기에_봇_토큰_붙여넣기
GUILD_ID=테스트용_서버_ID

server-bot/
├─ .env                  # 토큰·서버 ID (git 제외)
├─ .gitignore
├─ requirements.txt
├─ bot.py                # 진입점 (Cog 로드만 담당)
├─ config.py             # 설정 로딩
└─ cogs/
   ├─ poll.py            # #18 투표  ← 오늘 이사시킬 파일
   └─ admin.py           # 리로드 등 관리 명령어

import discord
from discord.ext import commands

class ServerBot(commands.Bot):
    def __init__(self):
        intents = discord.Intents.default()
        intents.message_content = True
        intents.members = True
        intents.polls = True
        super().__init__(command_prefix="!", intents=intents)

    async def setup_hook(self):
        for path in COGS_DIR.glob("*.py"):
            if path.stem.startswith("_"):
                continue
            await self.load_extension(f"cogs.{path.stem}")
            print(f"✅ cogs.{path.stem} 불러옴")

import datetime

import discord
from discord import app_commands
from discord.ext import commands

NUMBER_EMOJI = ["1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6️⃣", "7️⃣", "8️⃣", "9️⃣", "🔟"]

class Poll(commands.Cog):
    """투표 관련 명령어 모음"""

    def __init__(self, bot: commands.Bot):
        self.bot = bot

    @app_commands.command(name="투표", description="서버 투표를 만들어요")
    async def create_poll(self, interaction: discord.Interaction, 질문: str, 항목: str):
        labels = [x.strip() for x in 항목.split(",") if x.strip()]
        poll = discord.Poll(question=질문, duration=datetime.timedelta(hours=24))
        for index, label in enumerate(labels):
            poll.add_answer(text=label, emoji=NUMBER_EMOJI[index])
        await interaction.response.send_message("투표를 올렸어요! 📊", ephemeral=True)
        await interaction.channel.send(poll=poll)

    @commands.Cog.listener()
    async def on_poll_vote_add(self, user, answer):
        print(f"[투표] {user}님 → '{answer.text}'")

async def setup(bot: commands.Bot):
    await bot.add_cog(Poll(bot))

@commands.command(name="reload")
@commands.is_owner()
async def reload_cog(self, ctx: commands.Context, name: str):
    try:
        await self.bot.reload_extension(f"cogs.{name}")
    except commands.ExtensionNotLoaded:
        await self.bot.load_extension(f"cogs.{name}")
        await ctx.send(f"🆕 `{name}` 을 새로 불러왔어요!")
    except commands.ExtensionError as error:
        await ctx.send(f"❌ 실패: `{type(error).__name__}: {error}`")
    else:
        await ctx.send(f"🔄 `{name}` 을 다시 불러왔어요!")

class Poll(commands.Cog):
    async def cog_load(self):
        print("[Poll] 준비 완료!")   # DB 연결, 캐시 로딩 등

    async def cog_unload(self):
        print("[Poll] 정리했어요")     # 타이머 중지, 연결 닫기 등

# discord.py 2.x (2.7.1 기준) / Python 3.8+
# 디스코드 봇 만들기 #19 - Cog 구조화

import asyncio
import os
from pathlib import Path

import discord
from discord.ext import commands
from dotenv import load_dotenv

load_dotenv()
TOKEN = os.getenv("DISCORD_TOKEN")

BASE_DIR = Path(__file__).parent
COGS_DIR = BASE_DIR / "cogs"

class ServerBot(commands.Bot):
    def __init__(self):
        intents = discord.Intents.default()
        intents.message_content = True   # 특권 인텐트 (개발자 포털에서도 ON)
        intents.members = True           # 특권 인텐트 (#20 온보딩용 미리 준비)
        intents.polls = True             # 투표 이벤트용
        super().__init__(command_prefix="!", intents=intents)

    async def setup_hook(self):
        """봇이 접속하기 직전에 딱 한 번 실행돼요."""
        for path in sorted(COGS_DIR.glob("*.py")):
            if path.stem.startswith("_"):
                continue
            try:
                await self.load_extension(f"cogs.{path.stem}")
                print(f"✅ cogs.{path.stem} 불러옴")
            except commands.ExtensionError as error:
                print(f"❌ cogs.{path.stem} 실패: {error}")

    async def on_ready(self):
        print(f"{self.user} 로그인 성공! (Cog {len(self.cogs)}개)")

async def main():
    if not TOKEN:
        raise SystemExit(".env 에 DISCORD_TOKEN 이 없어요!")

    bot = ServerBot()
    async with bot:
        await bot.start(TOKEN)

if __name__ == "__main__":
    asyncio.run(main())

# discord.py 2.x / Python 3.8+
# #18 투표 기능을 Cog로 옮긴 버전

import datetime

import discord
from discord import app_commands
from discord.ext import commands

NUMBER_EMOJI = ["1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6️⃣", "7️⃣", "8️⃣", "9️⃣", "🔟"]
MAX_ANSWERS = 10
MAX_QUESTION_LEN = 300

class Poll(commands.Cog):
    """투표 관련 명령어 모음"""

    def __init__(self, bot: commands.Bot):
        self.bot = bot

    async def cog_load(self):
        print("[Poll] 준비 완료!")

    async def cog_unload(self):
        print("[Poll] 정리했어요")

    @app_commands.command(name="투표", description="서버 투표를 만들어요")
    @app_commands.describe(
        질문="투표에 띄울 질문 (최대 300자)",
        항목="선택지를 쉼표(,)로 구분해서 입력 (2~10개)",
        시간="진행 시간 (시간 단위, 기본 24)",
    )
    async def create_poll(
        self,
        interaction: discord.Interaction,
        질문: str,
        항목: str,
        시간: int = 24,
    ):
        labels = [x.strip() for x in 항목.split(",") if x.strip()]

        if not 2 <= len(labels) <= MAX_ANSWERS:
            await interaction.response.send_message(
                f"선택지는 2~{MAX_ANSWERS}개여야 해요! (지금 {len(labels)}개)",
                ephemeral=True,
            )
            return

        if len(질문) > MAX_QUESTION_LEN:
            await interaction.response.send_message(
                f"질문은 {MAX_QUESTION_LEN}자까지만 가능해요!", ephemeral=True
            )
            return

        poll = discord.Poll(
            question=질문,
            duration=datetime.timedelta(hours=시간),
        )
        for index, label in enumerate(labels):
            poll.add_answer(text=label, emoji=NUMBER_EMOJI[index])

        await interaction.response.send_message("투표를 올렸어요! 📊", ephemeral=True)
        await interaction.channel.send(poll=poll)

    @commands.Cog.listener()
    async def on_poll_vote_add(self, user, answer):
        print(f"[투표] {user}님 → '{answer.text}' (현재 {answer.vote_count}표)")

async def setup(bot: commands.Bot):
    await bot.add_cog(Poll(bot))

# discord.py 2.x / Python 3.8+
# 봇 관리용 명령어 (소유자 전용)

import discord
from discord.ext import commands

class Admin(commands.Cog):
    """봇 소유자 전용 관리 명령어"""

    def __init__(self, bot: commands.Bot):
        self.bot = bot

    @commands.command(name="reload")
    @commands.is_owner()
    async def reload_cog(self, ctx: commands.Context, name: str):
        """!reload poll  → cogs/poll.py 를 다시 불러와요"""
        try:
            await self.bot.reload_extension(f"cogs.{name}")
        except commands.ExtensionNotLoaded:
            try:
                await self.bot.load_extension(f"cogs.{name}")
            except commands.ExtensionError as error:
                await ctx.send(f"❌ 실패: `{type(error).__name__}: {error}`")
            else:
                await ctx.send(f"🆕 `{name}` 을 새로 불러왔어요!")
        except commands.ExtensionError as error:
            await ctx.send(f"❌ 실패: `{type(error).__name__}: {error}`")
        else:
            await ctx.send(f"🔄 `{name}` 을 다시 불러왔어요!")

    @commands.command(name="cogs")
    @commands.is_owner()
    async def list_cogs(self, ctx: commands.Context):
        """지금 올라가 있는 Cog 목록을 보여줘요"""
        names = "\n".join(f"· {name}" for name in self.bot.cogs) or "(없음)"
        await ctx.send(f"**불러온 Cog {len(self.bot.cogs)}개**\n{names}")

    @commands.command(name="sync")
    @commands.is_owner()
    async def sync_tree(self, ctx: commands.Context):
        """슬래시 명령어를 이 서버에 즉시 동기화해요"""
        self.bot.tree.copy_global_to(guild=ctx.guild)
        synced = await self.bot.tree.sync(guild=ctx.guild)
        await ctx.send(f"🌳 슬래시 명령어 {len(synced)}개 동기화 완료!")

    @reload_cog.error
    async def reload_error(self, ctx: commands.Context, error):
        if isinstance(error, commands.NotOwner):
            await ctx.send("이 명령어는 봇 소유자만 쓸 수 있어요 🔒")

async def setup(bot: commands.Bot):
    await bot.add_cog(Admin(bot))

[Poll] 준비 완료!
✅ cogs.admin 불러옴
✅ cogs.poll 불러옴
MojangBot#1234 로그인 성공! (Cog 2개)

!reload poll

[Poll] 정리했어요
[Poll] 준비 완료!
🔄 poll 을 다시 불러왔어요!