Add support for editing posts

forums/urls.py

@@ -10,6 +10,8 @@ urlpatterns = patterns('forums',
     url(r'^sticky/(?P<topicID>.*)', 'views.stickyTopic'),
     url(r'^reply/(?P<topicID>.*)', 'views.reply'),
     url(r'^reply$', 'views.reply'),
+    url(r'^edit/(?P<postID>.*)', 'views.editPost'),
+    url(r'^edit$', 'views.editPost'),
     url(r'^post/(?P<id>.*)$', 'views.post'),
     url(r'^preview$', 'views.preview'),
 )

forums/views.py

@@ -7,6 +7,7 @@ from django.shortcuts import render_to_response
 from django.http import HttpResponseRedirect
 from django.template import RequestContext
 from django.core.urlresolvers import reverse
+from django.core.exceptions import PermissionDenied
 from notification import models as notification
 
 def index(request):
@@ -44,7 +45,7 @@ def reply(request, topicID=None):
             notification.send([reply.parent.user], "forum_reply", {"reply": reply, 'notice_url': reverse('forums.views.post', kwargs={'id':reply.id}), 'notice_description': reply.topic().title})
         messages.info(request, "Reply successful")
         return HttpResponseRedirect(reverse('forums.views.post', kwargs={"id":reply.id}))
-    return render_to_response('forums/reply.html', {"post":parentPost, "form":form}, context_instance = RequestContext(request))
+    return render_to_response('forums/reply.html', {"parent":parentPost, "form":form}, context_instance = RequestContext(request))
 
 @login_required
 def newTopic(request, forumID=None):
@@ -83,6 +84,25 @@ def newTopic(request, forumID=None):
         return HttpResponseRedirect(reverse('forums.views.post', kwargs={'id': reply.id}))
     return render_to_response('forums/newTopic.html', {"forum":parentForum, "replyForm":replyForm, "topicForm": topicForm}, context_instance = RequestContext(request))
 
+@login_required
+def editPost(request, postID=None):
+    post = models.Post.objects.get(id__exact=postID)
+    if post.user != request.user and not request.user.has_perm('edit_posts'):
+      raise PermissionDenied
+    if request.method == 'POST':
+        form = forms.ReplyForm(request.POST)
+    else:
+        form = forms.ReplyForm(instance=post)
+    
+    if form.is_valid():
+        post.body = form.cleaned_data['body']
+        post.save()
+        messages.info(request, "Post updated.")
+        return HttpResponseRedirect(reverse('forums.views.post',
+          kwargs={"id":post.id}))
+    return render_to_response('forums/edit.html', {"post": post, "parent":
+      post.parent, "form": form}, context_instance = RequestContext(request))
+
 @permission_required('forums.delete_topic')
 def deleteTopic(request, topicID):
     topic = models.Topic.objects.get(id__exact=topicID)