TeamSpeak datebased GroupAssigner

cleaned up working release

+ properly register for servernotify events
+ process only voice_clients
+ properly check if client is already a group member
+ logging

1 files changed, 72 insertions, 0 deletions

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..532e4af
--- /dev/null
+++ b/main.py
@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+import logging
+
+import ts3
+
+from config import Config
+
+
+def watcher(conn):
+    # Register for events
+    # https://yat.qa/ressourcen/server-query-notify/#server
+    conn.servernotifyregister(event="server")
+
+    while True:
+        conn.send_keepalive()
+
+        try:
+            event = conn.wait_for_event(timeout=60)
+
+        except ts3.query.TS3TimeoutError:
+            pass
+
+        else:
+            # only parse entering clients info
+            if event.event == "notifycliententerview":
+
+                # skip query clients -- query client = 1 , voice client = 0
+                if event[0]['client_type'] == '0':
+
+                    # reasonid should be 0 not sure though
+                    if event[0]["reasonid"] == "0":
+
+                        user_grps = event.parsed[0]['client_servergroups'].split(sep=',')
+                        gid = Config.GID
+
+                        # only try to add nonmembers to group
+                        if str(gid) not in user_grps:
+
+                            cldbid = event.parsed[0]['client_database_id']
+
+                            # https://yat.qa/ressourcen/server-query-kommentare/
+                            # Usage: servergroupaddclient sgid={groupID} cldbid={clientDBID}
+                            try:
+                                cmd = conn.servergroupaddclient(sgid=gid, cldbid=cldbid)
+
+                                if cmd.error['id'] != '0':
+                                    logger.info(cmd.data[0].decode("utf-8"))
+
+                                # log process
+                                logmsg = '{client_nickname}:{client_database_id} - added to {gid}'
+                                logging.info(logmsg.format(**event[0], gid=gid))
+
+                            except KeyError:
+                                logger.error(str(event.parsed))
+                                pass
+
+
+if __name__ == "__main__":
+    # logging
+    logger = logging.getLogger()
+    logger.setLevel(logging.INFO)
+
+    logf = logging.FileHandler('info.log')
+    logf.setFormatter(logging.Formatter("%(asctime)s - %(levelname)-8s - %(message)s"))
+
+    logger.addHandler(logf)
+
+    with ts3.query.TS3Connection(Config.HOST, Config.PORT) as ts3conn:
+        ts3conn.login(client_login_name=Config.USER, client_login_password=Config.PW)
+        ts3conn.use(sid=Config.SID)
+        watcher(ts3conn)