from flask_wtf import FlaskForm from flask_wtf.file import FileAllowed from wtforms import StringField, SubmitField, FileField, TextAreaField, RadioField from wtforms.validators import DataRequired, Regexp, Optional, ValidationError from app import db import app.models as models import yaml import os import re REPERTOIRE_YAML = "./export/" if not os.path.exists(REPERTOIRE_YAML): os.makedirs(REPERTOIRE_YAML) def validate_ref_list(regex, message): def _validate_ref_list(form, field): refs = field.data.strip() + " " if re.match(regex, refs) == None: raise ValidationError(message) return _validate_ref_list class Form(FlaskForm): sauvegarder = SubmitField("Sauvegarder") charger = SubmitField("Charger") exporter = SubmitField("Exporter") fichier = FileField("Choisir fichier", validators=[FileAllowed(["yml"], "Fichier Yaml seulement!")]) importer = SubmitField("Importer") referentiel = RadioField("Liste des référentiels", validators=[Optional()]) class PNForm(Form): regex = "^PN\d$" code = StringField("Code", validators=[DataRequired(), Regexp("^PN\d$", message="Le code doit être de la forme PN0.")]) nom = StringField("Nom", validators=[DataRequired()] ) diminutif = StringField("Diminutif", validators=[DataRequired()] ) description = TextAreaField("Description", validators=[DataRequired()] ) type = RadioField("Type", choices=["1","2","3"], validators=[DataRequired()]) class ACForm(Form): regex = "^AC\d{4}$" code = StringField("Code", validators=[DataRequired(), Regexp("^AC\d{4}$", message="Le code doit être de la forme AC0000.")]) titre = StringField("Titre", validators=[DataRequired()] ) saes = StringField("SAEs", validators=[DataRequired(), validate_ref_list("^(\s*(SAE\d{2}\s+)*)$", message="Les saes doit être de la forme SAE00.")] ) class SAEForm(Form): regex = "^SAE\d{2}$" code = StringField("Code", validators=[DataRequired(), Regexp("^SAE\d{2}$", message="Le code doit être de la forme SAE00.")]) titre = StringField("Titre", validators=[DataRequired()] ) semestre = StringField("Semestre", validators=[DataRequired()] ) heures_encadrees = StringField("Heures encadrées", validators=[DataRequired()] ) heures_tp = StringField("Heures TP", validators=[DataRequired()] ) projet = StringField("Projet", validators=[DataRequired()] ) description = TextAreaField("Description", validators=[DataRequired()] ) coef = StringField("Coef.", validators=[DataRequired()] ) acs = StringField("ACs", validators=[DataRequired(), validate_ref_list("^(\s*(AC\d{4}\s+)*)$", message="Les acs être de la forme AC0000.")] ) ressources = StringField("Ressources", validators=[DataRequired(), validate_ref_list("^(\s*(R\d{3}\s+)*)$", message="Les ressources doit être de la forme R000.")] ) livrables = TextAreaField("Livrables", validators=[DataRequired()] ) motscles = StringField("Mots clés", validators=[DataRequired()] ) class RessourceForm(Form): regex = "^R\{3}$" code = StringField("Code", validators=[DataRequired(), Regexp("^R\{3}$", message="Le code doit être de la forme R000.")]) nom = StringField("Nom", validators=[DataRequired()] ) semestre = StringField("Semestre", validators=[DataRequired()] ) heures_formation = StringField("Heures formation", validators=[DataRequired()] ) heures_tp = StringField("Heures TP", validators=[DataRequired()] ) coef = StringField("Coef.", validators=[DataRequired()] ) acs = StringField("ACs", validators=[DataRequired(), validate_ref_list("^(\s*(AC\d{4}\s+)*)$", message="Les acs doit être de la forme AC0000.")] ) saes = StringField("SAEs", validators=[DataRequired(), validate_ref_list("^(\s*(SAE\d{2}\s+)*)$", message="Les saes doit être de la forme SAE00.")] ) prerequis = StringField("Prérequis", validators=[DataRequired()] ) contexte = TextAreaField("Contexte", validators=[DataRequired()] ) contenu = TextAreaField("Contenu", validators=[DataRequired()] ) motscles = StringField("Mots clés", validators=[DataRequired()] ) class CompetenceForm(Form): regex = "^RT\d$" code = StringField("Code", validators=[DataRequired(), Regexp("^RT\d$", message="Le code doit être de la forme RT0.")]) nom = StringField("Nom", validators=[DataRequired()] ) diminutif = StringField("Diminutif", validators=[DataRequired()] ) description = TextAreaField("Description", validators=[DataRequired()] ) composantes = TextAreaField("Composantes", validators=[DataRequired()] ) situations = TextAreaField("Situations", validators=[DataRequired()] ) niveaux = TextAreaField("Niveaux", validators=[DataRequired()] ) categorie_liste = ["saes","ressources","acs"] separateur = None def form_import(form): """ Si le bouton import a été appuyé et qu'il n'y a pas d'erreur d'import => importe le fichier yaml""" # Bouton import appuyé et fichier yaml selectionné if form.importer.data and len(form.fichier.errors) == 0: if form.fichier.data.filename == "": form.fichier.errors.append("Aucun fichier selectioné.") return form if re.match(form.regex, form.fichier.data.filename[:-4]) == None: form.fichier.errors.append("Mauvais type de référentiel!") return form fichier_Yaml = yaml.safe_load(form.fichier.data.read()) for categorie, valeur in fichier_Yaml.items(): if categorie in categorie_liste: form[categorie].process_data(" ".join(valeur)) else: form[categorie].process_data(valeur) form.validate_on_submit() # Réinitialise les messages d'erreur return form def form_export(form): """ Si le formulaire est valide => exporte dans un fichier yaml avec les informations du formulaire """ output = {} for categorie, valeur in list(form.data.items())[6:-1]: if categorie in categorie_liste: output[categorie] = [ referentiel for referentiel in valeur.split(separateur) ] else: output[categorie] = valeur fichier = REPERTOIRE_YAML + form.code.data + ".yml" with open(fichier, "w", encoding="utf8") as fid: fid.write(yaml.dump(output)) def form_charger(form): """ Si le bouton charger est appuyé et qu'un référentiel du BDD a été selectionné => remplie le formulaire avec ses informations """ if form.charger.data: if form.referentiel.data == None: form.referentiel.errors.append("Aucun référentiel n'a été selectionné!") else: temp = form.referentiel.data[1:-1].split() model = getattr(models, temp[0]) referentiel = model.query.filter_by(code=temp[1]).first() for categorie in list(referentiel.__dict__)[1:]: if categorie in categorie_liste: form[categorie].process_data(" ".join(ref.code for ref in referentiel.__dict__[categorie])) else: form[categorie].process_data(referentiel.__dict__[categorie]) form.validate_on_submit() return form def form_sauvegarder(form): """ Si le bouton sauvegarder est appuyé et que le formulaire est valide => ajoute le référentiel dans la table respectif du base de données """ if form.sauvegarder.data: model = getattr(models, "".join( c for c in form.code.data if not c.isdigit())) referentiel = model.query.filter_by(code=form.code.data).first() if referentiel == None: referentiel = model() # Si le référentiel contient une catégorie avec une liste de référentiels for categorie, valeur in list(form.data.items())[6:-1]: if categorie in categorie_liste: resultat = [] refs = [ ref for ref in form[categorie].data.split(separateur) ] ref_model = getattr(models, categorie.upper()[:-1]) for ref in refs: ref_bdd = ref_model.query.filter_by(code=ref).first() if ref_bdd == None: ref_new = ref_model(code=ref) resultat.append(ref_new) db.session.add(ref_new) else: resultat.append(ref_bdd) form[categorie].process_data(resultat) form.populate_obj(referentiel) db.session.add(referentiel) db.session.commit()