source: WAeUP_SRP/trunk/WAeUPTool.py @ 5529

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

pd_ft (fceokene) students start at level 500

  • Property svn:keywords set to Id
File size: 84.1 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 5518 2010-09-16 06:42:19Z 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)
[5511]968            elif brain.screening_type in ('pume2','pde2'):
[5373]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'
[5518]1125        elif da['entry_mode'].startswith('pde'):
1126            dsc['current_level'] = '500'           
[2575]1127        else:
1128            dsc['current_level'] = '100'
[3961]1129        dsc['current_session'] = es
[2514]1130        studycourse.getContent().edit(mapping=dsc)
[3349]1131        self.do_timing()
[2514]1132        #
1133        # payments folder
[2540]1134        student_object.invokeFactory('PaymentsFolder','payments')
1135        payments = getattr(student_object,'payments')
[3350]1136        #self.do_timing()
[2514]1137        dpay = {}
1138        dpay['Title'] = 'Payments'
1139        payments.getContent().edit(mapping=dpay)
[2540]1140        self.portal_workflow.doActionFor(payments,'open')
[3349]1141        self.do_timing()
[2580]1142        #
1143        # passport foto
1144        app_picture ="%s/import/images/%s/%s_passport.jpg" % (i_home,
1145                                                              brain.screening_type,
1146                                                              brain.reg_no)
[2675]1147        images_dir = getImagesDir(student_id)
1148        #images_dir = os.path.join("%s" % images_base,student_id)
1149        letter_dir,student_dir = os.path.split(images_dir)
1150        if not os.path.exists(letter_dir):
1151            os.mkdir(letter_dir)
[2580]1152        if not os.path.exists(images_dir):
1153            os.mkdir(images_dir)
1154        image_name = os.path.join(images_dir,"passport_%(student_id)s.jpg" % vars())
1155        if os.path.exists(app_picture):
1156            copy2(app_picture,image_name)
1157        else:
1158            logger.info('passport of %s/%s not found: %s' % (student_id,
1159                                                             brain.reg_no,
1160                                                             app_picture))
[2656]1161
[3349]1162        self.do_timing()
1163        self.print_timing()
1164        if with_timing:
1165            self.timer_step = 0
1166            self._v_timer_count += 1
1167            self._v_total += self.total_time
[2514]1168        return student_id
1169    ###)
1170
[2556]1171    security.declareProtected(ModifyPortalContent,'makeStudentLevel') ###(
[1194]1172    def makeStudentLevel(self,student_id):
1173        "create the StudyLevel for a returning Student"
1174        #import pdb;pdb.set_trace()
[1571]1175        logger = logging.getLogger('WAeUPTool.makeStudentLevel')
[1194]1176        students_folder = self.portal_url.getPortalObject().campus.students
1177        res = self.students_catalog(id=student_id)
1178        if res:
1179            st = res[0]
1180        course = st.course
1181        matric_no = st.matric_no
1182        level = st.level
1183        res = self.results_import(matric_no = matric_no)
1184        if res:
1185            results = res
[1571]1186        logger.info('%s creating Level %s' % (student_id,level))
[1194]1187        #
1188        # Level
1189        #
1190        student_obj = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1191        studycourse = getattr(student_obj,"study_course",None)
1192        self.portal_workflow.doActionFor(studycourse,'close_for_edit',dest_container=studycourse)
1193        l = getattr(studycourse,level,None)
1194        if l is None:
1195            studycourse.invokeFactory('StudentStudyLevel', level)
1196            l = getattr(studycourse, level)
1197            self.portal_workflow.doActionFor(l,'open',dest_container=l)
1198            l.getContent().edit(mapping={'Title': "Level %s" % level})
1199        ###)
[3984]1200       
1201       
1202    security.declareProtected(ModifyPortalContent,'makeStudentLevel') ###(
1203    def exportAllStudyLevels(self,student_id):
1204        "export the StudyLevels for a student"
1205        #import pdb;pdb.set_trace()
1206        logger = logging.getLogger('WAeUPTool.exportAllStudyLevels')
1207        students_folder = self.portal_url.getPortalObject().campus.students
1208        res = self.students_catalog(id=student_id)
1209        if res:
1210            st = res[0]
1211        course = st.course
1212        matric_no = st.matric_no
1213        level = st.level
1214        student_obj = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1215        studycourse = getattr(student_obj,"study_course",None)
[4800]1216        if studycourse:
1217            levels = studycourse.objectIds()
1218        else:
1219            levels = []
[3984]1220        stool = getToolByName(self, 'portal_schemas')
1221        schema = stool._getOb('student_study_level')
1222        fields = ['student_id','level']
1223        fields.extend(schema.keys())
1224        format = '"%(' + ')s","%('.join(fields) + ')s"'
1225        export_file = "%s/export/study_levels_removed.csv" % (i_home)
1226       
[3989]1227        if not os.path.exists(export_file): 
1228            file_handler = open(export_file,"a")
1229            headline = ','.join(fields)
1230            file_handler.write(headline +'\n')
1231        else:
1232            file_handler = open(export_file,"a")
1233       
[3984]1234        for level in levels:
1235            level_object = getattr(studycourse,level)
1236            level_content = level_object.getContent()
1237            d = {'student_id':student_id,'level':level}
1238            for field in schema.keys():
1239                d[field] = getattr(level_content,field,'')   
1240            line = format % d
1241            file_handler.write(line +'\n')               
1242           
1243        ###)       
[1194]1244
[2000]1245    security.declarePublic('getHallInfo') ###(
1246    def getHallInfo(self,bed):
1247        """return Hall Info"""
[828]1248        info = {}
[3406]1249        bedsplit = bed.split('_')
1250        if len(bedsplit) == 4:
1251            hall,block,room,letter = bed.split('_')
1252        else:
1253            info['maintenance_code'] = 'None'
1254            return info
[1846]1255        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="AccoHall",id=hall)
[828]1256        if res and len(res) == 1:
1257            hall_brain = res[0]
1258            hall_doc = hall_brain.getObject().getContent()
1259        else:
1260            return info
1261        info['hall_title'] = hall_brain.Title
1262        info['maintenance_code'] = hall_doc.maintenance_code
[1846]1263        res = ZCatalog.searchResults(self.portal_catalog_real,portal_type="ScratchCardBatch")
[828]1264        batch_doc = None
[5168]1265        # Pick the first pin batch which the right prefix. This may cause problems if old maintenance fee batches exist.
1266        # They must be removed first. 
[828]1267        for brain in res:
1268            if brain.id.startswith(info['maintenance_code']):
1269                batch_doc = brain.getObject().getContent()
1270                break
1271        if batch_doc is None:
[2913]1272            info['maintenance_fee'] = ''
[828]1273        else:
1274            info['maintenance_fee'] = batch_doc.cost
1275        return info
[1151]1276    ###)
[828]1277
[2396]1278    security.declareProtected(ModifyPortalContent,'removePictureFolder') ###(
1279    def removePictureFolder(self,student_id):
1280        """remove picture_folder by renaming it"""
[3807]1281        picture_path = getImagesDir(student_id)
1282        dest_path = os.path.join("%s" % images_base,'removed',student_id)
1283        dest_path = dest_path + "_removed"
1284        if os.path.exists(dest_path) or not os.path.exists(picture_path):
[2396]1285            return False
[3807]1286        os.rename(picture_path,dest_path)
[2396]1287        return True
1288    ###)
1289
1290    security.declareProtected(ModifyPortalContent,'restorePictureFolder') ###(
1291    def restorePictureFolder(self,student_id):
1292        """restore picture_folder by renaming it"""
[3807]1293        picture_path = getImagesDir(student_id)
1294        orig_path = os.path.join("%s" % images_base,'removed',student_id)
1295        orig_path = orig_path + "_removed"
1296        if os.path.exists(picture_path) or not os.path.exists(orig_path):
1297            return False       
[4558]1298        os.rename(orig_path,picture_path)
[2396]1299        return True
1300    ###)
1301
[2363]1302    security.declarePublic('picturesExist') ###(
[2420]1303    def picturesExist(self, ids,student_id=None):
[2363]1304        """check if pictures exist in the filesystem"""
[2420]1305        if student_id is None:
1306            student_id = self.getStudentId()
1307        if student_id is None:
1308            return False
[2675]1309        picture_path = getImagesDir(student_id)
1310        #picture_path = os.path.join(images_base,student_id)
[2363]1311        if not os.path.exists(picture_path):
1312            return False
1313        pictures  = [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1314        return set(ids).issubset(set(pictures))
1315    ###)
[3805]1316   
[3812]1317        security.declarePublic('picturePathExists') ###(
1318    def picturePathExists(self, student_id=None):
1319        """check if picture path exists in the filesystem"""
1320        if student_id is None:
1321            student_id = self.getStudentId()
1322        if student_id is None:
1323            return False
1324        picture_path = getImagesDir(student_id)
1325        if os.path.exists(picture_path):
1326            return True
1327        return False
1328    ###)
[3805]1329   
[3812]1330   
[3807]1331    security.declareProtected(ModifyPortalContent,'removeUnusedImageFolders') ###(
1332    def removeUnusedImageFolders(self):
[3806]1333        """check if an unused image folders exists in the filesystem"""
[3807]1334        mtool = self.portal_membership
1335        member = mtool.getAuthenticatedMember()
1336        member_id = str(member)
1337        logger = logging.getLogger('WAeUPTool.removeUnusedImageFolders')
[3806]1338        abc = os.listdir(images_base)
[3805]1339        ifolders = []
[3806]1340        for i in abc:
[3805]1341            picture_path = os.path.join(images_base,i)
1342            ifolders.extend(os.listdir(picture_path))
1343        unused_ids = []
1344        for id in ifolders:
1345            res = self.students_catalog(id=id)
1346            if not res:
1347                unused_ids.append(id)
[3807]1348                #import pdb;pdb.set_trace()   
1349                if not id.endswith('removed'):
1350                    removed = self.waeup_tool.removePictureFolder(id) 
1351                    if removed:
1352                        logger.info('%s: image folder %s successfully removed' % (member_id,id))
1353                    else:
1354                        logger.info('%s: image folder %s could not be removed' % (member_id,id))
1355        return
[3805]1356       
1357    ###)   
1358
[2364]1359    security.declarePublic('picturesList') ###(
1360    def picturesList(self):
1361        """check if pictures exist in the filesystem"""
1362        path = 'images'
1363        student_id = self.getStudentId()
[2677]1364        #picture_path = os.path.join(i_home,path,student_id)
1365        picture_path = getImagesDir(student_id)
[2364]1366        if not os.path.exists(picture_path):
1367            return []
1368        return [picture[:picture.rfind('_')] for picture in os.listdir(picture_path)]
1369    ###)
1370
[2114]1371    security.declarePublic('showFsPicture') ###(
1372    def showFsPicture(self,path):
1373        """return a picture from the filesystem"""
[2675]1374        #picture_path = os.path.join(i_home,path)
1375        picture_path = os.path.join(images_base,path)
[2335]1376        response = self.REQUEST.RESPONSE
[2342]1377        #import pdb;pdb.set_trace()
[2354]1378        registry = getToolByName(self, 'mimetypes_registry')
1379        mimetype = str(registry.lookupExtension(path.lower()) or
1380                    registry.lookupExtension('file.bin'))
[2114]1381        if os.path.exists(picture_path):
[2354]1382            response.setHeader('Content-type',mimetype)
[2114]1383            return open(picture_path).read()
[2335]1384        picture_path = os.path.join(i_home,'import',path)
1385        if os.path.exists(picture_path):
1386            return open(picture_path).read()
[2114]1387    ###)
1388
[1151]1389    security.declareProtected(ModifyPortalContent,'deleteAllCourses') ###(
1390    def deleteAllCourses(self,department="All"):
1391        ''' delete the courses'''
1392        pm = self.portal_membership
1393        member = pm.getAuthenticatedMember()
[1160]1394
[1151]1395        if str(member) not in ("henrik","joachim"):
1396            return "not possible"
1397        if department == "All":
1398            res = self.portal_catalog({'meta_type': 'Department'})
1399        if len(res) < 1:
1400            return "No Departments found"
[1160]1401
[1151]1402        deleted = []
1403        for dep in res:
1404            cf = dep.getObject().courses
1405            if cf:
1406                cf.manage_delObjects(ids=cf.objectIds())
1407                deleted.append("deleted Courses in %s" % dep.getId)
1408        return "\r".join(deleted)
[1160]1409    ###)
[1151]1410
[1572]1411    security.declareProtected(ModifyPortalContent,'getLogfileLines') ###(
1412    def getLogfileLines(self,filename="event.log",numlines=20):
1413        """Get last NUMLINES lines of logfile FILENAME.
[1160]1414
[1572]1415        Return last lines' of a file in the instances logfile directory as
1416        a list. The number of returned lines equals `numlines' or less. If
1417        less than `numlines' lines are available, the whole file ist
1418        returned. If the file can not be opened or some other error
1419        occurs, empty list is returend.
1420        """
1421        result = []
1422        lines_hit = 0
1423
1424        # We only handle files in instances' log directory...
1425        logpath = os.path.join(i_home, "log")
1426        filename = str(os.path.abspath( os.path.join( logpath, filename )))
1427        if not filename.startswith( logpath ):
1428            # Attempt to access file outside log-dir...
1429            return []
1430
1431        try:
1432            fd = file( filename, "rb" )
1433        except IOError:
1434            return []
1435        if not fd:
1436            return []
1437
1438        if os.linesep == None:
1439            linesep = '\n'
1440        else:
1441            linesep = os.linesep
1442
1443        # Try to find 'numlines' times a lineseparator, searching from end
1444        # and moving to the beginning of file...
1445        fd.seek( 0, 2) # Move to end of file...
1446        while lines_hit < numlines:
1447            if fd.read(1) == linesep[-1]: # This moves filedescriptor
1448                                          # one step forward...
1449                lines_hit += 1
1450            try:
1451                fd.seek( -2, 1) # Go two bytes back from current pos...
1452            except IOError:
1453                # We cannot go back two bytes. Maybe the file is too small...
1454                break
1455        fd.seek(2,1)
1456
1457        # Read all lines from current position...
1458        result = fd.readlines()
1459        # Remove line endings...
1460        result = [x.strip() for x in result]
1461        fd.close()
1462        return result
1463    ###)
1464
[1700]1465    security.declareProtected(ModifyPortalContent,"getCallbacksFromLog")###(
[1665]1466    def getCallbacksFromLog(self,filename):
1467        """fix Online Payment Transactions from Z2.log entries"""
1468        import transaction
1469        import random
1470        from cgi import parse_qs
1471        from urlparse import urlparse
1472        #from pdb import set_trace
1473        wftool = self.portal_workflow
1474        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
1475        students_folder = self.portal_url.getPortalObject().campus.students
1476        s = r'(?P<client_ip>\S+) - (?P<member_id>\S+) \['
1477        s += r'(?P<date>.*)\] "(?P<get>.*)" (?P<codes>\d+ \d+) "'
1478        s += r'(?P<intersw>.*)" "(?P<agent>.*)"'
1479        data = re.compile(s)
1480        start = True
1481        tr_count = 1
1482        total = 0
1483        #name = 'pume_results'
1484        #name = 'epaymentsuccessful_z2log2'
1485        name = filename
1486        no_import = []
1487        imported = []
1488        logger = logging.getLogger('WAeUPTool.getFailedTransactions')
1489        try:
1490            transactions = open("%s/import/%s" % (i_home,name),"rb").readlines()
1491        except:
1492            logger.error('Error reading %s' % name)
1493            return
1494        tas = []
1495        for line in transactions:
1496            dict = {}
1497            items = data.search(line)
1498            dict['idict'] = idict = items.groupdict()
1499            #print idict
1500            #from pdb import set_trace;set_trace()
1501            urlparsed = urlparse(idict['get'][4:])
1502            #print urlparsed
1503            path = urlparsed[2].split('/')
1504            dict['student_id'] = student_id = path[8]
1505            dict['payment_id'] = payment_id = path[10]
1506            dict['qs_dict'] = qs_dict = parse_qs(urlparsed[4])
1507            tas.append(dict)
1508            tr_count += 1
1509        return tas
1510    ###)
1511
[1620]1512    security.declareProtected(ModifyPortalContent,"importOnlinePaymentTransactions")###(
1513    def importOnlinePaymentTransactions(self):
1514        """load Online Payment Transactions from CSV values"""
1515        import transaction
1516        import random
1517        #from pdb import set_trace
1518        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
[1625]1519        opt = self.online_payments_import
[1620]1520        students_folder = self.portal_url.getPortalObject().campus.students
1521        start = True
1522        tr_count = 1
1523        total = 0
1524        #name = 'pume_results'
1525        name = 'OnlineTransactions'
1526        no_import = []
1527        imported = []
1528        logger = logging.getLogger('WAeUPTool.importOnlinePaymentTransactions')
1529        try:
1530            transactions = csv.DictReader(open("%s/import/%s.csv" % (i_home,name),"rb"))
1531        except:
1532            logger.error('Error reading %s.csv' % name)
1533            return
1534        for pay_transaction in transactions:
1535            if start:
1536                start = False
1537                logger.info('Start loading from %s.csv' % name)
1538                s = ','.join(['"%s"' % fn for fn in pay_transaction.keys()])
1539                no_import.append('%s,"Error"' % s)
1540                format = ','.join(['"%%(%s)s"' % fn for fn in pay_transaction.keys()])
1541                format_error = format + ',"%(Error)s"'
1542            data = {}
[1644]1543
[1781]1544            # format of the first file sent by Tayo
[1643]1545            #data['datetime'] = date = DateTime.DateTime(pay_transaction['Date'])
1546            #data['student_id'] = student_id = pay_transaction['Payer ID']
1547            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1548            #data['response_code'] = response_code = pay_transaction['Resp Code']
1549            #data['amount'] = amount = pay_transaction['Amount']
[1644]1550
[1781]1551            # format of the second file sent by Tayo
[1798]1552            #data['datetime'] = date = 0
1553            #data['student_id'] = student_id = pay_transaction['Payer ID']
1554            #data['order_id'] = order_id = pay_transaction['Order ID (Tranx Ref)']
1555            #data['response_code'] = response_code = '00'
1556            #data['amount'] = amount = pay_transaction['Amount']
[1813]1557
[1798]1558            # format of the third file sent by Kehinde
[1643]1559            data['datetime'] = date = 0
[1798]1560            data['student_id'] = student_id = pay_transaction['customer_id']
1561            data['order_id'] = order_id = pay_transaction['merchant_reference']
[1644]1562            data['response_code'] = response_code = '00'
[1813]1563            data['amount'] = amount = pay_transaction['Amount']
[1644]1564
[1620]1565            dup = False
1566            if response_code == "12":
1567                continue
1568            try:
1569                opt.addRecord(**data)
1570            except ValueError:
1571                dup = True
1572            #from pdb import set_trace;set_trace()
1573            if dup:
1574                if response_code == "00":
[1798]1575                    try:
1576                        opt.modifyRecord(**data)
1577                    except:
1578                        logger.info("duplicate uid, order_id %(order_id)s, student_id %(student_id)s, response_code %(response_code)s" % data)
1579                        continue
[1620]1580                else:
[1674]1581                    pay_transaction['Error'] = "Duplicate order_id"
[1620]1582                    no_import.append( format_error % pay_transaction)
[1674]1583                    logger.info("duplicate order_id %(order_id)s for %(student_id)s %(response_code)s" % data)
[1620]1584                    continue
1585            tr_count += 1
1586            if tr_count > 1000:
1587                if len(no_import) > 0:
1588                    open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1589                             '\n'.join(no_import) + '\n')
1590                    no_import = []
[1645]1591                em = '%d transactions committed\n' % (tr_count)
[1620]1592                transaction.commit()
1593                regs = []
1594                logger.info(em)
1595                total += tr_count
1596                tr_count = 0
1597        open("%s/import/%s_not_imported%s.csv" % (i_home,name,current),"a").write(
1598                                                '\n'.join(no_import))
1599        return self.REQUEST.RESPONSE.redirect("%s" % self.REQUEST.get('URL1'))
1600    ###)
[2237]1601
[3172]1602    security.declareProtected(ModifyPortalContent,"importData")###(
1603    def importData(self,filename,name,edit=False,bypass_queue_catalog=False):
1604        """load data from CSV values"""
1605        import transaction
1606        import random
1607        students_folder = self.portal_url.getPortalObject().campus.students
[3292]1608        uploads_folder = self.portal_url.getPortalObject().campus.uploads
[3172]1609        pending_only = False
[3177]1610        pend_str = '--'
[3172]1611        elapse = time.time()
1612        #
1613        # preparations
1614        #
[3177]1615        if filename == pend_str:
[3172]1616            pending_only = True
1617        importer_name = ''.join([part.capitalize() for part in name.split('_')])
[3191]1618        importer = eval("%sImport" % importer_name)(self)
[3172]1619        logger = importer.logger
1620        if importer.init_errors:
[3191]1621            logger.info(importer.init_errors)
1622            return importer.init_errors
[3172]1623        member = importer.member
[3320]1624        #current = importer.current
[3172]1625        import_date = importer.import_date
1626        #
1627        # not_imported
1628        #
1629        info = importer.info
1630        data_keys = importer.data_keys
[3181]1631        csv_keys = importer.csv_keys
[3172]1632        #csv_keys.extend(info.keys())
1633        headline_mapping = dict((k,k) for k in csv_keys)
1634        #
1635        # pending
1636        #
1637        pending_path = importer.pending_path
1638        pending_tmp = importer.pending_tmp
1639        pending_backup = importer.pending_backup
1640        pending_fn = importer.pending_fn
1641        imported_path = importer.imported_path
1642        imported_fn = importer.imported_fn
[3180]1643        commit_after = importer.commit_after
[3172]1644        pending = []
1645        pending_digests = []
1646        #total_added_to_pending = 0
1647        if not pending_only:
1648            pending,pending_digests = importer.makeIdLists()
1649            pending_at_start = len(pending)
1650        datafile = open(pending_tmp,"w")
1651        pending_csv_writer = csv.DictWriter(datafile,
1652                                                    csv_keys,
1653                                                    extrasaction='ignore')
1654        pending_csv_writer.writerow(headline_mapping)
1655        datafile.close()
1656        #
1657        # imported
1658        #
1659        if not os.path.exists(imported_path):
1660            datafile = open(imported_path,"w")
1661            imported_csv_writer = csv.DictWriter(datafile,
1662                                                 csv_keys,
1663                                                 extrasaction='ignore')
1664            imported_csv_writer.writerow(headline_mapping)
1665            datafile.close()
1666        start = True
1667        tr_count = 0
1668        total = 0
1669        total_added_to_imported = 0
1670        total_pending = 0
[3292]1671        import_source_done = ""
[3172]1672        if pending_only:
1673            import_source_path = pending_path
[2193]1674        else:
[3172]1675            import_source_path = "%s/import/%s.csv" % (i_home,filename)
[3292]1676            import_source_done = "%s/import/%s.done" % (i_home,filename)
[3172]1677        if not os.path.exists(import_source_path):
[3195]1678            fn = os.path.split(import_source_path)[1]
[3268]1679            em = 'no import file %(fn)s' % vars()
[3172]1680            return em
1681        import_source_fn = os.path.split(import_source_path)[1]
[3267]1682        if not pending_only:
1683            info['imported_from'] = import_source_fn
[3277]1684        headline = csv.reader(open(import_source_path,"rb")).next()
1685        if "import_mode" not in headline:
[3191]1686            msg = 'import_mode must be in heading'
1687            return msg
[3277]1688        invalid_keys = importer.checkHeadline(headline)
1689        if invalid_keys:
1690            return 'not ignorable key(s): "%s" found in heading' % ", ".join(invalid_keys)
[3201]1691
[3300]1692        import_keys = [k.strip() for k in headline if not (k.strip().startswith('ignore')
1693                                                         or k.strip() in info.keys())]
[3277]1694        # diff2schema = set(import_keys).difference(set(importer.schema.keys()))
1695        # diff2layout = set(import_keys).difference(set(importer.layout.keys()))
1696        # if diff2schema and diff2schema != set(['id',]):
1697        #     msg = 'not ignorable key(s): "%s" found in heading' % ", ".join(diff2schema)
1698        #     return msg
[3172]1699        #
1700        # start importing
1701        #
[2193]1702        try:
[3172]1703            reader = csv.DictReader(open(import_source_path,"rb"))
1704        except:
1705            msg = 'Error reading %s.csv' % filename
1706            logger.error(msg)
1707            return msg
1708        items = [item for item in reader]
1709        total_to_import = len(items)
[3181]1710        tti_float = float(total_to_import)
[3172]1711        if pending_only:
1712            pending_at_start = total_to_import
1713        count = 0
1714        imported = []
[3391]1715        old_commit_count = 0
[3204]1716        error_count = imported_count = 0
[3172]1717        already_in = 0
1718        for record in items:
1719            item = {}
[3198]1720            empty_value_keys = []
[3172]1721            for k,v in record.items():
[3346]1722                if k is None:
1723                    continue
[3172]1724                if v:
[3198]1725                    if v == EMPTY:
1726                        empty_value_keys += k,
1727                        v = ''
[3172]1728                    item[k.strip()] = v.strip()
1729            count += 1
1730            if start:
1731                start = False
1732                adapters = [MappingStorageAdapter(importer.schema, item)]
[3191]1733                logger.info('%(member)s starts import from %(import_source_fn)s' % vars())
[3172]1734            dm = DataModel(item, adapters,context=self)
1735            ds = DataStructure(data=item,datamodel=dm)
1736            error_string = ""
1737            total += 1
[3271]1738            import_mode = item.get('import_mode','')
[3191]1739            import_method = getattr(importer, '%(import_mode)s' % vars(),None )
1740            if import_method is None:
[3271]1741                error_string += "import_mode '%(import_mode)s' is invalid" % vars()
[3251]1742            elif (import_mode in importer.required_modes and
[3248]1743                not set(importer.required_keys[import_mode]).issubset(set(item.keys()))):
1744                diff2import = set(importer.required_keys[import_mode]).difference(set(item.keys()))
1745                error_string += 'required key(s): "%s" not found in record' % ", ".join(diff2import)
[3191]1746            else:
1747                for k in import_keys:
1748                    if k not in item.keys() or k not in importer.validators.keys():
1749                        continue
1750                    if not importer.validators[k](ds,mode=import_mode):
[3201]1751                        error_string += ' ++ '
1752                        error_string += "%s: %s" % (k,self.translation_service(ds.getError(k),
[3172]1753                                                                           ds.getErrorMapping(k)))
1754            if error_string:
1755                error = error_string
1756                id = ''
1757                mapping = item
[2699]1758            else:
[3172]1759                temp_item = item.copy()
1760                temp_item.update(dm)
[3185]1761                results = import_method(temp_item)
[3172]1762                id = results[0]
1763                error = results[1]
1764                mapping = results[2]
[3267]1765            #
1766            # restore some values
1767            #
[3198]1768            for k in empty_value_keys:
1769                mapping[k] = EMPTY
[3186]1770            if mapping.has_key('sex'):
1771                #import pdb;pdb.set_trace()
[3267]1772                if mapping['sex'] == True:
[3186]1773                    mapping['sex'] = 'F'
[3267]1774                elif mapping['sex'] == False:
[3186]1775                    mapping['sex'] = 'M'
[3267]1776            if pending_only:
1777                info['imported_from'] = item['imported_from']
[3181]1778            data_string = ", ".join("%s: %s" % (k,v) for k,v in mapping.items())
[3172]1779            info['error'] = error
[3326]1780            info['import_record_no'] = count + 1
[3172]1781            mapping.update(info)
[3178]1782            log_list = []
[3172]1783            if error:
[3204]1784                error_count += 1
[3172]1785                digest = makeDigest(mapping,data_keys)
1786                if digest not in pending_digests:
1787                    pending_digests += digest,
1788                    pending.append(mapping)
1789                    if not pending_only:
[3251]1790                        log_list += "record from %(import_source_fn)s added to %(pending_fn)s, %(data_string)s, %(error)s" % vars(),
[3172]1791                else:
1792                    already_in += 1
1793                    pass
1794            else:
1795                imported_count += 1
1796                imported += mapping,
[3251]1797                log_list += "record imported and added to %(imported_fn)s from %(import_source_fn)s, %(data_string)s" % vars(),
[3178]1798            if log_list:
1799                time_till_now = time.time() - elapse
[3204]1800                percent_finished = (error_count + imported_count)/tti_float*100
[3181]1801                log_list.insert(0,("%(percent_finished)6.3f %% done in %(time_till_now)3.2fs," % vars()),)
[3178]1802                logger.info(' '.join(log_list))
[3172]1803            finished = count > total_to_import - 1
[3277]1804            must_commit = (imported_count != old_commit_count) and (not imported_count % commit_after)
1805            if must_commit:
1806                old_commit_count = imported_count
[3181]1807
[3172]1808            if must_commit or finished:
1809                if len(imported):
1810                    transaction.commit()
1811                    datafile = open(imported_path,"a")
1812                    writer = csv.DictWriter(datafile,
1813                                            csv_keys,
1814                                            extrasaction='ignore')
1815                    writer.writerows(imported)
1816                    datafile.close()
1817                    total_added_to_imported += len(imported)
1818                    imported = []
1819                if len(pending) > 0:
1820                    datafile = open(pending_tmp,"a")
1821                    writer = csv.DictWriter(datafile,
1822                                            csv_keys,
1823                                            extrasaction='ignore')
1824                    writer.writerows(pending)
1825                    datafile.close()
1826                    total_pending += len(pending)
1827                    #total_added_to_pending += len(pending)
1828                    pending = []
1829                if not finished:
[3335]1830                    msg = '%(commit_after)d records imported and committed of total %(total_added_to_imported)d\n' % vars()
[3172]1831                    logger.info(msg)
1832        elapse = time.time() - elapse
[3271]1833        if os.path.exists(pending_path):
1834            copy2(pending_path,pending_backup)
[3172]1835        copy2(pending_tmp,pending_path)
[3191]1836        msg = "finished importing from %(import_source_fn)s in %(elapse).2f seconds, " % vars()
[3181]1837        msg += "%(count)d records totally read, %(total_added_to_imported)d added to %(imported_fn)s, " % vars()
[3172]1838        if pending_only:
1839            removed_pending = pending_at_start - total_pending
[3184]1840            msg += "%(removed_pending)d removed from %(pending_fn)s" % vars()
[2699]1841        else:
[3181]1842            added_pending = total_pending - pending_at_start
1843            msg += "%(added_pending)d added to %(pending_fn)s, %(already_in)s already in %(pending_fn)s" % vars()
1844        #msg += "%(total_pending)d totally written" % vars()    # this line does not make any sense
[3172]1845        logger.info(msg)
[3292]1846        if import_source_done:
1847            copy(import_source_path,import_source_done)
1848            os.remove(import_source_path)
1849            upload = getattr(uploads_folder,os.path.split(import_source_path)[1],None)
1850            if upload is not None:
1851                upload_doc = upload.getContent()
1852                mapping = {}
[3308]1853                #mapping['import_date'] = DateTime.DateTime()
1854                mapping['import_date'] = import_date
[3327]1855                mapping['imported_by'] = importer.imported_by
[3292]1856                mapping['import_message'] = msg
1857                upload_doc.edit(mapping = mapping)
[3178]1858        os.remove(pending_tmp)
[3172]1859        return msg
[2514]1860    ###)
1861
[2292]1862    security.declareProtected(ModifyPortalContent,"moveImagesToFS")###(
1863    def moveImagesToFS(self,student_id="O738726"):
1864        "move the images to the filesystem"
[2675]1865        images_dir = getImagesDir(student_id)
1866        #images_dir = os.path.join("%s" % images_base,student_id)
[2292]1867        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1868        stool = getToolByName(self, 'portal_schemas')
1869        schemas = ['student_application',
1870                   'student_clearance',
1871                   ]
1872        created = False
1873        for schema_id in schemas:
1874            schema = stool._getOb(schema_id)
1875            object = getattr(student_folder,schema_id[len('student_'):],None)
1876            if object is None:
1877                continue
1878            doc = object.getContent()
1879            for key in schema.keys():
1880                if schema[key].meta_type != "CPS Image Field":
1881                    continue
1882                #import pdb;pdb.set_trace()
[2351]1883                image = getattr(doc,key,None)
1884                if not image or not hasattr(image,"data"):
[2292]1885                    continue
1886                if not created:
1887                    if not os.path.exists(images_dir):
1888                        os.mkdir(images_dir)
1889                    created = True
1890                filename = os.path.join(images_dir,"%(key)s_%(student_id)s.jpg" % vars())
1891                open(filename,"wb").write(str(image.data))
[2500]1892    ###)
[2292]1893
[2300]1894    security.declareProtected(ModifyPortalContent,"movePassportToFS")###(
1895    def movePassportToFS(self,student_id="O738726"):
1896        "move the passports to the filesystem"
1897        images_dir = os.path.join("%s" % i_home,'passports')
1898        student_folder = getattr(self.portal_url.getPortalObject().campus.students,student_id)
1899        stool = getToolByName(self, 'portal_schemas')
1900        schemas = ['student_application',
1901                   #'student_clearance',
1902                   ]
1903        created = False
1904        for schema_id in schemas:
1905            schema = stool._getOb(schema_id)
1906            object = getattr(student_folder,schema_id[len('student_'):],None)
1907            if object is None:
1908                continue
1909            doc = object.getContent()
1910            for key in schema.keys():
1911                if schema[key].meta_type != "CPS Image Field":
1912                    continue
1913                #import pdb;pdb.set_trace()
1914                image = getattr(doc,key)
1915                if not hasattr(image,"data"):
1916                    continue
1917                if not created:
1918                    if not os.path.exists(images_dir):
1919                        os.mkdir(images_dir)
1920                    created = True
1921                filename = os.path.join(images_dir,"%(student_id)s.jpg" % vars())
1922                open(filename,"wb").write(str(image.data))
[2500]1923    ###)
[3824]1924   
1925   
1926    security.declarePublic('getConfigParams')
1927    def getConfigParams(self,conf_id="configuration"):
1928        conf = getattr(self.portal_url.getPortalObject().campus,conf_id)
1929        conf_obj = conf.getContent()
1930        stool = getToolByName(self, 'portal_schemas')
1931        schema = stool._getOb('configuration')
1932        d = {}
1933        for key in schema.keys():
1934            d[key] = getattr(conf_obj,key,None)
1935        return d
[2300]1936
[3824]1937
[5223]1938    security.declarePublic("changeWorkflowState")
[5183]1939    def changeWorkflowState(self, content, state_id, acquire_permissions=False,
1940                            portal_workflow=None, **kw):
[5223]1941        #Change the workflow state of an object
1942        #@param content: Content obj which state will be changed
1943        #@param state_id: name of the state to put on content
1944        #@param acquire_permissions: True->All permissions unchecked and on riles and
1945        #                            acquired
1946        #                            False->Applies new state security map
1947        #@param portal_workflow: Provide workflow tool (optimisation) if known
1948        #@param kw: change the values of same name of the state mapping
1949        #@return: None
1950       
[5183]1951   
1952        if portal_workflow is None:
1953            portal_workflow = getToolByName(content, 'portal_workflow')
1954   
1955        # Might raise IndexError if no workflow is associated to this type
1956
1957        wf_def = portal_workflow.getWorkflowsFor(content)[0]
1958        wf_id= wf_def.getId()
1959   
1960        wf_state = {
1961            'action': None,
1962            'actor': None,
1963            'comments': "Setting state to %s" % state_id,
1964            'review_state': state_id,
1965            'time': DateTime.DateTime(),
1966            }
1967   
1968        # Updating wf_state from keyword args
1969        for k in kw.keys():
1970            # Remove unknown items
1971            if not wf_state.has_key(k):
1972                del kw[k]
1973        if kw.has_key('review_state'):
1974            del kw['review_state']
1975        wf_state.update(kw)
1976   
1977        portal_workflow.setStatusOf(wf_id, content, wf_state)
1978   
1979        if acquire_permissions:
1980            # Acquire all permissions
1981            for permission in content.possible_permissions():
1982                content.manage_permission(permission, acquire=1)
1983        else:
1984            # Setting new state permissions
1985            wf_def.updateRoleMappingsFor(content)
1986   
1987        # Map changes to the catalogs
1988        content.reindexObject(idxs=['allowedRolesAndUsers', 'review_state'])
1989        return
1990       
1991       
1992       
1993    security.declareProtected(ModifyPortalContent,"changeStudentState")
1994    def changeStudentState(self, content, state_id, acquire_permissions=False,
1995                            portal_workflow=None, **kw):
1996        """Change the student state of an object
1997       
1998        ChangeWorkflowState + indexing students_catalog and check for registration state
1999
2000        """
2001   
2002        if portal_workflow is None:
2003            portal_workflow = getToolByName(content, 'portal_workflow')
2004   
2005        # Might raise IndexError if no workflow is associated to this type
2006
2007        wf_def = portal_workflow.getWorkflowsFor(content)[0]
2008        wf_id= wf_def.getId()
2009        wf = portal_workflow.getWorkflowById(wf_id)
2010
2011        if not wf.states.get(state_id):
2012            return 'reg state does not exist'
2013       
2014        wf_state = {
2015            'action': None,
2016            'actor': None,
2017            'comments': "Setting state to %s" % state_id,
2018            'review_state': state_id,
2019            'time': DateTime.DateTime(),
2020            }
2021   
2022        # Updating wf_state from keyword args
2023        for k in kw.keys():
2024            # Remove unknown items
2025            if not wf_state.has_key(k):
2026                del kw[k]
2027        if kw.has_key('review_state'):
2028            del kw['review_state']
2029        wf_state.update(kw)
2030       
2031        portal_workflow.setStatusOf(wf_id, content, wf_state)
2032   
2033        if acquire_permissions:
2034            # Acquire all permissions
2035            for permission in content.possible_permissions():
2036                content.manage_permission(permission, acquire=1)
2037        else:
2038            # Setting new state permissions
2039            wf_def.updateRoleMappingsFor(content)
2040   
2041        # Map changes to the catalogs
2042        content.reindexObject(idxs=['allowedRolesAndUsers', 'review_state'])
2043        data = {}
2044        #import pdb;pdb.set_trace()
2045        student_id = content.getId()
2046        res = self.students_catalog(id=student_id)
2047        data['id'] = student_id
2048        data['review_state'] = state_id
2049        self.students_catalog.modifyRecord(**data)       
2050       
2051        return       
2052
2053
[5191]2054    security.declarePublic('web_sniffer') ###(
2055    def web_sniffer(self,domain=None,params=None):
2056        "web sniffer based on http://python-dev.blogspot.com/2008/06/python-websniffer.html"
2057   
2058        import httplib
2059        import string
2060        import os
2061       
2062        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"}
2063       
2064        path = '/'
2065        method = 'GET'
2066        profile = opera
2067        #import pdb;pdb.set_trace()
2068        conn = httplib.HTTPSConnection(domain)
2069        conn.request(method, path, params, profile)
2070        stat = conn.getresponse()
2071        headers = stat.getheaders()
2072        version = stat.version
2073        status = stat.status
2074        reason = stat.reason
2075        data = stat.read()
2076        conn.close()
2077       
2078        #file_name = domain.replace(".", "-")
2079        #file_object = open(file_name, "w")
2080        #file_object.write(data)
2081        #file_object.close()
2082       
2083        print "\n"
2084        print "WEBSNIFFER 0.1"
2085        print "+++++++++++++++++"
2086        print "\n"
2087        print "Header (Client)"
2088        print "-----------------"
2089        for key, value in profile.iteritems():
2090            print "%s \t %s" % (key, value)
2091        print "-----------------"
2092        print "\n"
2093        print "Header (Server)"
2094        print "-----------------"
2095        print "HTTP \t %s" % (version)
2096        print "Status \t %s (%s)" % (status, reason)
2097        for key, value in headers:
2098            print "%s \t %s" % (key, value)
2099        print "-----------------"
2100        print "\n"
2101        #print "Quelltext"
2102        #print "-----------------"
2103        #print "Quelltext geschrieben in File:"
2104        #print file_name
2105        #print "-----------------"
2106        #print "\n"
2107
2108
2109
[828]2110InitializeClass(WAeUPTool)
Note: See TracBrowser for help on using the repository browser.