def show(category_id):
schema = CategorySchema()
category = Category.get(id=category_id)
if not category:
abort(404)
return schema.dumps(category)
def load_categories(self, data):
data['categories'] = [
Category.get(id=category_id)
for category_id in data['category_ids']
]
del data['category_ids']
return data
def find_dish(data):
field = data['field']
search = data['search']
if field == 'category':
category = Category.get(Category.title == search)
search = category.id
query_dish = Dish.select().where(get_model_field(Dish, field) == search)
return transform_query_to_array(query_dish)
def delete(category_id):
category = Category.get(id=category_id)
if not category:
abort(404)
category.delete()
db.commit()
return '', 204
def update(category_id):
schema = CategorySchema()
category = Category.get(id=category_id)
if not category:
abort(404)
try:
data = schema.load(request.get_json())
category.set(**data)
db.commit()
except ValidationError as err:
return jsonify({
'message': 'Validation failed',
'errors': err.messages
}), 422
return schema.dumps(category)
def get_categories(category_id):
schema = CategorySchema()
categories = Category.get(id=category_id)
return schema.dumps(categories)
