a3x/pizzabot
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Use new command handling system

Browse source
This commit is contained in:
Tulir Asokan 2018-12-18 00:54:33 +02:00
parent 3084193f86
commit 4c7bc0bf7d
1 changed files with 7 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
echo.py
View file

@ -1,34 +1,10 @@
from time import time from time import time
from maubot import Plugin, CommandSpec, Command, Argument, MessageEvent from maubot import Plugin, MessageEvent
from maubot.handlers import command
COMMAND_PING = "ping"
ARG_ECHO = "$echo"
COMMAND_ECHO = f"echo {ARG_ECHO}"
class EchoBot(Plugin): class EchoBot(Plugin):
async def start(self) -> None:
self.set_command_spec(CommandSpec(
commands=[Command(
syntax=COMMAND_PING,
description="Ping the bot",
), Command(
syntax=COMMAND_ECHO,
description="Echo something",
arguments={
ARG_ECHO: Argument(matches=".+", required=True,
description="The content to echo"),
},
)],
))
self.client.add_command_handler(COMMAND_PING, self.ping_handler)
self.client.add_command_handler(COMMAND_ECHO, self.echo_handler)
async def stop(self) -> None:
self.client.remove_command_handler(COMMAND_PING, self.ping_handler)
self.client.remove_command_handler(COMMAND_ECHO, self.echo_handler)
@staticmethod @staticmethod
def time_since(ms: int) -> str: def time_since(ms: int) -> str:
diff = int(time() * 1000) - ms diff = int(time() * 1000) - ms
@ -45,9 +21,11 @@ class EchoBot(Plugin):
days, hours = divmod(hours, 24) days, hours = divmod(hours, 24)
return f"{days} days, {hours} hours, {minutes} minutes and {seconds} seconds" return f"{days} days, {hours} hours, {minutes} minutes and {seconds} seconds"
@command.new("ping", help="Ping")
async def ping_handler(self, evt: MessageEvent) -> None: async def ping_handler(self, evt: MessageEvent) -> None:
await evt.reply(f"Pong! (ping took {self.time_since(evt.timestamp)} to arrive)") await evt.reply(f"Pong! (ping took {self.time_since(evt.timestamp)} to arrive)")
@staticmethod @command.new("echo", help="Repeat a message")
async def echo_handler(evt: MessageEvent) -> None: @command.argument("message", required=True)
await evt.respond(evt.content.command.arguments[ARG_ECHO]) async def echo_handler(self, evt: MessageEvent, message: str) -> None:
await evt.respond(message)