#!/usr/bin/env python3
from telnetlib import Telnet
import toml
from flask import Flask, jsonify, render_template_string
app = Flask(__name__)
def get_users():
with open("secret.toml") as infile:
cfg = toml.load(infile)
login = ("login %s %s\n" % (cfg["user"], cfg["pass"])).encode("utf-8")
with Telnet(cfg["host"], cfg["port"], 5) as tn:
tn.write(login)
tn.write(b"use 1 -virtual\n")
tn.write(b"clientlist\n")
tn.write(b"whoami\n")
tn.write(b"quit\n")
response = tn.read_until(b"virtualserver_status").decode("ascii")
return [x.split("=", 1)[1] for x in response.split()
if x.startswith("client_nickname") and "serveradmin" not in x]
@app.route("/")
def get_status():
return jsonify({"users": get_users()})
@app.route("/page")
def get_status_page():
users = get_users()
if len(users) == 0:
text = "No one in teamspeak!"
elif len(users) == 1:
text = f"Only {users[0]}"
else:
text = "I see the following people: " + ", ".join(users)
return render_template_string("""
Teamspeak Server Status
{% if scrolling %}
{% endif %}
""", scrolling=True, r=text)
if __name__ == "__main__":
app.run("0.0.0.0", 5000, debug=True, threaded=True)