Bug Fix: User avatar not being randomly set on register

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)