source: main/waeup.kofa/trunk/src/waeup/kofa/students/utils.py @ 9195

Last change on this file since 9195 was 9191, checked in by Henrik Bettermann, 12 years ago

Export pdf admission letter.
  • Property svn:keywords set to Id
File size: 21.5 KB
Line 
1## $Id: utils.py 9191 2012-09-16 19:23:06Z henrik $
2##
3## Copyright (C) 2011 Uli Fouquet & Henrik Bettermann
4## This program is free software; you can redistribute it and/or modify
5## it under the terms of the GNU General Public License as published by
6## the Free Software Foundation; either version 2 of the License, or
7## (at your option) any later version.
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 02111-1307 USA
17##
18"""General helper functions and utilities for the student section.
19"""
20import grok
21from time import time
22from reportlab.lib import colors
23from reportlab.lib.units import cm
24from reportlab.lib.pagesizes import A4
25from reportlab.lib.styles import getSampleStyleSheet
26from reportlab.platypus import Paragraph, Image, Table, Spacer
27from zope.component import getUtility, createObject
28from zope.formlib.form import setUpEditWidgets
29from zope.i18n import translate
30from waeup.kofa.interfaces import (
31    IExtFileStore, IKofaUtils, RETURNING, PAID, CLEARED)
32from waeup.kofa.interfaces import MessageFactory as _
33from waeup.kofa.students.interfaces import IStudentsUtils
34
35SLIP_STYLE = [
36    ('VALIGN',(0,0),(-1,-1),'TOP'),
37    #('FONT', (0,0), (-1,-1), 'Helvetica', 11),
38    ]
39
40CONTENT_STYLE = [
41    ('VALIGN',(0,0),(-1,-1),'TOP'),
42    #('FONT', (0,0), (-1,-1), 'Helvetica', 8),
43    #('TEXTCOLOR',(0,0),(-1,0),colors.white),
44    ('BACKGROUND',(0,0),(-1,0),colors.black),
45    ]
46
47FONT_SIZE = 10
48FONT_COLOR = 'black'
49
50def formatted_label(color=FONT_COLOR, size=FONT_SIZE):
51    tag1 ='<font color=%s size=%d>' % (color, size)
52    return tag1 + '%s:</font>'
53
54def trans(text, lang):
55    # shortcut
56    return translate(text, 'waeup.kofa', target_language=lang)
57
58def formatted_text(text, color=FONT_COLOR, size=FONT_SIZE):
59    """Turn `text`, `color` and `size` into an HTML snippet.
60
61    The snippet is suitable for use with reportlab and generating PDFs.
62    Wraps the `text` into a ``<font>`` tag with passed attributes.
63
64    Also non-strings are converted. Raw strings are expected to be
65    utf-8 encoded (usually the case for widgets etc.).
66
67    Finally, a br tag is added if widgets contain div tags
68    which are not supported by reportlab.
69
70    The returned snippet is unicode type.
71    """
72    try:
73        # In unit tests IKofaUtils has not been registered
74        portal_language = getUtility(IKofaUtils).PORTAL_LANGUAGE
75    except:
76        portal_language = 'en'
77    if not isinstance(text, unicode):
78        if isinstance(text, basestring):
79            text = text.decode('utf-8')
80        else:
81            text = unicode(text)
82    # Mainly for boolean values we need our customized
83    # localisation of the zope domain
84    text = translate(text, 'zope', target_language=portal_language)
85    text = text.replace('</div>', '<br /></div>')
86    tag1 = u'<font color="%s" size="%d">' % (color, size)
87    return tag1 + u'%s</font>' % text
88
89def generate_student_id():
90    students = grok.getSite()['students']
91    new_id = students.unique_student_id
92    return new_id
93
94def set_up_widgets(view, ignore_request=False):
95    view.adapters = {}
96    view.widgets = setUpEditWidgets(
97        view.form_fields, view.prefix, view.context, view.request,
98        adapters=view.adapters, for_display=True,
99        ignore_request=ignore_request
100        )
101
102def render_student_data(studentview):
103    """Render student table for an existing frame.
104    """
105    width, height = A4
106    set_up_widgets(studentview, ignore_request=True)
107    data_left = []
108    data_right = []
109    style = getSampleStyleSheet()
110    img = getUtility(IExtFileStore).getFileByContext(
111        studentview.context, attr='passport.jpg')
112    if img is None:
113        from waeup.kofa.browser import DEFAULT_PASSPORT_IMAGE_PATH
114        img = open(DEFAULT_PASSPORT_IMAGE_PATH, 'rb')
115    doc_img = Image(img.name, width=4*cm, height=4*cm, kind='bound')
116    data_left.append([doc_img])
117    #data.append([Spacer(1, 12)])
118    portal_language = getUtility(IKofaUtils).PORTAL_LANGUAGE
119
120    f_label = formatted_label(size=12) % _('Name')
121    f_label = Paragraph(f_label, style["Normal"])
122    f_text = formatted_text(studentview.context.display_fullname, size=12)
123    f_text = Paragraph(f_text, style["Normal"])
124    data_right.append([f_label,f_text])
125
126    for widget in studentview.widgets:
127        if 'name' in widget.name:
128            continue
129        f_label = formatted_label(size=12) % translate(
130            widget.label.strip(), 'waeup.kofa',
131            target_language=portal_language)
132        f_label = Paragraph(f_label, style["Normal"])
133        f_text = formatted_text(widget(), size=12)
134        f_text = Paragraph(f_text, style["Normal"])
135        data_right.append([f_label,f_text])
136
137    if hasattr(studentview.context, 'certcode'):
138        f_label = formatted_label(size=12) % _('Study Course')
139        f_label = Paragraph(f_label, style["Normal"])
140        f_text = formatted_text(
141            studentview.context['studycourse'].certificate.longtitle(), size=12)
142        f_text = Paragraph(f_text, style["Normal"])
143        data_right.append([f_label,f_text])
144
145        f_label = formatted_label(size=12) % _('Department')
146        f_label = Paragraph(f_label, style["Normal"])
147        f_text = formatted_text(
148            studentview.context[
149            'studycourse'].certificate.__parent__.__parent__.longtitle(),
150            size=12)
151        f_text = Paragraph(f_text, style["Normal"])
152        data_right.append([f_label,f_text])
153
154        f_label = formatted_label(size=12) % _('Faculty')
155        f_label = Paragraph(f_label, style["Normal"])
156        f_text = formatted_text(
157            studentview.context[
158            'studycourse'].certificate.__parent__.__parent__.__parent__.longtitle(),
159            size=12)
160        f_text = Paragraph(f_text, style["Normal"])
161        data_right.append([f_label,f_text])
162
163    table_left = Table(data_left,style=SLIP_STYLE)
164    table_right = Table(data_right,style=SLIP_STYLE, colWidths=[5*cm, 6*cm])
165    table = Table([[table_left, table_right],],style=SLIP_STYLE)
166    return table
167
168def render_table_data(tableheader,tabledata):
169    """Render children table for an existing frame.
170    """
171    data = []
172    #data.append([Spacer(1, 12)])
173    line = []
174    style = getSampleStyleSheet()
175    for element in tableheader:
176        field = formatted_text(element[0], color='white')
177        field = Paragraph(field, style["Normal"])
178        line.append(field)
179    data.append(line)
180    for ticket in tabledata:
181        line = []
182        for element in tableheader:
183              field = formatted_text(getattr(ticket,element[1],u' '))
184              field = Paragraph(field, style["Normal"])
185              line.append(field)
186        data.append(line)
187    table = Table(data,colWidths=[
188        element[2]*cm for element in tableheader], style=CONTENT_STYLE)
189    return table
190
191def get_signature_table(signatures, lang='en'):
192    """Return a reportlab table containing signature fields (with date).
193    """
194    style = getSampleStyleSheet()
195    space_width = 0.4  # width in cm of space between signatures
196    table_width = 16.0 # supposed width of signature table in cms
197    # width of signature cells in cm...
198    sig_col_width = table_width - ((len(signatures) - 1) * space_width)
199    sig_col_width = sig_col_width / len(signatures)
200    data = []
201    col_widths = [] # widths of columns
202
203    sig_style = [
204        ('VALIGN',(0,-1),(-1,-1),'TOP'),
205        ('FONT', (0,0), (-1,-1), 'Helvetica-BoldOblique', 12),
206        ('BOTTOMPADDING', (0,0), (-1,0), 36),
207        ('TOPPADDING', (0,-1), (-1,-1), 0),
208        ]
209    for num, elem in enumerate(signatures):
210        # draw a line above each signature cell (not: empty cells in between)
211        sig_style.append(
212            ('LINEABOVE', (num*2,-1), (num*2, -1), 1, colors.black))
213
214    row = []
215    for signature in signatures:
216        row.append(trans(_('Date:'), lang))
217        row.append('')
218        if len(signatures) > 1:
219            col_widths.extend([sig_col_width*cm, space_width*cm])
220        else:
221            col_widths.extend([sig_col_width/2*cm, sig_col_width/2*cm])
222            row.append('') # empty spaceholder on right
223    data.append(row[:-1])
224    data.extend(([''],)*3) # insert 3 empty rows...
225    row = []
226    for signature in signatures:
227        row.append(Paragraph(trans(signature, lang), style["Normal"]))
228        row.append('')
229    data.append(row[:-1])
230    table = Table(data, style=sig_style, repeatRows=len(data),
231                  colWidths=col_widths)
232    return table
233
234def docs_as_flowables(view, lang='en'):
235    """Create reportlab flowables out of scanned docs.
236    """
237    # XXX: fix circular import problem
238    from waeup.kofa.students.viewlets import FileManager
239    from waeup.kofa.browser import DEFAULT_IMAGE_PATH
240    from waeup.kofa.browser.pdf import NORMAL_STYLE, ENTRY1_STYLE
241    style = getSampleStyleSheet()
242    data = []
243
244    # Collect viewlets
245    fm = FileManager(view.context, view.request, view)
246    fm.update()
247    if fm.viewlets:
248        sc_translation = trans(_('Scanned Documents'), lang)
249        data.append(Paragraph(sc_translation, style["Heading3"]))
250        # Insert list of scanned documents
251        table_data = []
252        for viewlet in fm.viewlets:
253            f_label = Paragraph(trans(viewlet.label, lang), ENTRY1_STYLE)
254            img_path = getattr(getUtility(IExtFileStore).getFileByContext(
255                view.context, attr=viewlet.download_name), 'name', None)
256            f_text = Paragraph(trans(_('(not provided)'),lang), ENTRY1_STYLE)
257            if img_path is None:
258                pass
259            elif not img_path[-4:] in ('.jpg', '.JPG'):
260                # reportlab requires jpg images, I think.
261                f_text = Paragraph('%s (not displayable)' % (
262                    viewlet.title,), ENTRY1_STYLE)
263            else:
264                f_text = Image(img_path, width=2*cm, height=1*cm, kind='bound')
265            table_data.append([f_label, f_text])
266        if table_data:
267            # safety belt; empty tables lead to problems.
268            data.append(Table(table_data, style=SLIP_STYLE))
269    return data
270
271class StudentsUtils(grok.GlobalUtility):
272    """A collection of methods subject to customization.
273    """
274    grok.implements(IStudentsUtils)
275
276    def getReturningData(self, student):
277        """ Define what happens after school fee payment
278        depending on the student's senate verdict.
279
280        In the base configuration current level is always increased
281        by 100 no matter which verdict has been assigned.
282        """
283        new_level = student['studycourse'].current_level + 100
284        new_session = student['studycourse'].current_session + 1
285        return new_session, new_level
286
287    def setReturningData(self, student):
288        """ Define what happens after school fee payment
289        depending on the student's senate verdict.
290
291        This method folllows the same algorithm as getReturningData but
292        it also sets the new values.
293        """
294        new_session, new_level = self.getReturningData(student)
295        student['studycourse'].current_level = new_level
296        student['studycourse'].current_session = new_session
297        verdict = student['studycourse'].current_verdict
298        student['studycourse'].current_verdict = '0'
299        student['studycourse'].previous_verdict = verdict
300        return
301
302    def setPaymentDetails(self, category, student,
303            previous_session, previous_level):
304        """Create Payment object and set the payment data of a student for
305        the payment category specified.
306
307        """
308        p_item = u''
309        amount = 0.0
310        if previous_session:
311            p_session = previous_session
312            p_level = previous_level
313            p_current = False
314        else:
315            p_session = student['studycourse'].current_session
316            p_level = student['studycourse'].current_level
317            p_current = True
318        session = str(p_session)
319        try:
320            academic_session = grok.getSite()['configuration'][session]
321        except KeyError:
322            return _(u'Session configuration object is not available.'), None
323        if category == 'schoolfee':
324            try:
325                certificate = student['studycourse'].certificate
326                p_item = certificate.code
327            except (AttributeError, TypeError):
328                return _('Study course data are incomplete.'), None
329            if previous_session:
330                if previous_level == 100:
331                    amount = getattr(certificate, 'school_fee_1', 0.0)
332                else:
333                    amount = getattr(certificate, 'school_fee_2', 0.0)
334            else:
335                if student.state == CLEARED:
336                    amount = getattr(certificate, 'school_fee_1', 0.0)
337                elif student.state == RETURNING:
338                    # In case of returning school fee payment the payment session
339                    # and level contain the values of the session the student
340                    # has paid for.
341                    p_session, p_level = self.getReturningData(student)
342                    amount = getattr(certificate, 'school_fee_2', 0.0)
343                elif student.is_postgrad and student.state == PAID:
344                    # Returning postgraduate students also pay for the next session
345                    # but their level always remains the same.
346                    p_session += 1
347                    amount = getattr(certificate, 'school_fee_2', 0.0)
348        elif category == 'clearance':
349            try:
350                p_item = student['studycourse'].certificate.code
351            except (AttributeError, TypeError):
352                return _('Study course data are incomplete.'), None
353            amount = academic_session.clearance_fee
354        elif category == 'bed_allocation':
355            p_item = self.getAccommodationDetails(student)['bt']
356            amount = academic_session.booking_fee
357        if amount in (0.0, None):
358            return _('Amount could not be determined.' +
359                     ' Would you like to pay for a previous session?'), None
360        for key in student['payments'].keys():
361            ticket = student['payments'][key]
362            if ticket.p_state == 'paid' and\
363               ticket.p_category == category and \
364               ticket.p_item == p_item and \
365               ticket.p_session == p_session:
366                  return _('This type of payment has already been made.' +
367                           ' Would you like to pay for a previous session?'), None
368        payment = createObject(u'waeup.StudentOnlinePayment')
369        timestamp = ("%d" % int(time()*10000))[1:]
370        payment.p_id = "p%s" % timestamp
371        payment.p_category = category
372        payment.p_item = p_item
373        payment.p_session = p_session
374        payment.p_level = p_level
375        payment.p_current = p_current
376        payment.amount_auth = amount
377        return None, payment
378
379    def getAccommodationDetails(self, student):
380        """Determine the accommodation dates of a student.
381        """
382        d = {}
383        d['error'] = u''
384        hostels = grok.getSite()['hostels']
385        d['booking_session'] = hostels.accommodation_session
386        d['allowed_states'] = hostels.accommodation_states
387        d['startdate'] = hostels.startdate
388        d['enddate'] = hostels.enddate
389        d['expired'] = hostels.expired
390        # Determine bed type
391        studycourse = student['studycourse']
392        certificate = getattr(studycourse,'certificate',None)
393        entry_session = studycourse.entry_session
394        current_level = studycourse.current_level
395        if None in (entry_session, current_level, certificate):
396            return d
397        end_level = certificate.end_level
398        if current_level == 10:
399            bt = 'pr'
400        elif entry_session == grok.getSite()['hostels'].accommodation_session:
401            bt = 'fr'
402        elif current_level >= end_level:
403            bt = 'fi'
404        else:
405            bt = 're'
406        if student.sex == 'f':
407            sex = 'female'
408        else:
409            sex = 'male'
410        special_handling = 'regular'
411        d['bt'] = u'%s_%s_%s' % (special_handling,sex,bt)
412        return d
413
414    def selectBed(self, available_beds):
415        """Select a bed from a list of available beds.
416
417        In the base configuration we select the first bed found,
418        but can also randomize the selection if we like.
419        """
420        return available_beds[0]
421
422    def renderPDFAdmissionLetter(self, view, student=None):
423        """Render pdf admission letter.
424        """
425        # XXX: we have to fix the import problems here.
426        from waeup.kofa.browser.interfaces import IPDFCreator
427        from waeup.kofa.browser.pdf import format_html, NOTE_STYLE
428        if student is None:
429            return
430        style = getSampleStyleSheet()
431        creator = getUtility(IPDFCreator)
432        data = []
433        doc_title = view.label
434        author = '%s (%s)' % (view.request.principal.title,
435                              view.request.principal.id)
436        footer_text = view.label
437        if getattr(student, 'student_id', None) is not None:
438            footer_text = "%s - %s - " % (student.student_id, footer_text)
439
440        # Admission text
441        portal_language = getUtility(IKofaUtils).PORTAL_LANGUAGE
442        inst_name = grok.getSite()['configuration'].name
443        text = trans(_(
444            'This is to inform you that you have been provisionally'
445            ' admitted into ${a} as follows:', mapping = {'a': inst_name}),
446            portal_language)
447        html = format_html(text)
448        data.append(Paragraph(html, NOTE_STYLE))
449        data.append(Spacer(1, 20))
450
451        # Student data
452        data.append(render_student_data(view))
453
454        # Insert history
455        data.append(Spacer(1, 20))
456        datelist = student.history.messages[0].split()[0].split('-')
457        creation_date = u'%s/%s/%s' % (datelist[2], datelist[1], datelist[0])
458        text = trans(_(
459            'Your Kofa student record was created on ${a}.',
460            mapping = {'a': creation_date}),
461            portal_language)
462        html = format_html(text)
463        data.append(Paragraph(html, NOTE_STYLE))
464
465        # Create pdf stream
466        view.response.setHeader(
467            'Content-Type', 'application/pdf')
468        pdf_stream = creator.create_pdf(
469            data, None, doc_title, author=author, footer=footer_text,
470            note=None)
471        return pdf_stream
472
473    def renderPDF(self, view, filename='slip.pdf', student=None,
474                  studentview=None, tableheader=None, tabledata=None,
475                  note=None, signatures=None):
476        """Render pdf slips for various pages.
477        """
478        # XXX: we have to fix the import problems here.
479        from waeup.kofa.browser.interfaces import IPDFCreator
480        from waeup.kofa.browser.pdf import NORMAL_STYLE, ENTRY1_STYLE
481        style = getSampleStyleSheet()
482        creator = getUtility(IPDFCreator)
483        data = []
484        doc_title = view.label
485        author = '%s (%s)' % (view.request.principal.title,
486                              view.request.principal.id)
487        footer_text = view.label
488        if getattr(student, 'student_id', None) is not None:
489            footer_text = "%s - %s - " % (student.student_id, footer_text)
490
491        # Insert history
492        if not filename.startswith('payment'):
493            data.extend(creator.fromStringList(student.history.messages))
494
495        # Insert student data table
496        portal_language = getUtility(IKofaUtils).PORTAL_LANGUAGE
497        if student is not None:
498            bd_translation = trans(_('Base Data'), portal_language)
499            data.append(Paragraph(bd_translation, style["Heading3"]))
500            data.append(render_student_data(studentview))
501
502        # Insert widgets
503        if view.form_fields:
504            data.append(Paragraph(view.title, style["Heading3"]))
505            portal_language = getUtility(IKofaUtils).PORTAL_LANGUAGE
506            separators = getattr(self, 'SEPARATORS_DICT', {})
507            table = creator.getWidgetsTable(
508                view.form_fields, view.context, None, lang=portal_language,
509                separators=separators)
510            data.append(table)
511
512        # Insert scanned docs
513        data.extend(docs_as_flowables(view, portal_language))
514
515        # Insert content table (optionally on second page)
516        if tabledata and tableheader:
517            #data.append(PageBreak())
518            data.append(Spacer(1, 20))
519            data.append(Paragraph(view.content_title, style["Heading3"]))
520            contenttable = render_table_data(tableheader,tabledata)
521            data.append(contenttable)
522
523        # Insert signatures
524        if signatures:
525            data.append(Spacer(1, 20))
526            signaturetable = get_signature_table(signatures)
527            data.append(signaturetable)
528
529        view.response.setHeader(
530            'Content-Type', 'application/pdf')
531        try:
532            pdf_stream = creator.create_pdf(
533                data, None, doc_title, author=author, footer=footer_text,
534                note=note)
535        except IOError:
536            view.flash('Error in image file.')
537            return view.redirect(view.url(view.context))
538        return pdf_stream
539
540    VERDICTS_DICT = {
541        '0': _('(not yet)'),
542        'A': 'Successful student',
543        'B': 'Student with carryover courses',
544        'C': 'Student on probation',
545        }
546
547    SEPARATORS_DICT = {
548        }
549
550    #: A prefix used when generating new student ids. Each student id will
551    #: start with this string. The default is 'K' for ``Kofa``.
552    STUDENT_ID_PREFIX = u'K'
Note: See TracBrowser for help on using the repository browser.