ant-vue-admin/server/api/views.py

from django.shortcuts import render, redirect, HttpResponse
from django.http import JsonResponse, HttpResponseRedirect
from django.views.generic import View
# from . import models
from .models import User, Menu
from django.core.serializers import serialize
import json


# Create your views here.

def index(request):
    data = {
        'name': {
            'firstname': 'Tom',
            'lastname': 'john'
        },
        'age': 18,
        'sex': '男'
    }
    return JsonResponse(data, json_dumps_params={"ensure_ascii": False})

# class cbv(View):
#   def get(self,request):
#     print(request)
#     print(request.GET.get('username'))
#     return JsonResponse({'method':'get','data':request.GET})

#   def post(self,request):
#     return JsonResponse({'method':'post','data':request.POST})
#   def put(self,request):
#     return JsonResponse({'method':'put'})
#   def delete(self,request):
#     return JsonResponse({'method':'delete'})

# class extend_cbv(cbv):
#   # 重写post请求
#   def post(self,request):
#     return JsonResponse({'method':'post','msg':'重写post请求'},json_dumps_params={"ensure_ascii":False})


def format_useinfo(user):
    format_user = json.loads(
        serialize('json', user, ensure_ascii=False)
    )
    for item in format_user:
        res = item.get('fields')
        res['userId'] = item.get('pk')
        res['roles'] = [
            {
                'roleName': res.get('roleName'), 'value': res.get('value')
            }
        ]
        res.pop('roleName')
        res.pop('value')
        res.pop('password')
    return res


def format_uselist(list):
    json_user = json.loads(
        serialize('json', list, ensure_ascii=False)
    )
    res = []
    for item in json_user:
        item['fields'].pop('value')
        item['fields'].pop('token')
        user = item.get('fields')
        user['id'] = item.get('pk')
        # res['roles'] = [
        #     {
        #         'roleName': res.get('roleName'), 'value': res.get('value')
        #     }
        # ]
        # res.pop('roleName')
        # res.pop('value')
        # res.pop('password')
        res.append(user)
    return res


def format_login_result(user):
    format_user = json.loads(
        serialize('json', user, ensure_ascii=False)
    )
    for item in format_user:
        res = {}
        res['userId'] = item.get('pk')
        res['menus'] = item.get('menus')
        res['token'] = item.get('fields').get('token')
        res['roles'] = [
            {
                'roleName': item.get('fields').get('roleName'), 'value': item.get('fields').get('value')
            }
        ]
    return res


def format_menus(querySetObj, mode):
    # querySet 转json对象
    data = json.loads(serialize('json', querySetObj, ensure_ascii=False))
    # 定义一级菜单列表(无父id)
    menus = []
    # 子菜单列表
    childrens = []
    for item in data:
        item['fields']['id'] = item.get('pk')
        # mode True -- 否将某些字段合并处理成前台路由需要的数据结构;
        if mode:
            item['fields']['meta'] = {
                "title": item['fields'].pop('title'),
                "affix":  item['fields'].pop('affix'),
                "icon": item['fields'].pop('icon'),
            }
            if (item.get('fields').get('parent') == None):
                menus.append(item.get('fields'))
            else:
                childrens.append(item.get('fields'))

            for children in childrens:
                # 将列表转化成树形菜单
                getTree(children, menus, childrens)

        else:  # False -- 不必转成树结构, menuName 语言包需要到前台转中文
            item['fields']["menuName"] = item['fields'].pop("title")
            item['fields'].pop('affix')
            menus.append(item.get('fields'))

    return menus


"""
def getTree(children, menus, childrens)
根据传递过来的父菜单id，递归设置各层次父菜单的子菜单列表
:children: 单个子菜单
:param menus: 父菜单列表 一级菜单
:childrens： 子菜单列表，第二次调用
"""


def getTree(children, menus, childrens):
    # 遍历父菜单
    for m in menus:
        # 当前菜单的id 和 子菜单的父id 是否匹配
        if m.get('id') == children.get('parent'):
            if not m.get('children'):
                m['children'] = []
            if not children in m['children']:  # 判单子菜单是否已存在
                m['children'].append(children)

            if m.get('children'):
                for children in childrens:
                    # 判断子菜单是否还有匹配的子菜单
                    getTree(children, m.get('children'), childrens)


