add logging, @register future deprecation warning
parent
5b6ec72d0b
commit
0bb6eb6b6a
|
@ -1,11 +1,12 @@
|
|||
from enum import Enum
|
||||
from functools import wraps
|
||||
|
||||
from .log import logger
|
||||
|
||||
plugins = {}
|
||||
cmds = {}
|
||||
lstnrs = {}
|
||||
|
||||
|
||||
class OutputType(Enum):
|
||||
Message = 'message'
|
||||
Action = 'action'
|
||||
|
@ -24,6 +25,7 @@ class Output:
|
|||
|
||||
class _BasePlugin:
|
||||
enabled = True
|
||||
logger = logger
|
||||
|
||||
def enable(self):
|
||||
self.enabled = True
|
||||
|
@ -79,41 +81,42 @@ class Command(_BasePlugin):
|
|||
def action(msg):
|
||||
return Output(OutputType.Action, msg)
|
||||
|
||||
|
||||
def message(msg):
|
||||
return Output(OutputType.Message, msg)
|
||||
|
||||
|
||||
def _add_command(command, help_text, func):
|
||||
if command not in cmds:
|
||||
Command(command, help_text=help_text, run=func).add_command()
|
||||
else:
|
||||
cmds[command].update_plugin(help_text=help_text, run=func)
|
||||
|
||||
|
||||
def _ops_plugin(command, ops_msg, func):
|
||||
if command not in cmds:
|
||||
Command(command, ops=True, ops_msg=ops_msg).add_command()
|
||||
else:
|
||||
cmds[command].enable_ops(ops_msg)
|
||||
|
||||
|
||||
def _add_listener(name, func):
|
||||
Listener(name, run=func).add_listener()
|
||||
|
||||
|
||||
def clear_plugins():
|
||||
cmds.clear()
|
||||
lstnrs.clear()
|
||||
|
||||
|
||||
def register(command, help_text='N/A'):
|
||||
def command(command, help_text='N/A'):
|
||||
@wraps(command)
|
||||
def register_for_command(func):
|
||||
_add_command(command, help_text, func)
|
||||
return func
|
||||
return register_for_command
|
||||
|
||||
def register(command, help_text='N/A'):
|
||||
logger.warn('@register decorator has been deprecated in favor of @command. This will cause errors in future versions.')
|
||||
@wraps(command)
|
||||
def register_for_command(func):
|
||||
_add_command(command, help_text, func)
|
||||
return func
|
||||
return register_for_command
|
||||
|
||||
def listener(name):
|
||||
def register_as_listener(func):
|
||||
|
|
Loading…
Reference in New Issue