Skip to content

Commit

Permalink
Merge pull request #17 from dqrk0jeste/ispravljeno
Browse files Browse the repository at this point in the history
ispravljeno
  • Loading branch information
dqrk0jeste authored May 18, 2024
2 parents 378297f + 72aa4e3 commit d88d810
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 d88d810

Please sign in to comment.