rollbot3/src/plugins/teamspeak.py

import logging

import requests
from command_system import RollbotResponse, RollbotPlugin
from config import get_secret

TS3_QUERY = get_secret("teamspeak.query")


class teamspeak(RollbotPlugin):
    def __init__(self, bot, logger=logging.getLogger(__name__)):
        RollbotPlugin.__init__(self, "teamspeak", bot, logger=logger)

    def on_command(self, db, message):
        nicks = requests.get(TS3_QUERY).json()["users"]
        if len(nicks) == 0:
            return RollbotResponse(message, txt="No one in teamspeak!")
        if len(nicks) == 1:
            return RollbotResponse(message, txt="Only "+nicks[0])
        return RollbotResponse(message, txt="I see the following people: "+", ".join(nicks))


class speamteek(teamspeak):
    def __init__(self, bot, logger=logging.getLogger(__name__)):
        RollbotPlugin.__init__(self, "speamteek", bot, logger=logger)

    def on_command(self, db, message):
        r = super().on_command(db, message)
        if r.is_success:
            r.txt = r.txt[::-1]
        return r


class teamscream(teamspeak):
    def __init__(self, bot, logger=logging.getLogger(__name__)):
        RollbotPlugin.__init__(self, "teamscream", bot, logger=logger)

    def on_command(self, db, message):
        r = super().on_command(db, message)
        if r.is_success:
            r.txt = r.txt.upper()
        return r

class ts(teamspeak):
    def __init__(self, bot, logger=logging.getLogger(__name__)):
        RollbotPlugin.__init__(self, "ts", bot, logger=logger)

    def on_command(self, db, message):
        r = super().on_command(db, message)
        if r.is_success:
            shortNicks = requests.get(TS3_QUERY).json()["users"][0]
            r.txt = "i c : " + ", ".join(shortNicks)
        return r