Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from __future__ import unicode_literals
- from django.db import models
- from django.core.mail import send_mail
- from django.contrib.auth.models import PermissionsMixin
- from django.contrib.auth.base_user import AbstractBaseUser
- from django.utils.translation import ugettext_lazy as _
- from django.core.validators import RegexValidator
- from .managers import UserManager
- class User(AbstractBaseUser, PermissionsMixin):
- phone_regex = RegexValidator(regex=r'^+?1?d{9,15}$', message="")
- phone_number = models.CharField(validators=[phone_regex], unique=True)
- email = models.EmailField(_('email address'), unique=True)
- first_name = models.CharField(_('first name'), max_length=30, blank=True)
- last_name = models.CharField(_('last name'), max_length=30, blank=True)
- date_joined = models.DateTimeField(_('date joined'), auto_now_add=True)
- is_active = models.BooleanField(_('active'), default=True)
- avatar = models.ImageField(upload_to='avatars/', null=True, blank=True)
- objects = UserManager()
- USERNAME_FIELD = 'phone_number'
- REQUIRED_FIELDS = []
- class Meta:
- verbose_name = _('user')
- verbose_name_plural = _('users')
- def get_full_name(self):
- '''
- Returns the first_name plus the last_name, with a space in between.
- '''
- full_name = '%s %s' % (self.first_name, self.last_name)
- return full_name.strip()
- def get_short_name(self):
- '''
- Returns the short name for the user.
- '''
- return self.first_name
- def email_user(self, subject, message, from_email=None, **kwargs):
- '''
- Sends an email to this User.
- '''
- send_mail(subject, message, from_email, [self.email], **kwargs)
- def sms_user(self, message):
- '''
- Sends a sms to this User.
- '''
- pass
- class UserManager(BaseUserManager):
- use_in_migrations = True
- def _create_user(self, phone_number, password, **extra_fields):
- """
- Creates and saves a User with the given phone_number and password.
- """
- if not phone_number:
- raise ValueError('The given phone number must be set')
- #email = self.normalize_email(email)
- user = self.model(phone_number=phone_number, **extra_fields)
- user.set_password(password)
- user.save(using=self._db)
- return user
- def create_user(self, phone_number, password=None, **extra_fields):
- extra_fields.setdefault('is_superuser', False)
- return self._create_user(phone_number, password, **extra_fields)
- def create_superuser(self, phone_number, password, **extra_fields):
- extra_fields.setdefault('is_superuser', True)
- if extra_fields.get('is_superuser') is not True:
- raise ValueError('Superuser must have is_superuser=True.')
- return self._create_user(phone_number, password, **extra_fields)
Add Comment
Please, Sign In to add comment