45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
|
from rest_framework.decorators import api_view
|
||
|
|
from rest_framework.response import Response
|
||
|
|
from rest_framework import status
|
||
|
|
from .models import Project
|
||
|
|
from .serializers import ProjectSerializer, ProjectDetailSerializer
|
||
|
|
|
||
|
|
|
||
|
|
@api_view(['GET', 'POST'])
|
||
|
|
def project_list(request):
|
||
|
|
"""获取项目列表或创建新项目"""
|
||
|
|
if request.method == 'GET':
|
||
|
|
projects = Project.objects.all()
|
||
|
|
serializer = ProjectSerializer(projects, many=True)
|
||
|
|
return Response(serializer.data)
|
||
|
|
|
||
|
|
elif request.method == 'POST':
|
||
|
|
serializer = ProjectSerializer(data=request.data)
|
||
|
|
if serializer.is_valid():
|
||
|
|
serializer.save()
|
||
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
||
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
||
|
|
|
||
|
|
|
||
|
|
@api_view(['GET', 'PUT', 'DELETE'])
|
||
|
|
def project_detail(request, pk):
|
||
|
|
"""获取、更新或删除项目"""
|
||
|
|
try:
|
||
|
|
project = Project.objects.get(pk=pk)
|
||
|
|
except Project.DoesNotExist:
|
||
|
|
return Response({'error': 'Project not found'}, status=status.HTTP_404_NOT_FOUND)
|
||
|
|
|
||
|
|
if request.method == 'GET':
|
||
|
|
serializer = ProjectDetailSerializer(project)
|
||
|
|
return Response(serializer.data)
|
||
|
|
|
||
|
|
elif request.method == 'PUT':
|
||
|
|
serializer = ProjectSerializer(project, data=request.data)
|
||
|
|
if serializer.is_valid():
|
||
|
|
serializer.save()
|
||
|
|
return Response(serializer.data)
|
||
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
||
|
|
|
||
|
|
elif request.method == 'DELETE':
|
||
|
|
project.delete()
|
||
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|