tfjm-discord-bot/src/cogs/teams.py

238 lines
8.4 KiB
Python
Raw Normal View History

2020-04-27 12:46:04 +00:00
from collections import namedtuple
from typing import List, Tuple
2020-04-27 12:46:04 +00:00
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
2020-04-27 12:46:04 +00:00
@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`
"""
2020-04-27 12:46:04 +00:00
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}`"
2020-04-27 12:46:04 +00:00
)
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 !
2020-04-27 13:07:21 +00:00
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)
2020-04-27 13:07:21 +00:00
await ctx.author.add_roles(
team_role,
captain_role,
tournoi,
participant,
reason="Creation of team " + trigram,
2020-04-27 13:07:21 +00:00
)
2020-04-27 12:46:04 +00:00
2020-04-27 13:07:21 +00:00
await ctx.send(
f"L'équipe {team_role.mention} a été créée et son capitaine "
f"est {ctx.author.mention}"
)
2020-04-27 12:46:04 +00:00
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(
2020-04-28 18:08:57 +00:00
f"{member.mention} a été ajouté dans l'équipe {the_team[1].mention}"
)
2020-04-28 18:08:57 +00:00
@team.command(name="channel")
@commands.has_role(Role.CAPTAIN)
2020-04-28 18:08:57 +00:00
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`
"""
2020-04-28 18:08:57 +00:00
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.",
)
2020-04-28 19:03:35 +00:00
@team.command(name="voice", usage="Nom du salon")
@commands.has_role(Role.CAPTAIN)
async def team_channel(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.",
)
2020-04-27 12:46:04 +00:00
def setup(bot: Bot):
bot.add_cog(TeamsCog())