from side import Side

class Firkant:

    def __init__(self):
        self._venstre = None
        self._hoyre = None
        self._topp = None
        self._bunn = None

    def leggTilSide(self, side, besk):
        if besk == "venstre" and self._venstre == None:
            self._venstre = side
        elif besk == "hoyre" and self._hoyre == None:
            self._hoyre = side
        elif besk == "topp" and self._topp == None:
            self._topp = side
        elif besk == "bunn" and self._bunn == None:
            self._bunn = side
        else:
            print("Noe gikk galt")
    
    def fjernSide(self, side):
        if side == "venstre":
            self._venstre = None
        elif side == "hoyre":
            self._hoyre = None
        elif side == "topp":
            self._topp = None
        elif side == "bunn":
            self._bunn = None
        else:
            print("Ingen sider ble fjernet")
    
    def erFirkant(self):
        return self._venstre and self._hoyre and self._topp and self._bunn