Jocaru/wallamanta
1
0
Fork 0
Code Issues 15 Packages Projects Releases Wiki Activity

Prevent the bot from being used in groups

Browse Source
This commit is contained in:
Joan
2023-07-15 14:40:54 +02:00
parent 5e4b8725cb
commit 9e29628dc1
2 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
wallamanta/walladb.py
View File

@@ -77,6 +77,9 @@ def setup_db():
con.close() con.close()
def is_user_valid(telegram_user_id): def is_user_valid(telegram_user_id):
if telegram_user_id < 0:
ret = False
else:
con = connect_db() con = connect_db()
cur = con.cursor(prepared=True) cur = con.cursor(prepared=True)
params = (telegram_user_id,) params = (telegram_user_id,)

4
wallamanta/wallamanta.py
View File

@@ -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_id = helpers.get_telegram_user_id(update)
telegram_user_name = helpers.get_telegram_user_name(update) telegram_user_name = helpers.get_telegram_user_name(update)
#if not walladb.is_user_valid(telegram_user_id): #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) until = helpers.get_date_ahead(7)
walladb.add_test_user(telegram_user_id, telegram_user_name, until) walladb.add_test_user(telegram_user_id, telegram_user_name, until)
message = f"Periodo de prueba activado hasta el {helpers.get_spanish_date(until)}." message = f"Periodo de prueba activado hasta el {helpers.get_spanish_date(until)}."