130 lines
3.5 KiB
Python
130 lines
3.5 KiB
Python
# coding=utf-8
|
|
|
|
import imp
|
|
|
|
nemubotversion = 3.3
|
|
|
|
from . import DDGSearch
|
|
from . import WFASearch
|
|
from . import Wikipedia
|
|
|
|
def load(context):
|
|
global CONF
|
|
WFASearch.CONF = CONF
|
|
|
|
from hooks import Hook
|
|
add_hook("cmd_hook", Hook(define, "define"))
|
|
add_hook("cmd_hook", Hook(search, "search"))
|
|
add_hook("cmd_hook", Hook(search, "ddg"))
|
|
add_hook("cmd_hook", Hook(search, "g"))
|
|
add_hook("cmd_hook", Hook(calculate, "wa"))
|
|
add_hook("cmd_hook", Hook(calculate, "calc"))
|
|
add_hook("cmd_hook", Hook(wiki, "dico"))
|
|
add_hook("cmd_hook", Hook(wiki, "wiki"))
|
|
|
|
def reload():
|
|
imp.reload(DDGSearch)
|
|
imp.reload(WFASearch)
|
|
imp.reload(Wikipedia)
|
|
|
|
|
|
def define(msg):
|
|
if len(msg.cmds) <= 1:
|
|
return Response(msg.sender,
|
|
"Indicate a term to define",
|
|
msg.channel, nick=msg.nick)
|
|
|
|
s = DDGSearch.DDGSearch(' '.join(msg.cmds[1:]))
|
|
|
|
res = Response(msg.sender, channel=msg.channel)
|
|
|
|
res.append_message(s.definition)
|
|
|
|
return res
|
|
|
|
|
|
def search(msg):
|
|
if len(msg.cmds) <= 1:
|
|
return Response(msg.sender,
|
|
"Indicate a term to search",
|
|
msg.channel, nick=msg.nick)
|
|
|
|
s = DDGSearch.DDGSearch(' '.join(msg.cmds[1:]))
|
|
|
|
res = Response(msg.sender, channel=msg.channel, nomore="No more results",
|
|
count=" (%d more results)")
|
|
|
|
res.append_message(s.redirect)
|
|
res.append_message(s.abstract)
|
|
res.append_message(s.result)
|
|
res.append_message(s.answer)
|
|
|
|
for rt in s.relatedTopics:
|
|
res.append_message(rt)
|
|
|
|
return res
|
|
|
|
|
|
def calculate(msg):
|
|
if len(msg.cmds) <= 1:
|
|
return Response(msg.sender,
|
|
"Indicate a calcul to compute",
|
|
msg.channel, nick=msg.nick)
|
|
|
|
s = WFASearch.WFASearch(' '.join(msg.cmds[1:]))
|
|
|
|
if s.success:
|
|
res = Response(msg.sender, channel=msg.channel, nomore="No more results")
|
|
for result in s.nextRes:
|
|
res.append_message(result)
|
|
if (len(res.messages) > 0):
|
|
res.messages.pop(0)
|
|
return res
|
|
else:
|
|
return Response(msg.sender, s.error, msg.channel)
|
|
|
|
|
|
def wiki(msg):
|
|
if len(msg.cmds) <= 1:
|
|
return Response(msg.sender,
|
|
"Indicate a term to search",
|
|
msg.channel, nick=msg.nick)
|
|
if len(msg.cmds) > 2 and len(msg.cmds[1]) < 4:
|
|
lang = msg.cmds[1]
|
|
extract = 2
|
|
else:
|
|
lang = "fr"
|
|
extract = 1
|
|
if msg.cmds[0] == "dico":
|
|
site = "wiktionary.org"
|
|
section = 1
|
|
else:
|
|
site = "wikipedia.org"
|
|
section = 0
|
|
|
|
s = Wikipedia.Wikipedia(' '.join(msg.cmds[extract:]), lang, site, section)
|
|
|
|
res = Response(msg.sender, channel=msg.channel, nomore="No more results")
|
|
if site == "wiktionary.org":
|
|
tout = [result for result in s.nextRes if result.find("\x03\x16 :\x03\x16 ") != 0]
|
|
if len(tout) > 0:
|
|
tout.remove(tout[0])
|
|
defI=1
|
|
for t in tout:
|
|
if t.find("# ") == 0:
|
|
t = t.replace("# ", "%d. " % defI)
|
|
defI += 1
|
|
elif t.find("#* ") == 0:
|
|
t = t.replace("#* ", " * ")
|
|
res.append_message(t)
|
|
else:
|
|
for result in s.nextRes:
|
|
res.append_message(result)
|
|
|
|
if len(res.messages) > 0:
|
|
return res
|
|
else:
|
|
return Response(msg.sender,
|
|
"No information about " + " ".join(msg.cmds[extract:]),
|
|
msg.channel)
|