Melhoramentos no sistema de autorização...

- Economia de consultas ao salvar uma lista de usuários automaticamente autorizados;
- Correção de um algoritmo porco que não servia para nada (só checkava se o membro era do grupo de onde a mensagem veio, o que é uma clara tolice)

Author: prppedro

api.py

@@ -133,5 +133,4 @@ def isGroupMember(chat_id, user_id):
     if (response['ok'] == True):
         return True
     else:
-        return False
-
+        return False

data/config.json.dist

@@ -2,6 +2,10 @@
     "sudoers": [
         49681384
     ],
+    "authorized_chats": [
+        -7016014,
+        -1001049208929
+    ],
     "authorized_users": [
         49681384,
         14160874,
@@ -24,6 +28,9 @@
         102544375,
         52433246,
         78977189
+    ],
+    "automatically_authorized_users": [
+
     ],
     "enabled_plugins": {
         "^[!/]melembra(?:@MercuriumBot)? ?([0-9]+d)?([0-9]+h)?([0-9]+m)?([0-9]+s)?(?: (.+))?$": "melenbra",

plugins/stats.py

@@ -3,8 +3,12 @@
 from os import environ
 import requests
 import json
+import psycopg2 as psql
+from os import environ as env
 from api import send_message
 
+SQLPARAMS = "dbname=mercurium user=mercurium password=" + env['REBORNKEY']
+
 # Isso deve ser suficiente para transplantar as estatísticas para
 # um plugin independente
 def on_msg_received(msg, matches):
@@ -29,6 +33,12 @@ def return_statistics(chat):
     else:
         return "n ten isotirco nesste xet ;()"
 
+    # Conecta-se à base SQL
+    #conn = psql.connect(SQLPARAMS)
+
+
+
+
     dictofdicts = {}
 
     # Passa cada usuário do chat pro dictofdicts, onde a chave é o nome e o valor é a qtde.

reborn.py

@@ -80,16 +80,33 @@ def is_authorized(msg):
 
     if msg["from"]["id"] in config.config["authorized_users"]:
         return True
+    elif msg["from"]["id"] in config.config["automatically_authorized_users"]:
+        return True # Para usuários autorizados automaticamente com base no pertencimento
+                    # a grupos. Por ora, só podem ser removidos manualmente
     else:
         # Como eu não sei a ID de todos os indivíduos romanos, resolvi fazer uma query para ver se o caboclo
         # pertence ao chat...
-        # TODO: Criar um modo de adicionar o caboclo à lista de permissões
-        if api.isGroupMember(msg["chat"]["id"], msg["from"]["id"]) is True:
+
+        if isAuthorizedGroupMember(msg["from"]["id"]) is True:
+            # Adiciona o indivíduo de vez, economizando essa consulta burra e redundante
+            config.config["automatically_authorized_users"].append(msg["from"]["id"])
+            config.save_config()
             return True
         else:
             return False
 
 
+# Verifica se o caboclo faz parte de algum dos chats autorizados
+def isAuthorizedGroupMember(user_id):
+
+    for gid in config.config["authorized_chats"]:
+        if api.isGroupMember(gid, user_id):
+            return True
+
+    return False # Solução estranha, porém funcional e razoavelmente elegante
+
+
+
 def msg_matches(msg_text):
     for query, plugin in config.plugins.items():
         pattern = re.compile(query)