Mercurial > public > sg101

annotate gpp/bio/views.py @ 118:a20b2c492d55

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Reduced some sql queries by adding a select_related and monkey patching user profiles onto a user list in shoutbox and the bio/member's list, respectively.
author Brian Neal <bgneal@gmail.com>
date Sat, 24 Oct 2009 02:39:19 +0000
parents 62eb9cbbcffc
children 903ae6168071
rev   line source
gremmie@1 1 """
gremmie@1 2 Views for the bio application.
gremmie@1 3 """
gremmie@1 4
gremmie@1 5 from django.shortcuts import render_to_response
gremmie@1 6 from django.shortcuts import get_object_or_404
gremmie@1 7 from django.template import RequestContext
gremmie@1 8 from django.contrib import auth
gremmie@1 9 from django.http import HttpResponseRedirect
bgneal@92 10 from django.core.paginator import InvalidPage
gremmie@1 11 from django.core.urlresolvers import reverse
gremmie@1 12 from django.contrib.auth.decorators import login_required
gremmie@1 13
bgneal@29 14 from elsewhere.models import SocialNetworkForm
bgneal@29 15 from elsewhere.models import InstantMessengerForm
bgneal@29 16 from elsewhere.models import WebsiteForm
bgneal@29 17
gremmie@1 18 from bio.models import UserProfile
gremmie@1 19 from bio.forms import UploadAvatarForm
gremmie@1 20 from bio.forms import EditUserForm
gremmie@1 21 from bio.forms import EditUserProfileForm
gremmie@1 22 from core.paginator import DiggPaginator
gremmie@1 23
gremmie@1 24 #######################################################################
gremmie@1 25
gremmie@1 26 def member_list(request, type='user', page=1):
gremmie@1 27 if type == 'user':
gremmie@1 28 users = auth.models.User.objects.all().order_by('username')
gremmie@1 29 else:
gremmie@1 30 users = auth.models.User.objects.all().order_by('date_joined')
gremmie@1 31
gremmie@1 32 paginator = DiggPaginator(users, 10, body=5, tail=3, margin=3, padding=2)
gremmie@1 33 try:
gremmie@1 34 the_page = paginator.page(int(page))
gremmie@1 35 except InvalidPage:
gremmie@1 36 raise Http404
gremmie@1 37
bgneal@118 38 # Attach user profiles to each post to avoid using get_user_profile() in
bgneal@118 39 # the template.
bgneal@118 40 users = set(user.id for user in the_page.object_list)
bgneal@118 41
bgneal@118 42 profiles = UserProfile.objects.filter(user__id__in=users).select_related()
bgneal@118 43 user_profiles = dict((profile.user.id, profile) for profile in profiles)
bgneal@118 44
bgneal@118 45 for user in the_page.object_list:
bgneal@118 46 user.user_profile = user_profiles[user.id]
bgneal@118 47
gremmie@1 48 return render_to_response('bio/members.html', {
gremmie@1 49 'page': the_page,
gremmie@1 50 'type': type,
gremmie@1 51 },
gremmie@1 52 context_instance = RequestContext(request))
gremmie@1 53
gremmie@1 54 #######################################################################
gremmie@1 55
gremmie@1 56 @login_required
gremmie@1 57 def my_profile(request):
bgneal@44 58 profile = request.user.get_profile()
gremmie@1 59
gremmie@1 60 return render_to_response('bio/view_profile.html', {
gremmie@1 61 'subject': request.user,
gremmie@1 62 'profile': profile,
gremmie@1 63 'hide_email': False,
gremmie@1 64 'this_is_me': True,
gremmie@1 65 },
gremmie@1 66 context_instance = RequestContext(request))
gremmie@1 67
gremmie@1 68 #######################################################################
gremmie@1 69
gremmie@1 70 @login_required
gremmie@1 71 def view_profile(request, username):
gremmie@1 72
gremmie@1 73 user = get_object_or_404(auth.models.User, username = username)
gremmie@1 74 if user == request.user:
gremmie@1 75 return HttpResponseRedirect(reverse('bio.views.my_profile'))
gremmie@1 76
bgneal@44 77 profile = user.get_profile()
gremmie@1 78
gremmie@1 79 # work around MySQL's handling of Boolean
gremmie@1 80 hide_email = bool(profile.hide_email)
gremmie@1 81
gremmie@1 82 return render_to_response('bio/view_profile.html', {
gremmie@1 83 'subject': user,
gremmie@1 84 'profile': profile,
gremmie@1 85 'hide_email': hide_email,
gremmie@1 86 'this_is_me': False,
gremmie@1 87 },
gremmie@1 88 context_instance = RequestContext(request))
gremmie@1 89
gremmie@1 90 #######################################################################
gremmie@1 91
gremmie@1 92 @login_required
gremmie@1 93 def edit_profile(request):
gremmie@1 94 if request.method == 'POST':
gremmie@1 95 if request.POST.get('submit_button', 'Cancel') == 'Cancel':
gremmie@1 96 return HttpResponseRedirect(reverse('bio.views.my_profile'))
bgneal@44 97 profile = request.user.get_profile()
gremmie@1 98 user_form = EditUserForm(request.POST, instance=request.user)
gremmie@1 99 profile_form = EditUserProfileForm(request.POST, instance=profile)
gremmie@1 100 if user_form.is_valid() and profile_form.is_valid():
gremmie@1 101 user_form.save()
gremmie@1 102 profile = profile_form.save(commit=False)
gremmie@1 103 profile.user = request.user
gremmie@1 104 profile.save()
gremmie@1 105 return HttpResponseRedirect(reverse('bio.views.my_profile'))
gremmie@1 106 else:
bgneal@44 107 profile = request.user.get_profile()
gremmie@1 108 user_form = EditUserForm(instance=request.user)
gremmie@1 109 profile_form = EditUserProfileForm(instance=profile)
gremmie@1 110
gremmie@1 111 return render_to_response('bio/edit_profile.html', {
gremmie@1 112 'user_form': user_form,
gremmie@1 113 'profile_form': profile_form,
gremmie@1 114 },
gremmie@1 115 context_instance = RequestContext(request))
gremmie@1 116
gremmie@1 117 #######################################################################
gremmie@1 118
gremmie@1 119 @login_required
gremmie@1 120 def change_avatar(request):
gremmie@1 121 if request.method == 'POST':
gremmie@1 122 form = UploadAvatarForm(request.POST, request.FILES)
gremmie@1 123 if form.is_valid():
bgneal@44 124 profile = request.user.get_profile()
gremmie@1 125 file = form.get_file()
gremmie@1 126 if profile.avatar.name != '':
gremmie@1 127 profile.avatar.delete(save=False)
gremmie@1 128 if file is not None:
gremmie@1 129 profile.avatar.save(form.get_filename(), file, save=False)
gremmie@1 130 profile.save()
gremmie@1 131
gremmie@1 132 request.user.message_set.create(message='Avatar updated.')
gremmie@1 133 return HttpResponseRedirect(reverse('bio-me'))
gremmie@1 134 else:
gremmie@1 135 form = UploadAvatarForm()
gremmie@1 136
gremmie@1 137 return render_to_response('bio/avatar.html', {
gremmie@1 138 'form': form,
gremmie@1 139 },
gremmie@1 140 context_instance = RequestContext(request))
gremmie@1 141
bgneal@29 142 #######################################################################
gremmie@1 143
bgneal@29 144 @login_required
bgneal@29 145 def edit_elsewhere(request):
bgneal@29 146 im_id = 'id_im_%s' # to prevent duplicate ID in HTML output
bgneal@29 147 if request.method == 'POST':
bgneal@29 148 new_data = request.POST.copy()
bgneal@29 149
bgneal@29 150 # Add forms
bgneal@29 151 if new_data.get('sn-form') or new_data.get('im-form') or new_data.get('w-form'):
bgneal@29 152
bgneal@29 153 if new_data.get('sn-form'):
bgneal@29 154 sn_form = SocialNetworkForm(new_data)
bgneal@29 155 im_form = InstantMessengerForm(auto_id=im_id)
bgneal@29 156 w_form = WebsiteForm()
bgneal@29 157 form = sn_form
bgneal@29 158 elif new_data.get('im-form'):
bgneal@29 159 sn_form = SocialNetworkForm()
bgneal@29 160 im_form = InstantMessengerForm(new_data, auto_id=im_id)
bgneal@29 161 w_form = WebsiteForm()
bgneal@29 162 form = im_form
bgneal@29 163 elif new_data.get('w-form'):
bgneal@29 164 sn_form = SocialNetworkForm()
bgneal@29 165 im_form = InstantMessengerForm(auto_id=im_id)
bgneal@29 166 w_form = WebsiteForm(new_data)
bgneal@29 167 form = w_form
bgneal@29 168
bgneal@29 169 if form.is_valid():
bgneal@29 170 profile = form.save(commit=False)
bgneal@29 171 profile.user = request.user
bgneal@29 172 profile.save()
bgneal@29 173 return HttpResponseRedirect(request.path)
bgneal@29 174
bgneal@29 175 # Delete forms
bgneal@29 176 elif new_data.get('delete-sn-form') or new_data.get('delete-im-form') or new_data.get('delete-w-form'):
bgneal@29 177 delete_id = request.POST['delete_id']
bgneal@29 178
bgneal@29 179 if new_data.get('delete-sn-form'):
bgneal@29 180 request.user.social_network_profiles.get(id=delete_id).delete()
bgneal@29 181 elif new_data.get('delete-im-form'):
bgneal@29 182 request.user.instant_messenger_profiles.get(id=delete_id).delete()
bgneal@29 183 elif new_data.get('delete-w-form'):
bgneal@29 184 request.user.website_profiles.get(id=delete_id).delete()
bgneal@29 185
bgneal@29 186 return HttpResponseRedirect(request.path)
bgneal@29 187
bgneal@29 188 # WTF?
bgneal@29 189 else:
bgneal@29 190 return HttpResponseServerError
bgneal@29 191
bgneal@29 192 else:
bgneal@29 193 # Create blank forms
bgneal@29 194 sn_form = SocialNetworkForm()
bgneal@29 195 im_form = InstantMessengerForm(auto_id=im_id)
bgneal@29 196 w_form = WebsiteForm()
bgneal@29 197
bgneal@29 198 return render_to_response('bio/edit_elsewhere.html', {
bgneal@29 199 'sn_form': sn_form,
bgneal@29 200 'im_form': im_form,
bgneal@29 201 'w_form': w_form,
bgneal@29 202 },
bgneal@29 203 context_instance=RequestContext(request))
bgneal@29 204