# 请求---------------------------------------------------------------------
class login(View):
    def post(self, request):
        data = json.loads(request.body)
        user = User.objects.all().filter(username=data.get(
            'username'), password=data.get('password'))
        if user:
            res = format_login_result(user)
            return JsonResponse({'code': 0, 'msg': 'success', 'result': res}, json_dumps_params={"ensure_ascii": False})
        else:
            return JsonResponse({'code': 1, 'error': '用户名或密码不正确'}, json_dumps_params={"ensure_ascii": False})


class getUserInfoById(View):
    def get(self, request):
        userid = request.GET.get('userId')
        user = User.objects.all().filter(id=userid)
        if user:
            # res = {'codeList': ['1000', '3000', '5000']}
            res = format_useinfo(user)
            return JsonResponse({'code': 0, 'result': res, })
        else:
            return JsonResponse({'code': 1, 'result': {'message': 'error'}}, json_dumps_params={"ensure_ascii": False})


class getUserList(View):
    def get(self, request):
        users = User.objects.all()
        if users:
            res = format_uselist(users)
            return JsonResponse({'code': 0, 'result': res, }, json_dumps_params={"ensure_ascii": False})
        else:
            return JsonResponse({'code': 1, 'result': {'message': 'no role,create role'}}, json_dumps_params={"ensure_ascii": False})


class addUser(View):
    def post(self, request):
        data = json.loads(request.body)
        menuIdList = data.pop('menus')
        menus = Menu.objects.all().filter(id__in=menuIdList)
        # print(menus)
        # data.menus = menus
        print(data)
        print('--------------data------------')
        if User.objects.all().filter(username=data.get('username')):
            return JsonResponse({'code': 1, 'result': {'error': '该角色名已被占用'}}, json_dumps_params={"ensure_ascii": False})
        else:
            data['token'] = data.get('username') + 'token'
            userObj = User.objects.create(**data)
            userObj.menus.set(menus)

            return JsonResponse({'code': 0, 'result': {'data': '角色创建成功'}}, json_dumps_params={"ensure_ascii": False})


class editUser(View):
    def post(self, request):
        data = json.loads(request.body)
        user = User.objects.all().filter(id=data.get('id')).first()
        if user:
            menuIdList = data.pop('menus')
            menus = Menu.objects.all().filter(id__in=menuIdList)
            user.username = data.get('username')
            user.password = data.get('password')
            user.realName = data.get('realName')
            user.status = data.get('status')
            user.detail = data.get('detail')
            user.menus.set(menus)
            user.save()
            return JsonResponse({'code': 0, 'result': {'data': '修改成功'}}, json_dumps_params={"ensure_ascii": False})
        else:
            return JsonResponse({'code': 1, 'result': {'data': '查无此人'}}, json_dumps_params={"ensure_ascii": False})
            # userObj = User.objects.create(**data)
            # userObj.menus.set(menus)


class deleteUser(View):
    def post(self, request):
        data = json.loads(request.body)
        user = User.objects.all().filter(id=data.get('id'))
        try:
            if user:
                user.delete()
                return JsonResponse({'code': 0, 'result': {'data': '删除成功'}}, json_dumps_params={"ensure_ascii": False})
            else:
                return JsonResponse({'code': 1, 'result': {'data': '查无此人'}}, json_dumps_params={"ensure_ascii": False})
        except Exception as e:
            print(e)
            return JsonResponse({'code': 1, 'result': {'data': '服务端错误'}}, json_dumps_params={"ensure_ascii": False})


class getMenuListById(View):
    def post(self, request):
        # 获取参数id
        userId = json.loads(request.body).get('id')
        # 当前用户
        querSetUser = User.objects.all().filter(id=userId)
        user = format_useinfo(querSetUser)
        # 获取当前user 的menus(id集合)
        menuIdList = user.get('menus')
        # 匹配查询 id__in=[1,2,3]
        querySetMenu = Menu.objects.all().filter(id__in=menuIdList)
        menus = format_menus(querySetMenu, True)
        return JsonResponse({'code': 0, 'result': menus})


class getAllMenuList(View):
    def get(self, request):
        querySetObj = Menu.objects.all()
        menus = format_menus(querySetObj, False)
        return JsonResponse({'code': 0, 'result': {'format': True, 'menus': menus}})


class add_menu(View):
    def post(self, request):
        data = request.POST
        Menu.objects.create(**data)
        return JsonResponse({'code': 1, 'error': '菜单添加成功'})