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 date\r
  19 import decorators\r
  20 from forum.actions import EditProfileAction, FavoriteAction\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('-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         base_url = reverse('users') + '?name=%s&sort=%s&' % (suser, sortby)\r
  50 \r
  51     try:\r
  52         users = objects_list.page(page)\r
  53     except (EmptyPage, InvalidPage):\r
  54         users = objects_list.page(objects_list.num_pages)\r
  55 \r
  56     return render_to_response('users/users.html', {\r
  57                                 "users" : users,\r
  58                                 "suser" : suser,\r
  59                                 "keywords" : suser,\r
  60                                 "tab_id" : sortby,\r
  61                                 "context" : {\r
  62                                     'is_paginated' : is_paginated,\r
  63                                     'pages': objects_list.num_pages,\r
  64                                     'page': page,\r
  65                                     'has_previous': users.has_previous(),\r
  66                                     'has_next': users.has_next(),\r
  67                                     'previous': users.previous_page_number(),\r
  68                                     'next': users.next_page_number(),\r
  69                                     'base_url' : base_url\r
  70                                 }\r
  71 \r
  72                                 }, context_instance=RequestContext(request))\r
  73 \r
  74 def set_new_email(user, new_email, nomessage=False):\r
  75     if new_email != user.email:\r
  76         user.email = new_email\r
  77         user.email_isvalid = False\r
  78         user.save()\r
  79         #if settings.EMAIL_VALIDATION == 'on':\r
  80         #    send_new_email_key(user,nomessage=nomessage)    \r
  81 \r
  82 @login_required\r
  83 def edit_user(request, id):\r
  84     user = get_object_or_404(User, id=id)\r
  85     if not (request.user.is_superuser or request.user == user):\r
  86         return HttpResponseForbidden()\r
  87     if request.method == "POST":\r
  88         form = EditUserForm(user, request.POST)\r
  89         if form.is_valid():\r
  90             new_email = sanitize_html(form.cleaned_data['email'])\r
  91 \r
  92             set_new_email(user, new_email)\r
  93 \r
  94             if settings.EDITABLE_SCREEN_NAME:\r
  95                 user.username = sanitize_html(form.cleaned_data['username'])\r
  96             user.real_name = sanitize_html(form.cleaned_data['realname'])\r
  97             user.website = sanitize_html(form.cleaned_data['website'])\r
  98             user.location = sanitize_html(form.cleaned_data['city'])\r
  99             user.date_of_birth = sanitize_html(form.cleaned_data['birthday'])\r
 100             if user.date_of_birth == "None":\r
 101                 user.date_of_birth = '1900-01-01'\r
 102             user.about = sanitize_html(form.cleaned_data['about'])\r
 103 \r
 104             user.save()\r
 105             EditProfileAction(user=user, ip=request.META['REMOTE_ADDR']).save()\r
 106 \r
 107             return HttpResponseRedirect(user.get_profile_url())\r
 108     else:\r
 109         form = EditUserForm(user)\r
 110     return render_to_response('users/edit.html', {\r
 111                                                 'user': user,\r
 112                                                 'form' : form,\r
 113                                                 'gravatar_faq_url' : reverse('faq') + '#gravatar',\r
 114                                     }, context_instance=RequestContext(request))\r
 115 \r
 116 \r
 117 @login_required\r
 118 def user_powers(request, id, action, status):\r
 119     if not request.user.is_superuser:\r
 120         return HttpResponseForbidden()\r
 121 \r
 122     user = get_object_or_404(User, id=id)\r
 123     new_state = action == 'grant'\r
 124 \r
 125     if status == 'super':\r
 126         user.is_superuser = new_state\r
 127     elif status == 'staff':\r
 128         user.is_staff = new_state\r
 129     else:\r
 130         raise Http404()\r
 131 \r
 132     user.save()    \r
 133     return HttpResponseRedirect(user.get_profile_url())\r
 134 \r
 135 \r
 136 def user_view(template, tab_name, tab_description, page_title, private=False):\r
 137     def decorator(fn):\r
 138         def decorated(request, id, slug=None):\r
 139             user = get_object_or_404(User, id=id)\r
 140             if private and not (user == request.user or request.user.is_superuser):\r
 141                 return HttpResponseForbidden()\r
 142             context = fn(request, user)\r
 143 \r
 144             rev_page_title = user.username + " - " + page_title\r
 145 \r
 146             context.update({\r
 147                 "tab_name" : tab_name,\r
 148                 "tab_description" : tab_description,\r
 149                 "page_title" : rev_page_title,\r
 150                 "can_view_private": (user == request.user) or request.user.is_superuser\r
 151             })\r
 152             return render_to_response(template, context, context_instance=RequestContext(request))\r
 153         return decorated\r
 154     return decorator\r
 155 \r
 156 \r
 157 @user_view('users/stats.html', 'stats', _('user profile'), _('user overview'))\r
 158 def user_stats(request, user):\r
 159     questions = Question.objects.filter(author=user, deleted=None).order_by('-added_at')\r
 160     answers = Answer.objects.filter(author=user, deleted=None).order_by('-added_at')\r
 161 \r
 162     up_votes = user.vote_up_count\r
 163     down_votes = user.vote_down_count\r
 164     votes_today = user.get_vote_count_today()\r
 165     votes_total = int(settings.MAX_VOTES_PER_DAY)\r
 166 \r
 167     user_tags = Tag.objects.filter(Q(nodes__author=user) | Q(nodes__children__author=user)) \\r
 168         .annotate(user_tag_usage_count=Count('name')).order_by('-user_tag_usage_count')\r
 169 \r
 170     awards = [(Badge.objects.get(id=b['id']), b['count']) for b in\r
 171             Badge.objects.filter(awards__user=user).values('id').annotate(count=Count('cls')).order_by('-count')]\r
 172 \r
 173     return {\r
 174             "view_user" : user,\r
 175             "questions" : questions,\r
 176             "answers" : answers,\r
 177             "up_votes" : up_votes,\r
 178             "down_votes" : down_votes,\r
 179             "total_votes": up_votes + down_votes,\r
 180             "votes_today_left": votes_total-votes_today,\r
 181             "votes_total_per_day": votes_total,\r
 182             "user_tags" : user_tags[:50],\r
 183             "awards": awards,\r
 184             "total_awards" : len(awards),\r
 185         }\r
 186 \r
 187 @user_view('users/recent.html', 'recent', _('recent user activity'), _('recent activity'))\r
 188 def user_recent(request, user):\r
 189     activities = user.actions.exclude(action_type__in=("voteup", "votedown", "voteupcomment", "flag")).order_by('-action_date')[:USERS_PAGE_SIZE]\r
 190 \r
 191     return {"view_user" : user, "activities" : activities}\r
 192 \r
 193 \r
 194 @user_view('users/votes.html', 'votes', _('user vote record'), _('votes'), True)\r
 195 def user_votes(request, user):\r
 196     votes = user.votes.filter(node__deleted=None, node__node_type__in=("question", "answer")).order_by('-voted_at')[:USERS_PAGE_SIZE]\r
 197 \r
 198     return {"view_user" : user, "votes" : votes}\r
 199 \r
 200 \r
 201 @user_view('users/reputation.html', 'reputation', _('user reputation in the community'), _('user reputation'))\r
 202 def user_reputation(request, user):\r
 203     rep = list(user.reputes.order_by('date'))\r
 204     values = [r.value for r in rep]\r
 205     redux = lambda x, y: x+y     \r
 206 \r
 207     graph_data = simplejson.dumps([\r
 208             (time.mktime(rep[i].date.timetuple()) * 1000, reduce(redux, values[:i], 0))\r
 209             for i in range(len(values))\r
 210     ])\r
 211 \r
 212     return {"view_user": user, "reputation": reversed(rep), "graph_data": graph_data}\r
 213 \r
 214 @user_view('users/questions.html', 'favorites', _('favorite questions'),  _('favorite questions'))\r
 215 def user_favorites(request, user):\r
 216     favorites = FavoriteAction.objects.filter(user=user)\r
 217 \r
 218     return {"favorites" : favorites, "view_user" : user}\r
 219 \r
 220 @user_view('users/subscriptions.html', 'subscriptions', _('subscription settings'), _('subscriptions'), True)\r
 221 def user_subscriptions(request, user):\r
 222     if request.method == 'POST':\r
 223         form = SubscriptionSettingsForm(request.POST)\r
 224 \r
 225         if 'notswitch' in request.POST:\r
 226             user.subscription_settings.enable_notifications = not user.subscription_settings.enable_notifications\r
 227             user.subscription_settings.save()\r
 228 \r
 229             if user.subscription_settings.enable_notifications:\r
 230                 request.user.message_set.create(message=_('Notifications are now enabled'))\r
 231             else:\r
 232                 request.user.message_set.create(message=_('Notifications are now disabled'))\r
 233         else:\r
 234             form.is_valid()\r
 235             for k,v in form.cleaned_data.items():\r
 236                 setattr(user.subscription_settings, k, v)\r
 237 \r
 238             user.subscription_settings.save()\r
 239             request.user.message_set.create(message=_('New subscription settings are now saved'))\r
 240     else:\r
 241         form = SubscriptionSettingsForm(user.subscription_settings.__dict__)\r
 242 \r
 243     notificatons_on = user.subscription_settings.enable_notifications\r
 244 \r
 245     return {'view_user':user, 'notificatons_on': notificatons_on, 'form':form}\r
 246 \r
 247 @login_required\r
 248 def account_settings(request):\r
 249     logging.debug('')\r
 250     msg = request.GET.get('msg', '')\r
 251     is_openid = False\r
 252 \r
 253     return render_to_response('account_settings.html', {\r
 254         'msg': msg,\r
 255         'is_openid': is_openid\r
 256         }, context_instance=RequestContext(request))\r
 257 \r