2017-03-09 8 views
1

入れ子になったオブジェクトを更新する際に問題があります。 は、だから私はこの1つに類似した構造の二つのモデルがあります:完璧な作品を作成し、更新が動作していない、今までDjango-Rest-Framework。 2つのテーブルを更新する

class EmployeeEventUserSerializer(serializers.ModelSerializer): 
class Meta: 
    model = EmployeeEvent_Users 
    fields = ('id', 'employee',) 

class EmployeeEventSerializer(serializers.ModelSerializer): 
employee_event_users = EmployeeEventUserSerializer(required=True, many=True) 

class Meta: 
    model = EmployeeEvent 
    fields = ('event_name', 'start_date', 'end_date', 'employee_event_users',) 

def create(self, validated_data): 
    request = self.context.get('request') 
    employee_event = EmployeeEvent.objects.create(
     event_name=validated_data['event_name'], 
     start_date=validated_data['start_date'], 
     end_date=validated_data['end_date'], 
     # etc ... 
    ) 
    employeeEventUsers_data = validated_data.pop('employee_event_users', []) 

    for employeeEventUser_data in employeeEventUsers_data: 
     employeeEvent_users = EmployeeEvent_Users.objects.create(
      employee_event=employee_event, 
      employee=employeeEventUser_data['employee'], 

     ) 
    employee_event.employee_event_users = employeeEventUsers_data; 
    return employee_event 


def update(self, instance, validated_data): 

    instance.event_name = validated_data['event_name'] 
    instance.start_date = validated_data['start_date'] 
    instance.end_date = validated_data['end_date'] 
    instance.save() 

    instance.employee_event_users = validated_data.get('employee_event_users') 


    if instance.employee_event_users: 
     for employee_event_user in instance.employee_event_users: 
      employee_event_user_id = employee_event_user.get('employee_id', None) 
      if employee_event_user_id: 
       emp_user = EmployeeEvent_Users.objects.get(id=employee_event_user_id, employee_event=instance) 
       emp_user.employee = employee_event_user.get('employee', emp_user.employee) 
       emp_user.save() 
      else: 
       EmployeeEvent_Users.objects.create(employee_event=instance, **employee_event_user) 

    return instance 

class EmployeeEvent(models.Model): 
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) 
event_name = models.CharField(max_length=200) 
start_date = models.DateField() 
end_date = models.DateField() 


class EmployeeEvent_Users(models.Model): 
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) 
employee_event = models.ForeignKey(EmployeeEvent, on_delete=models.CASCADE, related_name='employee_event_employee_list') 

employee = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='employeeEvent_employee') 

を私は次のシリアライザを書いてきました。 EmployeeEventテーブルが正しく更新されていますが、EmployeeEvent_Userテーブルが更新されていません。私は以下の論理は正しいと思うが、何かが間違っている。どんな助けでも本当に感謝しています。誰も助けることができますか?

答えて

0

は、この更新方法は、任意のより良いあなたのために働くかどうかを確認してください:

def update(self, instance, validated_data): 
    instance.event_name = validated_data['event_name'] 
    instance.start_date = validated_data['start_date'] 
    instance.end_date = validated_data['end_date'] 

    # When updating children, I find it easier to delete them all 
    # and then re-add them instead of checking each one. It probably 
    # is not very efficient, however for small lists it should not 
    # matter much. 
    for emp in instance.employee_event_users.all(): 
     emp.delete() 

    # Now go through and re-add the EmployeeEvent_User 
    for employee_event_user in validated_data['employee_event_users']: 
     user = EmployeeEvent_Users.objects.create(employee_event=instance, **employee_event_user) 
     instance.employee_event_users.add(user) 

    instance.save() 
    return instance 

2つのことに注意すること:私は最後までinstance.save()を移動しました。テストケースでは、従業員を追加/編集する前にインスタンスを保存していました。次に、あなたのテストケースでは、辞書型のオブジェクトをemployee_event_usersに割り当てていました。実際のオブジェクトではありません。テストケースを変更して、新しいユーザーを作成し、リストに追加してから保存します(子供を削除してから再追加しない場合)。私は「『EmployeeEvent』オブジェクトには属性 『employee_event_users』がありません」というエラー を取得しています、上記のコードを試しながら

def update(self, instance, validated_data): 
    instance.event_name = validated_data['event_name'] 
    instance.start_date = validated_data['start_date'] 
    instance.end_date = validated_data['end_date'] 

    for employee_event_user in validated_data['employee_event_users']: 
     employee_event_user_id = employee_event_user.get('employee_id', None) 
     if employee_event_user_id: 
      emp_user = EmployeeEvent_Users.objects.get(id=employee_event_user_id, employee_event=instance) 
      emp_user.employee = employee_event_user.get('employee', emp_user.employee) 
      emp_user.save() 
     else: 
      user = EmployeeEvent_Users.objects.create(employee_event=instance, **employee_event_user) 
      user.save() 
      instance.employee_event_users.add(user) 

    instance.save() 
    return instance 
+0

:ここではそれは次のようになります。 どのような解決策ですか? –

+0

はい、あなたの 'serializers.py'では' employee_event_users'を 'employeeEvent_employee'に置き換えます。私はあなたのためにそれを解決すると思う。 – themanatuf

関連する問題