import OFS
import getpass
import AccessControl
from OFS import SimpleItem
import types
import time
from TraversableWrapper import Traversable
from Globals import Persistent
from AccessControl import ClassSecurityInfo
from Cruft import Cruft
from common import add_dtml, translate, perm_edit, perm_view
from Products.ZCatalog.CatalogAwareness import CatalogAware

class Viide(
     OFS.SimpleItem.Item,
     Persistent,
     AccessControl.Role.RoleManager,
     Traversable,
     Cruft,
     CatalogAware
     ):
     """ Viide kataloogis """
     meta_type = 'Viide'
     security = ClassSecurityInfo()
     security.declareObjectPublic()
     a=5
		 
     def __init__(self, aadress, nimi, kirjeldus, aine):
       "Viite eksemplari loomine"
       self.aadress=aadress
       self.nimi=nimi
       self.kirjeldus=kirjeldus
       self.aine=aine
class ViideteKataloog(
    OFS.Folder.Folder,
    OFS.SimpleItem.Item,
    Persistent,
    AccessControl.Role.RoleManager,
    Traversable,
    Cruft
    ):
    """ Koht viidete hoidmiseks """
    
    security = ClassSecurityInfo()
    security.declarePrivate('kysiUusNr')
    security.declareObjectPublic()
    viiteNr=0
    ained=[]
		
    def __init__(self):
       """ init, set id """
       self.id = 'viited'
			 
    security.declarePrivate('kysiUusNr')	
    def kysiUusNr(self):
       self.viiteNr=self.viiteNr+1
       while hasattr(self.aq_self, 'viide'+str(self.viiteNr)):
         self.viideNr=self.viideNr+1
       return self.viiteNr
			 
    security.declareProtected(perm_edit, 'lisaViide')	
    def lisaViide(self, aadress, nimi, kirjeldus, aine):
	   nr = str(self.kysiUusNr())
	   v = Viide(aadress, nimi, kirjeldus, aine)
	   v.id = nr
	   self._setObject(nr, v)
	   return "Lisatud"
	 
    security.declareProtected(perm_edit, 'kustutaViide')
    def kustutaViide(self, REQUEST, id):
      "kustutus"
      self._delObject(id)
      REQUEST.RESPONSE.redirect('viiteVaatamisVorm')
		 
    def viiteLisamisVorm_kuular(self, REQUEST, aadress, nimi, kirjeldus, aine):
      "Viite admete salvestamine"
      self.lisaViide(aadress, nimi, kirjeldus, aine)
      REQUEST.RESPONSE.redirect('viiteVaatamisVorm')

    def kysiViited(self):
      "Viidete leidmine"
      return self.objectValues('Viide')
    
    def kysiViited2(self, aine_id):
      "Viidete leidmine"
      m1=self.objectValues('Viide')
      m2=[]
      for x in m1:
          if x.aine==aine_id:
             m2.append(x)
      return m2
      
    def parooli_kuular(self, REQUEST, nimi, parool):
      "Parool"
      if parool=="kala" and nimi=="opetaja":
         REQUEST.RESPONSE.redirect('viiteVaatamisVorm')
      else:
         REQUEST.RESPONSE.redirect('sisenemine')
         
    def aineLisamisVorm_kuular(self, REQUEST, aine):
      "Aine salvestamine"
      self.lisaAine(aine)
      REQUEST.RESPONSE.redirect('aineVaatamisVorm')
      
    security.declareProtected(perm_edit, 'lisaAine')
    def lisaAine(self, aine):
           if not hasattr(self, 'ained'):
              self.ained=[]
	   self.ained.append(aine);
	   self.ained=self.ained
	   return "Lisatud"
	   
    def kysiAined(self):
      "Aine leidmine"
      return self.ained