# -*- coding: utf-8 -*- from django.shortcuts import render from django.views.generic import ListView from apps.AppInfo.models import MyUser, Equip, Group from django import forms from django.db.models import Q import re import datetime # 分配设备 class EquipAllot(ListView): def get(self, request): # 接收当前用户 # current_user = request.session.get('username') current_user = MyUser.objects.get(username="admin") # print("current_user:", current_user) # current_user = MyUser.objects.get(username=current_user) page = request.GET.get("page") uname = request.GET.get("uname") user_list = MyUser.objects.all() nums = user_list.count() return render(request, 'backstageNet/equipallot/equipallot.html', context={"nums": nums,"page":page,"uname":uname}) def post(self, request): pass def allot_user_list(request): if request.method == 'GET': page = int(request.GET.get('page')) print("page=", page) uname = request.GET.get("uname") # 接收当前用户 # current_user = request.session.get('username') current_user = MyUser.objects.get(username="admin") # print("current_user:", current_user) # current_user = MyUser.objects.get(username=current_user) # if current_user.is_superuser == True: user_list = MyUser.objects.all().order_by( "-date_joined")[(10*(page-1)):(page*10)] if uname: user_list = MyUser.objects.filter(username__contains=uname).order_by( "-date_joined")[(10*(page-1)):(page*10)] return render(request, 'backstageNet/equipallot/allot_user_list.html', context={'user_list': user_list}) elif request.method == 'POST': # current_user = request.session.get('username') # print("current_user:", current_user) current_user = MyUser.objects.get(username="admin") print("筛选：") f_name = request.POST.get('uname') page = int(request.POST.get('page')) print("page:", page) user_list = MyUser.objects.filter(username__contains=f_name) nums = user_list.count() user_list = user_list[(10*(page-1)):(page*10)] return render(request, 'backstageNet/equipallot/allot_user_list.html', context={'user_list': user_list, "nums": nums}) from django.views.decorators.csrf import csrf_exempt @csrf_exempt def allot_equip_list(request): # 当前用户已拥有的设备的列表： equip_list1 = [] # 未被分配的设备的列表： equip_list2 = [] # 当前页的设备id列表: equip_ids = [] if request.method == 'GET': # 选中需要进行分配设备的用户后：显示的设备列表； uname = request.GET.get('uname') print("uname=", uname) uname_obj = MyUser.objects.get(username=uname) page = int(request.GET.get('page')) print("page=", page) # 接收当前用户 # current_user = request.session.get('username') # print("current_user:", current_user) current_user = MyUser.objects.get(username="admin") if current_user.is_superuser == True: equip_list = Equip.objects.filter( Q(equip_user=None) | Q(equip_user=uname_obj)) equip_list = list(equip_list) print('type equip_list:', type(equip_list)) # print('equip_list:', equip_list) for i in equip_list: if i.equip_user != None: equip_list1.append(i) else: equip_list2.append(i) # print('equip_list1:', equip_list1) # print('equip_list2:', equip_list2) equip_list = equip_list1+equip_list2 equip_list = equip_list[(10*(page-1)):(page*10)] for i in equip_list: equip_ids.append(i.equip_id) equip_ids = ','.join(equip_ids) return render(request, 'backstageNet/equipallot/allot_equip_list.html', context={'equip_list': equip_list, 'equip_ids': equip_ids, "uname": uname}) elif request.method == 'POST': # current_user = request.session.get('username') # print("current_user:", current_user) current_user = MyUser.objects.get(username="admin") print("筛选：") uname = request.POST.get('uname') print("uname=", uname) uname_obj = MyUser.objects.get(username=uname) f_type = request.POST.get('type') f_id = request.POST.get('f_id') print("f_type:", f_type) print("f_id:", f_id) page = int(request.POST.get('page')) print("page:", page) # 000 if f_id == None and f_type == '0': equip_list = Equip.objects.filter( Q(equip_user=None) | Q(equip_user=uname_obj)) # 010 elif f_id == None and f_type != '0': equip_list = Equip.objects.filter(Q(equip_user=None, equip_type=f_type) | Q( equip_user=uname_obj, equip_type=f_type)) # 100 elif f_id != None and f_type == '0': equip_list = Equip.objects.filter( Q(equip_user=None, equip_id__contains=f_id) | Q(equip_user=uname_obj, equip_id__contains=f_id)) # 110 elif f_id != None and f_type != '0': equip_list = Equip.objects.filter(Q(equip_user=None, equip_id__contains=f_id, equip_type=f_type) | Q( equip_user=uname_obj, equip_id__contains=f_id, equip_type=f_type)) nums = equip_list.count() equip_list = equip_list[(10*(page-1)):(page*10)] for i in equip_list: equip_ids.append(i.equip_id) equip_ids = ','.join(equip_ids) return render(request, 'backstageNet/equipallot/allot_equip_list.html', context={'equip_list': equip_list, 'equip_ids': equip_ids, "nums": nums}) # 进入对应的用户分配设备页面 def allot(request): if request.method == 'GET': # current_user = request.session.get('username') current_user = MyUser.objects.get(username="admin") print("current_user:", current_user) # current_user = MyUser.objects.get(username=current_user) uname = request.GET.get('uname') uname_obj = MyUser.objects.get(username=uname) print("uname=", uname) page = request.GET.get("currpage") f_name = request.GET.get("f_name") nums = Equip.objects.filter( Q(equip_user=None) | Q(equip_user=uname_obj)).count() print('nums:', nums) return render(request, 'backstageNet/equipallot/allot.html', context={"nums": nums, "uname": uname,"currpage":page,"f_name":f_name}) if request.method == 'POST': # current_user = request.session.get('username') # print("current_user:", current_user) current_user = MyUser.objects.get(username="admin") uname = request.POST.get('uname') uname_obj = MyUser.objects.get(username=uname) print("uname=", uname) # 获得当前页的全部id：equip_ids equip_ids = request.POST.get('equip_ids') # print("equip_ids=", equip_ids) equip_ids = equip_ids.split(',') button = request.POST.get('submit') if button == "fenpei": # 获得选中的复选框: check_box_list = request.POST.getlist('check_box_list') print('check_box_list:', check_box_list) no_check_box_list = [] for i in equip_ids: if i in check_box_list: pass else: no_check_box_list.append(i) # 分配复选框选中的设备： for checked_id in check_box_list: choose_equip = Equip.objects.get(equip_id=checked_id) # 如果选中的设备没有用户：即选中的设备 设备用户为None： if choose_equip.equip_user == None: # 只有当选中的设备id对应的设备 没有用户时才分配 choose_equip.equip_user = uname_obj choose_equip.save() # 保存到数据库 # 取消分配:给当前用户取消已分配的设备； for a in no_check_box_list: choose_equip = Equip.objects.get(equip_id=a) choose_equip.equip_user = None # 把设备的设备用户改为None，表示取消分配 choose_equip.save() equip_list = Equip.objects.filter( Q(equip_user=None) | Q(equip_user=uname_obj)) user_equip_list = Equip.objects.filter(equip_user=uname_obj) equip_type_list = [] for i in user_equip_list: equip_type_list.append(i.equip_type.type_id) print('abcdef',equip_type_list) groups = Group.objects.filter(user=uname_obj) nums = equip_list.count() return render(request, 'backstageNet/equipallot/allot.html', context={'equip_list': equip_list, 'nums': nums, "uname": uname, 'allot_note': '设备分配成功！'})