forum_modules/akismet/startup.py

   1 from django.utils.translation import ugettext as _
   2 from django.http import HttpResponse, HttpResponseRedirect
   3 from django.utils import simplejson
   4 from django.utils.encoding import smart_str
   5 from django.shortcuts import render_to_response
   6 from forum.modules import decorate
   7 from forum import views
   8 from lib.akismet import Akismet
   9 from forum.settings import APP_URL, OSQA_VERSION
  10 from settings import WORDPRESS_API_KEY, REP_FOR_NO_SPAM_CHECK
  11 from forum.models.user import User
  12
  13 import settings
  14
  15 def can_bypass_spam_check(user):
  16     return user.is_authenticated() and (user.is_superuser or user.is_staff or cmp(user.reputation, REP_FOR_NO_SPAM_CHECK
  17                                                                                   ) > 0)
  18
  19
  20 def check_spam(param, comment_type):
  21     def wrapper(origin, request, *args, **kwargs):
  22         if request.POST and request.POST.get(param, None) and WORDPRESS_API_KEY and (not can_bypass_spam_check(
  23                 request.user)):
  24             comment = smart_str(request.POST[param])
  25
  26             data = {
  27             "user_ip":request.META["REMOTE_ADDR"],
  28             "user_agent":request.environ['HTTP_USER_AGENT'],
  29             "comment_type": comment_type,
  30             "comment":comment
  31             }
  32
  33             if request.user.is_authenticated():
  34                 data.update({
  35                 "comment_author":smart_str(request.user.username),
  36                 "comment_author_email":request.user.email,
  37                 "comment_author_url":request.user.website,
  38                 })
  39
  40             api = Akismet(settings.WORDPRESS_API_KEY, APP_URL, "OSQA/%s" % OSQA_VERSION)
  41             if api.comment_check(comment, data):
  42                 if request.is_ajax():
  43                     response = {
  44                     'success': False,
  45                     'error_message': _("Sorry, but akismet thinks your %s is spam.") % comment_type
  46                     }
  47                     return HttpResponse(simplejson.dumps(response), mimetype="application/json")
  48                 else:
  49                     return render_to_response('modules/akismet/foundspam.html', {
  50                     'action_name': comment_type
  51                     })
  52
  53         return origin(request, *args, **kwargs)
  54
  55     return wrapper
  56
  57
  58 decorate(views.writers.ask)(check_spam('text', _('question')))
  59 decorate(views.writers.answer)(check_spam('text', _('answer')))
  60 decorate(views.commands.comment)(check_spam('comment', _('comment')))
  61
  62