添加退出登录,菜单改为后端api

This commit is contained in:
xie7654
2025-07-01 20:43:50 +08:00
parent 7273ee275b
commit 61fb71133d
8 changed files with 58 additions and 23 deletions

View File

@@ -18,4 +18,5 @@ urlpatterns = [
path('login/', views.user.UserLogin.as_view()), path('login/', views.user.UserLogin.as_view()),
path('info/', views.user.UserInfo.as_view()), path('info/', views.user.UserInfo.as_view()),
path('codes/', views.user.Codes.as_view()), path('codes/', views.user.Codes.as_view()),
path('logout/', views.user.Logout.as_view()),
] ]

View File

@@ -15,6 +15,7 @@ class MenuMetaSerializer(serializers.ModelSerializer):
model = MenuMeta model = MenuMeta
fields = '__all__' fields = '__all__'
class MenuSerializer(CustomModelSerializer): class MenuSerializer(CustomModelSerializer):
"""菜单序列化器""" """菜单序列化器"""
parent = serializers.CharField(source='pid.name', read_only=True) parent = serializers.CharField(source='pid.name', read_only=True)
@@ -60,6 +61,13 @@ class MenuSerializer(CustomModelSerializer):
return super().update(instance, validated_data) return super().update(instance, validated_data)
class MenuUserSerializer(MenuSerializer):
def get_children(self, obj):
children = obj.children.exclude(type='button')
if children:
return MenuUserSerializer(children, many=True).data
return []
class MenuMetaViewSet(viewsets.ModelViewSet): class MenuMetaViewSet(viewsets.ModelViewSet):
"""菜单元数据视图集""" """菜单元数据视图集"""
@@ -106,6 +114,16 @@ class MenuViewSet(CustomModelViewSet):
def path_exists(self, request): def path_exists(self, request):
return self._build_response() return self._build_response()
@action(detail=False, methods=['get'], url_path='user_menu')
def user_menu(self, request):
user = self.request.user
if user.is_superuser:
menus = Menu.objects.filter(pid__isnull=True).exclude(type='button').order_by('sort')
else:
menus = Menu.objects.filter(pid__isnull=True,
role__users=user).exclude(type='button').order_by('sort').distinct()
menus_data = MenuUserSerializer(menus, many=True).data
return self._build_response(data=menus_data)
def update(self, request, *args, **kwargs): def update(self, request, *args, **kwargs):
partial = kwargs.pop('partial', False) partial = kwargs.pop('partial', False)

View File

@@ -5,6 +5,7 @@ from rest_framework.authtoken.views import ObtainAuthToken
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.views import APIView from rest_framework.views import APIView
from django.contrib.auth.hashers import make_password from django.contrib.auth.hashers import make_password
from rest_framework.permissions import IsAuthenticated
from system.models import User, Menu from system.models import User, Menu
from system.views.menu import MenuSerializer from system.views.menu import MenuSerializer
@@ -70,21 +71,17 @@ class UserInfo(APIView):
user_data = UserSerializer(user).data user_data = UserSerializer(user).data
if user.is_superuser: if user.is_superuser:
roles = ['admin'] roles = ['admin']
menus = Menu.objects.filter(pid__isnull=True).order_by('sort') # menus = Menu.objects.filter(pid__isnull=True).order_by('sort')
permissions = Menu.objects.filter(type='button').order_by('sort').values_list('auth_code', flat=True) # permissions = Menu.objects.filter(type='button').order_by('sort').values_list('auth_code', flat=True)
else: else:
roles = user.get_role_name roles = user.get_role_name
menus = Menu.objects.filter(pid__isnull=True, role__users=user).order_by('sort').distinct() # menus = Menu.objects.filter(pid__isnull=True, role__users=user).order_by('sort').distinct()
permissions = Menu.objects.filter(type='button', role__users=user).order_by('sort').distinct().values_list('auth_code', flat=True) # permissions = Menu.objects.filter(type='button', role__users=user).order_by('sort').distinct().values_list('auth_code', flat=True)
menus_data = MenuSerializer(menus, many=True).data # menus_data = MenuSerializer(menus, many=True).data
user_data['roles'] = roles
return Response({ return Response({
"code": 0, "code": 0,
"data": { "data": user_data,
"menus": menus_data,
"permissions": permissions,
"roles": roles,
"user": user_data,
},
"error": None, "error": None,
"message": "ok" "message": "ok"
}) })
@@ -120,3 +117,18 @@ class UserViewSet(CustomModelViewSet):
ordering_fields = ['create_time', 'id'] ordering_fields = ['create_time', 'id']
ordering = ['-create_time'] ordering = ['-create_time']
class Logout(APIView):
permission_classes = [IsAuthenticated]
def post(self, request, *args, **kwargs):
# user = request.user
# 删除用户的Token
# Token.objects.filter(user=user).delete()
return Response({
"code": 0,
"data": None,
"error": None,
"message": "登出成功"
})

View File

@@ -16,16 +16,16 @@
<link rel="icon" href="/favicon.ico" /> <link rel="icon" href="/favicon.ico" />
<script> <script>
// 生产环境下注入百度统计 // 生产环境下注入百度统计
if (window._VBEN_ADMIN_PRO_APP_CONF_) { // if (window._VBEN_ADMIN_PRO_APP_CONF_) {
var _hmt = _hmt || []; // var _hmt = _hmt || [];
(function () { // (function () {
var hm = document.createElement('script'); // var hm = document.createElement('script');
hm.src = // hm.src =
'https://hm.baidu.com/hm.js?b38e689f40558f20a9a686d7f6f33edf'; // 'https://hm.baidu.com/hm.js?b38e689f40558f20a9a686d7f6f33edf';
var s = document.getElementsByTagName('script')[0]; // var s = document.getElementsByTagName('script')[0];
s.parentNode.insertBefore(hm, s); // s.parentNode.insertBefore(hm, s);
})(); // })();
} // }
</script> </script>
</head> </head>
<body> <body>

View File

@@ -6,5 +6,7 @@ import { requestClient } from '#/api/request';
* 获取用户所有菜单 * 获取用户所有菜单
*/ */
export async function getAllMenusApi() { export async function getAllMenusApi() {
return requestClient.get<RouteRecordStringComponent[]>('/menu/all'); return requestClient.get<RouteRecordStringComponent[]>(
'/system/menu/user_menu',
);
} }

View File

@@ -128,7 +128,7 @@ watch(
<UserDropdown <UserDropdown
:avatar :avatar
:menus :menus
:text="userStore.userInfo?.realName" :text="userStore.userInfo?.username"
:description="userStore.userInfo?.email" :description="userStore.userInfo?.email"
tag-text="Pro" tag-text="Pro"
@logout="handleLogout" @logout="handleLogout"

View File

@@ -9,5 +9,6 @@ export const overridesPreferences = defineOverridesPreferences({
// overrides // overrides
app: { app: {
name: import.meta.env.VITE_APP_TITLE, name: import.meta.env.VITE_APP_TITLE,
accessMode: 'backend', // 或 'frontend'
}, },
}); });

View File

@@ -15,6 +15,7 @@ interface UserInfo extends BasicUserInfo {
* accessToken * accessToken
*/ */
token: string; token: string;
roles: [];
} }
export type { UserInfo }; export type { UserInfo };