def add_key():
""" Add key walmart """
user = model.User(current_user)
key = request.form['key']
user.set_key(key)
return redirect("/home")
def add_upc():
""" Add walmart object """
user = model.User(current_user)
user.set_upc(request.form['upc'])
if user.set_upc(request.form['upc']) != 1:
flash("Wrong upc or key!", category='error')
return redirect("/home")
def download():
""" Walmart data in csv format """
user = model.User(current_user)
csv = user.get_csv()
response = make_response(csv)
response.headers["Content-Disposition"] = "attachment; filename=database.csv"
return response
def data_array():
""" Data for datateble """
class JSONEncoder(json.JSONEncoder):
""" Fix error """
def default(self, o):
"""
Args:
o:
Returns:
"""
if isinstance(o, ObjectId):
return str(o)
if isinstance(o, (datetime.datetime, datetime.date)):
return str(o)
return json.JSONEncoder.default(self, o)
draw = int(request.form['draw'], base=0) + 1
user = model.User(current_user)
data = user.get_upc_limit_sort(int(request.form['start']), int(request.form['length']),\
request.form['columns[' + request.form['order[0][column]'] + '][data]'],\
(-1, 1)[request.form['order[0][dir]'] == 'asc'])
data = json.dumps({"draw": draw,
"recordsTotal": user.count_upc(),
"recordsFiltered": user.count_upc(),
"data":data}, cls=JSONEncoder)
return data
def del_upc():
""" Delete walmart object """
user = model.User(current_user)
user.del_upc(request.form['upc_del'])
return redirect("/home")
def Settings():
""" Return Settings.html and key user"""
user = model.User(current_user)
key = user.key_walmart()
return render_template('Settings.html', key=key)