Bot de sorteos para Discord en Python: botón para participar, ganadores al azar y repetir

discord.py
python-dotenv

DISCORD_TOKEN=pega-aquí-tu-token

import json
import os
import secrets
import time
from pathlib import Path

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

load_dotenv()  # lee el token del fichero .env

# Los sorteos se guardan junto al bot: un reinicio no los borra.
SORTEOS = Path("sorteos.json")
azar = secrets.SystemRandom()  # el azar del sistema: nadie puede adivinar quién gana


def leer() -> dict:
    return json.loads(SORTEOS.read_text("utf-8")) if SORTEOS.exists() else {}


def guardar(datos: dict) -> None:
    SORTEOS.write_text(json.dumps(datos, ensure_ascii=False, indent=2), "utf-8")


class Participar(discord.ui.View):
    def __init__(self):
        super().__init__(timeout=None)  # sin caducidad: el botón sigue valiendo tras reiniciar

    @discord.ui.button(label="Participar", emoji="🎉", style=discord.ButtonStyle.success, custom_id="sorteo:participar")
    async def participar(self, interaction: discord.Interaction, boton: discord.ui.Button):
        datos = leer()
        sorteo = datos.get(str(interaction.message.id))
        if sorteo is None or sorteo["terminado"]:
            await interaction.response.send_message("Este sorteo ya ha terminado.", ephemeral=True)
            return
        apuntados = sorteo["participantes"]
        if interaction.user.id in apuntados:
            apuntados.remove(interaction.user.id)
            texto = "Te has desapuntado. Si cambias de idea, vuelve a pulsar."
        else:
            apuntados.append(interaction.user.id)
            texto = f"¡Estás dentro! Sois {len(apuntados)}. Para desapuntarte, pulsa otra vez."
        guardar(datos)
        await interaction.response.send_message(texto, ephemeral=True)


class Bot(discord.Client):
    def __init__(self):
        super().__init__(intents=discord.Intents.default())
        self.tree = app_commands.CommandTree(self)

    async def setup_hook(self):
        self.add_view(Participar())  # vuelve a escuchar los botones ya publicados
        await self.tree.sync()  # publica /sorteo y /resortear
        revisar.start()


bot = Bot()


async def canal_de(sorteo: dict):
    return bot.get_channel(sorteo["canal"]) or await bot.fetch_channel(sorteo["canal"])


async def anunciar(id_mensaje: str, sorteo: dict, ganadores: list[int]) -> None:
    """Menciona a los ganadores respondiendo a la tarjeta del sorteo."""
    canal = await canal_de(sorteo)
    if ganadores:
        menciones = ", ".join(f"<@{g}>" for g in ganadores)
        texto = f"🎉 ¡Enhorabuena, {menciones}! Premio: **{sorteo['premio']}**."
    else:
        texto = f"Nadie se ha apuntado al sorteo de **{sorteo['premio']}**."
    tarjeta = discord.MessageReference(message_id=int(id_mensaje), channel_id=canal.id, fail_if_not_exists=False)
    await canal.send(texto, reference=tarjeta)


async def cerrar_tarjeta(id_mensaje: str, sorteo: dict) -> None:
    """Quita el botón y apunta cuántos participaron."""
    try:
        mensaje = await (await canal_de(sorteo)).fetch_message(int(id_mensaje))
        tarjeta = mensaje.embeds[0]
        tarjeta.color = 0x888888
        tarjeta.set_footer(text=f"Terminado · {len(sorteo['participantes'])} participantes")
        await mensaje.edit(embed=tarjeta, view=None)
    except (discord.HTTPException, IndexError):
        pass  # han borrado la tarjeta: el resultado ya está anunciado


