def client_edit():
try:
page = int(request.args.get('page', 1))
except ValueError:
page = 1
if request.form['year']:
clients = Client.objects.paginate(page=page, per_page=10)
for client in clients:
budget = ClientBudget.objects(
Q(client_id=client.id) & Q(date_app=function.datetime_convert(
datetime.date(int(request.form['year']), 1, 1)))).first()
clef = 'name[' + str(client.id) + ']'
if budget:
if float(request.form[clef]) > 0:
budget.montant = float(request.form[clef])
budget.save()
else:
budget.delete()
else:
budget_new = ClientBudget()
if float(request.form[clef]) > 0:
budget_new.montant = float(request.form[clef])
else:
budget_new.montant = float(0)
budget_new.date_app = function.datetime_convert(
datetime.date(int(request.form['year']), 1, 1))
budget_new.client_id = client.id
budget_new.save()
flash('Enregistrement effectue avec succes', 'success')
return redirect(
url_for('budget.client', page=page, year=request.form['year']))
