mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
199 lines
7.8 KiB
Python
199 lines
7.8 KiB
Python
# coding=utf-8
|
|
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
|
|
import sopel
|
|
import sopel.module
|
|
|
|
import os
|
|
|
|
from sopel_modules.SpiceBot_Events.System import bot_events_startup_register, bot_events_recieved, bot_events_trigger
|
|
from sopel_modules.SpiceBot_SBTools import bot_logging
|
|
|
|
import spicemanip
|
|
|
|
|
|
def setup(bot):
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "Starting setup procedure")
|
|
bot_events_startup_register(bot, ['2002'])
|
|
|
|
if 'SpiceBot_CommandsQuery' not in bot.memory:
|
|
bot.memory['SpiceBot_CommandsQuery'] = {"counts": {}, "commands": {}}
|
|
|
|
for comtype in ['module', 'nickname', 'rule']:
|
|
if comtype not in bot.memory['SpiceBot_CommandsQuery']['counts'].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['counts'][comtype] = 0
|
|
if comtype not in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][comtype] = dict()
|
|
|
|
filepathlisting = []
|
|
|
|
# main Modules directory
|
|
main_dir = os.path.dirname(os.path.abspath(sopel.__file__))
|
|
modules_dir = os.path.join(main_dir, 'modules')
|
|
filepathlisting.append(modules_dir)
|
|
|
|
# Home Directory
|
|
home_modules_dir = os.path.join(bot.config.homedir, 'modules')
|
|
if os.path.isdir(home_modules_dir):
|
|
filepathlisting.append(home_modules_dir)
|
|
|
|
# pypi installed
|
|
try:
|
|
import sopel_modules
|
|
for plugin_dir in set(sopel_modules.__path__):
|
|
for pathname in os.listdir(plugin_dir):
|
|
pypi_modules_dir = os.path.join(plugin_dir, pathname)
|
|
filepathlisting.append(pypi_modules_dir)
|
|
except Exception as e:
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "sopel_modules not loaded :" + str(e))
|
|
|
|
# Extra directories
|
|
filepathlist = []
|
|
for directory in bot.config.core.extra:
|
|
filepathlisting.append(directory)
|
|
|
|
for directory in filepathlisting:
|
|
for pathname in os.listdir(directory):
|
|
path = os.path.join(directory, pathname)
|
|
if (os.path.isfile(path) and path.endswith('.py') and not path.startswith('_')):
|
|
filepathlist.append(str(path))
|
|
|
|
# CoreTasks
|
|
ct_path = os.path.join(main_dir, 'coretasks.py')
|
|
filepathlist.append(ct_path)
|
|
|
|
for modulefile in filepathlist:
|
|
module_file_lines = []
|
|
module_file = open(modulefile, 'r')
|
|
lines = module_file.readlines()
|
|
for line in lines:
|
|
module_file_lines.append(line)
|
|
module_file.close()
|
|
|
|
dict_from_file = dict()
|
|
filelinelist = []
|
|
|
|
for line in module_file_lines:
|
|
|
|
if str(line).startswith("@"):
|
|
line = str(line)[1:]
|
|
|
|
# Commands
|
|
if str(line).startswith(tuple(["commands", "module.commands", "sopel.module.commands"])):
|
|
comtype = "module"
|
|
line = str(line).split("commands(")[-1]
|
|
line = str("(" + line)
|
|
validcoms = eval(str(line))
|
|
if isinstance(validcoms, tuple):
|
|
validcoms = list(validcoms)
|
|
else:
|
|
validcoms = [validcoms]
|
|
validcomdict = {"comtype": comtype, "validcoms": validcoms}
|
|
filelinelist.append(validcomdict)
|
|
elif str(line).startswith(tuple(["nickname_commands", "module.nickname_commands", "sopel.module.nickname_commands"])):
|
|
comtype = "nickname"
|
|
line = str(line).split("commands(")[-1]
|
|
line = str("(" + line)
|
|
validcoms = eval(str(line))
|
|
if isinstance(validcoms, tuple):
|
|
validcoms = list(validcoms)
|
|
else:
|
|
validcoms = [validcoms]
|
|
nickified = []
|
|
for nickcom in validcoms:
|
|
nickified.append(str(bot.nick) + " " + nickcom)
|
|
validcomdict = {"comtype": comtype, "validcoms": nickified}
|
|
filelinelist.append(validcomdict)
|
|
elif str(line).startswith(tuple(["rule", "module.rule", "sopel.module.rule"])):
|
|
comtype = "rule"
|
|
line = str(line).split("rule(")[-1]
|
|
validcoms = [str("(" + line)]
|
|
validcomdict = {"comtype": comtype, "validcoms": validcoms}
|
|
filelinelist.append(validcomdict)
|
|
|
|
if len(filelinelist):
|
|
bot.memory['SpiceBot_CommandsQuery']['counts'][comtype] += 1
|
|
|
|
for atlinefound in filelinelist:
|
|
|
|
comtype = atlinefound["comtype"]
|
|
validcoms = atlinefound["validcoms"]
|
|
|
|
# default command to filename
|
|
if "validcoms" not in dict_from_file.keys():
|
|
dict_from_file["validcoms"] = validcoms
|
|
|
|
maincom = dict_from_file["validcoms"][0]
|
|
if len(dict_from_file["validcoms"]) > 1:
|
|
comaliases = spicemanip.main(dict_from_file["validcoms"], '2+', 'list')
|
|
else:
|
|
comaliases = []
|
|
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][comtype][maincom] = dict_from_file
|
|
for comalias in comaliases:
|
|
if comalias not in bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][comtype][comalias] = {"aliasfor": maincom}
|
|
|
|
for comtype in ['module', 'nickname', 'rule']:
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "Found " + str(len(bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys())) + " " + comtype + " commands.")
|
|
|
|
bot_events_trigger(bot, 2002, "SpiceBot_CommandsQuery")
|
|
|
|
|
|
def shutdown(bot):
|
|
if "SpiceBot_CommandsQuery" in bot.memory:
|
|
del bot.memory["SpiceBot_CommandsQuery"]
|
|
|
|
|
|
@sopel.module.event('1004')
|
|
@sopel.module.rule('.*')
|
|
def bot_events_complete(bot, trigger):
|
|
bot_events_recieved(bot, trigger.event)
|
|
|
|
for comtype in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
|
|
if comtype not in ['module', 'nickname', 'rule']:
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "Found " + str(len(bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys())) + " " + comtype + " commands.")
|
|
|
|
|
|
def commandsquery_register(bot, command_type, validcoms, aliasfor=None):
|
|
|
|
if 'SpiceBot_CommandsQuery' not in bot.memory:
|
|
bot.memory['SpiceBot_CommandsQuery'] = {"counts": {}, "commands": {}}
|
|
|
|
if not isinstance(validcoms, list):
|
|
validcoms = [validcoms]
|
|
|
|
if command_type not in bot.memory['SpiceBot_CommandsQuery']['counts'].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['counts'][command_type] = 0
|
|
if command_type not in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][command_type] = dict()
|
|
bot.memory['SpiceBot_CommandsQuery']['counts'][command_type] += 1
|
|
|
|
dict_from_file = dict()
|
|
|
|
# default command to filename
|
|
if "validcoms" not in dict_from_file.keys():
|
|
dict_from_file["validcoms"] = validcoms
|
|
|
|
if not aliasfor:
|
|
|
|
maincom = dict_from_file["validcoms"][0]
|
|
if len(dict_from_file["validcoms"]) > 1:
|
|
comaliases = spicemanip.main(dict_from_file["validcoms"], '2+', 'list')
|
|
else:
|
|
comaliases = []
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][command_type][maincom] = dict_from_file
|
|
else:
|
|
comaliases = validcoms
|
|
|
|
for comalias in comaliases:
|
|
if comalias not in bot.memory['SpiceBot_CommandsQuery']['commands'][command_type].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][command_type][comalias] = {"aliasfor": aliasfor}
|
|
|
|
|
|
@sopel.module.event('2002')
|
|
@sopel.module.rule('.*')
|
|
def bot_events_setup(bot, trigger):
|
|
bot_events_recieved(bot, trigger.event)
|