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