ajout fonction save load

This commit is contained in:
vincent 2018-02-05 21:52:28 +01:00
parent 13499d5ee3
commit de9183f019

16
cave.py
View File

@ -1,9 +1,15 @@
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))
@ -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)
@ -50,8 +59,8 @@ class Etagere(object):
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))
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