Browse Source

switch ts3 to @as_plugin

ivan 5 years ago
parent
commit
d52b998811
1 changed files with 8 additions and 11 deletions
  1. 8 11
      src/plugins/teamspeak.py

+ 8 - 11
src/plugins/teamspeak.py

@@ -1,7 +1,7 @@
 import logging
 
 import requests
-from command_system import RollbotResponse, RollbotPlugin
+from command_system import RollbotResponse, RollbotPlugin, as_plugin
 from config import get_secret
 
 TS3_QUERY = get_secret("teamspeak.query")
@@ -24,7 +24,8 @@ 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)
+    def on_command(self, db, message): 
+        r = super().on_command(db, message)
         if r.is_success:
             r.txt = r.txt[::-1]
         return r
@@ -40,12 +41,8 @@ class teamscream(teamspeak):
             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):
-        nicks = requests.get(TS3_QUERY).json()["users"]
-        initials = [nick[0] for nick in nicks]
-        return RollbotResponse(message, txt="i c : "+", ".join(initials))
-
+@as_plugin
+def ts(message):
+    nicks = requests.get(TS3_QUERY).json()["users"]
+    initials = [nick[0] for nick in nicks]
+    return RollbotResponse(message, txt="i c : "+", ".join(initials))