Come aggiornare i valori in istanza quando si dispone di un custom.update() per aggiornare le relazioni molti-a-molti in serializzatore annidato scrivibile DRF

Question

Ho tre modelli Giocatore, squadra e appartenenza, dove giocatore e squadra hanno molti-to- molte relazioni che utilizzano l'abbonamento come modello intermedio.

class Player(models.Model): 
    name = models.CharField(max_length=254) 
    rating = models.FloatField(null=True) 
    install_ts = models.DateTimeField(auto_now_add=True, blank=True) 
    update_ts = models.DateTimeField(auto_now_add=True, blank=True) 


class Team(models.Model): 
    name = models.CharField(max_length=254) 
    rating = models.FloatField(null=True) 
    players = models.ManyToManyField(
      Player, 
      through='Membership', 
      through_fields=('team', 'player')) 
    is_active = models.BooleanField(default=True) 
    install_ts = models.DateTimeField(auto_now_add=True, blank=True) 
    update_ts = models.DateTimeField(auto_now_add=True, blank=True) 


class Membership(models.Model): 
    team = models.ForeignKey('Team') 
    player = models.ForeignKey('Player') 
    #date_of_joining = models.DateTimeField() 
    install_ts = models.DateTimeField(auto_now_add=True, blank=True) 
    update_ts = models.DateTimeField(auto_now_add=True, blank=True) 

Ora mi è stato richiesto di aggiornare questo abbonamento utilizzando il framework django rest. Ho provato ad aggiornare quelli usando Writable nested serializers scrivendo un custom .update() di serializzatore di squadra.

@transaction.atomic 
def update(self, instance, validated_data): 
    ''' 
    Cutomize the update function for the serializer to update the 
    related_field values. 
    ''' 

    if 'memberships' in validated_data: 
     instance = self._update_membership(instance, validated_data) 

     # remove memberships key from validated_data to use update method of 
     # base serializer class to update model fields 
     validated_data.pop('memberships', None) 

    return super(TeamSerializer, self).update(instance, validated_data) 


def _update_membership(self, instance, validated_data): 
    ''' 
    Update membership data for a team. 
    ''' 
    memberships = self.initial_data.get('memberships') 
    if isinstance(membership, list) and len(memberships) >= 1: 
     # make a set of incoming membership 
     incoming_player_ids = set() 

     try: 
      for member in memberships: 
       incoming_player_ids.add(member['id']) 
     except: 
      raise serializers.ValidationError(
       'id is required field in memberships objects.' 
      ) 

     Membership.objects.filter(
      team_id=instance.id 
     ).delete() 

     # add merchant member mappings 
     Membership.objects.bulk_create(
      [ 
       Membership(
        team_id=instance.id, 
        player_id=player 
       ) 
       for player in incoming_player_ids 
      ] 
     ) 
     return instance 
    else: 
     raise serializers.ValidationError(
       'memberships is not a list of objects' 
      ) 

Ora questo funziona bene per l'aggiornamento dei valori nel database per la tabella di appartenenza. L'unico problema che ho di fronte è che non sono in grado di aggiornare l'istanza prefetched nella memoria, che sulla richiesta PATCH a questa API aggiorna i valori nel database ma la risposta API mostra dati obsoleti.

La successiva richiesta di GET per la stessa risorsa fornisce dati aggiornati. Chiunque abbia lavorato con molti-a-molti in django e abbia scritto metodi di aggiornamento/creazione personalizzati per serializzatori nidificati scrivibili può aiutarmi a capire il modo possibile per risolvere questo problema.

Answer 1

In questo caso sono stato in grado di rintracciare lo issue. Stavo usando .prefetch_related() su queryset che sta causando l'istanza di utilizzare i dati prefetched per molte a molte relazioni. Ho due soluzioni che possono causare altri colpi al database per recuperare i dati aggiornati.

1. Non utilizzare .prefetch_related()

Un modo evidente è di non usare .prefetch_related() Che è non raccomandato in quanto si tradurrà in elevato numero di DB colpisce.

O

2. Recupero esempio modello aggiornato dopo l'aggiornamento molti-a-molti rapporti nel metodo di aggiornamento di serializzatore

instance = self.context['view'].get_queryset().get(
    id=instance.id 
) 

Modificato .update() di TeamSerializer

@transaction.atomic 
def update(self, instance, validated_data): 
    ''' 
    Cutomize the update function for the serializer to update the 
    related_field values. 
    ''' 

    if 'memberships' in validated_data: 
     instance = self._update_membership(instance, validated_data) 

     # remove memberships key from validated_data to use update method of 
     # base serializer class to update model fields 
     validated_data.pop('memberships', None) 

     # fetch updated model instance after updating many-to-many relations 
     instance = self.context['view'].get_queryset().get(
      id=instance.id 
     ) 
    return super(TeamSerializer, self).update(instance, validated_data) 

Se qualcuno ha una migliore approccio mi piacerebbe se possano aggiungere una risposta a questa domanda.

Answer 2

Penso che sia necessario "ricaricare" il campo giocatori con instance.players.all(), perché il set di query è già stato valutato nel metodo is_valid del serializzatore.

@transaction.atomic 
def update(self, instance, validated_data): 
    ''' 
    Cutomize the update function for the serializer to update the 
    related_field values. 
    ''' 

    if 'memberships' in validated_data: 
     instance = self._update_membership(instance, validated_data) 

     # remove memberships key from validated_data to use update method of 
     # base serializer class to update model fields 
     validated_data.pop('memberships', None) 
     instance.players.all() 

    return super(TeamSerializer, self).update(instance, validated_data)