nemesis
/
asmodee_matrix
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

welcome : ok

master
nemesis 2 years ago
parent
7e776ae498
commit
1705d754a6
1 changed files with 508 additions and 0 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 508
      asmodee.py

508
asmodee.py
View File

@ -11,6 +11,7 @@ import sys
import pickle
import threading
import asyncio
import nio
# Var
bot = None
@ -22,6 +23,11 @@ admins = [] # Admins du bot
modos = {} # Moderateurs des salons
modules = {} # Modules activés par salon
liste_mod = [] # Modules disponibles
citations = {} #addquote
bieres = {} # Boissons a offrir
salons = {} # liste de salons recommandés
welcome = {} # Messages d'accueil des salons
clock = {} # horloges pbta
# Connexion du bot
config = configparser.ConfigParser()
@ -44,6 +50,52 @@ creds = botlib.Creds(SERVER, USERNAME, PASSWORD)
bot = botlib.Bot(creds)
# Chargement des données persistantes
def horloge_read():
global clock
try:
with open("horloge", "rb") as fichier:
loader = pickle.Unpickler(fichier)
clock = loader.load()
except:
clock = {}
def citations_read():
global citations
try:
with open("citations", "rb") as fichier:
loader = pickle.Unpickler(fichier)
citations = loader.load()
except:
citations = {}
def bieres_read():
global bieres
try:
with open("bieres", "rb") as fichier:
loader = pickle.Unpickler(fichier)
bieres = loader.load()
except:
bieres = {}
def salons_read():
global salons
try:
with open("salons", "rb") as fichier:
loader = pickle.Unpickler(fichier)
salons = loader.load()
except:
salons = {}
def welcome_read():
global welcome
try:
with open("welcome", "rb") as fichier:
loader = pickle.Unpickler(fichier)
welcome = loader.load()
except:
welcome = {}
def modos_read():
global modos
try:
@ -69,6 +121,29 @@ def save_obj(room=None, event=None):
with open("modules", "wb") as fichier:
saver = pickle.Pickler(fichier)
saver.dump(modules)
if ('horloge' in liste_mod):
with open("horloge", "wb") as fichier:
saver = pickle.Pickler(fichier)
saver.dump(clock)
if ('quote' in liste_mod):
#with open("quotes", "wb") as fichier:
# saver = pickle.Pickler(fichier)
# saver.dump(quotes)
with open("citations", "wb") as fichier:
saver = pickle.Pickler(fichier)
saver.dump(citations)
if ('biere' in liste_mod):
with open("bieres", "wb") as fichier:
saver = pickle.Pickler(fichier)
saver.dump(bieres)
if ('salon' in liste_mod):
with open("salons", "wb") as fichier:
saver = pickle.Pickler(fichier)
saver.dump(salons)
if ('welcome' in liste_mod):
with open("welcome", "wb") as fichier:
saver = pickle.Pickler(fichier)
saver.dump(welcome)
def signal_handler(signal, frame): # Sauvegarder les données persistantes avant sortie
save_obj(None,None)
@ -671,7 +746,429 @@ async def liste(room, message): # Afficher un élement aléatoire d'une liste fo
await msg(room,nick + " obtient :\n" + '\n'.join(res),nick)
else:
await msg(room,"Utilisation : !liste <options/de/la/liste> [nombre d elements]",nick)
async def aff_hor(room, hor, valeur): # affichage des horloges
if (int(valeur) == 0):
await msg(room,hor + " : " + chr(128320) + "\n", "horloge")
elif (int(valeur) == 1):
await msg(room,hor + " : " + chr(128338) + "\n", "horloge")
elif (int(valeur) == 2):
await msg(room,hor + " : " + chr(128341) + "\n", "horloge")
elif (int(valeur) == 3):
await msg(room,hor + " : " + chr(128344) + "\n", "horloge")
elif (int(valeur) == 4):
await msg(room,hor + " : " + chr(128345) + "\n", "horloge")
elif (int(valeur) == 5):
await msg(room,hor + " : " + chr(128346) + "\n", "horloge")
elif (int(valeur) == 6):
await msg(room,hor + " : " + chr(128347) + "\n", "horloge")
else:
await msg(room,hor + " : " + str(valeur) + " \n", "horloge")
async def horloge(room, message): # Gestion des horloges PBTA
global modules
try:
rmod = modules[room.room_id, 'horloge']
except:
modules[room.room_id, 'horloge'] = False
if (modules[room.room_id, 'horloge']):
global clock
nick = getNick(room,message)
args = getMessage(message).split()
args.pop(0)
if (len(args) == 0):
# Liste des horloges du salon
for cle,valeur in clock.items():
if (cle[0] == room.room_id):
await aff_hor(room, cle[1], valeur)
elif (len(args) >= 2 and args[len(args)-1].isnumeric()):
# Place le niveau de l'horloge entre 0 et 6
nv = int(args[len(args)-1])
clock[room.room_id, ' '.join(args[0:-1])] = nv
elif (len(args) >= 2 and (args[len(args)-1][:1] == "+" or args[len(args)-1][:1] == "-")):
nv = 0
hor = ' '.join(args[0:-1])
if ((room.room_id, hor) in clock):
nv = clock[room.room_id, hor]
val = 0
if (args[len(args)-1][1:].isnumeric()):
val = int(args[len(args)-1][1:])
if (args[len(args)-1][:1] == "+"):
nv = nv + val
if (nv > 6):
nv = 6
else:
nv = nv - val
if (nv < 0):
nv = 0
clock[room.room_id, hor] = nv
await aff_hor(room, hor, nv)
elif (len(args) >= 2 and args[0] == "del"):
# Supprime une horloge
args.pop(0)
del clock[room.room_id, ' '.join(args)]
else:
await msg(room,"Utilisation : \n- !horloge : Affiche les horloges. \n- !horloge <nom> <niveau> : Place ou crée l'horloge au niveau spécifié.\n- !horloge del <nom> : Supprime l'horloge.",getUser(message))
async def reponses(room, phrases, content=False): # Afficher une réponse aléatoire dans la liste
i = random.randrange(0,len(phrases))
await msg(room,phrases[i],"reponses",content)
async def biere(room,message): # Gestion des boissons
global modules
try:
rmod = modules[room.room_id, 'biere']
except:
modules[room.room_id, 'biere'] = False
power = 0
if (modules[room.room_id, 'biere']):
sender = getUser(message)
mess = getMessage(message)
lmod = modos[room.room_id]
if (sender in admins or sender in lmod):
power = 1
global bieres
args = mess.split()
args.pop(0)
breuvages = {}
give = False
elu = ""
envoyeur = getNick(message)
try:
breuvages = bieres[room.room_id]
except:
breuvages = {}
if (len(args) > 1 and args[0] == "give"):
give = True
args.pop(0)
submit = " ".join(args).split(":")
elu = submit[0]
if (power == 0):
elu = elu.replace('@room', 'tout le monde')
try:
args = submit[1].split()
except:
args = ""
if (len(args) > 2 and args[0] == "add"):
#ajoute une biere
args.pop(0)
submit = " ".join(args).split(":")
if (len(submit) > 1):
# Ajout de la citation
nick = submit[0].strip()
nick = nick.replace('@room', 'tous')
submit.pop(0)
citation = ":".join(submit)
citation = citation.replace('@room', 'tout le monde')
tps = datetime.now().microsecond
breuvages[nick,tps] = citation
bieres[room.room_id] = breuvages
await msg(room,"Ajout du breuvage " + nick + ":" + citation,event['sender'])
else:
# message d'aide
await msg(room,"!biere add <nick>:<citation>",sender)
#elif (len(args) > 0 and args[0] == "list"):
# liste les quotes
# args.pop(0)
# salon = verifPMRoom(room,event)
# if (len(args) == 0):
# Liste les breuvages du salon
# reponse = "Voici les reponses du salon " + room.room_id + "\n"
# if (len(room.aliases) > 0):
# reponse = reponse + "(alias : " + room.aliases[0] + "\n"
# for cle,valeur in breuvages.items():
# reponse = reponse + cle[0] + " / " + str(cle[1]) + " : " + valeur + "\n"
# if (len(reponse) > 500):
# msg(salon,reponse,event['sender'])
# reponse = ""
# msg(salon,reponse,event['sender'])
# elif (args[0] == "all"):
# # tous les breuvages de tous les salons
# reponse = "Les breuvages de tous les salons : " + "\n"
# msg(salon,reponse,event['sender'])
# for rid,fame in bieres.items():
# reponse = "Voici les reponses du salon " + rid + "\n"
# for cle,valeur in fame.items():
# reponse = reponse + cle[0] + " / " + str(cle[1]) + " : " + valeur + "\n"
# if (len(reponse) > 500):
# msg(salon,reponse,event['sender'])
# reponse = ""
# msg(salon,reponse,event['sender'])
# else:
# # les breuvages du pseudo spécifié
# nick = " ".join(args)
# reponse = "Voici les reponses du breuvage " + nick + "\n"
# for cle,valeur in breuvages.items():
# if (cle[0] == nick):
# reponse = reponse + cle[0] + " / " + str(cle[1]) + " : " + valeur + "\n"
# msg(salon,reponse,event['sender'])
elif (len(args) > 1 and args[0] == "del"):
# salon = verifPMRoom(room,event)
salon = room.room_id
lmod = modos[room.room_id]
if (sender in admins):
id = args[1]
reponse = "Voici le breuvage supprimé : \n"
for rid,fame in bieres.items():
for cle,valeur in fame.items():
if (str(cle[1]) == id):
del fame[cle[0],cle[1]]
reponse = reponse + rid + " / " + cle[0] + " / " + str(cle[1]) + " : " + valeur + "\n"
break
await msg(salon,reponse,sender)
elif (sender in lmod):
id = args[1]
reponse = "Voici le breuvage supprimé : \n"
rid = room.room_id
fame = bieres[rid]
for cle,valeur in fame.items():
if (str(cle[1]) == id):
del fame[cle[0],cle[1]]
reponse = reponse + rid + " / " + cle[0] + " / " + str(cle[1]) + " : " + valeur + "\n"
break
await msg(salon,reponse,sender)
else:
await msg(salon,"Vous n'êtes pas autorisés à supprimer des breuvages.",sender)
elif (len(args) > 0):
# cite une reponse d'un breuvage precis
nick = " ".join(args)
phrases = []
for cle,valeur in breuvages.items():
if (cle[0] == nick):
if (give):
phrases.append(valeur + " pour " + elu + " de la part de " + envoyeur)
else:
phrases.append(valeur + " pour " + envoyeur)
if (len(phrases) > 0 ):
await reponses(room,phrases)
else:
await msg(room,"Je n'ai pas de " + nick + " en stock!",sender)
else:
# cite une réponse d'un breuvage aleatoire
phrases = []
for cle,valeur in breuvages.items():
if (give):
phrases.append(valeur + " pour " + elu + " de la part de " + envoyeur)
else:
phrases.append(valeur + " pour " + envoyeur)
if (len(phrases) > 0 ):
await reponses(room,phrases)
else:
await msg(room,"Je n'ai pas de breuvages pour ce salon",event['sender'])
async def boissons(room,message): # Gestion des alias de boissons
global modules
try:
rmod = modules[room.room_id, 'biere']
except:
modules[room.room_id, 'biere'] = False
if (modules[room.room_id, 'biere']):
sender = getUser(message)
mess = getMessage(message)
boisson = ""
elu = ""
args = mess.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):
mess = "!biere give " + elu + ":" + boisson
elif (len(elu) > 0):
mess = "!biere give " + elu
elif (len(boisson) > 0):
mess = "!biere " + boisson
else:
mess = "!biere"
message = sender + " : " + mess
await biere(room,message)
async def salon(room,message): # Gestion des listes de salons recommandes
global modules
try:
rmod = modules[room.room_id, 'salon']
except:
modules[room.room_id, 'salon'] = False
if (modules[room.room_id, 'salon']):
sender = getUser(message)
mess = getMessage(message)
global salons
args = mess.split()
args.pop(0)
famous = {}
try:
famous = salons[room.room_id]
except:
famous = {}
if (len(args) > 3 and args[0] == "add"):
#ajoute un salon recommande
ref = ""
desc = ""
args.pop(0)
if (args[0] == "desc"):
args.pop(0)
ref = "desc"
desc = " ".join(args)
else:
ref = args[0]
args.pop(0)
desc = " ".join(args)
famous[ref] = desc
salons[room.room_id] = famous
await msg(room,"Ajout du salon " + ref + " : " + desc,sender)
elif (len(args) > 1 and args[0] == "del"):
#salon = verifPMRoom(room,event)
salon = room.room_id
lmod = modos[room.room_id]
if (sender in admins):
id = args[1]
reponse = "Voici le salon supprimé : \n"
rid = room.room_id
fame = salons[rid]
for cle,valeur in fame.items():
if (str(cle) == id):
del fame[cle]
reponse = reponse + rid + " / " + cle + " : " + valeur + "\n"
break
await msg(room,reponse,sender)
elif (sender in lmod):
id = args[1]
reponse = "Voici le salon supprimé : \n"
rid = room.room_id
fame = salons[rid]
for cle,valeur in fame.items():
if (str(cle) == id):
del fame[cle]
reponse = reponse + rid + " / " + cle + " : " + valeur + "\n"
break
await msg(room,reponse,sender)
else:
await msg(salon,"Vous n'êtes pas autorisés à supprimer des salons.",sender)
elif (len(args) > 0):
# cite un salon precis
ref = args[0]
trouve = 0
for cle,valeur in famous.items():
if (cle == ref):
await msg(salon,cle + " : " + valeur,sender)
trouve = 1
break
if (trouve == 0):
await msg(room,"Je n'ai pas de salons nomme " + ref,sender)
else:
# cite tous les salons
reponse = ""
trouve = 0
if ("desc" in famous):
reponse = reponse + famous["desc"] + "\n"
for cle,valeur in famous.items():
if (cle != "desc"):
reponse = reponse + cle + " : " + valeur + "\n"
trouve = 1
if (trouve == 0):
await msg(room,"Je n'ai pas de salons recommandes pour ce salon",sender)
else:
await msg(room,reponse,sender)
async def accueil(room,message): # Mettre un message d'accueil
global modules
try:
rmod = modules[room.room_id, 'welcome']
except:
modules[room.room_id, 'welcome'] = False
if (modules[room.room_id, 'welcome']):
sender = getUser(message)
mess = getMessage(message)
lmod = modos[room.room_id]
if (sender in admins or sender in lmod):
global welcome
args = mess.split()
args.pop(0)
reponse = " ".join(args)
if (len(args) > 0):
welcome[room.room_id] = reponse
await msg(room,"Le message d'accueil est désormais : " + reponse,sender)
else:
await msg(salon,"Vous n'êtes pas autorisés à modifier le message d'accueil.",sender)
async def bienvenue(room,message): # Affichage d'un message d'accueil aux nouveaux venus
global modules
try:
rmod = modules[room.room_id, 'welcome']
except:
modules[room.room_id, 'welcome'] = False
if (modules[room.room_id, 'welcome']):
sender = message.sender
nick = room.user_name(message.state_key) if room.user_name(message.state_key) else message.state_key
membership = message.membership
prev_member = message.prev_membership
if room.name:
alias = room.name
elif room.canonical_alias:
alias = room.canonical_alias
elif room.names:
alias = room.names[0]
else:
alias = ""
reponse = ""
try:
reponse = welcome[room.room_id]
except:
reponse = ""
if (membership == "join" and prev_member != "join"):
await msg(room,"Bienvenue sur " + alias + ", " + nick + ". " + reponse,sender)
elif (membership == "leave" and prev_member != "leave"):
await msg(room,"Au revoir " + nick + "!",sender)
else:
print("current: " + membership + " prev: " + prev_member)
@bot.listener.on_custom_event(nio.RoomMemberEvent)
async def callBienvenue(room, message):
match = botlib.MessageMatch(room, message, bot)
if match.is_not_from_this_bot():
await bienvenue(room, message)
@bot.listener.on_message_event
async def callAccueil(room, message):
match = botlib.MessageMatch(room, message, bot, PREFIX)
if match.is_not_from_this_bot() and match.prefix() and match.command("accueil"):
await accueil(room, message)
@bot.listener.on_message_event
async def callSalon(room, message):
match = botlib.MessageMatch(room, message, bot, PREFIX)
if match.is_not_from_this_bot() and match.prefix() and match.command("salon"):
await salon(room, message)
@bot.listener.on_message_event
async def callBoisson(room, message):
match = botlib.MessageMatch(room, message, bot, "%")
if match.is_not_from_this_bot() and match.prefix():
await boissons(room, message)
@bot.listener.on_message_event
async def callBiere(room, message):
match = botlib.MessageMatch(room, message, bot, PREFIX)
if match.is_not_from_this_bot() and match.prefix() and match.command("biere"):
await biere(room, message)
@bot.listener.on_message_event
async def callHor(room, message):
match = botlib.MessageMatch(room, message, bot, PREFIX)
if match.is_not_from_this_bot() and match.prefix() and match.command("horloge"):
await horloge(room, message)
@bot.listener.on_message_event
async def callListe(room, message):
@ -736,6 +1233,17 @@ def main():
modules_read()
modos_read()
if ('horloge' in liste_mod):
horloge_read()
if ('quote' in liste_mod):
#quotes_read()
citations_read()
if ('biere' in liste_mod):
bieres_read()
if ('salon' in liste_mod):
salons_read()
if ('welcome' in liste_mod):
welcome_read()
if __name__ == "__main__":
main()

Write Preview
Loading…
Cancel
Save