import json
import pandas as pd
import os

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"]

    gui_data_keys = [
            "asm_id",
            "teilenummer",
            "datenbank",
                "pdm_kategorie_neu",
            "neuaufbau",
                "aus_halbzeug_neu",
                "halbzeugnummer_neu",
                "halbzeuglaenge_neu",
                "halbzeugoberflaeche_neu",
                "halbzeugfarbe_neu",
                "lebenszyklus_status_neu",
                "model3d_neu",
            "bezeichnung1_sivas", "bezeichnung1_neu",
            "bezeichnung2_sivas", "bezeichnung2_neu",
            "halbzeugOhneSaege_sivas", "halbzeugOhneSaege_neu",
            "ruestcode_sivas", "ruestcode_neu",
            "fertigungsart_sivas", "fertigungsart_neu",
            "allgemeintoleranz_sivas", "allgemeintoleranz_neu",
            "fosGruppe_sivas", "fosGruppe_neu",
            "kalkulationsart_sivas", "kalkulationsart_neu",
            "beschaffungsart_sivas", "beschaffungsart_neu",
            "mengeneinheit_sivas", "mengeneinheit_neu",
            "halbzeug_sivas", "halbzeug_neu",
            "tosgruppe_sivas", "tosgruppe_neu",
            "dispogruppe_sivas", "dispogruppe_neu",
            "dispolaufgruppe_sivas","dispolaufgruppe_neu",
            "hersteller_sivas", "hersteller_neu",
            "ersatzfuer_sivas", "ersatzfuer_neu",
            "ersetztdurch_sivas", "ersetztdurch_neu",
            "herstteilenr_sivas", "herstteilenr_neu",
            "werkstoffdiskr_sivas",
            "werkstofftyp_sivas", "werkstofftyp_neu",
            "material_sivas", "material_neu",
                "oberflaeche_neu",
                "farbe_neu",
            "version_sivas", "version_neu",
            "aenderungsindex_sivas", "aenderungsindex_neu",
            "aenderungstext_sivas", "aenderungstext_neu",
            "freigabevermerk_sivas", "freigabevermerk_neu",
            "zusatzinfos_auto_text",
            "zusatzinfos_bendef_text",
            # Felder der Comboboxen. benannt wie deren callbacks
            "cb_pdm_kategorie_neu",
            "cb_halbzeug_neu",
            "cb_halbzeugoberflaeche",
            "cb_halbzeugfarbe",
            "cb_lebenszyklus_status_neu",
            "cb_3Dmodel_neu",
            "cb_halbzeugOhneSaege_neu",
            "cb_ruestcode_neu",
            "cb_fertigungsart_neu",
            "cb_allgemeintoleranz_neu",
            "cb_fosGruppe_neu",
            "cb_kalkulationsart_neu",
            "cb_beschaffungsart_neu",
            "cb_mengeneinheit_neu",
            "cb_dispogruppe_neu",
            "cb_dispolaufgruppe_neu",
            "cb_hersteller_neu",
            "cb_werkstofftyp_neu",
            "cb_material_neu",
            "cb_oberflaeche_neu",
            "cb_farbe_neu", 
            # buttons
            "cb_suche_asm",
            "cb_speichern",
            "cb_migrieren",
            "cb_zurücksetzen",
            ]

    # dies sind die Excel Headel der SIVAS csv Export Datei
    # und wie diese in das Datenmodell der Gui integriert werden kann
    excel_to_gui = {
        "Teilenummer":"teilenummer",  
        "Bezeichnung1":"bezeichnung1_sivas",
        "Bezeichnung2":"bezeichnung2_sivas", 
        "Beschaffungsart":"beschaffungsart_sivas",
        "Mengeneinheit":"mengeneinheit_sivas",
        "Halbzeug":"halbzeug_sivas",
        "TOS_Gruppe":"tosgruppe_sivas",
        "Dispogruppe":"dispolaufgruppe_sivas",
        "Dispolaufgruppe":"dispolaufgruppe_sivas",
        "ErsatzFuer":"ersatzfuer_neu",
        "ErsetztDurch":"ersetztdurch_neu",
        "AktuellerIndex":"aenderungsindex_sivas",
        "Hersteller":"hersteller_sivas",
        "HerstellerTeilenummer":"herstteilenr_sivas",
        "Version":"version_sivas",
        "HalbzeugExtern":"halbzeug_sivas",
        "Werkstoff":"werkstofftyp_sivas",
        "Oberfläche":"cb_oberflaeche_neu",
        # "Truemmer":"truemmer",
        # "HzOhneSaegeliste":"",
        "Ruestcode":"ruestcode_sivas",
        "Fertigungsart":"fertigungsart_sivas",
        "FosGruppe":"fosGruppe_sivas",
        "Kalkulationsart":"kalkulationsart_sivas",
        #"Indexerfasser":"",
        "Aenderungstext":"aenderungstext_sivas",
        "Freigabevermerk":"freigabevermerk_sivas",
        # "Indexbearbeiter":"in"
    }

    def __init__(self):
        """ is initalized with its unique id"""
        self.red_ids = {}
        self.parent_children_mapping = {}
        self.data = {}
        self.gui_data = {}
        self.read_only_items = {}
        self.controller = None

    def set_controller(self, controller):
        self.controller = controller

    def set_asm_id(self, asm_id):
        self.set_gui_data("asm_id", asm_id)
    def get_asm_id(self):
        return self.get_gui_data("asm_id")

    def get_current_part_id(self):
        return self.get_gui_data("teilenummer")


    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

    # Methoden um die Verwendbarkeit einzelner Schalter auszugrauen
    def set_read_only_dict(self, data:dict):
        for k,v in data:
            self.set_read_only(k,v)
    def set_read_only(self, key, val):
        if key not in self.gui_data:
            raise Exception("this data can not be set "+ key)
        else:
            self.read_only_items[key] = val
    def get_read_only(self, key):
        return self.read_only_items[key]
    def get_all_read_only_data(self):
        return self.read_only_items

    # Daten zum Impfen merken
    def set_data_dict(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 "+ key)
        else:
            self.data[key] = val

    def get_data(self, key):
        """ gibt ein Datenstück zurück """
        return self.data[key]
    def get_all_data(self):
        """ gibt alle enthaltenen Daten zurück """
        return self.data

    # Daten die in der GUI gespeichert werden sollen
    def set_gui_dict(self, data:dict):
        """ setzt die Daten der GUI """
        for k,v in data.items():
            self.set_gui_data(k,v)
    def set_gui_data(self, key, val):
        """ merkt sich ein Datenatom für die GUI """
        if key not in self.gui_data_keys:
            raise Exception("this data can not be set "+ key)
        else:
            self.gui_data[key] = val

    def get_gui_data(self, key):
        if key in self.gui_data:
            return self.gui_data[key]
        else:
            return None
    def get_all_gui_data(self):
        return self.gui_data

    def csv2guidata(self, data):
        translated_data = dict()
        for k,v in data.items():
            if k in self.excel_to_gui and v is not None:
                translated_data[self.excel_to_gui[k]] = v

        # Spezial: Befüllung von Änderungstext mit dem Author, falls in der .csv vorhanden
        if "Indexerfasser" in data and "Aenderungstext" in data:
            translated_data["aenderungstext_sivas"] = '{aenderungstext} (von {author})'.format(aenderungstext=data["Aenderungstext"], author=data["Indexerfasser"])
            # Indexerfasser wird nicht in der gui verwaltet. Deshalb Modifikation aus den Rohdaten
        self.set_gui_dict(translated_data)
        return translated_data


    def sivas_data_from_csv(self, cid):
        path = os.path.join(self.controller.get_env("data"), cid + '.csv')
        if os.path.exists(path):
            data = self._load_csv(path)
            return data
        return None

    
    def _load_csv(self, path):
        """lädt die SIVAS csv Datei in die Felder für die Oberfläche """
        if not os.path.exists(path):
            return None
        df = pd.read_csv(path, sep=';')
        ret = dict()
        # baue aus der gegebenen Zeile ein dict
        d = df.to_dict()
        for k,v in d.items():
            value = v[0]
            # schmeisse alle NaN raus
            if pd.isna(value):
                value = None
            ret[k] = value
        return ret
    
    def from_json_file(self, subpath, cid):
        path = os.path.join(subpath, cid + '.json')
        if os.path.exists(path):
            self.load(path)
        return False
    
    def from_json(self, json_str):
        jsonobj = json.loads(json_str)

        if "gui_data" in jsonobj:
            gui_data = jsonobj["gui_data"]
            self.set_gui_dict(gui_data)
        else:
            raise Exception("no gui_data given in string")

        if "parent_children_mapping" in jsonobj:
            parent_children_mapping = jsonobj["parent_children_mapping"]
            self.set_parent_children(parent_children_mapping)
        else:
            raise Exception("no parent_children_mapping given in string")
        if "red_ids" in jsonobj:
            red_ids = jsonobj["red_ids"]
            self.set_red_ids(red_ids)
        return True

    def to_dict(self, tree=False ):
        jsonobj = dict()

        # die Oberflächendaten
        gui_data = self.get_all_gui_data()
        jsonobj["gui_data"] = gui_data

        # der Baum
        if tree:
            parent_children_mapping = self.get_parent_children()
            jsonobj["parent_children_mapping"] = parent_children_mapping

            # die ids der nicht migrierten Dateien
            red_ids = self.get_red_ids()
            jsonobj["red_ids"] = red_ids
        return jsonobj
    
    def to_json(self, tree=False):
        obj = self.to_dict(tree)
        return json.dumps(obj)

    def load(self, path):
        """loads all data of a model from a json string"""
        try:
            with open(path, 'r') as f:
                self.from_json(json.load(f))
        except Exception:
            raise Exception("guid_data, data, parent_children_mapping or red_ids not found")


    def save(self, path, tree=False):
        """saves all data of model to a json string"""
        with open(path, 'w') as f:
           json.dump(self.to_dict(tree), f)

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)

    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