class WorkgroupPostAdd(WorkgroupBase): template_name = 'workgroup/addpost.html' workgroup = None form = None def get(self,request,*args,**kwargs): from .forms import PostForm self.get_initial_context(request,*args,**kwargs) self.form = PostForm() self.context['form'] = self.form return render(request,self.template_name,self.context) def post(self,request,*args,**kwargs): from .forms import PostForm self.get_initial_context(request,*args,**kwargs) self.form = PostForm(request.POST) if (self.form.is_valid()): post = self.form.save(commit=False) post.user = request.user post.workgroup = self.workgroup post.save() return redirect('workgroup:home',workgroup_id=kwargs['workgroup_id'])
def post(self,request,*args,**kwargs): self.get_initial_context(request,*args,**kwargs) button_action = request.POST['button_action'] if button_action == 'save' : if (self.get_permssion(request)): form = PostForm(request.POST,instance=self.post) form.save() messages.info(request, "Post has been updated") else : messages.warning(request, "You don't have permission to edit this post") return redirect('workgroup:home',workgroup_id=kwargs['workgroup_id']) else : self.post.delete() messages.warning(request, "Post has been deleted") return redirect('workgroup:home',workgroup_id=kwargs['workgroup_id'])