Réécriture du script d'initialisation

ouais parce que on a supprimé des déconnaissances, du coup KeyError

diff --git a/init.py b/init.py
index 5625403e18faec4e87c5c4a33d6ed3e4d305329d..722daca6e938b7b82bab4acbcea1832802a3a806 100755 (executable)
--- a/init.py
+++ b/init.py
@@ -1,29 +1,32 @@
 #!/usr/bin/python
 # -*- coding:utf8 -*-
 
-# Pour initialiser deconnaisseur, si on n'a jamais joué dans ce dossier
-# parce que le dépôt git ne contient pas les fichiers modifiés, et
-# qu'ils sont indisensables au fonctionnement
+""" Pour initialiser deconnaisseur, si on n'a jamais joué dans ce dossier
+ parce que le dépôt git ne contient pas les fichiers modifiés, et
+ qu'ils sont indisensables au fonctionnement."""
 
 import pickle
 import re
 
 print "Ce script va initialiser deconnaisseur."
-print "Si des scores ou des énigmes déjà jouées sont mémorisées, ils seront effacés."
-go=raw_input("Continuer ?")
+go = raw_input("Initialiser les scores (écrase le fichier de scores) ? [y/N] ")
 
 if go.lower() in ["oui","o","yes","y"]:
     print "Création du fichier de scores vide"
     pickle.dump({},open("scores.pickle","w"))
-    print "Création des fichiers d'occurrences pour acoeur et irc.crans"
-    f=open("deconnaissances.txt")
-    t=f.read()
+
+go = raw_input("Initialiser les énigmes déjà jouées (écrase le fichier des déjà jouées) ? [y/N] ")
+if go.lower() in ["oui","o","yes","y"]:
+    print "Création des fichiers d'occurrences pour irc.crans"
+    f = open("deconnaissances.txt")
+    t = f.read()
     f.close()
-    l=re.findall("%\n(.*)\n(.*)\n(.*)\n(.*)\n(.*)\n",t)
-    dec={int(i[0]):list(i[1:]) for i in l if len(i)==5}
-    maxkey=max(dec.keys())
-    t="\n".join(["%-3s : %s"%(k,0) for k in range(maxkey+1)])
-    files=[open("played.%s.txt"%(serveur),"w") for serveur in ["crans","acoeur"]]
+    l = re.findall("%\n(.*)\n(.*)\n(.*)\n(.*)\n(.*)\n",t)
+    dico = {int(i[0]) : list(i[1:]) for i in l if len(i) == 5}
+    keys = dico.keys()
+    keys.sort()
+    t = "\n".join(["%-3s : %s" % (k,0) for k in keys])
+    files = [open("played.%s.txt" % (serveur), "w") for serveur in ["crans"]]
     for f in files:
         f.write(t)
         f.close()