62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import sys
|
|
import yaboli
|
|
from yaboli.utils import *
|
|
|
|
|
|
|
|
class InfoBot(yaboli.Bot):
|
|
"""
|
|
Display information about the clients connected to a room in its nick.
|
|
"""
|
|
|
|
def __init__(self):
|
|
super().__init__("()")
|
|
|
|
self.help_specific = (
|
|
"Displays information about the clients in a room in its nick:\n"
|
|
"(<people>P <bots>B <lurkers>L <bot-lurkers>N)\n\n"
|
|
"Github: https://github.com/Garmelon/infobot (complies with botrulez, including !kill and !restart)\n"
|
|
"Created by @Garmy using yaboli (https://github.com/Garmelon/yaboli)"
|
|
)
|
|
|
|
async def update_nick(self):
|
|
p = len(self.room.listing.get(types=["account", "agent"], lurker=False))
|
|
b = 1 + len(self.room.listing.get(types=["bot"], lurker=False))
|
|
l = len(self.room.listing.get(types=["account", "agent"], lurker=True))
|
|
n = len(self.room.listing.get(types=["bot"], lurker=True))
|
|
|
|
name = []
|
|
if p > 0: name.append(f"{p}P")
|
|
if b > 0: name.append(f"{b}B")
|
|
if l > 0: name.append(f"{l}L")
|
|
if n > 0: name.append(f"{n}N")
|
|
name = "\u0001(" + " ".join(name) + ")"
|
|
|
|
await self.set_nick(name)
|
|
|
|
async def on_join(self, session):
|
|
await self.update_nick()
|
|
|
|
async def on_part(self, session):
|
|
await self.update_nick()
|
|
|
|
async def on_nick(self, session_id, user_id, from_nick, to_nick):
|
|
await self.update_nick()
|
|
|
|
async def on_snapshot(self, user_id, session_id, version, sessions, messages, nick=None,
|
|
pm_with_nick=None, pm_with_user_id=None):
|
|
# Not needed because we're updating the nick anyways.
|
|
#super().on_snapshot(user_id, session_id, version, sessions, messages, nick, pm_with_nick,
|
|
#pm_with_user_id)
|
|
await self.update_nick()
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("USAGE:")
|
|
print(f" {sys.argv[0]} <room>")
|
|
return
|
|
|
|
run_bot(InfoBot, sys.argv[1])
|
|
|
|
if __name__ == "__main__":
|
|
main()
|