source: WAeUP_SRP/trunk/Accommodation.py @ 5213

Last change on this file since 5213 was 5208, checked in by Henrik Bettermann, 15 years ago

add student column to bedlist_modified
  • Property svn:keywords set to Id
File size: 12.8 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"' % vars())
48        modified = []
49        modified.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            modified_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                                st = bed_record.student
111                                bed_list.modifyRecord(**d)
112                                modified.append('"%(uid)s","%(bt)s","%(st)s"' % vars())
113                                modified_count += 1
114                                pass
115            hall_gen['count']= count
116            hall_gen['modified_count']= modified_count
117            beds_generated.append(hall_gen)
118        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
119        open("%s/export/bedlist_%s.csv" % (i_home,current),"w+").write('\n'.join(generated))
120        open("%s/export/bedlist_modified_%s.csv" % (i_home,current),"w+").write('\n'.join(modified))
121        return self.accommodation.acco_folder_view(beds_generated=beds_generated)
122    ###)
123
124    security.declareProtected(ModifyPortalContent,"importReservedBeds")###(
125    def importReservedBeds(self):
126        """load Reserved Beds from CSV"""
127        import transaction
128        import random
129        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
130        students_folder = self.portal_url.getPortalObject().campus.students
131        acco_folder = self.portal_url.getPortalObject().campus.accommodation
132        tr_count = 1
133        total = 0
134        name = 'ReservedBeds'
135        accommodation = self.portal_accommodation
136        students_cat = self.students_catalog
137        no_import = []
138        imported = []
139        logger = logging.getLogger('Accommodation.AccoFolder.importReservedBeds')
140        try:
141            beds = csv.DictReader(open("%s/import/%s.csv" % (i_home,name),"rb"))
142        except:
143            logger.error('Error reading %s.csv' % name)
144            return
145        halls = {}
146        start = True
147        for bed in beds:
148            if start:
149                start = False
150                logger.info('start loading from %s.csv' % name)
151                s = ','.join(['"%s"' % fn for fn in bed.keys()])
152                imported.append(s)
153                no_import.append('%s,"Error"' % s)
154                format = ','.join(['"%%(%s)s"' % fn for fn in bed.keys()])
155                format_error = format + ',"%(Error)s"'
156                no_certificate = "no certificate %s" % format
157            matric_no = bed.get('matric_no')
158            jamb_reg_no = bed.get('jamb_reg_no')
159            if matric_no != '':
160                res = students_cat(matric_no = matric_no)
161                if not res:
162                    bed['Error'] = "No such student"
163                    no_import.append( format_error % bed)
164                    continue
165                student = res[0]
166            elif jamb_reg_no != '':
167                res = students_cat(jamb_reg_no = jamb_reg_no)
168                if not res:
169                    bed['Error'] = "No such student"
170                    no_import.append( format_error % bed)
171                    continue
172                student = res[0]
173            else:
174                bed['Error'] = "No such student"
175                no_import.append( format_error % bed)
176                continue
177            sid = student.id
178            names = bed.get('name').split()
179            n = 0
180            for na in names:
181                if na.endswith(','):
182                    names[n] = na[:-1]
183                n += 1
184            s_names = student.name.split()
185            found = False
186            for sn in s_names:
187                if sn.upper() in names:
188                    found = True
189                    break
190            if not found:
191                bed['Error'] = "Name mismatch %s %s" % (str(names),str(s_names))
192                no_import.append( format_error % bed)
193                continue
194            hall = bed.get('hall')
195            if hall not in halls.keys():
196                hall_object = getattr(acco_folder,hall, None)
197                if hall_object is None:
198                    bed['Error'] = "No such hall"
199                    no_import.append( format_error % bed)
200                    continue
201                halls[hall] = hall_doc = hall_object.getContent()
202            bid = "%(hall)s_%(block)s_%(room)s_%(bed)s" % bed
203            res = accommodation(bed = bid)
204            psm = ''
205            while True:
206                if not res:
207                    psm = "No bed with this id"
208                    break
209                bed_brain = res[0]
210                if not bed_brain.bed_type.endswith("reserved"):
211                    psm = "Not a reserved bed"
212                    break
213                if bed_brain.student and bed_brain.student != NOT_OCCUPIED:
214                    psm = "Bed %s already reserved for %s" % (bid,bed_brain.student)
215                    break
216                if student.sex and not bed.get('block') in hall_doc.blocks_for_female:
217                    psm = "Sex does not match %s" % bid
218                    break
219                break
220            if psm != '':
221                bed['Error'] = psm
222                no_import.append( format_error % bed)
223                continue
224            already = accommodation(student=sid)
225            if already:
226                accommodation.modifyRecord(bed=already[0].bed,student=NOT_OCCUPIED)
227                student_obj = getattr(students_folder, student.id)
228                acco_id = "accommodation_%s" % self.getSessionId()[0]
229                if acco_id in student_obj.objectIds():
230                    acco_doc = getattr(student_obj, acco_id).getContent()
231                    info = self.getAccommodationInfo(sid)
232                    status = info['student_status']
233                    acco_doc.edit(mapping={'bed': bid,'student_status' : status})
234                    member = self.portal_membership.getAuthenticatedMember()
235                    logger.info('%s changed reserved bed %s for %s' % (member,bid,sid))
236            accommodation.modifyRecord(bed = bid, student = sid)
237            imported.append( format % bed)
238            tr_count += 1
239            if tr_count > 1000:
240                if len(no_import) > 0:
241                    open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
242                             '\n'.join(no_import) + '\n')
243                    no_import = []
244                open("%s/import/%simported%s.csv" % (i_home,name,current),"a").write(
245                                            '\n'.join(no_import) + "\n")
246                imported = []
247                em = '%d transactions commited total %s\n' % (tr_count,total)
248                transaction.commit()
249                regs = []
250                logger.info(em)
251                total += tr_count
252                tr_count = 0
253        #from pdb import set_trace;set_trace()
254        open("%s/import/%simported%s.csv" % (i_home,name,current),"a").write(
255                                            '\n'.join(imported))
256        open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
257                                                '\n'.join(no_import))
258        logger.info('finished loading from %s.csv' % name)
259        return self.REQUEST.RESPONSE.redirect("%s" % self.REQUEST.get('URL1'))
260
261
262InitializeClass(AccoFolder)
263
264def addAccoFolder(container, id, REQUEST=None, **kw):
265    """Add a AccoFolder."""
266    ob = AccoFolder(id, **kw)
267    return CPSBase_adder(container, ob, REQUEST=REQUEST)
268###)
269
270###)
271
272class AccoHall(CPSDocument): ###(
273    """
274    WAeUP AccoHall
275    """
276    meta_type = 'AccoHall'
277    portal_type = meta_type
278    security = ClassSecurityInfo()
279
280    security.declareProtected(View,"Title") ###(
281    def Title(self):
282        """compose title"""
283        content = self.getContent()
284        heading = getattr(content,'heading',None)
285        if heading is None:
286            return self.title
287        return heading
288    ###)
289
290InitializeClass(AccoHall)
291
292
293def addAccoHall(container, id, REQUEST=None, **kw):
294    """Add a AccoHall."""
295    ob = AccoHall(id, **kw)
296    return CPSBase_adder(container, ob, REQUEST=REQUEST)
297###)
Note: See TracBrowser for help on using the repository browser.