Guest User

Untitled

a guest
Oct 9th, 2018
112
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.85 KB | None | 0 0
  1. from django.db import models
  2. from django.contrib.auth.models import (BaseUserManager, AbstractBaseUser)
  3.  
  4.  
  5. STATUS_CHOICES = (
  6. (0, None),
  7. (1, 'siswa'),
  8. (2, 'guru')
  9. (3, 'staff'
  10. )
  11.  
  12.  
  13. class UserManager(BaseUserManager):
  14. def create_user(self, email, password, **kwargs):
  15. if not email:
  16. raise ValueError('User must have a valid email')
  17. if not password:
  18. raise ValueError('User must have a valid password')
  19.  
  20. user = self.model(email=self.normalize_email(email))
  21. user.set_password(password)
  22. user.save()
  23.  
  24. return user
  25.  
  26. def create_superuser(self, email, password, **kwargs):
  27. user = self.create_user(email, password, **kwargs)
  28. user.is_superuser = True
  29. user.is_admin = True
  30. user.is_staff = True
  31. user.is_active = True
  32. user.save()
  33.  
  34. return user
Add Comment
Please, Sign In to add comment