models.py
3.29 KB
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
from django.contrib.auth.models import BaseUserManager, PermissionsMixin
from django.db import models
from django.utils import timezone
from django.utils.translation import ugettext_lazy as _
from floraconcierge.mapping.model import User as ApiUser
class StaffManager(BaseUserManager):
def _create_user(self, email, is_staff, is_superuser, **extra_fields):
"""
Creates and saves a User with the given email and password.
"""
now = timezone.now()
if not email:
raise ValueError('The given email must be set')
email = self.normalize_email(email)
user = self.model(email=email,
is_staff=is_staff, is_active=True,
is_superuser=is_superuser, last_login=now,
date_joined=now, **extra_fields)
user.save(using=self._db)
return user
def create_user(self, email, **extra_fields):
return self._create_user(email, False, False, **extra_fields)
def create_superuser(self, email, password, **extra_fields):
return self._create_user(email, True, True, **extra_fields)
class User(PermissionsMixin):
email = models.EmailField(unique=True)
date_joined = models.DateTimeField(_('date joined'), default=timezone.now)
last_login = models.DateTimeField(_('last login'), default=timezone.now)
is_staff = models.BooleanField(_('staff status'), default=False,
help_text=_('Designates whether the user can log into this admin '
'site.'))
is_active = models.BooleanField(_('active'), default=True,
help_text=_('Designates whether this user should be treated as '
'active. Unselect this instead of deleting accounts.'))
USERNAME_FIELD = 'email'
REQUIRED_FIELDS = []
objects = StaffManager()
def __init__(self, *args, **kwargs):
api_user_model = kwargs.pop('api_user_model', None)
super(User, self).__init__(*args, **kwargs)
self._from_db = True
if api_user_model:
self.info = api_user_model
self._from_db = False
@classmethod
def from_api_user(cls, user):
assert isinstance(user, ApiUser)
try:
obj = cls.objects.get(email=user.email)
obj.info = user
except User.DoesNotExist:
obj = cls(api_user_model=user)
return obj
def save(self, force_insert=False, force_update=False, using=None,
update_fields=None):
if self._from_db or not update_fields:
return super(User, self).save(force_insert=force_insert, force_update=force_update, using=using,
update_fields=update_fields)
def get_email(self):
return self.email or self.info.email
def get_username(self):
return self.get_email()
def is_anonymous(self):
return self.info is not None
def is_authenticated(self):
return bool(self.info)
def __str__(self):
return self.get_email()
def __unicode__(self):
return self.get_email()
class Meta:
verbose_name = _('user')
verbose_name_plural = _('users')
swappable = 'AUTH_USER_MODEL'