source: main/waeup.kofa/trunk/src/waeup/kofa/hostels/hostel.py @ 15694

Last change on this file since 15694 was 15633, checked in by Henrik Bettermann, 5 years ago

Use factory to create beds and hostels.
  • Property svn:keywords set to Id
File size: 11.9 KB
Line 
1## $Id: hostel.py 15633 2019-10-03 10:09:25Z henrik $
2##
3## Copyright (C) 2011 Uli Fouquet & Henrik Bettermann
4## This program is free software; you can redistribute it and/or modify
5## it under the terms of the GNU General Public License as published by
6## the Free Software Foundation; either version 2 of the License, or
7## (at your option) any later version.
8##
9## This program is distributed in the hope that it will be useful,
10## but WITHOUT ANY WARRANTY; without even the implied warranty of
11## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12## GNU General Public License for more details.
13##
14## You should have received a copy of the GNU General Public License
15## along with this program; if not, write to the Free Software
16## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17##
18"""
19These are the hostels.
20"""
21import grok
22from zope.event import notify
23from zope.component import getUtility, createObject
24from zope.component.interfaces import IFactory
25from datetime import datetime
26from waeup.kofa.utils.helpers import attrs_to_fields
27from waeup.kofa.hostels.vocabularies import NOT_OCCUPIED
28from waeup.kofa.hostels.interfaces import IHostel, IBed
29from waeup.kofa.students.interfaces import IBedTicket
30from waeup.kofa.interfaces import IKofaUtils
31from waeup.kofa.interfaces import MessageFactory as _
32from waeup.kofa.utils.helpers import now
33
34class Hostel(grok.Container):
35    """This is a hostel.
36    """
37    grok.implements(IHostel)
38    grok.provides(IHostel)
39
40    @property
41    def bed_statistics(self):
42        total = len(self.keys())
43        booked = 0
44        for value in self.values():
45            if value.owner != NOT_OCCUPIED:
46                booked += 1
47        return {'booked':booked, 'total':total}
48
49    def clearHostel(self):
50        """Remove all beds
51        """
52        keys = [i for i in self.keys()] # create deep copy
53        for bed in keys:
54            del self[bed]
55        return
56
57    def addBed(self, bed):
58        """Add a bed.
59        """
60        if not IBed.providedBy(bed):
61            raise TypeError(
62                'Hostels contain only IBed instances')
63        self[bed.bed_id] = bed
64        return
65
66    def updateBeds(self):
67        """Fill hostel with beds or update beds.
68        """
69        added_counter = 0
70        modified_counter = 0
71        removed_counter = 0
72        modified_beds = u''
73
74        # Remove all empty beds. Occupied beds remain in hostel!
75        keys = list(self.keys()) # create list copy
76        for key in keys:
77            if self[key].owner == NOT_OCCUPIED:
78                del self[key]
79                self._p_changed = True
80                removed_counter += 1
81            else:
82                self[key].bed_number = 9999
83        remaining = len(keys) - removed_counter
84
85        blocks_for_female = getattr(self,'blocks_for_female',[])
86        blocks_for_male = getattr(self,'blocks_for_male',[])
87        beds_for_fresh = getattr(self,'beds_for_fresh',[])
88        beds_for_pre = getattr(self,'beds_for_pre',[])
89        beds_for_returning = getattr(self,'beds_for_returning',[])
90        beds_for_final = getattr(self,'beds_for_final',[])
91        beds_for_all = getattr(self,'beds_for_all',[])
92        all_blocks = blocks_for_female + blocks_for_male
93        all_beds = (beds_for_pre + beds_for_fresh +
94            beds_for_returning + beds_for_final + beds_for_all)
95        floor_base = 100
96        if self.rooms_per_floor > 99:
97            floor_base = 1000
98        for block in all_blocks:
99            sex = 'male'
100            if block in blocks_for_female:
101                sex = 'female'
102            for floor in range(1,int(self.floors_per_block)+1):
103                for room in range(1,int(self.rooms_per_floor)+1):
104                    for bed in all_beds:
105                        room_nr = floor*floor_base + room
106                        bt = 'all'
107                        if bed in beds_for_fresh:
108                            bt = 'fr'
109                        elif bed in beds_for_pre:
110                            bt = 'pr'
111                        elif bed in beds_for_final:
112                            bt = 'fi'
113                        elif bed in beds_for_returning:
114                            bt = 're'
115                        bt = u'%s_%s_%s' % (self.special_handling,sex,bt)
116                        uid = u'%s_%s_%d_%s' % (
117                            self.hostel_id,block,room_nr,bed)
118                        if uid in self:
119                            bed = self[uid]
120                            # Renumber remaining bed
121                            bed.bed_number = len(self) + 1 - remaining
122                            remaining -= 1
123                            if bed.bed_type != bt:
124                                bed.bed_type = bt
125                                modified_counter += 1
126                                modified_beds += '%s, ' % uid
127                                notify(grok.ObjectModifiedEvent(bed))
128                        else:
129                            bed = createObject(u'waeup.Bed')
130                            bed.bed_id = uid
131                            bed.bed_type = bt
132                            bed.bed_number = len(self) + 1 - remaining
133                            bed.owner = NOT_OCCUPIED
134                            self.addBed(bed)
135                            added_counter +=1
136        return removed_counter, added_counter, modified_counter, modified_beds
137
138    def writeLogMessage(self, view, message):
139        ob_class = view.__implemented__.__name__.replace('waeup.kofa.','')
140        self.__parent__.logger.info(
141            '%s - %s - %s' % (ob_class, self.__name__, message))
142        return
143
144Hostel = attrs_to_fields(Hostel)
145
146class Bed(grok.Container):
147    """This is a bed.
148    """
149    grok.implements(IBed)
150    grok.provides(IBed)
151
152    @property
153    def coordinates(self):
154        """Determine the coordinates from the bed_id.
155        """
156        return self.bed_id.split('_')
157
158    # The following property attributes are only needed
159    # for the exporter to ease evaluation with Excel.
160
161    @property
162    def hall(self):
163        return self.coordinates[0]
164
165    @property
166    def block(self):
167        return self.coordinates[1]
168
169    @property
170    def room(self):
171        return self.coordinates[2]
172
173    @property
174    def bed(self):
175        return self.coordinates[3]
176
177    @property
178    def special_handling(self):
179        return self.bed_type.split('_')[0]
180
181    @property
182    def sex(self):
183        return self.bed_type.split('_')[1]
184
185    @property
186    def bt(self):
187        return self.bed_type.split('_')[2]
188
189
190    def bookBed(self, student_id):
191        if self.owner == NOT_OCCUPIED:
192            self.owner = student_id
193            notify(grok.ObjectModifiedEvent(self))
194            return None
195        else:
196            return self.owner
197
198    def switchReservation(self):
199        """Reserves or unreserve bed respectively.
200        """
201        sh, sex, bt = self.bed_type.split('_')
202        hostel_id, block, room_nr, bed = self.coordinates
203        hostel = self.__parent__
204        beds_for_fresh = getattr(hostel,'beds_for_fresh',[])
205        beds_for_pre = getattr(hostel,'beds_for_pre',[])
206        beds_for_returning = getattr(hostel,'beds_for_returning',[])
207        beds_for_final = getattr(hostel,'beds_for_final',[])
208        bed_string = u'%s_%s_%s' % (block, room_nr, bed)
209        if bt == 'reserved':
210            bt = 'all'
211            if bed in beds_for_fresh:
212                bt = 'fr'
213            elif bed in beds_for_pre:
214                bt = 'pr'
215            elif bed in beds_for_final:
216                bt = 'fi'
217            elif bed in beds_for_returning:
218                bt = 're'
219            bt = u'%s_%s_%s' % (sh, sex, bt)
220            message = _(u'unreserved')
221        else:
222            bt = u'%s_%s_reserved' % (sh, sex)
223            message = _(u'reserved')
224        self.bed_type = bt
225        notify(grok.ObjectModifiedEvent(self))
226        return message
227
228    def releaseBed(self):
229        """Release bed.
230        """
231        if self.owner == NOT_OCCUPIED:
232            return
233        old_owner = self.owner
234        self.owner = NOT_OCCUPIED
235        notify(grok.ObjectModifiedEvent(self))
236        accommodation_session = grok.getSite()[
237            'hostels'].accommodation_session
238        try:
239            bedticket = grok.getSite()['students'][old_owner][
240                          'accommodation'][str(accommodation_session)]
241        except KeyError:
242            return '%s without bed ticket' % old_owner
243        bedticket.bed = None
244        tz = getUtility(IKofaUtils).tzinfo
245        timestamp = now(tz).strftime("%Y-%m-%d %H:%M:%S %Z")
246        bedticket.bed_coordinates = u'-- booking cancelled on %s --' % (
247            timestamp,)
248        return old_owner
249
250    def releaseBedIfMaintenanceNotPaid(self, n=7):
251        """Release bed if maintenance fee has not been paid on time.
252        Reserve bed so that it cannot be automatically booked by someone else.
253        """
254        if self.owner == NOT_OCCUPIED:
255            return
256        accommodation_session = grok.getSite()[
257            'hostels'].accommodation_session
258        try:
259            bedticket = grok.getSite()['students'][self.owner][
260                          'accommodation'][str(accommodation_session)]
261        except KeyError:
262            self.owner = NOT_OCCUPIED
263            sh, sex, bt = self.bed_type.split('_')
264            bt = u'%s_%s_reserved' % (sh, sex)
265            self.bed_type = bt
266            notify(grok.ObjectModifiedEvent(self))
267            return
268        if bedticket.maint_payment_made:
269            return
270        jetzt = datetime.utcnow()
271        days_ago = getattr(jetzt - bedticket.booking_date, 'days')
272        if days_ago >= n:
273            old_owner = self.owner
274            self.owner = NOT_OCCUPIED
275            sh, sex, bt = self.bed_type.split('_')
276            bt = u'%s_%s_reserved' % (sh, sex)
277            self.bed_type = bt
278            notify(grok.ObjectModifiedEvent(self))
279            bedticket.bed = None
280            tz = getUtility(IKofaUtils).tzinfo
281            timestamp = now(tz).strftime("%Y-%m-%d %H:%M:%S %Z")
282            bedticket.bed_coordinates = u'-- booking expired (%s) --' % (
283                timestamp,)
284            return old_owner
285        return
286
287    def writeLogMessage(self, view, message):
288        ob_class = view.__implemented__.__name__.replace('waeup.kofa.','')
289        self.__parent__.__parent__.logger.info(
290            '%s - %s - %s' % (ob_class, self.__name__, message))
291        return
292
293Bed = attrs_to_fields(Bed)
294
295class HostelFactory(grok.GlobalUtility):
296    """A factory for hostels.
297
298    We need this factory for the hostel processor.
299    """
300    grok.implements(IFactory)
301    grok.name(u'waeup.Hostel')
302    title = u"Create a new hostel.",
303    description = u"This factory instantiates new hostel instances."
304
305    def __call__(self, *args, **kw):
306        return Hostel()
307
308    def getInterfaces(self):
309        return implementedBy(Hostel)
310
311class BedFactory(grok.GlobalUtility):
312    """A factory for  beds.
313
314    We need this factory to ease customization.
315    """
316    grok.implements(IFactory)
317    grok.name(u'waeup.Bed')
318    title = u"Create a new bed.",
319    description = u"This factory instantiates new bed instances."
320
321    def __call__(self, *args, **kw):
322        return Bed()
323
324    def getInterfaces(self):
325        return implementedBy(Bed)
326
327
328@grok.subscribe(IBedTicket, grok.IObjectRemovedEvent)
329def handle_bedticket_removed(bedticket, event):
330    """If a bed ticket is deleted, we make sure that also the owner attribute
331    of the bed is cleared (set to NOT_OCCUPIED).
332    """
333    if bedticket.bed != None:
334        bedticket.bed.owner = NOT_OCCUPIED
335        notify(grok.ObjectModifiedEvent(bedticket.bed))
336
337@grok.subscribe(IBed, grok.IObjectRemovedEvent)
338def handle_bed_removed(bed, event):
339    """If a bed is deleted, we make sure that the bed object is
340    removed also from the owner's bed ticket.
341    """
342    if bed.owner == NOT_OCCUPIED:
343        return
344    accommodation_session = grok.getSite()['hostels'].accommodation_session
345    try:
346        bedticket = grok.getSite()['students'][bed.owner][
347                      'accommodation'][str(accommodation_session)]
348    except KeyError:
349        return
350    bedticket.bed = None
Note: See TracBrowser for help on using the repository browser.