source: WAeUP_SRP/trunk/WAeUPTables.py @ 1713

Last change on this file since 1713 was 1707, checked in by joachim, 18 years ago

new import function for createStudents,
catalog fields added
  • Property svn:keywords set to Id
File size: 23.8 KB
Line 
1#-*- mode: python; mode: fold -*-
2# (C) Copyright 2005 AixtraWare <http://aixtraware.de>
3# Author: Joachim Schmitz <js@aixtraware.de>
4#
5# This program is free software; you can redistribute it and/or modify
6# it under the terms of the GNU General Public License version 2 as published
7# by the Free Software Foundation.
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
17# 02111-1307, USA.
18#
19# $Id: WAeUPTables.py 1707 2007-04-25 11:58:39Z joachim $
20
21from zope.interface import implements
22from Globals import InitializeClass
23from Products.ZCatalog.ZCatalog import ZCatalog
24from Products.ZCatalog.ProgressHandler import ZLogHandler
25from AccessControl import ClassSecurityInfo
26from Products.CMFCore.permissions import ModifyPortalContent
27import urllib
28import DateTime,time
29import csv,re
30import logging
31import Globals
32p_home = Globals.package_home(globals())
33i_home = Globals.INSTANCE_HOME
34from Products.AdvancedQuery import Eq, Between, Le,In
35
36from interfaces import IWAeUPTable
37
38class AttributeHolder(object):
39    pass
40
41def dict2ob(dict):
42    ob = AttributeHolder()
43    for key, value in dict.items():
44        setattr(ob, key, value)
45    return ob
46
47
48class WAeUPTable(ZCatalog): ###(
49
50    implements(IWAeUPTable)
51    security = ClassSecurityInfo()
52
53    def refreshCatalog(self, clear=0, pghandler=None):
54        """ don't refresh for a normal table """
55
56        if self.REQUEST and self.REQUEST.RESPONSE:
57            self.REQUEST.RESPONSE.redirect(
58              URL1 +
59              '/manage_catalogAdvanced?manage_tabs_message=Catalog%20refresh%20not%20implemented')
60
61    def manage_catalogClear(self, REQUEST=None, RESPONSE=None, URL1=None):
62        """ clears the whole enchilada """
63        #self._catalog.clear()
64
65        if REQUEST and RESPONSE:
66            RESPONSE.redirect(
67              URL1 +
68              '/manage_catalogAdvanced?manage_tabs_message=Catalog%20Clearing%20disabled')
69
70    def addRecord(self, **data):
71        # The uid is the same as "bed".
72        uid = data[self.key]
73        res = self.searchResults({"%s" % self.key : uid})
74        if len(res) > 0:
75            raise ValueError("More than one record with uid %s" % uid)
76        self.catalog_object(dict2ob(data), uid=uid)
77        return uid
78
79    def deleteRecord(self, uid):
80        self.uncatalog_object(uid)
81
82    def searchAndSetRecord(self, **data):
83        raise NotImplemented
84
85    def modifyRecord(self, **data):
86        #records = self.searchResults(uid=uid)
87        uid = data[self.key]
88        records = self.searchResults({"%s" % self.key : uid})
89        if len(records) > 1:
90            # Can not happen, but anyway...
91            raise ValueError("More than one record with uid %s" % uid)
92        if len(records) == 0:
93            raise KeyError("No record for uid %s" % uid)
94        record = records[0]
95        record_data = {}
96        for field in self.schema() + self.indexes():
97            record_data[field] = getattr(record, field)
98        # Add the updated data:
99        record_data.update(data)
100        self.catalog_object(dict2ob(record_data), uid)
101
102    def reindexIndex(self, name, REQUEST,pghandler=None):
103        if isinstance(name, str):
104            name = (name,)
105        paths = self._catalog.uids.items()
106        i = 0
107        #import pdb;pdb.set_trace()
108        for p,rid in paths:
109            i += 1
110            metadata = self.getMetadataForRID(rid)
111            record_data = {}
112            for field in name:
113                record_data[field] = metadata.get(field)
114            uid = metadata.get(self.key)
115            self.catalog_object(dict2ob(record_data), uid, idxs=name,
116                                update_metadata=0)
117
118    security.declareProtected(ModifyPortalContent,"exportAllRecords")
119    def exportAllRecords(self):
120        "export a WAeUPTable"
121        #import pdb;pdb.set_trace()
122        fields = [field for field in self.schema()]
123        format = ','.join(['"%%(%s)s"' % fn for fn in fields])
124        csv = []
125        csv.append(','.join(['"%s"' % fn for fn in fields]))
126        for uid in self._catalog.uids:
127            records = self.searchResults({"%s" % self.key : uid})
128            if len(records) > 1:
129                # Can not happen, but anyway...
130                raise ValueError("More than one record with uid %s" % uid)
131            if len(records) == 0:
132                raise KeyError("No record for uid %s" % uid)
133            rec = records[0]
134            csv.append(format % rec)
135        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
136        open("%s/import/%s-%s.csv" % (i_home,self.getId(),current),"w+").write('\n'.join(csv))
137
138###)
139
140class AccommodationTable(WAeUPTable): ###(
141
142    meta_type = 'WAeUP Accommodation Tool'
143    name = "accommodation"
144    key = "bed"
145    def __init__(self):
146        WAeUPTable.__init__(self, 'portal_accommodation')
147
148    def searchAndReserveBed(self, student_id,bed_type):
149        records = self.searchResults({'student' : student_id})
150        if len(records) > 0:
151            return -1,"Student with Id %s already booked bed %s." % (student_id,records[0].bed)
152
153        records = [r for r in self.searchResults({'bed_type' : bed_type}) if not r.student]
154        #import pdb;pdb.set_trace()
155        if len(records) == 0:
156            return -2,"No bed available"
157        rec = records[0]
158        self.modifyRecord(bed=rec.bed,student=student_id)
159        s_logger = logging.getLogger('WAeUPTables.AccommodationTable.searchAndReserveBed')
160        s_logger.info('%s reserved bed %s' % (student_id,rec.bed))
161        return 1,rec.bed
162
163
164InitializeClass(AccommodationTable)
165
166###)
167
168class PinTable(WAeUPTable): ###(
169    from ZODB.POSException import ConflictError
170    meta_type = 'WAeUP Pin Tool'
171    name = "pins"
172    key = 'pin'
173    def __init__(self):
174        WAeUPTable.__init__(self, 'portal_pins')
175
176
177    def searchAndSetRecord(self, uid, student_id,prefix):
178        #records = self.searchResults(uid=uid)
179        records = self.searchResults(student = student_id)
180        #import pdb;pdb.set_trace()
181        if len(records) > 0:
182            for r in records:
183                if r.pin != uid and r.prefix_batch.startswith(prefix):
184                    return -2
185        records = self.searchResults({"%s" % self.key : uid})
186        if len(records) > 1:
187            # Can not happen, but anyway...
188            raise ValueError("More than one record with uid %s" % uid)
189        if len(records) == 0:
190            return -1
191        record = records[0]
192        if record.student == "":
193            record_data = {}
194            for field in self.schema() + self.indexes():
195                record_data[field] = getattr(record, field)
196            # Add the updated data:
197            record_data['student'] = student_id
198            try:
199                self.catalog_object(dict2ob(record_data), uid)
200                return 1
201            except ConflictError:
202                return 2
203        if record.student.upper() != student_id.upper():
204            return 0
205        if record.student.upper() == student_id.upper():
206            return 2
207        return -3
208
209InitializeClass(PinTable)
210
211###)
212
213class PumeResultsTable(WAeUPTable): ###(
214
215    meta_type = 'WAeUP PumeResults Tool'
216    name = "pumeresults"
217    key = "jamb_reg_no"
218    def __init__(self):
219        WAeUPTable.__init__(self, 'portal_pumeresults')
220
221
222InitializeClass(PumeResultsTable)
223
224###)
225
226class StudentsCatalog(WAeUPTable): ###(
227    security = ClassSecurityInfo()
228
229    meta_type = 'WAeUP Students Catalog'
230    name = "students_catalog"
231    key = "id"
232    affected_types = {   ###(
233                      'StudentApplication':
234                          {'id': 'application',
235                           'fields':
236                             ('jamb_reg_no',
237                              'entry_mode',
238                              'entry_level',
239                              'entry_session',
240                              )
241                              },
242                      'StudentClearance':
243                          {'id': 'clearance',
244                           'fields':
245                             ('matric_no',
246                              )
247                              },
248                         'StudentPersonal':
249                          {'id': 'personal',
250                           'fields':
251                             ('name',
252                              'sex',
253                              'email',
254                              'phone',
255                              )
256                              },
257                         'StudentStudyCourse':
258                          {'id': 'study_course',
259                           'fields':
260                             ('course',
261                              'faculty',
262                              'department',
263                              'level',
264                              'mode',
265                              'session',
266                              'verdict',
267                              )
268                              },
269                         }
270    ###)
271
272    def __init__(self):
273        WAeUPTable.__init__(self, 'students_catalog')
274        return
275
276    def manage_catalogClear(self, REQUEST=None, RESPONSE=None, URL1=None):
277        """ clears the whole enchilada """
278        self._catalog.clear()
279
280        if REQUEST and RESPONSE:
281            RESPONSE.redirect(
282              URL1 +
283              '/manage_catalogAdvanced?manage_tabs_message=Catalog%20Cleared')
284
285    def manage_catalogReindex(self, REQUEST, RESPONSE, URL1): ###(
286        """ clear the catalog, then re-index everything """
287
288        elapse = time.time()
289        c_elapse = time.clock()
290
291        pgthreshold = self._getProgressThreshold()
292        handler = (pgthreshold > 0) and ZLogHandler(pgthreshold) or None
293        self.refreshCatalog(clear=1, pghandler=handler)
294
295        elapse = time.time() - elapse
296        c_elapse = time.clock() - c_elapse
297
298        RESPONSE.redirect(
299            URL1 +
300            '/manage_catalogAdvanced?manage_tabs_message=' +
301            urllib.quote('Catalog Updated \n'
302                         'Total time: %s\n'
303                         'Total CPU time: %s' % (`elapse`, `c_elapse`)))
304    ###)
305
306    def get_from_doc_department(self,doc): ###(
307        "return the students department"
308        if doc is None:
309            return None
310        certificate_res = self.portal_catalog(id = doc.study_course)
311        if len(certificate_res) != 1:
312            return None
313        return certificate_res[0].getPath().split('/')[-3]
314
315    def get_from_doc_faculty(self,doc):
316        "return the students faculty"
317        if doc is None:
318            return None
319        certificate_res = self.portal_catalog(id = doc.study_course)
320        if len(certificate_res) != 1:
321            return None
322        return certificate_res[0].getPath().split('/')[-4]
323
324    def get_from_doc_level(self,doc):
325        "return the students level"
326        if doc is None:
327            return None
328        return getattr(doc,'current_level',None)
329
330    def get_from_doc_mode(self,doc):
331        "return the students mode"
332        if doc is None:
333            return None
334        cm = getattr(doc,'current_mode',None)
335        return cm
336       
337
338    def get_from_doc_session(self,doc):
339        "return the students current_session"
340        if doc is None:
341            return None
342        return getattr(doc,'current_session',None)
343
344    def get_from_doc_entry_session(self,doc):
345        "return the students entry_session"
346        if doc is None:
347            return None
348        es = getattr(doc,'entry_session',None)
349        if len(es) == 2:
350            return es
351        try:
352            digit = int(doc.jamb_reg_no[0])
353        except:
354            return "xx"
355        if digit < 8:
356            return "0%c" % doc.jamb_reg_no[0]
357        return "9%c" % doc.jamb_reg_no[0]
358
359    def get_from_doc_session(self,doc):
360        "return the students session"
361        if doc is None:
362            return None
363        return getattr(doc,'current_session',None)
364
365    def get_from_doc_course(self,doc):
366        "return the students study_course"
367        if doc is None:
368            return None
369        return getattr(doc,'study_course',None)
370
371    def get_from_doc_name(self,doc):
372        "return the students name from the personal"
373        if doc is None:
374            return None
375        return "%s %s %s" % (doc.firstname,doc.middlename,doc.lastname)
376
377    def get_from_doc_verdict(self,doc):
378        "return the students study_course"
379        if doc is None:
380            return None
381        return getattr(doc,'current_verdict',None)
382    ###)
383
384    def reindexIndex(self, name, REQUEST,pghandler=None): ###(
385        if isinstance(name, str):
386            name = (name,)
387        reindextypes = {}
388        reindex_special = []
389        #import pdb;pdb.set_trace()
390        for n in name:
391            if n in ("review_state","registered_courses"):
392                reindex_special.append(n)
393            else:
394                for pt in self.affected_types.keys():
395                    if n in self.affected_types[pt]['fields']:
396                        if reindextypes.has_key(pt):
397                            reindextypes[pt].append(n)
398                        else:
399                            reindextypes[pt]= [n]
400                        break
401        students = self.portal_catalog(portal_type="Student")
402        aq_portal = self.portal_catalog.evalAdvancedQuery
403        num_objects = len(students)
404        if pghandler:
405            pghandler.init('Refreshing catalog: %s' % self.absolute_url(1), num_objects)
406        noattr = set(('StudentClearance','StudentPersonal')) & set(reindextypes.keys())
407        for i in xrange(num_objects):
408            if pghandler: pghandler.report(i)
409            student_brain = students[i]
410            student_object = student_brain.getObject()
411            data = {}
412            modified = False
413            sid = data['id'] = student_brain.getId
414            if reindex_special and 'review_state' in reindex_special:
415                modified = True
416                data['review_state'] = student_brain.review_state
417            sub_objects = False
418            for pt in reindextypes.keys():
419                modified = True
420                try:
421                    doc = getattr(student_object,self.affected_types[pt]['id']).getContent()
422                    sub_objects = True
423                except:
424                    continue
425                for field in self.affected_types[pt]['fields']:
426                    if hasattr(self,'get_from_doc_%s' % field):
427                        data[field] = getattr(self,'get_from_doc_%s' % field)(doc)
428                    else:
429                        data[field] = getattr(doc,field)
430                #from pdb import set_trace;set_trace()
431            if not sub_objects and noattr:
432                import_res = self.returning_import(id = sid)
433                if not import_res:
434                    continue
435                import_record = import_res[0]
436                data['matric_no'] = import_record.matric_no
437                data['sex'] = import_record.Sex == 'F'
438                data['name'] = "%s %s %s" % (import_record.Firstname,
439                                             import_record.Middlename,
440                                             import_record.Lastname)
441                data['matric_no'] = import_record.Entryregno
442            if reindex_special and 'registered_courses' in reindex_special:
443                query = Eq('id','study_course') & Eq('path',student_brain.getPath())
444                brains = aq_portal(query)
445                while brains:
446                    study_course_path = brains[0].getPath()
447                    level_brains = self.portal_catalog(path = study_course_path,
448                                                       portal_type = "StudentStudyLevel")
449                    if not level_brains:
450                        break
451                    modified = True
452                    level_ids = [l.getId for l in level_brains]
453                    level_ids.sort()
454                    for l in level_brains:
455                        if l.getId == level_ids[-1]:
456                            level_path = l.getPath()
457                            break
458                    result_brains = self.portal_catalog(path = level_path,
459                                                        portal_type = "StudentCourseResult")
460                    course_ids = [cr.getId for cr in result_brains]
461                    courses = []
462                    for c in course_ids:
463                        if c.endswith('_co'):
464                            courses.append(c[:-3])
465                        else:
466                            courses.append(c)
467                    data['registered_courses'] = courses
468                    break
469            if modified:
470                self.modifyRecord(**data)
471        if pghandler: pghandler.finish()
472    ###)
473
474    def refreshCatalog(self, clear=0, pghandler=None): ###(
475        """ re-index everything we can find """
476        students_folder = self.portal_url.getPortalObject().campus.students
477
478        cat = self._catalog
479        paths = self._catalog.uids.items()
480        if clear:
481            paths = tuple(paths)
482            cat.clear()
483        students = self.portal_catalog(portal_type="Student")
484        num_objects = len(students)
485        if pghandler:
486            pghandler.init('Refreshing catalog: %s' % self.absolute_url(1), num_objects)
487        for i in xrange(num_objects):
488            if pghandler: pghandler.report(i)
489            student_brain = students[i]
490            spath = student_brain.getPath()
491            student_obj = student_brain.getObject()
492            data = {}
493            sid = data['id'] = student_brain.getId
494            data['review_state'] = student_brain.review_state
495            sub_objects = False
496            for pt in self.affected_types.keys():
497                modified = True
498                try:
499                    doc = getattr(student_object,self.affected_types[pt]['id']).getContent()
500                    sub_objects = True
501                except:
502                    continue
503                for field in self.affected_types[pt]['fields']:
504                    if hasattr(self,'get_from_doc_%s' % field):
505                        data[field] = getattr(self,'get_from_doc_%s' % field)(doc)
506                    else:
507                        data[field] = getattr(doc,field)
508                #from pdb import set_trace;set_trace()
509            if not sub_objects and noattr:
510                import_res = self.returning_import(id = sid)
511                if not import_res:
512                    continue
513                import_record = import_res[0]
514                data['matric_no'] = import_record.matric_no
515                data['sex'] = import_record.Sex == 'F'
516                data['name'] = "%s %s %s" % (import_record.Firstname,
517                                             import_record.Middlename,
518                                             import_record.Lastname)
519                data['matric_no'] = import_record.Entryregno
520##            sub_brains = self.portal_catalog(path = spath)
521##            if len(sub_brains) > 1:
522##                for sub_brain in sub_brains:
523##                    if not sub_brain.portal_type in self.affected_types.keys():
524##                        continue
525##                    doc = sub_brain.getObject().getContent()
526##                    for field in self.affected_types[sub_brain.portal_type]:
527##                        if hasattr(self,'get_from_doc_%s' % field):
528##                            data[field] = getattr(self,'get_from_doc_%s' % field)(doc)
529##                        else:
530##                            data[field] = getattr(doc,field)
531##            elif len(sub_brains) == 1:
532##                #import pdb;pdb.set_trace()
533##                import_res = self.returning_import(id = sid)
534##                if not import_res:
535##                    continue
536##                import_record = import_res[0]
537##                data['matric_no'] = import_record.matric_no
538##                data['sex'] = import_record.Sex == 'F'
539##                data['name'] = "%s %s %s" % (import_record.Firstname,
540##                                             import_record.Middlename,
541##                                             import_record.Lastname)
542##                data['matric_no'] = import_record.Entryregno
543            study_course = getattr(student_obj,'study_course',None)
544            current_level = data.get('level',None)
545            data['registered_courses'] = []
546            if study_course and current_level and current_level in study_course.objectIds():
547                level_obj = getattr(study_course,current_level)
548                courses = []
549                for c in level_obj.objectIds():
550                    if c.endswith('_co'):
551                        courses.append(c[:-3])
552                    else:
553                        courses.append(c)
554                data['registered_courses'] = courses
555            self.addRecord(**data)
556        if pghandler: pghandler.finish()
557    ###)
558
559
560    security.declarePrivate('notify_event_listener') ###(
561    def notify_event_listener(self,event_type,object,infos):
562        "listen for events"
563        pt = getattr(object,'portal_type',None)
564        mt = getattr(object,'meta_type',None)
565        students_catalog = self.students_catalog
566        data = {}
567        if pt == 'Student' and\
568           mt == 'CPS Proxy Folder' and\
569           event_type.startswith('workflow'):
570            data['id'] = object.getId()
571            data['review_state'] = self.portal_workflow.getInfoFor(object,'review_state',None)
572            #from pdb import set_trace;set_trace()
573            students_catalog.modifyRecord(**data)
574            return
575        if pt not in self.affected_types.keys():
576            return
577        if not infos.has_key('rpath'):
578            return
579        rpl = infos['rpath'].split('/')
580        if pt == 'Student' and event_type == "sys_add_object":
581            student_id = object.id
582            try:
583                self.addRecord(id = student_id)
584            except ValueError:
585                pass
586            return
587        elif pt == 'CourseResult' and mt == 'CPS Proxy Folder':
588            pass
589        if event_type not in ('sys_modify_object'):
590            #from pdb import set_trace;set_trace()
591            return
592        if mt == 'CPS Proxy Folder':
593            return
594        for field in self.affected_types[pt]:
595            if hasattr(self,'get_from_doc_%s' % field):
596                data[field] = getattr(self,'get_from_doc_%s' % field)(object)
597            else:
598                data[field] = getattr(object,field)
599        data['id'] = rpl[2]
600        students_catalog.modifyRecord(**data)
601    ###)
602
603
604InitializeClass(StudentsCatalog)
605
606###)
607
608class CoursesCatalog(WAeUPTable): ###(
609
610    meta_type = 'WAeUP Courses Catalog'
611    name = "students_catalog"
612    key = "code"
613    def __init__(self):
614        WAeUPTable.__init__(self, 'courses_catalog')
615
616
617InitializeClass(CoursesCatalog)
618###)
619
620class OnlinePaymentsImport(WAeUPTable): ###(
621
622    meta_type = 'WAeUP Online Payment Transactions'
623    name = "online_payments_import"
624    key = "order_id"
625    def __init__(self):
626        WAeUPTable.__init__(self, self.name)
627
628
629InitializeClass(CoursesCatalog)
630###)
631
632class ReturningImport(WAeUPTable): ###(
633
634    meta_type = 'Returning Import Table'
635    name = "returning_import"
636    key = "matric_no"
637    def __init__(self):
638        WAeUPTable.__init__(self, 'returning_import')
639
640
641InitializeClass(ReturningImport)
642###)
643
644class ResultsImport(WAeUPTable): ###(
645
646    meta_type = 'Results Import Table'
647    name = "results_import"
648    key = "key"
649    def __init__(self):
650        WAeUPTable.__init__(self, 'results_import')
651
652
653InitializeClass(ResultsImport)
654
655###)
656
657class PaymentsCatalog(WAeUPTable): ###(
658
659    meta_type = 'WAeUP Payments Catalog'
660    name = "students_catalog"
661    key = "id"
662    def __init__(self):
663        WAeUPTable.__init__(self, 'payments_catalog')
664
665
666InitializeClass(PaymentsCatalog)
667
668###)
669
670# BBB:
671AccomodationTable = AccommodationTable
Note: See TracBrowser for help on using the repository browser.