|
| 1 | +""" |
| 2 | +This module is a wrapper around django-easy-pdf so we can reuse code |
| 3 | +""" |
| 4 | +import io |
| 5 | +import os |
| 6 | +from django.conf import settings |
| 7 | +from django.http import HttpResponse |
| 8 | +from django.utils.dateformat import format |
| 9 | + |
| 10 | +from docxtpl import DocxTemplate |
| 11 | +from easy_pdf.rendering import render_to_pdf_response, render_to_pdf |
| 12 | + |
| 13 | +from markupfield_helpers.helpers import render_md |
| 14 | +from django.utils.html import mark_safe |
| 15 | + |
| 16 | + |
| 17 | +def _clean_split(text, separator='\n'): |
| 18 | + return [ |
| 19 | + t.replace('-', '').strip() |
| 20 | + for t in text.split('\n') |
| 21 | + if t.replace('-', '').strip() |
| 22 | + ] |
| 23 | + |
| 24 | + |
| 25 | +def _contract_context(contract, **context): |
| 26 | + start_date = contract.sponsorship.start_date |
| 27 | + context.update({ |
| 28 | + "contract": contract, |
| 29 | + "start_date": start_date, |
| 30 | + "start_day_english_suffix": format(start_date, "S"), |
| 31 | + "sponsor": contract.sponsorship.sponsor, |
| 32 | + "sponsorship": contract.sponsorship, |
| 33 | + "benefits": _clean_split(contract.benefits_list.raw), |
| 34 | + "legal_clauses": _clean_split(contract.legal_clauses.raw), |
| 35 | + "renewal": contract.sponsorship.renewal, |
| 36 | + }) |
| 37 | + previous_effective = contract.sponsorship.previous_effective_date |
| 38 | + context["previous_effective"] = previous_effective if previous_effective else "UNKNOWN" |
| 39 | + context["previous_effective_english_suffix"] = format(previous_effective, "S") if previous_effective else None |
| 40 | + return context |
| 41 | + |
| 42 | + |
| 43 | +def render_contract_to_pdf_response(request, contract, **context): |
| 44 | + template = "sponsors/admin/preview-contract.html" |
| 45 | + context = _contract_context(contract, **context) |
| 46 | + return render_to_pdf_response(request, template, context) |
| 47 | + |
| 48 | + |
| 49 | +def render_contract_to_pdf_file(contract, **context): |
| 50 | + template = "sponsors/admin/preview-contract.html" |
| 51 | + context = _contract_context(contract, **context) |
| 52 | + return render_to_pdf(template, context) |
| 53 | + |
| 54 | + |
| 55 | +def _gen_docx_contract(output, contract, **context): |
| 56 | + context = _contract_context(contract, **context) |
| 57 | + renewal = context["renewal"] |
| 58 | + if renewal: |
| 59 | + template = os.path.join(settings.TEMPLATES_DIR, "sponsors", "admin", "renewal-contract-template.docx") |
| 60 | + else: |
| 61 | + template = os.path.join(settings.TEMPLATES_DIR, "sponsors", "admin", "contract-template.docx") |
| 62 | + doc = DocxTemplate(template) |
| 63 | + doc.render(context) |
| 64 | + doc.save(output) |
| 65 | + return output |
| 66 | + |
| 67 | + |
| 68 | +def render_contract_to_docx_response(request, contract, **context): |
| 69 | + response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.wordprocessingml.document') |
| 70 | + response['Content-Disposition'] = 'attachment; filename=contract.docx' |
| 71 | + return _gen_docx_contract(output=response, contract=contract, **context) |
| 72 | + |
| 73 | + |
| 74 | +def render_contract_to_docx_file(contract, **context): |
| 75 | + fp = io.BytesIO() |
| 76 | + fp = _gen_docx_contract(output=fp, contract=contract, **context) |
| 77 | + fp.seek(0) |
| 78 | + return fp.read() |
0 commit comments