mirror of https://github.com/fspc/workstand.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
3.6 KiB
112 lines
3.6 KiB
from django.contrib.auth.models import (AbstractBaseUser, BaseUserManager,
|
|
PermissionsMixin)
|
|
from django.db import models
|
|
|
|
|
|
class CustomUserManager(BaseUserManager):
|
|
def create_user(self, email, password=None):
|
|
"""
|
|
Creates and saves a User with the given email and password.
|
|
:param email: str
|
|
:param password: str
|
|
:return: object `CustomUser`
|
|
"""
|
|
if not email:
|
|
raise ValueError('Users must have an email address')
|
|
|
|
user = self.model(
|
|
email=self.normalize_email(email),
|
|
)
|
|
|
|
user.set_password(password)
|
|
user.save(using=self._db)
|
|
return user
|
|
|
|
def create_superuser(self, email, password):
|
|
"""
|
|
Creates and saves a superuser with the given email, date of
|
|
birth and password.
|
|
:param email: str
|
|
:param password: str
|
|
:return: object `CustomUser`
|
|
"""
|
|
user = self.create_user(email, password=password)
|
|
user.is_admin = True
|
|
user.is_superuser = True
|
|
user.save(using=self._db)
|
|
return user
|
|
|
|
|
|
class CustomUser(AbstractBaseUser, PermissionsMixin):
|
|
email = models.EmailField(
|
|
verbose_name='email address',
|
|
max_length=255,
|
|
unique=True,
|
|
)
|
|
is_admin = models.BooleanField(default=False)
|
|
is_active = models.BooleanField(default=True)
|
|
|
|
objects = CustomUserManager()
|
|
|
|
USERNAME_FIELD = 'email'
|
|
|
|
@property
|
|
def is_staff(self):
|
|
# Simplest possible answer: All admins are staff
|
|
return self.is_admin
|
|
|
|
def get_short_name(self):
|
|
return self.email
|
|
|
|
def get_full_name(self):
|
|
return self.email
|
|
|
|
def __str__(self): # __unicode__ on Python 2
|
|
return self.email
|
|
|
|
class Meta:
|
|
verbose_name = 'User'
|
|
verbose_name_plural = 'Users'
|
|
|
|
|
|
class Member(models.Model):
|
|
user = models.OneToOneField(CustomUser, on_delete=models.CASCADE,
|
|
null=True)
|
|
email = models.EmailField(
|
|
verbose_name='email address',
|
|
max_length=255,
|
|
unique=False,
|
|
null=True,
|
|
blank=True,
|
|
)
|
|
email_consent = models.BooleanField(default=False, blank=False)
|
|
first_name = models.CharField(max_length=255, null=False, blank=False)
|
|
last_name = models.CharField(max_length=255, null=False, blank=False)
|
|
preferred_name = models.CharField(max_length=255, null=True, blank=True)
|
|
date_of_birth = models.DateField(null=True, blank=True)
|
|
guardian_name = models.CharField(max_length=255, null=True, blank=True)
|
|
phone = models.CharField(max_length=20, null=True, blank=True)
|
|
street = models.CharField(max_length=255, null=True, blank=True)
|
|
city = models.CharField(max_length=255, null=True, blank=True)
|
|
province = models.CharField(max_length=255, null=True, blank=True)
|
|
country = models.CharField(max_length=255, null=True, blank=True)
|
|
post_code = models.CharField(max_length=20, null=True, blank=False)
|
|
waiver = models.DateTimeField(null=True, blank=True)
|
|
is_active = models.BooleanField(default=True)
|
|
notes = models.TextField(null=True, blank=True)
|
|
suspended = models.BooleanField(default=False)
|
|
banned = models.BooleanField(default=False)
|
|
|
|
def get_full_name(self):
|
|
# The user is identified by their email address
|
|
return '{0} {1}'.format(self.first_name, self.last_name)
|
|
|
|
def get_short_name(self):
|
|
# The user is identified by their email address
|
|
if self.email:
|
|
return self.email
|
|
else:
|
|
return self.last_name
|
|
|
|
def __str__(self): # __unicode__ on Python 2
|
|
return self.email
|
|
|