File tree 2 files changed +10
-13
lines changed
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):
417
417
def test_get_serializer_comes_from_included_serializers (self ):
418
418
kwargs = {"pk" : self .author .id , "related_field" : "type" }
419
419
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" )
422
422
got = view .get_related_serializer_class ()
423
423
self .assertEqual (got , AuthorTypeSerializer )
424
-
425
- view .serializer_class .related_serializers = related_serializers
424
+ view .get_serializer_class ().related_serializers = related_serializers
426
425
427
426
def test_get_related_serializer_class_raises_error (self ):
428
427
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):
208
208
209
209
class AuthorViewSet (ModelViewSet ):
210
210
queryset = Author .objects .all ()
211
- serializer_classes = {
212
- "list" : AuthorListSerializer ,
213
- "retrieve" : AuthorDetailSerializer ,
214
- }
215
- serializer_class = AuthorSerializer # fallback
216
211
217
212
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 )
222
220
223
221
224
222
class CommentViewSet (ModelViewSet ):
You can’t perform that action at this time.
0 commit comments