|
|
@@ -1,5 +1,5 @@
|
|
|
from django.shortcuts import render, redirect, get_object_or_404
|
|
|
-from .forms import MultipleShiftForm, ShiftForm, EventForm
|
|
|
+from .forms import MultipleShiftForm, ShiftForm, EventForm, HelperForm
|
|
|
from .models import Shift, Employee, Location, Event, Helper
|
|
|
from django.utils.timezone import datetime, timedelta
|
|
|
from django.utils import timezone
|
|
|
@@ -229,6 +229,29 @@ def public(request):
|
|
|
return render(request, 'main/public.html', context)
|
|
|
|
|
|
|
|
|
+def create_helper(request):
|
|
|
+ if request.method == 'POST':
|
|
|
+ form = HelperForm(request.POST)
|
|
|
+ if form.is_valid():
|
|
|
+ form.save()
|
|
|
+ return redirect('current_week_shifts') # Redirect to a relevant page after saving
|
|
|
+ else:
|
|
|
+ form = HelperForm()
|
|
|
+ return render(request, 'main/create_helper.html', {'form': form})
|
|
|
+
|
|
|
+
|
|
|
+def edit_helper(request, pk):
|
|
|
+ helper = get_object_or_404(Helper, pk=pk)
|
|
|
+ if request.method == 'POST':
|
|
|
+ form = HelperForm(request.POST, instance=helper)
|
|
|
+ if form.is_valid():
|
|
|
+ form.save()
|
|
|
+ return redirect('current_week_shifts')
|
|
|
+ else:
|
|
|
+ form = HelperForm(instance=helper)
|
|
|
+ return render(request, 'main/edit_helper.html', {'form': form, 'helper': helper})
|
|
|
+
|
|
|
+
|
|
|
def edit_shift(request, pk):
|
|
|
shift = get_object_or_404(Shift, pk=pk)
|
|
|
if request.method == 'POST':
|
|
|
@@ -261,6 +284,14 @@ def delete_shift(request, pk):
|
|
|
return render(request, 'main/delete_shift.html', {'shift': shift})
|
|
|
|
|
|
|
|
|
+def delete_helper(request, pk):
|
|
|
+ helper = get_object_or_404(Helper, pk=pk)
|
|
|
+ if request.method == 'POST':
|
|
|
+ helper.delete()
|
|
|
+ return redirect('current_week_shifts')
|
|
|
+ return render(request, 'main/delete_helper.html', {'helper': helper})
|
|
|
+
|
|
|
+
|
|
|
def delete_event(request, pk):
|
|
|
event = get_object_or_404(Event, pk=pk)
|
|
|
if request.method == 'POST':
|