디스코드 봇 만들기#28레벨·경험치(XP) 시스템 만들기! 채팅할수록 레벨업, 랭킹판까지 (파이썬 예제)

pip install -U discord.py aiosqlite python-dotenv

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

import aiosqlite

DB_PATH = "levels.db"

async def init_db():
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute("""
            CREATE TABLE IF NOT EXISTS levels (
                guild_id INTEGER NOT NULL,
                user_id  INTEGER NOT NULL,
                xp       INTEGER NOT NULL DEFAULT 0,
                level    INTEGER NOT NULL DEFAULT 0,
                PRIMARY KEY (guild_id, user_id)
            )
        """)
        await db.commit()

def xp_for_next(level: int) -> int:
    """현재 레벨에서 다음 레벨까지 필요한 XP"""
    return 5 * (level ** 2) + 50 * level + 100

import random, time

XP_COOLDOWN = 60          # 초 단위
last_xp_time: dict[tuple[int, int], float] = {}

async def add_xp(guild_id: int, user_id: int, amount: int):
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            "INSERT INTO levels (guild_id, user_id, xp) VALUES (?, ?, ?) "
            "ON CONFLICT(guild_id, user_id) DO UPDATE SET xp = xp + ?",
            (guild_id, user_id, amount, amount),
        )
        await db.commit()

        cur = await db.execute(
            "SELECT xp, level FROM levels WHERE guild_id = ? AND user_id = ?",
            (guild_id, user_id),
        )
        xp, level = await cur.fetchone()

    # 레벨업 판정 (한 번에 여러 레벨 오를 수도 있어요)
    leveled_up = False
    while xp >= xp_for_next(level):
        xp -= xp_for_next(level)
        level += 1
        leveled_up = True

    if leveled_up:
        async with aiosqlite.connect(DB_PATH) as db:
            await db.execute(
                "UPDATE levels SET xp = ?, level = ? WHERE guild_id = ? AND user_id = ?",
                (xp, level, guild_id, user_id),
            )
            await db.commit()
    return level if leveled_up else None

@bot.tree.command(name="레벨", description="내 레벨과 경험치를 확인해요")
async def level_cmd(interaction: discord.Interaction, 유저: discord.Member = None):
    target = 유저 or interaction.user
    async with aiosqlite.connect(DB_PATH) as db:
        cur = await db.execute(
            "SELECT xp, level FROM levels WHERE guild_id = ? AND user_id = ?",
            (interaction.guild_id, target.id),
        )
        row = await cur.fetchone()

    xp, level = row if row else (0, 0)
    need = xp_for_next(level)
    filled = int(20 * xp / need)
    bar = "█" * filled + "░" * (20 - filled)

    embed = discord.Embed(title=f"{target.display_name} 님의 레벨", color=0x5865F2)
    embed.add_field(name="레벨", value=f"**{level}**", inline=True)
    embed.add_field(name="경험치", value=f"{xp} / {need}", inline=True)
    embed.add_field(name="진행도", value=f"`{bar}`", inline=False)
    embed.set_thumbnail(url=target.display_avatar.url)
    await interaction.response.send_message(embed=embed)

# 디스코드 봇 만들기 #28 - 레벨·경험치(XP) 시스템
# discord.py 2.x (검증 버전 2.7.1) / Python 3.10+
# pip install -U discord.py aiosqlite python-dotenv
import os
import random
import time

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

load_dotenv()
token = os.getenv("token")

DB_PATH = "levels.db"
XP_COOLDOWN = 60           # 경험치 지급 쿨다운(초)
XP_MIN, XP_MAX = 15, 25    # 메시지당 지급 XP 범위
LEVEL_UP_CHANNEL = None    # 특정 채널 ID를 넣으면 그곳에만 알림

intents = discord.Intents.default()
intents.message_content = True   # 개발자 포털에서도 켜야 해요!
intents.members = True

bot = commands.Bot(command_prefix="!", intents=intents)
last_xp_time: dict[tuple[int, int], float] = {}

def xp_for_next(level: int) -> int:
    return 5 * (level ** 2) + 50 * level + 100

async def init_db():
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            """
            CREATE TABLE IF NOT EXISTS levels (
                guild_id INTEGER NOT NULL,
                user_id  INTEGER NOT NULL,
                xp       INTEGER NOT NULL DEFAULT 0,
                level    INTEGER NOT NULL DEFAULT 0,
                PRIMARY KEY (guild_id, user_id)
            )
            """
        )
        await db.commit()

