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
|
# -*- coding: utf-8 -*-
#
# app 'account' models
# registration
#
from django.db import models
from django.contrib.auth.models import User
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'
NOTE: keep this model fields consistent with 'user_registered_callback'
"""
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")),
)
# choices for identify
IDENTIFIES = (
('OT', _("Other")),
('UG', _("Undergraudate")),
('MG', _("Master graudate")),
('PG', _("PhD graudate")),
('PD', _("Post-doctoral")),
('SR', _("Assistant researcher")),
('AR', _("Associate researcher")),
('DR', _("Distinguished researcher")),
('RE', _("Researcher")),
('IN', _("Instructor")),
('SP', _("Assistant professor")),
('AP', _("Associate professor")),
('PR', _("Professor")),
)
# 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)
identify = models.CharField(_("Identify"), max_length=2, choices=IDENTIFIES)
# 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')
def get_gender_value(self):
"""
return the corresponding value of user's gender
"""
genders_dict = dict((k, v) for k, v in self.GENDERS)
return genders_dict.get(self.gender)
def get_approved_value(self):
"""
return the corresponding value of is_approved for the user
"""
approved_dict = dict((k, v) for k, v in self.APPROVED_STATUS)
return approved_dict.get(self.is_approved)
def get_sponsored_value(self):
"""
return the corresponding value of is_sponsored for the user
"""
sponsored_dict = dict((k, v) for k, v in self.SPONSORED_STATUS)
return sponsored_dict.get(self.is_sponsored)
def get_identify_value(self):
"""
return the corresponding value of identify for the user
"""
identifies_dict = dict((k, v) for k, v in self.IDENTIFIES)
return identifies_dict.get(self.identify)
###### 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.identify = request.POST['identify']
profile.save()
### connect 'user_registered_callback' to signal
user_registered.connect(user_registered_callback)
|
