19 lines
505 B
Python
19 lines
505 B
Python
import os
|
|
from functools import wraps
|
|
|
|
# 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)
|
|
|
|
|
|
def protected(func):
|
|
@wraps(func)
|
|
async def wrapper(client, message):
|
|
if message.from_user.id not in allowed_ids:
|
|
return await message.reply_text("You are not on the list!")
|
|
return await func(client, message)
|
|
return wrapper
|