source: WAeUP_SRP/trunk/WAeUPTool.py @ 4241

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

resolve aaue #24

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