Add test

cms/pdf.py

@@ -1,24 +1,23 @@
 import os
 import tempfile
 
-from django.contrib.staticfiles.finders import find
 from django.conf import settings
+from django.contrib.staticfiles.finders import find
 
 from reportlab.lib import colors
 from reportlab.lib.enums import TA_LEFT, TA_CENTER
 from reportlab.lib.pagesizes import A4, landscape
+from reportlab.lib.styles import ParagraphStyle
 from reportlab.lib.units import cm
-from reportlab.lib.styles import ParagraphStyle as ps
 from reportlab.pdfgen import canvas
-
 from reportlab.platypus import (Frame, FrameBreak, Flowable, NextPageTemplate,
                                 Paragraph, PageTemplate, Preformatted, Spacer,
                                 SimpleDocTemplate, Table, TableStyle
                                 )
 
-style_normal = ps(name='CORPS', fontName='Helvetica', fontSize=9, alignment=TA_LEFT)
+style_normal = ParagraphStyle(name='CORPS', fontName='Helvetica', fontSize=9, alignment=TA_LEFT)
-style_bold = ps(name='CORPS', fontName='Helvetica-Bold', fontSize=10, alignment=TA_LEFT)
+style_bold = ParagraphStyle(name='CORPS', fontName='Helvetica-Bold', fontSize=10, alignment=TA_LEFT)
-style_footer = ps(name='CORPS', fontName='Helvetica', fontSize=7, alignment=TA_CENTER)
+style_footer = ParagraphStyle(name='CORPS', fontName='Helvetica', fontSize=7, alignment=TA_CENTER)
 
 LOGO_EPC = find('img/logo_EPC.png')
 LOGO_ESNE = find('img/logo_ESNE.png')
@@ -42,10 +41,8 @@ class HorLine(Flowable):
 class EpcBaseDocTemplate(SimpleDocTemplate):
     points = '.' * 93
 
-    def __init__(self, filename, section='', title='', orientation=A4):
+    def __init__(self, filename, section='', subject='', orientation=A4):
         path = os.path.join(tempfile.gettempdir(), filename)
-        self.section = section
-        self.title = title
         super().__init__(
             path,
             pagesize=orientation,
@@ -56,9 +53,11 @@ class EpcBaseDocTemplate(SimpleDocTemplate):
             id='first_table', showBoundary=0, leftPadding=0 * cm
         )
         self.story = []
-        #self.title = title
+        self.section = section
+        self.subject = subject
 
     def header(self, canvas, doc):
+        # Logos
         canvas.saveState()
         canvas.drawImage(
             LOGO_EPC, doc.leftMargin, doc.height - 1.5 * cm, 5 * cm, 3 * cm, preserveAspectRatio=True
@@ -67,13 +66,14 @@ class EpcBaseDocTemplate(SimpleDocTemplate):
             LOGO_ESNE, doc.width - 2.5 * cm, doc.height - 1.2 * cm, 5 * cm, 3 * cm, preserveAspectRatio=True
         )
 
+        # Section and subject
         x = doc.leftMargin
         y = doc.height - 2.5 * cm
         canvas.setFont('Helvetica-Bold', 10)
         canvas.line(x, y, x + doc.width, y)
         y -= 0.4 * cm
         canvas.drawString(x, y, self.section)
-        canvas.drawRightString(x + doc.width, y, self.title)
+        canvas.drawRightString(x + doc.width, y, self.subject)
         y -= 0.2 * cm
         canvas.line(x, y, x + doc.width, y)
 
@@ -86,30 +86,29 @@ class EpcBaseDocTemplate(SimpleDocTemplate):
 
     def later_header(self, canvas, doc):
         # Footer
+        canvas.saveState()
         canvas.line(doc.leftMargin, 1 * cm, doc.width + doc.leftMargin, 1 * cm)
         footer = Paragraph(settings.PDF_FOOTER_TEXT, style_footer)
         w, h = footer.wrap(doc.width, doc.bottomMargin)
         footer.drawOn(canvas, doc.leftMargin, h)
         canvas.restoreState()
 
-
+    def formating(self, text, len=25):
-    def formating(self, text):
+        return Preformatted(text, style_normal, maxLineLength=len)
-        return Preformatted(text, style_normal, maxLineLength=25)
 
     def normal_template_page(self):
         first_page_table_frame = Frame(self.leftMargin, self.bottomMargin, self.width + 1 * cm, self.height - 5 * cm,
                                        id='first_table', showBoundary=0, leftPadding=0 * cm)
