from collections import namedtuple from typing import List, Tuple import discord from discord.ext import commands from discord.ext.commands import Cog, Bot, group, Context from discord.utils import get, find from src.constants import * Team = namedtuple("Team", ["name", "trigram", "tournoi", "secret", "status"]) class TeamsCog(Cog, name="Teams"): def __init__(self): self.teams = self.load_teams() def load_teams(self): with open(TEAMS_FILE) as f: # first line is header lines = f.read().splitlines()[1:] teams = [ Team(*[field.strip('"') for field in line.split(";")]) for line in lines ] return teams def teams_for(self, member) -> List[Tuple[Team, discord.Role]]: """Return a list of pairs (role, team) corresponding to the teams of the member""" teams = [] for role in member.roles: team = get(self.teams, trigram=role.name) if team: teams.append((team, role)) return teams @group(name="team") async def team(self, ctx): """Groupe de commandes pour la gestion des équipes.""" @team.command(name="create") async def create_team(self, ctx: Context, trigram, team_secret): """ Permet aux capitaines de créer leur equipes. Pour utiliser cette commande, il faut ton trigram et ton code d'équipe. Tu peux ensuite écrire `!team create TRIGRAM SECRET` en gradant le point d'éclamation et en remplaçant `TRIGRAM` et `SECRET` par les bonnes valeurs. Le message que tu envoie sera immédiatement supprimé pour pas que d'autres voient ton secret, donc ne t'inquiète pas si il disparait. Exemple: `!team create FOX abq23j` """ await ctx.message.delete() team: Team = get(self.teams, trigram=trigram) role: discord.Role = get(ctx.guild.roles, name=trigram) captain_role = get(ctx.guild.roles, name=Role.CAPTAIN) if team is None: await ctx.send( f"{ctx.author.mention}: le trigram `{trigram}` " f"n'est pas valide. Es-tu sûr d'avoir le bon ?" ) elif role is not None: # Team exists captain = find(lambda m: captain_role in m.roles, role.members) await ctx.send( f"{ctx.author.mention}: l'équipe {trigram} " f"existe déjà. Tu peux demander a ton capitaine " f"{captain.mention} de t'ajouter à l'équipe avec " f"`!team add {ctx.author.name}`" ) elif team_secret != team.secret: await ctx.send( f"{ctx.author.mention}: ton secret n'est pas valide, " f"Tu peux le trouver sur https://inscription.tfjm.org/mon-equipe." ) else: # Team creation ! guild: discord.Guild = ctx.guild team_role = await guild.create_role( name=trigram, color=discord.Colour.from_rgb(255, 255, 255), reason="Creation of a new team", ) tournoi = get(guild.roles, name=team.tournoi) participant = get(guild.roles, name=Role.PARTICIPANT) await ctx.author.add_roles( team_role, captain_role, tournoi, participant, reason="Creation of team " + trigram, ) await ctx.send( f"L'équipe {team_role.mention} a été créée et son capitaine " f"est {ctx.author.mention}" ) diego = get(ctx.guild.members, display_name=DIEGO, top_role__name=Role.CNO) await ctx.author.send( f"Salut Capitaine !\n" "On va être amené à faire de nombreuses choses ensemble " "ces prochains jours, donc n'hésite pas à abuser de `!help`. " "Tu peux l'utiliser ici mais malheureusement tu ne pourra pas voir " "les commandes qui sont réservés aux capitaines. \n" "Une commande que tu peux avoir envie d'utiliser c'est " "`!team channel un-super-nom` pour créer une channel réservée à " "ton équipe. \n\n" f"Si tu as des suggestions pour que le bot permette à chacun d'avoir " f"une meilleure expérience ici, envoie un petit message à {diego.mention} ;)" ) @team.command(name="add") @commands.has_role(Role.CAPTAIN) async def team_add(self, ctx, member: discord.Member): """ (cap) Ajoute un membre a ton équipe. Commande réservée aux capitaines pour ajouter un membre dans leur équipe. Cela permet juste de donner les bons roles pour que les bonnes *channels* soient accessibles. Exemple: `!team add @Jean-Mich-Much` """ author_teams = self.teams_for(ctx.author) member_teams = self.teams_for(member) if member_teams: await ctx.send( f"{member.mention} est déjà dans une équipe " f"et ne peut pas être dans deux à la fois." ) elif len(author_teams) > 1: await ctx.send( f"Tu est dans plusieurs équipes, je ne sais " f"pas où l'ajouter. Il faut demander à un organisateur " f"de le faire." ) else: the_team = author_teams[0] tournoi = get(ctx.guild.roles, name=the_team[0].tournoi) participant = get(ctx.guild.roles, name=Role.PARTICIPANT) await member.add_roles( the_team[1], tournoi, participant, reason=f"{ctx.author.name} l'a ajouté à son équipe", ) await ctx.send( f"{member.mention} a été ajouté dans l'équipe {the_team[1].mention}" ) @team.command(name="channel") @commands.has_role(Role.CAPTAIN) async def team_channel(self, ctx, *channel_name): """ (cap) Crée une channel privée pour l'équipe Crée un endroit de discussion privé seulement pour l'équipe personne d'autre n'y aura accès. Exemple: `!team channel un-nom-sympa` """ if not channel_name: await ctx.send( "Tu dois mettre un nom d'équipe, par exemple " "`!team channel un-super-nom`" ) return channel_name = " ".join(channel_name) guild: discord.Guild = ctx.guild team_role = self.teams_for(ctx.author)[0][1] team_channel_category = get(guild.categories, name=TEAMS_CHANNEL_CATEGORY) await guild.create_text_channel( channel_name, overwrites={ guild.default_role: discord.PermissionOverwrite(read_messages=False), team_role: discord.PermissionOverwrite(read_messages=True), }, category=team_channel_category, reason=f"{ctx.author.name} à demandé une channel pour son équipe.", ) @team.command(name="voice", usage="Nom du salon") @commands.has_role(Role.CAPTAIN) async def team_voice(self, ctx, *channel_name): """ (cap) Crée une channel vocale privée pour l'équipe Crée un endroit de discussion privé seulement pour l'équipe personne d'autre n'y aura accès. Exemple: `!team voice un-nom-sympa` """ if not channel_name: await ctx.send( "Tu dois mettre un nom d'équipe, par exemple " "`!team channel un-super-nom`" ) return channel_name = " ".join(channel_name) guild: discord.Guild = ctx.guild team_role = self.teams_for(ctx.author)[0][1] team_channel_category = get(guild.categories, name=TEAMS_CHANNEL_CATEGORY) await guild.create_voice_channel( channel_name, overwrites={ guild.default_role: discord.PermissionOverwrite(read_messages=False), team_role: discord.PermissionOverwrite(read_messages=True), }, category=team_channel_category, reason=f"{ctx.author.name} à demandé une channel vocale pour son équipe.", ) def setup(bot: Bot): bot.add_cog(TeamsCog())