def post(self):
'''Creates a new Category '''
data = request.get_json()
categoryName = data.get('category_name')
categoryDesc = data.get('category_description')
catName_validator = name_validate(categoryName)
catDesc_validator = name_validate(categoryDesc)
if catName_validator is False:
return {"message": "category name cannot be blank"}, 422
if catDesc_validator is False:
return {
"message":
"category description cannot be blank, please"
" enter a valid description"
}, 422
user_id = get_jwt_identity()
if Category.query.filter_by(
user_id=user_id,
category_name=categoryName).first() is not None:
return {'message': 'Category already exists'}, 409
new_cat = Category(categoryName, categoryDesc, user_id)
new_cat.save()
return {
'category_id': new_cat.category_id,
'message': 'Category successfully created'
}, 201
def add(title, desc, father, icon):
assert isinstance(title, basestring), 'title is not string'
assert isinstance(desc, basestring), 'title is not string'
assert len(title) > 0, 'title name too short!'
assert len(desc) > 5, 'desc name too short!'
assert father is None or isinstance(father, ObjectId)
if father:
assert CategoryHelper.get(father)
new_category = Category()
new_category.title = title
new_category.desc = desc
if father:
new_category.father = father
if icon:
encoded = b64encode(icon.read())
new_category.icon.new_file()
new_category.icon.write(encoded)
new_category.icon.close()
new_category.save()
if father:
CategoryHelper.add_son(father, new_category.id)
return new_category
def create_category():
"""Create a category"""
name = request.json.get("name", '')
if not name:
return bad_request(405)
category = Category(name=name)
category.save()
return jsonify(category.to_dict()), 201
def dbcreate():
db.create_all()
client = Client("Placeholder Client")
category = Category("Placeholder Category")
client.save()
category.save()
client_id = client.id
category_id = category.id
feature = Feature("Demo Feature", "Demo Feature Description", 1, client_id,
category_id, datetime(2018, 1, 1))
feature.save()
def categories():
if request.method == "POST":
category_name = str(request.data.get('category_name', ''))
if category_name:
category = Category(category_name=category_name)
category.save()
response = jsonify({
'category_id':
category.category_id,
'category_name':
category.category_name,
'date_created':
category.date_created,
'date_modified':
category.date_modified,
'recipes':
url_for('recipes',
category_id=category.category_id,
_external=True)
})
response.status_code = 201
return response
else:
# GET
categories = Category.get_all()
results = []
for category in categories:
obj = {
'category_id':
category.category_id,
'category_name':
category.category_name,
'date_created':
category.date_created,
'date_modified':
category.date_modified,
'recipes':
url_for('recipes',
category_id=category.category_id,
_external=True)
}
results.append(obj)
response = jsonify(results)
response.status_code = 200
return response
def create_categories(user_id):
"""Adds categories to the database"""
category = Category.query.filter(Category.user_id == user_id).filter_by(
category_name=request.data['category_name']).first()
if not category:
if request.method == "POST":
category_name = str(request.data.get('category_name', ''))
if category_name is None:
response = {'message': 'No content provided.'}
return make_response(jsonify(response)), 400
category_name.strip()
if category_name:
if validate.validate_name(category_name) == "True":
category = Category(
category_name=category_name, user_id=user_id)
category.save()
response = jsonify(category.category_json())
return make_response(response), 201
response = {'message': 'Category name required.'}
return make_response(jsonify(response)), 422
# There is an existing category.
response = {'message': 'Category already exists.'}
return make_response(jsonify(response)), 409
def __call__(self, request):
if request.is_json:
data = request.get_json()
name = data["name"]
new_category = Category(name=name)
if new_category.save():
return {
"message":
f"category {new_category.name} has been created successfully."
}
else:
return {
"message":
f"category {new_category.name} could not be created"
}
else:
return {"error": "The request payload is not in JSON format"}
def add_category():
form = CategoryForm.AddForm()
if form.validate_on_submit():
category = Category()
if category.category_exist(request.form['name']):
flash('This category already exists', category='errorMessage')
return redirect(redirect_back())
save_category = category.save(owner_id=session['user_id'],
name=request.form['name'],
created=datetime.now())
if (save_category):
flash(' Category \'{0}\' has been successfully saved'.format(
category.name),
category='successMessage')
return redirect(redirect_back())
flash('Unable to save category', category='errorMessage')
flash(request.form, category='input')
return redirect(redirect_back())
error = form.errors
flash(error, category='error')
flash(request.form, category='input')
return redirect_back()
