Context navigation

source: WAeUP_SRP/base/WAeUPTool.py @ 2812

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

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

Download in other formats: