forum/views/writers.py

   1 # encoding:utf-8
   2 import os.path
   3 import time, datetime, random
   4 import logging
   5 from django.core.files.storage import FileSystemStorage
   6 from django.shortcuts import render_to_response, get_object_or_404
   7 from django.contrib.auth.decorators import login_required
   8 from django.http import HttpResponseRedirect, HttpResponse, HttpResponseForbidden, Http404
   9 from django.template import RequestContext
  10 from django.utils.html import *
  11 from django.utils import simplejson
  12 from django.utils.translation import ugettext as _
  13 from django.core.urlresolvers import reverse
  14 from django.core.exceptions import PermissionDenied
  15
  16 from forum.forms import *
  17 from forum.models import *
  18 from forum.const import *
  19 from forum.utils.forms import get_next_url
  20 from forum.views.readers import _get_tags_cache_json
  21 from forum.views.commands import SpamNotAllowedException
  22
  23 # used in index page
  24 INDEX_PAGE_SIZE = 20
  25 INDEX_AWARD_SIZE = 15
  26 INDEX_TAGS_SIZE = 100
  27 # used in tags list
  28 DEFAULT_PAGE_SIZE = 60
  29 # used in questions
  30 QUESTIONS_PAGE_SIZE = 10
  31 # used in answers
  32 ANSWERS_PAGE_SIZE = 10
  33
  34 def upload(request):#ajax upload file to a question or answer
  35     class FileTypeNotAllow(Exception):
  36         pass
  37     class FileSizeNotAllow(Exception):
  38         pass
  39     class UploadPermissionNotAuthorized(Exception):
  40         pass
  41
  42     #<result><msg><![CDATA[%s]]></msg><error><![CDATA[%s]]></error><file_url>%s</file_url></result>
  43     xml_template = "<result><msg><![CDATA[%s]]></msg><error><![CDATA[%s]]></error><file_url>%s</file_url></result>"
  44
  45     try:
  46         f = request.FILES['file-upload']
  47         # check upload permission
  48         if not request.user.can_upload_files():
  49             raise UploadPermissionNotAuthorized()
  50
  51         # check file type
  52         file_name_suffix = os.path.splitext(f.name)[1].lower()
  53
  54         if not file_name_suffix in ('.jpg', '.jpeg', '.gif', '.png', '.bmp', '.tiff', '.ico'):
  55             raise FileTypeNotAllow()
  56
  57         storage = FileSystemStorage(str(settings.UPFILES_FOLDER), str(settings.UPFILES_ALIAS))
  58         new_file_name = storage.save(f.name, f)
  59         # check file size
  60         # byte
  61         size = storage.size(new_file_name)
  62
  63         if size > float(settings.ALLOW_MAX_FILE_SIZE) * 1024 * 1024:
  64             storage.delete(new_file_name)
  65             raise FileSizeNotAllow()
  66
  67         result = xml_template % ('Good', '', str(settings.UPFILES_ALIAS) + new_file_name)
  68     except UploadPermissionNotAuthorized:
  69         result = xml_template % ('', _('uploading images is limited to users with >60 reputation points'), '')
  70     except FileTypeNotAllow:
  71         result = xml_template % ('', _("allowed file types are 'jpg', 'jpeg', 'gif', 'bmp', 'png', 'tiff'"), '')
  72     except FileSizeNotAllow:
  73         result = xml_template % ('', _("maximum upload file size is %sM") % settings.ALLOW_MAX_FILE_SIZE, '')
  74     except Exception, e:
  75         result = xml_template % ('', _('Error uploading file. Please contact the site administrator. Thank you. %s' % e), '')
  76
  77     return HttpResponse(result, mimetype="application/xml")
  78
  79
  80 def _create_post(request, post_cls, form, parent=None):
  81     post = post_cls()
  82
  83     if parent is not None:
  84         post.parent = parent
  85
  86     revision_data = dict(summary=_('Initial revision'), body=form.cleaned_data['text'])
  87
  88     if form.cleaned_data.get('title', None):
  89         revision_data['title'] = strip_tags(form.cleaned_data['title'].strip())
  90
  91     if form.cleaned_data.get('tags', None):
  92         revision_data['tagnames'] = form.cleaned_data['tags'].strip()
  93
  94     post.create_revision(request.user, **revision_data)
  95
  96     if form.cleaned_data['wiki']:
  97         post.wikify()
  98
  99     return HttpResponseRedirect(post.get_absolute_url())
 100
 101
 102
 103 def ask(request):
 104     if request.method == "POST" and "text" in request.POST:
 105         form = AskForm(request.POST)
 106         if form.is_valid():
 107             if request.user.is_authenticated():
 108                 data = {
 109                     "user_ip":request.META["REMOTE_ADDR"],
 110                     "user_agent":request.environ['HTTP_USER_AGENT'],
 111                     "comment_author":request.user.real_name,
 112                     "comment_author_email":request.user.email,
 113                     "comment_author_url":request.user.website,
 114                     "comment":request.POST['text']
 115                 }
 116                 if Node.isSpam(request.POST['text'], data):
 117                     raise SpamNotAllowedException("question")
 118
 119                 return _create_post(request, Question, form)
 120             else:
 121                 return HttpResponseRedirect(reverse('auth_action_signin', kwargs={'action': 'newquestion'}))
 122     elif request.method == "POST" and "go" in request.POST:
 123         form = AskForm({'title': request.POST['q']})
 124     else:
 125         form = AskForm()
 126
 127     #tags = _get_tags_cache_json()
 128     return render_to_response('ask.html', {
 129         'form' : form,
 130         #'tags' : tags,
 131         'email_validation_faq_url':reverse('faq') + '#validate',
 132         }, context_instance=RequestContext(request))
 133
 134 @login_required
 135 def edit_question(request, id):
 136     question = get_object_or_404(Question, id=id)
 137     if question.deleted and not request.user.can_view_deleted_post(question):
 138         raise Http404
 139     if request.user.can_edit_post(question):
 140         return _edit_question(request, question)
 141     elif request.user.can_retag_questions():
 142         return _retag_question(request, question)
 143     else:
 144         raise Http404
 145
 146 def _retag_question(request, question):
 147     if request.method == 'POST':
 148         form = RetagQuestionForm(question, request.POST)
 149         if form.is_valid():
 150             if form.has_changed():
 151                 active = question.active_revision
 152
 153                 question.create_revision(
 154                     request.user,
 155                     summary          = _('Retag'),
 156                     title            = active.title,
 157                     tagnames         = form.cleaned_data['tags'],
 158                     body             = active.body,
 159                 )
 160
 161             return HttpResponseRedirect(question.get_absolute_url())
 162     else:
 163         form = RetagQuestionForm(question)
 164     return render_to_response('question_retag.html', {
 165         'question': question,
 166         'form' : form,
 167         #'tags' : _get_tags_cache_json(),
 168     }, context_instance=RequestContext(request))
 169
 170 def _edit_question(request, question):
 171     if request.method == 'POST':
 172         revision_form = RevisionForm(question, data=request.POST)
 173         revision_form.is_valid()
 174         revision = question.revisions.get(revision=revision_form.cleaned_data['revision'])
 175
 176         if 'select_revision' in request.POST:
 177             form = EditQuestionForm(question, revision)
 178         else:
 179             form = EditQuestionForm(question, revision, data=request.POST)
 180
 181         if not 'select_revision' in request.POST and form.is_valid():
 182             if form.has_changed():
 183                 question.create_revision(
 184                     request.user,
 185                     summary          = form.cleaned_data['summary'],
 186                     title            = strip_tags(form.cleaned_data['title'].strip()),
 187                     tagnames         = form.cleaned_data['tags'].strip(),
 188                     body             = form.cleaned_data['text'],
 189                 )
 190
 191                 if form.cleaned_data.get('wiki', False):
 192                     question.wikify()
 193
 194             else:
 195                 if not revision == question.active_revision:
 196                     question.activate_revision(request.user, revision)
 197
 198             return HttpResponseRedirect(question.get_absolute_url())
 199     else:
 200         revision_form = RevisionForm(question)
 201         form = EditQuestionForm(question)
 202
 203     return render_to_response('question_edit.html', {
 204         'question': question,
 205         'revision_form': revision_form,
 206         'form' : form,
 207         #'tags' : _get_tags_cache_json()
 208     }, context_instance=RequestContext(request))
 209
 210 @login_required
 211 def edit_answer(request, id):
 212     answer = get_object_or_404(Answer, id=id)
 213     if answer.deleted and not request.user.can_view_deleted_post(answer):
 214         raise Http404
 215     elif not request.user.can_edit_post(answer):
 216         raise Http404
 217
 218     if request.method == "POST":
 219         revision_form = RevisionForm(answer, data=request.POST)
 220         revision_form.is_valid()
 221         revision = answer.revisions.get(revision=revision_form.cleaned_data['revision'])
 222
 223         if 'select_revision' in request.POST:
 224             form = EditAnswerForm(answer, revision)
 225         else:
 226             form = EditAnswerForm(answer, revision, data=request.POST)
 227
 228         if not 'select_revision' in request.POST and form.is_valid():
 229             if form.has_changed():
 230                 answer.create_revision(
 231                     request.user,
 232                     summary          = form.cleaned_data['summary'],
 233                     body             = form.cleaned_data['text'],
 234                 )
 235
 236                 if form.cleaned_data.get('wiki', False):
 237                     answer.wikify()
 238
 239             else:
 240                 if not revision == answer.active_revision:
 241                     answer.activate_revision(request.user, revision)
 242
 243             return HttpResponseRedirect(answer.get_absolute_url())
 244
 245     else:
 246         revision_form = RevisionForm(answer)
 247         form = EditAnswerForm(answer)
 248     return render_to_response('answer_edit.html', {
 249                               'answer': answer,
 250                               'revision_form': revision_form,
 251                               'form': form,
 252                               }, context_instance=RequestContext(request))
 253
 254 def answer(request, id):
 255     question = get_object_or_404(Question, id=id)
 256     if request.method == "POST":
 257         form = AnswerForm(question, request.POST)
 258         if form.is_valid():
 259             if request.user.is_authenticated():
 260                 data = {
 261                     "user_ip":request.META["REMOTE_ADDR"],
 262                     "user_agent":request.environ['HTTP_USER_AGENT'],
 263                     "comment_author":request.user.real_name,
 264                     "comment_author_email":request.user.email,
 265                     "comment_author_url":request.user.website,
 266                     "comment":request.POST['text']
 267                 }
 268                 if Node.isSpam(request.POST['text'], data):
 269                     raise SpamNotAllowedException("answer")
 270
 271                 return _create_post(request, Answer, form, question)
 272             else:
 273                 return HttpResponseRedirect(reverse('auth_action_signin', kwargs={'action': 'newquestion'}))
 274
 275     return HttpResponseRedirect(question.get_absolute_url())
 276