TreeWindow.py

from Box import Box
from View import View
from Vec import Vec


class TreeWindow:
 
    def __init__(self, view: View, img_closed:str, img_opened:str, img_list_empty:str, pos:Vec):
        self.view = view
        self.img_closed = img_closed
        self.img_opened = img_opened
        self.img_list_empty = img_list_empty
        
        f1 = Box(Vec(875,1088),90,view,None)
        e1,e2 = Box(Vec(589,923),64,view,f1),Box(Vec(1183,938),65,view,f1)
        d1,d2,d3,d4 = Box(Vec(429,708),52,view,e1),Box(Vec(724,652),56,view,e1),Box(Vec(1089,616),64,view,e2),Box(Vec(1320,745),63,view,e2)
        c1,c2,c3,c4,c5,c6,c7,c8 = Box(Vec(412,321),43,view,d1), Box(Vec(535,322),52,view,d1),Box(Vec(665,363),50,view,d2),Box(Vec(824,366),60,view,d2),Box(Vec(977,340),53,view, d3),Box(Vec(1175,325),48,view, d3),Box(Vec(1311,300),55,view,d4),Box(Vec(1422,366),63,view,d4)      
        b1,b2,b3,b4,b5,b6,b7,b8,b9,b10,b11,b12,b13,b14,b15,b16 = Box(Vec(355,151),32,view ,c1),Box(Vec(429,151),32,view ,c1),Box(Vec(500,151),32,view ,c2),Box(Vec(586,151),32,view ,c2),Box(Vec(658,145),32,view ,c3),Box(Vec(735,151),32,view ,c3),Box(Vec(825,165),32,view ,c4),Box(Vec(885,169),32,view ,c4),Box(Vec(958,185),32,view ,c5),Box(Vec(1024,172),32,view,c5),Box(Vec(1094,190),32,view,c6),Box(Vec(1186,184),32,view ,c6),Box(Vec(1281,179),32,view ,c7),Box(Vec(1358,165),32,view ,c7),Box(Vec(1431,174),32,view ,c8),Box(Vec(1500,170),32,view,c8),
        
        
        self.boxes = [b1,b2,b3,b4,b5,b6,b7,b8,b9,b10,b11,b12,b13,b14,b15,b16,
                      c1,c2,c3,c4,c5,c6,c7,c8,
                      d1,d2,d3,d4,
                      e1,e2,
                      f1]
        self.opened = False
        self.pos = pos
        self.width = 187 #width of the folder icon
        self.height = 193 #height of the folder icon

    def draw(self):
        if self.opened:
            self.view.draw_image(self.img_opened, self.pos)
            for boxes in self.boxes:
                print('a')
                boxes.draw()
        self.view.draw_image(self.img_closed, self.pos) # the folder is always showed