async def add_xp(guild_id: int, user_id: int, amount: int):
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            "INSERT INTO levels (guild_id, user_id, xp) VALUES (?, ?, ?) "
            "ON CONFLICT(guild_id, user_id) DO UPDATE SET xp = xp + ?",
            (guild_id, user_id, amount, amount),
        )
        await db.commit()
        cur = await db.execute(
            "SELECT xp, level FROM levels WHERE guild_id = ? AND user_id = ?",
            (guild_id, user_id),
        )
        xp, level = await cur.fetchone()

    leveled_up = False
    while xp >= xp_for_next(level):
        xp -= xp_for_next(level)
        level += 1
        leveled_up = True

    if leveled_up:
        async with aiosqlite.connect(DB_PATH) as db:
            await db.execute(
                "UPDATE levels SET xp = ?, level = ? WHERE guild_id = ? AND user_id = ?",
                (xp, level, guild_id, user_id),
            )
            await db.commit()
        return level
    return None

@bot.event
async def on_ready():
    await init_db()
    await bot.tree.sync()
    print(f"로그인 성공: {bot.user}")

@bot.event
async def on_message(message: discord.Message):
    if message.author.bot or message.guild is None:
        await bot.process_commands(message)
        return

    key = (message.guild.id, message.author.id)
    now = time.time()
    if now - last_xp_time.get(key, 0) >= XP_COOLDOWN:
        last_xp_time[key] = now
        new_level = await add_xp(*key, random.randint(XP_MIN, XP_MAX))
        if new_level:
            channel = (
                bot.get_channel(LEVEL_UP_CHANNEL) if LEVEL_UP_CHANNEL else message.channel
            )
            await channel.send(
                f"🎉 {message.author.mention} 님이 **레벨 {new_level}** 달성!"
            )

    await bot.process_commands(message)

@bot.tree.command(name="레벨", description="내 레벨과 경험치를 확인해요")
async def level_cmd(interaction: discord.Interaction, 유저: discord.Member = None):
    target = 유저 or interaction.user
    async with aiosqlite.connect(DB_PATH) as db:
        cur = await db.execute(
            "SELECT xp, level FROM levels WHERE guild_id = ? AND user_id = ?",
            (interaction.guild_id, target.id),
        )
        row = await cur.fetchone()

    xp, level = row if row else (0, 0)
    need = xp_for_next(level)
    filled = int(20 * xp / need)
    bar = "█" * filled + "░" * (20 - filled)

    embed = discord.Embed(title=f"{target.display_name} 님의 레벨", color=0x5865F2)
    embed.add_field(name="레벨", value=f"**{level}**", inline=True)
    embed.add_field(name="경험치", value=f"{xp} / {need}", inline=True)
    embed.add_field(name="진행도", value=f"`{bar}`", inline=False)
    embed.set_thumbnail(url=target.display_avatar.url)
    await interaction.response.send_message(embed=embed)

@bot.tree.command(name="랭킹", description="서버 상위 10명을 보여줘요")
async def rank_cmd(interaction: discord.Interaction):
    async with aiosqlite.connect(DB_PATH) as db:
        cur = await db.execute(
            "SELECT user_id, level, xp FROM levels WHERE guild_id = ? "
            "ORDER BY level DESC, xp DESC LIMIT 10",
            (interaction.guild_id,),
        )
        rows = await cur.fetchall()

    if not rows:
        await interaction.response.send_message("아직 기록이 없어요! 채팅을 쳐보세요 😄")
        return

    medals = ["🥇", "🥈", "🥉"]
    lines = []
    for i, (user_id, level, xp) in enumerate(rows):
        member = interaction.guild.get_member(user_id)
        name = member.display_name if member else f"(탈퇴한 유저 {user_id})"
        prefix = medals[i] if i < 3 else f"**{i + 1}.**"
        lines.append(f"{prefix} {name} — Lv.{level} ({xp} XP)")

    embed = discord.Embed(
        title=f"🏆 {interaction.guild.name} 랭킹 TOP 10",
        description="\n".join(lines),
        color=0x5865F2,
    )
    await interaction.response.send_message(embed=embed)

bot.run(token)

LEVEL_ROLES = {5: "수다쟁이", 10: "고인물", 20: "터줏대감"}

async def grant_role(member: discord.Member, level: int):
    role_name = LEVEL_ROLES.get(level)
    if not role_name:
        return
    role = discord.utils.get(member.guild.roles, name=role_name)
    if role:
        await member.add_roles(role, reason=f"레벨 {level} 달성")