diff --git a/MangoServer/PyAutoTest/auto_test/auto_user/views/user.py b/MangoServer/PyAutoTest/auto_test/auto_user/views/user.py index 8a0f94159c14776a81721266460786a6a6c133e4..9451d84577b5b046cbcb637956eda72785b8188f 100644 --- a/MangoServer/PyAutoTest/auto_test/auto_user/views/user.py +++ b/MangoServer/PyAutoTest/auto_test/auto_user/views/user.py @@ -138,7 +138,7 @@ class LoginViews(ViewSet): password = request.data.get('password') source_type = request.data.get('type') password = EncryptionTool.md5_32_small(**{'data': password}) - user_info = User.objects.filter(username=username, password=password).first() + user_info = User.objects.get(username=username, password=password) if not user_info: return ResponseData.fail(RESPONSE_MSG_0042) x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR') @@ -159,7 +159,7 @@ class LoginViews(ViewSet): "userName": user_info.username, "userId": user_info.id, "roleId": user_info.role.id if user_info.role else None, - "token": create_token({'id': user_info.id, 'username': user_info.username}), + "token": create_token({'id': user_info.id, 'username': user_info.username, 'nickname': user_info.nickname}), "selected_project": user_info.selected_project, "selected_environment": user_info.selected_environment, "roles": [ diff --git a/MangoServer/manage.py b/MangoServer/manage.py index 50263c36f8d3e32d8528094aaad48c8d6d062e02..be5dbf4b592843c10dc0a780db5d4fb781500741 100644 --- a/MangoServer/manage.py +++ b/MangoServer/manage.py @@ -8,7 +8,6 @@ import sys def main(): """Run administrative tasks.""" django_env = None - # 从参数中获取 --env 参数 for i, arg in enumerate(sys.argv): if '--env=' in arg: django_env = arg.split('=')[1]