디스코드 봇 만들기#18투표(Poll) 기능 만들기! 다수결로 정하는 서버 규칙 (파이썬 예제)

pip install -U discord.py python-dotenv

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

DISCORD_TOKEN=여기에_봇_토큰_붙여넣기

import datetime
import discord

poll = discord.Poll(
    question="오늘 저녁 뭐 먹을까요?",
    duration=datetime.timedelta(hours=24),
    multiple=False,
)

poll.add_answer(text="치킨", emoji="🍗")
poll.add_answer(text="피자", emoji="🍕")
poll.add_answer(text="굶기", emoji="😇")

await channel.send(poll=poll)

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

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

    poll = discord.Poll(
        question=질문,
        duration=datetime.timedelta(hours=시간),
        multiple=복수선택,
    )
    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)

message = await interaction.channel.fetch_message(메시지id)
poll = message.poll

total = poll.total_votes

for answer in poll.answers:
    percent = (answer.vote_count / total * 100) if total else 0
    bar = "█" * round(percent / 10) or "░"
    print(f"{answer.text}: {bar} {answer.vote_count}표 ({percent:.1f}%)")

if poll.is_finalised() and poll.victor_answer:
    print(f"🏆 1위: {poll.victor_answer.text}")

async for voter in poll.answers[0].voters():
    print(f"{voter} 님이 투표했어요!")

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

@bot.event
async def on_poll_vote_remove(user, answer):
    print(f"[취소] {user}님이 '{answer.text}' 투표를 취소했어요")

intents = discord.Intents.default()
intents.message_content = True   # 개발자 포털에서도 켜야 하는 특권 인텐트
intents.polls = True             # 투표 이벤트용

# discord.py 2.5+ / Python 3.8+
# 디스코드 봇 만들기 #18 - 투표(Poll) 기능

import os
import datetime

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

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

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

MAX_ANSWERS = 10        # 디스코드 제한
MAX_QUESTION_LEN = 300  # 디스코드 제한
MAX_ANSWER_LEN = 55     # 디스코드 제한
MAX_HOURS = 24 * 32     # 최대 32일

intents = discord.Intents.default()
intents.message_content = True
intents.polls = True

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

@bot.event
async def on_ready():
    try:
        synced = await bot.tree.sync()
        print(f"슬래시 명령어 {len(synced)}개 동기화 완료")
    except Exception as error:
        print(f"동기화 실패: {error}")
    print(f"{bot.user} 로그인 성공!")

@bot.tree.command(name="투표", description="서버 투표를 만들어요")
@app_commands.describe(
    질문="투표에 띄울 질문 (최대 300자)",
    항목="선택지를 쉼표(,)로 구분해서 입력 (2~10개)",
    시간="진행 시간 (시간 단위, 기본 24 / 최대 768)",
    복수선택="여러 개를 고를 수 있게 할지",
)
async def create_poll(
    interaction: discord.Interaction,
    질문: str,
    항목: str,
    시간: int = 24,
    복수선택: bool = False,
):
    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

    too_long = [x for x in labels if len(x) > MAX_ANSWER_LEN]
    if too_long:
        await interaction.response.send_message(
            f"선택지는 하나당 {MAX_ANSWER_LEN}자까지예요: {too_long[0][:20]}...",
            ephemeral=True,
        )
        return

    if not 1 <= 시간 <= MAX_HOURS:
        await interaction.response.send_message(
            f"진행 시간은 1~{MAX_HOURS}시간 사이여야 해요!", ephemeral=True
        )
        return

    poll = discord.Poll(
        question=질문,
        duration=datetime.timedelta(hours=시간),
        multiple=복수선택,
    )
    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)

@bot.tree.command(name="투표결과", description="투표 메시지의 현재 집계를 보여줘요")
@app_commands.describe(메시지id="투표가 달린 메시지의 ID")
async def poll_result(interaction: discord.Interaction, 메시지id: str):
    await interaction.response.defer(ephemeral=True)

    try:
        message = await interaction.channel.fetch_message(int(메시지id))
    except (ValueError, discord.NotFound):
        await interaction.followup.send(
            "그런 메시지를 찾지 못했어요. ID를 다시 확인해주세요!", ephemeral=True
        )
        return

    poll = message.poll
    if poll is None:
        await interaction.followup.send("이 메시지에는 투표가 없어요!", ephemeral=True)
        return

    total = poll.total_votes
    embed = discord.Embed(
        title=f"📊 {poll.question}",
        color=discord.Color.blurple(),
    )

    for answer in poll.answers:
        percent = (answer.vote_count / total * 100) if total else 0
        bar = "█" * round(percent / 10) or "░"
        name = f"{answer.emoji or ''} {answer.text}".strip()
        embed.add_field(
            name=name,
            value=f"`{bar}` **{answer.vote_count}표** ({percent:.1f}%)",
            inline=False,
        )

    status = "종료됨 ✅" if poll.is_finalised() else "진행 중 🟢"
    embed.set_footer(text=f"총 {total}표 · {status}")

    if poll.is_finalised() and poll.victor_answer is not None:
        embed.description = f"🏆 1위: **{poll.victor_answer.text}**"

    await interaction.followup.send(embed=embed, ephemeral=True)

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

@bot.event
async def on_poll_vote_remove(user, answer):
    print(f"[취소] {user}님이 '{answer.text}' 투표를 취소했어요")

bot.run(TOKEN)

/투표 질문:오늘 밤 뭐 돌릴까요? 항목:마인크래프트, 발로란트, 롤, 그냥 수다 시간:12 복수선택:False

슬래시 명령어 2개 동기화 완료
MojangBot#1234 로그인 성공!
[투표] 모장캠님 → '마인크래프트' (현재 1표)
[투표] 감자칩님 → '롤' (현재 1표)
[취소] 감자칩님이 '롤' 투표를 취소했어요

📊 오늘 밤 뭐 돌릴까요?
1️⃣ 마인크래프트   ██████ 6표 (60.0%)
2️⃣ 발로란트      ██ 2표 (20.0%)
3️⃣ 롤           █ 1표 (10.0%)
4️⃣ 그냥 수다      █ 1표 (10.0%)
총 10표 · 진행 중 🟢