source: WAeUP_SRP/trunk/Accommodation.py @ 408

#-*- mode: python; mode: fold -*-
from Globals import InitializeClass
from AccessControl import ClassSecurityInfo

from Products.CMFCore.utils import UniqueObject, getToolByName
from Products.CMFCore.permissions import View
from Products.CMFCore.permissions import ModifyPortalContent
from Products.CPSCore.CPSBase import CPSBase_adder, CPSBaseFolder 
#from Products.CPSCore.CPSBase import CPSBaseDocument as BaseDocument 
from Products.CPSDocument.CPSDocument import CPSDocument 
from Products.CPSCore.CPSBase import CPSBaseBTreeFolder as BaseBTreeFolder 
from Products.WAeUP_SRP.WAeUPTables import AccommodationTable

class AccoFolder(CPSDocument): ###(
    """
    WAeUP AccoFolder containing Accommodation halls 
    """
    meta_type = 'AccoFolder'
    portal_type = meta_type
    security = ClassSecurityInfo()
    
    security.declareProtected(ModifyPortalContent,"generateFreeBedsList") ###(
    def generateFreeBedsList(self):
        """
        generate the free Bedslist.
        """
        freelist = AccommodationTable()
        l = self.portal_catalog({'meta_type': "Accommodation"}) 
        halls = []
        for h in l:
            halls.append(h.getObject())
        for hall in halls:
            h = hall.getContent()
            reserved = [int(r) for r in h.reserved_rooms.split()]
            for block in range(1,int(h.nr_of_blocks)):
                for floor in range(1,int(h.nr_of_floors)):
                    for room in range(1,int(h.rooms_per_floor)):
                        for bed in 'ABCDEFGH'[:int(h.beds_per_room)]:
                            room_nr = floor*100 + room
                            if room_nr in reserved:
                                print room_nr
                                continue
                            bt = 'xx'
                            if bed in h.beds_for_fresh:
                                bt = 'fr'
                            elif bed in h.beds_for_returning:
                                bt = 're'
                            elif bed in h.beds_for_final:
                                bt = 'fi'
                            print bt
                            freelist.addRecord(bed = '%s_%d_%s' % (h.getId(),room_nr,bed),
                                               bed_type = bt)
        return self.accommodation.academics_contents()

###)

InitializeClass(AccoFolder)

def addAccoFolder(container, id, REQUEST=None, **kw):
    """Add a AccoFolder."""
    ob = AccoFolder(id, **kw)
    return CPSBase_adder(container, ob, REQUEST=REQUEST)
###)

class Accommodation(CPSDocument): ###(
    """
    WAeUP Accommodation containing Departments 
    """
    meta_type = 'Accommodation'
    portal_type = meta_type
    security = ClassSecurityInfo()
    
    security.declareProtected(View,"Title") ###(
    def Title(self):
        """compose title"""
        content = self.getContent()
        heading = getattr(content,'heading',None)
        if heading is None:
            return self.title
        return heading

###)

        
                            
                
InitializeClass(Accommodation)

def addAccommodation(container, id, REQUEST=None, **kw):
    """Add a Accommodation."""
    ob = Accommodation(id, **kw)
    return CPSBase_adder(container, ob, REQUEST=REQUEST)
###)