Example #1
0
    def post(self):
        data = request.get_json(force=True)

        if not data:
            return {'message': 'No data provided'}, 400

        if data['ts_id']:
            if TripStatusModel.find_by_newBudget(data['ts_id'],
                                                 data['trip_id'],
                                                 data['owner_id']):

                TripStatusModel.update_tableforSetBudget(
                    data['ts_id'], data['trip_id'], data['budget'])

                trip_status_id = TripStatusModel.get_tripstatus_idbyRecord(
                    data['trip_id'], data['owner_id'], data['budget'])

                return {
                    'message': 'Trip budget is set',
                    'trip_status_id': trip_status_id
                }
        else:
            new_entry = TripStatusModel(trip_id=data['trip_id'],
                                        trip_budget=data['budget'],
                                        owner_id=data['owner_id'])
            try:
                new_entry.save_to_db()
                trip_status_id = TripStatusModel.get_tripstatus_idbyRecord(
                    data['trip_id'], data['owner_id'], data['budget'])
                return {
                    'message': 'Trip budget is assinged',
                    'trip_status_id': trip_status_id
                }
            except Exception as e:
                return {'message': 'Something went wrong', 'error': e}
Example #2
0
    def post(self):
        data = request.get_json(force=True)

        if not data:
            return {'message': 'No data provided'}, 400

        new_trip_status = TripStatusModel(
            trip_id=data['trip_id'],
            trip_budget=data['trip_budget'],
            assigned_driver=data['assigned_driver'],
            is_confirmed_passenger=data['is_confirmed_passenger'],
            is_confirmed_driver=data['is_confirmed_driver'],
            trip_started=data['trip_started'],
            vehicle_no=data['vehicle_no'])

        try:
            new_trip_status.save_to_db()
            return {
                'message':
                'Trip Status {} created for '.format(data['trip_id'])
            }
        except Exception as e:
            return {
                'message': 'Something went wrong',
                'error': e,
                'data': new_trip_status
            }, 500