def scenario(request, scen_id):
"""Retrieve, update or delete a scenario"""
scenario = get_object_or_404(Scenario, id=scen_id)
if request.method == 'GET':
if (scenario.project.user.id != request.user.id and
scenario.project.is_private):
return Response(status=status.HTTP_404_NOT_FOUND)
serializer = ScenarioSerializer(scenario)
return Response(serializer.data)
elif scenario.project.user.id == request.user.id:
if request.method == 'PUT':
ctx = {'request': request}
serializer = ScenarioSerializer(scenario, data=request.data,
context=ctx)
if serializer.is_valid():
serializer.save()
return Response(serializer.data)
return Response(serializer.errors,
status=status.HTTP_400_BAD_REQUEST)
elif request.method == 'DELETE':
scenario.delete()
return Response(status=status.HTTP_204_NO_CONTENT)
else:
return Response(status=status.HTTP_404_NOT_FOUND)
def scenario(request, scen_id):
"""Retrieve, update or delete a scenario"""
scenario = get_object_or_404(Scenario, id=scen_id)
if request.method == 'GET':
if (scenario.project.user.id != request.user.id and
scenario.project.is_private):
return Response(status=status.HTTP_404_NOT_FOUND)
serializer = ScenarioSerializer(scenario)
return Response(serializer.data)
elif scenario.project.user.id == request.user.id:
if request.method == 'PUT':
ctx = {'request': request}
serializer = ScenarioSerializer(scenario, data=request.data,
context=ctx)
if serializer.is_valid():
serializer.save()
return Response(serializer.data)
return Response(serializer.errors,
status=status.HTTP_400_BAD_REQUEST)
elif request.method == 'DELETE':
scenario.delete()
return Response(status=status.HTTP_204_NO_CONTENT)
else:
return Response(status=status.HTTP_404_NOT_FOUND)
def scenarios(request):
"""Create a scenario for projects which authenticated user has access to"""
if request.method == 'POST':
serializer = ScenarioSerializer(data=request.data,
context={"request": request})
project_id = serializer.initial_data.get('project')
get_object_or_404(Project, id=project_id, user=request.user)
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)
def scenarios(request):
"""Create a scenario for projects which authenticated user has access to"""
if request.method == 'POST':
serializer = ScenarioSerializer(data=request.data,
context={"request": request})
project_id = serializer.initial_data.get('project')
get_object_or_404(Project, id=project_id, user=request.user)
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)
