Skip to content

Commit

Permalink
ispravljeno
Browse files Browse the repository at this point in the history
  • Loading branch information
urosh985 committed May 18, 2024
1 parent 378297f commit 72aa4e3
Showing 1 changed file with 14 additions and 15 deletions.
29 changes: 14 additions & 15 deletions example/api/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,21 +25,20 @@ def createComment(request):

# Takes category string (e.g. 'movie') and returns the matching model object, serializer object and author field name (e.g. Movie, MovieSerializer and 'director')
def handleMultipleCategories(category, short=False, author=False):
match category:
case 'movie':
resModel = Movie
authorField = 'director'
if short:
resSerializer = ShortMovieSerializer
else:
resSerializer = MovieSerializer
case 'book':
resModel = Book
authorField = 'authors'
if short:
resSerializer = ShortBookSerializer
else:
resSerializer = BookSerializer
if category == 'movie':
resModel = Movie
authorField = 'director'
if short:
resSerializer = ShortMovieSerializer
else:
resSerializer = MovieSerializer
elif category == 'book':
resModel = Book
authorField = 'authors'
if short:
resSerializer = ShortBookSerializer
else:
resSerializer = BookSerializer
if author:
return resModel, resSerializer, authorField
return resModel, resSerializer
Expand Down

0 comments on commit 72aa4e3

Please sign in to comment.