Как добавить дополнительные аргументы командной строки в createsuperuser - Django?

Ниже представлен класс UserManager:

class UserManager(BaseUserManager):

    def create_user(self, email, password, **extra_fields):
        if not email:
            raise ValueError(_('The Email must be set'))
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)        
        user.set_password(password)
        user.save()
        return user

    def create_superuser(self, email, password, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        extra_fields.setdefault('is_active', True)

        if extra_fields.get('is_staff') is not True:
            raise ValueError(_('Superuser must have is_staff=True.'))
        if extra_fields.get('is_superuser') is not True:
            raise ValueError(_('Superuser must have is_superuser=True.'))
        return self.create_user(email, password, **extra_fields)

В функции create_superuser я хотел бы добавить еще одно поле для установки tenant_id. Это поле должно быть взято из командной строки при использовании python manage.py createsuperuser. Если я добавляю tenant_id в функцию create_superuser следующим образом create_superuser(self, email, password, tenant_id,**extra_fields),, то получаю следующую ошибку UserManager.create_superuser() missing 1 required positional argument: 'tenant_id'. Есть ли способ передать tenant_id из командной строки?

Пробовали ли вы добавить его к созданию пользователя, например, так:

 class UserManager(BaseUserManager):
        def create_user(self, tenant_id email, password, **extra_fields):
            if not email:
                raise ValueError(_("The Email must be set"))
            email = self.normalize_email(email)
            user = self.model(email=email, **extra_fields)
            user.set_password(password)
            user.save()
            return user
    
        def create_superuser(self, tenant_id email, password, **extra_fields):
            extra_fields.setdefault("is_staff", True)
            extra_fields.setdefault("is_superuser", True)
            extra_fields.setdefault("is_active", True)
    
            if extra_fields.get("is_staff") is not True:
                raise ValueError(_("Superuser must have is_staff=True."))
            if extra_fields.get("is_superuser") is not True:
                raise ValueError(_("Superuser must have is_superuser=True."))
            return self.create_user(tenant_id email, password, **extra_fields)

и в models.py вставьте эти строки:

REQUIRED_FIELDS = ['tenant_id', 'email', 'password']

objects = MemberManager()
Вернуться на верх