project.py 557 B

1234567891011121314151617181920
  1. from django.shortcuts import get_object_or_404
  2. from rest_framework import status
  3. from rest_framework.decorators import permission_classes
  4. from rest_framework.response import Response
  5. from pycs_api.models import Project
  6. from pycs_api.serializers import ProjectSerializer
  7. from pycs_api.views.base import BaseViewSet
  8. class ProjectViewSet(BaseViewSet):
  9. serializer_class = ProjectSerializer
  10. def get_queryset(self):
  11. return self.request.user.projects
  12. def perform_create(self, serializer):
  13. serializer.save(user=self.request.user)