1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-27 13:34:28 +00:00
royalnet/royalpack/commands/ping.py
2020-07-01 00:11:14 +02:00

55 lines
1.7 KiB
Python

import asyncio
from typing import *
import royalnet
import royalnet.commands as rc
class PingCommand(rc.Command):
name: str = "ping"
description: str = "Display the status of the Herald network."
syntax: str = ""
async def run(self, args: rc.CommandArgs, data: rc.CommandData) -> None:
await data.reply("📶 Ping...")
telegram_c = self.interface.call_herald_event("telegram", "pong")
discord_c = self.interface.call_herald_event("discord", "pong")
constellation_c = self.interface.call_herald_event("constellation", "pong")
telegram_t = self.loop.create_task(telegram_c)
discord_t = self.loop.create_task(discord_c)
constellation_t = self.loop.create_task(constellation_c)
await asyncio.sleep(10)
try:
telegram_r = telegram_t.result()
except (asyncio.CancelledError, asyncio.InvalidStateError):
telegram_r = None
try:
discord_r = discord_t.result()
except (asyncio.CancelledError, asyncio.InvalidStateError):
discord_r = None
try:
constellation_r = constellation_t.result()
except (asyncio.CancelledError, asyncio.InvalidStateError):
constellation_r = None
lines = ["📶 [b]Pong![/b]", ""]
if telegram_r:
lines.append("🔵 Telegram")
else:
lines.append("🔴 Telegram")
if discord_r:
lines.append("🔵 Discord")
else:
lines.append("🔴 Discord")
if constellation_r:
lines.append("🔵 Constellation")
else:
lines.append("🔴 Constellation")
await data.reply("\n".join(lines))