Я использую в своем приложении некоторые user_types, такие как is_admin и is_landlord. Я хочу, чтобы всякий раз, когда пользователь создается с помощью python manage.py createsuperuser
. ему автоматически назначается is_admin.
модели.py
class UserManager(BaseUserManager):
def _create_user(self, email, password, is_staff, is_superuser, **extra_fields):
if not email:
raise ValueError('Users must have an email address')
now = datetime.datetime.now(pytz.utc)
email = self.normalize_email(email)
user = self.model(
email=email,
is_staff=is_staff,
is_active=True,
is_superuser=is_superuser,
last_login=now,
date_joined=now,
**extra_fields
)
user.set_password(password)
user.save(using=self._db)
return user
def create_user(self, email=None, password=None, **extra_fields):
return self._create_user(email, password, False, False, **extra_fields)
def create_superuser(self, email, password, **extra_fields):
user = self._create_user(email, password, True, True, **extra_fields)
user.save(using=self._db)
return user
class User(AbstractBaseUser, PermissionsMixin):
email = models.EmailField(max_length=254, unique=True)
is_staff = models.BooleanField(default=False)
is_superuser = models.BooleanField(default=False)
is_active = models.BooleanField(default=True)
last_login = models.DateTimeField(null=True, blank=True)
date_joined = models.DateTimeField(auto_now_add=True)
# CUSTOM USER FIELDS
name = models.CharField(max_length=30, blank=True, null=True)
telephone = models.IntegerField(blank=True, null=True)
USERNAME_FIELD = 'email'
EMAIL_FIELD = 'email'
REQUIRED_FIELDS = []
objects = UserManager()
def get_absolute_url(self):
return "/users/%i/" % (self.pk)
def get_email(self):
return self.email
class user_type(models.Model):
is_admin = models.BooleanField(default=False)
is_landlord = models.BooleanField(default=False)
user = models.OneToOneField(User, on_delete=models.CASCADE)
def __str__(self):
if self.is_landlord == True:
return User.get_email(self.user) + " - is_landlord"
else:
return User.get_email(self.user) + " - is_admin"
Примечание. Я хочу, чтобы созданный суперпользователь автоматически имел статус is_admin.
Поэтому я попробовал метод, который работает не так эффективно, но управляемо.
То, что я делаю, это простой трюк.
просмотры.py
def Dashboard(request, pk):
user = User.objects.get(id=request.user.pk)
landlord = user.is_staff & user.is_active
if not landlord:
reservations = Tables.objects.all()
available_ = Tables.objects.filter(status = "available")
unavailable_ = Tables.objects.filter(status = "unavailable")
bar_owners_ = user_type.objects.filter(is_landlord=True)
context = {"reservations":reservations, "available_":available_, "unavailable_":unavailable_, "bar_owners_":bar_owners_}
return render(request, "dashboard/super/admin/dashboard.html", context)
else:
"""Admin dashboard view."""
user = User.objects.get(pk=pk)
reservations = Tables.objects.filter(bar__user_id=user.id)
available =Tables.objects.filter(bar__user_id=user.id, status = "available")
unavailable =Tables.objects.filter(bar__user_id=user.id, status = "unavailable")
context = {"reservations":reservations, "available":available, "unavailable":unavailable}
return render(request, "dashboard/super/landlord/dashboard.html", context)
Поэтому я сообщаю Django, что если пользователь активен или для is_active установлено значение true, только тогда у него должен быть доступ к панели инструментов, иначе у него не будет к ней доступа.
Вы можете автоматически построить модель user_type
в функции create_superuser
:
class UserManager(BaseUserManager):
# …
def create_superuser(self, email, password, **extra_fields):
user = self._create_user(email, password, True, True, **extra_fields)
user.save(using=self._db)
user_type.objects.update_or_create(
user=user,
defaults = {
'is_admin': True
}
)
return user
Однако мне непонятно, почему вы используете класс user_type
, а не просто добавляете дополнительные поля в свою модель User
. Это сделает его более эффективным, поскольку они будут извлекаться в том же запросе, когда вы извлекаете пользователя, вошедшего в систему. Также удобнее просто получить доступ к атрибуту .is_landlord
из пользовательской модели.
Примечание. Модели в Django записываются в PerlCase, а не в змеином регистре. поэтому вы можете переименовать модель с
user_type
наUserType
.
@coderboy: спасибо, это действительно опечатка. Зафиксированный.
Спасибо за помощь. Я заметил, что вы использовали
user_type..objects
. Это правильно или должно бытьuser_type.objects
с одной точкой