Feat: exam creation page

recopytex/scripts/exam.py

@@ -4,22 +4,36 @@
 from datetime import datetime
 from pathlib import Path
 from prompt_toolkit import HTML
+from ..config import NO_ST_COLUMNS
+import pandas as pd
 import yaml
 from .getconfig import config
 
 
+def try_parsing_date(text, formats=["%Y-%m-%d", "%Y.%m.%d", "%Y/%m/%d"]):
+    for fmt in formats:
+        try:
+            return datetime.strptime(text[:10], fmt)
+        except ValueError:
+            pass
+    raise ValueError("no valid date format found")
+
+
 class Exam:
     def __init__(self, name, tribename, date, term, **kwrds):
         self._name = name
         self._tribename = tribename
-        try:
-            self._date = datetime.strptime(date, "%y%m%d")
-        except:
-            self._date = date
+
+        self._date = try_parsing_date(date)
 
         self._term = term
 
-        self._exercises = {}
+        try:
+            kwrds["exercices"]
+        except KeyError:
+            self._exercises = {}
+        else:
+            self._exercises = kwrds["exercices"]
 
     @property
     def name(self):
@@ -125,8 +139,21 @@ class Exam:
     def display(self, name):
         pass
 
-    def write(self):
+    def write_yaml(self):
         print(f"Sauvegarde temporaire dans {self.path('.yml')}")
         self.tribe_path.mkdir(exist_ok=True)
         with open(self.path(".yml"), "w") as f:
             f.write(yaml.dump(self.to_dict()))
+
+    def write_csv(self):
+        rows = self.to_row()
+
+        base_df = pd.DataFrame.from_dict(rows)[NO_ST_COLUMNS.keys()]
+        base_df.rename(columns=NO_ST_COLUMNS, inplace=True)
+
+        students = pd.read_csv(self.tribe_student_path)["Nom"]
+        for student in students:
+            base_df[student] = ""
+
+        self.tribe_path.mkdir(exist_ok=True)
+        base_df.to_csv(self.path(".csv"), index=False)