def charge_edit():
try:
page = int(request.args.get('page', 1))
except ValueError:
page = 1
if request.form['year']:
if page == 1:
offset = 0
else:
pages = page
pages -= 1
offset = pages * 10
charges = Charge.query().fetch(limit=10, offset=offset)
for char in charges:
budget = ChargeBudget.query(
ChargeBudget.charge_id == char.key,
ChargeBudget.date_app == function.date_convert(datetime.date(int(request.form['year']), 1, 1))
).get()
clef = 'name['+str(char.key.id())+']'
if budget:
if float(request.form[clef]) > 0:
budget.montant = float(request.form[clef])
budget.put()
else:
budget.key.delete()
else:
budget_new = ChargeBudget()
if float(request.form[clef]) > 0:
budget_new.montant = float(request.form[clef])
else:
budget_new.montant = float(0)
budget_new.date_app = function.date_convert(datetime.date(int(request.form['year']), 1, 1))
budget_new.charge_id = char.key
budget_new.put()
flash('Enregistrement effectue avec succes', 'success')
return redirect(url_for('budget.charge', page=page, year=str(request.form['year'])))
