mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 03:24:20 +00:00
🗑 Remove command module, pending refactor
This commit is contained in:
parent
a9cc5e7761
commit
1b8c721cc2
1 changed files with 0 additions and 121 deletions
|
@ -1,121 +0,0 @@
|
|||
"""
|
||||
Commands are used to quickly create single-message conversations
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
import royalnet.royaltyping as t
|
||||
|
||||
import logging
|
||||
import re
|
||||
|
||||
from . import teleporter
|
||||
from . import bullet
|
||||
from . import sentry
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class Command:
|
||||
"""
|
||||
A class that allows creating commands which can be called from the chat by entering a certain :attr:`.pattern` of
|
||||
characters:
|
||||
|
||||
.. code-block:: text
|
||||
|
||||
/echo Hello!
|
||||
# Hello!
|
||||
|
||||
.. todo:: Improve the docstring of :class:`.Command`.
|
||||
"""
|
||||
|
||||
def __init__(self,
|
||||
prefix: str,
|
||||
name: str,
|
||||
syntax: str,
|
||||
conversation: t.Conversation,
|
||||
*,
|
||||
pattern: str = r"^{prefix}{name} ?{syntax}",
|
||||
doc: str = ""):
|
||||
self.prefix: str = prefix
|
||||
"""
|
||||
The prefix used in the command (usually ``/`` or ``!``).
|
||||
"""
|
||||
|
||||
self.name: str = name
|
||||
"""
|
||||
The name of the command, usually all lowercase.
|
||||
"""
|
||||
|
||||
self.syntax: str = syntax
|
||||
"""
|
||||
A regex describing the syntax of the command, using named capture groups ``(?P<name>...)`` to capture arguments
|
||||
that should be passed to the function.
|
||||
"""
|
||||
|
||||
self.pattern: re.Pattern = re.compile(pattern.format(prefix=prefix, name=name, syntax=syntax))
|
||||
"""
|
||||
The compiled regex pattern.
|
||||
|
||||
By default, it :meth:`str.format` the passed string with the ``prefix``, ``name`` and ``syntax`` keyword
|
||||
arguments, but this behaviour can be changed by passing a different ``pattern`` to :meth:`__init__`.
|
||||
"""
|
||||
|
||||
self.doc: str = doc
|
||||
"""
|
||||
A string explaining how this command should be used. Useful for command lists or help commands.
|
||||
"""
|
||||
|
||||
self.conversation: t.Conversation = conversation
|
||||
"""
|
||||
The conversation to run when this command is called.
|
||||
"""
|
||||
|
||||
@classmethod
|
||||
def new(cls,
|
||||
prefix: str,
|
||||
name: str,
|
||||
syntax: str = "",
|
||||
*,
|
||||
pattern: str = r"^{prefix}{name}(?: {syntax})?",
|
||||
doc: str = ""):
|
||||
"""
|
||||
Create a new :class:`.Command` using the decorated function as :attr:`.conversation`.
|
||||
:return: The created :class:`.Command`.
|
||||
"""
|
||||
def decorator(f):
|
||||
log.debug(f"Making function {f} a teleporter...")
|
||||
teleporter_f = teleporter.teleport(is_async=True, validate_output=False)(f)
|
||||
|
||||
log.debug(f"Creating command: {prefix} {name} {syntax} - {doc}")
|
||||
return cls(prefix=prefix, name=name, syntax=syntax, conversation=teleporter_f, pattern=pattern, doc=doc)
|
||||
return decorator
|
||||
|
||||
async def run(self, _sentry: sentry.Sentry, **original_kwargs) -> t.Optional[t.Conversation]:
|
||||
"""
|
||||
A conversation which runs the command.
|
||||
"""
|
||||
log.debug(f"Getting a bullet from the queue...")
|
||||
b: bullet.Bullet = await _sentry
|
||||
if not isinstance(b, bullet.Message):
|
||||
log.debug(f"Bullet is not a message, returning...")
|
||||
|
||||
msg: bullet.Message = b
|
||||
log.debug(f"Getting text of {msg}...")
|
||||
text = await msg.text()
|
||||
|
||||
log.debug(f"Matching text {text} to {self.pattern}...")
|
||||
match: re.Match = self.pattern.search(text)
|
||||
if match is None:
|
||||
log.debug(f"Pattern didn't match, returning...")
|
||||
return
|
||||
|
||||
log.debug(f"Pattern matched, getting named groups...")
|
||||
match_kwargs: t.Dict[str, str] = match.groupdict()
|
||||
|
||||
log.debug(f"Running teleported function with match kwargs: {match_kwargs}")
|
||||
return await self.conversation(_sentry=_sentry, _msg=msg, **original_kwargs, **match_kwargs)
|
||||
|
||||
|
||||
__all__ = (
|
||||
"Command",
|
||||
)
|
Loading…
Reference in a new issue