def update(event, context): id = event['pathParameters']['id'] location = LocationModel.get(id) if location is None: raise Exception('Not found') return data = json.loads(event['body']) try: location.name = data.get('name', location.name) location.street_1 = data.get('street_1', location.street_1) location.street_2 = data.get('street_2', location.street_2) location.city = data.get('city', location.city) location.state = data.get('state', location.state) location.zip_code = data.get('zip_code', location.zip_code) location.type_id = data.get('type_id', location.type_id) location.save() response = {'statusCode': 200, 'body': json.dumps(dict(location))} except Exception as ex: response = { 'statusCode': 500, 'body': json.dumps({'errorMessage': 'There was a problem saving'}) } return response
def get(event, context): id = event['pathParameters']['id'] try: location = LocationModel.get(id) response = {'statusCode': 200, 'body': json.dumps(dict(location))} except Exception as ex: response = { 'statusCode': 500, 'body': json.dumps({'errorMessage': 'There was a problem deleting'}) } return response