mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-12-01 11:41:59 +00:00
32 lines
961 B
Python
32 lines
961 B
Python
|
from functools import wraps
|
||
|
|
from typing import Callable
|
||
|
|
|
||
|
|
from app.admin import BotAdminService
|
||
|
|
from utils.app.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
|