PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-22 15:36:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

🔧 修改BOT ADMIN权限检查修饰器获取依赖服务的方式

Browse Source
This commit is contained in:
洛水居室 2022-08-06 17:02:01 +08:00
parent 0414998e2d
commit 8b49356836
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/decorators/admins.py
View File

@ -8,9 +8,12 @@ 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):
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
@ -19,6 +22,7 @@ def bot_admins_rights_check(func: Callable) -> Callable:
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()