Garmelon/euphoria-infobot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add !hosts command

Browse source
This commit is contained in:
Joscha 2019-04-13 22:34:26 +00:00
parent e04548b0c5
commit 948f509999
1 changed files with 24 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

41
infobot.py
View file

@ -1,6 +1,7 @@
import logging import logging
import yaboli import yaboli
from yaboli.util import *
class InfoBot(yaboli.Bot): class InfoBot(yaboli.Bot):
@ -19,7 +20,7 @@ class InfoBot(yaboli.Bot):
"!recount {atmention} - Recount people in the room", "!recount {atmention} - Recount people in the room",
#"!detail {atmention} - Detailed list of clients in this room", #"!detail {atmention} - Detailed list of clients in this room",
#"!detail {atmention} @person - Detailed info regarding @person", #"!detail {atmention} @person - Detailed info regarding @person",
#"!hosts {atmention} [--mention] - Lists all hosts currently in this room", "!hosts [--ping] - Lists all hosts currently in this room",
"", "",
"Created by @Garmy using https://github.com/Garmelon/yaboli.", "Created by @Garmy using https://github.com/Garmelon/yaboli.",
"For additional info, try \"!help {atmention} <topic>\". Topics:", "For additional info, try \"!help {atmention} <topic>\". Topics:",
@ -48,6 +49,7 @@ class InfoBot(yaboli.Bot):
] ]
HELP_CHANGELOG = [ HELP_CHANGELOG = [
"(2019-04-13) re-add !hosts command",
"(2019-04-12) update to yaboli rewrite 5", "(2019-04-12) update to yaboli rewrite 5",
] ]
@ -63,6 +65,7 @@ class InfoBot(yaboli.Bot):
self.register_general("help", self.cmd_help_general, args=False) self.register_general("help", self.cmd_help_general, args=False)
self.register_specific("help", self.cmd_help_specific) self.register_specific("help", self.cmd_help_specific)
self.register_specific("recount", self.cmd_recount, args=False) self.register_specific("recount", self.cmd_recount, args=False)
self.register_general("hosts", self.cmd_hosts)
async def cmd_help_specific(self, room, message, args): async def cmd_help_specific(self, room, message, args):
if not args.has_args(): if not args.has_args():
@ -84,6 +87,26 @@ class InfoBot(yaboli.Bot):
await self.update_nick(room) await self.update_nick(room)
await message.reply("Recalibrated.") await message.reply("Recalibrated.")
async def cmd_hosts(self, room, message, args):
fancy = args.fancy()
ping = "mention" in fancy.optional or "ping" in fancy.optional
hosts = sorted(set(user.nick for user in room.users if user.is_manager))
lines = []
for host in hosts:
if ping:
lines.append(atmention(host))
else:
lines.append(host)
if lines:
lines = ["Hosts that are currently in this room:"] + lines
else:
lines = ["No hosts currently in this room."]
await message.reply("\n".join(lines))
# Updating the nick # Updating the nick
def format_nick(self, users): def format_nick(self, users):
@ -169,22 +192,6 @@ class InfoBot(yaboli.Bot):
# is_manager = "yes" if s.is_manager else "no" # is_manager = "yes" if s.is_manager else "no"
# return f"UID: {s.uid}\t| SID: {s.sid}\t| staff: {is_staff}\t| host: {is_manager}\t| nick: {s.nick!r}" # return f"UID: {s.uid}\t| SID: {s.sid}\t| staff: {is_staff}\t| host: {is_manager}\t| nick: {s.nick!r}"
# #
# @yaboli.command("hosts")
# async def command_hosts(self, room, message, argstr):
# flags, args, kwargs = self.parse_flags(self.parse_args(argstr))
# sessions = room.listing.get()
# sessions = sorted(set(s.nick for s in sessions if s.is_manager))
#
# if "ping" in kwargs:
# sessions = [mention(s) for s in sessions]
# else:
# sessions = [s for s in sessions]
#
# if sessions:
# text = "Hosts that are currently in this room:\n" + "\n".join(sessions)
# else:
# text = "No hosts currently in this room."
# await room.send(text, message.mid)
if __name__ == "__main__": if __name__ == "__main__":
yaboli.enable_logging(level=logging.DEBUG) yaboli.enable_logging(level=logging.DEBUG)