Hotel/hotel/rooms/views.py

63 lines
2.1 KiB
Python

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 *
import re
# 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']
template = loader.get_template('reservation.html')
error = ""
reservation = Reservation.objects.filter(room_number=request.POST['room'], begin_date__gte=begin_date, end_date__lte=end_date)
if reservation is not None:
status = "Pokój " + str(room_number) + " jest już zarezerwowany"
return HttpResponse(template.render({'status': status}, request))
else:
if request.method == "POST":
client_id = request.POST['id_number']
if check(client_id):
instance = Client.objects.get(id_number=client_id)
client = ClientForm(request.POST, instance=instance)
if not Client.objects.get(id_number=client_id):
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"
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)