def addpost(request):
if request.POST:
form = BlogPostForm(request.POST)
if form.is_valid():
form.save(request.user)
return HttpResponseRedirect('/')
else:
form = BlogPostForm()
return render(request,
'addpost.html',
{'form': form})
def edit(request, pk_id):
post = get_object_or_404(Post.objects.select_related(), pk=pk_id)
if request.user.is_staff or request.user.id == post.author.id:
if request.POST:
postform = BlogPostForm(request.POST)
if postform.is_valid():
postform.save(request.user, post)
return HttpResponseRedirect('/%s' % post.pk_id)
else:
postform = BlogPostForm({'title': post.title,
'body': post.body})
return render(request, 'edit.html', {'pk_id': pk_id, 'form': postform})
else:
messages.error(request, _('You can not edit this post, '
'because you are not the author '
'of it.'))
