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)