Я хочу присвоить значение user_company вошедшего пользователя полю user_company вновь созданного пользователя.

Когда пользователь создает запись пользователя для клиента, новый клиент должен иметь значение User.user_company текущего вошедшего пользователя.

В данной проблеме я хочу присвоить значение User.user_company вошедшего пользователя новому пользователю clientuser.user_company при вызове save() в представлении.

ниже представлен сериализатор с объектом clientuser.


class ClientSerializers(serializers.ModelSerializer):

    password2 = serializers.CharField(style={'input_type': 'password'}, write_only=True)
    client_name = serializers.CharField(style={'input_type' : 'text'}, required=True)


    class Meta:
        model = User
        fields = ['email', 'username', 'password', 'password2', 'user_type', 'client_name'] 
        extra_kwargs = {
                'password': {'write_only': True}, #dont want anyone to see the password
                'user_type': {'read_only': True},
        }

    
    def save(self):
        clientuser = User(
                    #creating a user record. it will record company fk
                    email=self.validated_data['email'],
                    username=self.validated_data['username'],
                    user_type = 3,
                    first_name = self.validated_data['client_name'])

        
        #validating the password
        password = self.validated_data['password']
        password2 = self.validated_data['password2']
        if password != password2:   #trying to match passwords.
            raise serializers.ValidationError({'password': 'Passwords must match.'})
        
        clientuser.set_password(password) #setting the password
        clientuser.save() #saving the user
        return clientuser

Я пробовал использовать

cur_usr = User()
param = 'user_company'
usr_comp = getattr(u, param)
print(f'usr_comp is {usr_comp})
Оператор

print печатает usr_comp is None в терминале

Я также пробовал использовать

curr_User_Company = User.user_company
user.user_company = curr_User_Company

в терминале появляется следующая строка

raise ValueError(
ValueError: Cannot assign "<django.db.models.fields.related_descriptors.ForwardManyToOneDescriptor object at 0x04B05F28>": "User.user_company" must be a "Company" instance.

Вот моя модель пользователя

class User(AbstractUser):
    email = models.EmailField(verbose_name="email", max_length=60, unique=True)
    user_type_data = ((1,"sysAdmin"),(2,"CompanyAdmin"), (3,"Client"), (4,"Employee"))
    user_type = models.IntegerField(choices=user_type_data, default=2)
    user_company = models.ForeignKey('Company', on_delete=models.CASCADE, null=True, blank=True)
    #if user is CompAdmin then company is the company he belongs to
    #if user is Client then company is the company he is serviced by
    #if user is Employee then company is the company he works for
    #if user is sysAdmin then company is null

Вот мое мнение

@csrf_exempt
@permission_classes([IsAuthenticated])
def ClientApi(request):
    if request.method == 'POST':
        data = JSONParser().parse(request)
        serializer = ClientSerializers(data=data)
        if serializer.is_valid():
            serializer.save()
            return JsonResponse(serializer.data, status=201)
        return JsonResponse(serializer.errors, status=400)

Я не уверен, нужно ли это, но если да, то вот пример данных, которые я пытался передать через Postman

{
        "email":"Maximillian@amgracing.com", 
        "username":"Maximillian", 
        "password":"AmgRacingBetterThanRedBull", 
        "password2":"AmgRacingBetterThanRedBull", 
        "client_name" : "MaximillianRacing" 
    }

Можно использовать следующее:

from django.contrib.auth import get_user

и это: 

 clientuser = User(
      #creating a user record. it will record company fk
      email=self.validated_data['email'],
      username=self.validated_data['username'],
      ///

      user_company = get_user(self.request).user_company /// this line get you user_company from logged user

Наконец-то я разобрался с этим. Я добавил параметр current_user в функцию save() и присвоил значение user_company в current_user.user_company

    def save(self, current_user):
        usr_comp = current_user.user_company
        clientUser = User(///
                          user_company = usr_comp)

В представлении ClientApi вот что я сделал

serializer.save(current_user = request.user)
Вернуться на верх