Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- from __future__ import print_function, unicode_literals
- from django.db import models
- from django.conf import settings
- from django.utils import timezone
- from django.core.mail import send_mail
- from django.utils.translation import ugettext_lazy as _
- from django.utils.encoding import python_2_unicode_compatible
- from django.contrib.auth.models import BaseUserManager, PermissionsMixin, AbstractBaseUser
- class UserManager(BaseUserManager):
- use_in_migrations = True
- def _create_user(self, email, password, **extra_fields):
- """
- Creates and saves a User with the given username, email and password.
- """
- if not email:
- raise ValueError('The given email must be set')
- email = self.normalize_email(email)
- user = self.model(email=email, **extra_fields)
- user.set_password(password)
- user.save(using=self._db)
- return user
- def create_user(self, email, password=None, **extra_fields):
- extra_fields.setdefault('is_staff', False)
- extra_fields.setdefault('is_superuser', False)
- return self._create_user(email, password, **extra_fields)
- def create_superuser(self, email, password, **extra_fields):
- extra_fields.setdefault('is_staff', True)
- extra_fields.setdefault('is_superuser', True)
- if extra_fields.get('is_staff') is not True:
- raise ValueError('Superuser must have is_staff=True.')
- if extra_fields.get('is_superuser') is not True:
- raise ValueError('Superuser must have is_superuser=True.')
- return self._create_user(email, password, **extra_fields)
- class User(AbstractBaseUser, PermissionsMixin):
- email = models.EmailField(
- verbose_name=_('Email Address'),
- max_length=255,
- unique=True,
- )
- first_name = models.CharField(
- verbose_name=_('first name'),
- max_length=50,
- blank=True
- )
- last_name = models.CharField(
- verbose_name=_('last name'),
- max_length=50,
- blank=True
- )
- is_staff = models.BooleanField(
- verbose_name=_('staff status'),
- default=False,
- help_text=_('Designates whether the user can log into this admin site.'),
- )
- is_active = models.BooleanField(
- verbose_name=_('active'),
- default=True,
- help_text=_(
- 'Designates whether this user should be treated as active. '
- 'Unselect this instead of deleting accounts.'
- ),
- )
- date_joined = models.DateTimeField(
- verbose_name=_('date joined'),
- auto_now_add=True
- )
- objects = UserManager()
- USERNAME_FIELD = 'email'
- class Meta:
- verbose_name = _('user')
- verbose_name_plural = _('users')
- def group_name(self):
- return ','.join([group.name for group in self.groups.all()])
- def is_callblade_admin(self):
- if self.groups.filter(name=settings.CALLBLADE_GROUP_ADMIN).exists():
- return True
- return False
- def is_admin(self):
- return self.is_superuser or self.is_callblade_admin()
- is_admin.boolean = True
- def is_buyer(self):
- if self.groups.filter(name=settings.BUYER_GROUP).exists() and self.buyer_set.exists():
- return True
- return False
- is_buyer.boolean = True
- def is_vendor(self):
- if self.groups.filter(name=settings.VENDOR_GROUP).exists() and self.vendor_set.exists():
- return True
- return False
- is_vendor.boolean = True
- 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)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement