コード例 #1
0
ファイル: views.py プロジェクト: jainam1995/vehicle-webapp
def theft_detail(request, pk):
    try:
        theft_report = TheftReport.objects.get(pk=pk)
    except TheftReport.DoesNotExist:
        return Response(status=status.HTTP_404_NOT_FOUND)

    if request.method == 'GET':
        serializer = TheftReportSerializer(theft_report)
        return Response(serializer.data)

    elif request.method == 'PUT':
        serializer = TheftReportSerializer(theft_report, data=request.data)
        if serializer.is_valid():
            serializer.save(reporter=request.user)
            return Response(serializer.data)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

    elif request.method == 'DELETE':
        TheftReport.delete(theft_report)
        return Response(status=status.HTTP_204_NO_CONTENT)
コード例 #2
0
ファイル: views.py プロジェクト: jainam1995/vehicle-webapp
def theft_report(request):
    if request.method == 'GET':
        theft_reports = TheftReport.objects.filter(reporter=request.user)
        serializer = TheftReportSerializer(theft_reports, many=True)
        return Response(serializer.data)

    elif request.method == 'POST':
        serializer = TheftReportSerializer(data=request.data)
        if serializer.is_valid():
            #check if vehicle exists in db
            vehicles_student = StudentVehicle.objects.filter(vehicle_registration_number=serializer.data['vehicle_pass_no'])
            vehicles_emp = EmployeeVehicle.objects.filter(vehicle_registration_number=serializer.data['vehicle_pass_no'])
            if(len(vehicles_student) > 0):
                serializer.save(reporter=request.user, stud_vehicle=vehicles_student[0])
            elif(len(vehicles_emp) > 0):
                serializer.save(reporter=request.user, emp_vehicle=vehicles_emp[0])
            else:
                #vehicle does not exist stud_vehicle and emp_vehicle are left blank
                serializer.save(reporter=request.user)
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)