122 lines
3.6 KiB
Python
122 lines
3.6 KiB
Python
import sys
|
|
import yaboli
|
|
|
|
class InfoBot(yaboli.Bot):
|
|
"""
|
|
Display information about the clients connected to a room in its nick.
|
|
"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
self.bot_description = ("This bot displays information about the clients connected to the\n"
|
|
"current room in its nick: (<people>P <bots>B <lurkers>L)\n\n"
|
|
"Created by @Garmy using yaboli (https://github.com/Garmelon/yaboli)\n"
|
|
"Github: https://github.com/Garmelon/infobot")
|
|
|
|
self.add_command("info", self.info_command, "Show more detailed info.",
|
|
("!info @bot [ --list[=<who>] | --name=<name> ]\n"
|
|
"--list=<who> : list user names and client/session ids\n"
|
|
"--name=<name> : list info for users with the name specified\n"
|
|
"<who> can be: people, bots, lurkers, all\n"
|
|
"<name> is the name of the person\n\n"
|
|
"Shows different information about the clients connected to the\n"
|
|
"current room."))
|
|
|
|
self.room.add_callback("sessions", self.update_nick)
|
|
|
|
self.update_nick()
|
|
|
|
def update_nick(self):
|
|
"""
|
|
update_nick(self) -> None
|
|
|
|
Change the name to display the correct values.
|
|
"""
|
|
|
|
nick = "\001({}P {}B {}L)".format(
|
|
len(self.room.get_people()),
|
|
len(self.room.get_bots()) + 1, # we are a bot
|
|
len(self.room.get_lurkers())
|
|
)
|
|
|
|
self.room.set_nick(nick)
|
|
|
|
def info_command(self, message, arguments, flags, options):
|
|
"""
|
|
info_command(message, arguments, flags, options) -> None
|
|
|
|
Send a more verbose message.
|
|
"""
|
|
|
|
if "name" in options and "list" in options:
|
|
msg = "Sorry, the --list and --name options can't be used simultaneously."
|
|
|
|
elif ("name" in options and options["name"] is not True) or "list" in options:
|
|
sessions = self.room.get_sessions() + [self.room.session]
|
|
clients = []
|
|
|
|
if "name" in options:
|
|
name = self.room.mentionable(options["name"]).lower()
|
|
|
|
clients = [c for c in sessions
|
|
if self.room.mentionable(c.name).lower() == name]
|
|
|
|
if name[:1] == "@":
|
|
name = name[1:]
|
|
clients.extend([c for c in sessions
|
|
if self.room.mentionable(c.name).lower() == name])
|
|
|
|
elif "list" in options:
|
|
if options["list"] is True or options["list"] == "all":
|
|
clients = self.room.s
|
|
elif options["list"] == "people":
|
|
clients = self.room.get_people()
|
|
elif options["list"] == "bots":
|
|
clients = self.room.get_bots()
|
|
elif options["list"] == "lurkers":
|
|
clients = self.room.get_lurkers()
|
|
|
|
if clients:
|
|
msg = ""
|
|
for client in sorted(clients, key=lambda c: c.name.lower()):
|
|
msg += "id={}, session_id={}, name={}\n".format(
|
|
repr(client.id),
|
|
repr(client.session_id),
|
|
repr(client.name)
|
|
)
|
|
|
|
msg = msg[:-1] # remove trailing newline
|
|
else:
|
|
msg = "No clients"
|
|
|
|
else:
|
|
people = len(self.room.get_people())
|
|
accounts = len(self.room.get_accounts())
|
|
bots = len(self.room.get_bots())
|
|
lurkers = len(self.room.get_lurkers())
|
|
|
|
msg = "people: {} (with accounts: {})\nbots: {}\nlurkers: {}\ntotal: {}"
|
|
msg = msg.format(people, accounts, bots, lurkers, people + bots + lurkers)
|
|
|
|
self.room.send_message(msg, message.id)
|
|
|
|
def main(rooms, imode=False):
|
|
manager = yaboli.BotManager(InfoBot, default_nick="infobot")
|
|
|
|
for room in rooms:
|
|
manager.create(room)
|
|
|
|
#if imode:
|
|
#manager.interactive_mode()
|
|
|
|
if __name__ == "__main__":
|
|
args = sys.argv[1:]
|
|
|
|
if "-i" in args:
|
|
imode = True
|
|
args.remove("-i")
|
|
else:
|
|
imode=False
|
|
|
|
main(args, imode=imode)
|