def school(id):
'''
GET: Returns school.
PUT: Updates school.
DELETE: Deletes school.
'''
school = School.query.get_or_404(id)
if request.method == 'PUT':
json = request.get_json()
address = Address.create_or_get(json['address']) if 'address' in json else None
school.name=json.get('name', '')
school.address=address
school.website=json.get('website', '')
school.category=json.get('category', '')
db.session.add(school)
db.session.commit()
return jsonify(school.json), status.HTTP_200_OK
elif request.method == 'DELETE':
db.session.delete(school)
db.session.commit()
return '', status.HTTP_204_NO_CONTENT
return jsonify(school.json), status.HTTP_200_OK
def company(id):
'''
GET: Returns company.
PUT: Updates company.
DELETE: Deletes company.
'''
company = Company.query.get_or_404(id)
if request.method == 'PUT':
json = request.get_json()
address = Address.create_or_get(json['address']) if 'address' in json else None
industry = Industry.create_or_get(json['industry']) if 'industry' in json else None
company.name=json.get('name', '')
company.address=address
company.industry=industry
company.website=json.get('website', '')
db.session.add(company)
db.session.commit()
return jsonify(company.json), status.HTTP_200_OK
elif request.method == 'DELETE':
db.session.delete(company)
db.session.commit()
return '', status.HTTP_204_NO_CONTENT
return jsonify(company.json), status.HTTP_200_OK
