mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-29 17:11:44 +00:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from functools import wraps
|
|
from typing import Callable
|
|
|
|
from core.admin.services import BotAdminService
|
|
from utils.service.inject import inject
|
|
|
|
|
|
def bot_admins_rights_check(func: Callable) -> Callable:
|
|
"""BOT ADMIN 权限检查"""
|
|
|
|
@inject
|
|
def get_bot_admin_service(bot_admin_service: BotAdminService = None):
|
|
return bot_admin_service
|
|
|
|
@wraps(func)
|
|
async def decorator(*args, **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)
|
|
bot_admin_service = get_bot_admin_service()
|
|
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
|