mirror of
https://github.com/TheCaduceus/FileStreamBot.git
synced 2026-01-15 16:33:25 -03:00
17 lines
564 B
Python
17 lines
564 B
Python
from pyrogram import Client
|
|
from pyrogram.types import Message, CallbackQuery
|
|
from typing import Union, Callable
|
|
from functools import wraps
|
|
from bot.config import Telegram
|
|
|
|
def verify_user(func: Callable):
|
|
|
|
@wraps(func)
|
|
async def decorator(client: Client, update: Union[Message, CallbackQuery]):
|
|
chat_id = str(update.from_user.id if update.from_user else update.chat.id)
|
|
|
|
if not Telegram.ALLOWED_USER_IDS or chat_id in Telegram.ALLOWED_USER_IDS:
|
|
return await func(client, update)
|
|
|
|
return decorator
|