Prevent the bot from being used in groups

wallamanta/walladb.py

@@ -77,15 +77,18 @@ def setup_db():
     con.close()
 
 def is_user_valid(telegram_user_id):
-    con = connect_db()
-    cur = con.cursor(prepared=True)
-    params = (telegram_user_id,)
-    cur.execute(f"SELECT * FROM users WHERE telegram_user_id=%s AND active=True", params)
-    try:
-        ret = cur.fetchone() != None
-    except:
+    if telegram_user_id < 0:
         ret = False
-    con.close()
+    else:
+        con = connect_db()
+        cur = con.cursor(prepared=True)
+        params = (telegram_user_id,)
+        cur.execute(f"SELECT * FROM users WHERE telegram_user_id=%s AND active=True", params)
+        try:
+            ret = cur.fetchone() != None
+        except:
+            ret = False
+        con.close()
     return ret
 
 def is_user_expired(telegram_user_id):

wallamanta/wallamanta.py

@@ -388,7 +388,9 @@ async def test_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
     telegram_user_id = helpers.get_telegram_user_id(update)
     telegram_user_name = helpers.get_telegram_user_name(update)
     #if not walladb.is_user_valid(telegram_user_id):
-    if walladb.get_user(telegram_user_id) == "NoName":
+    if telegram_user_id < 0:
+        message = "Este bot no se puede usar en grupos."
+    elif walladb.get_user(telegram_user_id) == "NoName":
         until = helpers.get_date_ahead(7)
         walladb.add_test_user(telegram_user_id, telegram_user_name, until)
         message = f"Periodo de prueba activado hasta el {helpers.get_spanish_date(until)}."