some refactoring, fixed db stuff and help section

pgbotlib/commands.py

@@ -1,6 +1,7 @@
 """ Respond to commands """
 
 import telethon
+import telethon.utils
 
 import pgbotlib.api
 import pgbotlib.dbstuff
@@ -16,7 +17,26 @@ class Commander:
     T_START = frozenset(['cmd_start'])
     T_START_E = frozenset(['cmd_start_enabled'])
     T_STOP = frozenset(['cmd_stop'])
-    T_STOP_D = frozenset(['cmd_stop_d'])
+    T_STOP_D = frozenset(['cmd_stop_disabled'])
+    DOC = """
+    Команды:
+        __.start__
+            запустить бота
+        __.stop__
+            остановить бота
+        __.list__
+            перечислить доступные токены
+        __.chat__
+            получить id текущего чата
+        __.users__
+            перечислить id пользователей
+        __.add token1[,token2,...] your phrase here__
+            добавить фразу your phrase here для реакции на токены
+        __.adduser id имя__
+            добавить пользователю имя
+        __.help__
+            вывести этот текст
+    """
 
     def __init__(self, config: dict,
                  client: telethon.TelegramClient,
@@ -24,7 +44,7 @@ class Commander:
                  db_conn: pgbotlib.dbstuff.DBConn,
                  namegen: pgbotlib.misc.NameGenerator,
                  responder: pgbotlib.response.Responder) -> None:
-        self.config = config
+        self.chats = config['chats']
         self.client = client
         self.admins = admins
         self.db_conn = db_conn
@@ -33,10 +53,9 @@ class Commander:
         self.available_tokens = [
                 str(token) for token, _ in self.responder.tokens]
 
-    def __add_entry(self, caller: int, command: str) -> bool:
+    def __add_response(self, caller: int, command: str) -> bool:
         if caller not in self.admins:
-            print('fuck off!')
-            return None
+            return 'а ты что ещё за хуй с горы?'
         input_tokens, phrase = command.strip().split(' ', 1)
         input_tokenset = frozenset(input_tokens.split(','))
         for token in input_tokenset:
@@ -44,26 +63,27 @@ class Commander:
                 return False
         query = 'INSERT INTO responses (tokens, response) values (%s,%s)'
         values = (','.join(sorted(input_tokenset)), phrase.strip())
-        return self.db_conn.update(query, values)
+        self.db_conn.update(query, values)
+        return 'да, господин!'
 
     def __add_user(self, caller: int, userspec: str) -> bool:
         if caller not in self.admins:
-            print('fuck off!')
-            return None
+            return 'а ты что ещё за хуй с горы?'
         user_id, names = userspec.strip().split(' ', 1)
         for name in names.strip().split(','):
             query = 'INSERT INTO names (tg_id, name) values(%s,%s)'
             values = (user_id, name)
             self.db_conn.update(query, values)
-        return True
+        return 'да, господин!'
+
 
     def __start_response(self) -> str:
-            if self.responder.enabled():
+            if self.responder.is_enabled():
                 return self.responder.get_response(self.T_START_E)
             return self.responder.get_response(self.T_START)
 
     def __stop_response(self) -> str:
-            if self.responder.enabled():
+            if self.responder.is_enabled():
                 return self.responder.get_response(self.T_STOP)
             return self.responder.get_response(self.T_STOP_D)
 
@@ -74,30 +94,38 @@ class Commander:
 
     async def action(self,
                      event: telethon.events.common.EventBuilder) -> None:
+        chat_id = telethon.utils.get_peer_id(event.message.peer_id)
+        if chat_id not in self.chats:
+            return None
         command = event.message.text
         sender = await event.get_sender()
         response = None
-        if command.startswith('/add '):
-            if self.__add_entry(sender.id, command[5:]):
-                response = 'success'
-            else:
-                response = 'failure'
-        elif command.startswith('/adduser '):
-            self.__add_user(sender.id, command[9:])
-        elif command == '/chat':
-            response = str(event.message.peer_id)
-        elif command == '/list':
+        if command.startswith('.add '):
+            try:
+                response = self.__add_response(sender.id, command[5:])
+            except Exception as e:
+                response = str(e)
+        elif command.startswith('.adduser '):
+            try:
+                response = self.__add_user(sender.id, command[9:])
+            except Exception as e:
+                response = str(e)
+        elif command == '.chat':
+            response = str(chat_id)
+        elif command == '.list':
             response = ', '.join(self.available_tokens)
-        elif command == '/users':
+        elif command == '.users':
             users = await self.client.get_participants(
                     entity=event.message.peer_id)
             response = self.__list_users(users)
-        elif command == '/start':
+        elif command == '.start':
             response = self.__start_response()
             self.responder.enable()
-        elif command == '/stop':
+        elif command == '.stop':
             response = self.__stop_response()
             self.responder.disable()
+        elif command == '.help':
+            response = self.DOC
         if response:
             await self.client.send_message(event.message.peer_id, response)
         return None