Removing things from the votes __init__ file
parent
813239c601
commit
c9706122a2
|
@ -40,7 +40,9 @@ from . import serializers
|
|||
from . import models
|
||||
from . import permissions
|
||||
from . import services
|
||||
from . import votes
|
||||
from .votes.utils import attach_votescount_to_queryset
|
||||
from .votes import services as votes_service
|
||||
from .votes import serializers as votes_serializers
|
||||
|
||||
|
||||
class ProjectAdminViewSet(ModelCrudViewSet):
|
||||
|
@ -50,7 +52,7 @@ class ProjectAdminViewSet(ModelCrudViewSet):
|
|||
|
||||
def get_queryset(self):
|
||||
qs = models.Project.objects.all()
|
||||
qs = votes.attach_votescount_to_queryset(qs, as_field="stars_count")
|
||||
qs = attach_votescount_to_queryset(qs, as_field="stars_count")
|
||||
return qs
|
||||
|
||||
def pre_save(self, obj):
|
||||
|
@ -71,7 +73,7 @@ class ProjectViewSet(ModelCrudViewSet):
|
|||
|
||||
def get_queryset(self):
|
||||
qs = models.Project.objects.all()
|
||||
qs = votes.attach_votescount_to_queryset(qs, as_field="stars_count")
|
||||
qs = attach_votescount_to_queryset(qs, as_field="stars_count")
|
||||
qs = qs.filter(Q(owner=self.request.user) |
|
||||
Q(members=self.request.user))
|
||||
return qs.distinct()
|
||||
|
@ -84,13 +86,13 @@ class ProjectViewSet(ModelCrudViewSet):
|
|||
@detail_route(methods=['post'], permission_classes=(IsAuthenticated,))
|
||||
def star(self, request, pk=None):
|
||||
project = self.get_object()
|
||||
votes.add_vote(project, user=request.user)
|
||||
votes_service.add_vote(project, user=request.user)
|
||||
return Response(status=status.HTTP_200_OK)
|
||||
|
||||
@detail_route(methods=['post'], permission_classes=(IsAuthenticated,))
|
||||
def unstar(self, request, pk=None):
|
||||
project = self.get_object()
|
||||
votes.remove_vote(project, user=request.user)
|
||||
votes_service.remove_vote(project, user=request.user)
|
||||
return Response(status=status.HTTP_200_OK)
|
||||
|
||||
@detail_route(methods=['get'])
|
||||
|
@ -328,13 +330,13 @@ class ProjectTemplateViewSet(ModelCrudViewSet):
|
|||
|
||||
|
||||
class FansViewSet(ModelCrudViewSet):
|
||||
serializer_class = votes.serializers.VoterSerializer
|
||||
list_serializer_class = votes.serializers.VoterSerializer
|
||||
serializer_class = votes_serializers.VoterSerializer
|
||||
list_serializer_class = votes_serializers.VoterSerializer
|
||||
permission_classes = (IsAuthenticated,)
|
||||
|
||||
def get_queryset(self):
|
||||
project = models.Project.objects.get(pk=self.kwargs.get("project_id"))
|
||||
return votes.get_voters(project)
|
||||
return votes_service.get_voters(project)
|
||||
|
||||
|
||||
class StarredViewSet(ModelCrudViewSet):
|
||||
|
@ -343,4 +345,4 @@ class StarredViewSet(ModelCrudViewSet):
|
|||
permission_classes = (IsAuthenticated,)
|
||||
|
||||
def get_queryset(self):
|
||||
return votes.get_voted(self.kwargs.get("user_id"), model=models.Project)
|
||||
return votes_service.get_voted(self.kwargs.get("user_id"), model=models.Project)
|
||||
|
|
|
@ -29,7 +29,9 @@ from taiga.base.api import ModelCrudViewSet
|
|||
|
||||
from taiga.projects.mixins.notifications import NotificationSenderMixin
|
||||
|
||||
from .. import votes
|
||||
from taiga.projects.votes.utils import attach_votescount_to_queryset
|
||||
from taiga.projects.votes import services as votes_service
|
||||
from taiga.projects.votes import serializers as votes_serializers
|
||||
from . import models
|
||||
from . import permissions
|
||||
from . import serializers
|
||||
|
@ -116,7 +118,7 @@ class IssueViewSet(NotificationSenderMixin, ModelCrudViewSet):
|
|||
|
||||
def get_queryset(self):
|
||||
qs = self.model.objects.all()
|
||||
qs = votes.attach_votescount_to_queryset(qs, as_field="votes_count")
|
||||
qs = attach_votescount_to_queryset(qs, as_field="votes_count")
|
||||
return qs
|
||||
|
||||
def pre_save(self, obj):
|
||||
|
@ -149,21 +151,21 @@ class IssueViewSet(NotificationSenderMixin, ModelCrudViewSet):
|
|||
@detail_route(methods=['post'], permission_classes=(IsAuthenticated,))
|
||||
def upvote(self, request, pk=None):
|
||||
issue = self.get_object()
|
||||
votes.add_vote(issue, user=request.user)
|
||||
votes_service.add_vote(issue, user=request.user)
|
||||
return Response(status=status.HTTP_200_OK)
|
||||
|
||||
@detail_route(methods=['post'], permission_classes=(IsAuthenticated,))
|
||||
def downvote(self, request, pk=None):
|
||||
issue = self.get_object()
|
||||
votes.remove_vote(issue, user=request.user)
|
||||
votes_service.remove_vote(issue, user=request.user)
|
||||
return Response(status=status.HTTP_200_OK)
|
||||
|
||||
|
||||
class VotersViewSet(ModelCrudViewSet):
|
||||
serializer_class = votes.serializers.VoterSerializer
|
||||
list_serializer_class = votes.serializers.VoterSerializer
|
||||
serializer_class = votes_serializers.VoterSerializer
|
||||
list_serializer_class = votes_serializers.VoterSerializer
|
||||
permission_classes = (IsAuthenticated,)
|
||||
|
||||
def get_queryset(self):
|
||||
issue = models.Issue.objects.get(pk=self.kwargs.get("issue_id"))
|
||||
return votes.get_voters(issue)
|
||||
return votes_service.get_voters(issue)
|
||||
|
|
|
@ -1,7 +0,0 @@
|
|||
from .services import add_vote
|
||||
from .services import remove_vote
|
||||
from .services import get_voters
|
||||
from .services import get_votes
|
||||
from .services import get_voted
|
||||
from .utils import attach_votescount_to_queryset
|
||||
from . import serializers
|
Loading…
Reference in New Issue