디스코드 봇 만들기#29화폐·상점(Economy) 시스템 만들기! 코인 벌어서 아이템 사기 (파이썬 예제)

pip install -U discord.py aiosqlite python-dotenv

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

import aiosqlite

DB_PATH = "economy.db"

async def init_db():
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute("""
            CREATE TABLE IF NOT EXISTS wallets (
                guild_id   INTEGER NOT NULL,
                user_id    INTEGER NOT NULL,
                coins      INTEGER NOT NULL DEFAULT 0,
                last_daily TEXT,
                PRIMARY KEY (guild_id, user_id)
            )
        """)
        await db.execute("""
            CREATE TABLE IF NOT EXISTS inventory (
                guild_id INTEGER NOT NULL,
                user_id  INTEGER NOT NULL,
                item_id  TEXT    NOT NULL,
                amount   INTEGER NOT NULL DEFAULT 0,
                PRIMARY KEY (guild_id, user_id, item_id)
            )
        """)
        await db.commit()

CURRENCY = "🪙"

SHOP_ITEMS = {
    "coffee": {"name": "☕ 커피 한 잔",      "price": 50,   "desc": "기분 전환용. 효과는 없어요"},
    "ticket": {"name": "🎟️ 이벤트 응모권",  "price": 300,  "desc": "주간 추첨 이벤트 참가권"},
    "vip":    {"name": "👑 VIP 역할권",     "price": 1000, "desc": "VIP 역할을 받을 수 있어요"},
}

# ❌ 이렇게 하면 안 돼요
row = await select_last_daily(...)      # 1) 마지막 출석일을 읽고
if row[0] != today:                     # 2) 오늘이 아니면
    await add_coins(...)                # 3) 코인을 준다

from datetime import datetime, timedelta, timezone

KST = timezone(timedelta(hours=9))
DAILY_REWARD = 200

async def claim_daily(guild_id: int, user_id: int) -> bool:
    today = datetime.now(KST).date().isoformat()   # '2026-09-17'
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            "INSERT OR IGNORE INTO wallets (guild_id, user_id, coins) VALUES (?, ?, 0)",
            (guild_id, user_id),
        )
        cur = await db.execute(
            "UPDATE wallets SET coins = coins + ?, last_daily = ? "
            "WHERE guild_id = ? AND user_id = ? "
            "AND (last_daily IS NULL OR last_daily <> ?)",
            (DAILY_REWARD, today, guild_id, user_id, today),
        )
        changed = cur.rowcount
        await db.commit()
    return changed > 0

async def buy_item(guild_id: int, user_id: int, item_id: str) -> bool:
    price = SHOP_ITEMS[item_id]["price"]
    async with aiosqlite.connect(DB_PATH) as db:
        cur = await db.execute(
            "UPDATE wallets SET coins = coins - ? "
            "WHERE guild_id = ? AND user_id = ? AND coins >= ?",
            (price, guild_id, user_id, price),
        )
        if cur.rowcount == 0:        # 잔액 부족 → 아무 일도 일어나지 않았어요
            await db.rollback()
            return False
        await db.execute(
            "INSERT INTO inventory (guild_id, user_id, item_id, amount) VALUES (?, ?, ?, 1) "
            "ON CONFLICT(guild_id, user_id, item_id) DO UPDATE SET amount = amount + 1",
            (guild_id, user_id, item_id),
        )
        await db.commit()
    return True

import discord

class ShopView(discord.ui.View):
    def __init__(self, author_id: int):
        super().__init__(timeout=60)
        self.author_id = author_id
        self.message: discord.Message | None = None

    async def interaction_check(self, interaction: discord.Interaction) -> bool:
        if interaction.user.id != self.author_id:
            await interaction.response.send_message(
                "이 상점 창은 다른 분이 연 거예요! `/상점`으로 직접 열어주세요 😅",
                ephemeral=True,
            )
            return False
        return True

    async def on_timeout(self):
        for child in self.children:
            child.disabled = True
        if self.message:
            await self.message.edit(view=self)

    @discord.ui.select(
        placeholder="살 물건을 골라주세요",
        options=[
            discord.SelectOption(
                label=info["name"],
                value=item_id,
                description=f"{info['price']} 코인 · {info['desc']}",
            )
            for item_id, info in SHOP_ITEMS.items()
        ],
    )
    async def pick_item(self, interaction: discord.Interaction, select: discord.ui.Select):
        item_id = select.values[0]
        info = SHOP_ITEMS[item_id]
        ok = await buy_item(interaction.guild_id, interaction.user.id, item_id)
        coins = await get_coins(interaction.guild_id, interaction.user.id)
        if ok:
            msg = f"{info['name']} 구매 완료! 남은 코인 **{coins}** {CURRENCY}"
        else:
            msg = f"코인이 모자라요 😢 (필요 {info['price']} / 보유 {coins})"
        await interaction.response.send_message(msg, ephemeral=True)

# discord.py 2.x (2.7.1 기준) / Python 3.10+
# pip install -U discord.py aiosqlite python-dotenv
import os
from datetime import datetime, timedelta, timezone

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

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

