adk23333/gitea_push2qq
Template
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: adk23333:5a331e773d88fd79cdd7048288a139deb0c9b64a
Branches Tags
adk23333:main
...
compare: adk23333:ff6c1627b0f230721ab19a8561533ded870421ec
Branches Tags
adk23333:main

5 Commits
5a331e773d ... ff6c1627b0

Author SHA1 Message Date
adk23333
ff6c1627b0 继续声明管理员的最高权限 2023-12-21 01:40:01 +08:00
adk23333
690f3bdd65 指令权限从用户改为群组 2023-12-21 01:32:12 +08:00
adk23333
8611ce65be 指令权限从用户改为群组 2023-12-21 01:27:29 +08:00
adk23333
57b1dd2a56 正式给指令添加权限限制 2023-12-21 01:25:47 +08:00
adk23333
31098af84b 正式给指令添加权限限制 2023-12-21 01:23:29 +08:00
2 changed files with 24 additions and 21 deletions
Show Stats Expand all files Collapse all files

10
cmds.py
View File

@ -64,9 +64,8 @@ def cmds(app, data):
action="store_true") action="store_true")
parser.add_argument('-g', '--group', help='组名') parser.add_argument('-g', '--group', help='组名')
parser.add_argument('-m', '--command', help='命令名') parser.add_argument('-m', '--command', help='命令名')
parser.add_argument('-au', '--adduser', help='添加用户到组', parser.add_argument('-au', '--adduser', help='添加用户/群到组',
action="store_true") action="store_true")
parser.add_argument('-u', '--user', help='用户QQ号')
args = parser.parse_args(sqt.args) args = parser.parse_args(sqt.args)
e: AsyncEnforcer = sqt.app.ctx.e e: AsyncEnforcer = sqt.app.ctx.e
@ -80,9 +79,9 @@ def cmds(app, data):
else: else:
msg = Message(content='缺失参数，请使用-h查看帮助', room_id=sqt.room_id) msg = Message(content='缺失参数，请使用-h查看帮助', room_id=sqt.room_id)
if args.adduser: if args.adduser:
if args.user and args.group: if args.group:
if await e.add_role_for_user(args.user, args.group): if await e.add_role_for_user(str(sqt.room_id), args.group):
msg = Message(content=f'添加成功：g, {args.user}, {args.group}', room_id=sqt.room_id) msg = Message(content=f'添加成功：g, {sqt.room_id}, {args.group}', room_id=sqt.room_id)
else: else:
msg = Message(content='添加失败，用户已在组内或其它错误', room_id=sqt.room_id) msg = Message(content='添加失败，用户已在组内或其它错误', room_id=sqt.room_id)
else: else:
@ -94,5 +93,6 @@ def cmds(app, data):
if msg == '': if msg == '':
msg = Message(content='参数错误，请使用-h查看帮助', room_id=sqt.room_id) msg = Message(content='参数错误，请使用-h查看帮助', room_id=sqt.room_id)
await sqt.app.ctx.sio.emit('sendMessage', msg.to_json()) await sqt.app.ctx.sio.emit('sendMessage', msg.to_json())
await e.save_policy()
return sio_decorator return sio_decorator

5
sio_model.py
View File

@ -3,7 +3,7 @@ from dataclasses import dataclass
from functools import wraps from functools import wraps
from typing import Optional, List, Union, Literal, Dict, Any from typing import Optional, List, Union, Literal, Dict, Any
from casbin import Enforcer from casbin import Enforcer, AsyncEnforcer
from pydantic import BaseModel, Field, model_validator from pydantic import BaseModel, Field, model_validator
from sanic import Sanic from sanic import Sanic
from sanic.log import logger from sanic.log import logger
@ -117,6 +117,9 @@ class SioDecorator:
room_id = self.data['roomId'] room_id = self.data['roomId']
if sender_id != self.app.ctx.sio_config.self_id: if sender_id != self.app.ctx.sio_config.self_id:
e: AsyncEnforcer = self.app.ctx.e
if (e.enforce(str(room_id), self._cmd[0][1:]) or
(await e.has_role_for_user(str(sender_id), 'admin'))):
parser = ArgumentParser(self._cmd[0]) parser = ArgumentParser(self._cmd[0])
sqt = SioRequest(app=self.app, sqt = SioRequest(app=self.app,