ajout fonction save load

cave.py

@@ -1,14 +1,20 @@
 
+import pickle
 class Etageres(object):
     def __init__(self):
         self.etageres = []
         self.nbetagere = 0
-
+    
+    @classmethod
+    def load_etageres(cls,file):
+        with open(file, 'rb') as data:
+            return pickle.load(data)
+           
     def add_etagere(self, largeur, hauteur):
         self.nbetagere += 1
         self.etageres.append(Etagere(largeur, hauteur, self))
 
-
+ 
     def nbbouteille(self):
         i = 0
         for etagere in self.etageres:
@@ -31,6 +37,9 @@ class Etageres(object):
         self.nbetagere -= 1
 
         return "etagere {} suprimer".format(etagere.num_etagere)
+    def save_etageres(self):
+        with open('company_data.pkl', 'wb') as output:
+             pickle.dump(self,output)
 
     def __iter__(self):
         return iter(self.etageres)
@@ -48,10 +57,10 @@ class Etagere(object):
         for bouteille in self.bouteilles:
             if (pos_hauteur > self.hauteur or pos_largeur > self.largeur or pos_largeur < 1 or pos_largeur < 1):
                 return "bouteille positioné en dehors des emplacement"
-                if (pos_largeur == bouteille.pos_largeur and pos_hauteur == bouteille.pos_hauteur):
-                    return "une bouteille existe déja a cette emplacement"
-        self.bouteilles.append(
-            Bouteille(pos_largeur, pos_hauteur, couleur, appelation, millesime, self))
+            if (pos_largeur == bouteille.pos_largeur and pos_hauteur == bouteille.pos_hauteur):
+                return "une bouteille existe déja a cette emplacement"
+                    
+        self.bouteilles.append(Bouteille(pos_largeur, pos_hauteur, couleur, appelation, millesime, self))
         return "OK"
 
     def remove_bouteille(self, pos_largeur, pos_hauteur):
@@ -94,4 +103,7 @@ print(new.etageres[0].bouteilles)
 print("vos etagére contiennes {} bouteille".format(new.nbbouteille()))
 print("vos etagére contiennes {} bouteille de rouge".format(
     new.nbbouteillecouleur("rouge")))
+new.save_etageres()
 print(new.remove_etagere(new.etageres[0]))
+del new
+