.. _fbf.plugs.core.core: core ~~~~ .. automodule:: fbf.plugs.core.core :show-inheritance: :members: :undoc-members: CODE ---- :: # fbf/plugs/core/core.py # # """ core bot commands. """ .. _fbf.plugs.core.core_fbf_imports: fbf imports -------------- :: from fbf.lib.aliases import setalias from fbf.lib.config import getmainconfig from fbf.utils.statdict import StatDict from fbf.utils.log import setloglevel, getloglevel from fbf.utils.timeutils import elapsedstring from fbf.utils.exception import handle_exception from fbf.lib.commands import cmnds from fbf.lib.examples import examples from fbf.lib.plugins import plugs from fbf.lib.boot import plugin_packages, getpluginlist, boot, getcmndtable, whatcommands from fbf.lib.persist import Persist from fbf.lib.reboot import reboot, reboot_stateful from fbf.lib.eventhandler import mainhandler from fbf.lib.fleet import getfleet from fbf.lib.partyline import partyline from fbf.lib.exit import globalshutdown from fbf.lib.runner import defaultrunner, cmndrunner, longrunner from fbf.lib.errors import NoSuchPlugin from fbf.utils.statdict import StatDict from fbf.utils.dol import Dol .. _fbf.plugs.core.core_basic_imports: basic imports ---------------- :: import time import threading import sys import re import os import copy import cgi import logging import functools .. _fbf.plugs.core.core_defines_: defines ---------- :: cpy = copy.deepcopy .. _fbf.plugs.core.core_reboot_command: reboot command ----------------- :: def handle_reboot(bot, ievent): """ no arguments - reboot the bot. """ ievent.reply("rebooting") #time.sleep(3) if ievent.rest == "cold": stateful = False else: stateful = True if stateful: if bot.type == "tornado": callback = functools.partial(reboot_stateful, bot, ievent, getfleet(), partyline) bot.server.io_loop.add_callback(callback) else: mainhandler.put(0, reboot_stateful, bot, ievent, getfleet(), partyline) else: getfleet().exit() mainhandler.put(0, reboot) cmnds.add("reboot", handle_reboot, "OPER") examples.add("reboot", "reboot the bot.", "reboot") .. _fbf.plugs.core.core_ping_command: ping command --------------- :: def handle_ping(bot, event): event.reply("pong") cmnds.add("ping", handle_ping, ["OPER", "USER"]) examples.add("ping", "ping/pong", "ping") .. _fbf.plugs.core.core_quit_command: quit command --------------- :: def handle_quit(bot, ievent): """ no arguments - disconnect from the server. """ ievent.reply("quiting") bot.exit() cmnds.add("quit", handle_quit, "OPER") examples.add("quit", "quit the bot.", "quit") .. _fbf.plugs.core.core_disabled_command: disabled command ------------------- :: def handle_disabled(bot, ievent): """ no arguments - show what plugins are disabled. """ ievent.reply("disabled plugins: ", bot.plugs.disabled()) cmnds.add('disabled', handle_disabled, ['OPER', 'USER', 'GUEST']) examples.add('disabled', 'show what plugins are disabled', 'disabled') .. _fbf.plugs.core.core_encoding_command: encoding command ------------------- :: def handle_encoding(bot, ievent): """ no arguments - show default encoding. """ ievent.reply('default encoding is %s' % bot.encoding or sys.getdefaultencoding()) cmnds.add('encoding', handle_encoding, ['USER', 'GUEST']) examples.add('encoding', 'show default encoding', 'encoding') .. _fbf.plugs.core.core_uptime_command: uptime command ----------------- :: def handle_uptime(bot, ievent): """ no arguments - show uptime. """ ievent.reply("uptime is %s" % elapsedstring(time.time()-bot.starttime)) cmnds.add('uptime', handle_uptime, ['USER', 'GUEST']) examples.add('uptime', 'show uptime of the bot', 'uptime') .. _fbf.plugs.core.core_list_command: list command --------------- :: def handle_available(bot, ievent): """ no arguments - show available plugins .. to enable use !plug-enable. """ if ievent.rest: ievent.reply("%s plugin has the following commands: " % ievent.rest, whatcommands(ievent.rest)) else: ievent.reply("available plugins: ", getpluginlist(), raw=True) ; return cmnds.add('list', handle_available, ['USER', 'GUEST']) examples.add('list', 'list available plugins', 'list') .. _fbf.plugs.core.core_commands_command: commands command ------------------- :: def handle_commands(bot, ievent): """ arguments: [] - show commands of plugin. """ try: plugin = ievent.args[0].lower() except IndexError: plugin = "" result = [] cmnds = getcmndtable() for cmnd, plugname in cmnds.items(): if plugname: if not plugin or plugin in plugname: result.append(cmnd) if result: result.sort() if not plugin: plugin = "FBFBOT" ievent.reply('%s has the following commands: ' % plugin, result) else: ievent.reply('no commands found for plugin %s' % plugin) cmnds.add('commands', handle_commands, ['USER', 'GUEST']) examples.add('commands', 'show commands of ', '1) commands core') .. _fbf.plugs.core.core_perm_command: perm command --------------- :: def handle_perm(bot, ievent): """ arguments: - get permission of command. """ try:cmnd = ievent.args[0] except IndexError: ievent.missing("") return try: perms = cmnds.perms(cmnd) except KeyError: ievent.reply("no %sw command registered") return if perms: ievent.reply("%s command needs %s permission" % (cmnd, perms)) else: ievent.reply("can't find perm for %s" % cmnd) cmnds.add('perm', handle_perm, ['USER', 'GUEST']) examples.add('perm', 'show permission of command', 'perm quit') .. _fbf.plugs.core.core_version_command: version command ------------------ :: def handle_version(bot, ievent): """ no arguments - show bot's version. """ from fbf.version import getversion version = getversion(bot.type.upper()) cfg = getmainconfig() if cfg.dbenable: version += " " + cfg.dbtype.upper() if ievent.rest and ievent.rest == "repo": try: from mercurial import context, hg, node, repo, ui repository = hg.repository(ui.ui(), '.') ctx = context.changectx(repository) tip = str(ctx.rev()) except: tip = None if tip: version2 = version + " HG " + tip else: version2 = version ievent.reply(version2) cmnds.add('version', handle_version, ['USER', 'GUEST']) examples.add('version', 'show version of the bot', 'version') setalias('v', "version") .. _fbf.plugs.core.core_whereis_command: whereis command ------------------ :: def handle_whereis(bot, ievent): """ arguments: - locate a command. """ try: cmnd = ievent.args[0] except IndexError: ievent.missing('') return cmnds.reloadcheck(bot, ievent, cmnd) plugin = cmnds.whereis(cmnd) if plugin: ievent.reply("%s command is in: %s" % (cmnd, plugin)) else: ievent.reply("can't find " + cmnd) cmnds.add('whereis', handle_whereis, ['USER', 'GUEST']) examples.add('whereis', 'whereis .. show in which plugins is', 'whereis test') .. _fbf.plugs.core.core_help-plug_command: help-plug command -------------------- :: def handle_helpplug(bot, ievent): """ arguments: - how help on plugin/command or show basic help msg. """ try: what = ievent.args[0] except (IndexError, TypeError): ievent.reply("available plugins: ", getpluginlist()) ievent.reply("see !help to get help on a plugin.") return cmnds.reloadcheck(bot, ievent, what) plugin = None modname = "" perms = [] for package in plugin_packages: try: modname = "%s.%s" % (package, what) try: plugin = plugs.load_mod(modname) if plugin: break except NoSuchPlugin: continue except(KeyError, ImportError): pass if not plugin: ievent.reply("no %s plugin loaded" % what) return try: phelp = plugin.__doc__ except (KeyError, AttributeError): ievent.reply('no description of %s plugin available' % what) return cmndresult = [] if phelp: counter = 1 for i, j in cmnds.items(): if what == j.plugname: try: descr = j.func.__doc__ if not descr: descr = "no description provided" try: cmndresult.append(" !%s - %s - perms: %s" % (i, descr, j.perms)) except KeyError: pass except AttributeError: pass counter += 1 if cmndresult and phelp: res = [] for r in cmndresult: if bot.type in ['web', ]: res.append("%s
" % r) elif bot.type in ['irc', ]: res.append(r.strip()) else: res.append(r) res.sort() what = what.upper() ievent.reply('help on plugin %s:
%s' % (what, phelp)) ievent.reply("commands: ", res, dot="count") else: if perms: ievent.reply('no commands available for permissions: %s' % ", ".join(perms)) else: ievent.reply("can't find help on %s" % what) cmnds.add('help-plug', handle_helpplug, ['USER', 'GUEST'], how="msg") examples.add('help-plug', 'get help on ', '1) help-plug 2) help-plug misc') .. _fbf.plugs.core.core_help-site_command: help-site command -------------------- :: def handle_helpsite(bot, event): """ arguments: - help commands that gives a url to the docs. """ if event.rest: target = cmnds.whereis(event.rest) target = target or event.rest where = bot.plugs.getmodule(target) if where: theplace = os.sep.join(where.split(".")[-2:]) event.reply("help for %s is at http:///docs/fbfbot/plugins/%s.html" % (event.rest.upper(), theplace)) else: event.reply("can't find a help url for %s" % event.rest) else: event.reply("documentation for fbfbot can be found at http:// or http://fbfbot.appspot.com/docs") event.reply('see !list for loaded plugins and "!help plugin" for a url to the plugin docs.') cmndhelp = cmnds.gethelp(event.rest) if cmndhelp: event.reply("docstring: ", cmndhelp.split("\n")) cmnds.add("help-site", handle_helpsite, ["OPER", "USER", "GUEST"]) examples.add("help-site", "show url pointing to the docs", "1) help 2) help rss") .. _fbf.plugs.core.core_help_command: help command --------------- :: def handle_help(bot, event): """ arguments: [] - show help. """ if not event.args: event.reply("documentation for fbfbot can be found at http:///docs/fbfbot") event.reply('see !list for loaded plugins and "!help plugin" for help on the plugin.') return t = event.args[0] logging.warn("help target is %s" % t) cmnds.reloadcheck(bot, event, t) where = cmnds.whereis(t) cmndhelp = cmnds.gethelp(t) if not cmndhelp: handle_helpplug(bot, event) ; return try: ex = examples[t].example except KeyError: ex = None p = cmnds.perms(t) res = "permissions: %s - location: %s - examples: %s" % (", ".join(p), where, ex) event.reply('help on %s command: %s' % (event.rest, cmndhelp), res.split(" - "), dot="
") cmnds.add("help", handle_help, ["OPER", "USER", "GUEST"]) examples.add("help", "show help of a command", "help rss-list") .. _fbf.plugs.core.core_apro_command: apro command --------------- :: def handle_apro(bot, ievent): """ arguments: - apropos (search) for commands. """ try: what = ievent.args[0] except IndexError: ievent.missing('') return result = cmnds.apropos(what) result.sort() if result: ievent.reply("commands matching %s: " % what, result) else: ievent.reply('no matching commands found for %s' % what) cmnds.add('apro', handle_apro, ['USER', 'GUEST']) examples.add('apro', 'apro .. search for commands that contain ', 'apro com') .. _fbf.plugs.core.core_whatcommands_command: whatcommands command ----------------------- :: def handle_whatcommands(bot, ievent): """ arguments: ') return result = cmnds res = [] for cmnd in list(result.values()): if cmnd and cmnd.perms and ievent.rest in cmnd.perms: res.append(cmnd.cmnd) res.sort() if not res: ievent.reply('no commands known for permission %s' % ievent.rest) else: ievent.reply('commands known for permission %s: ' % ievent.rest, res) cmnds.add('whatcommands', handle_whatcommands, ['USER', 'GUEST']) examples.add('whatcommands', 'show commands with permission ', 'whatcommands USER') .. _fbf.plugs.core.core_versions_command: versions command ------------------- :: def handle_versions(bot, ievent): """ no arguments - show versions of all loaded modules (if available). """ versions = {} allowed = ["fbf.", ] for mod in copy.copy(sys.modules): for allow in allowed: if mod.startswith(allow): try: versions[mod] = sys.modules[mod].__version__ except AttributeError as ex: pass try: versions['python'] = sys.version except AttributeError as ex: pass ievent.reply("versions ==> %s" % str(versions)) cmnds.add('versions', handle_versions, ['USER', 'GUEST']) examples.add('versions', 'show versions of all loaded modules', 'versions') .. _fbf.plugs.core.core_loglevel_command: loglevel command ------------------- :: def handle_loglevel(bot, event): """ arguments: - change loglevel of the bot. loglevel is on of debug, info, warn or error. """ if not event.rest: event.reply("loglevel is %s" % getloglevel()) ; return from fbf.lib.config import getmainconfig mainconfig = getmainconfig() mainconfig.loglevel = event.rest mainconfig.save() #mainhandler.put(4, setloglevel, event.rest) setloglevel(event.rest) event.done() cmnds.add("loglevel", handle_loglevel, "OPER") examples.add("loglevel", "set loglevel ot on of debug, info, warning or error", "loglevel warn") .. _fbf.plugs.core.core_threads_command: threads command ------------------ :: def handle_threads(bot, ievent): """ no arguments - show running threads. """ try: import threading except ImportError: ievent.reply("threading is not enabled.") return stats = StatDict() names = Dol() threadlist = threading.enumerate() for thread in threadlist: name = thread.getName() try: type = thread.type except AttributeError: type = name stats.upitem(type) try: names.add(type, thread.nowrunning) except AttributeError: pass if ievent.rest: result = [] for item, n in names.items(): if not n: continue res = "%s: " % item for name in n: res += "%s, " % name result.append(res[:-2]) result.sort() ievent.reply("nowrunning: ", result, dot="
") result = [] for item in stats.top(): result.append("%s = %s" % (item[0], item[1])) result.sort() ievent.reply(", ".join(result)) cmnds.add('threads', handle_threads, ['USER', 'GUEST']) examples.add('threads', 'show running threads', 'threads') .. _fbf.plugs.core.core_loaded_command: loaded command ----------------- :: def handle_loaded(bot, event): """ no arguments - show plugins in cache. """ res = [] for plug in list(plugs.keys()): res.append(plug.split(".")[-1]) event.reply("loaded plugins (cache): ", res) cmnds.add('loaded', handle_loaded, ['USER', 'GUEST']) examples.add('loaded', 'show list of loaded plugins', 'loaded') .. _fbf.plugs.core.core_statusline_command: statusline command --------------------- :: def handle_statusline(bot, event): """ no arguments - show a status line. """ event.reply("controlchars: %s - perms: %s" % (event.chan.data.cc, ", ".join(event.user.data.perms))) cmnds.add('statusline', handle_statusline, ['USER', 'GUEST']) examples.add('statusline', 'show status line', 'statusline') .. _fbf.plugs.core.core_topper_command: topper command ----------------- :: def handle_topper(bot, event): """ no arguments - show a 'topper' startus line. """ event.reply("forwards: %s - watched: %s - feeds: %s" % (", ".join(event.chan.data.forwards) or "none", ", ".join(event.chan.data.watched) or "none", ", ".join([str(x) for x in event.chan.data.feeds]) or "none")) cmnds.add('topper', handle_topper, ['USER', 'GUEST']) examples.add('topper', 'show topper line', 'topper') .. _fbf.plugs.core.core_running_command: running command ------------------ :: def handle_running(bot, event): """ no arguments - show running tasks. """ from fbf.lib.runner import allrunners res = [] for runner in allrunners: res.append("%s: %s" % (runner.name, str(runner.running()))) event.reply("runners: ", res) cmnds.add('running', handle_running, ['USER', 'GUEST']) examples.add('running', "show running tasks", "running") .. _fbf.plugs.core.core_descriptions_command: descriptions command ----------------------- :: def handle_descriptions(bot, event): """ no arguments - show descriptions of all plugins. """ bot.plugs.loadall() result = [] target = list(bot.plugs.keys()) target.sort() for modname in target: plug = bot.plugs.get(modname) if plug.__doc__: txt = plug.__doc__.replace("\n", "
") else: txt = "no docstring available" result.append("* %s plugin (%s) - %s" % (modname.split(".")[-1], modname, txt)) event.reply("descriptions:
", result, dot="

") cmnds.add('descriptions', handle_descriptions, ['USER', 'GUEST']) examples.add('descriptions', "show descriptions of all plugins", "descriptions") .. _fbf.plugs.core.core_stats_command: stats command ---------------- :: def handle_stats(bot, event): totalstats = StatDict() counter = 0 for modname in sys.modules: if modname.startswith("fbf"): plugname = modname.split(".")[-1] if event.args and plugname not in event.args: continue try: modstats = getattr(sys.modules[modname], "stats") except AttributeError: continue totalstats += modstats counter += 1 event.reply("stats results from %s modules: " % counter, totalstats) cmnds.add("stats", handle_stats, ["OPER", "USER"]) examples.add("stats", "show stats for all FBFBOT modules/plugins or specify a module/plugin", "1) stats 2) stats rss")