Compare commits
5 Commits
5a331e773d
...
ff6c1627b0
Author | SHA1 | Date | |
---|---|---|---|
ff6c1627b0 | |||
690f3bdd65 | |||
8611ce65be | |||
57b1dd2a56 | |||
31098af84b |
10
cmds.py
10
cmds.py
@ -64,9 +64,8 @@ def cmds(app, data):
|
||||
action="store_true")
|
||||
parser.add_argument('-g', '--group', help='组名')
|
||||
parser.add_argument('-m', '--command', help='命令名')
|
||||
parser.add_argument('-au', '--adduser', help='添加用户到组',
|
||||
parser.add_argument('-au', '--adduser', help='添加用户/群到组',
|
||||
action="store_true")
|
||||
parser.add_argument('-u', '--user', help='用户QQ号')
|
||||
|
||||
args = parser.parse_args(sqt.args)
|
||||
e: AsyncEnforcer = sqt.app.ctx.e
|
||||
@ -80,9 +79,9 @@ def cmds(app, data):
|
||||
else:
|
||||
msg = Message(content='缺失参数,请使用-h查看帮助', room_id=sqt.room_id)
|
||||
if args.adduser:
|
||||
if args.user and args.group:
|
||||
if await e.add_role_for_user(args.user, args.group):
|
||||
msg = Message(content=f'添加成功:g, {args.user}, {args.group}', room_id=sqt.room_id)
|
||||
if args.group:
|
||||
if await e.add_role_for_user(str(sqt.room_id), args.group):
|
||||
msg = Message(content=f'添加成功:g, {sqt.room_id}, {args.group}', room_id=sqt.room_id)
|
||||
else:
|
||||
msg = Message(content='添加失败,用户已在组内或其它错误', room_id=sqt.room_id)
|
||||
else:
|
||||
@ -94,5 +93,6 @@ def cmds(app, data):
|
||||
if msg == '':
|
||||
msg = Message(content='参数错误,请使用-h查看帮助', room_id=sqt.room_id)
|
||||
await sqt.app.ctx.sio.emit('sendMessage', msg.to_json())
|
||||
await e.save_policy()
|
||||
|
||||
return sio_decorator
|
||||
|
35
sio_model.py
35
sio_model.py
@ -3,7 +3,7 @@ from dataclasses import dataclass
|
||||
from functools import wraps
|
||||
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 sanic import Sanic
|
||||
from sanic.log import logger
|
||||
@ -117,22 +117,25 @@ class SioDecorator:
|
||||
room_id = self.data['roomId']
|
||||
|
||||
if sender_id != self.app.ctx.sio_config.self_id:
|
||||
parser = ArgumentParser(self._cmd[0])
|
||||
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])
|
||||
|
||||
sqt = SioRequest(app=self.app,
|
||||
parser=parser,
|
||||
args=self._cmd[1:],
|
||||
key=self._cmd[0],
|
||||
sender_id=sender_id,
|
||||
sender_name=sender_name,
|
||||
content=self._content,
|
||||
room_id=room_id,
|
||||
data=self.data)
|
||||
try:
|
||||
await func(sqt)
|
||||
except PrintMessage as e:
|
||||
msg = Message(content=str(e), room_id=room_id)
|
||||
await self.app.ctx.sio.emit('sendMessage', msg.to_json())
|
||||
sqt = SioRequest(app=self.app,
|
||||
parser=parser,
|
||||
args=self._cmd[1:],
|
||||
key=self._cmd[0],
|
||||
sender_id=sender_id,
|
||||
sender_name=sender_name,
|
||||
content=self._content,
|
||||
room_id=room_id,
|
||||
data=self.data)
|
||||
try:
|
||||
await func(sqt)
|
||||
except PrintMessage as e:
|
||||
msg = Message(content=str(e), room_id=room_id)
|
||||
await self.app.ctx.sio.emit('sendMessage', msg.to_json())
|
||||
|
||||
|
||||
@dataclass
|
||||
|
Loading…
Reference in New Issue
Block a user