source: WAeUP_SRP/trunk/WAeUPImport.py @ 4013

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

students.xml: fix directory to enable listStudents

WAeUPImport.py: make password editable by import

  • Property svn:keywords set to Id
File size: 64.1 KB
Line 
1#-*- mode: python; mode: fold -*-
2# (C) Copyright 2005 The WAeUP group  <http://www.waeup.org>
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: WAeUPImport.py 3999 2009-03-05 17:27:01Z henrik $
20"""The WAeUP Tool Box.
21"""
22
23#from AccessControl import ClassSecurityInfo
24#from Acquisition import aq_inner
25#from Acquisition import aq_parent
26#from Globals import DTMLFile
27#from Globals import InitializeClass
28from OFS.SimpleItem import SimpleItem
29from zExceptions import BadRequest
30
31#from Products.CMFCore.utils import getToolByName
32#from Products.CPSSchemas.DataStructure import DataStructure
33#from Products.CPSSchemas.DataModel import DataModel
34#from Products.CPSSchemas.StorageAdapter import MappingStorageAdapter
35from Products.CMFCore.ActionProviderBase import ActionProviderBase
36#from Products.CMFCore.permissions import View
37#from Products.ZCatalog.ZCatalog import ZCatalog
38#from Products.CMFCore.permissions import ModifyPortalContent
39#from Products.CMFCore.permissions import ManagePortal
40from Products.CMFCore.utils import UniqueObject
41#from Products.CMFCore.URLTool import URLTool
42from Products.CMFCore.utils import getToolByName
43from Globals import package_home,INSTANCE_HOME
44from Products.AdvancedQuery import Eq, Between, Le,In
45import csv,re,os,sys
46from shutil import copy2
47import DateTime,time
48import logging
49p_home = package_home(globals())
50i_home = INSTANCE_HOME
51from utils import makeDigest
52
53NO_KEY = '----'
54IGNORE = 'ignore'
55class WAeUPImport(UniqueObject, SimpleItem, ActionProviderBase): ###(
56    """ WAeUPImport """
57    required_modes = ('create',)
58
59    def __init__(self,waeup_tool): ###(
60        self.students_folder = waeup_tool.portal_url.getPortalObject().campus.students
61        self.member = member = waeup_tool.portal_membership.getAuthenticatedMember()
62        self.import_date = DateTime.DateTime()
63        self.imported_by = str(member)
64        #self.current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
65        self.waeup_tool = waeup_tool
66        self.academics_folder = waeup_tool.portal_url.getPortalObject().campus.academics
67        self.schema_tool = getToolByName(waeup_tool, 'portal_schemas')
68        self.layout_tool = getToolByName(waeup_tool, 'portal_layouts')
69        self.types_tool = getToolByName(waeup_tool, 'portal_types')
70        self.portal_workflow = getToolByName(waeup_tool, 'portal_workflow')
71        self.portal_url = getToolByName(waeup_tool, 'portal_url')
72        self.portal_catalog = waeup_tool.portal_catalog
73        self.portal_directories = waeup_tool.portal_directories
74        self.students_catalog = waeup_tool.students_catalog
75        self.courses_catalog = waeup_tool.courses_catalog
76        self.course_results = waeup_tool.course_results
77        self.payments_catalog = waeup_tool.payments_catalog
78        self.applicants_catalog = waeup_tool.applicants_catalog
79        #self.mode = mode
80        # self.import_method = getattr(self, '%s' % mode,None)
81        errors = []
82        # if self.import_method is None:
83        #     errors.append('No importer method %s' % mode)
84        self.pending_path = "%s/import/%s.pending" % (i_home,self.plural_name)
85        self.pending_tmp = "%s/import/%s.pending.tmp" % (i_home,self.plural_name)
86        self.pending_backup = "%s/import/%s.pending.old" % (i_home,self.plural_name)
87        self.pending_fn = os.path.split(self.pending_path)[1]
88        self.imported_path = "%s/import/%s.imported" % (i_home,self.plural_name)
89        self.imported_fn = os.path.split(self.imported_path)[1]
90        iname = "import_%s" % self.name
91        self.logger = logging.getLogger('WAeUPImport.%sImport' % self.plural_name.capitalize())
92        self.schema = self.schema_tool._getOb(iname,None)
93        #self.pending_schema = self.schema_tool._getOb("%s_pending" % iname,None)
94        self.layout = self.layout_tool._getOb(iname,None)
95        while True:
96            if self.schema is None:
97                errors.append('no schema %s' % iname)
98            # if self.pending_schema is None:
99            #     self.pending_schema = self.schema
100            if self.layout is None:
101                errors.append('no such layout %s' % iname)
102            if errors:
103                break
104            self.data_keys = self.schema.keys()
105            self.csv_keys = self.schema.keys()
106            info = {}
107            info['imported_from'] = ''
108            info['import_record_no'] = 0
109            info['imported_by'] = self.imported_by
110            info['import_date'] = self.import_date.strftime("%d/%m/%y %H:%M:%S")
111            info['error'] = ''
112            self.info = info
113            self.csv_keys.extend(self.info)
114            self.validators = {}
115            for widget in self.layout.keys():
116                self.validators[widget] = self.layout[widget].validate
117            self.required_keys = {}
118            for mode in self.required_modes:
119                self.required_keys[mode] = [self.layout.getIdUnprefixed(id)
120                                    for id,widget in self.layout.objectItems()
121                                    if widget.is_required]
122            break
123        self.init_errors = ','.join(errors)
124    ###)
125
126    def findStudent(self,mode,student_id=None, matric_no=None, jamb_reg_no=None): ###(
127        student_record = None
128        msg = ''
129        key_used = ''
130        while True:
131            if student_id:
132                key_used = 'student_id'
133                res = self.students_catalog(id = student_id)
134                if not res:
135                    msg = "no student with id %s" % student_id
136                    break
137                student_record = res[0]
138                if matric_no and student_record.matric_no:
139                    if  matric_no != student_record.matric_no:
140                        msg = "old matric_no %s overwritten with %s" % (student_record.matric_no,matric_no)
141                        #logger.info("%s, old matric_no %s overwritten with %s" % (student_record.id,student_record.matric_no,matric_no))
142                if jamb_reg_no and student_record.jamb_reg_no:
143                    if jamb_reg_no != student_record.jamb_reg_no:
144                        msg = "old reg_no %s overwritten with %s" % (student_record.jamb_reg_no,jamb_reg_no)
145                        #logger.info("%s, old reg_no %s overwritten with %s" % (student_record.id,student_record.jamb_reg_no,jamb_reg_no))
146            elif jamb_reg_no:
147                key_used = 'jamb_reg_no'
148                res = self.students_catalog(jamb_reg_no = jamb_reg_no)
149                if res:
150                    if mode == 'create':
151                        msg = "jamb_reg_no exists"
152                        break
153                else:
154                    if mode == 'edit':
155                        msg = "no student with jamb_reg_no %s" % jamb_reg_no
156                        break
157                student_record = res[0]
158            elif matric_no:
159                key_used = 'matric_no'
160                res = self.students_catalog(matric_no = matric_no)
161                if not res:
162                    msg = "no student with matric_no %s" % matric_no
163                    break
164                student_record = res[0]
165            else:
166                msg = "neither id, matric_no nor reg_no specified"
167            break
168        d = {}
169        d['student_record'] = student_record
170        d['key_used'] = key_used
171        d['msg'] = msg
172        return d
173    ###)
174
175    def makeIdLists(self): ###(
176        pending_digests = []
177        pending = []
178        # pending_student_ids = []
179        # pending_matric_nos = []
180        # pending_reg_ns = []
181        if os.path.exists(self.pending_path):
182            datafile = open(self.pending_path,"r")
183            reader = csv.reader(datafile)
184            old_headline = reader.next()
185            #datafile.seek(0)
186            pending_csv_reader = csv.DictReader(datafile,old_headline,)
187            #pending_csv_reader.next() # skip headline
188            for item in pending_csv_reader:
189                digest = makeDigest(item,self.data_keys)
190                if digest not in pending_digests:
191                    pending_digests += digest,
192                    pending += item,
193            datafile.close()
194            #copy2(self.pending_path,self.pending_backup)
195        return pending, pending_digests
196    ###)
197
198    def checkHeadline(self,headline): ###(
199        """ check the headline of a csv file """
200        import_keys = [k.strip() for k in headline if not (k.strip().startswith('ignore')
201                                                        or k.strip() in self.info.keys())]
202        # import_keys = [k.strip() for k in headline if not (k.strip() in self.info.keys())]
203        diff2schema = set(import_keys).difference(set(self.schema.keys()))
204        diff2layout = set(import_keys).difference(set(self.layout.keys()))
205        #if diff2schema and diff2schema != set(['id',]):
206        if diff2schema:
207            return list(diff2schema)
208        return []
209    ###)
210
211    def getHeadlineFields(self,headline,values): ###(
212        """ check the headline of a csv file """
213        # import_keys = [k.strip() for k in headline if not (k.strip().startswith('ignore')
214        #                                                 or k.strip() in self.info.keys())]
215        import_keys = [k.strip() for k in headline if not (k.strip() in self.info.keys())]
216        info_keys = [k.strip() for k in headline if k.strip() in self.info.keys()]
217        si = set(import_keys)
218        ss = set(self.schema.keys())
219
220        invalid_keys = si - ss
221        keys = []
222        i = 0
223        duplicates = False
224        singels = []
225        msg = ''
226        while True:
227            if len(values) != len(import_keys):
228                if len(values) == len(import_keys) + len(info_keys):
229                    msg += "fields from pending file in headline"
230                else:
231                    msg += "%d fields in headline but %d values" % (len(import_keys),len(values))
232                break
233            for k in import_keys:
234                if k in singels:
235                    keys += (k,'%s' % k,values[i],'(duplicate)'),
236                    msg += ("duplicate %s," % k)
237                    keys[singels.index(k)] = (k,'%s' % k,values[singels.index(k)],'(duplicate)')
238                elif k in invalid_keys and not k.startswith(IGNORE):
239                    keys += (k,NO_KEY,values[i],'(invalid)'),
240                else:
241                    keys += (k,k,values[i],''),
242                i += 1
243                singels += k,
244            break
245        return msg,keys
246    ###)
247###)
248
249class ApplicationImport(WAeUPImport):###(
250    name = "application"
251    plural_name = "%ss" % name
252    commit_after = 1000
253
254    def create(self,mapping):###(
255        reg_no = mapping.get('reg_no')
256        msg = ''
257        while True:
258            try:
259                self.applicants_catalog.addRecord(**mapping)
260            except ValueError:
261                msg =  "applicant record with reg_no %s already exists" % reg_no
262            break
263        return reg_no,msg,mapping
264    ###)
265
266    def edit(self,mapping):###(
267        reg_no = mapping.get('reg_no')
268        status = mapping.get('status')
269        msg = ''
270        while True:
271            res = self.applicants_catalog(reg_no = reg_no)
272            if len(res):
273                if res[0].status == 'created' and status != 'created':
274                    msg =  "student object with id %s for %s already created, status cannot be changed" % (res[0].student_id, reg_no)
275                elif status == 'created' and res[0].status != 'created':
276                    msg =  "student object for %s has not yet been created, status cannot be set to 'created'" % (reg_no)
277                else:
278                    self.applicants_catalog.modifyRecord(**mapping)
279            else:
280                msg =  "applicant record with reg_no %s does not exist" % reg_no
281            break
282        return reg_no,msg,mapping
283    ###)
284
285###)
286
287class CertificateImport(WAeUPImport):###(
288    name = "certificate"
289    plural_name = "%ss" % name
290    commit_after = 100000
291
292    def create(self,mapping):###(
293        if getattr(self,'_v_certificate_list',None) is None:
294            self._v_certificate_list = []
295        if getattr(self,'_v_department_certificates',None) is None:
296            departments = self.portal_catalog(portal_type = "Department")
297            self._v_department_certificates = {}
298            for department  in departments:
299                certificates_container = getattr(department.getObject(),"certificates",None)
300                self._v_department_certificates[department.getId] = {'container': certificates_container,
301                                                            'certificates': certificates_container.objectIds(),
302                                                            }
303        department_id = mapping['department_code']
304        msg = ''
305        certificate_id = mapping.get('code')
306        while True:
307            department_certificates = self._v_department_certificates.get(department_id,None)
308            if department_certificates is None:
309                msg =  "No Department with ID: %s" % department_id
310                break
311            certificates_container = department_certificates['container']
312            certificates = department_certificates['certificates']
313            if certificate_id in self._v_certificate_list:
314                msg =  "Duplicate Certificate ID: %s" % department_id
315                break
316            if certificate_id in certificates:
317                msg =  "Duplicate Certificate ID: %s" % department_id
318                break
319            try:
320                certificates_container.invokeFactory('Certificate', certificate_id)
321            except BadRequest,E:
322                msg =  "%s" % E
323                break
324            self._v_certificate_list.append(certificate_id)
325            certificate = getattr(certificates_container,certificate_id)
326            certificate.getContent().edit(mapping=mapping)
327            break
328        return certificate_id,msg,mapping
329    ###)
330
331    def edit(self,mapping):###(
332        certificate_id = mapping.get('code')
333        res = self.portal_catalog(id=certificate_id)
334        msg = ''
335        while True:
336            if not res:
337                msg =  "no certificate with id: %s" % certificate_id
338                break
339            c = res[0].getObject()
340            c.getContent().edit(mapping=mapping)
341            break
342        return certificate_id,msg,mapping
343    ###)
344###)
345
346class CourseImport(WAeUPImport):###(
347    name = "course"
348    plural_name = "%ss" % name
349    commit_after = 1000
350
351    def create(self,mapping):###(
352        if getattr(self,'_v_course_list',None) is None:
353            self._v_course_list = []
354        if getattr(self,'_v_department_courses',None) is None:
355            departments = self.portal_catalog(portal_type = "Department")
356            self._v_department_courses = {}
357            for department in departments:
358                courses_container = getattr(department.getObject(),"courses",None)
359                if courses_container is not None:
360                    self._v_department_courses[department.getId] = {'container': courses_container,
361                                                                    'courses': courses_container.objectIds(),
362                                                                   }
363        department_id = mapping['department_code']
364        course_id = mapping.get('code','')
365        msg = ''
366        while True:
367            department_courses = self._v_department_courses.get(department_id,None)
368            if department_courses is None:
369                msg =  "no department with id: %(department_id)s" % vars()
370                break
371            courses_container = department_courses['container']
372            courses = department_courses['courses']
373            if course_id in self._v_course_list:
374                msg =  "duplicate course id: %(course_id)s" % vars()
375                break
376            if course_id in courses:
377                msg =  "course %(course_id)s already exists in department %(department_id)s" % vars()
378                break
379            try:
380                courses_container.invokeFactory('Course', course_id)
381            except BadRequest,E:
382                msg =  "%s" % E
383                break
384            self._v_course_list.append(course_id)
385            course = getattr(courses_container,course_id)
386            course.getContent().edit(mapping=mapping)
387            break
388        return course_id,msg,mapping
389    ###)
390
391    def edit(self,mapping): ###(
392        course_id = mapping.get('code','')
393        course = self.courses_catalog.getRecordByKey(course_id)
394        msg = ''
395        while True:
396            if course is None:
397                msg =  "no course with id: %s" % course_id
398                break
399            course_object = getattr(getattr(getattr(getattr(self.academics_folder,course.faculty),
400                                    course.department),
401                                    'courses'),
402                                    course_id)
403            course_object.getContent().edit(mapping=mapping)
404            break
405        return course_id,msg,mapping
406    ###)
407
408    def remove(self,mapping): ###(
409        course_id = mapping.get('code','')
410        course = self.courses_catalog.getRecordByKey(course_id)
411        msg = ''
412        while True:
413            if course is None:
414                msg =  "no course with id: %s" % course_id
415                break
416            courses = getattr(getattr(getattr(self.academics_folder,course.faculty),
417                                    course.department),
418                                    'courses')
419            courses.manage_delObjects((course_id),)     
420            break
421        return course_id,msg,mapping
422    ###)   
423   
424###)
425
426class CourseResultImport(WAeUPImport):###(
427    """ CourseresultImport """
428    name = "course_result"
429    plural_name = "%ss" % name
430    commit_after = 1000000
431    required_modes = ('create','edit','remove')
432
433    def create(self,mapping):###(
434        students_folder = self.portal_url.getPortalObject().campus.students
435        if getattr(self,'_v_courses',None) is None:
436            res = self.courses_catalog()
437            self._v_courses = {}
438            for brain in res:
439                self._v_courses[brain.code] = brain
440        if getattr(self,'_v_level_created',None) is None:
441            self._v_level_created = []
442        if getattr(self,'_v_student_study_course',None) is None:
443            self._v_student_study_course = {}
444        msg = ''
445        key = ''
446        matric_no = mapping.get('matric_no','')
447        id = mapping.get('id','')
448        while True:
449            course_id = mapping.get('code')
450            if course_id not in self._v_courses.keys():
451                msg = "no course with id: %s" % course_id
452                break
453            result = self.findStudent('create',student_id=id,matric_no=matric_no)
454            msg = result['msg']
455            if msg:
456                break
457            student_record = result['student_record']
458            student_id = mapping['student_id']  = student_record.id
459            level_id = mapping.get('level_id','')
460            code = mapping.get('code','')
461            level_ident = "%(student_id)s_%(level_id)s" % vars()
462            if level_ident not in self._v_level_created:
463                context = self._v_student_study_course.get(student_id,None)
464                if context is None:
465                    try:
466                        context = getattr(getattr(students_folder,
467                                                "%(student_id)s" % vars()),
468                                        'study_course')
469                        self._v_student_study_course[student_id] = context
470                    except:
471                        msg = "could not create level %(level_id)s for %(student_id)s" % vars()
472                        break
473                if level_id not in context.objectIds():
474                    context.invokeFactory('StudentStudyLevel',"%(level_id)s" % vars())
475                    level = getattr(context,"%(level_id)s" % vars())
476                    self.portal_workflow.doActionFor(level,'open')
477                    session_id = mapping.get('session_id','')
478                    if session_id:
479                        level.getContent().edit(mapping={'session': "%s" % session_id,})
480                    self.portal_workflow.doActionFor(level,'close')
481                    self._v_level_created += level_ident,
482            mapping['key'] = key = "%(student_id)s|%(level_id)s|%(code)s" % vars()
483            #overwrite semester and credits in create mode
484            for k in ('semester','credits',):
485                mapping[k] = getattr(self._v_courses[course_id],k)
486            try:
487                self.course_results.addRecord(**mapping)
488            except ValueError:
489                msg = "course result already exists: %s" % key
490            break
491        return key,msg,mapping
492    ###)
493
494    def edit(self,mapping): ###(
495        msg = ''
496        key = ''
497        matric_no = mapping.get('matric_no','')
498        id = mapping.get('id','')
499        while True:
500            result = self.findStudent('edit',student_id=id,matric_no=matric_no)
501            msg = result['msg']
502            if msg:
503                break
504            student_record = result['student_record']
505            student_id = student_record.id
506            level_id = mapping.get('level_id','')
507            code = mapping.get('code','')
508            #code = mapping['code']
509            mapping['key'] = key = "%(student_id)s|%(level_id)s|%(code)s" % vars()
510            if self.course_results.getRecordByKey(key) is None:
511                msg =  "no course result with key %(key)s" % vars()
512                break
513            self.course_results.modifyRecord(**mapping)
514            break
515        return key,msg,mapping
516    ###)
517
518    def remove(self,mapping):###(
519        key = ''
520        msg = ''
521        matric_no = mapping.get('matric_no','')
522        id = mapping.get('id','')
523        while True:
524            result = self.findStudent('edit',student_id=id,matric_no=matric_no)
525            msg = result['msg']
526            if msg:
527                break
528            student_record = result['student_record']
529            student_id = student_record.id
530            level_id = mapping.get('level_id','')
531            code = mapping.get('code','')
532            key = "%(student_id)s|%(level_id)s|%(code)s" % vars()
533            if self.course_results.getRecordByKey(key) is None:
534                msg =  "no course result with key %(key)s" % vars()
535                break
536            self.course_results.deleteRecord(key)
537            break
538        return key,msg,mapping
539    ###)
540###)
541
542class CertificateCourseImport(WAeUPImport):###(
543    name = "certificate_course"
544    plural_name = "%ss" % name
545    commit_after = 1000
546    required_modes = ('create','edit')
547
548    def create(self,mapping):
549        if getattr(self,'_v_courses',None) is None:
550            res = self.courses_catalog()
551            self._v_courses= [course.code for course in res]
552        if getattr(self,'_v_ceritficates',None) is None:
553            res = self.portal_catalog(portal_type = "Certificate")
554            self._v_certificates = {}
555            for cert in res:
556                self._v_certificates[cert.getId] = cert.getObject()
557        msg = ''
558        while True:
559            certificate_course_id = mapping.get('code')
560            if certificate_course_id not in self._v_courses:
561                msg =  "no course with id: %s" % certificate_course_id
562                break
563            cert_id = mapping['certificate_code']
564            cert = self._v_certificates.get(cert_id,None)
565            if cert is None:
566                msg =  "no certificate with id: %s" % cert_id
567                break
568            level_id = mapping.get('level')
569            level = getattr(cert,level_id,None)
570            if level is None:
571                cert.invokeFactory('StudyLevel', level_id)
572                level = getattr(cert,level_id,None)
573            elif hasattr(level,certificate_course_id):
574                msg =  "duplicate certificate course id: %(certificate_course_id)s " % vars()
575                msg += "in %(cert_id)s/ %(level_id)s" % vars()
576                break
577            level.invokeFactory('CertificateCourse', certificate_course_id)
578            c = getattr(level,certificate_course_id)
579            c.getContent().edit(mapping=mapping)
580            break
581        return certificate_course_id,msg,mapping
582    ###)
583
584class DepartmentImport(WAeUPImport):###(
585    name = "department"
586    plural_name = "%ss" % name
587    commit_after = 1000
588
589    def create(self,mapping):###(
590        "create a department in the correct faculty"
591        faculty_id = mapping['faculty_code']
592        msg = ''
593        if getattr(self,'_v_faculties',None) is None:
594            res = self.portal_catalog(portal_type = "Faculty")
595            self._v_faculties = {}
596            for f in res:
597                self._v_faculties[f.getId] = f.getObject()
598        department_id = mapping.get('code','')
599        while True:
600            faculty = self._v_faculties.get(faculty_id,None)
601            if faculty is None:
602                msg =  "no faculty with id: %s" % faculty_id
603                break
604            else:
605                d = getattr(faculty,department_id,None)
606                if d is None or d.portal_type == "Faculty":
607                    try:
608                        faculty.invokeFactory('Department', department_id)
609                    except BadRequest,E:
610                        msg =  "%s" % E
611                        break
612                    d = getattr(faculty,department_id)
613                    d.invokeFactory('CoursesFolder','courses')
614                    courses = getattr(d,'courses')
615                    dict = {'Title': 'Courses'}
616                    courses.getContent().edit(mapping=dict)
617                    d.invokeFactory('CertificatesFolder','certificates')
618                    certificates = getattr(d,'certificates')
619                    dict = {'Title': 'Certificates'}
620                    certificates.getContent().edit(mapping=dict)
621                    d.getContent().edit(mapping=mapping)
622            break
623        return department_id,msg,mapping
624    ###)
625
626    def edit(self,mapping): ###(
627        "edit a department in the correct faculty"
628        academics_folder = self.portal_url.getPortalObject().campus.academics
629        faculty_id = mapping['faculty_code']
630        department_id = mapping.get('code','')
631        msg = ''
632        while True:
633            try:
634                d = getattr(getattr(academics_folder,faculty_id),department_id,None)
635            except KeyError:
636                msg =  "department %s or faculty %s wrong" % (department_id,faculty_id)
637                break
638            if d is None or d.portal_type == "Faculty":
639                msg =  "department %s not found" % (department_id)
640                break
641            d.getContent().edit(mapping=mapping)
642            break
643        return department_id,msg,mapping
644    ###)
645###)
646
647class FacultyImport(WAeUPImport):###(
648    name = "faculty"
649    plural_name = "faculties"
650    commit_after = 100
651   
652
653    def create(self,mapping): ###(
654        "create a faculty"
655        academics_folder = self.portal_url.getPortalObject().campus.academics
656        faculty_id = mapping.get('code','')
657        msg = ''
658        while True:
659            if faculty_id in academics_folder.objectIds():
660                msg =  "faculty with id: %s exists" % faculty_id
661                break
662            try:
663                academics_folder.invokeFactory('Faculty', faculty_id)
664            except BadRequest,E:
665                msg =  "%s" % E
666                break
667            f = getattr(academics_folder,faculty_id,None)
668            f.getContent().edit(mapping=mapping)
669            break
670        return faculty_id,msg,mapping
671        ###)
672
673    def edit(self,mapping): ###(
674        "edit a faculty"
675        academics_folder = self.portal_url.getPortalObject().campus.academics
676        faculty_id = mapping['code']
677        msg = ''
678        while True:
679            f = getattr(academics_folder,faculty_id,None)
680            if f is None:
681                msg =  "faculty with id: %s does not exist" % faculty_id
682            f.getContent().edit(mapping=mapping)
683            break
684        return faculty_id,msg,mapping
685    ###)
686###)
687
688class StudentImport(WAeUPImport):###(
689    name = "student"
690    plural_name = "%ss" % name
691    commit_after = 100
692
693    field2types_student = {   ###(
694                      'StudentApplication':
695                          {'id': 'application',
696                           #'title': 'Application Data',
697                           'wf_transition_return': 'close',
698                           'wf_transition_admit': 'remain',
699                           'wf_transition_graduate': 'close',
700                           'wf_transition_pay_school_fee': 'close',
701                           'wf_transition_validate_courses': 'close',
702                           'fields':
703                             ('jamb_reg_no',
704                              'entry_mode',
705                              'entry_session',
706                              'jamb_score',
707                              'app_email',
708                              'app_mobile',
709                              'jamb_age',
710                              'jamb_state',
711                              'jamb_lga',
712                              'jamb_sex',
713                              'app_ac_pin',
714                              'app_reg_pin',
715                              'app_ac_date',                             
716                              )
717                              },
718                      'StudentClearance':
719                          {'id': 'clearance',
720                           #'title': 'Clearance/Eligibility Record',
721                           'wf_transition_return': 'close',
722                           'wf_transition_admit': 'remain',
723                           'wf_transition_graduate': 'close',
724                           'wf_transition_pay_school_fee': 'close',
725                           'wf_transition_validate_courses': 'close',
726                           'fields':
727                              ('matric_no',
728                              'nationality',
729                              'lga',
730                              'birthday',
731                              'clr_ac_pin',
732                              'request_date',
733                              'cleared_date',
734                              'clearance_officer',
735                              )
736                              },
737                         'StudentPersonal':
738                          {'id': 'personal',
739                           #'title': 'Personal Data',
740                           'wf_transition_return': 'open',
741                           'wf_transition_admit': 'remain',
742                           'wf_transition_graduate': 'close',
743                           'wf_transition_pay_school_fee': 'open',
744                           'wf_transition_validate_courses': 'open',
745                           'fields':
746                             ('firstname',
747                              'middlename',
748                              'lastname',
749                              'sex',
750                              'email',
751                              'phone',
752                              'perm_address',
753                              'marit_stat',
754                              'disabled',
755                              )
756                              },
757                         'StudentStudyCourse':
758                          {'id': 'study_course',
759                           #'title': 'Study Course',
760                           'wf_transition_return': 'open',
761                           'wf_transition_admit': 'remain',
762                           'wf_transition_graduate': 'close',
763                           'wf_transition_pay_school_fee': 'open',
764                           'wf_transition_validate_courses': 'open',
765                           'fields':
766                             ('study_course',
767                              'current_level',
768                              'current_session',
769                              'current_mode',  #is no longer used and visible but can still be imported
770                              'current_verdict',
771                              'previous_verdict',
772                              )
773                              },
774                         # 'StudentStudyLevel':
775                         #  {'id': 'current_level',
776                         #   'title': '',
777                         #   'wf_transition_return': 'open',
778                         #   'wf_transition_admit': 'remain',
779                         #   'fields':
780                         #     ('verdict',
781                         #      'session',
782                         #      )
783                         #      },
784                         'PaymentsFolder':
785                          {'id': 'payments',
786                           #'title': 'Payments',
787                           'wf_transition_return': 'open',
788                           'wf_transition_admit': 'open',
789                           'wf_transition_graduate': 'close',
790                           'wf_transition_pay_school_fee': 'open',
791                           'wf_transition_validate_courses': 'open',
792                           'fields':
793                             ()
794                              },
795                         }
796    ###)
797
798    def create(self,mapping): ###(
799        "create student records due import"
800        logger = logging.getLogger('WAeUPImport.StudentImport.create')
801        students_folder = self.portal_url.getPortalObject().campus.students
802        jamb_reg_no = mapping.get('jamb_reg_no',None)
803        matric_no = mapping.get('matric_no',None)
804        entry_mode = mapping.get('entry_mode',None)
805        password = mapping.get('password',None)
806        msg = ''
807        student_id = mapping.get('id',None)
808        while True:
809            if student_id:
810                #msg = "student_id must not be specified in create mode"
811                #break
812                res = self.students_catalog(id = student_id)
813                if res:
814                    msg = "student_id %s already assigned" % res[0].id
815                    break               
816            else:
817                student_id = self.waeup_tool.generateStudentId('?')
818            if jamb_reg_no and not entry_mode == 'transfer':
819                res = self.students_catalog(jamb_reg_no = jamb_reg_no)
820                if res:
821                    msg = "jamb_reg_no already assigned to student %s" % res[0].id
822                    break
823            if matric_no:
824                res = self.students_catalog(matric_no = matric_no)
825                if res:
826                    msg = "matric_no already assigned to student %s" % res[0].id
827                    break
828            if not matric_no and not jamb_reg_no:
829                msg = "jamb_reg_no or matric_no must be specified"
830                break
831           
832            students_folder.invokeFactory('Student', student_id)
833            student_obj = getattr(students_folder,student_id)
834            f2t = self.field2types_student
835            d = {}
836            transition = mapping.get('reg_transition','admit')
837            if transition not in ('admit','return','pay_school_fee','validate_courses','graduate'):
838                msg = "no valid transition provided"
839                break
840            for pt in f2t.keys():
841                student_obj.invokeFactory(pt,f2t[pt]['id'])
842                sub_obj = getattr(student_obj,f2t[pt]['id'])
843                sub_doc = sub_obj.getContent()
844                #d['Title'] = f2t[pt]['title']
845                for field in f2t[pt]['fields']:
846                    d[field] = mapping.get(field,'')
847
848                if pt == "StudentApplication":
849                    #d['jamb_sex']  = 'M'
850                    #if mapping.get('sex'):
851                    #    d['jamb_sex']  = 'F'
852                    d['jamb_firstname'] = mapping.get('firstname',None)
853                    d['jamb_middlename'] = mapping.get('middlename',None)
854                    d['jamb_lastname'] = mapping.get('lastname',None)
855
856                # if pt == "StudyCourse":
857                #     for von,zu in (('entry_mode','current_mode'),
858                #                    ('entry_session','current_session')):
859                #         if mapping.get(zu,None) is None and mapping.get(von,None) is not None:
860                #             d[zu] = mapping[von]
861                sub_doc.edit(mapping = d)
862
863                #import pdb;pdb.set_trace()
864                new_state = f2t[pt]['wf_transition_%(transition)s' % vars()]
865                if new_state != "remain":
866                    self.portal_workflow.doActionFor(sub_obj,new_state,dest_container=sub_obj)
867            self.portal_workflow.doActionFor(student_obj,transition)
868            student_obj.manage_setLocalRoles(student_id, ['Owner',])
869            mapping['id'] = student_id
870            if password:
871                self.waeup_tool.makeStudentMember(student_id,password)
872            break
873        return student_id,msg,mapping
874    ###)
875
876    def remove(self,mapping): ###(
877        "remove student records due import"
878        logger = logging.getLogger('WAeUPImport.StudentImport.remove')
879        students_folder = self.portal_url.getPortalObject().campus.students
880        stdir = self.portal_directories.students
881        student_id = mapping.get('id',None)
882        msg = ''
883        export_file = "%s/export/students_removed.csv" % (i_home)
884        reimport_file = "%s/export/students_for_reimport.csv" % (i_home)
885       
886        toexport_for_reimport = {
887       
888                    'application':
889                             ('jamb_reg_no',
890                              'entry_mode',
891                              'entry_session',
892                              'jamb_score',
893                              'app_email',
894                              'app_mobile',
895                              'jamb_age',
896                              'jamb_state',
897                              'jamb_lga',
898                              'jamb_sex',
899                              'app_ac_pin',
900                              'app_reg_pin',
901                              'app_ac_date',                             
902                              ),
903                    'clearance':
904                              ('matric_no',
905                              'nationality',
906                              'lga',
907                              'birthday',
908                              'clr_ac_pin',
909                              'request_date',
910                              'cleared_date',
911                              'clearance_officer',
912                              ),
913                    'study_course':
914                              ('study_course',
915                              'current_level',
916                              'current_session',
917                              'current_verdict',
918                              'previous_verdict',
919                              ),
920                    'personal': 
921                             ('firstname',
922                              'middlename',
923                              'lastname',
924                              'sex',
925                              'email',
926                              'phone',
927                              'perm_address',
928                              'marit_stat',
929                              'disabled',
930                              ),
931                   }
932
933        sub_types = ['StudentApplication','StudentClearance','StudentPersonal','StudentStudyCourse']
934        non_image_keys = {}
935        for i in sub_types:
936            pt_ob = getattr(self.types_tool,i)
937            sub_schemas = pt_ob.schemas
938            for i in sub_schemas:
939                if i.startswith('student_'):
940                    schema = getattr(self.schema_tool,i)
941                    field_id = i.replace('student_','')
942                    non_image_keys[field_id] = [key for key in schema.keys()
943                                                if schema[key].meta_type != "CPS Image Field"]
944       
945        # export field names must be unique but, unfortunately, firstname, middlename and lastname are not
946        # thus we have to omit them
947        non_image_keys['clearance'].remove('firstname')
948        non_image_keys['clearance'].remove('middlename')
949        non_image_keys['clearance'].remove('lastname')
950        toexport = non_image_keys
951
952        while True:
953            if hasattr(students_folder,student_id):
954                # begin export
955                line = []
956                fields = ['student_id','reg_state','password']
957                line_for_reimport = []
958                fields_for_reimport = ['student_id','reg_state','password','import_mode']               
959                #fields = []
960                #for f in self.students_catalog.schema():
961                #    fields.append(f)               
962                for k in toexport.keys():
963                    for f in toexport[k]:
964                        fields.append(f)
965                for k in toexport_for_reimport.keys():
966                    for f in toexport_for_reimport[k]:
967                        fields_for_reimport.append(f)                       
968                if not os.path.exists(export_file):       
969                    headline = ','.join(fields)
970                    open(export_file,"a").write(headline +'\n')
971                if not os.path.exists(reimport_file): 
972                    headline_for_reimport = ','.join(fields_for_reimport)                   
973                    open(reimport_file,"a").write(headline_for_reimport +'\n')                   
974                format = '"%(' + ')s","%('.join(fields) + ')s"'
975                format_for_reimport = '"%(' + ')s","%('.join(fields_for_reimport) + ')s"'
976                res = self.students_catalog(id = student_id)
977                student = res[0]
978                student_obj = getattr(students_folder,student_id)
979                #d = student.getFormattedStudentEntry(student)
980                d = {'student_id':student_id,'reg_state':student.review_state,'import_mode':'create'}
981                for k in toexport.keys()[0:]:
982                    try:
983                        object = getattr(student_obj,k)
984                        object_doc = object.getContent()
985                    except:
986                        continue
987                    for f in toexport[k]:
988                        d[f] = getattr(object_doc,f,'')
989                if hasattr(stdir, student_id):
990                    d['password'] = self.waeup_tool.getCredential(student_id)
991                    stdir.deleteEntry(student_id)
992                self.waeup_tool.removePictureFolder(student_id)       
993                line.append(format % d)
994                open(export_file,"a").write('\n'.join(line) +'\n')
995                line_for_reimport.append(format_for_reimport % d)
996                open(reimport_file,"a").write('\n'.join(line_for_reimport) +'\n')               
997                self.course_results.exportRemoveAllCourses(student_id = student_id, export = True, remove = True)
998                self.payments_catalog.exportRemoveAllPayments(student_id = student_id, export = True, remove = True)
999                self.waeup_tool.exportAllStudyLevels(student_id = student_id)
1000                # end export
1001                students_folder.manage_delObjects((student_id),)         
1002            else:
1003                msg = "Student object not found"
1004                break
1005     
1006            break
1007        return student_id,msg,mapping     
1008
1009
1010    def edit(self,mapping): ###(
1011        "edit student records due import"
1012        wftool = self.portal_workflow
1013        logger = logging.getLogger('WAeUPImport.StudentImport.edit')
1014        students_folder = self.portal_url.getPortalObject().campus.students
1015        student_id = mapping.get('id',None)
1016        jamb_reg_no = mapping.get('jamb_reg_no',None)
1017        matric_no = mapping.get('matric_no',None)
1018        editable_keys = mapping.keys()
1019        password = mapping.get('password',None)
1020        msg = ''
1021        while True:
1022            if student_id:
1023                res = self.students_catalog(id = student_id)
1024                if not res:
1025                    msg = "no student with id %s" % student_id
1026                    break
1027                student_record = res[0]
1028                if matric_no and student_record.matric_no:
1029                    if  matric_no != student_record.matric_no and not matric_no == 'transferred':
1030                        res = self.students_catalog(matric_no = matric_no)
1031                        if res:
1032                            msg = "matric_no already assigned to student %s" % res[0].id
1033                            break
1034                        msg = "old matric_no %s overwritten with %s" % (student_record.matric_no,matric_no)
1035                        #logger.info("%s, old matric_no %s overwritten with %s" % (student_record.id,student_record.matric_no,matric_no))
1036                if jamb_reg_no and student_record.jamb_reg_no:
1037                    if jamb_reg_no != student_record.jamb_reg_no:
1038                        res = self.students_catalog(jamb_reg_no = jamb_reg_no)
1039                        if res:
1040                            msg = "jamb_reg_no already assigned to student %s" % res[0].id
1041                            break
1042                        msg = "old jamb_reg_no %s overwritten with %s" % (student_record.jamb_reg_no,jamb_reg_no)
1043                        #logger.info("%s, old reg_no %s overwritten with %s" % (student_record.id,student_record.jamb_reg_no,jamb_reg_no))
1044            elif jamb_reg_no:
1045                res = self.students_catalog(jamb_reg_no = jamb_reg_no)
1046                if not res:
1047                    msg = "no student with jamb_reg_no %s" % jamb_reg_no
1048                    break
1049                student_record = res[0]
1050                editable_keys.remove('jamb_reg_no')
1051            elif matric_no:
1052                res = self.students_catalog(matric_no = matric_no)
1053                if not res:
1054                    msg = "no student with matric_no %s" % matric_no
1055                    break
1056                student_record = res[0]
1057                editable_keys.remove('matric_no')
1058            student_id = student_record.id
1059            student_obj = getattr(students_folder,student_id)
1060            f2t = self.field2types_student
1061            d = {}
1062            any_change = False
1063            #special treatment for StudentStudyLevel
1064            d['verdict']  = mapping.get('current_verdict','')
1065            d['session']  = mapping.get('current_session','')
1066            current_level = mapping.get('current_level','')
1067            transition = mapping.get('reg_transition',None)
1068            # the validate_courses import transition is not really useful because it does not execute validate_courses.py
1069            if transition and transition not in ('admit','return','graduate','pay_school_fee','validate_courses'):
1070                msg = "no valid transition provided"
1071                break     
1072            while d['session'] and d['verdict'] and current_level:
1073                sub_obj = getattr(student_obj,'study_course',None)
1074                if sub_obj is None:
1075                    break
1076                level_obj = getattr(sub_obj,current_level,None)
1077                if  level_obj is None:
1078                    break
1079                any_change = True
1080                level_obj.getContent().edit(mapping = d)
1081                try:
1082                    wftool.doActionFor(level_obj,'close')
1083                except:
1084                    pass
1085                break
1086           
1087            for pt in f2t.keys():
1088                #if pt == "StudentApplication":
1089                #    d['jamb_sex']  = 'M'
1090                #    if mapping.get('sex'):
1091                #        d['jamb_sex']  = 'F'
1092                intersect = set(f2t[pt]['fields']).intersection(set(editable_keys))
1093                #import pdb;pdb.set_trace()
1094                object_id = f2t[pt]['id']
1095                sub_obj = getattr(student_obj,object_id,None)
1096                if intersect and pt not in ('StudentStudyLevel',):
1097                    if sub_obj is None:
1098                        try:
1099                            student_obj.invokeFactory(pt,object_id)
1100                        except:
1101                            continue
1102                        sub_obj = getattr(student_obj,object_id)
1103                        #if f2t[pt]['title'] != '':
1104                        #    d['Title'] = f2t[pt]['title']
1105                    sub_doc = sub_obj.getContent()
1106                    for field in intersect:
1107                        changed = False
1108                        if getattr(sub_doc,field,None) != mapping.get(field,''):
1109                            any_change = True
1110                            changed = True
1111                            d[field] = mapping.get(field,'')
1112                        if changed:
1113                            sub_doc.edit(mapping = d)
1114                if transition:
1115                    new_state = f2t[pt]['wf_transition_%(transition)s' % vars()]
1116                    if new_state != "remain":
1117                        try:
1118                            self.portal_workflow.doActionFor(sub_obj,new_state,dest_container=sub_obj)
1119                        except:
1120                            pass
1121            if transition:
1122                try:
1123                    self.portal_workflow.doActionFor(student_obj,transition)
1124                except:
1125                    msg = "reg_transition not allowed"
1126            if password:
1127                self.waeup_tool.editPassword(student_id,password)
1128            break
1129        return student_id,msg,mapping
1130    ###)
1131###)
1132
1133class StudentStudyLevelImport(WAeUPImport):###(
1134    """ StudentStudyLevelImport """
1135    name = "student_study_level"
1136    plural_name = "%ss" % name
1137    commit_after = 1000000
1138    required_modes = ('create',)
1139
1140    def create(self,mapping): ###(
1141        "edit student levels and create StudentStudyLevel object if not existent"
1142        wftool = self.portal_workflow
1143        logger = logging.getLogger('WAeUPImport.StudentStudyLevelImport.create')
1144        students_folder = self.portal_url.getPortalObject().campus.students
1145        student_id = mapping.get('id',None)
1146        matric_no = mapping.get('matric_no',None)
1147        editable_keys = mapping.keys()
1148        key = ''
1149        msg = ''
1150        while True:
1151            result = self.findStudent('create',student_id=student_id,matric_no=matric_no)
1152            msg = result['msg']
1153            if msg:
1154                break
1155            student_record = result['student_record']
1156            student_id = student_record.id
1157            mapping['id'] = student_id
1158            session = mapping.get('session','')
1159            level = key = mapping.get('code','')
1160            if not level.isdigit():
1161                msg = 'student %s: imported level is empty' % (student_id,)
1162                break
1163            study_course_obj = getattr(getattr(students_folder,student_id),'study_course',None)
1164            if study_course_obj is None:
1165                msg = 'student %s: no study_course object' % student_id
1166                break
1167            level_obj = getattr(study_course_obj,level,None)
1168            if  level_obj is None:
1169                # The only difference to the edit method is that we create a StudentStudyLevel object
1170                try:
1171                    study_course_obj.invokeFactory('StudentStudyLevel',"%s" % level)
1172                    level_obj = getattr(context,"%s" % level)
1173                except:
1174                    continue
1175            level_obj.getContent().edit(mapping = mapping)
1176            level_review_state = wftool.getInfoFor(level_obj,'review_state',None)
1177            if level_review_state != "closed":
1178                wftool.doActionFor(level_obj,'close')
1179            break
1180        return key,msg,mapping
1181    ###)
1182
1183    def edit(self,mapping): ###(
1184        "edit student levels"
1185        wftool = self.portal_workflow
1186        logger = logging.getLogger('WAeUPImport.StudentStudyLevelImport.edit')
1187        students_folder = self.portal_url.getPortalObject().campus.students
1188        student_id = mapping.get('id',None)
1189        matric_no = mapping.get('matric_no',None)
1190        key = ''
1191        msg = ''
1192        while True:
1193            result = self.findStudent('create',student_id=student_id,matric_no=matric_no)
1194            msg = result['msg']
1195            if msg:
1196                break
1197            student_record = result['student_record']
1198            student_id = student_record.id
1199            mapping['id'] = student_id
1200            session = mapping.get('session','')
1201            level = key = mapping.get('code','')
1202            #import pdb;pdb.set_trace()
1203            if not level.isdigit():
1204                msg = 'student %s: imported level is empty' % (student_id,)
1205                break
1206            study_course_obj = getattr(getattr(students_folder,student_id),'study_course',None)
1207            if study_course_obj is None:
1208                msg = 'student %s: no study_course object' % student_id
1209                break
1210            level_obj = getattr(study_course_obj,level,None)
1211            if  level_obj is None:
1212                msg = 'student %s: no study_level object for level %s' % (student_id,level)
1213                break
1214            level_obj.getContent().edit(mapping = mapping)
1215            break
1216        return key,msg,mapping
1217    ###)
1218###)
1219
1220class VerdictImport(WAeUPImport):###(
1221    """ VerdictImport """
1222    name = "verdict"
1223    plural_name = "%ss" % name
1224    commit_after = 1000
1225    required_modes = ('create','edit')
1226
1227    def create(self,mapping): ###(
1228        "edit student verdicts and create StudentStudyLevel object if not existent"
1229        wftool = self.portal_workflow
1230        logger = logging.getLogger('WAeUPImport.VerdictImport.create')
1231        students_folder = self.portal_url.getPortalObject().campus.students
1232        student_id = mapping.get('id',None)
1233        matric_no = mapping.get('matric_no',None)
1234        editable_keys = mapping.keys()
1235        key = ''
1236        while True:
1237            result = self.findStudent('create',student_id=student_id,matric_no=matric_no)
1238            #student_record,msg = self.getStudentRecord(mapping)
1239            msg = result['msg']
1240            if msg:
1241                break
1242            student_record = result['student_record']
1243            student_id = student_record.id
1244            mapping['id'] = student_id
1245            d = {}
1246            #import pdb;pdb.set_trace()
1247            any_change = False
1248            #special treatment for StudentStudyLevel
1249            current_session = d['session'] = mapping.get('current_session','')
1250            if current_session and student_record.session != current_session:
1251                msg = 'student %s: imported session %s does not match current_session %s' % (student_id,
1252                                                                                            current_session,
1253                                                                                            student_record.session)
1254                break
1255            current_level = mapping.get('current_level','')
1256            if not current_level.isdigit():
1257                msg = 'student %s: imported level is empty' % (student_id,)
1258                break
1259            if current_level and student_record.level != current_level:
1260                msg = 'student %s: imported level %s does not match current_level %s' % (student_id,
1261                                                                                        current_level,
1262                                                                                        student_record.level)
1263                break
1264            student_review_state =  student_record.review_state
1265            if student_review_state == 'deactivated':
1266                msg = "student %s in review_state %s" % (student_id, student_review_state)
1267                break
1268            if student_review_state not in ('courses_validated','returning'):
1269                msg = "student %s in wrong review_state %s" % (student_id, student_review_state)
1270                break
1271            student_obj = getattr(students_folder,student_id)
1272            # f2t = self.field2types_student
1273            study_course_obj = getattr(student_obj,'study_course',None)
1274            if study_course_obj is None:
1275                msg = 'student %s: no study_course object' % student_id
1276                break
1277            level_obj = getattr(study_course_obj,current_level,None)
1278
1279            if  level_obj is None:
1280                # The only difference to the edit method is that we create a StudentStudyLevel object
1281                try:
1282                    study_course_obj.invokeFactory('StudentStudyLevel',"%s" % current_level)
1283                    level_obj = getattr(context,"%s" % current_level)
1284                    level_obj.portal_workflow.doActionFor(level,'open')
1285                except:
1286                    continue
1287                #msg = 'student %s: no study_level object for level %s' % (student_id,
1288                #                                                                current_level)
1289                #break
1290
1291            verdict = d['verdict'] = d['current_verdict']  = mapping.get('current_verdict','')
1292
1293            #if verdict == student_record.verdict:
1294            #    msg = 'student %s: verdict already set to %s' % (student_id,
1295            #                                                            verdict)
1296
1297            level_review_state = wftool.getInfoFor(level_obj,'review_state',None)
1298            if level_review_state != "closed":
1299                wftool.doActionFor(level_obj,'close')
1300                # msg = 'student %s: level %s is not closed' % (student_id,
1301                #                                                      current_level)
1302
1303            study_course_obj.getContent().edit(mapping = d)
1304            level_obj.getContent().edit(mapping = d)
1305            if student_review_state != "returning":
1306                wftool.doActionFor(student_obj,'return')
1307            # try:
1308            #     wftool.doActionFor(level_obj,'close')
1309            # except:
1310            #     pass
1311            break
1312        return student_id,msg,mapping
1313    ###)
1314
1315
1316    def edit(self,mapping): ###(
1317        "edit student verdicts"
1318        wftool = self.portal_workflow
1319        logger = logging.getLogger('WAeUPImport.VerdictImport.edit')
1320        students_folder = self.portal_url.getPortalObject().campus.students
1321        student_id = mapping.get('id',None)
1322        matric_no = mapping.get('matric_no',None)
1323        editable_keys = mapping.keys()
1324        key = ''
1325        while True:
1326            result = self.findStudent('edit',student_id=student_id,matric_no=matric_no)
1327            #student_record,msg = self.getStudentRecord(mapping)
1328            msg = result['msg']
1329            if msg:
1330                break
1331            student_record = result['student_record']
1332            key_used = result['key_used']
1333            if key_used in editable_keys:
1334                editable_keys.remove(key_used)
1335            student_id = student_record.id
1336            mapping['id'] = student_id
1337            d = {}
1338            #import pdb;pdb.set_trace()
1339            any_change = False
1340            #special treatment for StudentStudyLevel
1341            current_session = d['session'] = mapping.get('current_session','')
1342            if current_session and student_record.session != current_session:
1343                msg = 'student %s: imported session %s does not match current_session %s' % (student_id,
1344                                                                                            current_session,
1345                                                                                            student_record.session)
1346                break
1347            current_level = mapping.get('current_level','')
1348            if not current_level.isdigit():
1349                msg = 'student %s: imported level is empty' % (student_id,)
1350                break
1351            if current_level and student_record.level != current_level:
1352                msg = 'student %s: imported level %s does not match current_level %s' % (student_id,
1353                                                                                        current_level,
1354                                                                                        student_record.level)
1355                break
1356            student_review_state =  student_record.review_state
1357            if student_review_state == 'deactivated':
1358                msg = "student %s in review_state %s" % (student_id, student_review_state)
1359                break
1360            if student_review_state not in ('courses_validated','returning'):
1361                msg = "student %s in wrong review_state %s" % (student_id, student_review_state)
1362                break
1363            student_obj = getattr(students_folder,student_id)
1364            # f2t = self.field2types_student
1365            study_course_obj = getattr(student_obj,'study_course',None)
1366            if study_course_obj is None:
1367                msg = 'student %s: no study_course object' % student_id
1368                break
1369            level_obj = getattr(study_course_obj,current_level,None)
1370            if  level_obj is None:
1371                msg = 'student %s: no study_level object for level %s' % (student_id,
1372                                                                                current_level)
1373                break
1374            verdict = d['verdict'] = d['current_verdict']  = mapping.get('current_verdict','')
1375
1376            #if verdict == student_record.verdict:
1377            #    msg = 'student %s: verdict already set to %s' % (student_id,
1378            #                                                            verdict)
1379
1380            level_review_state = wftool.getInfoFor(level_obj,'review_state',None)
1381            if level_review_state != "closed":
1382                wftool.doActionFor(level_obj,'close')
1383                # msg = 'student %s: level %s is not closed' % (student_id,
1384                #                                                      current_level)
1385
1386            study_course_obj.getContent().edit(mapping = d)
1387            level_obj.getContent().edit(mapping = d)
1388            if student_review_state != "returning":
1389                wftool.doActionFor(student_obj,'return')
1390            # try:
1391            #     wftool.doActionFor(level_obj,'close')
1392            # except:
1393            #     pass
1394            break
1395        return student_id,msg,mapping
1396    ###)
1397
1398   
1399   
1400class PaymentImport(WAeUPImport):###(
1401    """ PaymentImport """
1402    name = "payment"
1403    plural_name = "%ss" % name
1404    commit_after = 1000000
1405    required_modes = ('create',)
1406
1407    def create(self,mapping): ###(
1408        "create payment object"
1409        wftool = self.portal_workflow
1410        logger = logging.getLogger('WAeUPImport.PaymentImport.create')
1411        students_folder = self.portal_url.getPortalObject().campus.students
1412        student_id = mapping.get('id',None)
1413        order_id = key= mapping.get('order_id',None)
1414        payment_type = mapping.get('type',None)
1415        editable_keys = mapping.keys()
1416        key = ''
1417        msg = ''
1418        while True:
1419            result = self.findStudent('create',student_id=student_id)
1420            msg = result['msg']
1421            if msg:
1422                break           
1423            if payment_type == 'sc':
1424                hyphen = order_id.rfind('-')
1425                if not hyphen > -1:
1426                    msg = '%s: wrong order_id of sc payment' % student_id
1427                    break
1428                payment_object_id = 'p' + order_id[hyphen+1:]
1429            elif payment_type == 'online':
1430                payment_object_id = 'p' + order_id[6:]
1431            else:
1432                msg = '%s: no payment type defined' % student_id
1433                break
1434            student_record = result['student_record']
1435            student_id = student_record.id
1436
1437            mapping['id'] = student_id
1438            payments_folder = getattr(getattr(students_folder,student_id),'payments',None)
1439            if payments_folder is None:
1440                msg = '%s: no payments folder' % student_id
1441                break
1442            if getattr(payments_folder,payment_object_id,False):
1443                msg = '%s: payment object with id %s exists' % (student_id,payment_object_id)
1444                break
1445            try:
1446                #import pdb;pdb.set_trace()
1447                payments_folder.invokeFactory('Payment',"%s" % payment_object_id)
1448                payment_obj = getattr(payments_folder,"%s" % payment_object_id)
1449            except:
1450                msg = '%s: payment object %s cannot be created' % (student_id,payment_object_id)
1451                break
1452            payment_obj.getContent().edit(mapping = mapping)
1453            wftool.doActionFor(payment_obj,'close')
1454            break
1455        return key,msg,mapping
1456    ###)
1457   
1458   
1459###)
Note: See TracBrowser for help on using the repository browser.