135 lines
6.6 KiB
Python
135 lines
6.6 KiB
Python
|
import random
|
||
|
|
|
||
|
|
from bot import traductions as tr
|
||
|
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
def to_str(entier):
|
||
|
|
return str(entier).replace("1", "a").replace("2", "b").replace("3", "c").replace("4", "d").replace("5", "e") \
|
||
|
|
.replace("6", "f").replace("7", "g").replace("8", "h").replace("9", "i").replace("0", "j")
|
||
|
|
|
||
|
|
|
||
|
|
class MainClass:
|
||
|
|
name = "avalon"
|
||
|
|
|
||
|
|
def __init__(self, guild):
|
||
|
|
self.guild = guild
|
||
|
|
# Init database
|
||
|
|
self.curent_games = []
|
||
|
|
self.current_waiting_players = []
|
||
|
|
self.current_roles = []
|
||
|
|
self.current_players = []
|
||
|
|
with self.guild.bot.database.cursor() as cursor:
|
||
|
|
sql_init = "CREATE TABLE IF NOT EXISTS {guild_id}avalon (" \
|
||
|
|
" id int(5) NOT NULL AUTO_INCREMENT PRIMARY KEY," \
|
||
|
|
" nb_joueurs int(5) NOT NULL," \
|
||
|
|
" gentil_a varchar(50) NOT NULL," \
|
||
|
|
" gentil_b varchar(50) NOT NULL," \
|
||
|
|
" gentil_c varchar(50)," \
|
||
|
|
" gentil_d varchar(50)," \
|
||
|
|
" gentil_e varchar(50)," \
|
||
|
|
" gentil_f varchar(50)," \
|
||
|
|
" mechant_a varchar(50) NOT NULL," \
|
||
|
|
" mechant_b varchar(50)," \
|
||
|
|
" mechant_c varchar(50)," \
|
||
|
|
" mechant_d varchar(50)," \
|
||
|
|
" merlin varchar(50) NOT NULL," \
|
||
|
|
" assassin varchar(50) NOT NULL," \
|
||
|
|
" mordred varchar(50)," \
|
||
|
|
" perceval varchar(50)," \
|
||
|
|
" morgane varchar(50)," \
|
||
|
|
" oberon varchar(50)," \
|
||
|
|
" vainqueur varchar(50)" \
|
||
|
|
")".format(guild_id=self.guild.id)
|
||
|
|
cursor.execute(sql_init)
|
||
|
|
|
||
|
|
async def start(self, msg, command, args):
|
||
|
|
if len(self.current_waiting_players) >= 5:
|
||
|
|
await msg.channel.send(
|
||
|
|
tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonstart"]["notenoughtplayers"])
|
||
|
|
return
|
||
|
|
elif len(self.current_waiting_players) != len(self.current_roles):
|
||
|
|
await msg.channel.send(tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonstart"]["rolesnotmatch"])
|
||
|
|
return
|
||
|
|
else:
|
||
|
|
self.current_games.append({
|
||
|
|
"channel": msg.channel.id,
|
||
|
|
"players": self.current_waiting_players,
|
||
|
|
"gentils": [],
|
||
|
|
"mechants": [],
|
||
|
|
"merlin": None,
|
||
|
|
"assassin": None,
|
||
|
|
"mordred": None,
|
||
|
|
"perceval": None,
|
||
|
|
"morgane": None,
|
||
|
|
"oberon": None
|
||
|
|
})
|
||
|
|
for player in self.current_waiting_players:
|
||
|
|
role = random.choose(self.current_roles)
|
||
|
|
self.current_roles.remove()
|
||
|
|
self.current_waiting_players = []
|
||
|
|
|
||
|
|
async def quit(self, msg, command, args):
|
||
|
|
if msg.author.id in self.current_waiting_players:
|
||
|
|
self.current_waiting_players.remove(msg.author.id)
|
||
|
|
await msg.channel.send(tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonquit"]["quit"]
|
||
|
|
.format(player_id=msg.author.id,
|
||
|
|
nb_players=len(self.current_waiting_players)))
|
||
|
|
elif msg.author.id in self.current_players:
|
||
|
|
self.quitting_players.append(msg.author.id)
|
||
|
|
to_del = []
|
||
|
|
# Verify if everyone want to quit game
|
||
|
|
for game in self.curent_games:
|
||
|
|
stop_game = True
|
||
|
|
for game_player in game.players:
|
||
|
|
if game_player not in self.quitting_player:
|
||
|
|
stop_game = False
|
||
|
|
if stop_game:
|
||
|
|
to_del.append(game)
|
||
|
|
for game in to_del:
|
||
|
|
self.curent_games.remove(game)
|
||
|
|
await msg.channel.send(tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonquit"]["alreadyplaying"]
|
||
|
|
.format(player_id=msg.authpr.id))
|
||
|
|
|
||
|
|
async def join(self, msg, command, args):
|
||
|
|
# Personne pas déjà en train d'attendre, ni en train de jouer
|
||
|
|
if msg.author.id not in self.current_waiting_players + self.current_players:
|
||
|
|
self.current_waiting_players.append(msg.author.id)
|
||
|
|
await msg.channel.send(
|
||
|
|
tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonjoin"]["join"]
|
||
|
|
.format(player_id=msg.author.id,
|
||
|
|
nb_players=len(self.current_waiting_players)))
|
||
|
|
if len(self.current_waiting_players) >= 5:
|
||
|
|
await msg.channel.send(tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonjoin"]["canplay"]
|
||
|
|
.format(prefix=self.guild.config["prefix"]))
|
||
|
|
elif msg.author.id in self.current_players:
|
||
|
|
await msg.channel.send(
|
||
|
|
tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonjoin"]["alreadyplaying"]
|
||
|
|
.format(player_id=msg.author.id))
|
||
|
|
elif msg.author.id in self.current_waiting_players:
|
||
|
|
await msg.channel.send(tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonjoin"]["alreadywaiting"]
|
||
|
|
.format(player_id=msg.author.id))
|
||
|
|
|
||
|
|
async def stats(self, msg, command, args):
|
||
|
|
with self.guild.bot.database.cursor() as cursor:
|
||
|
|
cursor.execute("SELECT id,nb_joueurs,vainqueur FROM {guild_id}avalon;".format(guild_id=self.guild.id))
|
||
|
|
results = cursor.fetchall()
|
||
|
|
nb_games = len(results)
|
||
|
|
nb_victoire_gentil = len(list([result for result in results if results["vainqueur"] == "gentil"]))
|
||
|
|
nb_victoire_mechant = nb_games - nb_victoire_gentil
|
||
|
|
await msg.channel.send(tr.tr[self.guild.config["lang"]]["modules"]["avalon"]["avalonstats"]
|
||
|
|
.format(nb_games=nb_games,
|
||
|
|
nb_victoire_gentil=nb_victoire_gentil,
|
||
|
|
nb_victoire_mechant=nb_victoire_mechant))
|
||
|
|
|
||
|
|
async def on_message(self, msg):
|
||
|
|
if msg.content.startswith(self.guild.config["prefix"]):
|
||
|
|
command, *args = msg.content.lstrip(self.guild.config["prefix"]).split(" ")
|
||
|
|
if command == "avalonstats":
|
||
|
|
await self.stats(msg, command, args)
|
||
|
|
elif command == "avalonjoin":
|
||
|
|
await self.join(msg, command, args)
|
||
|
|
elif command == "avalonquit":
|
||
|
|
await self.quit(msg, command, args)
|
||
|
|
return
|