forum/views/users.py

   1 from django.contrib.auth.decorators import login_required\r
   2 from forum.models import User\r
   3 from django.db.models import Q, Count\r
   4 from django.core.paginator import Paginator, EmptyPage, InvalidPage\r
   5 from django.template.defaultfilters import slugify\r
   6 from django.contrib.contenttypes.models import ContentType\r
   7 from django.core.urlresolvers import reverse\r
   8 from django.shortcuts import render_to_response, get_object_or_404\r
   9 from django.template import RequestContext\r
  10 from django.http import HttpResponse, HttpResponseForbidden, HttpResponseRedirect, Http404\r
  11 from django.utils.translation import ugettext as _\r
  12 from django.utils.http import urlquote_plus\r
  13 from django.utils.html import strip_tags\r
  14 from django.utils import simplejson\r
  15 from django.core.urlresolvers import reverse\r
  16 from forum.forms import *\r
  17 from forum.utils.html import sanitize_html\r
  18 from datetime import datetime, date\r
  19 import decorators\r
  20 from forum.actions import EditProfileAction, FavoriteAction, BonusRepAction, SuspendAction\r
  21 \r
  22 import time\r
  23 \r
  24 USERS_PAGE_SIZE = 35# refactor - move to some constants file\r
  25 \r
  26 def users(request):\r
  27     is_paginated = True\r
  28     sortby = request.GET.get('sort', 'reputation')\r
  29     suser = request.REQUEST.get('q', "")\r
  30     try:\r
  31         page = int(request.GET.get('page', '1'))\r
  32     except ValueError:\r
  33         page = 1\r
  34 \r
  35     if suser == "":\r
  36         if sortby == "newest":\r
  37             objects_list = Paginator(User.objects.all().order_by('-date_joined'), USERS_PAGE_SIZE)\r
  38         elif sortby == "last":\r
  39             objects_list = Paginator(User.objects.all().order_by('date_joined'), USERS_PAGE_SIZE)\r
  40         elif sortby == "user":\r
  41             objects_list = Paginator(User.objects.all().order_by('username'), USERS_PAGE_SIZE)\r
  42         # default\r
  43         else:\r
  44             objects_list = Paginator(User.objects.all().order_by('-is_active', '-reputation'), USERS_PAGE_SIZE)\r
  45         base_url = reverse('users') + '?sort=%s&' % sortby\r
  46     else:\r
  47         sortby = "reputation"\r
  48         objects_list = Paginator(User.objects.filter(username__icontains=suser).order_by('-reputation'), USERS_PAGE_SIZE\r
  49                                  )\r
  50         base_url = reverse('users') + '?name=%s&sort=%s&' % (suser, sortby)\r
  51 \r
  52     try:\r
  53         users = objects_list.page(page)\r
  54     except (EmptyPage, InvalidPage):\r
  55         users = objects_list.page(objects_list.num_pages)\r
  56 \r
  57     return render_to_response('users/users.html', {\r
  58     "users" : users,\r
  59     "suser" : suser,\r
  60     "keywords" : suser,\r
  61     "tab_id" : sortby,\r
  62     "context" : {\r
  63     'is_paginated' : is_paginated,\r
  64     'pages': objects_list.num_pages,\r
  65     'page': page,\r
  66     'has_previous': users.has_previous(),\r
  67     'has_next': users.has_next(),\r
  68     'previous': users.previous_page_number(),\r
  69     'next': users.next_page_number(),\r
  70     'base_url' : base_url\r
  71     }\r
  72 \r
  73     }, context_instance=RequestContext(request))\r
  74 \r
  75 def set_new_email(user, new_email, nomessage=False):\r
  76     if new_email != user.email:\r
  77         user.email = new_email\r
  78         user.email_isvalid = False\r
  79         user.save()\r
  80     #if settings.EMAIL_VALIDATION == 'on':\r
  81     #    send_new_email_key(user,nomessage=nomessage)\r
  82 \r
  83 @login_required\r
  84 def edit_user(request, id):\r
  85     user = get_object_or_404(User, id=id)\r
  86     if not (request.user.is_superuser or request.user == user):\r
  87         return HttpResponseForbidden()\r
  88     if request.method == "POST":\r
  89         form = EditUserForm(user, request.POST)\r
  90         if form.is_valid():\r
  91             new_email = sanitize_html(form.cleaned_data['email'])\r
  92 \r
  93             set_new_email(user, new_email)\r
  94 \r
  95             if settings.EDITABLE_SCREEN_NAME:\r
  96                 user.username = sanitize_html(form.cleaned_data['username'])\r
  97             user.real_name = sanitize_html(form.cleaned_data['realname'])\r
  98             user.website = sanitize_html(form.cleaned_data['website'])\r
  99             user.location = sanitize_html(form.cleaned_data['city'])\r
 100             user.date_of_birth = form.cleaned_data['birthday']\r
 101             if user.date_of_birth == "None":\r
 102                 user.date_of_birth = datetime(1900, 1, 1, 0, 0)\r
 103             user.about = sanitize_html(form.cleaned_data['about'])\r
 104 \r
 105             user.save()\r
 106             EditProfileAction(user=user, ip=request.META['REMOTE_ADDR']).save()\r
 107 \r
 108             return HttpResponseRedirect(user.get_profile_url())\r
 109     else:\r
 110         form = EditUserForm(user)\r
 111     return render_to_response('users/edit.html', {\r
 112     'user': user,\r
 113     'form' : form,\r
 114     'gravatar_faq_url' : reverse('faq') + '#gravatar',\r
 115     }, context_instance=RequestContext(request))\r
 116 \r
 117 \r
 118 @login_required\r
 119 def user_powers(request, id, action, status):\r
 120     if not request.user.is_superuser:\r
 121         return HttpResponseForbidden()\r
 122 \r
 123     user = get_object_or_404(User, id=id)\r
 124     new_state = action == 'grant'\r
 125 \r
 126     if status == 'super':\r
 127         user.is_superuser = new_state\r
 128     elif status == 'staff':\r
 129         user.is_staff = new_state\r
 130     else:\r
 131         raise Http404()\r
 132 \r
 133     user.save()\r
 134     return HttpResponseRedirect(user.get_profile_url())\r
 135 \r
 136 \r
 137 @decorators.command\r
 138 def award_points(request, id):\r
 139     if (not request.POST) and request.POST.get('points', None):\r
 140         raise decorators.CommandException(_("Invalid request type"))\r
 141 \r
 142     if not request.user.is_superuser:\r
 143         raise decorators.CommandException(_("Only superusers are allowed to award reputation points"))\r
 144 \r
 145     user = get_object_or_404(User, id=id)\r
 146     points = int(request.POST['points'])\r
 147 \r
 148     extra = dict(message=request.POST.get('message', ''), awarding_user=request.user.id, value=points)\r
 149 \r
 150     BonusRepAction(user=user, extra=extra).save(data=dict(value=points))\r
 151 \r
 152     return dict(reputation=user.reputation)\r
 153 \r
 154 \r
 155 @decorators.command\r
 156 def suspend(request, id):\r
 157     user = get_object_or_404(User, id=id)\r
 158 \r
 159     if not request.POST:\r
 160         if user.is_suspended():\r
 161             suspension = user.suspension\r
 162             suspension.cancel(ip=request.META['REMOTE_ADDR'])\r
 163             return decorators.RefreshPageCommand()\r
 164         else:\r
 165             return render_to_response('users/suspend_user.html')\r
 166 \r
 167     if not request.user.is_superuser:\r
 168         raise decorators.CommandException(_("Only superusers can ban other users"))\r
 169 \r
 170     data = {\r
 171     'bantype': request.POST.get('bantype', 'indefinetly').strip(),\r
 172     'publicmsg': request.POST.get('publicmsg', _('Bad behaviour')),\r
 173     'privatemsg': request.POST.get('privatemsg', None) or request.POST.get('publicmsg', ''),\r
 174     'suspender': request.user.id\r
 175     }\r
 176 \r
 177     if data['bantype'] == 'forxdays':\r
 178         try:\r
 179             data['forxdays'] = int(request.POST['forxdays'])\r
 180         except:\r
 181             raise decorators.CommandException(_('Invalid numeric argument for the number of days.'))\r
 182 \r
 183     SuspendAction(user=user, ip=request.META['REMOTE_ADDR']).save(data=data)\r
 184 \r
 185     return decorators.RefreshPageCommand()\r
 186 \r
 187 def user_view(template, tab_name, tab_description, page_title, private=False):\r
 188     def decorator(fn):\r
 189         def decorated(request, id, slug=None):\r
 190             user = get_object_or_404(User, id=id)\r
 191             if private and not (user == request.user or request.user.is_superuser):\r
 192                 return HttpResponseForbidden()\r
 193             context = fn(request, user)\r
 194 \r
 195             rev_page_title = user.username + " - " + page_title\r
 196 \r
 197             context.update({\r
 198             "tab_name" : tab_name,\r
 199             "tab_description" : tab_description,\r
 200             "page_title" : rev_page_title,\r
 201             "can_view_private": (user == request.user) or request.user.is_superuser\r
 202             })\r
 203             return render_to_response(template, context, context_instance=RequestContext(request))\r
 204 \r
 205         return decorated\r
 206 \r
 207     return decorator\r
 208 \r
 209 \r
 210 @user_view('users/stats.html', 'stats', _('user profile'), _('user overview'))\r
 211 def user_stats(request, user):\r
 212     questions = Question.objects.filter_state(deleted=False).filter(author=user).order_by('-added_at')\r
 213     answers = Answer.objects.filter_state(deleted=False).filter(author=user).order_by('-added_at')\r
 214 \r
 215     up_votes = user.vote_up_count\r
 216     down_votes = user.vote_down_count\r
 217     votes_today = user.get_vote_count_today()\r
 218     votes_total = int(settings.MAX_VOTES_PER_DAY)\r
 219 \r
 220     user_tags = Tag.objects.filter(Q(nodes__author=user) | Q(nodes__children__author=user)) \\r
 221         .annotate(user_tag_usage_count=Count('name')).order_by('-user_tag_usage_count')\r
 222 \r
 223     awards = [(Badge.objects.get(id=b['id']), b['count']) for b in\r
 224               Badge.objects.filter(awards__user=user).values('id').annotate(count=Count('cls')).order_by('-count')]\r
 225 \r
 226     return {\r
 227     "view_user" : user,\r
 228     "questions" : questions,\r
 229     "answers" : answers,\r
 230     "up_votes" : up_votes,\r
 231     "down_votes" : down_votes,\r
 232     "total_votes": up_votes + down_votes,\r
 233     "votes_today_left": votes_total-votes_today,\r
 234     "votes_total_per_day": votes_total,\r
 235     "user_tags" : user_tags[:50],\r
 236     "awards": awards,\r
 237     "total_awards" : len(awards),\r
 238     }\r
 239 \r
 240 @user_view('users/recent.html', 'recent', _('recent user activity'), _('recent activity'))\r
 241 def user_recent(request, user):\r
 242     activities = user.actions.exclude(\r
 243             action_type__in=("voteup", "votedown", "voteupcomment", "flag", "newpage", "editpage")).order_by(\r
 244             '-action_date')[:USERS_PAGE_SIZE]\r
 245 \r
 246     return {"view_user" : user, "activities" : activities}\r
 247 \r
 248 \r
 249 @user_view('users/votes.html', 'votes', _('user vote record'), _('votes'), True)\r
 250 def user_votes(request, user):\r
 251     votes = user.votes.exclude(node__state_string__contains="(deleted").filter(\r
 252             node__node_type__in=("question", "answer")).order_by('-voted_at')[:USERS_PAGE_SIZE]\r
 253 \r
 254     return {"view_user" : user, "votes" : votes}\r
 255 \r
 256 \r
 257 @user_view('users/reputation.html', 'reputation', _('user reputation in the community'), _('user reputation'))\r
 258 def user_reputation(request, user):\r
 259     rep = list(user.reputes.order_by('date'))\r
 260     values = [r.value for r in rep]\r
 261     redux = lambda x, y: x+y\r
 262 \r
 263     graph_data = simplejson.dumps([\r
 264     (time.mktime(rep[i].date.timetuple()) * 1000, reduce(redux, values[:i], 0))\r
 265     for i in range(len(values))\r
 266     ])\r
 267 \r
 268     rep = user.reputes.filter(action__canceled=False).order_by('-date')[0:20]\r
 269 \r
 270     return {"view_user": user, "reputation": rep, "graph_data": graph_data}\r
 271 \r
 272 @user_view('users/questions.html', 'favorites', _('favorite questions'), _('favorite questions'))\r
 273 def user_favorites(request, user):\r
 274     favorites = FavoriteAction.objects.filter(canceled=False, user=user)\r
 275 \r
 276     return {"favorites" : favorites, "view_user" : user}\r
 277 \r
 278 @user_view('users/subscriptions.html', 'subscriptions', _('subscription settings'), _('subscriptions'), True)\r
 279 def user_subscriptions(request, user):\r
 280     if request.method == 'POST':\r
 281         form = SubscriptionSettingsForm(request.POST)\r
 282 \r
 283         if 'notswitch' in request.POST:\r
 284             user.subscription_settings.enable_notifications = not user.subscription_settings.enable_notifications\r
 285             user.subscription_settings.save()\r
 286 \r
 287             if user.subscription_settings.enable_notifications:\r
 288                 request.user.message_set.create(message=_('Notifications are now enabled'))\r
 289             else:\r
 290                 request.user.message_set.create(message=_('Notifications are now disabled'))\r
 291 \r
 292         form.is_valid()\r
 293         for k, v in form.cleaned_data.items():\r
 294             setattr(user.subscription_settings, k, v)\r
 295 \r
 296         user.subscription_settings.save()\r
 297         request.user.message_set.create(message=_('New subscription settings are now saved'))\r
 298     else:\r
 299         form = SubscriptionSettingsForm(user.subscription_settings.__dict__)\r
 300 \r
 301     notificatons_on = user.subscription_settings.enable_notifications\r
 302 \r
 303     return {'view_user':user, 'notificatons_on': notificatons_on, 'form':form}\r
 304 \r
 305 @login_required\r
 306 def account_settings(request):\r
 307     logging.debug('')\r
 308     msg = request.GET.get('msg', '')\r
 309     is_openid = False\r
 310 \r
 311     return render_to_response('account_settings.html', {\r
 312     'msg': msg,\r
 313     'is_openid': is_openid\r
 314     }, context_instance=RequestContext(request))\r
 315 \r