source: WAeUP_SRP/trunk/WAeUPTool.py @ 5378

Last change on this file since 5378 was 5373, checked in by Henrik Bettermann, 14 years ago

use jamb_reg_no if provided

  • Property svn:keywords set to Id
File size: 84.0 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 5373 2010-08-05 06:20:15Z 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
[4302]386        while self.students_catalog(id = sid) or self.waeup_tool.picturePathExists(sid) or self.removed_student_ids(id = sid):
[5019]387            logger.info('skipped used id %s' % sid)
[1720]388            sid = "%c%d" % (letter,r.randint(99999,1000000))
[1151]389        return sid
[1460]390    ###)
[1415]391
392    def generatePassword(self,s=None): ###(
393        import random
394        r = random
395        ##if letter not in ('ABCDEFGIHKLMNOPQRSTUVWXY'):
396        if s is None:
397            s = 'abcdefghklmnpqrstuvwxy23456789'
398        pw = ''
399        while len(pw) < 6:
400            pw += r.choice(s)
401        return pw
[1151]402    ###)
[828]403
[2645]404    security.declareProtected(ModifyPortalContent, 'dumpSchoolfeePayments') ###(
405    def dumpSchoolfeePayments(self):
406        "dump paid schoolfees"
[2666]407        mtool = self.portal_membership
408        member = mtool.getAuthenticatedMember()
[2645]409        logger = logging.getLogger('WAeUPTool.dumpSchoolfees')
410        aq_student = self.students_catalog.evalAdvancedQuery
411        query = In('review_state',('schoolfee_paid',
412                                   'courses_registered',
413                                   'courses_validated',
414                                   ))
415        res = aq_student(query)
[2666]416        #import pdb;pdb.set_trace()
[2645]417        l = []
418        logger.info("start for %d" % len(res))
419        count = 1
420        log_after = 100
421        for student in res:
[2656]422            if not count % log_after:
[2645]423                logger.info("processed %d total %d" % (log_after,count))
424            count += 1
425            fee_dict =self.getSchoolFee(student)
426            fulltime = student.mode.endswith('_ft')
427            d = {}
428            d['student_id'] = student.id
[2666]429            d['name'] = student.name
430            d['amount'] = fee_dict.get(new_returning)
[2645]431            l += d,
[2666]432        csv_name = self.dumpListToCSV(l,'payments')
433        logger.info('%s dumped payments to %s' % (member,export_file))
[2645]434    ###)
435
[2666]436    security.declarePublic('dumpListToCSV') ###(
437    def dumpListToCSV(self,l,filename,fields=None):
438        """dump a list of dicts to a CSV file"""
439        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
440        export_file = "%s/export/%s_%s.csv" % (i_home,filename,current,)
441        if fields is None:
442            fields = l[0].keys()
443        headline = ','.join(fields)
444        out = open(export_file,"wb")
445        out.write(headline +'\n')
446        out.close()
447        out = open(export_file,"a")
448        csv_writer = csv.DictWriter(out,fields,)
449        csv_writer.writerows(l)
450        return export_file
451    ###)
452
[2643]453    security.declareProtected(ManagePortal, 'listMembers') ###(
454    def listMembers(self):
455        "list all members"
456        mtool = self.portal_membership
457        member = mtool.getAuthenticatedMember()
458        logger = logging.getLogger('WAeUPTool.listMembers')
459        if str(member) not in ('admin','joachim'):
460            logger.info('%s tried to list members' % (member))
461            return None
462        members = self.portal_directories.members
463        all = members.listEntryIdsAndTitles()
464        l = []
465        for user_id,name in all:
466            d = {}
467            d['user_id'] = user_id
468            d['name'] = name
469            d['pw'] = getattr(getattr(members,user_id),'password')
[2645]470            d['email'] = getattr(getattr(members,user_id),'email')
471            d['groups'] = " ".join(getattr(getattr(members,user_id),'groups'))
472            d['roles'] = " ".join(getattr(getattr(members,user_id),'roles'))
[2643]473            l += d,
474        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
475        export_file = "%s/export/member_list_%s.csv" % (i_home,current,)
[3272]476        logger.info('%s dumped member list to %s' % (member,export_file))
[2643]477        fields = l[0].keys()
478        headline = ','.join(fields)
479        out = open(export_file,"wb")
480        out.write(headline +'\n')
481        out.close()
482        out = open(export_file,"a")
483        csv_writer = csv.DictWriter(out,fields,)
484        csv_writer.writerows(l)
485    ###)
[3300]486
[3272]487    security.declareProtected(ManagePortal, 'listStudents') ###(
488    def listStudents(self):
489        "list all students"
490        mtool = self.portal_membership
491        member = mtool.getAuthenticatedMember()
492        logger = logging.getLogger('WAeUPTool.listStudents')
493        if str(member) not in ('admin','joachim'):
494            logger.info('%s tried to list students' % (member))
495            return None
496        students = self.portal_directories.students
497        all = students.listEntryIdsAndTitles()
498        l = []
499        for user_id,name in all:
500            d = {}
501            d['user_id'] = user_id
502            d['name'] = name
503            d['pw'] = getattr(getattr(students,user_id),'password')
504            d['email'] = getattr(getattr(students,user_id),'email')
505            d['groups'] = " ".join(getattr(getattr(students,user_id),'groups'))
506            d['roles'] = " ".join(getattr(getattr(students,user_id),'roles'))
507            l += d,
508        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
509        export_file = "%s/export/student_list_%s.csv" % (i_home,current,)
510        logger.info('%s dumped student list to %s' % (member,export_file))
511        fields = l[0].keys()
512        headline = ','.join(fields)
513        out = open(export_file,"wb")
514        out.write(headline +'\n')
515        out.close()
516        out = open(export_file,"a")
517        csv_writer = csv.DictWriter(out,fields,)
518        csv_writer.writerows(l)
[3300]519    ###)
[2656]520
[1890]521    security.declareProtected(ManagePortal, 'removeDeletedDocIds') ###(
522    def removeDeletedDocIds(self, max=1000):
523        """
524        remove deleted docids from repository commit after max
525        """
526        logger = logging.getLogger('WAeUPTool.removeDeletedDocIds')
527        repository = getToolByName(self, 'portal_repository')
528        pxtool = getToolByName(self, 'portal_proxies')
[2632]529        logger.info('start')
[1890]530        pxtool_infos = pxtool.getRevisionsUsed()
[2632]531        logger.info('found  %d used revisions ' % (len(pxtool_infos)))
[1895]532
[1890]533        nb_revs = 0
534        docids_d = {} # all docids
535        unused_docids_d = {} # all docids that are unused
536        ids_unused_revs_docids = [] # ids for revs of unused docids
[2632]537        unused_ids = [] # ids for unused revs
[1890]538        total = 0
[1895]539        idlist = repository.objectIds()
[2632]540        to_delete = 0
541        found = False
[1895]542        for id in idlist:
[1890]543            docid, rev = repository._splitId(id)
544            if docid is None:
545                logger.info("invalid doc_id %s" % docid)
546                continue
547            nb_revs += 1
548            if not pxtool_infos.has_key(docid):
[2632]549                found = True
550                to_delete += 1
551                unused_ids.append(id)
[1890]552            elif not pxtool_infos[docid].has_key(rev):
[2632]553                found = True
554                to_delete += 1
555                unused_ids.append(id)
556            if found and not to_delete % max:
557                found = False
[1890]558                #import pdb;pdb.set_trace()
[2632]559                repository.manage_delObjects(unused_ids)
[1890]560                transaction.commit()
[2632]561                logger.info('removed %d total %d unused docids ' % (max,to_delete))
562        else:
563            if unused_ids:
564                repository.manage_delObjects(unused_ids)
565                transaction.commit()
566        logger.info('finished removing %d unused docids ' % (to_delete))
[3473]567    ###)
[1890]568
[3858]569    security.declarePublic('getCredential') ###(
[1250]570    def getCredential(self,student_id):
571        student_entry = getattr(self.portal_directories.students,student_id,None)
[3858]572        #import pdb;pdb.set_trace()
573        #if not self.isStaff():
574        #    mtool = self.portal_membership
575        #    member = mtool.getAuthenticatedMember()
576        #    logger = logging.getLogger('WAeUPTool.getCredential')
577        #    logger.info('%s tried to access password of %s' % (member,student_id))
578        #    return None
[1250]579        if student_entry is None:
[4022]580            return "not set"
[1263]581        return getattr(student_entry,"password","not set")
[1261]582    ###)
[1250]583
[2747]584    security.declarePublic('checkPassword') ###(
[1467]585    def checkPassword(self,student_id,password):
[1460]586        student_entry = getattr(self.portal_directories.students,student_id,None)
587        if student_entry is None:
588            return False
589        return getattr(student_entry,"password","not set") == password
[2747]590    ###)
[2712]591
[2747]592    security.declarePublic('checkGenericPassword') ###(
[2710]593    def checkGenericPassword(self,member_id):
594        member_entry = getattr(self.portal_directories.members,member_id,None)
595        if member_entry is None:
596            return False
[2712]597        ltool = getToolByName(self, 'portal_layouts')
[2710]598        unsecure_words = ltool._getOb('members')['w__password'].check_words
[2712]599        password = getattr(member_entry,"password","not set")
600        is_unsecure = password in unsecure_words
601        if is_unsecure:
[2713]602            logger = logging.getLogger('WAeUPTool.checkGenericPassword')
[2712]603            logger.info('Member %s tried to log in with unsecure password %s' %(member_id,password))
604        return is_unsecure
[2747]605    ###)
[1460]606
[4027]607    security.declarePublic('editPassword') ###(
[1467]608    def editPassword(self,student_id,password):
609        "edit a student password"
610        student_entry = getattr(self.portal_directories.students,student_id,None)
611        if student_entry is None:
[1571]612            return
[1467]613        setattr(student_entry,'password',password)
614    ###)
615
[2555]616    security.declareProtected(ModifyPortalContent,'doCommit') ###(
[1401]617    def doCommit(self,logger=None):
618        "commit some transactions"
619        transaction.commit()
620    ###)
621
[1285]622    security.declarePublic('loadStudentFoto') ###(
[1813]623    def loadStudentFoto(self,student,filename,folder):
[1285]624        "return a student passport picture"
[2413]625        #import pdb;pdb.set_trace()
626        picture ="%s/import/%s/%s" % (i_home,folder,filename)
627        student_id = student.getId()
[2675]628        images_dir = getImagesDir(student_id)
[2413]629        if not os.path.exists(images_dir):
630            os.mkdir(images_dir)
631        image_name = os.path.join(images_dir,"passport_%(student_id)s.jpg" % vars())
632        for extension in ('.jpg','.JPG'):
633            fullname = "%(picture)s%(extension)s" % vars()
634            if os.path.exists(fullname):
635                copy2(fullname,image_name)
636                return "successfully copied passport picture"
[2467]637        return "passport picture not found: %s.jpg or .JPG" % picture
[2413]638    ###)
639
[1285]640
[1170]641    security.declareProtected(ModifyPortalContent,'createOne') ###(
[1194]642    def createOne(self,students_folder,student_brain,letter,commit=False):
643        sid = self.waeup_tool.generateStudentId(letter)
[1170]644        students_folder.invokeFactory('Student', sid)
645        student = getattr(students_folder,sid)
646        self.portal_workflow.doActionFor(student,'return')
647        student.manage_setLocalRoles(sid, ['Owner',])
648        matric_no = student_brain.matric_no
649        jamb_reg_no = student_brain.Entryregno
650        self.students_catalog.addRecord(id = sid,
651                                           matric_no = matric_no,
652                                           jamb_reg_no = jamb_reg_no,
653                                           sex = student_brain.Sex == "F",
654                                           name = "%s %s %s" % (student_brain.Firstname,
655                                                                student_brain.Middlename,
656                                                                student_brain.Lastname)
657                                        )
658        if commit:
659            transaction.commit()
660        return sid,jamb_reg_no
661    ###)
662
[3840]663    # Functions without docstring cannot be called from outside (via the URL),
664    # thus we can declare addStudent public.
[3838]665    security.declarePublic('addStudent') ###(
[1752]666    def addStudent(self,dict):
[1415]667        students_folder = self.portal_url.getPortalObject().campus.students
668        sid = self.waeup_tool.generateStudentId('?')
669        students_folder.invokeFactory('Student', sid)
670        student_obj = getattr(students_folder,sid)
[3481]671        f2t = StudentImport.field2types_student
[1772]672        #from pdb import set_trace; set_trace()
673        d = {}
[2297]674        #d['jamb_sex']  = 'M'
675        #if dict.get('sex'):
676        #    d['jamb_sex']  = 'F'
[1904]677
[1898]678        entry_session = dict.get('entry_session')
[2454]679        if entry_session == self.getSessionId()[0]:
[1898]680            wfaction = 'admit'
[1991]681            wft = 'wf_transition_admit'
[1900]682            password = None
[1904]683        else:
[1900]684            wfaction = 'return'
[1991]685            wft = 'wf_transition_return'
[1900]686            password = self.generatePassword()
[1904]687            self.makeStudentMember(sid,password)
688
[1728]689        for pt in f2t.keys():
[5023]690            if pt == 'StudentAccommodation':
691                continue
[1728]692            student_obj.invokeFactory(pt,f2t[pt]['id'])
693            sub_obj = getattr(student_obj,f2t[pt]['id'])
694            sub_doc = sub_obj.getContent()
[1898]695            #self.portal_workflow.doActionFor(sub_obj,'open',dest_container=sub_obj)
[3481]696            #d['Title'] = f2t[pt]['title']
[1728]697            for field in f2t[pt]['fields']:
[1749]698                d[field] = dict.get(field,'')
[1728]699            sub_doc.edit(mapping = d)
[1898]700            new_state = f2t[pt][wft]
[1772]701            if new_state != "remain":
702                self.portal_workflow.doActionFor(sub_obj,new_state,dest_container=sub_obj)
[1415]703        self.portal_workflow.doActionFor(student_obj,wfaction)
704        student_obj.manage_setLocalRoles(sid, ['Owner',])
705        return sid,password
706    ###)
707
[1151]708    security.declarePublic('getCertificateBrain') ###(
709    def getCertificateBrain(self,cert_id):
710        "do it"
[1849]711        res = ZCatalog.searchResults(self.portal_catalog_real,
[1151]712                                {'portal_type':"Certificate",
713                                      'id': cert_id})
714        if res:
715            return res[0]
716        return None
717    ###)
[1160]718
[1756]719    security.declareProtected(ModifyPortalContent,'get_csv_filenames') ###(
720    def get_csv_filenames(self):
721        "do it"
[1759]722        files = [file for file in os.listdir("%s/import/" % (i_home))
[3143]723                 if file.endswith('.csv') and (file.find('imported') == -1 and
724                                               file.find('pending') == -1)]
[1756]725        return files
726    ###)
727
[1151]728    security.declarePublic('findStudentByMatricelNo') ###(
729    def findStudentByMatricelNo(self,matric_no):
730        "do it"
[1849]731        res = ZCatalog.searchResults(self.portal_catalog_real,
[1151]732                                {'portal_type':"StudentClearance",
733                                 'SearchableText': matric_no})
734        if res:
735            return res[0]
736        return None
737    ###)
738
[3817]739
740    security.declarePublic('getOfficerName') ###(
741    def getOfficerName(self,mid):
742        """get the real name of a member"""
743        membership = self.portal_membership
744        member_record = membership.getMemberById(mid)
[5210]745        if member_record:
746            firstname = member_record.getProperty('givenName','')
747            lastname = member_record.getProperty('sn','')
748            return firstname + ' ' + lastname.upper()
749        else:
750            return mid
[3817]751    ###)
752
753
[2557]754    security.declarePublic('makeStudentMember') ###(
[1151]755    def makeStudentMember(self,sid,password='uNsEt'):
756        """make the student a member"""
757        membership = self.portal_membership
758        membership.addMember(sid,
759                             password ,
760                             roles=('Member',
761                                     'Student',
762                                     ),
763                             domains='',
764                             properties = {'memberareaCreationFlag': False,
765                                           'homeless': True},)
766        member = membership.getMemberById(sid)
767        self.portal_registration.afterAdd(member, sid, password, None)
[1261]768        #self.manage_setLocalRoles(sid, ['Owner',])
[1151]769    ###)
[1160]770
[2560]771    security.declareProtected(View,'makeStudentData') ###(
[1158]772    def makeStudentData(self,student_id,email=None,phone_nr=None):
[1151]773        "create Datastructure for a returning Student"
[1406]774        #import pdb;pdb.set_trace()
[1571]775        logger = logging.getLogger('WAeUPTool.makeStudentData')
[1151]776        students_folder = self.portal_url.getPortalObject().campus.students
[1794]777        #res = self.students_catalog(id=student_id)
778        #if res:
779        #    st = res[0]
780        #res = self.returning_import(matric_no = st.matric_no)
[1798]781        res = self.returning_import(id = student_id)
[1151]782        if res:
[1160]783            student = res[0]
[1794]784        else:
[1816]785            logger.info('Id %s not found in returning_import' % student_id)
[1794]786            return
[1580]787        logger.info('%s creates data structure' % student_id)
[1794]788        s_results = self.results_import(matric_no = student.matric_no)
[1816]789        if s_results:
790            lnr = self.getLevelFromResultsCosCode(s_results)
791            level = "%d00" % lnr
792            verdict,eligible = self.getVerdict(s_results[0].Verdict)
[2454]793            #if eligible:
794            #    level = "%d00" % (lnr + 1)
[1816]795        else:
796            logger.info('matric_no %s not found in results_import' % student.matric_no)
[2656]797            level = ''
798            verdict = ''
[1171]799        #student should not be allowed to perform this transition
[1174]800        #wftool = self.portal_workflow
801        #wftool.doActionFor(student,'return')
[1151]802        certcode_org = student.Coursemajorcode
803        certcode = makeCertificateCode(certcode_org)
804        certificate_brain = self.getCertificateBrain(certcode)
805        if not certificate_brain:
806            em = 'Certificate %s org-code %s not found\n' % (certcode, certcode_org)
807            logger.info(em)
808        matric_no = student.matric_no
809        sid = student_id
810        student_obj = getattr(students_folder,sid)
[3114]811        if not getattr(student_obj,'application'):
812            student_obj.invokeFactory('StudentApplication','application')
[1151]813        application = student_obj.application
[1169]814        self.portal_workflow.doActionFor(application,'open',dest_container=application)
[1151]815        da = {'Title': 'Application Data'}
816        student_obj.invokeFactory('StudentPersonal','personal')
817        da['jamb_reg_no'] = student.Entryregno
[1462]818        em = self.getEntryMode(student.Entryregno)
[1401]819        da['entry_mode'] = em
[1151]820        personal = student_obj.personal
821        self.portal_workflow.doActionFor(personal,'open',dest_container=personal)
822        dp = {'Title': 'Personal Data'}
823        student_obj.invokeFactory('StudentClearance','clearance')
824        clearance = student_obj.clearance
[1169]825        self.portal_workflow.doActionFor(clearance,'open',dest_container=clearance)
[1151]826        dc = {'Title': 'Clearance/Eligibility Record'}
827        dc['matric_no'] = matric_no
828        state = student.State
829        lga = student.LGA
830        if state and lga:
831            lga =  state + ' / ' + lga
832        else:
833            lga = "None"
[1174]834        da['jamb_lga'] = dc['lga'] = lga
[1173]835        da['app_email'] = dp['email'] = email
836        da['app_mobile'] = dp['phone'] = phone_nr
[1411]837        dp['firstname'] = student.Firstname
838        dp['middlename'] = student.Middlename
839        dp['lastname'] = student.Lastname
840        da['jamb_lastname'] = "%s %s %s" % (student.Firstname,student.Middlename,student.Lastname)
[1174]841        da['jamb_sex'] = student.Sex
[1151]842        dp['sex'] = student.Sex == 'F'
843        dp['perm_address'] = student.Permanent_Address
844        application.getContent().edit(mapping=da)
[1169]845        self.portal_workflow.doActionFor(application,'close',dest_container=application)
[1151]846        personal.getContent().edit(mapping=dp)
847        clearance.getContent().edit(mapping=dc)
[1169]848        self.portal_workflow.doActionFor(clearance,'close',dest_container=clearance)
[1151]849        #
850        # Study Course
851        #
852        student_obj.invokeFactory('StudentStudyCourse','study_course')
853        studycourse = student_obj.study_course
854        self.portal_workflow.doActionFor(studycourse,'open',dest_container=studycourse)
855        dsc = {}
856        dsc['study_course'] = certcode
[1401]857        dsc['current_level'] = level
858        dsc['current_verdict'] = verdict
[3780]859        dsc['current_mode'] = em   #no longer used
[2466]860        dsc['current_session'] = '05'
[1151]861        studycourse.getContent().edit(mapping=dsc)
862        #
863        # Level
864        #
[3473]865        # l = getattr(studycourse,level,None)
866        # if l is None:
867        #     studycourse.invokeFactory('StudentStudyLevel', level)
868        #     l = getattr(studycourse, level)
869        #     self.portal_workflow.doActionFor(l,'open',dest_container=l)
870        #     l.getContent().edit(mapping={'Title': "Level %s" % level})
871        ###)
[1160]872
[3349]873    def init_timing(self): ###(
874        if self.with_timing:
875            if not hasattr(self,'_v_step_times'):
876                self._v_timer_count = 0
877                self._v_total = 0
878                self._v_step_times = {}
879                current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
880                self._v_timer_file = "%s/export/timing_%s.csv" % (i_home,current,)
881            self.timer_step = 0
882            self.total_time = 0
883            self.elapse = time.time()
884            self.i_elapse = time.time()
885            self.c_elapse = time.clock()
[3473]886    ###)
[3349]887
888    def do_timing(self): ###(
889        if self.with_timing:
890            try:
891                raise 'dummy'
892            except:
893                frame = sys.exc_traceback.tb_frame.f_back
894                locals = frame.f_locals
895                globals = frame.f_globals
896                functionname = frame.f_code.co_name
897                filename = os.path.basename(frame.f_code.co_filename)
898                lineno = frame.f_lineno
899                mod_line = "%(functionname)s:%(lineno)s" % vars()
900            i_time = time.time() - self.i_elapse
901            td = {}
902            if self._v_step_times.has_key(mod_line):
903                a_time = self._v_step_times[mod_line]['a_time'] + i_time
904                td['a_time'] = a_time
905            else:
906                td['a_time'] = i_time
907            td['i_time'] = i_time
908            self._v_step_times[mod_line] = td
909            self.i_time = i_time
910            self.total_time += i_time
911            self.timer_step +=1
912            self.i_elapse = time.time()
[3473]913    ###)
[3349]914
915    security.declareProtected(ModifyPortalContent,'print_timing') ###( ###(
916    def print_timing(self):
917        if self.with_timing:
918            l = []
919            timer_count = self._v_timer_count + 1
920            mod_lines = self._v_step_times.keys()
921            mod_lines.sort(cmp,reverse=0)
922            for mod_line in mod_lines:
923                td = self._v_step_times[mod_line]
924                i_time = td['i_time']
925                a_time = td['a_time']/(self._v_timer_count + 1)
926                l += ("%(mod_line)s,%(i_time)6.2f,%(a_time)6.2f,%(timer_count)d" % vars()),
927            total_time = self.total_time
928            total_avarage = self._v_total / timer_count
929            l += ("total,%(total_time)6.4f,%(total_avarage)6.4f,%(timer_count)d" % vars()),
930            print "\r\n".join(l)
931            out = open(self._v_timer_file,'a')
932            out.write("\r\n".join(l))
933            out.close()
[3473]934    ###)
[3349]935
936    security.declareProtected(ModifyPortalContent,'get_timing_data') ###( ###(
937    def get_timing_data(self):
938        if self.with_timing:
939            timer_count = self._v_timer_count + 1
940            results = {}
941            for k,d in self._v_step_times.items():
942                dd = {}
943                dd['a_time'] = d['a_time']/timer_count
944                dd['i_time'] = d['i_time']
945                dd['count'] = timer_count
946                results[k] = dd
947            dd = {}
948            dd['a_time'] = self._v_total / timer_count
949            dd['i_time'] = self.total_time
950            dd['count'] = timer_count
951            results["total"] = dd
952            return results
953    ###)
954
[2556]955    security.declareProtected(ModifyPortalContent,'admitOneStudent') ###(
[3349]956    def admitOneStudent(self,brain,entry_session,pin_password,with_timing=False):
[2514]957        "create Datastructure for an admitted Student"
958        #import pdb;pdb.set_trace()
[3349]959        logger = logging.getLogger('WAeUPTool.admitOneStudent')
960        self.with_timing = with_timing
[3681]961
[4539]962        try:
[5188]963            if brain.screening_type in ('dp','pt','ct','cest','sandwich','sandwich2008') and brain.serial and brain.entry_session:
[4539]964                if brain.course1:
965                    reg_no = "%s%s/%s" % (brain.course1[:3],brain.serial,brain.entry_session)
966                else:
967                    reg_no = "%s%s/%s" % (brain.course_admitted[:3],brain.serial,brain.entry_session)
[5373]968            elif brain.screening_type == 'pume2':
969                reg_no = brain.jamb_reg_no.upper()
[4532]970            else:
[4539]971                reg_no = brain.reg_no
972        except:
973            logger.info('applicant record %s has errors' % (brain.reg_no))
974            return
[3705]975
[3681]976        #ignore argument entry_session
977        if not brain.entry_session:
978            logger.info('no entry_session for %s provided' % (reg_no))
979            return
[3961]980        else:
981            es = brain.entry_session
982        if len(es) == 1:
983            es = '0%c' % es
[3681]984
[2727]985        if not hasattr(self,"_v_certificates"):
[3339]986            self._v_certificates = self.getCertificatesDict()
[2514]987        students_folder = self.portal_url.getPortalObject().campus.students
[3705]988
[3264]989        res = self.students_catalog(jamb_reg_no = reg_no)
[3198]990        if res:
[3800]991            logger.info('student with reg_no %s exists (%s)' % (reg_no,res[0].id))
[3198]992            return
[2514]993        if brain.status != "admitted":
[3264]994            logger.info('status of %s is %s' % (reg_no,brain.status))
[2540]995            return
[2514]996        pin_parts = brain.pin.split('-')
[2755]997        if pin_parts and len(pin_parts) != 3:
[3264]998            logger.info('invalid pin %s for %s' % (brain.pin,reg_no))
[2540]999            return
[3764]1000        if not brain.course_admitted:
1001            logger.info('no course_admitted provided for %s' % (reg_no))
1002            return           
[2727]1003        if brain.course_admitted not in self._v_certificates:
[3264]1004            logger.info('certificate %s not found for %s' % (brain.course_admitted,reg_no))
[2727]1005            return
[2959]1006        if brain.sex not in (True,False):
[3264]1007            logger.info('sex of %s not available' % (reg_no))
[3002]1008            return
[3349]1009        self.init_timing()
[2540]1010        student_id = self.generateStudentId('?')
[2514]1011        students_folder.invokeFactory('Student', student_id)
1012        student_object = getattr(students_folder,student_id)
[3349]1013        self.do_timing()
[2576]1014        if pin_password:
1015            password = pin_parts[2]
1016            self.makeStudentMember(student_id,password = password)
[2540]1017        student_object.manage_setLocalRoles(student_id, ['Owner',])
[3349]1018        self.do_timing()
[3264]1019        #logger.info("creating %s reg_no %s" % (student_id,reg_no))
[2514]1020        #
1021        # application
1022        #
[2540]1023        student_object.invokeFactory('StudentApplication','application')
1024        application = student_object.application
[2575]1025        #self.portal_workflow.doActionFor(application,'open',dest_container=application)
[3350]1026        #self.do_timing()
[2514]1027        da = {'Title': 'Application Data'}
[3264]1028        da['jamb_reg_no'] = reg_no
[2576]1029
[2540]1030        sex = 'M'
1031        if brain.sex:
1032            sex = 'F'
1033        da['jamb_sex'] = sex
[2670]1034        da['jamb_age'] = brain.jamb_age
[3341]1035        da['app_reg_pin'] = brain.pin
[2670]1036        da['jamb_lga'] = brain.jamb_lga
1037        da['jamb_state'] = brain.jamb_state
[2575]1038        da['jamb_score'] = brain.aggregate
[2514]1039        da['app_email'] = brain.email
[2540]1040        da['app_mobile'] = brain.phone
[3339]1041
[3775]1042        # entry_mode cannot be retrieved from the certtificate
1043        # because ug_ft has two different entry modes
1044                       
[3780]1045        if brain.entry_mode:                      # does not happen because import_application
1046            da['entry_mode'] = brain.entry_mode   # schema has no field entry_mode
[3775]1047        elif brain.screening_type == 'pume': 
[3776]1048            da['entry_mode'] = 'ume_ft' 
[3775]1049        elif brain.screening_type == 'pde': 
[3776]1050            da['entry_mode'] = 'de_ft' 
[3775]1051        else:
1052            da['entry_mode'] = self._v_certificates[brain.course_admitted]['study_mode']
1053       
1054        #elif brain.screening_type == 'pce': 
1055        #   da['entry_mode'] = 'pce' 
1056        #elif brain.screening_type == 'prence': 
1057        #   da['entry_mode'] = 'prence' 
1058        #else: 
1059        #   da['entry_mode'] = '' 
[3776]1060     
[3339]1061
[3681]1062        #da['entry_session'] = entry_session
[3961]1063        da['entry_session'] = es
[2575]1064        da['jamb_lastname'] = brain.lastname
1065        da['jamb_middlename'] = brain.middlenames   # different field names!
[2670]1066        da['jamb_firstname'] = brain.firstname
[2575]1067        da['screening_application_date'] = brain.application_date
1068        da['date_of_birth'] = brain.date_of_birth
1069        da['jamb_first_cos'] = brain.course1
1070        da['jamb_second_cos'] = brain.course2
1071        da['course3'] = brain.course3
1072        da['screening_type'] = brain.screening_type
1073        da['screening_score'] = brain.screening_score
1074        da['screening_date'] = brain.screening_date
1075        da['hq_type'] = brain.hq_type
1076        da['hq_grade'] = brain.hq_grade
1077        da['aos'] = brain.aos
[2576]1078
[2514]1079        application.getContent().edit(mapping=da)
[3349]1080        self.do_timing()
[2575]1081        #self.portal_workflow.doActionFor(application,'close',dest_container=application)
[2514]1082        #
1083        # personal
1084        #
[2540]1085        student_object.invokeFactory('StudentPersonal','personal')
1086        personal = student_object.personal
[2575]1087        #self.portal_workflow.doActionFor(personal,'open',dest_container=personal)
[3350]1088        #self.do_timing()
[2514]1089        dp = {'Title': 'Personal Data'}
1090        dp['sex'] = brain.sex
1091        dp['email'] = brain.email
[2540]1092        dp['phone'] = brain.phone
[2514]1093        dp['lastname'] = brain.lastname
[2575]1094        dp['middlename'] = brain.middlenames   # different field names!
[2576]1095        dp['firstname'] = brain.firstname
[2514]1096        personal.getContent().edit(mapping=dp)
[3349]1097        self.do_timing()
[2514]1098        #
1099        # clearance
1100        #
[2540]1101        student_object.invokeFactory('StudentClearance','clearance')
1102        clearance = student_object.clearance
[2575]1103        #self.portal_workflow.doActionFor(clearance,'open',dest_container=clearance)
[2514]1104        dc = {'Title': 'Clearance/Eligibility Record'}
1105        dc['lga'] = brain.lga
[2670]1106        dc['birthday'] = brain.date_of_birth
[2514]1107        clearance.getContent().edit(mapping=dc)
[3349]1108        self.do_timing()
[2575]1109        #self.portal_workflow.doActionFor(clearance,'close',dest_container=clearance)
[2514]1110        #
1111        # study Course
1112        #
[2540]1113        student_object.invokeFactory('StudentStudyCourse','study_course')
1114        studycourse = student_object.study_course
[2575]1115        #self.portal_workflow.doActionFor(studycourse,'open',dest_container=studycourse)
[3350]1116        #self.do_timing()
[2514]1117        dsc = {}
1118        dsc['study_course'] = brain.course_admitted
1119        dsc['current_verdict'] = ''
[3780]1120        dsc['current_mode'] = da['entry_mode'] # no longer used
[2575]1121        if da['entry_mode'].startswith('de'):
1122            dsc['current_level'] = '200'
[3040]1123        elif da['entry_mode'].startswith('pre'):
1124            dsc['current_level'] = '000'
[2575]1125        else:
1126            dsc['current_level'] = '100'
[3961]1127        dsc['current_session'] = es
[2514]1128        studycourse.getContent().edit(mapping=dsc)
[3349]1129        self.do_timing()
[2514]1130        #
1131        # payments folder
[2540]1132        student_object.invokeFactory('PaymentsFolder','payments')
1133        payments = getattr(student_object,'payments')
[3350]1134        #self.do_timing()
[2514]1135        dpay = {}
1136        dpay['Title'] = 'Payments'
1137        payments.getContent().edit(mapping=dpay)
[2540]1138        self.portal_workflow.doActionFor(payments,'open')
[3349]1139        self.do_timing()
[2580]1140        #
1141        # passport foto
1142        app_picture ="%s/import/images/%s/%s_passport.jpg" % (i_home,
1143                                                              brain.screening_type,
1144                                                              brain.reg_no)
[2675]1145        images_dir = getImagesDir(student_id)
1146        #images_dir = os.path.join("%s" % images_base,student_id)
1147        letter_dir,student_dir = os.path.split(images_dir)
1148        if not os.path.exists(letter_dir):
1149            os.mkdir(letter_dir)
[2580]1150        if not os.path.exists(images_dir):
1151            os.mkdir(images_dir)
1152        image_name = os.path.join(images_dir,"passport_%(student_id)s.jpg" % vars())
1153        if os.path.exists(app_picture):
1154            copy2(app_picture,image_name)
1155        else:
1156            logger.info('passport of %s/%s not found: %s' % (student_id,
1157                                                             brain.reg_no,
1158                                                             app_picture))
[2656]1159
[3349]1160        self.do_timing()
1161        self.print_timing()
1162        if with_timing:
1163            self.timer_step = 0
1164            self._v_timer_count += 1
1165            self._v_total += self.total_time
[2514]1166        return student_id
1167    ###)
1168
[2556]1169    security.declareProtected(ModifyPortalContent,'makeStudentLevel') ###(
[1194]1170    def makeStudentLevel(self,student_id):
1171        "create the StudyLevel for a returning Student"
1172        #import pdb;pdb.set_trace()
[1571]1173        logger = logging.getLogger('WAeUPTool.makeStudentLevel')
[1194]1174        students_folder = self.portal_url.getPortalObject().campus.students
1175        res = self.students_catalog(id=student_id)
1176        if res:
1177            st = res[0]
1178        course = st.course
1179        matric_no = st.matric_no
1180        level = st.level
1181        res = self.results_import(matric_no = matric_no)
1182        if res:
1183            results = res
[1571]1184        logger.info('%s creating Level %s' % (student_id,level))
[1194]1185        #
1186        # Level
1187        #
1188        student_obj = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1189        studycourse = getattr(student_obj,"study_course",None)
1190        self.portal_workflow.doActionFor(studycourse,'close_for_edit',dest_container=studycourse)
1191        l = getattr(studycourse,level,None)
1192        if l is None:
1193            studycourse.invokeFactory('StudentStudyLevel', level)
1194            l = getattr(studycourse, level)
1195            self.portal_workflow.doActionFor(l,'open',dest_container=l)
1196            l.getContent().edit(mapping={'Title': "Level %s" % level})
1197        ###)
[3984]1198       
1199       
1200    security.declareProtected(ModifyPortalContent,'makeStudentLevel') ###(
1201    def exportAllStudyLevels(self,student_id):
1202        "export the StudyLevels for a student"
1203        #import pdb;pdb.set_trace()
1204        logger = logging.getLogger('WAeUPTool.exportAllStudyLevels')
1205        students_folder = self.portal_url.getPortalObject().campus.students
1206        res = self.students_catalog(id=student_id)
1207        if res:
1208            st = res[0]
1209        course = st.course
1210        matric_no = st.matric_no
1211        level = st.level
1212        student_obj = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1213        studycourse = getattr(student_obj,"study_course",None)
[4800]1214        if studycourse:
1215            levels = studycourse.objectIds()
1216        else:
1217            levels = []
[3984]1218        stool = getToolByName(self, 'portal_schemas')
1219        schema = stool._getOb('student_study_level')
1220        fields = ['student_id','level']
1221        fields.extend(schema.keys())
1222        format = '"%(' + ')s","%('.join(fields) + ')s"'
1223        export_file = "%s/export/study_levels_removed.csv" % (i_home)
1224       
[3989]1225        if not os.path.exists(export_file): 
1226            file_handler = open(export_file,"a")
1227            headline = ','.join(fields)
1228            file_handler.write(headline +'\n')
1229        else:
1230            file_handler = open(export_file,"a")
1231       
[3984]1232        for level in levels:
1233            level_object = getattr(studycourse,level)
1234            level_content = level_object.getContent()
1235            d = {'student_id':student_id,'level':level}
1236            for field in schema.keys():
1237                d[field] = getattr(level_content,field,'')   
1238            line = format % d
1239            file_handler.write(line +'\n')               
1240           
1241        ###)       
[1194]1242
[2000]1243    security.declarePublic('getHallInfo') ###(
1244    def getHallInfo(self,bed):
1245        """return Hall Info"""
[828]1246        info = {}
[3406]1247        bedsplit = bed.split('_')
1248        if len(bedsplit) == 4:
1249            hall,block,room,letter = bed.split('_')
1250        else:
1251            info['maintenance_code'] = 'None'
1252            return info
[1846]1253        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="AccoHall",id=hall)
[828]1254        if res and len(res) == 1:
1255            hall_brain = res[0]
1256            hall_doc = hall_brain.getObject().getContent()
1257        else:
1258            return info
1259        info['hall_title'] = hall_brain.Title
1260        info['maintenance_code'] = hall_doc.maintenance_code
[1846]1261        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="ScratchCardBatch")
[828]1262        batch_doc = None
[5168]1263        # Pick the first pin batch which the right prefix. This may cause problems if old maintenance fee batches exist.
1264        # They must be removed first. 
[828]1265        for brain in res:
1266            if brain.id.startswith(info['maintenance_code']):
1267                batch_doc = brain.getObject().getContent()
1268                break
1269        if batch_doc is None:
[2913]1270            info['maintenance_fee'] = ''
[828]1271        else:
1272            info['maintenance_fee'] = batch_doc.cost
1273        return info
[1151]1274    ###)
[828]1275
[2396]1276    security.declareProtected(ModifyPortalContent,'removePictureFolder') ###(
1277    def removePictureFolder(self,student_id):
1278        """remove picture_folder by renaming it"""
[3807]1279        picture_path = getImagesDir(student_id)
1280        dest_path = os.path.join("%s" % images_base,'removed',student_id)
1281        dest_path = dest_path + "_removed"
1282        if os.path.exists(dest_path) or not os.path.exists(picture_path):
[2396]1283            return False
[3807]1284        os.rename(picture_path,dest_path)
[2396]1285        return True
1286    ###)
1287
1288    security.declareProtected(ModifyPortalContent,'restorePictureFolder') ###(
1289    def restorePictureFolder(self,student_id):
1290        """restore picture_folder by renaming it"""
[3807]1291        picture_path = getImagesDir(student_id)
1292        orig_path = os.path.join("%s" % images_base,'removed',student_id)
1293        orig_path = orig_path + "_removed"
1294        if os.path.exists(picture_path) or not os.path.exists(orig_path):
1295            return False       
[4558]1296        os.rename(orig_path,picture_path)
[2396]1297        return True
1298    ###)
1299
[2363]1300    security.declarePublic('picturesExist') ###(
[2420]1301    def picturesExist(self, ids,student_id=None):
[2363]1302        """check if pictures exist in the filesystem"""
[2420]1303        if student_id is None:
1304            student_id = self.getStudentId()
1305        if student_id is None:
1306            return False
[2675]1307        picture_path = getImagesDir(student_id)
1308        #picture_path = os.path.join(images_base,student_id)
[2363]1309        if not os.path.exists(picture_path):
1310            return False
1311        pictures  = [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1312        return set(ids).issubset(set(pictures))
1313    ###)
[3805]1314   
[3812]1315        security.declarePublic('picturePathExists') ###(
1316    def picturePathExists(self, student_id=None):
1317        """check if picture path exists in the filesystem"""
1318        if student_id is None:
1319            student_id = self.getStudentId()
1320        if student_id is None:
1321            return False
1322        picture_path = getImagesDir(student_id)
1323        if os.path.exists(picture_path):
1324            return True
1325        return False
1326    ###)
[3805]1327   
[3812]1328   
[3807]1329    security.declareProtected(ModifyPortalContent,'removeUnusedImageFolders') ###(
1330    def removeUnusedImageFolders(self):
[3806]1331        """check if an unused image folders exists in the filesystem"""
[3807]1332        mtool = self.portal_membership
1333        member = mtool.getAuthenticatedMember()
1334        member_id = str(member)
1335        logger = logging.getLogger('WAeUPTool.removeUnusedImageFolders')
[3806]1336        abc = os.listdir(images_base)
[3805]1337        ifolders = []
[3806]1338        for i in abc:
[3805]1339            picture_path = os.path.join(images_base,i)
1340            ifolders.extend(os.listdir(picture_path))
1341        unused_ids = []
1342        for id in ifolders:
1343            res = self.students_catalog(id=id)
1344            if not res:
1345                unused_ids.append(id)
[3807]1346                #import pdb;pdb.set_trace()   
1347                if not id.endswith('removed'):
1348                    removed = self.waeup_tool.removePictureFolder(id) 
1349                    if removed:
1350                        logger.info('%s: image folder %s successfully removed' % (member_id,id))
1351                    else:
1352                        logger.info('%s: image folder %s could not be removed' % (member_id,id))
1353        return
[3805]1354       
1355    ###)   
1356
[2364]1357    security.declarePublic('picturesList') ###(
1358    def picturesList(self):
1359        """check if pictures exist in the filesystem"""
1360        path = 'images'
1361        student_id = self.getStudentId()
[2677]1362        #picture_path = os.path.join(i_home,path,student_id)
1363        picture_path = getImagesDir(student_id)
[2364]1364        if not os.path.exists(picture_path):
1365            return []
1366        return [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1367    ###)
1368
[2114]1369    security.declarePublic('showFsPicture') ###(
1370    def showFsPicture(self,path):
1371        """return a picture from the filesystem"""
[2675]1372        #picture_path = os.path.join(i_home,path)
1373        picture_path = os.path.join(images_base,path)
[2335]1374        response = self.REQUEST.RESPONSE
[2342]1375        #import pdb;pdb.set_trace()
[2354]1376        registry = getToolByName(self, 'mimetypes_registry')
1377        mimetype = str(registry.lookupExtension(path.lower()) or
1378                    registry.lookupExtension('file.bin'))
[2114]1379        if os.path.exists(picture_path):
[2354]1380            response.setHeader('Content-type',mimetype)
[2114]1381            return open(picture_path).read()
[2335]1382        picture_path = os.path.join(i_home,'import',path)
1383        if os.path.exists(picture_path):
1384            return open(picture_path).read()
[2114]1385    ###)
1386
[1151]1387    security.declareProtected(ModifyPortalContent,'deleteAllCourses') ###(
1388    def deleteAllCourses(self,department="All"):
1389        ''' delete the courses'''
1390        pm = self.portal_membership
1391        member = pm.getAuthenticatedMember()
[1160]1392
[1151]1393        if str(member) not in ("henrik","joachim"):
1394            return "not possible"
1395        if department == "All":
1396            res = self.portal_catalog({'meta_type': 'Department'})
1397        if len(res) < 1:
1398            return "No Departments found"
[1160]1399
[1151]1400        deleted = []
1401        for dep in res:
1402            cf = dep.getObject().courses
1403            if cf:
1404                cf.manage_delObjects(ids=cf.objectIds())
1405                deleted.append("deleted Courses in %s" % dep.getId)
1406        return "\r".join(deleted)
[1160]1407    ###)
[1151]1408
[1572]1409    security.declareProtected(ModifyPortalContent,'getLogfileLines') ###(
1410    def getLogfileLines(self,filename="event.log",numlines=20):
1411        """Get last NUMLINES lines of logfile FILENAME.
[1160]1412
[1572]1413        Return last lines' of a file in the instances logfile directory as
1414        a list. The number of returned lines equals `numlines' or less. If
1415        less than `numlines' lines are available, the whole file ist
1416        returned. If the file can not be opened or some other error
1417        occurs, empty list is returend.
1418        """
1419        result = []
1420        lines_hit = 0
1421
1422        # We only handle files in instances' log directory...
1423        logpath = os.path.join(i_home, "log")
1424        filename = str(os.path.abspath( os.path.join( logpath, filename )))
1425        if not filename.startswith( logpath ):
1426            # Attempt to access file outside log-dir...
1427            return []
1428
1429        try:
1430            fd = file( filename, "rb" )
1431        except IOError:
1432            return []
1433        if not fd:
1434            return []
1435
1436        if os.linesep == None:
1437            linesep = '\n'
1438        else:
1439            linesep = os.linesep
1440
1441        # Try to find 'numlines' times a lineseparator, searching from end
1442        # and moving to the beginning of file...
1443        fd.seek( 0, 2) # Move to end of file...
1444        while lines_hit < numlines:
1445            if fd.read(1) == linesep[-1]: # This moves filedescriptor
1446                                          # one step forward...
1447                lines_hit += 1
1448            try:
1449                fd.seek( -2, 1) # Go two bytes back from current pos...
1450            except IOError:
1451                # We cannot go back two bytes. Maybe the file is too small...
1452                break
1453        fd.seek(2,1)
1454
1455        # Read all lines from current position...
1456        result = fd.readlines()
1457        # Remove line endings...
1458        result = [x.strip() for x in result]
1459        fd.close()
1460        return result
1461    ###)
1462
[1700]1463    security.declareProtected(ModifyPortalContent,"getCallbacksFromLog")###(
[1665]1464    def getCallbacksFromLog(self,filename):
1465        """fix Online Payment Transactions from Z2.log entries"""
1466        import transaction
1467        import random
1468        from cgi import parse_qs
1469        from urlparse import urlparse
1470        #from pdb import set_trace
1471        wftool = self.portal_workflow
1472        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
1473        students_folder = self.portal_url.getPortalObject().campus.students
1474        s = r'(?P<client_ip>\S+) - (?P<member_id>\S+) \['
1475        s += r'(?P<date>.*)\] "(?P<get>.*)" (?P<codes>\d+ \d+) "'
1476        s += r'(?P<intersw>.*)" "(?P<agent>.*)"'
1477        data = re.compile(s)
1478        start = True
1479        tr_count = 1
1480        total = 0
1481        #name = 'pume_results'
1482        #name = 'epaymentsuccessful_z2log2'
1483        name = filename
1484        no_import = []
1485        imported = []
1486        logger = logging.getLogger('WAeUPTool.getFailedTransactions')
1487        try:
1488            transactions = open("%s/import/%s" % (i_home,name),"rb").readlines()
1489        except:
1490            logger.error('Error reading %s' % name)
1491            return
1492        tas = []
1493        for line in transactions:
1494            dict = {}
1495            items = data.search(line)
1496            dict['idict'] = idict = items.groupdict()
1497            #print idict
1498            #from pdb import set_trace;set_trace()
1499            urlparsed = urlparse(idict['get'][4:])
1500            #print urlparsed
1501            path = urlparsed[2].split('/')
1502            dict['student_id'] = student_id = path[8]
1503            dict['payment_id'] = payment_id = path[10]
1504            dict['qs_dict'] = qs_dict = parse_qs(urlparsed[4])
1505            tas.append(dict)
1506            tr_count += 1
1507        return tas
1508    ###)
1509
[1620]1510    security.declareProtected(ModifyPortalContent,"importOnlinePaymentTransactions")###(
1511    def importOnlinePaymentTransactions(self):
1512        """load Online Payment Transactions from CSV values"""
1513        import transaction
1514        import random
1515        #from pdb import set_trace
1516        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
[1625]1517        opt = self.online_payments_import
[1620]1518        students_folder = self.portal_url.getPortalObject().campus.students
1519        start = True
1520        tr_count = 1
1521        total = 0
1522        #name = 'pume_results'
1523        name = 'OnlineTransactions'
1524        no_import = []
1525        imported = []
1526        logger = logging.getLogger('WAeUPTool.importOnlinePaymentTransactions')
1527        try:
1528            transactions = csv.DictReader(open("%s/import/%s.csv" % (i_home,name),"rb"))
1529        except:
1530            logger.error('Error reading %s.csv' % name)
1531            return
1532        for pay_transaction in transactions:
1533            if start:
1534                start = False
1535                logger.info('Start loading from %s.csv' % name)
1536                s = ','.join(['"%s"' % fn for fn in pay_transaction.keys()])
1537                no_import.append('%s,"Error"' % s)
1538                format = ','.join(['"%%(%s)s"' % fn for fn in pay_transaction.keys()])
1539                format_error = format + ',"%(Error)s"'
1540            data = {}
[1644]1541
[1781]1542            # format of the first file sent by Tayo
[1643]1543            #data['datetime'] = date = DateTime.DateTime(pay_transaction['Date'])
1544            #data['student_id'] = student_id = pay_transaction['Payer ID']
1545            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1546            #data['response_code'] = response_code = pay_transaction['Resp Code']
1547            #data['amount'] = amount = pay_transaction['Amount']
[1644]1548
[1781]1549            # format of the second file sent by Tayo
[1798]1550            #data['datetime'] = date = 0
1551            #data['student_id'] = student_id = pay_transaction['Payer ID']
1552            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1553            #data['response_code'] = response_code = '00'
1554            #data['amount'] = amount = pay_transaction['Amount']
[1813]1555
[1798]1556            # format of the third file sent by Kehinde
[1643]1557            data['datetime'] = date = 0
[1798]1558            data['student_id'] = student_id = pay_transaction['customer_id']
1559            data['order_id'] = order_id = pay_transaction['merchant_reference']
[1644]1560            data['response_code'] = response_code = '00'
[1813]1561            data['amount'] = amount = pay_transaction['Amount']
[1644]1562
[1620]1563            dup = False
1564            if response_code == "12":
1565                continue
1566            try:
1567                opt.addRecord(**data)
1568            except ValueError:
1569                dup = True
1570            #from pdb import set_trace;set_trace()
1571            if dup:
1572                if response_code == "00":
[1798]1573                    try:
1574                        opt.modifyRecord(**data)
1575                    except:
1576                        logger.info("duplicate uid, order_id %(order_id)s, student_id %(student_id)s, response_code %(response_code)s" % data)
1577                        continue
[1620]1578                else:
[1674]1579                    pay_transaction['Error'] = "Duplicate order_id"
[1620]1580                    no_import.append( format_error % pay_transaction)
[1674]1581                    logger.info("duplicate order_id %(order_id)s for %(student_id)s %(response_code)s" % data)
[1620]1582                    continue
1583            tr_count += 1
1584            if tr_count > 1000:
1585                if len(no_import) > 0:
1586                    open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1587                             '\n'.join(no_import) + '\n')
1588                    no_import = []
[1645]1589                em = '%d transactions committed\n' % (tr_count)
[1620]1590                transaction.commit()
1591                regs = []
1592                logger.info(em)
1593                total += tr_count
1594                tr_count = 0
1595        open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1596                                                '\n'.join(no_import))
1597        return self.REQUEST.RESPONSE.redirect("%s" % self.REQUEST.get('URL1'))
1598    ###)
[2237]1599
[3172]1600    security.declareProtected(ModifyPortalContent,"importData")###(
1601    def importData(self,filename,name,edit=False,bypass_queue_catalog=False):
1602        """load data from CSV values"""
1603        import transaction
1604        import random
1605        students_folder = self.portal_url.getPortalObject().campus.students
[3292]1606        uploads_folder = self.portal_url.getPortalObject().campus.uploads
[3172]1607        pending_only = False
[3177]1608        pend_str = '--'
[3172]1609        elapse = time.time()
1610        #
1611        # preparations
1612        #
[3177]1613        if filename == pend_str:
[3172]1614            pending_only = True
1615        importer_name = ''.join([part.capitalize() for part in name.split('_')])
[3191]1616        importer = eval("%sImport" % importer_name)(self)
[3172]1617        logger = importer.logger
1618        if importer.init_errors:
[3191]1619            logger.info(importer.init_errors)
1620            return importer.init_errors
[3172]1621        member = importer.member
[3320]1622        #current = importer.current
[3172]1623        import_date = importer.import_date
1624        #
1625        # not_imported
1626        #
1627        info = importer.info
1628        data_keys = importer.data_keys
[3181]1629        csv_keys = importer.csv_keys
[3172]1630        #csv_keys.extend(info.keys())
1631        headline_mapping = dict((k,k) for k in csv_keys)
1632        #
1633        # pending
1634        #
1635        pending_path = importer.pending_path
1636        pending_tmp = importer.pending_tmp
1637        pending_backup = importer.pending_backup
1638        pending_fn = importer.pending_fn
1639        imported_path = importer.imported_path
1640        imported_fn = importer.imported_fn
[3180]1641        commit_after = importer.commit_after
[3172]1642        pending = []
1643        pending_digests = []
1644        #total_added_to_pending = 0
1645        if not pending_only:
1646            pending,pending_digests = importer.makeIdLists()
1647            pending_at_start = len(pending)
1648        datafile = open(pending_tmp,"w")
1649        pending_csv_writer = csv.DictWriter(datafile,
1650                                                    csv_keys,
1651                                                    extrasaction='ignore')
1652        pending_csv_writer.writerow(headline_mapping)
1653        datafile.close()
1654        #
1655        # imported
1656        #
1657        if not os.path.exists(imported_path):
1658            datafile = open(imported_path,"w")
1659            imported_csv_writer = csv.DictWriter(datafile,
1660                                                 csv_keys,
1661                                                 extrasaction='ignore')
1662            imported_csv_writer.writerow(headline_mapping)
1663            datafile.close()
1664        start = True
1665        tr_count = 0
1666        total = 0
1667        total_added_to_imported = 0
1668        total_pending = 0
[3292]1669        import_source_done = ""
[3172]1670        if pending_only:
1671            import_source_path = pending_path
[2193]1672        else:
[3172]1673            import_source_path = "%s/import/%s.csv" % (i_home,filename)
[3292]1674            import_source_done = "%s/import/%s.done" % (i_home,filename)
[3172]1675        if not os.path.exists(import_source_path):
[3195]1676            fn = os.path.split(import_source_path)[1]
[3268]1677            em = 'no import file %(fn)s' % vars()
[3172]1678            return em
1679        import_source_fn = os.path.split(import_source_path)[1]
[3267]1680        if not pending_only:
1681            info['imported_from'] = import_source_fn
[3277]1682        headline = csv.reader(open(import_source_path,"rb")).next()
1683        if "import_mode" not in headline:
[3191]1684            msg = 'import_mode must be in heading'
1685            return msg
[3277]1686        invalid_keys = importer.checkHeadline(headline)
1687        if invalid_keys:
1688            return 'not ignorable key(s): "%s" found in heading' % ", ".join(invalid_keys)
[3201]1689
[3300]1690        import_keys = [k.strip() for k in headline if not (k.strip().startswith('ignore')
1691                                                         or k.strip() in info.keys())]
[3277]1692        # diff2schema = set(import_keys).difference(set(importer.schema.keys()))
1693        # diff2layout = set(import_keys).difference(set(importer.layout.keys()))
1694        # if diff2schema and diff2schema != set(['id',]):
1695        #     msg = 'not ignorable key(s): "%s" found in heading' % ", ".join(diff2schema)
1696        #     return msg
[3172]1697        #
1698        # start importing
1699        #
[2193]1700        try:
[3172]1701            reader = csv.DictReader(open(import_source_path,"rb"))
1702        except:
1703            msg = 'Error reading %s.csv' % filename
1704            logger.error(msg)
1705            return msg
1706        items = [item for item in reader]
1707        total_to_import = len(items)
[3181]1708        tti_float = float(total_to_import)
[3172]1709        if pending_only:
1710            pending_at_start = total_to_import
1711        count = 0
1712        imported = []
[3391]1713        old_commit_count = 0
[3204]1714        error_count = imported_count = 0
[3172]1715        already_in = 0
1716        for record in items:
1717            item = {}
[3198]1718            empty_value_keys = []
[3172]1719            for k,v in record.items():
[3346]1720                if k is None:
1721                    continue
[3172]1722                if v:
[3198]1723                    if v == EMPTY:
1724                        empty_value_keys += k,
1725                        v = ''
[3172]1726                    item[k.strip()] = v.strip()
1727            count += 1
1728            if start:
1729                start = False
1730                adapters = [MappingStorageAdapter(importer.schema, item)]
[3191]1731                logger.info('%(member)s starts import from %(import_source_fn)s' % vars())
[3172]1732            dm = DataModel(item, adapters,context=self)
1733            ds = DataStructure(data=item,datamodel=dm)
1734            error_string = ""
1735            total += 1
[3271]1736            import_mode = item.get('import_mode','')
[3191]1737            import_method = getattr(importer, '%(import_mode)s' % vars(),None )
1738            if import_method is None:
[3271]1739                error_string += "import_mode '%(import_mode)s' is invalid" % vars()
[3251]1740            elif (import_mode in importer.required_modes and
[3248]1741                not set(importer.required_keys[import_mode]).issubset(set(item.keys()))):
1742                diff2import = set(importer.required_keys[import_mode]).difference(set(item.keys()))
1743                error_string += 'required key(s): "%s" not found in record' % ", ".join(diff2import)
[3191]1744            else:
1745                for k in import_keys:
1746                    if k not in item.keys() or k not in importer.validators.keys():
1747                        continue
1748                    if not importer.validators[k](ds,mode=import_mode):
[3201]1749                        error_string += ' ++ '
1750                        error_string += "%s: %s" % (k,self.translation_service(ds.getError(k),
[3172]1751                                                                           ds.getErrorMapping(k)))
1752            if error_string:
1753                error = error_string
1754                id = ''
1755                mapping = item
[2699]1756            else:
[3172]1757                temp_item = item.copy()
1758                temp_item.update(dm)
[3185]1759                results = import_method(temp_item)
[3172]1760                id = results[0]
1761                error = results[1]
1762                mapping = results[2]
[3267]1763            #
1764            # restore some values
1765            #
[3198]1766            for k in empty_value_keys:
1767                mapping[k] = EMPTY
[3186]1768            if mapping.has_key('sex'):
1769                #import pdb;pdb.set_trace()
[3267]1770                if mapping['sex'] == True:
[3186]1771                    mapping['sex'] = 'F'
[3267]1772                elif mapping['sex'] == False:
[3186]1773                    mapping['sex'] = 'M'
[3267]1774            if pending_only:
1775                info['imported_from'] = item['imported_from']
[3181]1776            data_string = ", ".join("%s: %s" % (k,v) for k,v in mapping.items())
[3172]1777            info['error'] = error
[3326]1778            info['import_record_no'] = count + 1
[3172]1779            mapping.update(info)
[3178]1780            log_list = []
[3172]1781            if error:
[3204]1782                error_count += 1
[3172]1783                digest = makeDigest(mapping,data_keys)
1784                if digest not in pending_digests:
1785                    pending_digests += digest,
1786                    pending.append(mapping)
1787                    if not pending_only:
[3251]1788                        log_list += "record from %(import_source_fn)s added to %(pending_fn)s, %(data_string)s, %(error)s" % vars(),
[3172]1789                else:
1790                    already_in += 1
1791                    pass
1792            else:
1793                imported_count += 1
1794                imported += mapping,
[3251]1795                log_list += "record imported and added to %(imported_fn)s from %(import_source_fn)s, %(data_string)s" % vars(),
[3178]1796            if log_list:
1797                time_till_now = time.time() - elapse
[3204]1798                percent_finished = (error_count + imported_count)/tti_float*100
[3181]1799                log_list.insert(0,("%(percent_finished)6.3f %% done in %(time_till_now)3.2fs," % vars()),)
[3178]1800                logger.info(' '.join(log_list))
[3172]1801            finished = count > total_to_import - 1
[3277]1802            must_commit = (imported_count != old_commit_count) and (not imported_count % commit_after)
1803            if must_commit:
1804                old_commit_count = imported_count
[3181]1805
[3172]1806            if must_commit or finished:
1807                if len(imported):
1808                    transaction.commit()
1809                    datafile = open(imported_path,"a")
1810                    writer = csv.DictWriter(datafile,
1811                                            csv_keys,
1812                                            extrasaction='ignore')
1813                    writer.writerows(imported)
1814                    datafile.close()
1815                    total_added_to_imported += len(imported)
1816                    imported = []
1817                if len(pending) > 0:
1818                    datafile = open(pending_tmp,"a")
1819                    writer = csv.DictWriter(datafile,
1820                                            csv_keys,
1821                                            extrasaction='ignore')
1822                    writer.writerows(pending)
1823                    datafile.close()
1824                    total_pending += len(pending)
1825                    #total_added_to_pending += len(pending)
1826                    pending = []
1827                if not finished:
[3335]1828                    msg = '%(commit_after)d records imported and committed of total %(total_added_to_imported)d\n' % vars()
[3172]1829                    logger.info(msg)
1830        elapse = time.time() - elapse
[3271]1831        if os.path.exists(pending_path):
1832            copy2(pending_path,pending_backup)
[3172]1833        copy2(pending_tmp,pending_path)
[3191]1834        msg = "finished importing from %(import_source_fn)s in %(elapse).2f seconds, " % vars()
[3181]1835        msg += "%(count)d records totally read, %(total_added_to_imported)d added to %(imported_fn)s, " % vars()
[3172]1836        if pending_only:
1837            removed_pending = pending_at_start - total_pending
[3184]1838            msg += "%(removed_pending)d removed from %(pending_fn)s" % vars()
[2699]1839        else:
[3181]1840            added_pending = total_pending - pending_at_start
1841            msg += "%(added_pending)d added to %(pending_fn)s, %(already_in)s already in %(pending_fn)s" % vars()
1842        #msg += "%(total_pending)d totally written" % vars()    # this line does not make any sense
[3172]1843        logger.info(msg)
[3292]1844        if import_source_done:
1845            copy(import_source_path,import_source_done)
1846            os.remove(import_source_path)
1847            upload = getattr(uploads_folder,os.path.split(import_source_path)[1],None)
1848            if upload is not None:
1849                upload_doc = upload.getContent()
1850                mapping = {}
[3308]1851                #mapping['import_date'] = DateTime.DateTime()
1852                mapping['import_date'] = import_date
[3327]1853                mapping['imported_by'] = importer.imported_by
[3292]1854                mapping['import_message'] = msg
1855                upload_doc.edit(mapping = mapping)
[3178]1856        os.remove(pending_tmp)
[3172]1857        return msg
[2514]1858    ###)
1859
[2292]1860    security.declareProtected(ModifyPortalContent,"moveImagesToFS")###(
1861    def moveImagesToFS(self,student_id="O738726"):
1862        "move the images to the filesystem"
[2675]1863        images_dir = getImagesDir(student_id)
1864        #images_dir = os.path.join("%s" % images_base,student_id)
[2292]1865        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1866        stool = getToolByName(self, 'portal_schemas')
1867        schemas = ['student_application',
1868                   'student_clearance',
1869                   ]
1870        created = False
1871        for schema_id in schemas:
1872            schema = stool._getOb(schema_id)
1873            object = getattr(student_folder,schema_id[len('student_'):],None)
1874            if object is None:
1875                continue
1876            doc = object.getContent()
1877            for key in schema.keys():
1878                if schema[key].meta_type != "CPS Image Field":
1879                    continue
1880                #import pdb;pdb.set_trace()
[2351]1881                image = getattr(doc,key,None)
1882                if not image or not hasattr(image,"data"):
[2292]1883                    continue
1884                if not created:
1885                    if not os.path.exists(images_dir):
1886                        os.mkdir(images_dir)
1887                    created = True
1888                filename = os.path.join(images_dir,"%(key)s_%(student_id)s.jpg" % vars())
1889                open(filename,"wb").write(str(image.data))
[2500]1890    ###)
[2292]1891
[2300]1892    security.declareProtected(ModifyPortalContent,"movePassportToFS")###(
1893    def movePassportToFS(self,student_id="O738726"):
1894        "move the passports to the filesystem"
1895        images_dir = os.path.join("%s" % i_home,'passports')
1896        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1897        stool = getToolByName(self, 'portal_schemas')
1898        schemas = ['student_application',
1899                   #'student_clearance',
1900                   ]
1901        created = False
1902        for schema_id in schemas:
1903            schema = stool._getOb(schema_id)
1904            object = getattr(student_folder,schema_id[len('student_'):],None)
1905            if object is None:
1906                continue
1907            doc = object.getContent()
1908            for key in schema.keys():
1909                if schema[key].meta_type != "CPS Image Field":
1910                    continue
1911                #import pdb;pdb.set_trace()
1912                image = getattr(doc,key)
1913                if not hasattr(image,"data"):
1914                    continue
1915                if not created:
1916                    if not os.path.exists(images_dir):
1917                        os.mkdir(images_dir)
1918                    created = True
1919                filename = os.path.join(images_dir,"%(student_id)s.jpg" % vars())
1920                open(filename,"wb").write(str(image.data))
[2500]1921    ###)
[3824]1922   
1923   
1924    security.declarePublic('getConfigParams')
1925    def getConfigParams(self,conf_id="configuration"):
1926        conf = getattr(self.portal_url.getPortalObject().campus,conf_id)
1927        conf_obj = conf.getContent()
1928        stool = getToolByName(self, 'portal_schemas')
1929        schema = stool._getOb('configuration')
1930        d = {}
1931        for key in schema.keys():
1932            d[key] = getattr(conf_obj,key,None)
1933        return d
[2300]1934
[3824]1935
[5223]1936    security.declarePublic("changeWorkflowState")
[5183]1937    def changeWorkflowState(self, content, state_id, acquire_permissions=False,
1938                            portal_workflow=None, **kw):
[5223]1939        #Change the workflow state of an object
1940        #@param content: Content obj which state will be changed
1941        #@param state_id: name of the state to put on content
1942        #@param acquire_permissions: True->All permissions unchecked and on riles and
1943        #                            acquired
1944        #                            False->Applies new state security map
1945        #@param portal_workflow: Provide workflow tool (optimisation) if known
1946        #@param kw: change the values of same name of the state mapping
1947        #@return: None
1948       
[5183]1949   
1950        if portal_workflow is None:
1951            portal_workflow = getToolByName(content, 'portal_workflow')
1952   
1953        # Might raise IndexError if no workflow is associated to this type
1954
1955        wf_def = portal_workflow.getWorkflowsFor(content)[0]
1956        wf_id= wf_def.getId()
1957   
1958        wf_state = {
1959            'action': None,
1960            'actor': None,
1961            'comments': "Setting state to %s" % state_id,
1962            'review_state': state_id,
1963            'time': DateTime.DateTime(),
1964            }
1965   
1966        # Updating wf_state from keyword args
1967        for k in kw.keys():
1968            # Remove unknown items
1969            if not wf_state.has_key(k):
1970                del kw[k]
1971        if kw.has_key('review_state'):
1972            del kw['review_state']
1973        wf_state.update(kw)
1974   
1975        portal_workflow.setStatusOf(wf_id, content, wf_state)
1976   
1977        if acquire_permissions:
1978            # Acquire all permissions
1979            for permission in content.possible_permissions():
1980                content.manage_permission(permission, acquire=1)
1981        else:
1982            # Setting new state permissions
1983            wf_def.updateRoleMappingsFor(content)
1984   
1985        # Map changes to the catalogs
1986        content.reindexObject(idxs=['allowedRolesAndUsers', 'review_state'])
1987        return
1988       
1989       
1990       
1991    security.declareProtected(ModifyPortalContent,"changeStudentState")
1992    def changeStudentState(self, content, state_id, acquire_permissions=False,
1993                            portal_workflow=None, **kw):
1994        """Change the student state of an object
1995       
1996        ChangeWorkflowState + indexing students_catalog and check for registration state
1997
1998        """
1999   
2000        if portal_workflow is None:
2001            portal_workflow = getToolByName(content, 'portal_workflow')
2002   
2003        # Might raise IndexError if no workflow is associated to this type
2004
2005        wf_def = portal_workflow.getWorkflowsFor(content)[0]
2006        wf_id= wf_def.getId()
2007        wf = portal_workflow.getWorkflowById(wf_id)
2008
2009        if not wf.states.get(state_id):
2010            return 'reg state does not exist'
2011       
2012        wf_state = {
2013            'action': None,
2014            'actor': None,
2015            'comments': "Setting state to %s" % state_id,
2016            'review_state': state_id,
2017            'time': DateTime.DateTime(),
2018            }
2019   
2020        # Updating wf_state from keyword args
2021        for k in kw.keys():
2022            # Remove unknown items
2023            if not wf_state.has_key(k):
2024                del kw[k]
2025        if kw.has_key('review_state'):
2026            del kw['review_state']
2027        wf_state.update(kw)
2028       
2029        portal_workflow.setStatusOf(wf_id, content, wf_state)
2030   
2031        if acquire_permissions:
2032            # Acquire all permissions
2033            for permission in content.possible_permissions():
2034                content.manage_permission(permission, acquire=1)
2035        else:
2036            # Setting new state permissions
2037            wf_def.updateRoleMappingsFor(content)
2038   
2039        # Map changes to the catalogs
2040        content.reindexObject(idxs=['allowedRolesAndUsers', 'review_state'])
2041        data = {}
2042        #import pdb;pdb.set_trace()
2043        student_id = content.getId()
2044        res = self.students_catalog(id=student_id)
2045        data['id'] = student_id
2046        data['review_state'] = state_id
2047        self.students_catalog.modifyRecord(**data)       
2048       
2049        return       
2050
2051
[5191]2052    security.declarePublic('web_sniffer') ###(
2053    def web_sniffer(self,domain=None,params=None):
2054        "web sniffer based on http://python-dev.blogspot.com/2008/06/python-websniffer.html"
2055   
2056        import httplib
2057        import string
2058        import os
2059       
2060        opera = {"User-Agent":"Opera/9.20 (Windows NT 6.0; U; en)", "Accept-Charset":"ISO-8859-1,UTF-8;q=0.7,*;q=0.7", "Cache-Control":"no", "Accept-Language":"de,en;q=0.7,en-us;q=0.3", "Referer":"http://fake_referer.net"}
2061       
2062        path = '/'
2063        method = 'GET'
2064        profile = opera
2065        #import pdb;pdb.set_trace()
2066        conn = httplib.HTTPSConnection(domain)
2067        conn.request(method, path, params, profile)
2068        stat = conn.getresponse()
2069        headers = stat.getheaders()
2070        version = stat.version
2071        status = stat.status
2072        reason = stat.reason
2073        data = stat.read()
2074        conn.close()
2075       
2076        #file_name = domain.replace(".", "-")
2077        #file_object = open(file_name, "w")
2078        #file_object.write(data)
2079        #file_object.close()
2080       
2081        print "\n"
2082        print "WEBSNIFFER 0.1"
2083        print "+++++++++++++++++"
2084        print "\n"
2085        print "Header (Client)"
2086        print "-----------------"
2087        for key, value in profile.iteritems():
2088            print "%s \t %s" % (key, value)
2089        print "-----------------"
2090        print "\n"
2091        print "Header (Server)"
2092        print "-----------------"
2093        print "HTTP \t %s" % (version)
2094        print "Status \t %s (%s)" % (status, reason)
2095        for key, value in headers:
2096            print "%s \t %s" % (key, value)
2097        print "-----------------"
2098        print "\n"
2099        #print "Quelltext"
2100        #print "-----------------"
2101        #print "Quelltext geschrieben in File:"
2102        #print file_name
2103        #print "-----------------"
2104        #print "\n"
2105
2106
2107
[828]2108InitializeClass(WAeUPTool)
Note: See TracBrowser for help on using the repository browser.