app.py

   1 #!/usr/bin/env python3
   2 # -*- coding: utf-8 -*-
   3
   4 import flask
   5 import io
   6 import matplotlib.pyplot as plt
   7 import base64
   8 from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
   9
  10 from configuration import config_init,CONFIG
  11 from gestion_erreurs import initialise_erreurs, debug, niveau_debug, warning
  12 import gestion_donnees as donnees
  13 from trace_courbe import cree_figure
  14 import faq as f
  15
  16
  17 def initialise_mode_beta():
  18     global beta
  19     hote = flask.request.host
  20     if hote[:4] == "beta":
  21         print("** Mode bêta !**")
  22         return True
  23     else:
  24         return False
  25
  26
  27 app = flask.Flask(__name__)
  28
  29 @app.route('/',methods=['POST','GET'])
  30 def index():
  31     beta=initialise_mode_beta()
  32     liste_err = initialise_erreurs()
  33     val_form = config_init()
  34     ## Si on a chargé un fichier de données
  35     if flask.request.method=="POST":
  36         if 'fichier_donnees' in flask.request.files:
  37             ## charger les données dans le formulaire
  38             fichier = flask.request.files['fichier_donnees']
  39             chaine = fichier.read()
  40             val_form2 = donnees.fichier_json_vers_configdonnees(chaine,liste_err)
  41             # Si jamais le formulaire est vide, c'est qu'il y a eu une erreur !
  42             if val_form2 == {}:
  43                 return flask.render_template("index.html",err=liste_err[2]+liste_err[1]+liste_err[0],valform=val_form, CONFIG=CONFIG, beta=beta)
  44             val_form.update(val_form2)
  45             val_form["fichier_importe"] = 1 # Pour noter qu'on a importé un fichier
  46
  47     return flask.render_template("index.html",err=liste_err[1],valform=val_form, CONFIG=CONFIG, beta=beta)
  48
  49 @app.route('/apropos')
  50 def apropos():
  51     beta=initialise_mode_beta()
  52     return flask.render_template("apropos.html",err=[], beta=beta)
  53
  54
  55 @app.route("/courbe/<ext>", methods=['POST'])
  56 def courbe_image(ext):
  57     liste_err = initialise_erreurs()
  58     data = flask.request.form
  59
  60     # récupérer les données du formulaire proprement
  61     config,listes_jours,listes_donnees = donnees.web_vers_python(data,liste_err)
  62     debug(" * On a récupéré et traité les données du formulaire web, "+str(listes_jours)+str(listes_donnees),liste_err)
  63
  64     # Gérer les enfants additionnels
  65     enfants_add = donnees.gere_enfants_additionnels(data, flask.request.files, liste_err)
  66     debug("Enfants additionnels récupérés : "+str(enfants_add), liste_err)
  67
  68     # Récupérer le texte à exporter
  69     texte = donnees.donnees_vers_json(listes_jours["poids"],listes_donnees["poids"],listes_jours["taille"],listes_donnees["taille"], config)
  70     debug("texte prêt à être exporté : "+texte,liste_err)
  71
  72     # noter le nom de l'enfant pour l'export
  73     nomenfant = donnees.simplifie_nom(config['nom'])
  74     # Les noms des autres enfants c'est sympa
  75     nomsenfantsplus = ""
  76     for dicoenfant in enfants_add:
  77         nomsenfantsplus += "_"+donnees.simplifie_nom(dicoenfant["poids"][0]["nom"])
  78
  79     existe_courbe = False
  80     # créer les figures
  81     try:
  82         debug("création des figures...",liste_err)
  83         textes_images = {}
  84         liste_extracalculs = []
  85
  86         for typed in CONFIG["liste_typedonnees"]:
  87             if config["tracevide"] == "oui" or listes_jours[typed] != []:
  88                 existe_courbe = True # Une courbe (au moins) a été tracée
  89                 debug("On trace la courbe de "+typed,liste_err)
  90                 fig = cree_figure(config,listes_jours[typed],listes_donnees[typed],typed,liste_extracalculs, liste_err, enfants_add)
  91                 output = io.BytesIO()
  92                 FigureCanvas(fig).print_png(output)
  93                 plt.close(fig)
  94                 textes_images[typed] = base64.b64encode(output.getvalue()).decode("ascii")
  95             else:
  96                 textes_images[typed] = ""
  97
  98         result = "success"
  99     except:
 100         result = "fail"
 101         liste_err[0].append("Impossible de générer les courbes ! Config : "+str(config))
 102
 103     if not(existe_courbe):
 104         warning("Aucune courbe n'a été tracée. C'est probablement parce que les données sont vides et que l'option \"tracer les courbes vides\" n'a pas été cochée.", liste_err)
 105
 106     if ext == "b64":
 107         reponse = { "result":result,
 108             "export_txt": texte,
 109             "nomenfant": nomenfant,
 110             "nomsenfantsplus": nomsenfantsplus}
 111         if result == "success":
 112             reponse["messages"] = liste_err[0]+liste_err[1]
 113             reponse["calculextra"] = liste_extracalculs
 114         else:
 115             reponse["messages"] = liste_err[2]+liste_err[1]+liste_err[0]
 116
 117         for typed in CONFIG["liste_typedonnees"]:
 118             reponse["image_"+typed] = textes_images.get(typed,"")
 119
 120         return flask.jsonify(reponse)
 121
 122     elif ext == "png" and result == "success": # à voir cette partie car il faudrait voir ce qu'on y renvoie... vu
 123         # qu'il n'y a plus une courbe mais plusieurs. Là ça renverra la dernière générée.
 124         return flask.Response(output.getvalue(), mimetype='image/png')
 125     elif ext == "png":
 126         return flask.render_template("erreur_fatale.html",erreur="Impossible de générer la courbe")
 127     else:
 128         return flask.render_template("erreur_fatale.html",erreur="Le type d'image demandé n'est pas reconnu")
 129
 130
 131 @app.route("/contact")
 132 def contact():
 133     beta=initialise_mode_beta()
 134     return flask.render_template("contact.html",err=[], beta=beta)
 135
 136 @app.route("/faq")
 137 def faq():
 138     beta=initialise_mode_beta()
 139
 140     table_faq = f.lire_fichier_csv_simple(f.fichier_FAQ)
 141     l_categ, l_categsimple, table_qr = f.extraire_tables_par_cat(table_faq)
 142
 143     return flask.render_template("faq.html",lcateg=l_categ,lcategsimple= l_categsimple, tableqr=table_qr,err=[], beta=beta)
 144
 145 @app.route("/changelog")
 146 def changelog():
 147     beta=initialise_mode_beta()
 148
 149     table_version = f.lire_fichier_csv_simple(f.fichier_changelog)
 150
 151     return flask.render_template("changelog.html",table = table_version, nblignes=CONFIG["nb_lignes_changelog"], err=[], beta=beta)
 152
 153 if __name__ == "__main__":
 154     print("Mode debug maison : "+str(niveau_debug))
 155     app.run(host='0.0.0.0',debug=True)