# coding: utf-8

import os
import os.path
import codecs
import json

#----------------------------------------------------------
class Database_cl(object): 
#----------------------------------------------------------
    # da es hier nur darum geht, die Daten dauerhaft zu speichern,
    # wird ein sehr einfacher Ansatz verwendet:
    # - es können Daten zu genau 15 Teams gespeichert werden
    # - je Team werden 2 Teilnehmer mit Namen, Vornamen und Matrikelnummer
    #   berücksichtigt
    # - die Daten werden als eine JSON-Datei abgelegt
    

    #-------------------------------------------------------
    def __init__(self): 
    #-------------------------------------------------------
        self.data = {}
        self.data['Studenten'] = {}
        self.data['Lehrende'] = {}
        self.data['Firmen'] = {}
        self.data['Angebote'] = {}
        self.ReadAll()


    #------------------------------------------------------- 
    def Read(self, category):
    #-------------------------------------------------------
        path = 'data/' + category
        if not(os.path.exists(path)):
            os.makedirs(path)
        categoryDir = os.listdir(path)
        for fileName in categoryDir:
            if fileName.endswith('.json') and fileName != 'last.json':
                file = codecs.open(os.path.join('data', category, fileName), 'rU', 'utf-8')
                fileContent = file.read()
                id = fileName[:-5]
                self.data[category][id] = json.loads(fileContent)

    #------------------------------------------------------- 
    def ReadAll(self):
    #-------------------------------------------------------
        self.Read('Studenten')
        self.Read('Lehrende')
        self.Read('Firmen')
        self.Read('Angebote')    

    #-------------------------------------------------------
    def ReadEntry(self, category = None, id = None): 
    #-------------------------------------------------------
        print("ReadEntry: ", category, id)
        self.Read(category)
        data = None
        if id == None:
            data = self.data
        else:
            if id in self.data[category]: 
                data = self.data[category][id]
        
        print(data, "\n")
        return data

    #-------------------------------------------------------
    def Save(self, data, category): 
    #-------------------------------------------------------
        status_b = False
        id = data['id']
        print("ID: ", id, "\n")
        if(id != "None"):
            if id in self.data[category]:
                file = codecs.open(os.path.join('data', category, id+'.json'), 'w', 'utf-8')
                file.write(json.dumps(data, indent=3, ensure_ascii=True))
                file.close()
                self.Read(category)
                status_b = True
        else:
            data['id'] = self.IdNext(category)
            file = codecs.open(os.path.join('data', category, data['id']+'.json'), 'w', 'utf-8')
            file.write(json.dumps(data, indent=3, ensure_ascii=True))
            file.close()
            self.Read(category)
            status_b = True
        
        return status_b

    #-------------------------------------------------------
    def Delete(self, category, id): 
    #-------------------------------------------------------
        status_b = False
        if(category == 'Studenten'):
            for angebote in self.data['Angebote']:
                if(self.data['Angebote'][angebote]['Student'] == id):
                    os.remove(os.path.join('data', 'Angebote', angebote+'.json'))
        elif(category == 'Lehrende'):
            for angebote in self.data['Angebote']:
                if(self.data['Angebote'][angebote]['Lehrende'] == id):
                    os.remove(os.path.join('data', 'Angebote', angebote+'.json'))
        elif(category == 'Firmen'):
            for firmen in self.data['Firmen']:
                if(firmen == id):
                    fn = self.data['Firmen'][firmen]['Name']
            for angebote in self.data['Angebote']:
                if(self.data['Angebote'][angebote]['Firma'] == fn):
                    os.remove(os.path.join('data', 'Angebote', angebote+'.json'))

        if id in self.data[category]:
            os.remove(os.path.join('data', category, id+'.json'))
            del self.data[category][id]
        
        return status_b

    #-------------------------------------------------------
    def GetDefault(self, category): 
    #-------------------------------------------------------
        if(category == 'Studenten'):
            return {'Name':'', 'Vorname':'', 'Matrikelnummer':''}
        elif(category == 'Lehrende'):
            return {'Titel':'', 'Name':'', 'Vorname':'', 'Lehrgebiet':''}
        elif(category == 'Firmen'):
            return {'Name':'', 'Branche':'', 'Schwerpunkt':'', 'Sitz':'', 'Anzahl Mitarbeiter':''}
        elif(category == 'Angebote'):
            return {'Name':'', 'Firma':'', 'Beschreibung':'', 'Voraussetzungen':'', 'Firmenbetreuer':'', 'Status':'Angebot', 'Lehrender':'', 'ZeitraumVon':'', 'ZeitraumBis':'', 'Student':''}

    #-------------------------------------------------------
    def IdNext(self, category):
    #-------------------------------------------------------
        path = 'data/' + category + '/last.json'
        if(os.path.isfile(path)):
            file = open(os.path.join('data', category, 'last.json'), 'r+')
            last = file.read()
            last = str(int(last)+1)
            file.seek(0)
            file.write(last)
            file.close()
        else:
            file = open(os.path.join('data', category, 'last.json'), 'w+')
            last = str(int(0))
            file.write(last)
            file.close()
        return last
            
# EOF