File tree Expand file tree Collapse file tree 2 files changed +10
-13
lines changed Expand file tree Collapse file tree 2 files changed +10
-13
lines changed Original file line number Diff line number Diff line change @@ -417,12 +417,11 @@ def test_get_related_serializer_class_many(self):
417417    def  test_get_serializer_comes_from_included_serializers (self ):
418418        kwargs  =  {"pk" : self .author .id , "related_field" : "type" }
419419        view  =  self ._get_view (kwargs )
420-         related_serializers  =  view .serializer_class .related_serializers 
421-         delattr (view .serializer_class , "related_serializers" )
420+         related_serializers  =  view .get_serializer_class () .related_serializers 
421+         delattr (view .get_serializer_class () , "related_serializers" )
422422        got  =  view .get_related_serializer_class ()
423423        self .assertEqual (got , AuthorTypeSerializer )
424- 
425-         view .serializer_class .related_serializers  =  related_serializers 
424+         view .get_serializer_class ().related_serializers  =  related_serializers 
426425
427426    def  test_get_related_serializer_class_raises_error (self ):
428427        kwargs  =  {"pk" : self .author .id , "related_field" : "unknown" }
Original file line number Diff line number Diff line change @@ -208,17 +208,15 @@ class NoFiltersetEntryViewSet(EntryViewSet):
208208
209209class  AuthorViewSet (ModelViewSet ):
210210    queryset  =  Author .objects .all ()
211-     serializer_classes  =  {
212-         "list" : AuthorListSerializer ,
213-         "retrieve" : AuthorDetailSerializer ,
214-     }
215-     serializer_class  =  AuthorSerializer   # fallback 
216211
217212    def  get_serializer_class (self ):
218-         try :
219-             return  self .serializer_classes .get (self .action , self .serializer_class )
220-         except  AttributeError :
221-             return  self .serializer_class 
213+         serializer_classes  =  {
214+             "list" : AuthorListSerializer ,
215+             "retrieve" : AuthorDetailSerializer ,
216+         }
217+ 
218+         action  =  getattr (self , "action" , "" )
219+         return  serializer_classes .get (action , AuthorSerializer )
222220
223221
224222class  CommentViewSet (ModelViewSet ):
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments