Feat: écriture du fichier csv

recopytex/scripts/recopytex.py

@@ -5,10 +5,12 @@ import click
 from pathlib import Path
 import sys
 import papermill as pm
+import pandas as pd
 from datetime import datetime
 
 from .getconfig import config
 from .prompts import prompt_exam, prompt_exercise
+from ..config import NO_ST_COLUMNS
 
 
 @click.group()
@@ -42,7 +44,8 @@ def exam_dict2element(exam):
                     "term": exam["term"],
                     "assessment": exam["name"],
                     "date": exam["date"].strftime("%d/%m/%Y"),
-                    "exercice": ex["name"],
+                    "exercise": ex["name"],
+                    "question": q["id"],
                     **q,
                 }
             )
@@ -56,7 +59,23 @@ def new_exam():
     exam["exercices"] = []
     for ex in range(int(exam["ExQty"])):
         exam["exercices"].append(prompt_exercise(ex + 1))
-    print(exam_dict2element(exam))
+
+    elements = exam_dict2element(exam)
+
+    base_df = pd.DataFrame.from_dict(elements)[NO_ST_COLUMNS.keys()]
+    base_df.rename(columns=NO_ST_COLUMNS, inplace=True)
+
+    students = pd.read_csv(exam["tribe"]["students"])["Nom"]
+    for student in students:
+        base_df[student] = ""
+
+    dest = (
+        Path(config["source"])
+        / exam["tribe"]["name"]
+        / f"{exam['date'].strftime('%y%m%d')}_{exam['name']}.csv"
+    )
+    base_df.to_csv(dest, index=False)
+    print(f"Le fichier note a été enregistré à {dest}")
 
 
 @cli.command()