!2 解决新建用户没有关联角色时,会关联所有角色BUG。

Merge pull request !2 from liujian8670/N/A
This commit is contained in:
ktianc 2023-04-23 04:24:26 +00:00 committed by Gitee
commit 93fbfbbe39
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F

View File

@ -63,9 +63,10 @@ class UserDal(DalBase):
data.password = self.model.get_password_hash(password) data.password = self.model.get_password_hash(password)
data.avatar = data.avatar if data.avatar else settings.DEFAULT_AVATAR data.avatar = data.avatar if data.avatar else settings.DEFAULT_AVATAR
obj = self.model(**data.dict(exclude={'role_ids'})) obj = self.model(**data.dict(exclude={'role_ids'}))
roles = await RoleDal(self.db).get_datas(limit=0, id=("in", data.role_ids), v_return_objs=True) if data.role_ids:
for role in roles: roles = await RoleDal(self.db).get_datas(limit=0, id=("in", data.role_ids), v_return_objs=True)
obj.roles.append(role) for role in roles:
obj.roles.append(role)
await self.flush(obj) await self.flush(obj)
return await self.out_dict(obj, v_options, v_return_obj, v_schema) return await self.out_dict(obj, v_options, v_return_obj, v_schema)