디스코드 봇 만들기#22슬래시 명령어 자동완성(Autocomplete) 만들기! 금지어를 골라서 추가·삭제 (파이썬 예제)

pip install -U discord.py python-dotenv

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

from typing import List

import discord
from discord import app_commands

async def fruit_autocomplete(
    interaction: discord.Interaction,
    current: str,
) -> List[app_commands.Choice[str]]:
    fruits = ["사과", "바나나", "포도", "수박"]
    return [
        app_commands.Choice(name=fruit, value=fruit)
        for fruit in fruits
    ]

@app_commands.command(name="과일")
@app_commands.autocomplete(과일=fruit_autocomplete)
async def pick_fruit(interaction: discord.Interaction, 과일: str):
    await interaction.response.send_message(f"{과일} 골랐네요!")

MAX_CHOICES = 25          # 디스코드가 보여주는 후보 최대 개수
MAX_NAME_LEN = 100        # Choice.name 글자 수 제한

def to_choices(words: List[str]) -> List[app_commands.Choice[str]]:
    return [
        app_commands.Choice(name=word[:MAX_NAME_LEN], value=word[:MAX_NAME_LEN])
        for word in words[:MAX_CHOICES]
    ]

matched = [w for w in words if current.lower() in w.lower()]
return to_choices(matched)

async def preset_autocomplete(self, interaction, current: str):
    category = interaction.namespace.분류      # ← 먼저 고른 값!
    pool = PRESETS.get(category, [])
    if not pool:                               # 아직 안 골랐으면 전체
        pool = [w for words in PRESETS.values() for w in words]

    already = set(load_words(interaction.guild_id or 0))
    matched = [w for w in pool if current.lower() in w.lower() and w not in already]
    return to_choices(matched)

WORD_FILE = Path("data/banned_words.json")

def load_all() -> Dict[str, List[str]]:
    if WORD_FILE.exists():
        with WORD_FILE.open(encoding="utf-8") as f:
            return json.load(f)
    return {}

def load_words(guild_id: int) -> List[str]:
    return load_all().get(str(guild_id), [])

