def animalOperation(category=None):
print(app.root_path)
if checkSession(session,["Admin","Manager","Staff"]):
global newUsers
global newSponsors
global severity
# newUsers = userDB.get_new_users()
form = getAnimalForm(category) if category != None else None
cats = Category.get_all()
addForm = True
severAnimals = insertAnimal.health_count()
if request.method == "GET":
if category == None:
form = None
addForm = False
else:
addForm = True
if form and form.validate_on_submit():
imagePath = uploadImage(form.image.data,category)
newAnimal = getAnimal(form.__dict__['_fields'],category)
newAnimal.update({'image':[imagePath],"dateAdded":datetime.utcnow(),'status':'healthy'})
_id = insertAnimal.insert_animal(newAnimal.__dict__,category)
log.insert_one({
"inserted_by": session['current_user']['username'],
"animal": _id,
'operation':'add',
'category': category,
"date_added":datetime.now()
})
return redirect(url_for("user.staffHome"))
return render_template("staff/animals.html",profile="profile.png",categories=cats,addForm=addForm,category=category,form=form,newUsers=newUsers,newSponsors=newSponsors,severity=severity,health_dist=severAnimals)
else:
return redirect(url_for("animals.zooAnimals"))
def get_all():
"""
Get all categories from a given retailer
"""
logger.debug("Testing connection with one product")
retailer = ctype = request.args.get('retailer')
cats = Category.get_all(retailer=retailer)
if not cats:
raise errors.ApiError("invalid_request",
"Could not fetch data from Postgres Categories")
return jsonify(cats)
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 get_categories():
"""Get category list"""
results = [category.to_dict() for category in Category.get_all()]
return jsonify(results), 200
