source: WAeUP_SRP/base/WAeUPTool.py @ 3590

Last change on this file since 3590 was 3492, checked in by joachim, 16 years ago

fix traceback for anonymous users,
do not show log-message, when student views academic session.

  • Property svn:keywords set to Id
File size: 72.8 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 3492 2008-05-06 11:36:36Z joachim $
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
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        if not hasattr(self,"_v_certificates"):
973            self._v_certificates = self.getCertificatesDict()
974        students_folder = self.portal_url.getPortalObject().campus.students
975        if brain.screening_type in ('cest','sandwich',):
976            reg_no = "%s%s/07" % (brain.course1[:3],brain.serial)
977        else:
978            reg_no = brain.reg_no
979        res = self.students_catalog(jamb_reg_no = reg_no)
980        if res:
981            logger.info('student with this reg_no already exists %s with id %s' % (reg_no,res[0].id))
982            return
983        if brain.status != "admitted":
984            logger.info('status of %s is %s' % (reg_no,brain.status))
985            return
986        pin_parts = brain.pin.split('-')
987        if pin_parts and len(pin_parts) != 3:
988            logger.info('invalid pin %s for %s' % (brain.pin,reg_no))
989            return
990        if brain.course_admitted not in self._v_certificates:
991            logger.info('certificate %s not found for %s' % (brain.course_admitted,reg_no))
992            return
993        if brain.sex not in (True,False):
994            logger.info('sex of %s not available' % (reg_no))
995            return
996        self.init_timing()
997        student_id = self.generateStudentId('?')
998        students_folder.invokeFactory('Student', student_id)
999        student_object = getattr(students_folder,student_id)
1000        self.do_timing()
1001        if pin_password:
1002            password = pin_parts[2]
1003            self.makeStudentMember(student_id,password = password)
1004        student_object.manage_setLocalRoles(student_id, ['Owner',])
1005        self.do_timing()
1006        #logger.info("creating %s reg_no %s" % (student_id,reg_no))
1007        #
1008        # application
1009        #
1010        student_object.invokeFactory('StudentApplication','application')
1011        application = student_object.application
1012        #self.portal_workflow.doActionFor(application,'open',dest_container=application)
1013        #self.do_timing()
1014        da = {'Title': 'Application Data'}
1015        da['jamb_reg_no'] = reg_no
1016
1017        sex = 'M'
1018        if brain.sex:
1019            sex = 'F'
1020        da['jamb_sex'] = sex
1021        da['jamb_age'] = brain.jamb_age
1022        da['app_reg_pin'] = brain.pin
1023        da['jamb_lga'] = brain.jamb_lga
1024        da['jamb_state'] = brain.jamb_state
1025        da['jamb_score'] = brain.aggregate
1026        da['app_email'] = brain.email
1027        da['app_mobile'] = brain.phone
1028
1029        da['entry_mode'] = self._v_certificates[brain.course_admitted]['study_mode']
1030
1031        da['entry_session'] = entry_session
1032        da['jamb_lastname'] = brain.lastname
1033        da['jamb_middlename'] = brain.middlenames   # different field names!
1034        da['jamb_firstname'] = brain.firstname
1035        da['screening_application_date'] = brain.application_date
1036        da['date_of_birth'] = brain.date_of_birth
1037        da['jamb_first_cos'] = brain.course1
1038        da['jamb_second_cos'] = brain.course2
1039        da['course3'] = brain.course3
1040        da['screening_type'] = brain.screening_type
1041        da['screening_score'] = brain.screening_score
1042        da['screening_date'] = brain.screening_date
1043        da['hq_type'] = brain.hq_type
1044        da['hq_grade'] = brain.hq_grade
1045        da['aos'] = brain.aos
1046
1047        application.getContent().edit(mapping=da)
1048        self.do_timing()
1049        #self.portal_workflow.doActionFor(application,'close',dest_container=application)
1050        #
1051        # personal
1052        #
1053        student_object.invokeFactory('StudentPersonal','personal')
1054        personal = student_object.personal
1055        #self.portal_workflow.doActionFor(personal,'open',dest_container=personal)
1056        #self.do_timing()
1057        dp = {'Title': 'Personal Data'}
1058        dp['sex'] = brain.sex
1059        dp['email'] = brain.email
1060        dp['phone'] = brain.phone
1061        dp['lastname'] = brain.lastname
1062        dp['middlename'] = brain.middlenames   # different field names!
1063        dp['firstname'] = brain.firstname
1064        personal.getContent().edit(mapping=dp)
1065        self.do_timing()
1066        #
1067        # clearance
1068        #
1069        student_object.invokeFactory('StudentClearance','clearance')
1070        clearance = student_object.clearance
1071        #self.portal_workflow.doActionFor(clearance,'open',dest_container=clearance)
1072        dc = {'Title': 'Clearance/Eligibility Record'}
1073        dc['lga'] = brain.lga
1074        dc['birthday'] = brain.date_of_birth
1075        clearance.getContent().edit(mapping=dc)
1076        self.do_timing()
1077        #self.portal_workflow.doActionFor(clearance,'close',dest_container=clearance)
1078        #
1079        # study Course
1080        #
1081        student_object.invokeFactory('StudentStudyCourse','study_course')
1082        studycourse = student_object.study_course
1083        #self.portal_workflow.doActionFor(studycourse,'open',dest_container=studycourse)
1084        #self.do_timing()
1085        dsc = {}
1086        dsc['study_course'] = brain.course_admitted
1087        dsc['current_verdict'] = ''
1088        dsc['current_mode'] = da['entry_mode']
1089        if da['entry_mode'].startswith('de'):
1090            dsc['current_level'] = '200'
1091        elif da['entry_mode'].startswith('pre'):
1092            dsc['current_level'] = '000'
1093        else:
1094            dsc['current_level'] = '100'
1095        dsc['current_session'] = entry_session
1096        studycourse.getContent().edit(mapping=dsc)
1097        self.do_timing()
1098        #
1099        # payments folder
1100        student_object.invokeFactory('PaymentsFolder','payments')
1101        payments = getattr(student_object,'payments')
1102        #self.do_timing()
1103        dpay = {}
1104        dpay['Title'] = 'Payments'
1105        payments.getContent().edit(mapping=dpay)
1106        self.portal_workflow.doActionFor(payments,'open')
1107        self.do_timing()
1108        #
1109        # passport foto
1110        app_picture ="%s/import/images/%s/%s_passport.jpg" % (i_home,
1111                                                              brain.screening_type,
1112                                                              brain.reg_no)
1113        images_dir = getImagesDir(student_id)
1114        #images_dir = os.path.join("%s" % images_base,student_id)
1115        letter_dir,student_dir = os.path.split(images_dir)
1116        if not os.path.exists(letter_dir):
1117            os.mkdir(letter_dir)
1118        if not os.path.exists(images_dir):
1119            os.mkdir(images_dir)
1120        image_name = os.path.join(images_dir,"passport_%(student_id)s.jpg" % vars())
1121        if os.path.exists(app_picture):
1122            copy2(app_picture,image_name)
1123        else:
1124            logger.info('passport of %s/%s not found: %s' % (student_id,
1125                                                             brain.reg_no,
1126                                                             app_picture))
1127
1128        self.do_timing()
1129        self.print_timing()
1130        if with_timing:
1131            self.timer_step = 0
1132            self._v_timer_count += 1
1133            self._v_total += self.total_time
1134        return student_id
1135    ###)
1136
1137    security.declareProtected(ModifyPortalContent,'makeStudentLevel') ###(
1138    def makeStudentLevel(self,student_id):
1139        "create the StudyLevel for a returning Student"
1140        #import pdb;pdb.set_trace()
1141        logger = logging.getLogger('WAeUPTool.makeStudentLevel')
1142        students_folder = self.portal_url.getPortalObject().campus.students
1143        res = self.students_catalog(id=student_id)
1144        if res:
1145            st = res[0]
1146        course = st.course
1147        matric_no = st.matric_no
1148        level = st.level
1149        res = self.results_import(matric_no = matric_no)
1150        if res:
1151            results = res
1152        logger.info('%s creating Level %s' % (student_id,level))
1153        #
1154        # Level
1155        #
1156        student_obj = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1157        studycourse = getattr(student_obj,"study_course",None)
1158        self.portal_workflow.doActionFor(studycourse,'close_for_edit',dest_container=studycourse)
1159        l = getattr(studycourse,level,None)
1160        if l is None:
1161            studycourse.invokeFactory('StudentStudyLevel', level)
1162            l = getattr(studycourse, level)
1163            self.portal_workflow.doActionFor(l,'open',dest_container=l)
1164            l.getContent().edit(mapping={'Title': "Level %s" % level})
1165        ###)
1166
1167    security.declarePublic('getHallInfo') ###(
1168    def getHallInfo(self,bed):
1169        """return Hall Info"""
1170        info = {}
1171        bedsplit = bed.split('_')
1172        if len(bedsplit) == 4:
1173            hall,block,room,letter = bed.split('_')
1174        else:
1175            info['maintenance_code'] = 'None'
1176            return info
1177        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="AccoHall",id=hall)
1178        if res and len(res) == 1:
1179            hall_brain = res[0]
1180            hall_doc = hall_brain.getObject().getContent()
1181        else:
1182            return info
1183        info['hall_title'] = hall_brain.Title
1184        info['maintenance_code'] = hall_doc.maintenance_code
1185        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="ScratchCardBatch")
1186        batch_doc = None
1187        for brain in res:
1188            if brain.id.startswith(info['maintenance_code']):
1189                batch_doc = brain.getObject().getContent()
1190                break
1191        if batch_doc is None:
1192            info['maintenance_fee'] = ''
1193        else:
1194            info['maintenance_fee'] = batch_doc.cost
1195        return info
1196    ###)
1197
1198    security.declareProtected(ModifyPortalContent,'removePictureFolder') ###(
1199    def removePictureFolder(self,student_id):
1200        """remove picture_folder by renaming it"""
1201        path = 'images'
1202        picture_path = os.path.join(i_home,path,student_id)
1203        if not os.path.exists(picture_path):
1204            return False
1205        os.rename(picture_path,picture_path + "_removed")
1206        return True
1207    ###)
1208
1209    security.declareProtected(ModifyPortalContent,'restorePictureFolder') ###(
1210    def restorePictureFolder(self,student_id):
1211        """restore picture_folder by renaming it"""
1212        path = 'images'
1213        picture_path = os.path.join(i_home,path,student_id)
1214        if not os.path.exists(picture_path + "_removed"):
1215            return False
1216        os.rename(picture_path + "_removed",picture_path)
1217        return True
1218    ###)
1219
1220    security.declarePublic('picturesExist') ###(
1221    def picturesExist(self, ids,student_id=None):
1222        """check if pictures exist in the filesystem"""
1223        if student_id is None:
1224            student_id = self.getStudentId()
1225        if student_id is None:
1226            return False
1227        picture_path = getImagesDir(student_id)
1228        #picture_path = os.path.join(images_base,student_id)
1229        if not os.path.exists(picture_path):
1230            return False
1231        pictures  = [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1232        return set(ids).issubset(set(pictures))
1233    ###)
1234
1235    security.declarePublic('picturesList') ###(
1236    def picturesList(self):
1237        """check if pictures exist in the filesystem"""
1238        path = 'images'
1239        student_id = self.getStudentId()
1240        #picture_path = os.path.join(i_home,path,student_id)
1241        picture_path = getImagesDir(student_id)
1242        if not os.path.exists(picture_path):
1243            return []
1244        return [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1245    ###)
1246
1247    security.declarePublic('showFsPicture') ###(
1248    def showFsPicture(self,path):
1249        """return a picture from the filesystem"""
1250        #picture_path = os.path.join(i_home,path)
1251        picture_path = os.path.join(images_base,path)
1252        response = self.REQUEST.RESPONSE
1253        #import pdb;pdb.set_trace()
1254        registry = getToolByName(self, 'mimetypes_registry')
1255        mimetype = str(registry.lookupExtension(path.lower()) or
1256                    registry.lookupExtension('file.bin'))
1257        if os.path.exists(picture_path):
1258            response.setHeader('Content-type',mimetype)
1259            return open(picture_path).read()
1260        picture_path = os.path.join(i_home,'import',path)
1261        if os.path.exists(picture_path):
1262            return open(picture_path).read()
1263    ###)
1264
1265    security.declareProtected(ModifyPortalContent,'deleteAllCourses') ###(
1266    def deleteAllCourses(self,department="All"):
1267        ''' delete the courses'''
1268        pm = self.portal_membership
1269        member = pm.getAuthenticatedMember()
1270
1271        if str(member) not in ("henrik","joachim"):
1272            return "not possible"
1273        if department == "All":
1274            res = self.portal_catalog({'meta_type': 'Department'})
1275        if len(res) < 1:
1276            return "No Departments found"
1277
1278        deleted = []
1279        for dep in res:
1280            cf = dep.getObject().courses
1281            if cf:
1282                cf.manage_delObjects(ids=cf.objectIds())
1283                deleted.append("deleted Courses in %s" % dep.getId)
1284        return "\r".join(deleted)
1285    ###)
1286
1287    security.declareProtected(ModifyPortalContent,'getLogfileLines') ###(
1288    def getLogfileLines(self,filename="event.log",numlines=20):
1289        """Get last NUMLINES lines of logfile FILENAME.
1290
1291        Return last lines' of a file in the instances logfile directory as
1292        a list. The number of returned lines equals `numlines' or less. If
1293        less than `numlines' lines are available, the whole file ist
1294        returned. If the file can not be opened or some other error
1295        occurs, empty list is returend.
1296        """
1297        result = []
1298        lines_hit = 0
1299
1300        # We only handle files in instances' log directory...
1301        logpath = os.path.join(i_home, "log")
1302        filename = str(os.path.abspath( os.path.join( logpath, filename )))
1303        if not filename.startswith( logpath ):
1304            # Attempt to access file outside log-dir...
1305            return []
1306
1307        try:
1308            fd = file( filename, "rb" )
1309        except IOError:
1310            return []
1311        if not fd:
1312            return []
1313
1314        if os.linesep == None:
1315            linesep = '\n'
1316        else:
1317            linesep = os.linesep
1318
1319        # Try to find 'numlines' times a lineseparator, searching from end
1320        # and moving to the beginning of file...
1321        fd.seek( 0, 2) # Move to end of file...
1322        while lines_hit < numlines:
1323            if fd.read(1) == linesep[-1]: # This moves filedescriptor
1324                                          # one step forward...
1325                lines_hit += 1
1326            try:
1327                fd.seek( -2, 1) # Go two bytes back from current pos...
1328            except IOError:
1329                # We cannot go back two bytes. Maybe the file is too small...
1330                break
1331        fd.seek(2,1)
1332
1333        # Read all lines from current position...
1334        result = fd.readlines()
1335        # Remove line endings...
1336        result = [x.strip() for x in result]
1337        fd.close()
1338        return result
1339    ###)
1340
1341    security.declareProtected(ModifyPortalContent,"getCallbacksFromLog")###(
1342    def getCallbacksFromLog(self,filename):
1343        """fix Online Payment Transactions from Z2.log entries"""
1344        import transaction
1345        import random
1346        from cgi import parse_qs
1347        from urlparse import urlparse
1348        #from pdb import set_trace
1349        wftool = self.portal_workflow
1350        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
1351        students_folder = self.portal_url.getPortalObject().campus.students
1352        s = r'(?P<client_ip>\S+) - (?P<member_id>\S+) \['
1353        s += r'(?P<date>.*)\] "(?P<get>.*)" (?P<codes>\d+ \d+) "'
1354        s += r'(?P<intersw>.*)" "(?P<agent>.*)"'
1355        data = re.compile(s)
1356        start = True
1357        tr_count = 1
1358        total = 0
1359        #name = 'pume_results'
1360        #name = 'epaymentsuccessful_z2log2'
1361        name = filename
1362        no_import = []
1363        imported = []
1364        logger = logging.getLogger('WAeUPTool.getFailedTransactions')
1365        try:
1366            transactions = open("%s/import/%s" % (i_home,name),"rb").readlines()
1367        except:
1368            logger.error('Error reading %s' % name)
1369            return
1370        tas = []
1371        for line in transactions:
1372            dict = {}
1373            items = data.search(line)
1374            dict['idict'] = idict = items.groupdict()
1375            #print idict
1376            #from pdb import set_trace;set_trace()
1377            urlparsed = urlparse(idict['get'][4:])
1378            #print urlparsed
1379            path = urlparsed[2].split('/')
1380            dict['student_id'] = student_id = path[8]
1381            dict['payment_id'] = payment_id = path[10]
1382            dict['qs_dict'] = qs_dict = parse_qs(urlparsed[4])
1383            tas.append(dict)
1384            tr_count += 1
1385        return tas
1386    ###)
1387
1388    security.declareProtected(ModifyPortalContent,"importOnlinePaymentTransactions")###(
1389    def importOnlinePaymentTransactions(self):
1390        """load Online Payment Transactions from CSV values"""
1391        import transaction
1392        import random
1393        #from pdb import set_trace
1394        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
1395        opt = self.online_payments_import
1396        students_folder = self.portal_url.getPortalObject().campus.students
1397        start = True
1398        tr_count = 1
1399        total = 0
1400        #name = 'pume_results'
1401        name = 'OnlineTransactions'
1402        no_import = []
1403        imported = []
1404        logger = logging.getLogger('WAeUPTool.importOnlinePaymentTransactions')
1405        try:
1406            transactions = csv.DictReader(open("%s/import/%s.csv" % (i_home,name),"rb"))
1407        except:
1408            logger.error('Error reading %s.csv' % name)
1409            return
1410        for pay_transaction in transactions:
1411            if start:
1412                start = False
1413                logger.info('Start loading from %s.csv' % name)
1414                s = ','.join(['"%s"' % fn for fn in pay_transaction.keys()])
1415                no_import.append('%s,"Error"' % s)
1416                format = ','.join(['"%%(%s)s"' % fn for fn in pay_transaction.keys()])
1417                format_error = format + ',"%(Error)s"'
1418            data = {}
1419
1420            # format of the first file sent by Tayo
1421            #data['datetime'] = date = DateTime.DateTime(pay_transaction['Date'])
1422            #data['student_id'] = student_id = pay_transaction['Payer ID']
1423            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1424            #data['response_code'] = response_code = pay_transaction['Resp Code']
1425            #data['amount'] = amount = pay_transaction['Amount']
1426
1427            # format of the second file sent by Tayo
1428            #data['datetime'] = date = 0
1429            #data['student_id'] = student_id = pay_transaction['Payer ID']
1430            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1431            #data['response_code'] = response_code = '00'
1432            #data['amount'] = amount = pay_transaction['Amount']
1433
1434            # format of the third file sent by Kehinde
1435            data['datetime'] = date = 0
1436            data['student_id'] = student_id = pay_transaction['customer_id']
1437            data['order_id'] = order_id = pay_transaction['merchant_reference']
1438            data['response_code'] = response_code = '00'
1439            data['amount'] = amount = pay_transaction['Amount']
1440
1441            dup = False
1442            if response_code == "12":
1443                continue
1444            try:
1445                opt.addRecord(**data)
1446            except ValueError:
1447                dup = True
1448            #from pdb import set_trace;set_trace()
1449            if dup:
1450                if response_code == "00":
1451                    try:
1452                        opt.modifyRecord(**data)
1453                    except:
1454                        logger.info("duplicate uid, order_id %(order_id)s, student_id %(student_id)s, response_code %(response_code)s" % data)
1455                        continue
1456                else:
1457                    pay_transaction['Error'] = "Duplicate order_id"
1458                    no_import.append( format_error % pay_transaction)
1459                    logger.info("duplicate order_id %(order_id)s for %(student_id)s %(response_code)s" % data)
1460                    continue
1461            tr_count += 1
1462            if tr_count > 1000:
1463                if len(no_import) > 0:
1464                    open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1465                             '\n'.join(no_import) + '\n')
1466                    no_import = []
1467                em = '%d transactions committed\n' % (tr_count)
1468                transaction.commit()
1469                regs = []
1470                logger.info(em)
1471                total += tr_count
1472                tr_count = 0
1473        open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1474                                                '\n'.join(no_import))
1475        return self.REQUEST.RESPONSE.redirect("%s" % self.REQUEST.get('URL1'))
1476    ###)
1477
1478    security.declareProtected(ModifyPortalContent,"importData")###(
1479    def importData(self,filename,name,edit=False,bypass_queue_catalog=False):
1480        """load data from CSV values"""
1481        import transaction
1482        import random
1483        students_folder = self.portal_url.getPortalObject().campus.students
1484        uploads_folder = self.portal_url.getPortalObject().campus.uploads
1485        pending_only = False
1486        pend_str = '--'
1487        elapse = time.time()
1488        #
1489        # preparations
1490        #
1491        if filename == pend_str:
1492            pending_only = True
1493        importer_name = ''.join([part.capitalize() for part in name.split('_')])
1494        importer = eval("%sImport" % importer_name)(self)
1495        logger = importer.logger
1496        if importer.init_errors:
1497            logger.info(importer.init_errors)
1498            return importer.init_errors
1499        member = importer.member
1500        #current = importer.current
1501        import_date = importer.import_date
1502        #
1503        # not_imported
1504        #
1505        info = importer.info
1506        data_keys = importer.data_keys
1507        csv_keys = importer.csv_keys
1508        #csv_keys.extend(info.keys())
1509        headline_mapping = dict((k,k) for k in csv_keys)
1510        #
1511        # pending
1512        #
1513        pending_path = importer.pending_path
1514        pending_tmp = importer.pending_tmp
1515        pending_backup = importer.pending_backup
1516        pending_fn = importer.pending_fn
1517        imported_path = importer.imported_path
1518        imported_fn = importer.imported_fn
1519        commit_after = importer.commit_after
1520        pending = []
1521        pending_digests = []
1522        #total_added_to_pending = 0
1523        if not pending_only:
1524            pending,pending_digests = importer.makeIdLists()
1525            pending_at_start = len(pending)
1526        datafile = open(pending_tmp,"w")
1527        pending_csv_writer = csv.DictWriter(datafile,
1528                                                    csv_keys,
1529                                                    extrasaction='ignore')
1530        pending_csv_writer.writerow(headline_mapping)
1531        datafile.close()
1532        #
1533        # imported
1534        #
1535        if not os.path.exists(imported_path):
1536            datafile = open(imported_path,"w")
1537            imported_csv_writer = csv.DictWriter(datafile,
1538                                                 csv_keys,
1539                                                 extrasaction='ignore')
1540            imported_csv_writer.writerow(headline_mapping)
1541            datafile.close()
1542        start = True
1543        tr_count = 0
1544        total = 0
1545        total_added_to_imported = 0
1546        total_pending = 0
1547        import_source_done = ""
1548        if pending_only:
1549            import_source_path = pending_path
1550        else:
1551            import_source_path = "%s/import/%s.csv" % (i_home,filename)
1552            import_source_done = "%s/import/%s.done" % (i_home,filename)
1553        if not os.path.exists(import_source_path):
1554            fn = os.path.split(import_source_path)[1]
1555            em = 'no import file %(fn)s' % vars()
1556            return em
1557        import_source_fn = os.path.split(import_source_path)[1]
1558        if not pending_only:
1559            info['imported_from'] = import_source_fn
1560        headline = csv.reader(open(import_source_path,"rb")).next()
1561        if "import_mode" not in headline:
1562            msg = 'import_mode must be in heading'
1563            return msg
1564        invalid_keys = importer.checkHeadline(headline)
1565        if invalid_keys:
1566            return 'not ignorable key(s): "%s" found in heading' % ", ".join(invalid_keys)
1567
1568        import_keys = [k.strip() for k in headline if not (k.strip().startswith('ignore')
1569                                                         or k.strip() in info.keys())]
1570        # diff2schema = set(import_keys).difference(set(importer.schema.keys()))
1571        # diff2layout = set(import_keys).difference(set(importer.layout.keys()))
1572        # if diff2schema and diff2schema != set(['id',]):
1573        #     msg = 'not ignorable key(s): "%s" found in heading' % ", ".join(diff2schema)
1574        #     return msg
1575        #
1576        # start importing
1577        #
1578        try:
1579            reader = csv.DictReader(open(import_source_path,"rb"))
1580        except:
1581            msg = 'Error reading %s.csv' % filename
1582            logger.error(msg)
1583            return msg
1584        items = [item for item in reader]
1585        total_to_import = len(items)
1586        tti_float = float(total_to_import)
1587        if pending_only:
1588            pending_at_start = total_to_import
1589        count = 0
1590        imported = []
1591        old_commit_count = 0
1592        error_count = imported_count = 0
1593        already_in = 0
1594        for record in items:
1595            item = {}
1596            empty_value_keys = []
1597            for k,v in record.items():
1598                if k is None:
1599                    continue
1600                if v:
1601                    if v == EMPTY:
1602                        empty_value_keys += k,
1603                        v = ''
1604                    item[k.strip()] = v.strip()
1605            count += 1
1606            if start:
1607                start = False
1608                adapters = [MappingStorageAdapter(importer.schema, item)]
1609                logger.info('%(member)s starts import from %(import_source_fn)s' % vars())
1610            dm = DataModel(item, adapters,context=self)
1611            ds = DataStructure(data=item,datamodel=dm)
1612            error_string = ""
1613            total += 1
1614            import_mode = item.get('import_mode','')
1615            import_method = getattr(importer, '%(import_mode)s' % vars(),None )
1616            if import_method is None:
1617                error_string += "import_mode '%(import_mode)s' is invalid" % vars()
1618            elif (import_mode in importer.required_modes and
1619                not set(importer.required_keys[import_mode]).issubset(set(item.keys()))):
1620                diff2import = set(importer.required_keys[import_mode]).difference(set(item.keys()))
1621                error_string += 'required key(s): "%s" not found in record' % ", ".join(diff2import)
1622            else:
1623                for k in import_keys:
1624                    if k not in item.keys() or k not in importer.validators.keys():
1625                        continue
1626                    if not importer.validators[k](ds,mode=import_mode):
1627                        error_string += ' ++ '
1628                        error_string += "%s: %s" % (k,self.translation_service(ds.getError(k),
1629                                                                           ds.getErrorMapping(k)))
1630            if error_string:
1631                error = error_string
1632                id = ''
1633                mapping = item
1634            else:
1635                temp_item = item.copy()
1636                temp_item.update(dm)
1637                results = import_method(temp_item)
1638                id = results[0]
1639                error = results[1]
1640                mapping = results[2]
1641            #
1642            # restore some values
1643            #
1644            for k in empty_value_keys:
1645                mapping[k] = EMPTY
1646            if mapping.has_key('sex'):
1647                #import pdb;pdb.set_trace()
1648                if mapping['sex'] == True:
1649                    mapping['sex'] = 'F'
1650                elif mapping['sex'] == False:
1651                    mapping['sex'] = 'M'
1652            if pending_only:
1653                info['imported_from'] = item['imported_from']
1654            data_string = ", ".join("%s: %s" % (k,v) for k,v in mapping.items())
1655            info['error'] = error
1656            info['import_record_no'] = count + 1
1657            mapping.update(info)
1658            log_list = []
1659            if error:
1660                error_count += 1
1661                digest = makeDigest(mapping,data_keys)
1662                if digest not in pending_digests:
1663                    pending_digests += digest,
1664                    pending.append(mapping)
1665                    if not pending_only:
1666                        log_list += "record from %(import_source_fn)s added to %(pending_fn)s, %(data_string)s, %(error)s" % vars(),
1667                else:
1668                    already_in += 1
1669                    pass
1670            else:
1671                imported_count += 1
1672                imported += mapping,
1673                log_list += "record imported and added to %(imported_fn)s from %(import_source_fn)s, %(data_string)s" % vars(),
1674            if log_list:
1675                time_till_now = time.time() - elapse
1676                percent_finished = (error_count + imported_count)/tti_float*100
1677                log_list.insert(0,("%(percent_finished)6.3f %% done in %(time_till_now)3.2fs," % vars()),)
1678                logger.info(' '.join(log_list))
1679            finished = count > total_to_import - 1
1680            must_commit = (imported_count != old_commit_count) and (not imported_count % commit_after)
1681            if must_commit:
1682                old_commit_count = imported_count
1683
1684            if must_commit or finished:
1685                if len(imported):
1686                    transaction.commit()
1687                    datafile = open(imported_path,"a")
1688                    writer = csv.DictWriter(datafile,
1689                                            csv_keys,
1690                                            extrasaction='ignore')
1691                    writer.writerows(imported)
1692                    datafile.close()
1693                    total_added_to_imported += len(imported)
1694                    imported = []
1695                if len(pending) > 0:
1696                    datafile = open(pending_tmp,"a")
1697                    writer = csv.DictWriter(datafile,
1698                                            csv_keys,
1699                                            extrasaction='ignore')
1700                    writer.writerows(pending)
1701                    datafile.close()
1702                    total_pending += len(pending)
1703                    #total_added_to_pending += len(pending)
1704                    pending = []
1705                if not finished:
1706                    msg = '%(commit_after)d records imported and committed of total %(total_added_to_imported)d\n' % vars()
1707                    logger.info(msg)
1708        elapse = time.time() - elapse
1709        if os.path.exists(pending_path):
1710            copy2(pending_path,pending_backup)
1711        copy2(pending_tmp,pending_path)
1712        msg = "finished importing from %(import_source_fn)s in %(elapse).2f seconds, " % vars()
1713        msg += "%(count)d records totally read, %(total_added_to_imported)d added to %(imported_fn)s, " % vars()
1714        if pending_only:
1715            removed_pending = pending_at_start - total_pending
1716            msg += "%(removed_pending)d removed from %(pending_fn)s" % vars()
1717        else:
1718            added_pending = total_pending - pending_at_start
1719            msg += "%(added_pending)d added to %(pending_fn)s, %(already_in)s already in %(pending_fn)s" % vars()
1720        #msg += "%(total_pending)d totally written" % vars()    # this line does not make any sense
1721        logger.info(msg)
1722        if import_source_done:
1723            copy(import_source_path,import_source_done)
1724            os.remove(import_source_path)
1725            upload = getattr(uploads_folder,os.path.split(import_source_path)[1],None)
1726            if upload is not None:
1727                upload_doc = upload.getContent()
1728                mapping = {}
1729                #mapping['import_date'] = DateTime.DateTime()
1730                mapping['import_date'] = import_date
1731                mapping['imported_by'] = importer.imported_by
1732                mapping['import_message'] = msg
1733                upload_doc.edit(mapping = mapping)
1734        os.remove(pending_tmp)
1735        return msg
1736    ###)
1737
1738    security.declareProtected(ModifyPortalContent,"moveImagesToFS")###(
1739    def moveImagesToFS(self,student_id="O738726"):
1740        "move the images to the filesystem"
1741        images_dir = getImagesDir(student_id)
1742        #images_dir = os.path.join("%s" % images_base,student_id)
1743        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1744        stool = getToolByName(self, 'portal_schemas')
1745        schemas = ['student_application',
1746                   'student_clearance',
1747                   ]
1748        created = False
1749        for schema_id in schemas:
1750            schema = stool._getOb(schema_id)
1751            object = getattr(student_folder,schema_id[len('student_'):],None)
1752            if object is None:
1753                continue
1754            doc = object.getContent()
1755            for key in schema.keys():
1756                if schema[key].meta_type != "CPS Image Field":
1757                    continue
1758                #import pdb;pdb.set_trace()
1759                image = getattr(doc,key,None)
1760                if not image or not hasattr(image,"data"):
1761                    continue
1762                if not created:
1763                    if not os.path.exists(images_dir):
1764                        os.mkdir(images_dir)
1765                    created = True
1766                filename = os.path.join(images_dir,"%(key)s_%(student_id)s.jpg" % vars())
1767                open(filename,"wb").write(str(image.data))
1768    ###)
1769
1770    security.declareProtected(ModifyPortalContent,"movePassportToFS")###(
1771    def movePassportToFS(self,student_id="O738726"):
1772        "move the passports to the filesystem"
1773        images_dir = os.path.join("%s" % i_home,'passports')
1774        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1775        stool = getToolByName(self, 'portal_schemas')
1776        schemas = ['student_application',
1777                   #'student_clearance',
1778                   ]
1779        created = False
1780        for schema_id in schemas:
1781            schema = stool._getOb(schema_id)
1782            object = getattr(student_folder,schema_id[len('student_'):],None)
1783            if object is None:
1784                continue
1785            doc = object.getContent()
1786            for key in schema.keys():
1787                if schema[key].meta_type != "CPS Image Field":
1788                    continue
1789                #import pdb;pdb.set_trace()
1790                image = getattr(doc,key)
1791                if not hasattr(image,"data"):
1792                    continue
1793                if not created:
1794                    if not os.path.exists(images_dir):
1795                        os.mkdir(images_dir)
1796                    created = True
1797                filename = os.path.join(images_dir,"%(student_id)s.jpg" % vars())
1798                open(filename,"wb").write(str(image.data))
1799    ###)
1800
1801InitializeClass(WAeUPTool)
Note: See TracBrowser for help on using the repository browser.