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
|
from django.shortcuts import render
from django.http import HttpResponseRedirect
from django.template import RequestContext
from django.db import transaction
from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.conf import settings
from datetime import datetime
from pgcommitfest.mailqueue.util import send_template_mail
from models import UserProfile, UserExtraEmail
from forms import UserProfileForm, MailForm
from util import generate_random_token
@login_required
@transaction.atomic
def userprofile(request):
(profile, created) = UserProfile.objects.get_or_create(user=request.user)
form = mailform = None
if request.method == 'POST':
if request.POST['submit'] == 'Save':
form = UserProfileForm(request.user, request.POST, instance=profile)
if form.is_valid():
form.save()
messages.add_message(request, messages.INFO, "User profile saved.")
return HttpResponseRedirect('.')
elif request.POST['submit'] == 'Add email':
mailform = MailForm(request.POST)
if mailform.is_valid():
m = UserExtraEmail(user=request.user,
email=mailform.cleaned_data['email'],
confirmed=False,
token=generate_random_token(),
tokensent=datetime.now())
m.save()
send_template_mail(settings.NOTIFICATION_FROM,
request.user.username,
m.email,
'Your email address for commitfest.postgresql.org',
'extra_email_mail.txt',
{'token': m.token, 'user': m.user})
messages.info(request, "A confirmation token has been sent to %s" % m.email)
return HttpResponseRedirect('.')
else:
messages.error(request, "Invalid submit button pressed! Nothing saved.")
return HttpResponseRedirect('.')
if not form:
form = UserProfileForm(request.user, instance=profile)
if not mailform:
mailform = MailForm()
extramails = UserExtraEmail.objects.filter(user=request.user)
return render(request, 'userprofileform.html', {
'form': form,
'extramails': extramails,
'mailform': mailform,
})
@login_required
@transaction.atomic
def deletemail(request):
try:
id = int(request.META['QUERY_STRING'])
except ValueError:
messages.error(request, "Invalid format of id in query string")
return HttpResponseRedirect('../')
try:
e = UserExtraEmail.objects.get(user=request.user, id=id)
except UserExtraEmail.DoesNotExist:
messages.error(request, "Specified email address does not exist on this user")
return HttpResponseRedirect('../')
messages.info(request, "Email address %s deleted." % e.email)
e.delete()
return HttpResponseRedirect('../')
@login_required
@transaction.atomic
def confirmemail(request, tokenhash):
try:
e = UserExtraEmail.objects.get(user=request.user, token=tokenhash)
if e.confirmed:
messages.warning(request, "This email address has already been confirmed.")
else:
# Ok, it's not confirmed. So let's do that now
e.confirmed = True
e.token = ''
e.save()
messages.info(request, "Email address %s added to profile." % e.email)
except UserExtraEmail.DoesNotExist:
messages.error(request, "Token %s was not found for your user. It may be because it has already been used?" % tokenhash)
return HttpResponseRedirect("../../")
|
