Поле «Сотрудники» - это список пользователей, которые являются сотрудниками данного пользователя. Каждый пользователь может иметь каждого пользователя в качестве сотрудника, пользователь также может быть сотрудником и иметь сотрудников.
У меня проблема, когда он автоматически добавляет пользователя в качестве сотрудника ко всем сотрудникам, добавленным к пользователю. Поэтому, когда вы добавляете пользователя 2 в качестве сотрудника к пользователю 1, он также добавляет пользователя 1 в качестве сотрудника пользователя 2.
Как я могу сделать так, чтобы он не менял другие записи? Я просто хочу добавить сотрудников к добавляемому пользователю.
Я использую Django REST 2.2. Поле сотрудников является полем ManyToMany, это правильно для моего использования?
введите описание изображения здесь
users / models.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;
users / serializers.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