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