WEB/Praktikum3/bt/app/database.py

# coding: utf-8

import os
import os.path
import codecs

import json

#----------------------------------------------------------
class Database_cl(object):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self, type_spl):
   #-------------------------------------------------------
      self.type_s = type_spl
      self.path_s = os.path.join('data', type_spl)
      self.data_o = {}
      self.readData_p()

   #-------------------------------------------------------
   def create_px(self, data_opl):
   #-------------------------------------------------------
      # Überprüfung der Datenn müsste ergänzt werden!
      id_s = self.nextId_p()
      data_opl['id'] = id_s
      # Datei erzeugen
      file_o = codecs.open(os.path.join(self.path_s , id_s+'.dat'), 'w', 'utf-8')
      file_o.write(json.dumps(data_opl, indent=3, ensure_ascii=True))
      file_o.close()

      self.data_o[id_s] = data_opl

      return id_s
      
   #-------------------------------------------------------
   def read_px(self, id_spl = None):
   #-------------------------------------------------------
      # hier zur Vereinfachung:
      # Aufruf ohne id: alle Einträge liefern
      data_o = None
      if id_spl == None:
         data_o = self.data_o
      elif id_spl == '0':
         data_o = self.getDefault_px()
      else:
         if id_spl in self.data_o:
            data_o = self.data_o[id_spl]

      return data_o

   #-------------------------------------------------------
   def update_px(self, id_spl, data_opl):
   #-------------------------------------------------------
      # Überprüfung der Daten müsste ergänzt werden!
      status_b = False
      if id_spl in self.data_o:
         # Datei aktualisieren
         file_o = codecs.open(os.path.join(self.path_s, id_spl+'.dat'), 'w', 'utf-8')
         file_o.write(json.dumps(data_opl, indent=3, ensure_ascii=True))
         file_o.close()

         self.data_o[id_spl] = data_opl
         status_b = True
      
      return status_b

   #-------------------------------------------------------
   def delete_px(self, id_spl):
   #-------------------------------------------------------
      status_b = False
      if id_spl in self.data_o:
         # Datei entfernen
         os.remove(os.path.join(self.path_s, id_spl+'.dat'))

         del self.data_o[id_spl]
         status_b = True
      
      return status_b
         
   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------
      # sollte überschrieben werden!
      return {}

   #-------------------------------------------------------
   def readData_p(self):
   #-------------------------------------------------------

      files_a = os.listdir(self.path_s)
      for fileName_s in files_a:
         if fileName_s.endswith('.dat') and fileName_s != 'maxid.dat':
            file_o = codecs.open(os.path.join(self.path_s, fileName_s), 'rU', 'utf-8')
            content_s = file_o.read()
            file_o.close()
            id_s = fileName_s[:-4]
            self.data_o[id_s] = json.loads(content_s)

   #-------------------------------------------------------
   def nextId_p(self):
   #-------------------------------------------------------
      file_o = open(os.path.join(self.path_s, 'maxid.dat'), 'r+')
      maxId_s = file_o.read()
      maxId_s = str(int(maxId_s)+1)
      file_o.seek(0)
      file_o.write(maxId_s)
      file_o.close()

      return maxId_s


#----------------------------------------------------------
class LoadDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('projekt')
      super().__init__('komponente')
      super().__init__('qsmitarbeiter')
      super().__init__('swentwickler')
      super().__init__('projekt')
      super().__init__('fehler')
      super().__init__('katursache')
      super().__init__('katfehler')

#----------------------------------------------------------
class ProjektDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('projekt')

   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------

      return {
         'name': '',
         'id': ''
      }

#----------------------------------------------------------
class KomponenteDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('komponente')

   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------

      return {
         'name': '',
         'id': '',
         'projektid': ''
      }

   #-------------------------------------------------------
   def deleteDependencies_px(self, id):
   #-------------------------------------------------------
      komponenten = {}
      for komponente in list(self.data_o):
         print(komponente)
         if self.data_o[komponente]['projektid'] == id:
            self.delete_px(komponente)
            komponenten[komponente] = komponente
      return komponenten


#----------------------------------------------------------
class QsMitarbeiterDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('qsmitarbeiter')

   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------

      return {
         'name': '',
         'id': ''
      }

#----------------------------------------------------------
class SwEntwicklerDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('swentwickler')

   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------

      return {
         'name': '',
         'id': ''
      }

#----------------------------------------------------------
class KatfehlerDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('katfehler')

   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------

      return {
         'name': '',
         'id': ''
      }

#----------------------------------------------------------
class KatursacheDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('katursache')

   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------

      return {
         'name': '',
         'id': ''
      }

#----------------------------------------------------------
class FehlerDatabase_cl(Database_cl):
#----------------------------------------------------------

   #-------------------------------------------------------
   def __init__(self):
   #-------------------------------------------------------
      super().__init__('fehler')

   #-------------------------------------------------------
   def getDefault_px(self):
   #-------------------------------------------------------

      return {
         'name': '',
         'id': '',
         'komponenteid': '',
         'katfehlerid': '',
         'katursacheid': '',
         'qsmitarbeiterid': '',
         'swentwicklerid': '',
         'status': '',
         'beschreibung': '',
         'beschreibungursache': '',
         'zeiterfasst': '',
         'zeitbehoben': ''
      }

   #-------------------------------------------------------
   def readErkannt_px(self):
   #-------------------------------------------------------
      # hier zur Vereinfachung:
      # Aufruf ohne id: alle Einträge liefern
      data_o = {}
      for entry in self.data_o:
         if self.data_o[entry]['status'] != "geprueft":
            data_o[entry] = self.data_o[entry]

      return data_o

   #-------------------------------------------------------
   def readBehoben_px(self):
   #-------------------------------------------------------
      # hier zur Vereinfachung:
      # Aufruf ohne id: alle Einträge liefern
      data_o = {}
      for entry in self.data_o:
         if self.data_o[entry]['status'] == "geprueft":
            data_o[entry] = self.data_o[entry]
      return data_o

   #-------------------------------------------------------
   def deleteDependencies_px(self, id):
   #-------------------------------------------------------
      for fehler in list(self.data_o):
         if self.data_o[fehler]['komponenteid'] == id:
            self.delete_px(fehler)
# EOF