diff options
Diffstat (limited to 'website/managers.py')
-rw-r--r-- | website/managers.py | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/website/managers.py b/website/managers.py new file mode 100644 index 0000000..c55b931 --- /dev/null +++ b/website/managers.py @@ -0,0 +1,49 @@ +from django.db import models +from django.contrib.auth.models import ( + BaseUserManager, AbstractBaseUser +) + +class DrupalUserManager(BaseUserManager): + def create_user(self, password=None): + user = self.model() + if not email: + raise ValueError('Users must have an email address') + + user.set_password(password) + user.save(using=self._db) + return user + + def create_superuser(self, username, password): + user = self.model() + user.username = username + user.password = password + user.is_admin = True + user.save(using=self._db) + return user + + + + + + + + + + + + + + + + + + + + + + + + + + + |