Context navigation

source: WAeUP_SRP/base/WAeUPTool.py @ 3548

Last change on this file since 3548 was 3492, checked in by joachim, 17 years ago
fix traceback for anonymous users, do not show log-message, when student views academic session.
Property svn:keywords set to `Id`
File size: 72.8 KB

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

Note: See TracBrowser for help on using the repository browser.

Download in other formats: