1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
|
# -*- coding: utf-8 -*-
# Copyright (C) 1998-2012 by the Free Software Foundation, Inc.
#
# This file is part of HyperKitty.
#
# HyperKitty is free software: you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# HyperKitty is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# HyperKitty. If not, see <http://www.gnu.org/licenses/>.
#
# Author: Aamir Khan <syst3m.w0rm@gmail.com>
#
import logging
from django.conf import settings
from django.core.urlresolvers import reverse
from django.core.exceptions import SuspiciousOperation, ObjectDoesNotExist
from django.contrib.auth import authenticate, login, get_backends
from django.contrib.auth.decorators import login_required
from django.contrib.auth.models import User
from django.contrib.auth.views import login as django_login_view
from django.shortcuts import render, redirect
from django.utils.http import is_safe_url
from django.utils.timezone import utc, get_current_timezone
#from django.utils.translation import gettext as _
from social_auth.backends import SocialAuthBackend
from hyperkitty.models import UserProfile, Rating, Favorite, LastView
from hyperkitty.views.forms import RegistrationForm, UserProfileForm
from hyperkitty.lib import get_store, FLASH_MESSAGES, paginate
logger = logging.getLogger(__name__)
def login_view(request, *args, **kwargs):
if "extra_context" not in kwargs:
kwargs["extra_context"] = {}
if "backends" not in kwargs["extra_context"]:
kwargs["extra_context"]["backends"] = []
# Note: sorry but I really find the .setdefault() method non-obvious and
# harder to re-read that the lines above.
for backend in get_backends():
if not isinstance(backend, SocialAuthBackend):
continue # It should be checked using duck-typing instead
kwargs["extra_context"]["backends"].append(backend.name)
return django_login_view(request, *args, **kwargs)
@login_required
def user_profile(request, user_email=None):
if not request.user.is_authenticated():
return redirect('user_login')
store = get_store(request)
# try to render the user profile.
try:
user_profile = request.user.get_profile()
except ObjectDoesNotExist:
user_profile = UserProfile.objects.create(user=request.user)
if request.method == 'POST':
form = UserProfileForm(request.POST)
if form.is_valid():
request.user.first_name = form.cleaned_data["first_name"]
request.user.last_name = form.cleaned_data["last_name"]
user_profile.timezone = form.cleaned_data["timezone"]
request.user.save()
user_profile.save()
redirect_url = reverse('user_profile')
redirect_url += "?msg=updated-ok"
return redirect(redirect_url)
else:
form = UserProfileForm(initial={
"first_name": request.user.first_name,
"last_name": request.user.last_name,
"timezone": get_current_timezone(),
})
# Favorites
try:
favorites = Favorite.objects.filter(user=request.user)
except Favorite.DoesNotExist:
favorites = []
for fav in favorites:
thread = store.get_thread(fav.list_address, fav.threadid)
fav.thread = thread
# Flash messages
flash_messages = []
flash_msg = request.GET.get("msg")
if flash_msg:
flash_msg = { "type": FLASH_MESSAGES[flash_msg][0],
"msg": FLASH_MESSAGES[flash_msg][1] }
flash_messages.append(flash_msg)
context = {
'user_profile' : user_profile,
'form': form,
'favorites': favorites,
'flash_messages': flash_messages,
}
return render(request, "user_profile.html", context)
def user_registration(request):
if not settings.USE_INTERNAL_AUTH and \
request.META["SERVER_NAME"] != "testserver": # work with unit tests
raise SuspiciousOperation
redirect_to = request.REQUEST.get("next", reverse("root"))
if not is_safe_url(url=redirect_to, host=request.get_host()):
redirect_to = settings.LOGIN_REDIRECT_URL
if request.user.is_authenticated():
# Already registered, redirect back to index page
return redirect(redirect_to)
if request.POST:
form = RegistrationForm(request.POST)
if form.is_valid():
u = User.objects.create_user(form.cleaned_data['username'],
form.cleaned_data['email'],
form.cleaned_data['password1'])
u.is_active = True
u.save()
user = authenticate(username=form.cleaned_data['username'],
password=form.cleaned_data['password1'])
if user is not None:
logger.debug(user)
if user.is_active:
login(request, user)
return redirect(redirect_to)
else:
form = RegistrationForm()
context = {
'form': form,
'next': redirect_to,
}
return render(request, 'register.html', context)
@login_required
def last_views(request):
store = get_store(request)
# Last viewed threads
try:
last_views = LastView.objects.filter(user=request.user
).order_by("view_date")
except Favorite.DoesNotExist:
last_views = []
last_views = paginate(last_views, request.GET.get('lvpage'))
for last_view in last_views:
thread = store.get_thread(last_view.list_address, last_view.threadid)
if thread.date_active.replace(tzinfo=utc) > last_view.view_date:
# small optimization: only query the replies if necessary
# XXX: Storm-specific (count method)
thread.unread = thread.replies_after(last_view.view_date).count()
else:
thread.unread = 0
last_view.thread = thread
return render(request, 'ajax/last_views.html', {
"last_views": last_views,
})
@login_required
def votes(request):
store = get_store(request)
# Votes
try:
votes = Rating.objects.filter(user=request.user)
except Rating.DoesNotExist:
votes = []
votes = paginate(votes, request.GET.get('vpage'))
for vote in votes:
vote.message = store.get_message_by_hash_from_list(
vote.list_address, vote.messageid)
return render(request, 'ajax/votes.html', {
"votes": votes,
})
|