DB_PATH = "economy.db"
KST = timezone(timedelta(hours=9))
DAILY_REWARD = 200          # 출석 보상 코인
CURRENCY = "🪙"

SHOP_ITEMS = {
    "coffee": {"name": "☕ 커피 한 잔",      "price": 50,   "desc": "기분 전환용. 효과는 없어요"},
    "ticket": {"name": "🎟️ 이벤트 응모권",  "price": 300,  "desc": "주간 추첨 이벤트 참가권"},
    "vip":    {"name": "👑 VIP 역할권",     "price": 1000, "desc": "VIP 역할을 받을 수 있어요"},
}

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

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


# ---------------- DB ----------------
async def init_db():
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            """
            CREATE TABLE IF NOT EXISTS wallets (
                guild_id   INTEGER NOT NULL,
                user_id    INTEGER NOT NULL,
                coins      INTEGER NOT NULL DEFAULT 0,
                last_daily TEXT,
                PRIMARY KEY (guild_id, user_id)
            )
            """
        )
        await db.execute(
            """
            CREATE TABLE IF NOT EXISTS inventory (
                guild_id INTEGER NOT NULL,
                user_id  INTEGER NOT NULL,
                item_id  TEXT    NOT NULL,
                amount   INTEGER NOT NULL DEFAULT 0,
                PRIMARY KEY (guild_id, user_id, item_id)
            )
            """
        )
        await db.commit()


async def get_coins(guild_id: int, user_id: int) -> int:
    async with aiosqlite.connect(DB_PATH) as db:
        cur = await db.execute(
            "SELECT coins FROM wallets WHERE guild_id = ? AND user_id = ?",
            (guild_id, user_id),
        )
        row = await cur.fetchone()
    return row[0] if row else 0


async def claim_daily(guild_id: int, user_id: int) -> bool:
    today = datetime.now(KST).date().isoformat()
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            "INSERT OR IGNORE INTO wallets (guild_id, user_id, coins) VALUES (?, ?, 0)",
            (guild_id, user_id),
        )
        cur = await db.execute(
            "UPDATE wallets SET coins = coins + ?, last_daily = ? "
            "WHERE guild_id = ? AND user_id = ? "
            "AND (last_daily IS NULL OR last_daily <> ?)",
            (DAILY_REWARD, today, guild_id, user_id, today),
        )
        changed = cur.rowcount
        await db.commit()
    return changed > 0


async def buy_item(guild_id: int, user_id: int, item_id: str) -> bool:
    price = SHOP_ITEMS[item_id]["price"]
    async with aiosqlite.connect(DB_PATH) as db:
        cur = await db.execute(
            "UPDATE wallets SET coins = coins - ? "
            "WHERE guild_id = ? AND user_id = ? AND coins >= ?",
            (price, guild_id, user_id, price),
        )
        if cur.rowcount == 0:
            await db.rollback()
            return False
        await db.execute(
            "INSERT INTO inventory (guild_id, user_id, item_id, amount) VALUES (?, ?, ?, 1) "
            "ON CONFLICT(guild_id, user_id, item_id) DO UPDATE SET amount = amount + 1",
            (guild_id, user_id, item_id),
        )
        await db.commit()
    return True


async def transfer(guild_id: int, sender_id: int, receiver_id: int, amount: int) -> bool:
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            "INSERT OR IGNORE INTO wallets (guild_id, user_id, coins) VALUES (?, ?, 0)",
            (guild_id, receiver_id),
        )
        cur = await db.execute(
            "UPDATE wallets SET coins = coins - ? "
            "WHERE guild_id = ? AND user_id = ? AND coins >= ?",
            (amount, guild_id, sender_id, amount),
        )
        if cur.rowcount == 0:
            await db.rollback()
            return False
        await db.execute(
            "UPDATE wallets SET coins = coins + ? WHERE guild_id = ? AND user_id = ?",
            (amount, guild_id, receiver_id),
        )
        await db.commit()
    return True


# ---------------- 상점 UI ----------------
class ShopView(discord.ui.View):
    def __init__(self, author_id: int):
        super().__init__(timeout=60)
        self.author_id = author_id
        self.message: discord.Message | None = None

    async def interaction_check(self, interaction: discord.Interaction) -> bool:
        if interaction.user.id != self.author_id:
            await interaction.response.send_message(
                "이 상점 창은 다른 분이 연 거예요! `/상점`으로 직접 열어주세요 😅",
                ephemeral=True,
            )
            return False
        return True

    async def on_timeout(self):
        for child in self.children:
            child.disabled = True
        if self.message:
            await self.message.edit(view=self)

    @discord.ui.select(
        placeholder="살 물건을 골라주세요",
        options=[
            discord.SelectOption(
                label=info["name"],
                value=item_id,
                description=f"{info['price']} 코인 · {info['desc']}",
            )
            for item_id, info in SHOP_ITEMS.items()
        ],
    )
    async def pick_item(self, interaction: discord.Interaction, select: discord.ui.Select):
        item_id = select.values[0]
        info = SHOP_ITEMS[item_id]
        ok = await buy_item(interaction.guild_id, interaction.user.id, item_id)
        coins = await get_coins(interaction.guild_id, interaction.user.id)
        if ok:
            msg = f"{info['name']} 구매 완료! 남은 코인 **{coins}** {CURRENCY}"
        else:
            msg = f"코인이 모자라요 😢 (필요 {info['price']} / 보유 {coins})"
        await interaction.response.send_message(msg, ephemeral=True)


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