-        later_pages_table_frame = Frame(self.leftMargin, self.bottomMargin, self.width + 1 * cm, self.height - 2 * cm,
+        later_pages_table_frame = Frame(self.leftMargin, self.bottomMargin, self.width + 1 * cm, self.height - 5 * cm,
                                         id='later_table', showBoundary=0, leftPadding=0 * cm)
         # Page template
         first_page = PageTemplate(id='FirstPage', frames=[first_page_table_frame], onPage=self.header)
-        later_pages = PageTemplate(id='LaterPages', frames=[later_pages_table_frame], onPage=self.later_header)
+        later_pages = PageTemplate(id='LaterPages', frames=[later_pages_table_frame], onPage=self.header)
         self.addPageTemplates([first_page, later_pages])
         self.story = [NextPageTemplate(['*', 'LaterPages'])]
 
     def six_semester_template_page(self):
         frame_title = Frame(self.leftMargin, 24*cm, self.width, 1*cm, showBoundary=0, leftPadding=0)
-        frame_size = (7 * cm, 6 * cm,)
         w, h = (7.5 * cm, 6.5 * cm,)
 
         x = [self.leftMargin, 11 * cm] * 3
@@ -148,58 +147,6 @@ class NumberedCanvas(canvas.Canvas):
         self.setFont("Helvetica", 7)
         self.drawString(self._pagesize[0] / 2, 1 * cm, "Page {0} de {1}".format(self._pageNumber, page_count))
 
-"""
-class EpcBaseDocTemplate(SimpleDocTemplate):
-    def __init__(self, filename, title='', pagesize=A4):
-        super().__init__(filename, pagesize=pagesize, _pageBreakQuick=0, lefMargin=1.5 * cm, bottomMargin=1.5 * cm,
-                         topMargin=1.5 * cm, rightMargin=1.5 * cm)
-        self.story = []
-        self.title = title
-
-    def header(self, canvas, doc):
-
-        canvas.saveState()
-        canvas.drawImage(LOGO_EPC, doc.leftMargin, doc.height - 0.5 * cm, 5 * cm, 3 * cm, preserveAspectRatio=True)
-        canvas.drawImage(LOGO_ESNE, doc.width - 2 * cm, doc.height - 0.5 * cm, 5 * cm, 3 * cm,
-                         preserveAspectRatio=True)
-        canvas.line(doc.leftMargin, doc.height - 0.5 * cm, doc.width + doc.leftMargin, doc.height - 0.5 * cm)
-        canvas.drawString(doc.leftMargin, doc.height - 1.1 * cm, FILIERE)
-        canvas.drawRightString(doc.width + doc.leftMargin, doc.height - 1.1 * cm, self.title)
-        canvas.line(doc.leftMargin, doc.height - 1.3 * cm, doc.width + doc.leftMargin, doc.height - 1.3 * cm)
-        canvas.restoreState()
-
-    def later_header(self, canvas, doc):
-        canvas.saveState()
-        canvas.line(doc.leftMargin, doc.height + 1 * cm, doc.width + doc.leftMargin, doc.height + 1 * cm)
-        canvas.drawString(doc.leftMargin, doc.height + 0.5 * cm, FILIERE)
-        canvas.drawRightString(doc.width + doc.leftMargin, doc.height + 0.5 * cm, self.title)
-        canvas.line(doc.leftMargin, doc.height + 0.2 * cm, doc.width + doc.leftMargin, doc.height + 0.2 * cm)
-        canvas.restoreState()
-
-    def normal_template_page(self):
-        first_page_table_frame = Frame(self.leftMargin, self.bottomMargin, self.width + 1 * cm, self.height - 4 * cm,
-                                       id='first_table', showBoundary=0, leftPadding=0 * cm)
-        later_pages_table_frame = Frame(self.leftMargin, self.bottomMargin, self.width + 1 * cm, self.height - 2 * cm,
-                                        id='later_table', showBoundary=0, leftPadding=0 * cm)
-        # Page template
-        first_page = PageTemplate(id='FirstPage', frames=[first_page_table_frame], onPage=self.header)
-        later_pages = PageTemplate(id='LaterPages', frames=[later_pages_table_frame], onPage=self.later_header)
-        self.addPageTemplates([first_page, later_pages])
-        self.story = [NextPageTemplate(['*', 'LaterPages'])]
-
-    def six_semester_template_page(self):
-        frame_size = (8 * cm, 6.5 * cm,)
-        w, h = frame_size
-
-        x = [self.leftMargin, 12 * cm] * 3
-        y = [17 * cm, 17 * cm, 10 * cm, 10 * cm, 3 * cm, 3 * cm]
-        frames = [Frame(x[f], y[f], width=w, height=h, showBoundary=0, leftPadding=0) for f in range(6)]
-        # Frame for total periods
-        frames.append(Frame(self.leftMargin, self.bottomMargin, self.width, 1.5 * cm, leftPadding=0))
-        # Page template
-        frame_page = PageTemplate(id='FirstPage', frames=frames, onPage=self.header)
-        self.addPageTemplates(frame_page)
-"""
 
 class ModuleDescriptionPdf(EpcBaseDocTemplate):
     """
@@ -219,7 +166,7 @@ class ModuleDescriptionPdf(EpcBaseDocTemplate):
             for sc in c.souscompetence_set.all():
                 str_sous_competence += '- {0} (voir {1})\n'.format(sc.nom, c.code)
 
-        str_res = ' \n'.join(['- {0}'.format(c.nom) for c in module.ressource_set.all()])
+        str_res = ' \n'.join(['- {0}'.format(c.nom) for c in module.ressource_set.all()])  # for future use
 
         str_objectif = ' \n'.join(['- {0}'.format(c.nom) for c in module.objectif_set.all()])
 
@@ -247,14 +194,13 @@ class ModuleDescriptionPdf(EpcBaseDocTemplate):
             formated_data.append(
                 [
                     Preformatted(foo[0], style_normal, maxLineLength=15),
-                    Preformatted(foo[1], style_normal, maxLineLength=97),
+                    Preformatted(foo[1], style_normal, maxLineLength=85),
                 ]
             )
 
         t = Table(data=formated_data, colWidths=[4 * cm, 13 * cm])
         t.hAlign = TA_LEFT
-        t.setStyle(
+        t.setStyle(tblstyle=TableStyle(
-            TableStyle(
                 [
                     ('SIZE', (0, 0), (-1, -1), 7),
                     ('ALIGN', (0, 0), (-1, -1), 'LEFT'),
@@ -281,7 +227,7 @@ class FormationPlanPdf(EpcBaseDocTemplate):
         return Preformatted(el1, style_normal, maxLineLength=length)
 
     def produce(self, domain, process):
-
+        print(domain[0], process[0])
         data = [
             ['Domaines', 'Processus', 'Sem1', 'Sem2', 'Sem3', 'Sem4', 'Sem5', 'Sem6'],
             [self.formating(domain[0]), self.formating(process[0], 60), 'M01', '', '', '', '', ''],
@@ -303,8 +249,7 @@ class FormationPlanPdf(EpcBaseDocTemplate):
             data=data, colWidths=[7 * cm, 9 * cm, 1.5 * cm, 1.5 * cm, 1.5 * cm, 1.5 * cm, 1.5 * cm, 1.5 * cm],
             spaceBefore=0.5 * cm, spaceAfter=1 * cm
         )
-        t.setStyle(
+        t.setStyle(tblstyle=TableStyle(
-            TableStyle(
                 [
                     ('SIZE', (0, 0), (-1, -1), 8),
                     ('FONT', (0, 0), (-1, 0), 'Helvetica-Bold'),

cms/tests.py

@@ -1,3 +1,26 @@
-from django.test import TestCase
 
+from django.conf import settings
+from django.contrib.auth.models import User
+from django.core import mail
+from django.test import TestCase, override_settings
+from django.urls import reverse
 # Create your tests here.
+
+class PdfTestCase(TestCase):
+
+    @classmethod
+    def setUpTestData(cls):
+
+        User.objects.create_superuser('me', 'me@example.org', 'mepassword')
+
+    def setUp(self):
+        self.client.login(username='me', password='mepassword')
+
+    def test_plan_pdf(self):
+        response = self.client.get(reverse('plan-pdf'))
+        self.assertEqual(
+            response['Content-Disposition'],
+            'attachment; filename="plan_formation.pdf"'
+        )
+        self.assertEqual(response['Content-Type'], 'application/pdf')
+        self.assertGreater(len(response.content), 200)