0

员工字段是用户的员工列表。每个用户都可以将每个用户都作为员工，用户也可以是员工并拥有员工。

我有一个问题，它会自动将用户作为员工添加到添加到用户的所有员工中。因此，当您将用户 2 作为员工添加到用户 1 时，它也会将用户 1 添加为用户 2 的员工。

我怎样才能使它不改变其他记录？我只是想将员工添加到正在添加的用户中。

我正在使用 Django REST 2.2。employees 字段是 ManyToMany 字段，这对我的用法是否正确？

在此处输入图像描述

在此处输入图像描述

用户/模型.py

class MyUserManager(BaseUserManager):
def create_user(self, email, password=None):
    if not email:
        raise ValueError('Users must have an email address')

    user = self.model(
        email=self.normalize_email(email),
    )

    user.set_password(password)
    user.save(using=self._db)
    return user

def create_superuser(self, email, password):
    user = self.create_user(
        email=self.normalize_email(email),
        password=password,
    )
    user.is_admin = True
    user.is_staff = True
    user.is_superuser = True
    user.save(using=self._db)
    return user

class CustomUser(AbstractBaseUser):
    username = None
    email = models.EmailField(verbose_name='email address', max_length=255, unique=True)
    date_joined = models.DateTimeField(verbose_name='date joined', auto_now_add=True)
    last_login = models.DateTimeField(verbose_name='last login', auto_now=True)
    is_admin = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    is_superuser = models.BooleanField(default=False)

    is_employee = models.BooleanField('is_employee', default=False)
    employees = models.ManyToManyField("self", related_name='employees', blank=True)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    objects = MyUserManager()

    def __str__(self):
        return self.email

    def has_perm(self, perm, obj=None):
        return self.is_admin

    def has_module_perms(selfself, app_label):
        return True;

用户/序列化程序.py

class CustomUserSerializer(serializers.ModelSerializer):

class Meta:
    model = CustomUser
    fields = ('id', 'email', 'password', 'is_employee', 'employees')

def create(self, validated_data):
    validated_data['password'] = make_password(validated_data.get('password'))
    return super(CustomUserSerializer, self).create(validated_data)

def update(self, instance, validated_data):
    instance.email = validated_data.get('email')
    instance.set_password(validated_data.get('password', instance.password))
    instance.is_employee = validated_data.get('is_employee')
    instance.employees.set(validated_data.get('employees'))
    instance.save()
    return instance
4

2 回答 2

1

尝试设置ManyToManyField.symmetrical为False

于 2020-07-12T12:27:37.137 回答
0

ManyToMany关系对你的情况是错误的。因为一个用户不能是多个雇主的雇员，只能是一个雇主的雇员。

因此，您应该简单地创建一个多对一关系：

class CustomUser(AbstractBaseUser):
    ...
    employer = models.ForeignKey('self', related_name='employees', on_delete=models.SET_NULL)

因此，每个用户将有一个employer字段，并进行employees设置。而且您可能可以摆脱is_employee字段，因为您可以通过检查是否employer设置了字段来检查用户是否是员工。

于 2020-07-11T22:08:17.987 回答