adding plan and target edit

rowers/views.py

@@ -27,6 +27,9 @@ import codecs
 import isodate
 
 from django.shortcuts import render
+
+from django.views.generic.edit import UpdateView
+
 from django.http import (
     HttpResponse, HttpResponseRedirect,
     HttpResponseForbidden, HttpResponseNotAllowed,
@@ -125,6 +128,9 @@ from rowers.tasks_standalone import addcomment2
 from django.contrib import messages
 from async_messages import messages as a_messages
 
+from django.contrib.admin.widgets import AdminDateWidget,AdminTimeWidget,AdminSplitDateTime
+
+
 import requests
 import json
 from rest_framework.renderers import JSONRenderer
@@ -14170,6 +14176,41 @@ def virtualevent_submit_result_view(request,id=0):
                       'w_form':w_form,
                       })
 
+@user_passes_test(hasplannedsessions,login_url="/", redirect_field_name=None)
+def rower_delete_trainingtarget(request,id=0):
+    try:
+        target = TrainingTarget.objects.get(id=id)
+    except TrainingPlan.DoesNotExist:
+        raise Http404("Training Plan Does Not Exist")
+        
+    if checkaccessuser(request.user,target.rower):
+        target.delete()
+        messages.info(request,"We have deleted the training plan")
+    else:
+        raise PermissionDenied("Access denied")
+
+    url = reverse(rower_create_trainingplan)
+
+    return HttpResponseRedirect(url)
+    
+
+@user_passes_test(hasplannedsessions,login_url="/", redirect_field_name=None)
+def rower_delete_trainingplan(request,id=0):
+    try:
+        plan = TrainingPlan.objects.get(id=id)
+    except TrainingPlan.DoesNotExist:
+        raise Http404("Training Plan Does Not Exist")
+        
+    if checkaccessuser(request.user,plan.rower):
+        plan.delete()
+        messages.info(request,"We have deleted the training plan")
+    else:
+        raise PermissionDenied("Access denied")
+
+    url = reverse(rower_create_trainingplan)
+
+    return HttpResponseRedirect(url)
+
 @user_passes_test(hasplannedsessions,login_url="/", redirect_field_name=None)
 def rower_create_trainingplan(request,id=0):
 
@@ -14228,3 +14269,43 @@ def rower_create_trainingplan(request,id=0):
     
                     
         
+class TrainingPlanUpdate(UpdateView):
+    model = TrainingPlan
+    template_name = 'trainingplan_edit.html'
+    form_class = TrainingPlanForm
+
+    def get_success_url(self):
+        return reverse(rower_create_trainingplan)
+
+    def form_valid(self, form):
+        form.instance.user = self.request.user
+        form.instance.post_date = datetime.datetime.now()
+        plan = form.save()
+        return super(TrainingPlanUpdate, self).form_valid(form)
+
+    def get_object(self, *args, **kwargs):
+        obj = super(TrainingPlanUpdate, self).get_object(*args, **kwargs)
+        if not checkaccessuser(self.request.user,obj.rower):
+            raise Http404
+        return obj
+
+class TrainingTargetUpdate(UpdateView):
+    model = TrainingTarget
+    template_name = 'trainingplan_edit.html'
+    form_class = TrainingTargetForm
+
+    def get_success_url(self):
+        return reverse(rower_create_trainingplan)
+
+    def form_valid(self, form):
+        form.instance.user = self.request.user
+        form.instance.post_date = datetime.datetime.now()
+        plan = form.save()
+        return super(TrainingTargetUpdate, self).form_valid(form)
+
+    def get_object(self, *args, **kwargs):
+        obj = super(TrainingTargetUpdate, self).get_object(*args, **kwargs)
+        if not checkaccessuser(self.request.user,obj.rower):
+            raise Http404
+        return obj
+