source: WAeUP_SRP/trunk/WAeUPTool.py @ 4249

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

resolve aaue #24

  • Property svn:keywords set to Id
File size: 77.2 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: WAeUPTool.py 4027 2009-03-25 16:53:32Z henrik $
20"""The WAeUP Tool Box.
21"""
22
23from AccessControl import ClassSecurityInfo
24from Acquisition import aq_inner
25from Acquisition import aq_parent
26from Globals import DTMLFile
27from Globals import InitializeClass
28from OFS.SimpleItem import SimpleItem
29from zExceptions import BadRequest
30
31from Products.CMFCore.utils import getToolByName
32from Products.CPSSchemas.DataStructure import DataStructure
33from Products.CPSSchemas.DataModel import DataModel
34from Products.CPSSchemas.StorageAdapter import MappingStorageAdapter
35from Products.CMFCore.ActionProviderBase import ActionProviderBase
36from Products.CMFCore.permissions import View
37from Products.ZCatalog.ZCatalog import ZCatalog
38from Products.CMFCore.permissions import ModifyPortalContent
39from Products.CMFCore.permissions import ManagePortal
40from Products.CMFCore.utils import UniqueObject
41from Products.CMFCore.URLTool import URLTool
42from Products.CMFCore.utils import getToolByName
43from Students import makeCertificateCode
44from Globals import package_home,INSTANCE_HOME
45from WAeUPImport import ApplicationImport,CertificateImport,CertificateCourseImport
46from WAeUPImport import CourseImport,CourseResultImport,StudentStudyLevelImport,PaymentImport
47from WAeUPImport import DepartmentImport,FacultyImport,StudentImport,VerdictImport
48from utils import makeDigest
49import DateTime,time
50import logging
51import transaction
52import csv,re,os,sys
53import md5
54from shutil import copy2,copy
55from Products.AdvancedQuery import Eq, Between, Le,In
56
57p_home = package_home(globals())
58i_home = INSTANCE_HOME
59images_base = os.path.join(i_home,"images")
60EMPTY = 'XXX'
61
62def getImagesDir(student_id):
63    return os.path.join("%s" % images_base,student_id[0],student_id)
64
65def getObject(object,name):
66    if object.hasObject(name):
67        return getattr(object,name)
68    return None
69
70class WAeUPTool(UniqueObject, SimpleItem, ActionProviderBase):
71    """WAeUP tool"""
72
73    id = 'waeup_tool'
74    meta_type = 'WAeUP Tool'
75    _actions = ()
76    security = ClassSecurityInfo()
77    security.declareObjectProtected(View)
78    manage_options = ( ActionProviderBase.manage_options
79                     + SimpleItem.manage_options
80                     )
81
82    security.declareProtected(View,'re_split') ###(
83    def re_split(self,split_string,string):
84        return re.split(split_string,string)
85    ###)
86
87    security.declareProtected(View,'difference') ###(
88    def difference(self,l1,l2):
89        return set(l1).difference(set(l2))
90    ###)
91
92    def rwrite(self,s): ###(
93        response = self.REQUEST.RESPONSE
94        response.setHeader('Content-type','text/html; charset=ISO-8859-15')
95        response.write("%s<br />\r\n" % s)
96    ###)
97
98    def addtodict(self,d,key,item): ###(
99        d[key].append(item)
100        return d[key]
101    ###)
102
103    def sleep(self,secs): ###(
104        "sleep"
105        import time
106        time.sleep(secs)
107        return
108    ###)
109
110    security.declareProtected(View,'updateRoleMappingsFor') ###(
111    def updateRoleMappingsFor(self,wf_definition,ob):
112        "do so for public"
113        wf_def = getattr(self.portal_workflow,wf_definition)
114        wf_def.updateRoleMappingsFor(ob)
115    ###)
116
117    security.declareProtected(View,'getStatesLgas') ###(
118    def getStatesLgas(self):
119        """return lga info"""
120        voc = getattr(self.portal_vocabularies,'local_gov_areas')
121        states = []
122        lgas  = []
123        d = {}
124        wd = {}
125        for k,v in voc.items():
126            parts = v.split(' / ')
127            if len(parts) == 1:
128                state = parts[0].lower()
129                lga = ""
130            elif len(parts) == 2:
131                state = "_".join(re.split('[^a-zA-Z0-9/]',parts[0].lower()))
132                lga = "-".join(re.split('[^a-zA-Z0-9/]',parts[1].lower()))
133            else:
134                continue
135            if state not in states:
136                states.append(state)
137            if lga not in lgas:
138                lgas.append(lga)
139            words = re.split('[^a-zA-Z0-9/]',k)
140            words.sort()
141            wd[k] = words
142            d[k] = v
143        mapping = {}
144        mapping['word_dict'] = wd
145        mapping['lga_dict'] = d
146        mapping['states'] = states
147        mapping['lgas'] = lgas
148        return mapping
149    ###)
150
151    security.declareProtected(View,'findLga') ###(
152    def findLga(self,words,words_dict):
153        words = re.split('[^a-zA-Z0-9/]',words)
154        lga_words = []
155        for word in words:
156            if word:
157                lga_words += word.strip().lower(),
158        lga_words.sort()
159        state_lga = ''
160        while not state_lga:
161            for k,l in words_dict.items():
162                if lga_words == l:
163                    state_lga = k
164                    break
165            break
166        return state_lga
167    ###)
168    security.declareProtected(View,'getAccessInfo') ###(
169    def getAccessInfo(self,context):
170        "return a dict with access_info"
171        logger = logging.getLogger('WAeUPTool.getAccessInfo')
172        mtool = self.portal_membership
173        member = mtool.getAuthenticatedMember()
174        member_id = str(member)
175        info = {}
176        is_anonymous = info['is_anonymous'] = mtool.isAnonymousUser()
177        is_student = info['is_student'] = ord(member_id[1]) > 48 and ord(member_id[1]) <= 57
178        is_staff = info['is_staff'] = not is_anonymous and not is_student
179        roles = member.getRolesInContext(context)
180        info['is_sectionofficer'] = not is_student and ("SectionOfficer" in roles or
181                                                        "SectionManager" in roles or
182                                                        "Manager" in roles)
183        info['is_clearanceofficer'] = not is_student and ("ClearanceOfficer" in roles)
184        #is_allowed = info['is_allowed'] = not is_anonymous
185        requested_id = context.getStudentId()
186        student_id  = None
187        if not is_anonymous and requested_id:
188            if (is_student and member_id == requested_id) or is_staff:
189                student_id = requested_id
190            else:   
191                logger.info('%s tried to access %s of %s' % (member_id,context.portal_type,requested_id))
192        info['student_id'] = student_id
193        return info
194    ###)
195
196    security.declareProtected(ModifyPortalContent,'openLog') ###(
197    def openLog(self,name):
198        """open a log file"""
199        version = 1
200        path = "%s/log/%s_%d.log" % (i_home,name,version)
201        while os.path.exists(path):
202            version += 1
203            path = "%s/log/%s_%d.log" % (i_home,name,version)
204        log = open(path,"w")
205        return log
206    ###)
207
208    security.declareProtected(ModifyPortalContent,'bypassQueueCatalog') ###(
209    def bypassQueueCatalog(self,enable=True):
210        """bypass the QueueCatalog by setting all indexes to process imediate,
211        if enable is True (default) the old settings are restored
212        """
213
214    ###)
215
216    security.declareProtected(ModifyPortalContent,'measureOaT') ###(
217    def measureOaT(self,method="a",probe="1000",nr_pts="1"):
218        """measure Object access Time"""
219        import random
220        if hasattr(self,'portal_catalog_real'):
221            aq_portal = self.portal_catalog_real.evalAdvancedQuery
222        else:
223            aq_portal = self.portal_catalog.evalAdvancedQuery
224        nr_pts = int(nr_pts)
225        probe = int(probe)
226        intervall = probe/10
227        objects = ("application","clearance","personal")
228        portal_types = ("StudentApplication","StudentClearance","StudentPersonal")
229        #i = random.randrange(num_objects)
230        count = 0
231        found = 0
232        not_found = 0
233        t_found = 0
234        t_not_found = 0
235        time_found = time_not_found = 0.0
236        t_time_found = t_time_not_found = 0.0
237        accessed = []
238        t_min = 1000
239        t_max = 0
240        #import pdb;pdb.set_trace()
241        students = self.portal_catalog(portal_type="Student")
242        num_students = len(students)
243        if method == "d":
244            query = Eq('path','/uniben/campus/students') & In('portal_type',portal_types[:nr_pts])
245            res = aq_portal(query)
246            brains = {}
247            for r in res:
248                sid = r.relative_path.split('/')[-2]
249                if brains.has_key(sid):
250                    brains[sid][r.portal_type] = r
251                else:
252                    brains[sid] = {r.portal_type : r}
253            brains_list = brains.keys()
254            num_objects = len(brains_list)
255        else:
256            num_objects = num_students
257        print "="*40
258        print "method: %s probes: %d nr_pts: %d num_objects: %d" % (method,
259                                                                        probe,
260                                                                        nr_pts,
261                                                                        num_objects)
262        print "nr found/not time found/not min/max"
263        elapse = time.time()
264        i_elapse = time.time()
265        c_elapse = time.clock()
266        for c in range(1,probe + 1):
267            i = random.randrange(num_objects)
268            if method in ('a','b','c'):
269                student_brain = students[i]
270            elif method == "d":
271                #import pdb;pdb.set_trace()
272                student_brain = brains[brains_list[i]]
273            if method == "c":
274                query = Eq('path',student_brain.getPath()) & In('portal_type',portal_types[:nr_pts])
275                res = aq_portal(query)
276                this_portal_types = [r.portal_type for r in res]
277            for i in range(nr_pts):
278                oid = objects[i]
279                if method == "a":
280                    try:
281                        student_path = student_brain.getPath()
282                        path = "%s/%s" % (student_path,oid)
283                        doc = self.unrestrictedTraverse(path).getContent()
284                        found += 1
285                        i_time = time.time() - i_elapse
286                        time_found += i_time
287                    except:
288                        not_found += 1
289                        i_time = time.time() - i_elapse
290                        time_not_found += i_time
291                        pass
292                elif method == "b":
293                    try:
294                        student_object = student_brain.getObject()
295                        doc = getattr(student_object,oid).getContent()
296                        found += 1
297                        i_time = time.time() - i_elapse
298                        time_found += i_time
299                    except:
300                        i_time = time.time() - i_elapse
301                        time_not_found += i_time
302                        not_found += 1
303                        pass
304                elif method == "c":
305                    if portal_types[i] in this_portal_types:
306                        found += 1
307                        doc = res[this_portal_types.index(portal_types[i])].getObject().getContent()
308                        i_time = time.time() - i_elapse
309                        time_found += i_time
310                    else:
311                        not_found += 1
312                        i_time = time.time() - i_elapse
313                        time_not_found += i_time
314                elif method == "d":
315                    if student_brain.has_key(portal_types[i]):
316                        found += 1
317                        doc = student_brain[portal_types[i]].getObject().getContent()
318                        i_time = time.time() - i_elapse
319                        time_found += i_time
320                    else:
321                        not_found += 1
322                        i_time = time.time() - i_elapse
323                        time_not_found += i_time
324                i_elapse = time.time()
325            if c and (c % intervall == 0):
326                #i_time = time.time() - i_elapse
327                t_per = 0.0
328                if found:
329                    t_per = time_found/found
330                if t_per > t_max:
331                    t_max = t_per
332                if t_per > 0.0 and t_per < t_min:
333                    t_min = t_per
334                itf = 0.0
335                if found:
336                    itf = time_found/found
337                itnf = 0.0
338                if not_found :
339                    itnf = time_not_found / not_found
340                interval_time = time_found + time_not_found
341                s = "%(c)d: %(found)d/%(not_found)d " % vars()
342                s += "%(interval_time)6.2f %(itf)6.4f/%(itnf)6.4f " % vars()
343                s += "%(t_min)6.4f/%(t_max)6.4f" %  vars()
344                print s
345                t_found += found
346                t_not_found += not_found
347                t_time_found += time_found
348                t_time_not_found += time_not_found
349                time_found = time_not_found = 0.0
350                found = not_found = 0
351        # t_found += found
352        # t_not_found += not_found
353        elapse = time.time() - elapse
354        itf = 0.0
355        if t_found:
356            itf = t_time_found/t_found
357        itnf = 0.0
358        if t_not_found:
359            itnf = t_time_not_found / t_not_found
360        #c_elapse = time.clock() - c_elapse
361        s = "%(probe)d: %(t_found)d/%(t_not_found)d " % vars()
362        s += "%(elapse)6.2f %(itf)6.4f/%(itnf)6.4f " % vars()
363        s += "%(t_min)6.4f/%(t_max)6.4f" %  vars()
364        print "-"*40
365        print s
366        rel_found = float(t_found)/probe
367        rel_not_found = float(t_not_found)/probe
368        estimated_total_time = num_objects*(rel_found*itf + rel_not_found*itnf)
369        print estimated_total_time
370    ###)
371
372    security.declareProtected(ModifyPortalContent,'writeLog') ###(
373    def writeLog(self,logfile,s):
374        """write to the log file"""
375        logfile.write(s)
376    ###)
377
378    def generateStudentId(self,letter): ###(
379        import random
380        logger = logging.getLogger('WAeUPTool.generateStudentId')
381        r = random
382        if letter == '?':
383            letter= r.choice('ABCDEFGHKLMNPQRSTUVWXY')
384        sid = "%c%d" % (letter,r.randint(99999,1000000))
385        students = self.portal_url.getPortalObject().campus.students
386        while self.students_catalog(id = sid) or self.waeup_tool.picturePathExists(sid):
387            if not self.students_catalog(id = sid):
388                logger.info('only picture path %s exists' % sid)
389            sid = "%c%d" % (letter,r.randint(99999,1000000))
390        return sid
391    ###)
392
393    def generatePassword(self,s=None): ###(
394        import random
395        r = random
396        ##if letter not in ('ABCDEFGIHKLMNOPQRSTUVWXY'):
397        if s is None:
398            s = 'abcdefghklmnpqrstuvwxy23456789'
399        pw = ''
400        while len(pw) < 6:
401            pw += r.choice(s)
402        return pw
403    ###)
404
405    security.declareProtected(ModifyPortalContent, 'dumpSchoolfeePayments') ###(
406    def dumpSchoolfeePayments(self):
407        "dump paid schoolfees"
408        mtool = self.portal_membership
409        member = mtool.getAuthenticatedMember()
410        logger = logging.getLogger('WAeUPTool.dumpSchoolfees')
411        aq_student = self.students_catalog.evalAdvancedQuery
412        query = In('review_state',('schoolfee_paid',
413                                   'courses_registered',
414                                   'courses_validated',
415                                   ))
416        res = aq_student(query)
417        #import pdb;pdb.set_trace()
418        l = []
419        logger.info("start for %d" % len(res))
420        count = 1
421        log_after = 100
422        for student in res:
423            if not count % log_after:
424                logger.info("processed %d total %d" % (log_after,count))
425            count += 1
426            fee_dict =self.getSchoolFee(student)
427            fulltime = student.mode.endswith('_ft')
428            d = {}
429            d['student_id'] = student.id
430            d['name'] = student.name
431            d['amount'] = fee_dict.get(new_returning)
432            l += d,
433        csv_name = self.dumpListToCSV(l,'payments')
434        logger.info('%s dumped payments to %s' % (member,export_file))
435    ###)
436
437    security.declarePublic('dumpListToCSV') ###(
438    def dumpListToCSV(self,l,filename,fields=None):
439        """dump a list of dicts to a CSV file"""
440        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
441        export_file = "%s/export/%s_%s.csv" % (i_home,filename,current,)
442        if fields is None:
443            fields = l[0].keys()
444        headline = ','.join(fields)
445        out = open(export_file,"wb")
446        out.write(headline +'\n')
447        out.close()
448        out = open(export_file,"a")
449        csv_writer = csv.DictWriter(out,fields,)
450        csv_writer.writerows(l)
451        return export_file
452    ###)
453
454    security.declareProtected(ManagePortal, 'listMembers') ###(
455    def listMembers(self):
456        "list all members"
457        mtool = self.portal_membership
458        member = mtool.getAuthenticatedMember()
459        logger = logging.getLogger('WAeUPTool.listMembers')
460        if str(member) not in ('admin','joachim'):
461            logger.info('%s tried to list members' % (member))
462            return None
463        members = self.portal_directories.members
464        all = members.listEntryIdsAndTitles()
465        l = []
466        for user_id,name in all:
467            d = {}
468            d['user_id'] = user_id
469            d['name'] = name
470            d['pw'] = getattr(getattr(members,user_id),'password')
471            d['email'] = getattr(getattr(members,user_id),'email')
472            d['groups'] = " ".join(getattr(getattr(members,user_id),'groups'))
473            d['roles'] = " ".join(getattr(getattr(members,user_id),'roles'))
474            l += d,
475        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
476        export_file = "%s/export/member_list_%s.csv" % (i_home,current,)
477        logger.info('%s dumped member list to %s' % (member,export_file))
478        fields = l[0].keys()
479        headline = ','.join(fields)
480        out = open(export_file,"wb")
481        out.write(headline +'\n')
482        out.close()
483        out = open(export_file,"a")
484        csv_writer = csv.DictWriter(out,fields,)
485        csv_writer.writerows(l)
486    ###)
487
488    security.declareProtected(ManagePortal, 'listStudents') ###(
489    def listStudents(self):
490        "list all students"
491        mtool = self.portal_membership
492        member = mtool.getAuthenticatedMember()
493        logger = logging.getLogger('WAeUPTool.listStudents')
494        if str(member) not in ('admin','joachim'):
495            logger.info('%s tried to list students' % (member))
496            return None
497        students = self.portal_directories.students
498        all = students.listEntryIdsAndTitles()
499        l = []
500        for user_id,name in all:
501            d = {}
502            d['user_id'] = user_id
503            d['name'] = name
504            d['pw'] = getattr(getattr(students,user_id),'password')
505            d['email'] = getattr(getattr(students,user_id),'email')
506            d['groups'] = " ".join(getattr(getattr(students,user_id),'groups'))
507            d['roles'] = " ".join(getattr(getattr(students,user_id),'roles'))
508            l += d,
509        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
510        export_file = "%s/export/student_list_%s.csv" % (i_home,current,)
511        logger.info('%s dumped student list to %s' % (member,export_file))
512        fields = l[0].keys()
513        headline = ','.join(fields)
514        out = open(export_file,"wb")
515        out.write(headline +'\n')
516        out.close()
517        out = open(export_file,"a")
518        csv_writer = csv.DictWriter(out,fields,)
519        csv_writer.writerows(l)
520    ###)
521
522    security.declareProtected(ManagePortal, 'removeDeletedDocIds') ###(
523    def removeDeletedDocIds(self, max=1000):
524        """
525        remove deleted docids from repository commit after max
526        """
527        logger = logging.getLogger('WAeUPTool.removeDeletedDocIds')
528        repository = getToolByName(self, 'portal_repository')
529        pxtool = getToolByName(self, 'portal_proxies')
530        logger.info('start')
531        pxtool_infos = pxtool.getRevisionsUsed()
532        logger.info('found  %d used revisions ' % (len(pxtool_infos)))
533
534        nb_revs = 0
535        docids_d = {} # all docids
536        unused_docids_d = {} # all docids that are unused
537        ids_unused_revs_docids = [] # ids for revs of unused docids
538        unused_ids = [] # ids for unused revs
539        total = 0
540        idlist = repository.objectIds()
541        to_delete = 0
542        found = False
543        for id in idlist:
544            docid, rev = repository._splitId(id)
545            if docid is None:
546                logger.info("invalid doc_id %s" % docid)
547                continue
548            nb_revs += 1
549            if not pxtool_infos.has_key(docid):
550                found = True
551                to_delete += 1
552                unused_ids.append(id)
553            elif not pxtool_infos[docid].has_key(rev):
554                found = True
555                to_delete += 1
556                unused_ids.append(id)
557            if found and not to_delete % max:
558                found = False
559                #import pdb;pdb.set_trace()
560                repository.manage_delObjects(unused_ids)
561                transaction.commit()
562                logger.info('removed %d total %d unused docids ' % (max,to_delete))
563        else:
564            if unused_ids:
565                repository.manage_delObjects(unused_ids)
566                transaction.commit()
567        logger.info('finished removing %d unused docids ' % (to_delete))
568    ###)
569
570    security.declarePublic('getCredential') ###(
571    def getCredential(self,student_id):
572        student_entry = getattr(self.portal_directories.students,student_id,None)
573        #import pdb;pdb.set_trace()
574        #if not self.isStaff():
575        #    mtool = self.portal_membership
576        #    member = mtool.getAuthenticatedMember()
577        #    logger = logging.getLogger('WAeUPTool.getCredential')
578        #    logger.info('%s tried to access password of %s' % (member,student_id))
579        #    return None
580        if student_entry is None:
581            return "not set"
582        return getattr(student_entry,"password","not set")
583    ###)
584
585    security.declarePublic('checkPassword') ###(
586    def checkPassword(self,student_id,password):
587        student_entry = getattr(self.portal_directories.students,student_id,None)
588        if student_entry is None:
589            return False
590        return getattr(student_entry,"password","not set") == password
591    ###)
592
593    security.declarePublic('checkGenericPassword') ###(
594    def checkGenericPassword(self,member_id):
595        member_entry = getattr(self.portal_directories.members,member_id,None)
596        if member_entry is None:
597            return False
598        ltool = getToolByName(self, 'portal_layouts')
599        unsecure_words = ltool._getOb('members')['w__password'].check_words
600        password = getattr(member_entry,"password","not set")
601        is_unsecure = password in unsecure_words
602        if is_unsecure:
603            logger = logging.getLogger('WAeUPTool.checkGenericPassword')
604            logger.info('Member %s tried to log in with unsecure password %s' %(member_id,password))
605        return is_unsecure
606    ###)
607
608    security.declarePublic('editPassword') ###(
609    def editPassword(self,student_id,password):
610        "edit a student password"
611        student_entry = getattr(self.portal_directories.students,student_id,None)
612        if student_entry is None:
613            return
614        setattr(student_entry,'password',password)
615    ###)
616
617    security.declareProtected(ModifyPortalContent,'doCommit') ###(
618    def doCommit(self,logger=None):
619        "commit some transactions"
620        transaction.commit()
621    ###)
622
623    security.declarePublic('loadStudentFoto') ###(
624    def loadStudentFoto(self,student,filename,folder):
625        "return a student passport picture"
626        #import pdb;pdb.set_trace()
627        picture ="%s/import/%s/%s" % (i_home,folder,filename)
628        student_id = student.getId()
629        images_dir = getImagesDir(student_id)
630        if not os.path.exists(images_dir):
631            os.mkdir(images_dir)
632        image_name = os.path.join(images_dir,"passport_%(student_id)s.jpg" % vars())
633        for extension in ('.jpg','.JPG'):
634            fullname = "%(picture)s%(extension)s" % vars()
635            if os.path.exists(fullname):
636                copy2(fullname,image_name)
637                return "successfully copied passport picture"
638        return "passport picture not found: %s.jpg or .JPG" % picture
639    ###)
640
641
642    security.declareProtected(ModifyPortalContent,'createOne') ###(
643    def createOne(self,students_folder,student_brain,letter,commit=False):
644        sid = self.waeup_tool.generateStudentId(letter)
645        students_folder.invokeFactory('Student', sid)
646        student = getattr(students_folder,sid)
647        self.portal_workflow.doActionFor(student,'return')
648        student.manage_setLocalRoles(sid, ['Owner',])
649        matric_no = student_brain.matric_no
650        jamb_reg_no = student_brain.Entryregno
651        self.students_catalog.addRecord(id = sid,
652                                           matric_no = matric_no,
653                                           jamb_reg_no = jamb_reg_no,
654                                           sex = student_brain.Sex == "F",
655                                           name = "%s %s %s" % (student_brain.Firstname,
656                                                                student_brain.Middlename,
657                                                                student_brain.Lastname)
658                                        )
659        if commit:
660            transaction.commit()
661        return sid,jamb_reg_no
662    ###)
663
664    # Functions without docstring cannot be called from outside (via the URL),
665    # thus we can declare addStudent public.
666    security.declarePublic('addStudent') ###(
667    def addStudent(self,dict):
668        students_folder = self.portal_url.getPortalObject().campus.students
669        sid = self.waeup_tool.generateStudentId('?')
670        students_folder.invokeFactory('Student', sid)
671        student_obj = getattr(students_folder,sid)
672        f2t = StudentImport.field2types_student
673        #from pdb import set_trace; set_trace()
674        d = {}
675        #d['jamb_sex']  = 'M'
676        #if dict.get('sex'):
677        #    d['jamb_sex']  = 'F'
678
679        entry_session = dict.get('entry_session')
680        if entry_session == self.getSessionId()[0]:
681            wfaction = 'admit'
682            wft = 'wf_transition_admit'
683            password = None
684        else:
685            wfaction = 'return'
686            wft = 'wf_transition_return'
687            password = self.generatePassword()
688            self.makeStudentMember(sid,password)
689
690        for pt in f2t.keys():
691            student_obj.invokeFactory(pt,f2t[pt]['id'])
692            sub_obj = getattr(student_obj,f2t[pt]['id'])
693            sub_doc = sub_obj.getContent()
694            #self.portal_workflow.doActionFor(sub_obj,'open',dest_container=sub_obj)
695            #d['Title'] = f2t[pt]['title']
696            for field in f2t[pt]['fields']:
697                d[field] = dict.get(field,'')
698            sub_doc.edit(mapping = d)
699            new_state = f2t[pt][wft]
700            if new_state != "remain":
701                self.portal_workflow.doActionFor(sub_obj,new_state,dest_container=sub_obj)
702        self.portal_workflow.doActionFor(student_obj,wfaction)
703        student_obj.manage_setLocalRoles(sid, ['Owner',])
704        return sid,password
705    ###)
706
707    security.declarePublic('getCertificateBrain') ###(
708    def getCertificateBrain(self,cert_id):
709        "do it"
710        res = ZCatalog.searchResults(self.portal_catalog_real,
711                                {'portal_type':"Certificate",
712                                      'id': cert_id})
713        if res:
714            return res[0]
715        return None
716    ###)
717
718    security.declareProtected(ModifyPortalContent,'get_csv_filenames') ###(
719    def get_csv_filenames(self):
720        "do it"
721        files = [file for file in os.listdir("%s/import/" % (i_home))
722                 if file.endswith('.csv') and (file.find('imported') == -1 and
723                                               file.find('pending') == -1)]
724        return files
725    ###)
726
727    security.declarePublic('findStudentByMatricelNo') ###(
728    def findStudentByMatricelNo(self,matric_no):
729        "do it"
730        res = ZCatalog.searchResults(self.portal_catalog_real,
731                                {'portal_type':"StudentClearance",
732                                 'SearchableText': matric_no})
733        if res:
734            return res[0]
735        return None
736    ###)
737
738
739    security.declarePublic('getOfficerName') ###(
740    def getOfficerName(self,mid):
741        """get the real name of a member"""
742        membership = self.portal_membership
743        member_record = membership.getMemberById(mid)
744        return member_record.getProperty('fullname',None)
745    ###)
746
747
748    security.declarePublic('makeStudentMember') ###(
749    def makeStudentMember(self,sid,password='uNsEt'):
750        """make the student a member"""
751        membership = self.portal_membership
752        membership.addMember(sid,
753                             password ,
754                             roles=('Member',
755                                     'Student',
756                                     ),
757                             domains='',
758                             properties = {'memberareaCreationFlag': False,
759                                           'homeless': True},)
760        member = membership.getMemberById(sid)
761        self.portal_registration.afterAdd(member, sid, password, None)
762        #self.manage_setLocalRoles(sid, ['Owner',])
763    ###)
764
765    security.declareProtected(View,'makeStudentData') ###(
766    def makeStudentData(self,student_id,email=None,phone_nr=None):
767        "create Datastructure for a returning Student"
768        #import pdb;pdb.set_trace()
769        logger = logging.getLogger('WAeUPTool.makeStudentData')
770        students_folder = self.portal_url.getPortalObject().campus.students
771        #res = self.students_catalog(id=student_id)
772        #if res:
773        #    st = res[0]
774        #res = self.returning_import(matric_no = st.matric_no)
775        res = self.returning_import(id = student_id)
776        if res:
777            student = res[0]
778        else:
779            logger.info('Id %s not found in returning_import' % student_id)
780            return
781        logger.info('%s creates data structure' % student_id)
782        s_results = self.results_import(matric_no = student.matric_no)
783        if s_results:
784            lnr = self.getLevelFromResultsCosCode(s_results)
785            level = "%d00" % lnr
786            verdict,eligible = self.getVerdict(s_results[0].Verdict)
787            #if eligible:
788            #    level = "%d00" % (lnr + 1)
789        else:
790            logger.info('matric_no %s not found in results_import' % student.matric_no)
791            level = ''
792            verdict = ''
793        #student should not be allowed to perform this transition
794        #wftool = self.portal_workflow
795        #wftool.doActionFor(student,'return')
796        certcode_org = student.Coursemajorcode
797        certcode = makeCertificateCode(certcode_org)
798        certificate_brain = self.getCertificateBrain(certcode)
799        if not certificate_brain:
800            em = 'Certificate %s org-code %s not found\n' % (certcode, certcode_org)
801            logger.info(em)
802        matric_no = student.matric_no
803        sid = student_id
804        student_obj = getattr(students_folder,sid)
805        if not getattr(student_obj,'application'):
806            student_obj.invokeFactory('StudentApplication','application')
807        application = student_obj.application
808        self.portal_workflow.doActionFor(application,'open',dest_container=application)
809        da = {'Title': 'Application Data'}
810        student_obj.invokeFactory('StudentPersonal','personal')
811        da['jamb_reg_no'] = student.Entryregno
812        em = self.getEntryMode(student.Entryregno)
813        da['entry_mode'] = em
814        personal = student_obj.personal
815        self.portal_workflow.doActionFor(personal,'open',dest_container=personal)
816        dp = {'Title': 'Personal Data'}
817        student_obj.invokeFactory('StudentClearance','clearance')
818        clearance = student_obj.clearance
819        self.portal_workflow.doActionFor(clearance,'open',dest_container=clearance)
820        dc = {'Title': 'Clearance/Eligibility Record'}
821        dc['matric_no'] = matric_no
822        state = student.State
823        lga = student.LGA
824        if state and lga:
825            lga =  state + ' / ' + lga
826        else:
827            lga = "None"
828        da['jamb_lga'] = dc['lga'] = lga
829        da['app_email'] = dp['email'] = email
830        da['app_mobile'] = dp['phone'] = phone_nr
831        dp['firstname'] = student.Firstname
832        dp['middlename'] = student.Middlename
833        dp['lastname'] = student.Lastname
834        da['jamb_lastname'] = "%s %s %s" % (student.Firstname,student.Middlename,student.Lastname)
835        da['jamb_sex'] = student.Sex
836        dp['sex'] = student.Sex == 'F'
837        dp['perm_address'] = student.Permanent_Address
838        application.getContent().edit(mapping=da)
839        self.portal_workflow.doActionFor(application,'close',dest_container=application)
840        personal.getContent().edit(mapping=dp)
841        clearance.getContent().edit(mapping=dc)
842        self.portal_workflow.doActionFor(clearance,'close',dest_container=clearance)
843        #
844        # Study Course
845        #
846        student_obj.invokeFactory('StudentStudyCourse','study_course')
847        studycourse = student_obj.study_course
848        self.portal_workflow.doActionFor(studycourse,'open',dest_container=studycourse)
849        dsc = {}
850        dsc['study_course'] = certcode
851        dsc['current_level'] = level
852        dsc['current_verdict'] = verdict
853        dsc['current_mode'] = em   #no longer used
854        dsc['current_session'] = '05'
855        studycourse.getContent().edit(mapping=dsc)
856        #
857        # Level
858        #
859        # l = getattr(studycourse,level,None)
860        # if l is None:
861        #     studycourse.invokeFactory('StudentStudyLevel', level)
862        #     l = getattr(studycourse, level)
863        #     self.portal_workflow.doActionFor(l,'open',dest_container=l)
864        #     l.getContent().edit(mapping={'Title': "Level %s" % level})
865        ###)
866
867    def init_timing(self): ###(
868        if self.with_timing:
869            if not hasattr(self,'_v_step_times'):
870                self._v_timer_count = 0
871                self._v_total = 0
872                self._v_step_times = {}
873                current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
874                self._v_timer_file = "%s/export/timing_%s.csv" % (i_home,current,)
875            self.timer_step = 0
876            self.total_time = 0
877            self.elapse = time.time()
878            self.i_elapse = time.time()
879            self.c_elapse = time.clock()
880    ###)
881
882    def do_timing(self): ###(
883        if self.with_timing:
884            try:
885                raise 'dummy'
886            except:
887                frame = sys.exc_traceback.tb_frame.f_back
888                locals = frame.f_locals
889                globals = frame.f_globals
890                functionname = frame.f_code.co_name
891                filename = os.path.basename(frame.f_code.co_filename)
892                lineno = frame.f_lineno
893                mod_line = "%(functionname)s:%(lineno)s" % vars()
894            i_time = time.time() - self.i_elapse
895            td = {}
896            if self._v_step_times.has_key(mod_line):
897                a_time = self._v_step_times[mod_line]['a_time'] + i_time
898                td['a_time'] = a_time
899            else:
900                td['a_time'] = i_time
901            td['i_time'] = i_time
902            self._v_step_times[mod_line] = td
903            self.i_time = i_time
904            self.total_time += i_time
905            self.timer_step +=1
906            self.i_elapse = time.time()
907    ###)
908
909    security.declareProtected(ModifyPortalContent,'print_timing') ###( ###(
910    def print_timing(self):
911        if self.with_timing:
912            l = []
913            timer_count = self._v_timer_count + 1
914            mod_lines = self._v_step_times.keys()
915            mod_lines.sort(cmp,reverse=0)
916            for mod_line in mod_lines:
917                td = self._v_step_times[mod_line]
918                i_time = td['i_time']
919                a_time = td['a_time']/(self._v_timer_count + 1)
920                l += ("%(mod_line)s,%(i_time)6.2f,%(a_time)6.2f,%(timer_count)d" % vars()),
921            total_time = self.total_time
922            total_avarage = self._v_total / timer_count
923            l += ("total,%(total_time)6.4f,%(total_avarage)6.4f,%(timer_count)d" % vars()),
924            print "\r\n".join(l)
925            out = open(self._v_timer_file,'a')
926            out.write("\r\n".join(l))
927            out.close()
928    ###)
929
930    security.declareProtected(ModifyPortalContent,'get_timing_data') ###( ###(
931    def get_timing_data(self):
932        if self.with_timing:
933            timer_count = self._v_timer_count + 1
934            results = {}
935            for k,d in self._v_step_times.items():
936                dd = {}
937                dd['a_time'] = d['a_time']/timer_count
938                dd['i_time'] = d['i_time']
939                dd['count'] = timer_count
940                results[k] = dd
941            dd = {}
942            dd['a_time'] = self._v_total / timer_count
943            dd['i_time'] = self.total_time
944            dd['count'] = timer_count
945            results["total"] = dd
946            return results
947    ###)
948
949    security.declareProtected(ModifyPortalContent,'admitOneStudent') ###(
950    def admitOneStudent(self,brain,entry_session,pin_password,with_timing=False):
951        "create Datastructure for an admitted Student"
952        #import pdb;pdb.set_trace()
953        logger = logging.getLogger('WAeUPTool.admitOneStudent')
954        self.with_timing = with_timing
955
956        if brain.screening_type in ('cest','sandwich',):
957            reg_no = "%s%s/%s" % (brain.course1[:3],brain.serial,brain.entry_session)
958        else:
959            reg_no = brain.reg_no
960
961        #ignore argument entry_session
962        if not brain.entry_session:
963            logger.info('no entry_session for %s provided' % (reg_no))
964            return
965        else:
966            es = brain.entry_session
967        if len(es) == 1:
968            es = '0%c' % es
969
970        if not hasattr(self,"_v_certificates"):
971            self._v_certificates = self.getCertificatesDict()
972        students_folder = self.portal_url.getPortalObject().campus.students
973
974        res = self.students_catalog(jamb_reg_no = reg_no)
975        if res:
976            logger.info('student with reg_no %s exists (%s)' % (reg_no,res[0].id))
977            return
978        if brain.status != "admitted":
979            logger.info('status of %s is %s' % (reg_no,brain.status))
980            return
981        pin_parts = brain.pin.split('-')
982        if pin_parts and len(pin_parts) != 3:
983            logger.info('invalid pin %s for %s' % (brain.pin,reg_no))
984            return
985        if not brain.course_admitted:
986            logger.info('no course_admitted provided for %s' % (reg_no))
987            return           
988        if brain.course_admitted not in self._v_certificates:
989            logger.info('certificate %s not found for %s' % (brain.course_admitted,reg_no))
990            return
991        if brain.sex not in (True,False):
992            logger.info('sex of %s not available' % (reg_no))
993            return
994        self.init_timing()
995        student_id = self.generateStudentId('?')
996        students_folder.invokeFactory('Student', student_id)
997        student_object = getattr(students_folder,student_id)
998        self.do_timing()
999        if pin_password:
1000            password = pin_parts[2]
1001            self.makeStudentMember(student_id,password = password)
1002        student_object.manage_setLocalRoles(student_id, ['Owner',])
1003        self.do_timing()
1004        #logger.info("creating %s reg_no %s" % (student_id,reg_no))
1005        #
1006        # application
1007        #
1008        student_object.invokeFactory('StudentApplication','application')
1009        application = student_object.application
1010        #self.portal_workflow.doActionFor(application,'open',dest_container=application)
1011        #self.do_timing()
1012        da = {'Title': 'Application Data'}
1013        da['jamb_reg_no'] = reg_no
1014
1015        sex = 'M'
1016        if brain.sex:
1017            sex = 'F'
1018        da['jamb_sex'] = sex
1019        da['jamb_age'] = brain.jamb_age
1020        da['app_reg_pin'] = brain.pin
1021        da['jamb_lga'] = brain.jamb_lga
1022        da['jamb_state'] = brain.jamb_state
1023        da['jamb_score'] = brain.aggregate
1024        da['app_email'] = brain.email
1025        da['app_mobile'] = brain.phone
1026
1027        # entry_mode cannot be retrieved from the certtificate
1028        # because ug_ft has two different entry modes
1029                       
1030        if brain.entry_mode:                      # does not happen because import_application
1031            da['entry_mode'] = brain.entry_mode   # schema has no field entry_mode
1032        elif brain.screening_type == 'pume': 
1033            da['entry_mode'] = 'ume_ft' 
1034        elif brain.screening_type == 'pde': 
1035            da['entry_mode'] = 'de_ft' 
1036        else:
1037            da['entry_mode'] = self._v_certificates[brain.course_admitted]['study_mode']
1038       
1039        #elif brain.screening_type == 'pce': 
1040        #   da['entry_mode'] = 'pce' 
1041        #elif brain.screening_type == 'prence': 
1042        #   da['entry_mode'] = 'prence' 
1043        #else: 
1044        #   da['entry_mode'] = '' 
1045     
1046
1047        #da['entry_session'] = entry_session
1048        da['entry_session'] = es
1049        da['jamb_lastname'] = brain.lastname
1050        da['jamb_middlename'] = brain.middlenames   # different field names!
1051        da['jamb_firstname'] = brain.firstname
1052        da['screening_application_date'] = brain.application_date
1053        da['date_of_birth'] = brain.date_of_birth
1054        da['jamb_first_cos'] = brain.course1
1055        da['jamb_second_cos'] = brain.course2
1056        da['course3'] = brain.course3
1057        da['screening_type'] = brain.screening_type
1058        da['screening_score'] = brain.screening_score
1059        da['screening_date'] = brain.screening_date
1060        da['hq_type'] = brain.hq_type
1061        da['hq_grade'] = brain.hq_grade
1062        da['aos'] = brain.aos
1063
1064        application.getContent().edit(mapping=da)
1065        self.do_timing()
1066        #self.portal_workflow.doActionFor(application,'close',dest_container=application)
1067        #
1068        # personal
1069        #
1070        student_object.invokeFactory('StudentPersonal','personal')
1071        personal = student_object.personal
1072        #self.portal_workflow.doActionFor(personal,'open',dest_container=personal)
1073        #self.do_timing()
1074        dp = {'Title': 'Personal Data'}
1075        dp['sex'] = brain.sex
1076        dp['email'] = brain.email
1077        dp['phone'] = brain.phone
1078        dp['lastname'] = brain.lastname
1079        dp['middlename'] = brain.middlenames   # different field names!
1080        dp['firstname'] = brain.firstname
1081        personal.getContent().edit(mapping=dp)
1082        self.do_timing()
1083        #
1084        # clearance
1085        #
1086        student_object.invokeFactory('StudentClearance','clearance')
1087        clearance = student_object.clearance
1088        #self.portal_workflow.doActionFor(clearance,'open',dest_container=clearance)
1089        dc = {'Title': 'Clearance/Eligibility Record'}
1090        dc['lga'] = brain.lga
1091        dc['birthday'] = brain.date_of_birth
1092        clearance.getContent().edit(mapping=dc)
1093        self.do_timing()
1094        #self.portal_workflow.doActionFor(clearance,'close',dest_container=clearance)
1095        #
1096        # study Course
1097        #
1098        student_object.invokeFactory('StudentStudyCourse','study_course')
1099        studycourse = student_object.study_course
1100        #self.portal_workflow.doActionFor(studycourse,'open',dest_container=studycourse)
1101        #self.do_timing()
1102        dsc = {}
1103        dsc['study_course'] = brain.course_admitted
1104        dsc['current_verdict'] = ''
1105        dsc['current_mode'] = da['entry_mode'] # no longer used
1106        if da['entry_mode'].startswith('de'):
1107            dsc['current_level'] = '200'
1108        elif da['entry_mode'].startswith('pre'):
1109            dsc['current_level'] = '000'
1110        else:
1111            dsc['current_level'] = '100'
1112        dsc['current_session'] = es
1113        studycourse.getContent().edit(mapping=dsc)
1114        self.do_timing()
1115        #
1116        # payments folder
1117        student_object.invokeFactory('PaymentsFolder','payments')
1118        payments = getattr(student_object,'payments')
1119        #self.do_timing()
1120        dpay = {}
1121        dpay['Title'] = 'Payments'
1122        payments.getContent().edit(mapping=dpay)
1123        self.portal_workflow.doActionFor(payments,'open')
1124        self.do_timing()
1125        #
1126        # passport foto
1127        app_picture ="%s/import/images/%s/%s_passport.jpg" % (i_home,
1128                                                              brain.screening_type,
1129                                                              brain.reg_no)
1130        images_dir = getImagesDir(student_id)
1131        #images_dir = os.path.join("%s" % images_base,student_id)
1132        letter_dir,student_dir = os.path.split(images_dir)
1133        if not os.path.exists(letter_dir):
1134            os.mkdir(letter_dir)
1135        if not os.path.exists(images_dir):
1136            os.mkdir(images_dir)
1137        image_name = os.path.join(images_dir,"passport_%(student_id)s.jpg" % vars())
1138        if os.path.exists(app_picture):
1139            copy2(app_picture,image_name)
1140        else:
1141            logger.info('passport of %s/%s not found: %s' % (student_id,
1142                                                             brain.reg_no,
1143                                                             app_picture))
1144
1145        self.do_timing()
1146        self.print_timing()
1147        if with_timing:
1148            self.timer_step = 0
1149            self._v_timer_count += 1
1150            self._v_total += self.total_time
1151        return student_id
1152    ###)
1153
1154    security.declareProtected(ModifyPortalContent,'makeStudentLevel') ###(
1155    def makeStudentLevel(self,student_id):
1156        "create the StudyLevel for a returning Student"
1157        #import pdb;pdb.set_trace()
1158        logger = logging.getLogger('WAeUPTool.makeStudentLevel')
1159        students_folder = self.portal_url.getPortalObject().campus.students
1160        res = self.students_catalog(id=student_id)
1161        if res:
1162            st = res[0]
1163        course = st.course
1164        matric_no = st.matric_no
1165        level = st.level
1166        res = self.results_import(matric_no = matric_no)
1167        if res:
1168            results = res
1169        logger.info('%s creating Level %s' % (student_id,level))
1170        #
1171        # Level
1172        #
1173        student_obj = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1174        studycourse = getattr(student_obj,"study_course",None)
1175        self.portal_workflow.doActionFor(studycourse,'close_for_edit',dest_container=studycourse)
1176        l = getattr(studycourse,level,None)
1177        if l is None:
1178            studycourse.invokeFactory('StudentStudyLevel', level)
1179            l = getattr(studycourse, level)
1180            self.portal_workflow.doActionFor(l,'open',dest_container=l)
1181            l.getContent().edit(mapping={'Title': "Level %s" % level})
1182        ###)
1183       
1184       
1185    security.declareProtected(ModifyPortalContent,'makeStudentLevel') ###(
1186    def exportAllStudyLevels(self,student_id):
1187        "export the StudyLevels for a student"
1188        #import pdb;pdb.set_trace()
1189        logger = logging.getLogger('WAeUPTool.exportAllStudyLevels')
1190        students_folder = self.portal_url.getPortalObject().campus.students
1191        res = self.students_catalog(id=student_id)
1192        if res:
1193            st = res[0]
1194        course = st.course
1195        matric_no = st.matric_no
1196        level = st.level
1197        student_obj = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1198        studycourse = getattr(student_obj,"study_course",None)
1199        levels = studycourse.objectIds()
1200       
1201        stool = getToolByName(self, 'portal_schemas')
1202        schema = stool._getOb('student_study_level')
1203        fields = ['student_id','level']
1204        fields.extend(schema.keys())
1205        format = '"%(' + ')s","%('.join(fields) + ')s"'
1206        export_file = "%s/export/study_levels_removed.csv" % (i_home)
1207       
1208        if not os.path.exists(export_file): 
1209            file_handler = open(export_file,"a")
1210            headline = ','.join(fields)
1211            file_handler.write(headline +'\n')
1212        else:
1213            file_handler = open(export_file,"a")
1214       
1215        for level in levels:
1216            level_object = getattr(studycourse,level)
1217            level_content = level_object.getContent()
1218            d = {'student_id':student_id,'level':level}
1219            for field in schema.keys():
1220                d[field] = getattr(level_content,field,'')   
1221            line = format % d
1222            file_handler.write(line +'\n')               
1223           
1224        ###)       
1225
1226    security.declarePublic('getHallInfo') ###(
1227    def getHallInfo(self,bed):
1228        """return Hall Info"""
1229        info = {}
1230        bedsplit = bed.split('_')
1231        if len(bedsplit) == 4:
1232            hall,block,room,letter = bed.split('_')
1233        else:
1234            info['maintenance_code'] = 'None'
1235            return info
1236        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="AccoHall",id=hall)
1237        if res and len(res) == 1:
1238            hall_brain = res[0]
1239            hall_doc = hall_brain.getObject().getContent()
1240        else:
1241            return info
1242        info['hall_title'] = hall_brain.Title
1243        info['maintenance_code'] = hall_doc.maintenance_code
1244        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="ScratchCardBatch")
1245        batch_doc = None
1246        for brain in res:
1247            if brain.id.startswith(info['maintenance_code']):
1248                batch_doc = brain.getObject().getContent()
1249                break
1250        if batch_doc is None:
1251            info['maintenance_fee'] = ''
1252        else:
1253            info['maintenance_fee'] = batch_doc.cost
1254        return info
1255    ###)
1256
1257    security.declareProtected(ModifyPortalContent,'removePictureFolder') ###(
1258    def removePictureFolder(self,student_id):
1259        """remove picture_folder by renaming it"""
1260        picture_path = getImagesDir(student_id)
1261        dest_path = os.path.join("%s" % images_base,'removed',student_id)
1262        dest_path = dest_path + "_removed"
1263        if os.path.exists(dest_path) or not os.path.exists(picture_path):
1264            return False
1265        os.rename(picture_path,dest_path)
1266        return True
1267    ###)
1268
1269    security.declareProtected(ModifyPortalContent,'restorePictureFolder') ###(
1270    def restorePictureFolder(self,student_id):
1271        """restore picture_folder by renaming it"""
1272        picture_path = getImagesDir(student_id)
1273        orig_path = os.path.join("%s" % images_base,'removed',student_id)
1274        orig_path = orig_path + "_removed"
1275        if os.path.exists(picture_path) or not os.path.exists(orig_path):
1276            return False       
1277        os.rename(picture_path + "_removed",picture_path)
1278        return True
1279    ###)
1280
1281    security.declarePublic('picturesExist') ###(
1282    def picturesExist(self, ids,student_id=None):
1283        """check if pictures exist in the filesystem"""
1284        if student_id is None:
1285            student_id = self.getStudentId()
1286        if student_id is None:
1287            return False
1288        picture_path = getImagesDir(student_id)
1289        #picture_path = os.path.join(images_base,student_id)
1290        if not os.path.exists(picture_path):
1291            return False
1292        pictures  = [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1293        return set(ids).issubset(set(pictures))
1294    ###)
1295   
1296        security.declarePublic('picturePathExists') ###(
1297    def picturePathExists(self, student_id=None):
1298        """check if picture path exists in the filesystem"""
1299        if student_id is None:
1300            student_id = self.getStudentId()
1301        if student_id is None:
1302            return False
1303        picture_path = getImagesDir(student_id)
1304        if os.path.exists(picture_path):
1305            return True
1306        return False
1307    ###)
1308   
1309   
1310    security.declareProtected(ModifyPortalContent,'removeUnusedImageFolders') ###(
1311    def removeUnusedImageFolders(self):
1312        """check if an unused image folders exists in the filesystem"""
1313        mtool = self.portal_membership
1314        member = mtool.getAuthenticatedMember()
1315        member_id = str(member)
1316        logger = logging.getLogger('WAeUPTool.removeUnusedImageFolders')
1317        abc = os.listdir(images_base)
1318        ifolders = []
1319        for i in abc:
1320            picture_path = os.path.join(images_base,i)
1321            ifolders.extend(os.listdir(picture_path))
1322        unused_ids = []
1323        for id in ifolders:
1324            res = self.students_catalog(id=id)
1325            if not res:
1326                unused_ids.append(id)
1327                #import pdb;pdb.set_trace()   
1328                if not id.endswith('removed'):
1329                    removed = self.waeup_tool.removePictureFolder(id) 
1330                    if removed:
1331                        logger.info('%s: image folder %s successfully removed' % (member_id,id))
1332                    else:
1333                        logger.info('%s: image folder %s could not be removed' % (member_id,id))
1334        return
1335       
1336    ###)   
1337
1338    security.declarePublic('picturesList') ###(
1339    def picturesList(self):
1340        """check if pictures exist in the filesystem"""
1341        path = 'images'
1342        student_id = self.getStudentId()
1343        #picture_path = os.path.join(i_home,path,student_id)
1344        picture_path = getImagesDir(student_id)
1345        if not os.path.exists(picture_path):
1346            return []
1347        return [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1348    ###)
1349
1350    security.declarePublic('showFsPicture') ###(
1351    def showFsPicture(self,path):
1352        """return a picture from the filesystem"""
1353        #picture_path = os.path.join(i_home,path)
1354        picture_path = os.path.join(images_base,path)
1355        response = self.REQUEST.RESPONSE
1356        #import pdb;pdb.set_trace()
1357        registry = getToolByName(self, 'mimetypes_registry')
1358        mimetype = str(registry.lookupExtension(path.lower()) or
1359                    registry.lookupExtension('file.bin'))
1360        if os.path.exists(picture_path):
1361            response.setHeader('Content-type',mimetype)
1362            return open(picture_path).read()
1363        picture_path = os.path.join(i_home,'import',path)
1364        if os.path.exists(picture_path):
1365            return open(picture_path).read()
1366    ###)
1367
1368    security.declareProtected(ModifyPortalContent,'deleteAllCourses') ###(
1369    def deleteAllCourses(self,department="All"):
1370        ''' delete the courses'''
1371        pm = self.portal_membership
1372        member = pm.getAuthenticatedMember()
1373
1374        if str(member) not in ("henrik","joachim"):
1375            return "not possible"
1376        if department == "All":
1377            res = self.portal_catalog({'meta_type': 'Department'})
1378        if len(res) < 1:
1379            return "No Departments found"
1380
1381        deleted = []
1382        for dep in res:
1383            cf = dep.getObject().courses
1384            if cf:
1385                cf.manage_delObjects(ids=cf.objectIds())
1386                deleted.append("deleted Courses in %s" % dep.getId)
1387        return "\r".join(deleted)
1388    ###)
1389
1390    security.declareProtected(ModifyPortalContent,'getLogfileLines') ###(
1391    def getLogfileLines(self,filename="event.log",numlines=20):
1392        """Get last NUMLINES lines of logfile FILENAME.
1393
1394        Return last lines' of a file in the instances logfile directory as
1395        a list. The number of returned lines equals `numlines' or less. If
1396        less than `numlines' lines are available, the whole file ist
1397        returned. If the file can not be opened or some other error
1398        occurs, empty list is returend.
1399        """
1400        result = []
1401        lines_hit = 0
1402
1403        # We only handle files in instances' log directory...
1404        logpath = os.path.join(i_home, "log")
1405        filename = str(os.path.abspath( os.path.join( logpath, filename )))
1406        if not filename.startswith( logpath ):
1407            # Attempt to access file outside log-dir...
1408            return []
1409
1410        try:
1411            fd = file( filename, "rb" )
1412        except IOError:
1413            return []
1414        if not fd:
1415            return []
1416
1417        if os.linesep == None:
1418            linesep = '\n'
1419        else:
1420            linesep = os.linesep
1421
1422        # Try to find 'numlines' times a lineseparator, searching from end
1423        # and moving to the beginning of file...
1424        fd.seek( 0, 2) # Move to end of file...
1425        while lines_hit < numlines:
1426            if fd.read(1) == linesep[-1]: # This moves filedescriptor
1427                                          # one step forward...
1428                lines_hit += 1
1429            try:
1430                fd.seek( -2, 1) # Go two bytes back from current pos...
1431            except IOError:
1432                # We cannot go back two bytes. Maybe the file is too small...
1433                break
1434        fd.seek(2,1)
1435
1436        # Read all lines from current position...
1437        result = fd.readlines()
1438        # Remove line endings...
1439        result = [x.strip() for x in result]
1440        fd.close()
1441        return result
1442    ###)
1443
1444    security.declareProtected(ModifyPortalContent,"getCallbacksFromLog")###(
1445    def getCallbacksFromLog(self,filename):
1446        """fix Online Payment Transactions from Z2.log entries"""
1447        import transaction
1448        import random
1449        from cgi import parse_qs
1450        from urlparse import urlparse
1451        #from pdb import set_trace
1452        wftool = self.portal_workflow
1453        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
1454        students_folder = self.portal_url.getPortalObject().campus.students
1455        s = r'(?P<client_ip>\S+) - (?P<member_id>\S+) \['
1456        s += r'(?P<date>.*)\] "(?P<get>.*)" (?P<codes>\d+ \d+) "'
1457        s += r'(?P<intersw>.*)" "(?P<agent>.*)"'
1458        data = re.compile(s)
1459        start = True
1460        tr_count = 1
1461        total = 0
1462        #name = 'pume_results'
1463        #name = 'epaymentsuccessful_z2log2'
1464        name = filename
1465        no_import = []
1466        imported = []
1467        logger = logging.getLogger('WAeUPTool.getFailedTransactions')
1468        try:
1469            transactions = open("%s/import/%s" % (i_home,name),"rb").readlines()
1470        except:
1471            logger.error('Error reading %s' % name)
1472            return
1473        tas = []
1474        for line in transactions:
1475            dict = {}
1476            items = data.search(line)
1477            dict['idict'] = idict = items.groupdict()
1478            #print idict
1479            #from pdb import set_trace;set_trace()
1480            urlparsed = urlparse(idict['get'][4:])
1481            #print urlparsed
1482            path = urlparsed[2].split('/')
1483            dict['student_id'] = student_id = path[8]
1484            dict['payment_id'] = payment_id = path[10]
1485            dict['qs_dict'] = qs_dict = parse_qs(urlparsed[4])
1486            tas.append(dict)
1487            tr_count += 1
1488        return tas
1489    ###)
1490
1491    security.declareProtected(ModifyPortalContent,"importOnlinePaymentTransactions")###(
1492    def importOnlinePaymentTransactions(self):
1493        """load Online Payment Transactions from CSV values"""
1494        import transaction
1495        import random
1496        #from pdb import set_trace
1497        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
1498        opt = self.online_payments_import
1499        students_folder = self.portal_url.getPortalObject().campus.students
1500        start = True
1501        tr_count = 1
1502        total = 0
1503        #name = 'pume_results'
1504        name = 'OnlineTransactions'
1505        no_import = []
1506        imported = []
1507        logger = logging.getLogger('WAeUPTool.importOnlinePaymentTransactions')
1508        try:
1509            transactions = csv.DictReader(open("%s/import/%s.csv" % (i_home,name),"rb"))
1510        except:
1511            logger.error('Error reading %s.csv' % name)
1512            return
1513        for pay_transaction in transactions:
1514            if start:
1515                start = False
1516                logger.info('Start loading from %s.csv' % name)
1517                s = ','.join(['"%s"' % fn for fn in pay_transaction.keys()])
1518                no_import.append('%s,"Error"' % s)
1519                format = ','.join(['"%%(%s)s"' % fn for fn in pay_transaction.keys()])
1520                format_error = format + ',"%(Error)s"'
1521            data = {}
1522
1523            # format of the first file sent by Tayo
1524            #data['datetime'] = date = DateTime.DateTime(pay_transaction['Date'])
1525            #data['student_id'] = student_id = pay_transaction['Payer ID']
1526            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1527            #data['response_code'] = response_code = pay_transaction['Resp Code']
1528            #data['amount'] = amount = pay_transaction['Amount']
1529
1530            # format of the second file sent by Tayo
1531            #data['datetime'] = date = 0
1532            #data['student_id'] = student_id = pay_transaction['Payer ID']
1533            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1534            #data['response_code'] = response_code = '00'
1535            #data['amount'] = amount = pay_transaction['Amount']
1536
1537            # format of the third file sent by Kehinde
1538            data['datetime'] = date = 0
1539            data['student_id'] = student_id = pay_transaction['customer_id']
1540            data['order_id'] = order_id = pay_transaction['merchant_reference']
1541            data['response_code'] = response_code = '00'
1542            data['amount'] = amount = pay_transaction['Amount']
1543
1544            dup = False
1545            if response_code == "12":
1546                continue
1547            try:
1548                opt.addRecord(**data)
1549            except ValueError:
1550                dup = True
1551            #from pdb import set_trace;set_trace()
1552            if dup:
1553                if response_code == "00":
1554                    try:
1555                        opt.modifyRecord(**data)
1556                    except:
1557                        logger.info("duplicate uid, order_id %(order_id)s, student_id %(student_id)s, response_code %(response_code)s" % data)
1558                        continue
1559                else:
1560                    pay_transaction['Error'] = "Duplicate order_id"
1561                    no_import.append( format_error % pay_transaction)
1562                    logger.info("duplicate order_id %(order_id)s for %(student_id)s %(response_code)s" % data)
1563                    continue
1564            tr_count += 1
1565            if tr_count > 1000:
1566                if len(no_import) > 0:
1567                    open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1568                             '\n'.join(no_import) + '\n')
1569                    no_import = []
1570                em = '%d transactions committed\n' % (tr_count)
1571                transaction.commit()
1572                regs = []
1573                logger.info(em)
1574                total += tr_count
1575                tr_count = 0
1576        open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1577                                                '\n'.join(no_import))
1578        return self.REQUEST.RESPONSE.redirect("%s" % self.REQUEST.get('URL1'))
1579    ###)
1580
1581    security.declareProtected(ModifyPortalContent,"importData")###(
1582    def importData(self,filename,name,edit=False,bypass_queue_catalog=False):
1583        """load data from CSV values"""
1584        import transaction
1585        import random
1586        students_folder = self.portal_url.getPortalObject().campus.students
1587        uploads_folder = self.portal_url.getPortalObject().campus.uploads
1588        pending_only = False
1589        pend_str = '--'
1590        elapse = time.time()
1591        #
1592        # preparations
1593        #
1594        if filename == pend_str:
1595            pending_only = True
1596        importer_name = ''.join([part.capitalize() for part in name.split('_')])
1597        importer = eval("%sImport" % importer_name)(self)
1598        logger = importer.logger
1599        if importer.init_errors:
1600            logger.info(importer.init_errors)
1601            return importer.init_errors
1602        member = importer.member
1603        #current = importer.current
1604        import_date = importer.import_date
1605        #
1606        # not_imported
1607        #
1608        info = importer.info
1609        data_keys = importer.data_keys
1610        csv_keys = importer.csv_keys
1611        #csv_keys.extend(info.keys())
1612        headline_mapping = dict((k,k) for k in csv_keys)
1613        #
1614        # pending
1615        #
1616        pending_path = importer.pending_path
1617        pending_tmp = importer.pending_tmp
1618        pending_backup = importer.pending_backup
1619        pending_fn = importer.pending_fn
1620        imported_path = importer.imported_path
1621        imported_fn = importer.imported_fn
1622        commit_after = importer.commit_after
1623        pending = []
1624        pending_digests = []
1625        #total_added_to_pending = 0
1626        if not pending_only:
1627            pending,pending_digests = importer.makeIdLists()
1628            pending_at_start = len(pending)
1629        datafile = open(pending_tmp,"w")
1630        pending_csv_writer = csv.DictWriter(datafile,
1631                                                    csv_keys,
1632                                                    extrasaction='ignore')
1633        pending_csv_writer.writerow(headline_mapping)
1634        datafile.close()
1635        #
1636        # imported
1637        #
1638        if not os.path.exists(imported_path):
1639            datafile = open(imported_path,"w")
1640            imported_csv_writer = csv.DictWriter(datafile,
1641                                                 csv_keys,
1642                                                 extrasaction='ignore')
1643            imported_csv_writer.writerow(headline_mapping)
1644            datafile.close()
1645        start = True
1646        tr_count = 0
1647        total = 0
1648        total_added_to_imported = 0
1649        total_pending = 0
1650        import_source_done = ""
1651        if pending_only:
1652            import_source_path = pending_path
1653        else:
1654            import_source_path = "%s/import/%s.csv" % (i_home,filename)
1655            import_source_done = "%s/import/%s.done" % (i_home,filename)
1656        if not os.path.exists(import_source_path):
1657            fn = os.path.split(import_source_path)[1]
1658            em = 'no import file %(fn)s' % vars()
1659            return em
1660        import_source_fn = os.path.split(import_source_path)[1]
1661        if not pending_only:
1662            info['imported_from'] = import_source_fn
1663        headline = csv.reader(open(import_source_path,"rb")).next()
1664        if "import_mode" not in headline:
1665            msg = 'import_mode must be in heading'
1666            return msg
1667        invalid_keys = importer.checkHeadline(headline)
1668        if invalid_keys:
1669            return 'not ignorable key(s): "%s" found in heading' % ", ".join(invalid_keys)
1670
1671        import_keys = [k.strip() for k in headline if not (k.strip().startswith('ignore')
1672                                                         or k.strip() in info.keys())]
1673        # diff2schema = set(import_keys).difference(set(importer.schema.keys()))
1674        # diff2layout = set(import_keys).difference(set(importer.layout.keys()))
1675        # if diff2schema and diff2schema != set(['id',]):
1676        #     msg = 'not ignorable key(s): "%s" found in heading' % ", ".join(diff2schema)
1677        #     return msg
1678        #
1679        # start importing
1680        #
1681        try:
1682            reader = csv.DictReader(open(import_source_path,"rb"))
1683        except:
1684            msg = 'Error reading %s.csv' % filename
1685            logger.error(msg)
1686            return msg
1687        items = [item for item in reader]
1688        total_to_import = len(items)
1689        tti_float = float(total_to_import)
1690        if pending_only:
1691            pending_at_start = total_to_import
1692        count = 0
1693        imported = []
1694        old_commit_count = 0
1695        error_count = imported_count = 0
1696        already_in = 0
1697        for record in items:
1698            item = {}
1699            empty_value_keys = []
1700            for k,v in record.items():
1701                if k is None:
1702                    continue
1703                if v:
1704                    if v == EMPTY:
1705                        empty_value_keys += k,
1706                        v = ''
1707                    item[k.strip()] = v.strip()
1708            count += 1
1709            if start:
1710                start = False
1711                adapters = [MappingStorageAdapter(importer.schema, item)]
1712                logger.info('%(member)s starts import from %(import_source_fn)s' % vars())
1713            dm = DataModel(item, adapters,context=self)
1714            ds = DataStructure(data=item,datamodel=dm)
1715            error_string = ""
1716            total += 1
1717            import_mode = item.get('import_mode','')
1718            import_method = getattr(importer, '%(import_mode)s' % vars(),None )
1719            if import_method is None:
1720                error_string += "import_mode '%(import_mode)s' is invalid" % vars()
1721            elif (import_mode in importer.required_modes and
1722                not set(importer.required_keys[import_mode]).issubset(set(item.keys()))):
1723                diff2import = set(importer.required_keys[import_mode]).difference(set(item.keys()))
1724                error_string += 'required key(s): "%s" not found in record' % ", ".join(diff2import)
1725            else:
1726                for k in import_keys:
1727                    if k not in item.keys() or k not in importer.validators.keys():
1728                        continue
1729                    if not importer.validators[k](ds,mode=import_mode):
1730                        error_string += ' ++ '
1731                        error_string += "%s: %s" % (k,self.translation_service(ds.getError(k),
1732                                                                           ds.getErrorMapping(k)))
1733            if error_string:
1734                error = error_string
1735                id = ''
1736                mapping = item
1737            else:
1738                temp_item = item.copy()
1739                temp_item.update(dm)
1740                results = import_method(temp_item)
1741                id = results[0]
1742                error = results[1]
1743                mapping = results[2]
1744            #
1745            # restore some values
1746            #
1747            for k in empty_value_keys:
1748                mapping[k] = EMPTY
1749            if mapping.has_key('sex'):
1750                #import pdb;pdb.set_trace()
1751                if mapping['sex'] == True:
1752                    mapping['sex'] = 'F'
1753                elif mapping['sex'] == False:
1754                    mapping['sex'] = 'M'
1755            if pending_only:
1756                info['imported_from'] = item['imported_from']
1757            data_string = ", ".join("%s: %s" % (k,v) for k,v in mapping.items())
1758            info['error'] = error
1759            info['import_record_no'] = count + 1
1760            mapping.update(info)
1761            log_list = []
1762            if error:
1763                error_count += 1
1764                digest = makeDigest(mapping,data_keys)
1765                if digest not in pending_digests:
1766                    pending_digests += digest,
1767                    pending.append(mapping)
1768                    if not pending_only:
1769                        log_list += "record from %(import_source_fn)s added to %(pending_fn)s, %(data_string)s, %(error)s" % vars(),
1770                else:
1771                    already_in += 1
1772                    pass
1773            else:
1774                imported_count += 1
1775                imported += mapping,
1776                log_list += "record imported and added to %(imported_fn)s from %(import_source_fn)s, %(data_string)s" % vars(),
1777            if log_list:
1778                time_till_now = time.time() - elapse
1779                percent_finished = (error_count + imported_count)/tti_float*100
1780                log_list.insert(0,("%(percent_finished)6.3f %% done in %(time_till_now)3.2fs," % vars()),)
1781                logger.info(' '.join(log_list))
1782            finished = count > total_to_import - 1
1783            must_commit = (imported_count != old_commit_count) and (not imported_count % commit_after)
1784            if must_commit:
1785                old_commit_count = imported_count
1786
1787            if must_commit or finished:
1788                if len(imported):
1789                    transaction.commit()
1790                    datafile = open(imported_path,"a")
1791                    writer = csv.DictWriter(datafile,
1792                                            csv_keys,
1793                                            extrasaction='ignore')
1794                    writer.writerows(imported)
1795                    datafile.close()
1796                    total_added_to_imported += len(imported)
1797                    imported = []
1798                if len(pending) > 0:
1799                    datafile = open(pending_tmp,"a")
1800                    writer = csv.DictWriter(datafile,
1801                                            csv_keys,
1802                                            extrasaction='ignore')
1803                    writer.writerows(pending)
1804                    datafile.close()
1805                    total_pending += len(pending)
1806                    #total_added_to_pending += len(pending)
1807                    pending = []
1808                if not finished:
1809                    msg = '%(commit_after)d records imported and committed of total %(total_added_to_imported)d\n' % vars()
1810                    logger.info(msg)
1811        elapse = time.time() - elapse
1812        if os.path.exists(pending_path):
1813            copy2(pending_path,pending_backup)
1814        copy2(pending_tmp,pending_path)
1815        msg = "finished importing from %(import_source_fn)s in %(elapse).2f seconds, " % vars()
1816        msg += "%(count)d records totally read, %(total_added_to_imported)d added to %(imported_fn)s, " % vars()
1817        if pending_only:
1818            removed_pending = pending_at_start - total_pending
1819            msg += "%(removed_pending)d removed from %(pending_fn)s" % vars()
1820        else:
1821            added_pending = total_pending - pending_at_start
1822            msg += "%(added_pending)d added to %(pending_fn)s, %(already_in)s already in %(pending_fn)s" % vars()
1823        #msg += "%(total_pending)d totally written" % vars()    # this line does not make any sense
1824        logger.info(msg)
1825        if import_source_done:
1826            copy(import_source_path,import_source_done)
1827            os.remove(import_source_path)
1828            upload = getattr(uploads_folder,os.path.split(import_source_path)[1],None)
1829            if upload is not None:
1830                upload_doc = upload.getContent()
1831                mapping = {}
1832                #mapping['import_date'] = DateTime.DateTime()
1833                mapping['import_date'] = import_date
1834                mapping['imported_by'] = importer.imported_by
1835                mapping['import_message'] = msg
1836                upload_doc.edit(mapping = mapping)
1837        os.remove(pending_tmp)
1838        return msg
1839    ###)
1840
1841    security.declareProtected(ModifyPortalContent,"moveImagesToFS")###(
1842    def moveImagesToFS(self,student_id="O738726"):
1843        "move the images to the filesystem"
1844        images_dir = getImagesDir(student_id)
1845        #images_dir = os.path.join("%s" % images_base,student_id)
1846        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1847        stool = getToolByName(self, 'portal_schemas')
1848        schemas = ['student_application',
1849                   'student_clearance',
1850                   ]
1851        created = False
1852        for schema_id in schemas:
1853            schema = stool._getOb(schema_id)
1854            object = getattr(student_folder,schema_id[len('student_'):],None)
1855            if object is None:
1856                continue
1857            doc = object.getContent()
1858            for key in schema.keys():
1859                if schema[key].meta_type != "CPS Image Field":
1860                    continue
1861                #import pdb;pdb.set_trace()
1862                image = getattr(doc,key,None)
1863                if not image or not hasattr(image,"data"):
1864                    continue
1865                if not created:
1866                    if not os.path.exists(images_dir):
1867                        os.mkdir(images_dir)
1868                    created = True
1869                filename = os.path.join(images_dir,"%(key)s_%(student_id)s.jpg" % vars())
1870                open(filename,"wb").write(str(image.data))
1871    ###)
1872
1873    security.declareProtected(ModifyPortalContent,"movePassportToFS")###(
1874    def movePassportToFS(self,student_id="O738726"):
1875        "move the passports to the filesystem"
1876        images_dir = os.path.join("%s" % i_home,'passports')
1877        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1878        stool = getToolByName(self, 'portal_schemas')
1879        schemas = ['student_application',
1880                   #'student_clearance',
1881                   ]
1882        created = False
1883        for schema_id in schemas:
1884            schema = stool._getOb(schema_id)
1885            object = getattr(student_folder,schema_id[len('student_'):],None)
1886            if object is None:
1887                continue
1888            doc = object.getContent()
1889            for key in schema.keys():
1890                if schema[key].meta_type != "CPS Image Field":
1891                    continue
1892                #import pdb;pdb.set_trace()
1893                image = getattr(doc,key)
1894                if not hasattr(image,"data"):
1895                    continue
1896                if not created:
1897                    if not os.path.exists(images_dir):
1898                        os.mkdir(images_dir)
1899                    created = True
1900                filename = os.path.join(images_dir,"%(student_id)s.jpg" % vars())
1901                open(filename,"wb").write(str(image.data))
1902    ###)
1903   
1904   
1905    security.declarePublic('getConfigParams')
1906    def getConfigParams(self,conf_id="configuration"):
1907        conf = getattr(self.portal_url.getPortalObject().campus,conf_id)
1908        conf_obj = conf.getContent()
1909        stool = getToolByName(self, 'portal_schemas')
1910        schema = stool._getOb('configuration')
1911        d = {}
1912        for key in schema.keys():
1913            d[key] = getattr(conf_obj,key,None)
1914        return d
1915
1916
1917InitializeClass(WAeUPTool)
Note: See TracBrowser for help on using the repository browser.