mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
96 lines
3.9 KiB
Python
96 lines
3.9 KiB
Python
# coding=utf-8
|
|
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
|
|
import sopel
|
|
import sopel.module
|
|
|
|
from sopel_modules.SpiceBot_Events.System import bot_events_check
|
|
from sopel_modules.SpiceBot_SBTools import sopel_triggerargs, similar_list, letters_in_string
|
|
|
|
|
|
import spicemanip
|
|
|
|
|
|
@sopel.module.rule('^\?(.*)')
|
|
def query_detection(bot, trigger):
|
|
|
|
while not bot_events_check(bot, '2002'):
|
|
pass
|
|
|
|
triggerargs, triggercommand = sopel_triggerargs(bot, trigger, 'query_command')
|
|
|
|
# command issued, check if valid
|
|
if not triggercommand or not len(triggercommand):
|
|
return
|
|
|
|
if not letters_in_string(triggercommand):
|
|
return
|
|
|
|
commands_list = dict()
|
|
for commandstype in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
|
|
if commandstype != 'rule':
|
|
for com in bot.memory['SpiceBot_CommandsQuery']['commands'][commandstype].keys():
|
|
if com not in commands_list.keys():
|
|
if commandstype == 'nickname':
|
|
commands_list[str(bot.nick) + " " + com] = bot.memory['SpiceBot_CommandsQuery']['commands'][commandstype][com]
|
|
if "aliasfor" in commands_list[str(bot.nick) + " " + com].keys():
|
|
aliasforcom = commands_list[str(bot.nick) + " " + com]["aliasfor"]
|
|
commands_list[str(bot.nick) + " " + com]["aliasfor"] = str(bot.nick) + " " + aliasforcom
|
|
else:
|
|
commands_list[com] = bot.memory['SpiceBot_CommandsQuery']['commands'][commandstype][com]
|
|
|
|
botnick_coms = list(bot.memory['SpiceBot_CommandsQuery']['commands']['nickname'].keys())
|
|
# botnick_coms.extend(bot.memory['SpiceBot_CommandsQuery']['nickrules'])
|
|
|
|
if triggercommand.lower().startswith(str(bot.nick).lower()):
|
|
searchlist = botnick_coms
|
|
searchitem = spicemanip.main([triggercommand].extend(triggerargs), 0)
|
|
else:
|
|
searchlist = list(commands_list.keys())
|
|
searchitem = triggercommand
|
|
|
|
if searchitem.endswith("+"):
|
|
|
|
searchitem = searchitem[:-1]
|
|
if not searchitem or not len(searchitem):
|
|
return
|
|
|
|
if searchitem.lower() not in searchlist:
|
|
dispmsg = ["Cannot find any alias commands: No valid commands match " + str(searchitem) + "."]
|
|
closestmatches = similar_list(bot, searchitem, searchlist, 10, 'reverse')
|
|
if len(closestmatches):
|
|
dispmsg.append("The following commands match " + str(searchitem) + ": " + spicemanip.main(closestmatches, 'andlist') + ".")
|
|
bot.notice(dispmsg, trigger.nick)
|
|
return
|
|
|
|
realcom = searchitem
|
|
if "aliasfor" in commands_list[searchitem].keys():
|
|
realcom = commands_list[searchitem]["aliasfor"]
|
|
validcomlist = commands_list[realcom]["validcoms"]
|
|
bot.notice("The following commands match " + str(searchitem) + ": " + spicemanip.main(validcomlist, 'andlist') + ".", trigger.nick)
|
|
return
|
|
|
|
if searchitem.endswith("?"):
|
|
|
|
searchitem = searchitem[:-1]
|
|
if not searchitem or not len(searchitem):
|
|
return
|
|
|
|
closestmatches = similar_list(bot, searchitem, searchlist, 10, 'reverse')
|
|
if not len(closestmatches):
|
|
bot.notice("Cannot find any similar commands for " + str(searchitem) + ".", trigger.nick)
|
|
else:
|
|
bot.notice("The following commands may match " + str(searchitem) + ": " + spicemanip.main(closestmatches, 'andlist') + ".", trigger.nick)
|
|
return
|
|
|
|
commandlist = []
|
|
for command in searchlist:
|
|
if command.lower().startswith(str(searchitem).lower()):
|
|
commandlist.append(command)
|
|
|
|
if not len(commandlist):
|
|
bot.notice("No commands start with " + str(searchitem) + ".", trigger.nick)
|
|
else:
|
|
bot.notice("The following commands start with " + str(searchitem) + ": " + spicemanip.main(commandlist, 'andlist') + ".", trigger.nick)
|