From e5a08b75447fe21247ebee701f47d2df0a03f6a1 Mon Sep 17 00:00:00 2001
From: Nemesis <nemesis@ombreport.info>
Date: Tue, 24 Jul 2018 17:13:18 +0200
Subject: [PATCH] biere %

---
 asmodee.py | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 70 insertions(+), 1 deletion(-)

diff --git a/asmodee.py b/asmodee.py
index 8928ba7..d216e11 100755
--- a/asmodee.py
+++ b/asmodee.py
@@ -33,6 +33,8 @@ clock = {}
 stats = {}
 quotes = {}
 bieres = {}
+welcome = {}
+membres = {}
 admins = [] # Admins du bot
 
 def horloge_read():
@@ -70,6 +72,25 @@ def bieres_read():
 			bieres = loader.load()
 	except:
 		bieres = {}
+
+def welcome_read():
+	global welcome
+	try:
+		with open("welcome", "rb") as fichier:
+			loader = pickle.Unpickler(fichier)
+			welcome = loader.load()
+	except:
+		welcome = {}
+
+def membres_read():
+	global membres
+	try:
+		with open("membres", "rb") as fichier:
+			loader = pickle.Unpickler(fichier)
+			membres = loader.load()
+	except:
+		membres = {}
+
 	
 def signal_handler(signal, frame):
 	with open("horloge", "wb") as fichier:
@@ -84,6 +105,12 @@ def signal_handler(signal, frame):
 	with open("bieres", "wb") as fichier:
 		saver = pickle.Pickler(fichier)
 		saver.dump(bieres)
+	with open("welcome", "wb") as fichier:
+		saver = pickle.Pickler(fichier)
+		saver.dump(welcome)
+	with open("membres", "wb") as fichier:
+		saver = pickle.Pickler(fichier)
+		saver.dump(membres)
 	sys.exit(0)
 
 class Userstats: #Pour garder des stats sur les users
@@ -1168,7 +1195,41 @@ def biere(room,event):
 		else:
 			room.send_text("Je n'ai pas de breuvages pour ce salon")
 
+def boissons(room,event):
+	boisson = ""
+	elu = ""
+	
+	args = event['content']['body'].split()
+	try:
+		boisson = args[0][1:]
+	except:
+		boisson = ""
+	args.pop(0)
+
+	if (len(args) > 0):
+		elu = " ".join(args)
 	
+	if (len(elu) > 0 and len(boisson) > 0):
+		event['content']['body'] = "!biere give " + elu + ":" + boisson
+	elif (len(elu) > 0):
+		event['content']['body'] = "!biere give " + elu
+	elif (len(boisson) > 0):
+		event['content']['body'] = "!biere " + boisson
+	else:
+		event['content']['body'] = "!biere"
+
+def accueil(room,event):
+	if (event['sender'] in admins):
+		global welcome
+		args = event['content']['body'].split()
+		reponse = " ".join(args.pop(0))
+		if (len(args) > 0):
+			welcome[room.room_id] = reponse
+			salon.send_text("Le message d'accueil est désormais : " + reponse)
+			
+	else:
+		salon.send_text("Vous n'êtes pas autorisés à modifier le message d'accueil.")
+		
 def invitations(room_id, state):
 	global bot
 	try:
@@ -1267,7 +1328,15 @@ def main():
 	bieres_read()
 	biere_handler = MCommandHandler("biere", biere)
 	bot.add_handler(biere_handler)
-	
+	boisson_handler = MCommandHandler("", boissons, '%')
+	bot.add_handler(boisson_handler)	
+
+	# Accueil
+	welcome_read()
+	membres_read()
+	welcome_handler = MCommandHandler("accueil", accueil)
+	bot.add_handler(welcome_handler)
+
 	#test_handler = MCommandHandler("test", verifPMRoom)
 	#bot.add_handler(test_handler)