Fix minor problems with user IDs and datetime
This commit is contained in:
@@ -1,18 +1,22 @@
|
||||
import os
|
||||
from functools import wraps
|
||||
|
||||
from telegram_downloader_bot.logger import log
|
||||
|
||||
# Comma separated list of Telegram IDs that this bot will respond to
|
||||
allowed_ids_raw = os.getenv("ALLOWED_IDS", "")
|
||||
allowed_ids = allowed_ids_raw.split(",")
|
||||
|
||||
print(allowed_ids_raw)
|
||||
print(allowed_ids)
|
||||
allowed_ids = [int(x) for x in allowed_ids]
|
||||
|
||||
|
||||
def protected(func):
|
||||
@wraps(func)
|
||||
async def wrapper(client, message):
|
||||
if message.from_user.id not in allowed_ids:
|
||||
if int(message.from_user.id) not in allowed_ids:
|
||||
log.warning(
|
||||
f"User with ID {message.from_user.id} attempted to text this bot!")
|
||||
log.info(
|
||||
f"Only users allowed are: {' '.join(allowed_ids)}")
|
||||
return await message.reply_text("You are not on the list!")
|
||||
return await func(client, message)
|
||||
return wrapper
|
||||
|
||||
Reference in New Issue
Block a user