From 8c02ce86c7f549d372b7dda5ae85ed76f9101c25 Mon Sep 17 00:00:00 2001 From: youhuo Date: Thu, 18 Aug 2022 16:27:28 +0800 Subject: [PATCH] feat: set first user is admin when there is no admin in db --- services/auth_service.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/services/auth_service.py b/services/auth_service.py index 8918a63..7a2bee0 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 -- Gitee