kl3z
367 lines
17 KiB
Python
367 lines
17 KiB
Python
#!pip install discord.py python-dotenv
|
|
import discord
|
|
from discord.ext import commands
|
|
from discord.ui import View, Select, Button, Modal, TextInput
|
|
from datetime import datetime, timedelta
|
|
import os
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
jogador_em_progresso = None
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
intents.reactions = True
|
|
intents.guilds = True
|
|
intents.members = True
|
|
bot = commands.Bot(command_prefix="!", intents=intents)
|
|
EMOJI_TANK = "🛡️"
|
|
EMOJI_HEALER = "💚"
|
|
EMOJI_DPS = "⚔️"
|
|
DUNGEONS = [
|
|
"Eco-Dome Al'dani",
|
|
"Ara-Kara, City of Echoes",
|
|
"The Dawnbreaker",
|
|
"Operation: Floodgate",
|
|
"Priory of the Sacred Flame",
|
|
"Halls of Atonement",
|
|
"Tazavesh, the Veiled Market: Streets of Wonder",
|
|
"Tazavesh, the Veiled Market: So'leah's Gambit"
|
|
]
|
|
DIFICULDADES = [str(i) for i in range(0, 21)] + ["20+"]
|
|
LIMITES = {"Tank": 1, "Healer": 1, "DPS": 3}
|
|
grupo_mensagem_id = None
|
|
inscritos = {"Tank": [], "Healer": [], "DPS": []}
|
|
classes_escolhidas = {}
|
|
dungeon_escolhida = DUNGEONS[2]
|
|
dificuldade_escolhida = "10"
|
|
data_formatada = "06/08/2025"
|
|
hora_escolhida = "22:00"
|
|
alteracoes_feitas = {"dungeon": False, "dificuldade": False, "data": False}
|
|
def format_grupo_embed():
|
|
def format_role(role):
|
|
players = inscritos[role]
|
|
if not players:
|
|
return "None"
|
|
return "\n".join(f"{p} ({classes_escolhidas.get(p, 'sem classe')})" for p in players)
|
|
|
|
embed = discord.Embed(
|
|
title=f"Dungeon: {dungeon_escolhida}",
|
|
description=(
|
|
f"Dificuldade: {dificuldade_escolhida}\n"
|
|
f"Marcação: {data_formatada} às {hora_escolhida}"
|
|
),
|
|
color=0x00ffcc
|
|
)
|
|
embed.add_field(name=f"{EMOJI_TANK} Tank", value=format_role("Tank"), inline=False)
|
|
embed.add_field(name=f"{EMOJI_HEALER} Healer", value=format_role("Healer"), inline=False)
|
|
embed.add_field(name=f"{EMOJI_DPS} DPS", value=format_role("DPS"), inline=False)
|
|
embed.set_footer(text="Bot created by Kl3z")
|
|
return embed
|
|
|
|
async def criar_grupo_customizado(canal):
|
|
global grupo_mensagem_id, inscritos, dungeon_escolhida, dificuldade_escolhida, data_formatada, hora_escolhida, alteracoes_feitas
|
|
|
|
inscritos = {"Tank": [], "Healer": [], "DPS": []}
|
|
classes_escolhidas.clear()
|
|
dungeon_escolhida = DUNGEONS[2]
|
|
dificuldade_escolhida = "10"
|
|
data_formatada = "06/08/2025"
|
|
hora_escolhida = "22:00"
|
|
alteracoes_feitas = {"dungeon": False, "dificuldade": False, "data": False}
|
|
|
|
embed = format_grupo_embed()
|
|
mensagem = await canal.send(embed=embed, view=DungeonView())
|
|
grupo_mensagem_id = mensagem.id
|
|
|
|
await canal.send(
|
|
"Escolher a tua **Role**:",
|
|
view=RoleView("Novo jogador", canal.id, grupo_mensagem_id)
|
|
)
|
|
|
|
class StackDropdown(Select):
|
|
def __init__(self):
|
|
options = [
|
|
discord.SelectOption(label="Plate Stack"),
|
|
discord.SelectOption(label="Leather Stack"),
|
|
discord.SelectOption(label="Mail Stack"),
|
|
discord.SelectOption(label="Cloth Stack")
|
|
]
|
|
super().__init__(placeholder="Escolhe o tipo de stack", options=options)
|
|
|
|
async def callback(self, interaction: discord.Interaction):
|
|
escolha = self.values[0]
|
|
forum_channel = discord.utils.get(interaction.guild.channels, name="lfg", type=discord.ChannelType.forum)
|
|
|
|
if not forum_channel:
|
|
await interaction.response.send_message("❌ Canal de fórum 'lfg' não encontrado ou não é do tipo fórum.", ephemeral=True)
|
|
return
|
|
|
|
# Cria o post no fórum
|
|
thread = await forum_channel.create_thread(name=escolha, content=escolha)
|
|
|
|
# Apaga a mensagem original com a combobox
|
|
await interaction.message.delete()
|
|
|
|
# Evita erro de timeout
|
|
await interaction.response.defer()
|
|
|
|
|
|
|
|
|
|
|
|
@bot.command(name="bot")
|
|
async def bot_command(ctx):
|
|
await ctx.send(view=StackView())
|
|
|
|
class StackView(View):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.add_item(StackDropdown())
|
|
|
|
CLASSES_POR_ROLE = {
|
|
"Tank": [
|
|
("Protection Paladin", "https://wow.zamimg.com/images/wow/icons/large/spell_holy_avengershield.jpg"),
|
|
("Protection Warrior", "https://wow.zamimg.com/images/wow/icons/large/spell_warrior_protetion_spec.jpg"),
|
|
("Blood Death Knight", "https://wow.zamimg.com/images/wow/icons/large/spell_deathknight_bloodpresence.jpg"),
|
|
("Vengeance Demon Hunter","https://wow.zamimg.com/images/wow/icons/large/ability_demonhunter_specdps.jpg"),
|
|
("Brewmaster Monk", "https://wow.zamimg.com/images/wow/icons/large/spell_monk_brewmaster_spec.jpg"),
|
|
("Guardian Druid", "https://wow.zamimg.com/images/wow/icons/large/ability_racial_bearform.jpg")
|
|
],
|
|
"Healer": [
|
|
("Discipline Priest", "https://wow.zamimg.com/images/wow/icons/large/spell_holy_powerwordshield.jpg"),
|
|
("Restoration Shaman", "https://wow.zamimg.com/images/wow/icons/large/spell_nature_magicimmunity.jpg"),
|
|
("Mistweaver Monk", "https://wow.zamimg.com/images/wow/icons/large/spell_monk_mistweaver_spec.jpg"),
|
|
("Restoration Druid", "https://wow.zamimg.com/images/wow/icons/large/spell_nature_healingtouch.jpg"),
|
|
("Holy Paladin", "https://wow.zamimg.com/images/wow/icons/large/spell_holy_holybolt.jpg"),
|
|
("Preservation Evoker", "https://wow.zamimg.com/images/wow/icons/large/spell_spec_evoker_preservation.jpg")
|
|
],
|
|
"DPS": [
|
|
("Arcane Mage", "https://wow.zamimg.com/images/wow/icons/large/spell_holy_magicalsentry.jpg"),
|
|
("Balance Druid", "https://wow.zamimg.com/images/wow/icons/large/spell_nature_starfall.jpg"),
|
|
("Unholy Death Knight", "https://wow.zamimg.com/images/wow/icons/large/spell_deathknight_unholypresence.png"),
|
|
("Retribution Paladin", "https://wow.zamimg.com/images/wow/icons/large/spell_holy_auraoflight.jpg"),
|
|
("Fire Mage", "https://wow.zamimg.com/images/wow/icons/large/spell_fire_flamebolt.jpg"),
|
|
("Elemental Shaman", "https://wow.zamimg.com/images/wow/icons/large/spell_nature_lightning.jpg"),
|
|
("Enhancement Shaman", "https://wow.zamimg.com/images/wow/icons/large/spell_shaman_improvedstormstrike.jpg"),
|
|
("Marksmanship Hunter", "https://wow.zamimg.com/images/wow/icons/large/ability_hunter_focusedaim.jpg"),
|
|
("Havoc Demon Hunter", "https://wow.zamimg.com/images/wow/icons/large/ability_demonhunter_specdps.jpg"),
|
|
("Beast Mastery Hunter", "https://wow.zamimg.com/images/wow/icons/large/ability_hunter_bestialdiscipline.jpg"),
|
|
("Assassination Rogue", "https://wow.zamimg.com/images/wow/icons/large/ability_rogue_deadlybrew.jpg"),
|
|
("Outlaw Rogue", "https://wow.zamimg.com/images/wow/icons/large/ability_rogue_waylay.jpg"),
|
|
("Shadow Priest", "https://wow.zamimg.com/images/wow/icons/large/spell_shadow_shadowwordpain.jpg"),
|
|
("Frost Death Knight", "https://wow.zamimg.com/images/wow/icons/large/spell_deathknight_frostpresence.png"),
|
|
("Windwalker Monk", "https://wow.zamimg.com/images/wow/icons/large/spell_monk_windwalkerspec.jpg"),
|
|
("Fury Warrior", "https://wow.zamimg.com/images/wow/icons/large/ability_warrior_innerrage.jpg"),
|
|
("Arms Warrior", "https://wow.zamimg.com/images/wow/icons/large/ability_warrior_savageblow.jpg"),
|
|
("Affliction Warlock", "https://wow.zamimg.com/images/wow/icons/large/spell_shadow_deathcoil.jpg"),
|
|
("Destruction Warlock", "https://wow.zamimg.com/images/wow/icons/large/spell_shadow_rainoffire.jpg"),
|
|
("Feral Druid", "https://wow.zamimg.com/images/wow/icons/large/ability_druid_catform.jpg"),
|
|
("Devastation Evoker", "https://wow.zamimg.com/images/wow/icons/large/spell_spec_evoker_devastation.jpg"),
|
|
("Survival Hunter", "https://wow.zamimg.com/images/wow/icons/large/ability_hunter_camouflage.jpg"),
|
|
("Frost Mage", "https://wow.zamimg.com/images/wow/icons/large/spell_frost_frostbolt02.jpg")
|
|
]
|
|
}
|
|
class DungeonDropdown(Select):
|
|
def __init__(self):
|
|
options = [discord.SelectOption(label=d, value=d) for d in DUNGEONS]
|
|
super().__init__(placeholder="Escolhe a dungeon", min_values=1, max_values=1, options=options)
|
|
|
|
async def callback(self, interaction: discord.Interaction):
|
|
global dungeon_escolhida
|
|
if alteracoes_feitas["dungeon"]:
|
|
await interaction.response.send_message("❌ A dungeon já foi escolhida e não pode ser alterada novamente.", ephemeral=True)
|
|
return
|
|
dungeon_escolhida = self.values[0]
|
|
alteracoes_feitas["dungeon"] = True
|
|
|
|
mensagem = await interaction.channel.fetch_message(grupo_mensagem_id)
|
|
nova_view = DungeonView()
|
|
if all(alteracoes_feitas.values()):
|
|
nova_view.clear_items()
|
|
|
|
await mensagem.edit(embed=format_grupo_embed(), view=nova_view)
|
|
if isinstance(interaction.channel, discord.Thread):
|
|
await interaction.channel.edit(name=f"{dungeon_escolhida} - Key {dificuldade_escolhida}")
|
|
|
|
await interaction.response.defer()
|
|
|
|
class DificuldadeDropdown(Select):
|
|
def __init__(self):
|
|
options = [discord.SelectOption(label=f"Chave {d}", value=d) for d in DIFICULDADES]
|
|
super().__init__(placeholder="Escolhe a dificuldade", min_values=1, max_values=1, options=options)
|
|
|
|
async def callback(self, interaction: discord.Interaction):
|
|
global dificuldade_escolhida
|
|
if alteracoes_feitas["dificuldade"]:
|
|
await interaction.response.send_message("❌ A dificuldade já foi definida e não pode ser alterada novamente.", ephemeral=True)
|
|
return
|
|
dificuldade_escolhida = self.values[0]
|
|
alteracoes_feitas["dificuldade"] = True
|
|
mensagem = await interaction.channel.fetch_message(grupo_mensagem_id)
|
|
nova_view = DungeonView()
|
|
if all(alteracoes_feitas.values()):
|
|
nova_view.clear_items()
|
|
await mensagem.edit(embed=format_grupo_embed(), view=nova_view)
|
|
if isinstance(interaction.channel, discord.Thread):
|
|
await interaction.channel.edit(name=f"{dungeon_escolhida} - Key {dificuldade_escolhida}")
|
|
|
|
await interaction.response.defer()
|
|
|
|
class DataModal(Modal, title="Definir Data da Dungeon"):
|
|
dia = TextInput(label="Dia do mês (1-31)", placeholder="Ex: 6", max_length=2)
|
|
hora = TextInput(label="Hora (HH:MM)", placeholder="Ex: 22:30", max_length=5)
|
|
async def on_submit(self, interaction: discord.Interaction):
|
|
global data_formatada, hora_escolhida
|
|
if alteracoes_feitas["data"]:
|
|
await interaction.response.send_message("❌ A data e hora já foram definidas e não podem ser alteradas novamente.", ephemeral=True)
|
|
return
|
|
try:
|
|
dia_num = int(self.dia.value)
|
|
if not 1 <= dia_num <= 31:
|
|
raise ValueError("Dia fora do intervalo")
|
|
hora_val = datetime.strptime(self.hora.value, "%H:%M").strftime("%H:%M")
|
|
hoje = datetime.now()
|
|
mes = hoje.month
|
|
ano = hoje.year
|
|
if dia_num < hoje.day:
|
|
mes += 1
|
|
if mes > 12:
|
|
mes = 1
|
|
ano += 1
|
|
data_completa = datetime(year=ano, month=mes, day=dia_num)
|
|
except Exception:
|
|
await interaction.response.send_message("❌ Data ou hora inválida.", ephemeral=True)
|
|
return
|
|
data_formatada = data_completa.strftime("%d/%m/%Y")
|
|
hora_escolhida = hora_val
|
|
alteracoes_feitas["data"] = True
|
|
mensagem = await interaction.channel.fetch_message(grupo_mensagem_id)
|
|
nova_view = DungeonView()
|
|
if all(alteracoes_feitas.values()):
|
|
nova_view.clear_items()
|
|
await mensagem.edit(embed=format_grupo_embed(), view=nova_view)
|
|
await interaction.response.defer()
|
|
class BotaoData(Button):
|
|
def __init__(self):
|
|
super().__init__(label="🗓️ Definir Data", style=discord.ButtonStyle.primary)
|
|
async def callback(self, interaction: discord.Interaction):
|
|
await interaction.response.send_modal(DataModal())
|
|
class DungeonView(View):
|
|
def __init__(self):
|
|
super().__init__(timeout=None)
|
|
if not alteracoes_feitas["dungeon"]:
|
|
self.add_item(DungeonDropdown())
|
|
if not alteracoes_feitas["dificuldade"]:
|
|
self.add_item(DificuldadeDropdown())
|
|
if not alteracoes_feitas["data"]:
|
|
self.add_item(BotaoData())
|
|
class RoleDropdown(Select):
|
|
def __init__(self, jogador, ctx_channel_id, role_msg_id):
|
|
self.jogador = jogador
|
|
self.ctx_channel_id = ctx_channel_id
|
|
self.role_msg_id = role_msg_id
|
|
options = []
|
|
for role in LIMITES:
|
|
if len(inscritos[role]) < LIMITES[role]:
|
|
options.append(discord.SelectOption(label=role, description="Escolher esta role"))
|
|
|
|
placeholder = "Escolhe a tua role" if options else "Todas as roles preenchidas"
|
|
super().__init__(placeholder=placeholder, options=options, disabled=(len(options) == 0))
|
|
|
|
async def callback(self, interaction: discord.Interaction):
|
|
global jogador_em_progresso
|
|
jogador = interaction.user.display_name
|
|
if jogador_em_progresso and jogador_em_progresso != jogador:
|
|
await interaction.response.send_message(
|
|
f"⏳ Aguarda que {jogador_em_progresso} termine a escolha da classe.",
|
|
ephemeral=True
|
|
)
|
|
return
|
|
jogador_em_progresso = jogador
|
|
role = self.values[0]
|
|
for r in inscritos:
|
|
if jogador in inscritos[r]:
|
|
inscritos[r].remove(jogador)
|
|
if jogador not in inscritos[role] and len(inscritos[role]) < LIMITES[role]:
|
|
inscritos[role].append(jogador)
|
|
canal = bot.get_channel(self.ctx_channel_id)
|
|
if isinstance(canal, discord.Thread):
|
|
await canal.join()
|
|
await interaction.message.delete()
|
|
mensagem = await canal.fetch_message(grupo_mensagem_id)
|
|
await mensagem.edit(embed=format_grupo_embed())
|
|
await canal.send(
|
|
f"{jogador}, agora escolhe a tua classe para **{role}**:",
|
|
view=ClasseView(role, jogador, self.ctx_channel_id)
|
|
)
|
|
await interaction.response.defer()
|
|
else:
|
|
jogador_em_progresso = None
|
|
await interaction.response.send_message("❌ Esta role já está cheia ou ocorreu um erro.", ephemeral=True)
|
|
class ClasseDropdown(Select):
|
|
def __init__(self, role, jogador):
|
|
self.jogador = jogador
|
|
options = [discord.SelectOption(label=nome, value=nome) for nome, _ in CLASSES_POR_ROLE[role]]
|
|
super().__init__(placeholder=f"Escolhe a classe ({role})", options=options)
|
|
async def callback(self, interaction: discord.Interaction):
|
|
global jogador_em_progresso
|
|
classes_escolhidas[self.jogador] = self.values[0]
|
|
canal = interaction.channel
|
|
mensagem = await canal.fetch_message(grupo_mensagem_id)
|
|
await mensagem.edit(embed=format_grupo_embed())
|
|
await interaction.message.delete()
|
|
jogador_em_progresso = None
|
|
if any(len(inscritos[r]) < LIMITES[r] for r in LIMITES):
|
|
await canal.send(
|
|
"Escolher a tua **Role**:",
|
|
view=RoleView(interaction.user.display_name, canal.id, grupo_mensagem_id)
|
|
)
|
|
await interaction.response.defer()
|
|
class ClasseView(View):
|
|
def __init__(self, role, jogador, ctx_channel_id):
|
|
super().__init__()
|
|
dropdown = ClasseDropdown(role, jogador)
|
|
dropdown.ctx_channel_id = ctx_channel_id
|
|
self.add_item(dropdown)
|
|
import asyncio
|
|
async def agendar_remocao_thread(thread: discord.Thread, data_str: str, hora_str: str):
|
|
try:
|
|
alvo = datetime.strptime(f"{data_str} {hora_str}", "%d/%m/%Y %H:%M")
|
|
agora = datetime.now()
|
|
segundos_ate_apagar = (alvo + timedelta(minutes=30) - agora).total_seconds()
|
|
if segundos_ate_apagar > 0:
|
|
await asyncio.sleep(segundos_ate_apagar)
|
|
await thread.delete()
|
|
print(f"[INFO] Thread '{thread.name}' apagada automaticamente após 30 minutos.")
|
|
else:
|
|
print(f"[INFO] Tempo já passou, thread '{thread.name}' não foi agendada.")
|
|
except Exception as e:
|
|
print(f"[ERRO] ao agendar remoção: {e}")
|
|
@bot.command(name="criargrupo")
|
|
async def criar_grupo(ctx):
|
|
global grupo_mensagem_id, inscritos, dungeon_escolhida, dificuldade_escolhida, data_formatada, hora_escolhida, alteracoes_feitas
|
|
inscritos = {"Tank": [], "Healer": [], "DPS": []}
|
|
classes_escolhidas.clear()
|
|
dungeon_escolhida = DUNGEONS[2]
|
|
dificuldade_escolhida = "10"
|
|
data_formatada = "06/08/2025"
|
|
hora_escolhida = "22:00"
|
|
alteracoes_feitas = {"dungeon": False, "dificuldade": False, "data": False}
|
|
embed = format_grupo_embed()
|
|
mensagem = await ctx.send(embed=embed, view=DungeonView())
|
|
grupo_mensagem_id = mensagem.id
|
|
if isinstance(ctx.channel, discord.Thread):
|
|
asyncio.create_task(agendar_remocao_thread(ctx.channel, data_formatada, hora_escolhida))
|
|
role_msg = await ctx.send(
|
|
"Escolher a tua **Role**:",
|
|
view=RoleView(ctx.author.display_name, ctx.channel.id, grupo_mensagem_id)
|
|
)
|
|
class RoleView(View):
|
|
def __init__(self, jogador, ctx_channel_id, role_msg_id):
|
|
super().__init__(timeout=None)
|
|
self.add_item(RoleDropdown(jogador, ctx_channel_id, role_msg_id))
|
|
@bot.event
|
|
async def on_ready():
|
|
print(f"Bot ligado como {bot.user}")
|
|
bot.run(os.getenv("DISCORD_TOKEN"))
|
|
|