class CadItem():
    """
    This item contains all data for writing all necessary data to the solid edg cad file 

    """
    # diese Daten bilden dann die Impfdaten
    data_keys = [ "Author", "Teilenummer", "StartRevision",
        "Bezeichnung1", "Bezeichnung2", "Bezeichnung4", "KategorieNE",
        "BeschaffungsartNE", "TOS-Gruppe_ERP", "Dispogruppe",
        "Dispolaufgruppe", "WerkstofftypNE", "WerkstoffNE",
        "Oberflaechenbehandlung", "Farbe", "KalkulationsartNE",
        "RuestcodeAusw", "Fertigungsart", "Allgemeintoleranz",
        "HalbzeugInternAngabe", "Halbzeug", "HalbzeugOhneSaegeliste",
        "EinheitNE", "FOS-GruppeNE", "Hersteller", "HerstellerTeilenummer",
        "Serien-Status", "Kantennorm", "Oberflaechennorm", "Massnorm",
        "Tolerierungsnorm", "TOS-Gruppe", "TOS_Schoenenberger_Produkt",
        "TOS_Automation", "TOS_Maschinenelement",
        "TOS_Schoenenberger_Komponenten", "TOS_Pneumatik", "TOS_Konfig",
        "TOS_ILS", "TOS_MAS", "TOS_Spinnereikomponente",
        "TOS_Betriebsmittel", "TOS_Geruest", "TOS_FremdTransport",
        "TOS_MAS_Konfig", "TOS_Schaltschrank", "TOS_Schalt_Einbau",
        "TOS_Dezentral", "TOS_Automation_Sensorik", "TOS_Kabel",
        "TOS_Installation", "TOS_Automation_Sonstiges",
        "TOS_Verbindungselement", "TOS_Lager", "TOS_Kraft_Drehmoment",
        "TOS_Sicherungselement", "TOS_Bedienungselement", "TOS_ILS_2",
        "TOS_ILS_Foerderer", "TOS_MAS_Fahrstecke", "TOS_Trolley",
        "TOS_Trolley_Projekte", "TOS_Trolley_Omniflo",
        "TOS_Trolley_Spinnerei", "TOS_Trolley_ILS", "Änderungsbeschreibung",
        "Freigabevermerk", "ZusaetzlicheInformationen"]

    def __init__(self, asm_id):
        """ is initalized with its unique id"""
        self.set_asm_id(asm_id)
        self.red_ids = {}
        self.parent_children_mapping = {}
        self.data = {}
        self.gui_data = {}

    def set_asm_id(self, asm_id):
        self.asm_id = asm_id
    def get_current_asm_id(self):
        return self.asm_id

    def get_current_part_id(self):
        return self.prt_id
    def set_part_id(self, prt_id):
        self.prt_id = prt_id


    def set_parent_children(self, parent_children_mapping):
        """ contains all id relations below, data of subassemblies etc. in
            a dict of the kind parent-children
        """
        self.parent_children_mapping = parent_children_mapping
    def get_parent_children(self):
        return self.parent_children_mapping

    def set_red_ids(self, red_ids):
        """ these ids are not yet known by the PDM"""
        self.red_ids = red_ids

    def get_red_ids(self):
        return self.red_ids

    def set_data(self, data:dict):
        for k,v in data:
            self.set_data(k,v)
    def set_data(self, key, val):
        if key not in self.data_keys:
            raise Exception("this data can not be set")
        else:
            self.data[key] = val

    def get_data(self, key):
        return self.data[key]

    def load():
        pass
    def save():
        pass

class CadItemRepository:
    """ 
    container class for all CadItems in the local folder
    also just managed by the controller
    """
    def __init__(self):
        self.caditems = []

    def add(self, caditem):
        self.caditems.append(caditem)

    def get(self, id):
        for caditem in self.caditems:
            if caditem.id == id:
                return caditem
        return None

    def get_all(self):
        return self.caditems

    def update(self, old_item, new_item):
        self.delete(old_item.id)
        self.add(new_item)
        raise Exception("CadItem not found")

    def delete(self, id):
        for p in self.caditems:
            if p.id == id:
                self.caditems.remove(p)
                return
        raise Exception("CadItem not found")

if __name__ == "__main__":
    pass