On va se simplifier la vie pour l'année prochaine : on enregistre et on charge le résultat du tirage dans un fichier JSON ! :)

tirkdo.py

@@ -1,7 +1,11 @@
 #!/usr/bin/python
 
+import datetime
+import json
 import random
+import sys
 
+annee = datetime.date.today().year
 participants = [
     'Maman',
     'Papa',
@@ -21,17 +25,16 @@ on_evite = {
   'Didi': 'Johnny',
 }
 
-annee_passee = {
-  'Nico': 'Ben',
-  'Didi': 'Nico',
-  'Mémé': 'Papa',
-  'Maman': 'Mémé',
-  'Charlotte': 'Ludo',
-  'Johnny': 'Charlotte',
-  'Papa': 'Didi',
-  'Ludo': 'Maman',
-  'Ben': 'Johnny',
-}
+# On charge le résultat de l'année dernière
+nom_fichier = sys.argv[1] if len(sys.argv) > 1 else f"{annee-1}.json"
+try:
+    with open(nom_fichier, "r", encoding="utf-8") as fd:
+        annee_passee = json.load(fd)
+        print(f"Résultat de l'an passée chargé depuis le fichier {nom_fichier} :")
+        print("\n".join([f"- {offre} => {recois}" for offre, recois in annee_passee.items()]))
+except FileNotFoundError:
+    print(f"Échec de chargement du tirage de l'an passé depuis le fichier {nom_fichier}.")
+    sys.exit(1)
 
 class EchecTirage(Exception):
     pass
@@ -69,6 +72,7 @@ def tirage(nb_tentatives_max=99):
     return result
 
 # On procède au tirage
+print("On procède au tirage... Suspense !")
 result = None
 while not result:
     try:
@@ -77,6 +81,13 @@ while not result:
         print("Échec du tirage, on recommence !")
         result = None
 
-print("Résultat du tirage :")
+# On affiche le résultat
+print("Résultat du tirage pour l'année {annee} :")
 for offre, recois in result.items():
     print(f" - {offre} -> {recois}")
+
+# On enregistre le tirage pour l'année passée
+nom_fichier = f"{annee}.json"
+with open(nom_fichier, "w", encoding="utf-8") as fichier:
+    fichier.write(json.dumps(result, indent=2, ensure_ascii=False))
+print(f"Résultat enregistré dans le fichier {nom_fichier}.")