Skip to content

Commit 89a6f59

Browse files
authored
Replace self.queryset with Viewset method equivalents (#84)
1 parent 06893fb commit 89a6f59

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vote/views.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
class VoteMixin:
88
def get_instance(self, pk):
9-
return self.queryset.get(pk=pk)
9+
return self.get_queryset().get(pk=pk)
1010

1111
@action(detail=True, methods=('post', 'delete'))
1212
def vote(self, request, pk):
@@ -16,7 +16,7 @@ def vote(self, request, pk):
1616
action = request.data.get('action', 'up')
1717
voted = getattr(obj.votes, action)(user_id)
1818
if voted:
19-
post_voted.send(sender=self.queryset.model,
19+
post_voted.send(sender=self.get_serializer_class().Meta.model,
2020
obj=obj,
2121
user_id=user_id,
2222
action=action)
@@ -25,7 +25,7 @@ def vote(self, request, pk):
2525
else:
2626
deleted = obj.votes.delete(user_id)
2727
if deleted:
28-
post_voted.send(sender=self.queryset.model,
28+
post_voted.send(sender=self.get_serializer_class().Meta.model,
2929
obj=obj,
3030
user_id=user_id,
3131
action='delete')

0 commit comments

Comments
 (0)