source: WAeUP_SRP/trunk/WAeUPTool.py @ 3787

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