134 lines
3.8 KiB
Python
134 lines
3.8 KiB
Python
# coding=utf-8
|
|
|
|
from datetime import timedelta
|
|
from queue import Queue
|
|
import re
|
|
import subprocess
|
|
from threading import Thread
|
|
|
|
from nemubot.hooks import hook
|
|
from nemubot.message import Text
|
|
from nemubot.message.visitor import AbstractVisitor
|
|
|
|
nemubotversion = 3.4
|
|
|
|
queue = Queue()
|
|
spk_th = None
|
|
last = None
|
|
|
|
SMILEY = list()
|
|
CORRECTIONS = list()
|
|
|
|
def load(context):
|
|
for smiley in context.config.getNodes("smiley"):
|
|
if smiley.hasAttribute("txt") and smiley.hasAttribute("mood"):
|
|
SMILEY.append((smiley.getAttribute("txt"), smiley.getAttribute("mood")))
|
|
print ("%d smileys loaded" % len(SMILEY))
|
|
|
|
for correct in context.config.getNodes("correction"):
|
|
if correct.hasAttribute("bad") and correct.hasAttribute("good"):
|
|
CORRECTIONS.append((" " + (correct.getAttribute("bad") + " "), (" " + correct.getAttribute("good") + " ")))
|
|
print ("%d corrections loaded" % len(CORRECTIONS))
|
|
|
|
|
|
class Speaker(Thread):
|
|
|
|
def run(self):
|
|
global queue, spk_th
|
|
while not queue.empty():
|
|
sentence = queue.get_nowait()
|
|
lang = "fr"
|
|
subprocess.call(["espeak", "-v", lang, "--", sentence])
|
|
queue.task_done()
|
|
|
|
spk_th = None
|
|
|
|
|
|
class SpeakerVisitor(AbstractVisitor):
|
|
|
|
def __init__(self, last):
|
|
self.pp = ""
|
|
self.last = last
|
|
|
|
|
|
def visit_Text(self, msg):
|
|
force = (self.last is None)
|
|
|
|
if force or msg.date - self.last.date > timedelta(0, 500):
|
|
self.pp += "A %d heure %d : " % (msg.date.hour, msg.date.minute)
|
|
force = True
|
|
|
|
if force or msg.channel != self.last.channel:
|
|
if msg.to_response == msg.to:
|
|
self.pp += "sur %s. " % (", ".join(msg.to))
|
|
else:
|
|
self.pp += "en message priver. "
|
|
|
|
action = False
|
|
if msg.message.find("ACTION ") == 0:
|
|
self.pp += "%s " % msg.frm
|
|
msg.message = msg.message.replace("ACTION ", "")
|
|
action = True
|
|
for (txt, mood) in SMILEY:
|
|
if msg.message.find(txt) >= 0:
|
|
self.pp += "%s %s : " % (msg.frm, mood)
|
|
msg.message = msg.message.replace(txt, "")
|
|
action = True
|
|
break
|
|
|
|
if not action and (force or msg.frm != self.last.frm):
|
|
self.pp += "%s dit : " % msg.frm
|
|
|
|
if re.match(".*https?://.*", msg.message) is not None:
|
|
msg.message = re.sub(r'https?://([^/]+)[^ ]*', " U.R.L \\1", msg.message)
|
|
|
|
self.pp += msg.message
|
|
|
|
|
|
def visit_DirectAsk(self, msg):
|
|
res = Text("%s: %s" % (msg.designated, msg.message),
|
|
server=msg.server, date=msg.date,
|
|
to=msg.to, frm=msg.frm)
|
|
res.accept(self)
|
|
|
|
|
|
def visit_Command(self, msg):
|
|
res = Text("Bang %s%s%s" % (msg.cmd,
|
|
" " if len(msg.args) else "",
|
|
" ".join(msg.args)),
|
|
server=msg.server, date=msg.date,
|
|
to=msg.to, frm=msg.frm)
|
|
res.accept(self)
|
|
|
|
|
|
def visit_OwnerCommand(self, msg):
|
|
res = Text("Owner Bang %s%s%s" % (msg.cmd,
|
|
" " if len(msg.args) else "",
|
|
" ".join(msg.args)),
|
|
server=msg.server, date=msg.date,
|
|
to=msg.to, frm=msg.frm)
|
|
res.accept(self)
|
|
|
|
|
|
@hook("in")
|
|
def treat_for_speak(msg):
|
|
if not msg.frm_owner:
|
|
append_message(msg)
|
|
|
|
def append_message(msg):
|
|
global last, spk_th
|
|
|
|
if hasattr(msg, "message") and msg.message.find("TYPING ") == 0:
|
|
return
|
|
if last is not None and last.message == msg.message:
|
|
return
|
|
|
|
vprnt = SpeakerVisitor(last)
|
|
msg.accept(vprnt)
|
|
queue.put_nowait(vprnt.pp)
|
|
last = msg
|
|
|
|
if spk_th is None:
|
|
spk_th = Speaker()
|
|
spk_th.start()
|