@bot.tree.command(name="출석", description="하루 한 번 코인을 받아요")
async def daily_cmd(interaction: discord.Interaction):
    ok = await claim_daily(interaction.guild_id, interaction.user.id)
    coins = await get_coins(interaction.guild_id, interaction.user.id)
    if ok:
        await interaction.response.send_message(
            f"출석 완료! **+{DAILY_REWARD}** {CURRENCY} (현재 {coins} {CURRENCY})"
        )
    else:
        await interaction.response.send_message(
            "오늘은 이미 출석했어요! 한국 시간 자정에 초기화돼요 ⏰", ephemeral=True
        )


@bot.tree.command(name="지갑", description="내 코인과 인벤토리를 확인해요")
@app_commands.describe(유저="다른 사람의 지갑도 볼 수 있어요")
async def wallet_cmd(interaction: discord.Interaction, 유저: discord.Member | None = None):
    target = 유저 or interaction.user
    coins = await get_coins(interaction.guild_id, target.id)

    async with aiosqlite.connect(DB_PATH) as db:
        cur = await db.execute(
            "SELECT item_id, amount FROM inventory "
            "WHERE guild_id = ? AND user_id = ? AND amount > 0",
            (interaction.guild_id, target.id),
        )
        rows = await cur.fetchall()

    lines = [
        f"{SHOP_ITEMS[item_id]['name']} × {amount}"
        for item_id, amount in rows
        if item_id in SHOP_ITEMS
    ]

    embed = discord.Embed(title=f"{target.display_name} 님의 지갑", color=0x5865F2)
    embed.add_field(name="보유 코인", value=f"**{coins}** {CURRENCY}", inline=False)
    embed.add_field(
        name="인벤토리",
        value="\n".join(lines) if lines else "아직 아무것도 없어요 🛒",
        inline=False,
    )
    embed.set_thumbnail(url=target.display_avatar.url)
    await interaction.response.send_message(embed=embed)


@bot.tree.command(name="상점", description="코인으로 아이템을 사요")
async def shop_cmd(interaction: discord.Interaction):
    coins = await get_coins(interaction.guild_id, interaction.user.id)
    embed = discord.Embed(
        title="🛒 상점",
        description=f"보유 코인: **{coins}** {CURRENCY}\n아래에서 살 물건을 골라주세요!",
        color=0x5865F2,
    )
    for info in SHOP_ITEMS.values():
        embed.add_field(
            name=f"{info['name']} — {info['price']} {CURRENCY}",
            value=info["desc"],
            inline=False,
        )

    view = ShopView(interaction.user.id)
    await interaction.response.send_message(embed=embed, view=view)
    view.message = await interaction.original_response()


@bot.tree.command(name="송금", description="다른 멤버에게 코인을 보내요")
@app_commands.describe(받는사람="코인을 받을 사람", 금액="보낼 코인 수")
async def pay_cmd(interaction: discord.Interaction, 받는사람: discord.Member, 금액: int):
    if 금액 < 1:
        await interaction.response.send_message("1코인 이상만 보낼 수 있어요!", ephemeral=True)
        return
    if 받는사람.bot or 받는사람.id == interaction.user.id:
        await interaction.response.send_message(
            "봇이나 자기 자신에게는 보낼 수 없어요 🙅", ephemeral=True
        )
        return

    ok = await transfer(interaction.guild_id, interaction.user.id, 받는사람.id, 금액)
    if ok:
        await interaction.response.send_message(
            f"{interaction.user.mention} → {받는사람.mention} **{금액}** {CURRENCY} 송금 완료!"
        )
    else:
        await interaction.response.send_message("코인이 모자라요 😢", ephemeral=True)


bot.run(token)

@bot.tree.command(name="코인지급", description="(관리자) 코인을 지급하거나 회수해요")
@app_commands.describe(대상="지급할 사람", 금액="음수를 넣으면 회수돼요")
@app_commands.checks.has_permissions(administrator=True)
async def grant_cmd(interaction: discord.Interaction, 대상: discord.Member, 금액: int):
    async with aiosqlite.connect(DB_PATH) as db:
        await db.execute(
            "INSERT OR IGNORE INTO wallets (guild_id, user_id, coins) VALUES (?, ?, 0)",
            (interaction.guild_id, 대상.id),
        )
        await db.execute(
            "UPDATE wallets SET coins = MAX(0, coins + ?) "
            "WHERE guild_id = ? AND user_id = ?",
            (금액, interaction.guild_id, 대상.id),
        )
        await db.commit()

    coins = await get_coins(interaction.guild_id, 대상.id)
    await interaction.response.send_message(
        f"{대상.display_name} 님의 코인을 **{금액:+d}** 조정했어요. (현재 {coins} {CURRENCY})"
    )