diff options
| -rw-r--r-- | users/admin.py | 3 | ||||
| -rw-r--r-- | users/forms.py | 5 |
2 files changed, 7 insertions, 1 deletions
diff --git a/users/admin.py b/users/admin.py index f5116502..8d54136f 100644 --- a/users/admin.py +++ b/users/admin.py @@ -1,6 +1,7 @@ from django.contrib import admin # Register your models here. -from .models import UserProfile +from .models import UserProfile, TokenStore admin.site.register(UserProfile) +admin.site.register(TokenStore) diff --git a/users/forms.py b/users/forms.py index d66e560d..50179abd 100644 --- a/users/forms.py +++ b/users/forms.py @@ -7,6 +7,8 @@ from django.template.loader import render_to_string from django.utils.html import strip_tags from .accountFunctions import store_token from .mail_send import send_email +from random import choice +from blog.context_processors import avatar_list class RegisterForm(forms.Form): username = forms.CharField(label='Username', max_length=30, min_length=4) @@ -47,6 +49,9 @@ class RegisterForm(forms.Form): ) 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) |