cave/cave.py

import json 
from tkinter import *

class Etageres(object):
    def __init__(self):
        self.etageres = []
        self.nbetagere = 0
        self.couleur =  {"rouge":"red","blanc":"white","rose":"pink"}
    
    @classmethod
    def load_etageres(cls,file):
        try:
            with open(file, 'rb') as json_data:
                data=json.load(json_data)
        except FileNotFoundError:
            print ("no file to load")
            return Etageres()
        loadobj=Etageres()
        for etagere in data["etageres"]: 
            print("etagere:")  
            print (etagere) 
            loadobj.add_etagere(etagere["largeur"],etagere["hauteur"])
            for bouteille in etagere["bouteilles"]:
                loadobj.etageres[loadobj.nbetagere-1].add_bouteille(bouteille["pos_largeur"],bouteille["pos_hauteur"],bouteille["couleur"],bouteille["millesime"],bouteille["appelation"])
        return loadobj

            
            
           
    def add_etagere(self, largeur, hauteur):
        self.nbetagere += 1
        self.etageres.append(Etagere    (largeur, hauteur, self))

    def click_add_etagere(self):

        child_window=Toplevel(self.fenetre)
        Label(child_window,text="select largeur").pack()
        largeur=StringVar()
        Entry(child_window,textvariable=largeur).pack()
        
        Label(child_window,text="hauteur").pack()
        hauteur=StringVar()
        Entry(child_window,textvariable=hauteur).pack()
        
        Button(child_window,text="OK",command=lambda:add(largeur.get(),hauteur.get())).pack()

        def add(largeur,hauteur):
            child_window.destroy()
            self.nbetagere += 1
            newetagere=Etagere(int(largeur),int(hauteur),self)
            
            self.etageres.append(newetagere)
            newetagere.generate_frame(self.fenetre)
            
       

    def generate_windows(self):
        self.fenetre=Tk()
        self.fenetre.title(string="cave a vin")
        for etagere in self.etageres:
            etagere.generate_frame(self.fenetre)
        
        framecontrol=LabelFrame(self.fenetre,text="command pannel")
        framecontrol.grid(column=2,row=1)
        Button(framecontrol,text="add etagere",command=lambda:self.click_add_etagere()).pack()
        Button(framecontrol,text="remove etagere").pack()
        Button(framecontrol,text="save",command=lambda:self.save_etageres("cave")).pack()

    def mainloop(self):
        self.fenetre.mainloop()

    def nbbouteille(self):
        i = 0
        for etagere in self.etageres:
            for bouteille in etagere:
                i += 1

        return i

    def nbbouteillecouleur(self, couleur):
        i = 0
        for etagere in self.etageres:
            for bouteille in etagere:
                if bouteille.couleur == couleur:
                    i += 1
        return i

    def remove_etagere(self, etagere):
        self.etageres.remove(etagere)

        self.nbetagere -= 1

        return "etagere {} suprimer".format(etagere.num_etagere)
    def save_etageres(self,file):
        with open(file, 'w') as output:
            data={"etageres":[]}
            for etagere in self.etageres:
                data["etageres"].append(etagere.save())
            print(data)
            json.dump(data,output,indent=4)
            print ("save ok")

    def __iter__(self):
        return iter(self.etageres)


