app/pe/pe_jury.py

@@ -100,126 +100,127 @@ class JuryPE(object):
             meme_programme: si True, impose un même programme pour tous les étudiants participant au jury,
                             si False, permet des programmes differents
         """
-        "L'année du diplome"
         self.diplome = diplome
+        "L'année du diplome"
 
-        "La formation associée au diplome"
         self.formation_id = formation_id
+        "La formation associée au diplome"
 
-        "Un zip où ranger les fichiers générés"
         self.nom_export_zip = f"Jury_PE_{self.diplome}"
-        self.zipdata = io.BytesIO()
-        self.zipfile = ZipFile(self.zipdata, "w")
+        "Nom du zip où ranger les fichiers générés"
 
-        """Chargement des étudiants à prendre en compte dans le jury"""
+        self.zipdata = io.BytesIO()
+        with ZipFile(self.zipdata, "w") as zipfile:
+            # Chargement des étudiants à prendre en compte dans le jury
             pe_comp.pe_print(
                 f"""*** Recherche et chargement des étudiants diplômés en {
                     self.diplome} pour la formation {self.formation_id}"""
             )
-        self.etudiants = EtudiantsJuryPE(self.diplome)  # Les infos sur les étudiants
+            self.etudiants = EtudiantsJuryPE(
+                self.diplome
+            )  # Les infos sur les étudiants
             self.etudiants.find_etudiants(self.formation_id)
             self.diplomes_ids = self.etudiants.diplomes_ids
 
-        """Génère les semestres taggués (avec le calcul des moyennes) pour le jury PE"""
+            # Génère les semestres taggués (avec le calcul des moyennes) pour le jury PE
             pe_comp.pe_print("*** Génère les semestres taggués")
             self.semestres_taggues = compute_semestres_tag(self.etudiants)
 
             if pe_comp.PE_DEBUG:
-            """Intègre le bilan des semestres taggués au zip final"""
-            for fid in self.semestres_taggues:
-                formsemestretag = self.semestres_taggues[fid]
+                # Intègre le bilan des semestres taggués au zip final
+                for formsemestretag in self.semestres_taggues.values():
                     filename = formsemestretag.nom.replace(" ", "_") + ".csv"
                     pe_comp.pe_print(f"    - Export csv de {filename} ")
                     self.add_file_to_zip(
-                    filename, formsemestretag.str_tagtable(), path="details_semestres"
+                        zipfile,
+                        filename,
+                        formsemestretag.str_tagtable(),
+                        path="details_semestres",
                     )
 
-        """Génère les trajectoires (combinaison de semestres suivis
-        par un étudiant pour atteindre le semestre final d'un aggrégat)
-        """
+            # Génère les trajectoires (combinaison de semestres suivis
+            # par un étudiant pour atteindre le semestre final d'un aggrégat)
             pe_comp.pe_print(
                 "*** Génère les trajectoires (différentes combinaisons de semestres) des étudiants"
             )
             self.trajectoires = TrajectoiresJuryPE(self.diplome)
             self.trajectoires.cree_trajectoires(self.etudiants)
 
-        """Génère les moyennes par tags des trajectoires"""
-        pe_comp.pe_print("*** Calcule les moyennes par tag des trajectoires possibles")
+            # Génère les moyennes par tags des trajectoires
+            pe_comp.pe_print(
+                "*** Calcule les moyennes par tag des trajectoires possibles"
+            )
             self.trajectoires_tagguees = compute_trajectoires_tag(
                 self.trajectoires, self.etudiants, self.semestres_taggues
             )
 
             if pe_comp.PE_DEBUG:
-            """Intègre le bilan des trajectoires tagguées au zip final"""
-            for trajectoire_id in self.trajectoires_tagguees:
-                trajectoire_tagguee = self.trajectoires_tagguees[trajectoire_id]
+                # Intègre le bilan des trajectoires tagguées au zip final
+                for trajectoire_tagguee in self.trajectoires_tagguees.values():
                     filename = trajectoire_tagguee.get_repr().replace(" ", "_") + ".csv"
                     pe_comp.pe_print(f"    - Export csv de {filename} ")
                     self.add_file_to_zip(
+                        zipfile,
                         filename,
                         trajectoire_tagguee.str_tagtable(),
                         path="details_semestres",
                     )
 
-        """Génère les interclassements (par promo et) par (nom d') aggrégat"""
+            # Génère les interclassements (par promo et) par (nom d') aggrégat
             pe_comp.pe_print("*** Génère les interclassements par aggrégat")
             self.interclassements_taggues = compute_interclassements(
                 self.etudiants, self.trajectoires, self.trajectoires_tagguees
             )
 
             if pe_comp.PE_DEBUG:
-            """Intègre le bilan des aggrégats (par promo) au zip final"""
-            for nom_aggregat in self.interclassements_taggues:
-                interclass_tag = self.interclassements_taggues[nom_aggregat]
+                # Intègre le bilan des aggrégats (par promo) au zip final
+                for interclass_tag in self.interclassements_taggues.values():
                     filename = interclass_tag.get_repr().replace(" ", "_") + ".csv"
                     pe_comp.pe_print(f"    - Export csv de {filename} ")
                     self.add_file_to_zip(
+                        zipfile,
                         filename,
                         interclass_tag.str_tagtable(),
                         path="details_semestres",
                     )
 
-        """Synthèse des éléments du jury PE"""
+            # Synthèse des éléments du jury PE
             self.synthese = self.synthetise_juryPE()
 
             # Export des données => mode 1 seule feuille -> supprimé
             pe_comp.pe_print("*** Export du jury de synthese")
-        filename = "synthese_jury_" + str(self.diplome) + ".xlsx"
-        with pd.ExcelWriter(filename, engine="openpyxl") as writer:
-            for onglet in self.synthese:
-                df = self.synthese[onglet]
-                df.to_excel(
-                    writer, onglet, index=True, header=True
-                )  # écriture dans l'onglet
-                # worksheet = writer.sheets[onglet]  # l'on
+            output = io.BytesIO()
+
+            with pd.ExcelWriter(output, engine="openpyxl") as writer:
+                for onglet, df in self.synthese.items():
+                    # écriture dans l'onglet:
+                    df.to_excel(writer, onglet, index=True, header=True)
+            output.seek(0)
 
             self.add_file_to_zip(
-            filename,
-            open(filename, "rb").read(),
+                zipfile, f"synthese_jury_{self.diplome}.xlsx", output.read()
             )
 
         # Fin !!!! Tada :)
 
-    def add_file_to_zip(self, filename: str, data, path=""):
-        """Add a file to our zip
+    def add_file_to_zip(self, zipfile: ZipFile, filename: str, data, path=""):
+        """Add a file to given zip
         All files under NOM_EXPORT_ZIP/
         path may specify a subdirectory
 
         Args:
+             zipfile: ZipFile
              filename: Le nom du fichier à intégrer au zip
              data: Les données du fichier
              path: Un dossier dans l'arborescence du zip
         """
         path_in_zip = os.path.join(path, filename)  # self.nom_export_zip,
-        self.zipfile.writestr(path_in_zip, data)
+        zipfile.writestr(path_in_zip, data)
 
-    def get_zipped_data(self):
+    def get_zipped_data(self) -> io.BytesIO | None:
         """returns file-like data with a zip of all generated (CSV) files.
-        Reset file cursor at the beginning !
+        Warning: reset stream to the begining.
         """
-        if self.zipfile:
-            self.zipfile.close()
-            self.zipfile = None
         self.zipdata.seek(0)
         return self.zipdata