def delete(charge_id):
charges = Charge.get_by_id(charge_id)
from ..budget.models_budget import ChargeBudget
bugdet = ChargeBudget.query(ChargeBudget.charge_id == charges.key).count()
if bugdet:
flash("Impossible de supprimer cet element", "warning")
else:
charges.key.delete()
flash("Suppression reussie", "success")
return redirect(url_for("charge.index"))
def edit(charge_id=None):
if charge_id:
charges = Charge.get_by_id(charge_id)
form = FormCharge(obj=charges)
else:
charges = Charge()
form = FormCharge()
success = False
if form.validate_on_submit():
entreprise = Societe.query().get()
charges.libelle = form.libelle.data
charges.societe = entreprise.key
charges.put()
flash("Enregistement effectue avec succes", "success")
success = True
return render_template("charge/edit.html", **locals())
