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
|
# Registration form
import string
from random import choice
from django import forms
from django.contrib.auth.models import User
from django.template.loader import render_to_string
from django.utils.html import strip_tags
from blog.context_processors import avatar_list
from users.models import UserProfile
from .accountFunctions import store_token
from .mail_send import send_email
class RegisterForm(forms.Form):
username = forms.CharField(label='Username', max_length=30, min_length=4)
email = forms.EmailField(label='Email')
password1 = forms.CharField(label='Password', widget=forms.PasswordInput, min_length=8)
password2 = forms.CharField(label='Password (again)', widget=forms.PasswordInput, min_length=8)
captcha = forms.CharField(label='Captcha', max_length=6)
expected_captcha = None
protected_usernames = [
'admin',
'administrator',
'root',
'thatcomputerscientist',
'skippy',
'system',
'test',
'user',
'webmaster',
'www',
'postmaster',
'hostmaster',
'info',
'support',
]
allowed_chars = string.ascii_letters + string.digits
def __init__(self, *args, **kwargs):
if 'expected_captcha' in kwargs:
self.expected_captcha = kwargs.pop('expected_captcha')
super().__init__(*args, **kwargs)
def clean(self):
cleaned_data = super().clean()
password1 = cleaned_data.get('password1')
password2 = cleaned_data.get('password2')
captcha = cleaned_data.get('captcha')
if password1 and password2:
if password1 != password2:
raise forms.ValidationError('Passwords do not match.')
if len(password1) < 8:
raise forms.ValidationError('Password must be at least 8 characters long.')
if str.lower(captcha) != str.lower(self.expected_captcha):
raise forms.ValidationError('Captcha does not match.')
if User.objects.filter(username=cleaned_data.get('username')).exists():
raise forms.ValidationError('Username already exists.')
if cleaned_data.get('username').lower() in self.protected_usernames:
raise forms.ValidationError('Username not allowed. Please choose another.')
for char in cleaned_data.get('username'):
if char not in self.allowed_chars:
raise forms.ValidationError('Username contains invalid characters. Only A-Z, a-z, and 0-9 are allowed.')
if User.objects.filter(email=cleaned_data.get('email')).exists():
raise forms.ValidationError('Email already exists.')
return cleaned_data
def save(self, request):
user = User.objects.create_user(
username=self.cleaned_data.get('username').lower(),
email=self.cleaned_data.get('email').lower(),
password=self.cleaned_data.get('password1'),
)
user.save()
user_profile = UserProfile.objects.create(user=user)
avatar_dir = choice(list(avatar_list().keys()))
avatar_file = choice(avatar_list()[avatar_dir])
user_profile.avatar_url = avatar_dir + '/' + avatar_file.replace('.gif', '')
user_profile.save()
uid, token = store_token(token_type='verifyemail', user=user, email=user.email)
# Send verification email
subject = 'Verify your email address'
message = render_to_string('verification_email.html', {
'user': user.username if user.first_name is None else user.first_name,
'site_name': 'That Computer Scientist',
'uid': uid,
'token': token,
'protocol': 'https://' if request.is_secure() else 'http://',
'domain': request.get_host(),
})
message = strip_tags(message)
# send_mail(subject, message, 'That Computer Scientist <' + settings.EMAIL_HOST_USER + '>', [user.email], fail_silently=False)
if (send_email(sender='[email protected]', sender_name='That Computer Scientist', recipient=user.email, subject=subject, body_html=message, body_text=message)):
return user
else:
return user
class UpdateUserDetailsForm(forms.Form):
first_name = forms.CharField(label='First name', max_length=30, required=False, widget=forms.TextInput(attrs={'placeholder': 'First name'}))
last_name = forms.CharField(label='Last name', max_length=30, required=False, widget=forms.TextInput(attrs={'placeholder': 'Last name'}))
location = forms.CharField(label='Location', max_length=30, required=False, widget=forms.TextInput(attrs={'placeholder': 'Location'}))
bio = forms.CharField(label='Bio', max_length=500, required=False, widget=forms.Textarea(attrs={'placeholder': 'Bio'}))
is_public = forms.ChoiceField(label='Activity Visibility', choices=((True, 'Public'), (False, 'Private')), widget=forms.RadioSelect)
email_public = forms.ChoiceField(label='Email Visibility', choices=((True, 'Public'), (False, 'Private')), widget=forms.RadioSelect)
def __init__(self, *args, **kwargs):
self.user = kwargs.pop('user')
super().__init__(*args, **kwargs)
def clean(self):
cleaned_data = super().clean()
return cleaned_data
def save(self):
self.user.first_name = self.cleaned_data.get('first_name')
self.user.last_name = self.cleaned_data.get('last_name')
self.user.save()
user_profile = UserProfile.objects.get(user=self.user)
user_profile.location = self.cleaned_data.get('location')
user_profile.bio = self.cleaned_data.get('bio')
user_profile.is_public = self.cleaned_data.get('is_public')
user_profile.email_public = self.cleaned_data.get('email_public')
user_profile.save()
return (self.user, user_profile)
|
