def list_company(request):
"""
List all companies, or create a new company.
"""
if request.method == 'GET':
companies = Company.objects.all()
serializer = CompanySerializer(companies, many=True)
return JSONResponse(serializer.data)
elif request.method == 'POST':
data = JSONParser().parse(request)
serializer = CompanySerializer(data=data)
if serializer.is_valid():
serializer.save()
return JSONResponse(serializer.data, status=201)
return JSONResponse(serializer.errors, status=400)
def company_detail(request, pk):
"""
Retrieve, update or delete a company.
"""
try:
company = Company.objects.get(pk=pk)
except Company.DoesNotExist:
return HttpResponse(status=404)
if request.method == 'GET':
serializer = CompanySerializer(company)
return JSONResponse(serializer.data)
elif request.method == 'PUT':
data = JSONParser().parse(request)
serializer = CompanySerializer(company, data=data)
if serializer.is_valid():
serializer.save()
return JSONResponse(serializer.data)
return JSONResponse(serializer.errors, status=400)
elif request.method == 'DELETE':
company.delete()
return HttpResponse(status=204)