def save_words(guild_id: int, words: List[str]) -> None:
    data = load_all()
    data[str(guild_id)] = words
    WORD_FILE.parent.mkdir(parents=True, exist_ok=True)
    with WORD_FILE.open("w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=2)

# 1) import 추가
from cogs.wordadmin import load_words

# 2) COMPILED_BANNED 상수를 함수로 교체
def banned_patterns(guild_id: int):
    """기본 금지어 + /금지어추가 로 등록한 금지어"""
    words = BANNED_PATTERNS + load_words(guild_id)
    # re.compile 은 내부 캐시(512개)를 쓰기 때문에 매번 불러도 저렴해요
    return [re.compile(word, re.IGNORECASE) for word in words]

# 3) on_message 안에서 함수를 호출
for pattern in banned_patterns(message.guild.id):
    ...

# discord.py 2.x (2.7.1 기준) / Python 3.8+
# 디스코드 봇 만들기 #22 - 슬래시 명령어 자동완성(Autocomplete)

import json
from pathlib import Path
from typing import Dict, List

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

WORD_FILE = Path("data/banned_words.json")
MAX_CHOICES = 25          # 디스코드가 보여주는 자동완성 후보 최대 개수
MAX_NAME_LEN = 100        # Choice.name 글자 수 제한

# /금지어추가 의 분류별 추천 프리셋
PRESETS: Dict[str, List[str]] = {
    "욕설": ["바보", "멍청이", "등신", "머저리"],
    "광고": [r"discord\.gg/\w+", r"https?://\S*(도박|토토)", r"카톡\s*문의", r"텔레\s*@\w+"],
    "도배": [r"(.)\1{9,}", r"[ㅋㅎ]{15,}"],
}

# ---------- 저장소 (#14 JSON 편 방식) ----------
def load_all() -> Dict[str, List[str]]:
    if WORD_FILE.exists():
        with WORD_FILE.open(encoding="utf-8") as f:
            return json.load(f)
    return {}

def load_words(guild_id: int) -> List[str]:
    """automod.py 에서도 이 함수를 가져다 씁니다."""
    return load_all().get(str(guild_id), [])

def save_words(guild_id: int, words: List[str]) -> None:
    data = load_all()
    data[str(guild_id)] = words
    WORD_FILE.parent.mkdir(parents=True, exist_ok=True)
    with WORD_FILE.open("w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=2)

def to_choices(words: List[str]) -> List[app_commands.Choice[str]]:
    """문자열 목록을 Choice 목록으로. 25개·100자 제한을 여기서 한 번에 처리해요."""
    return [
        app_commands.Choice(name=word[:MAX_NAME_LEN], value=word[:MAX_NAME_LEN])
        for word in words[:MAX_CHOICES]
    ]

class WordAdmin(commands.Cog):
    """자동완성으로 금지어를 추가·삭제하는 관리 Cog"""

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

    # ---------- 자동완성 콜백 ----------
    async def preset_autocomplete(
        self,
        interaction: discord.Interaction,
        current: str,
    ) -> List[app_commands.Choice[str]]:
        """분류(namespace)에 따라 다른 후보를 돌려주는 연동 자동완성"""
        category = interaction.namespace.분류          # 먼저 고른 값 읽기
        pool = PRESETS.get(category, [])
        if not pool:                                   # 분류를 아직 안 골랐으면 전체
            pool = [word for words in PRESETS.values() for word in words]

        already = set(load_words(interaction.guild_id or 0))
        matched = [w for w in pool if current.lower() in w.lower() and w not in already]
        return to_choices(matched)

    async def registered_autocomplete(
        self,
        interaction: discord.Interaction,
        current: str,
    ) -> List[app_commands.Choice[str]]:
        """이미 등록된 금지어만 후보로 (삭제용)"""
        words = load_words(interaction.guild_id or 0)
        matched = [w for w in words if current.lower() in w.lower()]
        return to_choices(matched)

    # ---------- 명령어 ----------
    @app_commands.command(name="금지어추가", description="금지어를 추가해요 (자동완성 추천)")
    @app_commands.describe(분류="추천 목록 분류", 단어="추가할 단어 또는 정규식")
    @app_commands.choices(
        분류=[app_commands.Choice(name=key, value=key) for key in PRESETS]
    )
    @app_commands.autocomplete(단어=preset_autocomplete)
    @app_commands.checks.has_permissions(manage_messages=True)
    async def add_word(
        self,
        interaction: discord.Interaction,
        분류: app_commands.Choice[str],
        단어: str,
    ):
        단어 = 단어.strip()
        if not 단어:
            await interaction.response.send_message("빈 값은 추가할 수 없어요 🙅", ephemeral=True)
            return

        words = load_words(interaction.guild_id)
        if 단어 in words:
            await interaction.response.send_message(
                f"`{단어}` 는 이미 등록돼 있어요.", ephemeral=True
            )
            return

        words.append(단어)
        save_words(interaction.guild_id, words)
        await interaction.response.send_message(
            f"✅ `{단어}` 를 **{분류.value}** 금지어로 추가했어요. (총 {len(words)}개)",
            ephemeral=True,
        )

    @app_commands.command(name="금지어삭제", description="등록된 금지어를 골라서 삭제해요")
    @app_commands.describe(단어="삭제할 금지어 (등록된 것만 후보로 떠요)")
    @app_commands.autocomplete(단어=registered_autocomplete)
    @app_commands.checks.has_permissions(manage_messages=True)
    async def remove_word(self, interaction: discord.Interaction, 단어: str):
        words = load_words(interaction.guild_id)
        if 단어 not in words:
            await interaction.response.send_message(
                f"`{단어}` 는 등록되어 있지 않아요. `/금지어목록` 으로 확인해보세요.",
                ephemeral=True,
            )
            return

        words.remove(단어)
        save_words(interaction.guild_id, words)
        await interaction.response.send_message(
            f"🗑️ `{단어}` 를 삭제했어요. (남은 {len(words)}개)", ephemeral=True
        )

    @app_commands.command(name="금지어목록", description="등록된 금지어를 모두 보여줘요")
    async def list_words(self, interaction: discord.Interaction):
        words = load_words(interaction.guild_id)
        if not words:
            await interaction.response.send_message(
                "아직 등록된 금지어가 없어요. `/금지어추가` 로 넣어보세요!", ephemeral=True
            )
            return

        embed = discord.Embed(
            title="🛡️ 등록된 금지어",
            description="\n".join(f"`{word}`" for word in words[:50]),
            color=discord.Color.from_str("#5865F2"),
        )
        embed.set_footer(text=f"총 {len(words)}개")
        await interaction.response.send_message(embed=embed, ephemeral=True)

    # ---------- 권한 에러 처리 ----------
    @add_word.error
    @remove_word.error
    async def permission_error(self, interaction: discord.Interaction, error):
        if isinstance(error, app_commands.MissingPermissions):
            await interaction.response.send_message(
                "이 명령어는 **메시지 관리** 권한이 있어야 쓸 수 있어요 🔒", ephemeral=True
            )

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

server-bot/
├─ .env
├─ .gitignore
├─ bot.py
├─ data/
│  ├─ warns.json
│  └─ banned_words.json   # ← 오늘 자동 생성돼요
└─ cogs/
   ├─ admin.py
   ├─ poll.py             # #18 투표
   ├─ onboarding.py       # #20 환영·자동역할
   ├─ automod.py          # #21 (오늘 3줄 수정!)
   └─ wordadmin.py        # ← 오늘 추가!

✅ cogs.admin 불러옴
✅ cogs.automod 불러옴
✅ cogs.onboarding 불러옴
✅ cogs.poll 불러옴
✅ cogs.wordadmin 불러옴
MojangBot#1234 로그인 성공! (Cog 5개)

@테스터 🚫 사용할 수 없는 표현이에요. 경고 1/3회 누적됐어요.