diff --git a/services/auth_service.py b/services/auth_service.py index 8918a6303248ae49a6916d0534639e3c9bcbe755..7a2bee0d33048d8a095b985033b66a2bc8600a34 100644 --- a/services/auth_service.py +++ b/services/auth_service.py @@ -45,7 +45,10 @@ async def logout_jwt(name): async def register_user(data): data.update(dict({'token': create_private_secret()})) try: - await User().save(data) + user = await User().save(data) + if not await User().query_obj_one(User.role == User_Role.ADMIN.value): + user.role = User_Role.ADMIN + await user.update() except IntegrityError: if await User.query_obj_one(User.nick_name == data['nick_name']): return ERROR_DUPLICATED_USER_NAME, False