from django.http import HttpResponse from django.template import loader from django.shortcuts import get_object_or_404, render from .models import * from .forms import * from django.db.models import Q import re import datetime # Create your views here. def index(request): rooms = Room.objects.all() rooms = rooms.order_by('room_number') template = loader.get_template('index.html') return HttpResponse(template.render({'rooms': rooms}, request)) def reservation(request): room_number, begin_date, end_date = request.POST['room'], request.POST['begindate'], request.POST['enddate'] try: begin_datetime = datetime.datetime.strptime(begin_date, "%Y-%m-%d") end_datetime = datetime.datetime.strptime(end_date, "%Y-%m-%d") except: return HttpResponse(loader.get_template('error.html').render({'error': "Błąd przy datach"}, request)) if begin_datetime == '' or end_datetime == '' or end_datetime < begin_datetime or begin_datetime < datetime.datetime.today() or end_datetime < datetime.datetime.today(): return HttpResponse(loader.get_template('error.html').render({'error': "Błąd przy datach"}, request)) template = loader.get_template('reservation.html') error = "" reservation = Reservation.objects.filter(room_number=room_number).filter(Q(begin_date__range=(begin_datetime, end_datetime)) | Q(end_date__range=(begin_datetime, end_datetime))).first() if not reservation is None: status = "Pokój " + str(room_number) + " jest już zarezerwowany" return HttpResponse(template.render({'status': status}, request)) else: if request.method == "POST": try: client_id = request.POST['id_number'] if check(client_id): if Client.objects.filter(id_number=client_id).first() is None: client = ClientForm(request.POST) client.save() r = Reservation() r.room_number = get_object_or_404(Room, room_number=room_number) r.client_id = get_object_or_404(Client, id_number=client_id) r.begin_date = begin_date r.end_date = end_date r.save() status = "Pokój " + str(room_number) + " został pomyślnie zarejestrowany" return HttpResponse(template.render({'status': status}, request)) error = "Numer dowodu nieprawidłowy" except: pass form = ClientForm() template = loader.get_template('form.html') return HttpResponse(template.render({'form': form, 'error': error, 'room_number': room_number, 'begin_date': begin_date, 'end_date': end_date}, request)) def check(numer): pattern = re.compile(r'[A-Z][A-Z][A-Z]\d\d\d\d\d\d') return pattern.match(numer)