Python会议室预定

from django.shortcuts import render, redirect, HttpResponse # Create your views here. from .models import * import datetime import json from django.http import JsonResponse from django.contrib import auth def index(request): current_date = datetime.datetime.now().date() # 取到当前时间 book_date = request.GET.get("book_date", current_date.strftime("%Y-%m-%d")) # 取到预定时间 book_date = datetime.datetime.strptime(book_date, "%Y-%m-%d") # 格式化时间 time_choices = Book.time_choices # 取到model里定义的时间 room_list = Room.objects.all() # 取到所有房间 book_list = Book.objects.filter(date=book_date) # 筛选当前时间的所有预定信息 html = "" for room in room_list: # 取到每一个的房间信息 s = '<tr><td>{0}({1})</td>'.format(room.caption, room.num) for item in time_choices: # 取到每一个可以预定的时间段 flag = False for book in book_list: # 取到每一个预定信息book if book.room.caption == room.caption and book.time_id == item[0]: # 如果每一个预定信息的房间名称和时间段匹配成功 flag = True break if flag: # 如果预定信息存在 if request.user.pk != book.user.pk: # 如果当前登录用户不是预定用户 s += '<td class="active_other item" room_id="{0}" time_id="{1}">{2}</td>'.format(room.pk, item[0], book.user.username) else: # 如果当前登录用户就是预定用户 s += '<td class="active item" room_id="{0}" time_id="{1}">{2}</td>'.format(room.pk, item[0], book.user.username) else: # 如果当前房间当前时段没人预定,显示空白 s += '<td class="item" room_id="{0}" time_id="{1}"></td>'.format(room.pk, item[0]) s += "</tr>" html += s # 拼出tr单元格 return render(request, 'index.html', locals()) def book(request): print(request.POST) response = {'status': True, 'msg': None, 'data': None} try: choice_date = request.POST.get('date') # 取出从前端get的数据 choice_date = datetime.datetime.strptime(choice_date, '%Y-%m-%d').date() # 转换日期成指定格式 post_data = json.loads(request.POST.get('data')) # 取出ajax传来的数据,并反序列化 # 优化 既在添加字典又在取消字典中 == 不添加 也不取消 for room_id, time_list in post_data['DEL'].items(): # 循环字典中每一个删除信息 if room_id not in post_data['ADD']: # 如果循环到的删除房间 没在 添加房间的字典中 continue # 结束当前循环进入下一个循环 for time_id in list(time_list): # 循环删除字典中每个房间的时间段 if time_id in post_data['ADD'][room_id]: # 如果这个时间段在当前循环房间 添加预订 的字典中 post_data['ADD'][room_id].remove(time_id) # 从添加预订的字典中移除 这个时间段 post_data['DEL'][room_id].remove(time_id) # 从取消预订的字典中移除 这个时间段 # 增加预定 book_obj_list = [] for room_id, time_list in post_data['ADD'].items(): for time_id in time_list: obj = Book(room_id=room_id, time_id=time_id, user_id=request.user.pk, date=choice_date) book_obj_list.append(obj) # 添加到预订房间的 列表中 Book.objects.bulk_create(book_obj_list) # 批量创建到数据库 # 删除会议室预定信息 print(post_data['DEL']) from django.db.models import Q remove_booking = Q() for room_id, time_id_list in post_data['DEL'].items(): for time_id in time_id_list: temp = Q() temp.connector = 'AND' temp.children.append(('user_id', request.user.pk,)) temp.children.append(('date', choice_date)) temp.children.append(('room_id', room_id,)) temp.children.append(('time_id', time_id,)) remove_booking.add(temp, 'OR') if remove_booking: Book.objects.filter(remove_booking).delete() except Exception as e: response['status'] = False response['msg'] = str(e) return JsonResponse(response) def login(request): if request.method == "POST": user = request.POST.get("user") pwd = request.POST.get("pwd") user = auth.authenticate(username=user, password=pwd) # 用auth 模块对当前登录用户进行验证 if user: auth.login(request, user) return redirect("/index/") return render(request, "login.html")