@tasks.loop(seconds=30)
async def revisar():
    ahora = time.time()
    for id_mensaje, sorteo in leer().items():
        if sorteo["terminado"] or sorteo["fin"] > ahora:
            continue
        # Primero se cierra (nadie más se apunta) y luego se elige.
        datos = leer()
        datos[id_mensaje]["terminado"] = True
        guardar(datos)
        sorteo = datos[id_mensaje]
        apuntados = sorteo["participantes"]
        ganadores = azar.sample(apuntados, min(sorteo["ganadores"], len(apuntados)))
        try:
            await anunciar(id_mensaje, sorteo, ganadores)
            await cerrar_tarjeta(id_mensaje, sorteo)
        except discord.HTTPException as error:
            print(f"No he podido anunciar el sorteo {id_mensaje}: {error!r}")
        datos = leer()  # otra vez: mientras tanto, otros se han podido apuntar a otros sorteos
        datos[id_mensaje]["ganadores_ids"] = ganadores
        guardar(datos)


@revisar.before_loop
async def antes_de_revisar():
    await bot.wait_until_ready()


@bot.tree.command(name="sorteo", description="Empieza un sorteo en este canal")
@app_commands.guild_only()
@app_commands.default_permissions(manage_guild=True)
@app_commands.describe(premio="Qué se sortea", minutos="Cuánto dura: de 1 minuto a 28 días (40320)", ganadores="Cuántos ganan")
async def sorteo(interaction: discord.Interaction, premio: str,
                 minutos: app_commands.Range[int, 1, 40320], ganadores: app_commands.Range[int, 1, 20] = 1):
    fin = int(time.time()) + minutos * 60
    tarjeta = discord.Embed(
        title=f"🎁 Sorteo: {premio}",
        # <t:…:R> es una marca de tiempo de Discord: «dentro de 2 horas», en la hora de cada uno.
        description=f"Pulsa **Participar** para apuntarte.\nTermina <t:{fin}:R> (<t:{fin}:f>).\nGanadores: {ganadores}",
        color=0x5B4AE8,
    )
    tarjeta.set_footer(text=f"Lo organiza {interaction.user.display_name}")
    await interaction.response.send_message(embed=tarjeta, view=Participar())
    mensaje = await interaction.original_response()
    datos = leer()
    datos[str(mensaje.id)] = {
        "canal": interaction.channel_id, "premio": premio, "ganadores": ganadores, "fin": fin,
        "participantes": [], "terminado": False, "ganadores_ids": [],
    }
    guardar(datos)


@bot.tree.command(name="resortear", description="Elige otro ganador de un sorteo terminado")
@app_commands.guild_only()
@app_commands.default_permissions(manage_guild=True)
@app_commands.describe(id_mensaje="Clic derecho en la tarjeta del sorteo → «Copiar ID del mensaje»", cuantos="Cuántos ganadores nuevos")
async def resortear(interaction: discord.Interaction, id_mensaje: str, cuantos: app_commands.Range[int, 1, 20] = 1):
    id_mensaje = id_mensaje.strip()
    sorteo = leer().get(id_mensaje)
    canal = bot.get_channel(sorteo["canal"]) if sorteo else None
    # Que sea de este servidor: el id de otro servidor no vale aquí.
    if sorteo is None or not sorteo["terminado"] or getattr(getattr(canal, "guild", None), "id", None) != interaction.guild_id:
        await interaction.response.send_message("No encuentro un sorteo terminado con ese id en este servidor.", ephemeral=True)
        return
    # Sin repetir: quien ya ganó no entra otra vez.
    quedan = [p for p in sorteo["participantes"] if p not in sorteo["ganadores_ids"]]
    if not quedan:
        await interaction.response.send_message("Ya no queda nadie a quien elegir.", ephemeral=True)
        return
    nuevos = azar.sample(quedan, min(cuantos, len(quedan)))
    await interaction.response.send_message("Elegido. Lo anuncio en el canal del sorteo.", ephemeral=True)
    await anunciar(id_mensaje, sorteo, nuevos)
    datos = leer()
    datos[id_mensaje]["ganadores_ids"] += nuevos
    guardar(datos)


@bot.event
async def on_ready():
    print(f"Conectado como {bot.user}")


bot.run(os.environ["DISCORD_TOKEN"])