mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-29 09:07:13 +00:00
32 lines
972 B
Python
32 lines
972 B
Python
from functools import wraps
|
|
from typing import Callable
|
|
|
|
from apps.admin.services import BotAdminService
|
|
from utils.apps.inject import inject
|
|
|
|
|
|
def bot_admins_rights_check(func: Callable) -> Callable:
|
|
"""BOT ADMIN 权限检查"""
|
|
|
|
@wraps(func)
|
|
@inject
|
|
async def decorator(*args, bot_admin_service: BotAdminService, **kwargs):
|
|
if len(args) == 3:
|
|
# self update context
|
|
_, update, context = args
|
|
elif len(args) == 2:
|
|
# update context
|
|
update, context = args
|
|
else:
|
|
return await func(*args, **kwargs)
|
|
if bot_admin_service is None:
|
|
raise RuntimeError("bot_admin_service is None")
|
|
admin_list = await bot_admin_service.get_admin_list()
|
|
if update.message.from_user.id in admin_list:
|
|
return await func(*args, **kwargs)
|
|
else:
|
|
await update.message.reply_text("权限不足")
|
|
return None
|
|
|
|
return decorator
|