class Etagere(object):
    def __init__(self, largeur, hauteur, etageres):
        self.num_etagere = etageres.nbetagere
        self.largeur = largeur
        self.hauteur = hauteur
        self.etageres = etageres
        self.bouteilles = []

    def add_bouteille(self, pos_largeur, pos_hauteur, couleur, appelation, millesime):
        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))
        return "OK"

    def remove_bouteille(self, pos_largeur, pos_hauteur):
        i = 0
        for bouteille in self.bouteilles:
            if (pos_largeur == bouteille.pos_largeur and pos_hauteur == bouteille.pos_hauteur):
                self.bouteilles.remove(bouteille)
                return ( "bouteille suprimé")
            else:
                i += 1
        return ( "aucune bouteille troué a cette emplacement")

    def generate_frame(self,root):

        self.frame=LabelFrame(root,text=self)
        for hauteur in range(self.hauteur):
            for largeur in range(self.largeur):
                isset = False
                for bouteille in self:
                    if (largeur == bouteille.pos_largeur and hauteur == bouteille.pos_hauteur):
                        bouteille.generate_boutton(self.frame)
                        isset = True
                        break
                if isset == False:
                    Cassier_vide(largeur,hauteur,self)
        self.frame.grid(row=self.num_etagere)        
        

    def click_remove_bouteille(self,bouteille):
        self.remove_bouteille(bouteille.pos_largeur, bouteille.pos_hauteur)
        bouteille.boutton.destroy()
        Cassier_vide(bouteille.pos_largeur,bouteille.pos_hauteur,self)
        self.frame["text"]=self
        

    def click_add_bouteille(self,cassiervide):

        child_window=Toplevel(self.etageres.fenetre)
        Label(child_window,text="select color").pack()
        couleur=StringVar()
        couleur.set("rouge")
        OptionMenu(child_window,couleur,*self.etageres.couleur.keys()).pack()
        
        Label(child_window,text="enter millesime").pack()
        millesime=StringVar()
        Entry(child_window,textvariable=millesime).pack()
        
        Label(child_window,text="enter appelation").pack()
        appelation=StringVar()
        Entry(child_window,textvariable=appelation).pack()
        
        Button(child_window,text="OK",command=lambda:add(cassiervide,couleur.get(),millesime.get(),appelation.get())).pack()

        def add(cassiervide,couleur,millesime,appelation):
            child_window.destroy()
            newbouteille=Bouteille(cassiervide.pos_largeur, cassiervide.pos_hauteur, couleur, appelation, millesime, self)
            self.bouteilles.append(newbouteille)
            cassiervide.boutton.destroy()
            del cassiervide
            newbouteille.generate_boutton(self.frame)
            self.frame["text"]=self

    def save(self):
        data={"largeur": self.largeur,"hauteur": self.hauteur,"bouteilles":[]}
        for bouteille in self.bouteilles:
            data["bouteilles"].append(bouteille.save())
        return data

    def __repr__(self):
        return "etagére numéros {} de largeur {} et de hauteur {} et contient {} bouteille(s) ".format(self.num_etagere, self.largeur, self.hauteur, len(self.bouteilles))

    def __iter__(self):
        return iter(self.bouteilles)

class Cassier_vide(object):
    
    def __init__(self, pos_largeur, pos_hauteur,etagere):
        self.pos_largeur = pos_largeur
        self.pos_hauteur = pos_hauteur
        self.etagere = etagere
        self.boutton=Button(etagere.frame,text="cassier vide",command=lambda:self.etagere.click_add_bouteille(self))
        self.boutton.grid(row=self.pos_hauteur,column=self.pos_largeur)
        

class Bouteille(object):

    def __init__(self, pos_largeur, pos_hauteur, couleur, appelation, millesime, etagere):
        self.pos_largeur = pos_largeur
        self.pos_hauteur = pos_hauteur
        
        self.couleur = couleur

        self.millesime = millesime
        self.appelation = appelation
        self.etagere = etagere

    def generate_boutton(self,frame):

        self.boutton=Button(frame,text=self,command=lambda:self.etagere.click_remove_bouteille(self))
        self.boutton["bg"]=self.etagere.etageres.couleur[self.couleur]
        self.boutton.grid(row=self.pos_hauteur,column=self.pos_largeur)
        

    def save(self):
        data={ "pos_largeur" : self.pos_largeur,"pos_hauteur" : self.pos_hauteur, "couleur" : self.couleur, "millesime" : self.millesime,"appelation" : self.appelation}
        return data
    
    def __repr__(self):
        return "{},{} \n".format(self.appelation, self.millesime)

new = Etageres.load_etageres("cave")
#new =Etageres()
#new.add_etagere(10, 3)
#new.add_etagere(4,4)
#print(new.etageres[0].add_bouteille(1, 1, "rouge", "saint emilion", 2005))
#print(new.etageres[0].add_bouteille(9, 1, "rose", "gewurtz", 2010))
#print(new.etageres[0].add_bouteille(5, 2, "rouge", "saint emilion", 2006))
#
#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")))
#
#print(new.couleur)

new.generate_windows()
new.mainloop()