Как создать таблицу в программе createsuperuser

У меня есть две таблицы Employee и Sector, таблица employee имеет для внешнего ключа свойство sector code (sectorCode) таблицы Sector. Таблица Employee наследуется от класса AbstractBaseUser. Я хочу создать суперпользователя с помощью команды python manage.py createsuperuser. Я получаю ошибку: ValueError: Cannot assign "'Code1'": "Employee.sectorCode" must be a "Sector" instance. (Я добавил в таблицу Sector строку NameSector1; Code1)

Я ввожу следующие значения:

λ python manage.py createsuperuser
registrationNumber: 001
Name: TestN1
FirstName: TestFN1
sectorCode: Code1
Password: ...

Error ...

Как я могу инстанцировать класс сектора в диалоге ?

models.py

from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager

class MyUserManager(BaseUserManager):
    
    def create_user(self, registrationNumber, firstName, name, sectorCode, password=None):
        if not firstName: raise ValueError("firstName required")
        if not name: raise ValueError("name required")
        if not registrationNumber: raise ValueError("registrationNumber required")
        if not sectorCode: raise ValueError("sectorCode required")
        
        user=self.model(firstName = firstName, name = name, registrationNumber = registrationNumber, sectorCode = sectorCode)
        user.set_password(password); user.save()
        return user
    
    def create_superuser(self, firstName, name, registrationNumber, sectorCode, password=None):
        user=self.create_user(firstName = firstName, name = name, registrationNumber = registrationNumber, sectorCode = sectorCode, password = password)
        user.is_admin=True; user.is_superuser=True
        user.save()
        return user

class Sector(models.Model):
    nameSector = models.CharField(verbose_name = "nameSector", max_length=50)
    sectorCode = models.CharField(verbose_name = "sectorCode", max_length=3, primary_key=True)
    
    class Meta: db_table = "Sector"
    
class Employee(AbstractBaseUser):
    firstName = models.CharField(verbose_name = "firstName", max_length=20)
    name = models.CharField(verbose_name = "name", max_length=20)
    registrationNumber = models.CharField(verbose_name="registrationNumber", max_length=20, primary_key=True)
    sectorCode = models.ForeignKey(Sector, on_delete=models.CASCADE)
    
    USERNAME_FIELD="registrationNumber"
    REQUIRED_FIELDS = ["name", "firstName", "sectorCode"]
    objects = MyUserManager()
    class Meta: db_table = "Employee"

Вы можете начать с того, чтобы убедиться, что ваша база данных синхронизирована с вашими моделями...

Например:

    $ django-admin.py migrate --fake-initial ...

    $ django-admin.py createsuperuser bob

Если вы знаете, что сектор существует, вы можете работать с:

class MyUserManager(BaseUserManager):
    def create_user(
        self, registrationNumber, firstName, name, sectorCode, password=None
    ):
        if not firstName:
            raise ValueError("firstName required")
        if not name:
            raise ValueError("name required")
        if not registrationNumber:
            raise ValueError("registrationNumber required")
        if not sectorCode:
            raise ValueError("sectorCode required")

        user = self.model(
            firstName=firstName,
            name=name,
            registrationNumber=registrationNumber,
            sectorCode_id=sectorCode,
        )
        user.set_password(password)
        user.save()
        return user

    # …

Если вы хотите создать Sector на случай, если этот отсутствует, вы можете использовать:

class MyUserManager(BaseUserManager):
    def create_user(
        self, registrationNumber, firstName, name, sectorCode, password=None
    ):
        if not firstName:
            raise ValueError("firstName required")
        if not name:
            raise ValueError("name required")
        if not registrationNumber:
            raise ValueError("registrationNumber required")
        if not sectorCode:
            raise ValueError("sectorCode required")

        sector, __ = Sector.objects.get_or_create(
            sectorCode=sectorCode, defaults={'nameSector': sectorCode}
        )

        user = self.model(
            firstName=firstName,
            name=name,
            registrationNumber=registrationNumber,
            sectorCode=sector,
        )
        user.set_password(password)
        user.save()
        return user

    # …

Code1, однако не будет действительным, так как код сектора имеет максимальную длину в три символа.

Примечание: обычно имена полей в модели Django записываются в snake_case, а не PascalCase, поэтому должно быть: first_name вместо firstName.

Вернуться на верх