gd/config/datafunc.py

#modulo per la gestione della lista 'data' e dei relativo valori
from .views import *
from azienda.models import *
from amministratore.models import *

class DataDict(object):
  def __init__(self):
    self.data = dict()
    self.data['HeaderTitle'] = getConfig('HeaderTitle')
    self.data['Navbar']=True
    self.data['ShowSelectAzienda'] = True
    self.data['ShowSelectSede'] = True

  def getData(self,voice=None):
    if not voice:
      return self.data
    returnvalue = None
    try:
      returnvalue = self.data[voice]
    except KeyError:
      print('getData, chiave errata!')
    return returnvalue

  def setData(self,data,valore):
    if data in self.data:
      self.data[data] = valore
      return self.data
    self.data[data]=valore
    return self.data

class Define(object):
  def __init__(self,request):
    self.AziendaId = None
    self.SedeId = None
    self.AdminId = None

    if 'AziendaId' in request.session:
      self.AziendaId = request.session['AziendaId']
    if 'SedeId' in request.session:
      self.SedeId = request.session['SedeId']
    if 'AdminId' in request.session:
      self.AdminId = request.session['AdminId']

  def getAzienda(self):
    azienda=None
    if self.AziendaId:
      try:
        azienda = Azienda.objects.get(pk=self.AziendaId)
      except Azienda.DoesNotExist as dne:
        print('azienda non esiste')
    return azienda

  def getAziendaId(self):
    return self.AziendaId

  def getSede(self):
    sede=None
    if self.AziendaId:
      try:
        sede = Sede.objects.get(pk=self.SedeId)
      except Sede.DoesNotExist as dne:
        print('Sede Non esiste')
    return sede

  def getSedeId(self):
    return self.SedeId

  def getAdmin(self):
    admin = None
    if self.AdminId:
      try:
        admin = Amministratore.objects.get(pk=self.AdminId)
      except Amministratore.DoesNotExist as dne:
        print('Amministratore non esiste')
    return admin

  def getAdminId(self):
    return self.AdminId

  def setAziendaId(self,id=None):
    self.data['AziendaId'] = id
    return self.getAziendaId()

  def setSedeId(self,id=None):
    self.data['SedeId'] = id
    return self.getSedeId()

  def setAdminId(self,id=None):
    self.data['AdminId'] = id
    return self.getAdminId()