sintonia/api/libretime_api/managers.py

23 lines
728 B
Python
Raw Normal View History

2020-01-30 14:47:36 +01:00
from django.contrib.auth.models import BaseUserManager
2021-05-27 16:23:02 +02:00
2020-01-30 14:47:36 +01:00
class UserManager(BaseUserManager):
def create_user(self, username, type, email, first_name, last_name, password):
2021-05-27 16:23:02 +02:00
user = self.model(
username=username,
type=type,
email=email,
first_name=first_name,
last_name=last_name,
)
2020-01-30 14:47:36 +01:00
user.set_password(password)
user.save(using=self._db)
return user
def create_superuser(self, username, email, first_name, last_name, password):
2021-05-27 16:23:02 +02:00
user = self.create_user(username, "A", email, first_name, last_name, password)
2020-01-30 14:47:36 +01:00
return user
def get_by_natural_key(self, username):
return self.get(username=username)