22 lines
709 B
Python
22 lines
709 B
Python
from functools import wraps
|
|
|
|
from telegram_downloader_bot.logger import log
|
|
from telegram_downloader_bot.settings import settings
|
|
|
|
allowed_ids = settings.allowed_ids.split(",")
|
|
allowed_ids = [int(x) for x in allowed_ids]
|
|
|
|
|
|
def protected(func):
|
|
@wraps(func)
|
|
async def wrapper(client, message):
|
|
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
|