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
|
# -*- coding: utf-8 -*-
#
# app 'account' models
# registration
#
from django.db import models
from django.contrib.auth.models import User
from django.contrib import admin
from django.utils.translation import ugettext_lazy as _
from registration.signals import user_registered
class UserProfile(models.Model):
"""
custom user profile
connected with signal 'user_registered' sent by 'django-registration'
"""
# XXX: keep consistent with GENDERS in 'forms.UserRegForm'
GENDERS = (
('M', _("Male")),
('F', _("Female")),
('X', _("Secret")),
)
# status choices of is_approved
APPROVED_STATUS = (
('Y', _("Yes")),
('N', _("No")),
('C', _("Checking")),
)
# status choices of is_sponsored
SPONSORED_STATUS = (
('Y', _("Yes")),
('N', _("No")),
('C', _("Checking")),
)
# model fields
# FK default backward manager name 'userprofile_set'
user = models.ForeignKey(User, unique=True, verbose_name=_("Username"))
realname = models.CharField(_("Name"), max_length=30)
gender = models.CharField(_("Gender"), max_length=1, choices=GENDERS)
institute = models.CharField(_("Institute"), max_length=100)
# store the infomation about approval and sponsorship
is_approved = models.CharField(_("Is approved"), max_length=1,
choices=APPROVED_STATUS, default='C')
is_sponsored = models.CharField(_("Is sponsored"), max_length=1,
choices=SPONSORED_STATUS, default='C')
class Meta:
verbose_name = _('user profile')
verbose_name_plural = _('user profiles')
def __unicode__(self):
return u'UserProfile for %s' % self.user
def get_approved(self, *args, **kwargs):
"""
return list of approved object
"""
return self.objects.filter(is_approved='Y')
def get_sponsored(self, *args, **kwargs):
"""
return list of sponsored object
"""
return self.objects.filter(is_sponsored='Y')
###### signal callback ######
def user_registered_callback(sender, user, request, **kwargs):
"""
callback of signal 'user_registered' from 'django-registration'
to create custom user profile
ref: http://johnparsons.net/index.php/2013/06/28/creating-profiles-with-django-registration/
"""
profile = UserProfile(user = user)
profile.realname = request.POST['realname']
profile.gender = request.POST['gender']
profile.institute = request.POST['institute']
profile.save()
### connect 'user_registered_callback' to signal
user_registered.connect(user_registered_callback)
### add to adim
admin.site.register([
UserProfile,
])
|
