source: WAeUP_SRP/trunk/Accommodation.py @ 4157

Last change on this file since 4157 was 3870, checked in by Henrik Bettermann, 16 years ago

reenable selection of beds for returning students
  • Property svn:keywords set to Id
File size: 12.7 KB
Line 
1#-*- mode: python; mode: fold -*-
2from Globals import InitializeClass
3from AccessControl import ClassSecurityInfo
4
5from Products.CMFCore.utils import UniqueObject, getToolByName
6from Products.CMFCore.permissions import View
7from Products.CMFCore.permissions import ModifyPortalContent
8from Products.CPSCore.CPSBase import CPSBase_adder, CPSBaseFolder
9#from Products.CPSCore.CPSBase import CPSBaseDocument as BaseDocument
10from Products.CPSDocument.CPSDocument import CPSDocument
11from Products.CPSCore.CPSBase import CPSBaseBTreeFolder as BaseBTreeFolder
12from Products.WAeUP_SRP.WAeUPTables import AccommodationTable,NOT_OCCUPIED
13import logging
14import csv,re,os
15import Globals
16import DateTime
17import re
18p_home = Globals.package_home(globals())
19i_home = Globals.INSTANCE_HOME
20
21
22class AccoFolder(CPSDocument): ###(
23    """
24    WAeUP AccoFolder containing Accommodation halls
25    """
26    meta_type = 'AccoFolder'
27    portal_type = meta_type
28    security = ClassSecurityInfo()
29
30    security.declareProtected(View,"Title")
31    def Title(self):
32        """compose title"""
33        return "Accommodation Section"
34
35    security.declareProtected(ModifyPortalContent,"generateFreeBedsList") ###(
36    def generateFreeBedsList(self):
37        """
38        generate the free Bedslist.
39        """
40        logger = logging.getLogger('Accommodation.AccoFolder.generateFreeBedsList')
41        member = self.portal_membership.getAuthenticatedMember()
42        logger.info('%s generated bed list' % member)
43        bed_list = self.portal_accommodation
44        l = self.portal_catalog({'meta_type': "AccoHall"})
45        halls = []
46        generated = []
47        generated.append('"Bed","BedType","Student"' % vars())
48        modyfied = []
49        modyfied.append('"Bed","BedType","Student"' % vars())
50        beds_generated = []
51        for h in l:
52            halls.append(h.getObject())
53        for hall in halls:
54            #import pdb;pdb.set_trace()
55            hall_doc = hall.getContent()
56            hall_gen = {}
57            hall_gen['name'] = hall.Title
58            count = 0
59            modyfied_count = 0
60            reserved = [(r.split('/')[0],int(r.split('/')[1])) for r in re.split(',|\.| ',hall_doc.reserved_rooms)
61                                     if r]
62            #import pdb;pdb.set_trace()                         
63            #for block in range(1,int(hall_doc.nr_of_blocks)+1):
64            # for block in 'ABCDEFGHIJ'[:int(hall_doc.nr_of_blocks)]:
65            blocks_for_female = getattr(hall_doc,'blocks_for_female',[])
66            blocks_for_male = getattr(hall_doc,'blocks_for_male',[])
67            beds_for_fresh = getattr(hall_doc,'beds_for_fresh',[])
68            beds_for_pre = getattr(hall_doc,'beds_for_pre',[])
69            beds_for_returning = getattr(hall_doc,'beds_for_returning',[])
70            beds_for_final = getattr(hall_doc,'beds_for_final',[])
71            if set(blocks_for_female).intersection(set(blocks_for_male)):
72                return self.accommodation.acco_folder_view(beds_generated=beds_generated)
73            for block in blocks_for_female + blocks_for_male:
74                sex = 'male'
75                if block in blocks_for_female:
76                    sex = 'female'
77                for floor in range(1,int(hall_doc.nr_of_floors)+1):
78                    for room in range(1,int(hall_doc.rooms_per_floor)+1):
79                        for bed in 'ABCDEFGHIKLMNOPQRSTUVWXYZ'[:int(hall_doc.beds_per_room)]:
80                            room_nr = floor*100 + room
81                            bt = 'all'
82                            if (block,room_nr) in reserved:
83                                bt = "reserved"
84                            elif hall_doc.special_handling and not hall_doc.special_handling.startswith("no"):
85                                bt = hall_doc.special_handling
86                            elif bed in beds_for_fresh:
87                                bt = 'fr'
88                            elif bed in beds_for_pre:
89                                bt = 'pr'
90                            elif bed in beds_for_final:
91                                bt = 'fi'
92                            elif bed in beds_for_returning:
93                                bt = 're'                               
94                            if hall_doc.special_handling.startswith("no_"):
95                                bt += "_" + hall_doc.special_handling[3:]
96                            bt = "%(sex)s_%(bt)s" % vars()
97                            uid = '%s_%s_%d_%s' % (hall.getId(),block,room_nr,bed)
98                            d = {}
99                            d['bed'] = uid
100                            d['bed_type'] = bt
101                            d['sort_id'] = getattr(hall_doc,'sort_id',0)
102                            d['hall'] = hall.getId()
103                            bed_record = bed_list.getRecordByKey(uid)
104                            if bed_record is None:
105                                d['student'] = NOT_OCCUPIED
106                                bed_list.addRecord(**d)
107                                count +=1
108                                generated.append('"%(uid)s","%(bt)s"' % vars())
109                            elif bed_record.bed_type != bt:
110                                bed_list.modifyRecord(**d)
111                                modyfied.append('"%(uid)s","%(bt)s"' % vars())
112                                modyfied_count += 1
113                                pass
114            hall_gen['count']= count
115            hall_gen['modyfied_count']= modyfied_count
116            beds_generated.append(hall_gen)
117        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
118        open("%s/export/bedlist_%s.csv" % (i_home,current),"w+").write('\n'.join(generated))
119        open("%s/export/bedlist_modyfied_%s.csv" % (i_home,current),"w+").write('\n'.join(modyfied))
120        return self.accommodation.acco_folder_view(beds_generated=beds_generated)
121    ###)
122
123    security.declareProtected(ModifyPortalContent,"importReservedBeds")###(
124    def importReservedBeds(self):
125        """load Reserved Beds from CSV"""
126        import transaction
127        import random
128        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
129        students_folder = self.portal_url.getPortalObject().campus.students
130        acco_folder = self.portal_url.getPortalObject().campus.accommodation
131        tr_count = 1
132        total = 0
133        name = 'ReservedBeds'
134        accommodation = self.portal_accommodation
135        students_cat = self.students_catalog
136        no_import = []
137        imported = []
138        logger = logging.getLogger('Accommodation.AccoFolder.importReservedBeds')
139        try:
140            beds = csv.DictReader(open("%s/import/%s.csv" % (i_home,name),"rb"))
141        except:
142            logger.error('Error reading %s.csv' % name)
143            return
144        halls = {}
145        start = True
146        for bed in beds:
147            if start:
148                start = False
149                logger.info('start loading from %s.csv' % name)
150                s = ','.join(['"%s"' % fn for fn in bed.keys()])
151                imported.append(s)
152                no_import.append('%s,"Error"' % s)
153                format = ','.join(['"%%(%s)s"' % fn for fn in bed.keys()])
154                format_error = format + ',"%(Error)s"'
155                no_certificate = "no certificate %s" % format
156            matric_no = bed.get('matric_no')
157            jamb_reg_no = bed.get('jamb_reg_no')
158            if matric_no != '':
159                res = students_cat(matric_no = matric_no)
160                if not res:
161                    bed['Error'] = "No such student"
162                    no_import.append( format_error % bed)
163                    continue
164                student = res[0]
165            elif jamb_reg_no != '':
166                res = students_cat(jamb_reg_no = jamb_reg_no)
167                if not res:
168                    bed['Error'] = "No such student"
169                    no_import.append( format_error % bed)
170                    continue
171                student = res[0]
172            else:
173                bed['Error'] = "No such student"
174                no_import.append( format_error % bed)
175                continue
176            sid = student.id
177            names = bed.get('name').split()
178            n = 0
179            for na in names:
180                if na.endswith(','):
181                    names[n] = na[:-1]
182                n += 1
183            s_names = student.name.split()
184            found = False
185            for sn in s_names:
186                if sn.upper() in names:
187                    found = True
188                    break
189            if not found:
190                bed['Error'] = "Name mismatch %s %s" % (str(names),str(s_names))
191                no_import.append( format_error % bed)
192                continue
193            hall = bed.get('hall')
194            if hall not in halls.keys():
195                hall_object = getattr(acco_folder,hall, None)
196                if hall_object is None:
197                    bed['Error'] = "No such hall"
198                    no_import.append( format_error % bed)
199                    continue
200                halls[hall] = hall_doc = hall_object.getContent()
201            bid = "%(hall)s_%(block)s_%(room)s_%(bed)s" % bed
202            res = accommodation(bed = bid)
203            psm = ''
204            while True:
205                if not res:
206                    psm = "No bed with this id"
207                    break
208                bed_brain = res[0]
209                if not bed_brain.bed_type.endswith("reserved"):
210                    psm = "Not a reserved bed"
211                    break
212                if bed_brain.student and bed_brain.student != NOT_OCCUPIED:
213                    psm = "Bed %s already reserved for %s" % (bid,bed_brain.student)
214                    break
215                if student.sex and not bed.get('block') in hall_doc.blocks_for_female:
216                    psm = "Sex does not match %s" % bid
217                    break
218                break
219            if psm != '':
220                bed['Error'] = psm
221                no_import.append( format_error % bed)
222                continue
223            already = accommodation(student=sid)
224            if already:
225                accommodation.modifyRecord(bed=already[0].bed,student=NOT_OCCUPIED)
226                student_obj = getattr(students_folder, student.id)
227                acco_id = "accommodation_%s" % self.getSessionId()[0]
228                if acco_id in student_obj.objectIds():
229                    acco_doc = getattr(student_obj, acco_id).getContent()
230                    info = self.getAccommodationInfo(sid)
231                    status = info['student_status']
232                    acco_doc.edit(mapping={'bed': bid,'student_status' : status})
233                    member = self.portal_membership.getAuthenticatedMember()
234                    logger.info('%s changed reserved bed %s for %s' % (member,bid,sid))
235            accommodation.modifyRecord(bed = bid, student = sid)
236            imported.append( format % bed)
237            tr_count += 1
238            if tr_count > 1000:
239                if len(no_import) > 0:
240                    open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
241                             '\n'.join(no_import) + '\n')
242                    no_import = []
243                open("%s/import/%simported%s.csv" % (i_home,name,current),"a").write(
244                                            '\n'.join(no_import) + "\n")
245                imported = []
246                em = '%d transactions commited total %s\n' % (tr_count,total)
247                transaction.commit()
248                regs = []
249                logger.info(em)
250                total += tr_count
251                tr_count = 0
252        #from pdb import set_trace;set_trace()
253        open("%s/import/%simported%s.csv" % (i_home,name,current),"a").write(
254                                            '\n'.join(imported))
255        open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
256                                                '\n'.join(no_import))
257        logger.info('finished loading from %s.csv' % name)
258        return self.REQUEST.RESPONSE.redirect("%s" % self.REQUEST.get('URL1'))
259
260
261InitializeClass(AccoFolder)
262
263def addAccoFolder(container, id, REQUEST=None, **kw):
264    """Add a AccoFolder."""
265    ob = AccoFolder(id, **kw)
266    return CPSBase_adder(container, ob, REQUEST=REQUEST)
267###)
268
269###)
270
271class AccoHall(CPSDocument): ###(
272    """
273    WAeUP AccoHall
274    """
275    meta_type = 'AccoHall'
276    portal_type = meta_type
277    security = ClassSecurityInfo()
278
279    security.declareProtected(View,"Title") ###(
280    def Title(self):
281        """compose title"""
282        content = self.getContent()
283        heading = getattr(content,'heading',None)
284        if heading is None:
285            return self.title
286        return heading
287    ###)
288
289InitializeClass(AccoHall)
290
291
292def addAccoHall(container, id, REQUEST=None, **kw):
293    """Add a AccoHall."""
294    ob = AccoHall(id, **kw)
295    return CPSBase_adder(container, ob, REQUEST=REQUEST)
296###)
Note: See TracBrowser for help on using the repository browser.