support @utils.export on functions, to export those functions

This commit is contained in:
jesopo 2020-02-19 17:22:37 +00:00
parent b277463fee
commit 70db97f64e
3 changed files with 17 additions and 17 deletions

View file

@ -39,10 +39,6 @@ class CoinParseException(Exception):
pass
class Module(ModuleManager.BaseModule):
def on_load(self):
self.exports.add("command-spec.coins", self._coins_spec)
self.exports.add("command-spec.coinsmany", self._coins_many_spec)
def _coin_spec_parse(self, word):
try:
out = decimal.Decimal(word)
@ -54,11 +50,14 @@ class Module(ModuleManager.BaseModule):
else:
raise utils.parse.SpecTypeError(
"Please provide a positive coin amount")
@utils.export("command-spec.coins")
def _coins_spec(self, server, channel, user, args):
if args:
return self._coin_spec_parse(args[0]), 1
else:
raise utils.parse.SpecTypeError("No coin amount provided")
@utils.export("command-spec.coinsmany")
def _coins_many_spec(self, server, channel, user, args):
out = []
for arg in args:

View file

@ -256,7 +256,13 @@ class ModuleManager(object):
module_title = (getattr(module_object, "_name", None) or
definition.name.title())
# @utils.hook() magic
# per-module @export magic
if utils.decorators.has_magic(module_object):
magic = utils.decorators.get_magic(module_object)
for key, value in magic.get_exports():
context_exports.add(key, value)
# per-function @hook/@export magic
for attribute_name in dir(module_object):
attribute = getattr(module_object, attribute_name)
if (inspect.ismethod(attribute) and
@ -265,12 +271,8 @@ class ModuleManager(object):
for hook, kwargs in magic.get_hooks():
context_events.on(hook)._hook(attribute, kwargs=kwargs)
# @utils.export() magic
if utils.decorators.has_magic(module_object):
magic = utils.decorators.get_magic(module_object)
for key, value in magic.get_exports():
context_exports.add(key, value)
context_exports.add(key, attribute)
branch, commit = utils.git_commit(bot.directory)

View file

@ -37,12 +37,11 @@ def hook(event: str, **kwargs):
magic.add_hook(event, kwargs)
return func
return _hook_func
def export(setting: str, value: typing.Any):
def _export_func(module):
magic = get_magic(module)
magic.add_export(setting, value)
return module
return _export_func
def export(setting: str, value: typing.Any=None):
def _export(obj: typing.Any):
get_magic(obj).add_export(setting, value)
return obj
return _export
def _kwarg(key: str, value: typing.Any, func: typing.Any):
magic = get_magic(func)