mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
115 lines
4.1 KiB
Python
115 lines
4.1 KiB
Python
# coding=utf8
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
"""
|
|
This is the SpiceBot search system.
|
|
"""
|
|
import sopel
|
|
|
|
from sopel_modules.spicemanip import spicemanip
|
|
|
|
import sopel_modules.SpiceBot as SpiceBot
|
|
|
|
|
|
@SpiceBot.prerun('module')
|
|
@sopel.module.commands('search', 'find', 'googled', 'lookup')
|
|
def search_main(bot, trigger, botcom):
|
|
|
|
searchterm = spicemanip(botcom.dict['args'], 0) or None
|
|
if not searchterm:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "Not sure what you want me to look for.")
|
|
return
|
|
|
|
searchdict = {
|
|
"query": searchterm,
|
|
}
|
|
|
|
searchreturn = SpiceBot.search.search(searchdict)
|
|
if not searchreturn:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], 'I cannot find anything about that')
|
|
else:
|
|
bot.osd(["[Information search for '" + str(searchterm) + "']", str(searchreturn)])
|
|
|
|
|
|
@SpiceBot.prerun('module', "search_prefix")
|
|
@sopel.module.commands('(.*)')
|
|
def gifapi_triggers(bot, trigger, botcom):
|
|
|
|
if botcom.dict['com'] not in list(SpiceBot.search.valid_api.keys()):
|
|
return
|
|
|
|
searchterm = spicemanip(botcom.dict['args'], 0) or None
|
|
if not searchterm:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "Not sure what you want me to look for.")
|
|
return
|
|
|
|
searchdict = {
|
|
"query": searchterm,
|
|
"query_type": botcom.dict["realcom"],
|
|
}
|
|
bot.say(str(searchdict))
|
|
|
|
searchreturn = SpiceBot.search.search(searchdict)
|
|
if not searchreturn:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "[" + botcom.dict["realcom"].upper() + " search for '" + str(searchterm) + "']" + 'I cannot find anything about that')
|
|
else:
|
|
bot.osd(["[" + botcom.dict["realcom"].upper() + " search for '" + str(searchterm) + "']", str(searchreturn)])
|
|
|
|
|
|
@SpiceBot.prerun('module')
|
|
@sopel.module.commands("where", "whereis")
|
|
def search_where(bot, trigger, botcom):
|
|
|
|
searchterm = spicemanip(botcom.dict['args'], 0) or None
|
|
if not searchterm:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "Not sure what you want me to look for.")
|
|
return
|
|
searchdict = {
|
|
"query": searchterm,
|
|
"query_type": "gmaps",
|
|
}
|
|
|
|
searchreturn = SpiceBot.search.search(searchdict)
|
|
if not searchreturn:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "[Location search for " + str(searchterm) + "] " + 'I cannot find anything about that')
|
|
else:
|
|
bot.osd(["[Location search for " + str(searchterm) + "]", str(searchreturn)])
|
|
|
|
|
|
@SpiceBot.prerun('module')
|
|
@sopel.module.commands("what", "whatis")
|
|
def search_what(bot, trigger, botcom):
|
|
|
|
searchterm = spicemanip(botcom.dict['args'], 0) or None
|
|
if not searchterm:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "Not sure what you want me to look for.")
|
|
return
|
|
searchdict = {
|
|
"query": searchterm,
|
|
}
|
|
|
|
searchreturn = SpiceBot.search.search(searchdict)
|
|
if not searchreturn:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "[Information search for '" + str(searchterm) + "'] " + 'I cannot find anything about that')
|
|
else:
|
|
bot.osd(["[Information search for '" + str(searchterm) + "']", str(searchreturn)])
|
|
|
|
|
|
@SpiceBot.prerun('module')
|
|
@sopel.module.commands("youtubes", "video")
|
|
def search_video(bot, trigger, botcom):
|
|
|
|
searchterm = spicemanip(botcom.dict['args'], 0) or None
|
|
if not searchterm:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "Not sure what you want me to look for.")
|
|
return
|
|
searchdict = {
|
|
"query": searchterm,
|
|
"query_type": "youtube",
|
|
}
|
|
|
|
searchreturn = SpiceBot.search.search(searchdict)
|
|
if not searchreturn:
|
|
SpiceBot.messagelog.messagelog_error(botcom.dict["log_id"], "[Youtube search for '" + str(searchterm) + "'] " + 'I cannot find anything about that')
|
|
else:
|
|
bot.osd(["[Youtube search for '" + str(searchterm) + "']", str(searchreturn)])
|