Allows to create the root PDF

OneGov · Nov 15, 2018 · 0cef50a · 0cef50a
1 parent c9c04d9
commit 0cef50a
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 3 deletions.
diff --git a/onegov/agency/layouts/agency.py b/onegov/agency/layouts/agency.py
@@ -33,6 +33,11 @@ def breadcrumbs(self):
     def editbar_links(self):
         if self.request.is_manager:
             return [
+                Link(
+                    text=_("Create PDF"),
+                    url=self.request.link(self.model, 'create-pdf'),
+                    attrs={'class': 'create-pdf'}
+                ),
                 LinkGroup(
                     title=_("Add"),
                     links=[

diff --git a/onegov/agency/tests/test_layouts.py b/onegov/agency/tests/test_layouts.py
@@ -67,6 +67,7 @@ def test_agency_collection_layout():
     request.is_manager = True
     layout = AgencyCollectionLayout(model, request)
     assert list(hrefs(layout.editbar_links)) == [
+        'ExtendedAgencyCollection/create-pdf',
         'ExtendedAgencyCollection/new'
     ]
 

diff --git a/onegov/agency/tests/test_views.py b/onegov/agency/tests/test_views.py
@@ -158,11 +158,10 @@ def test_views(client):
     assert pdf.content_type == 'application/pdf'
     assert pdf.content_length
 
-    client.app.root_pdf = BytesIO(pdf.body)
-
     agencies = client.get('/organizations')
+    agencies = agencies.click("PDF erstellen").form.submit().follow()
+
     pdf = agencies.click("Gesamter Staatskalender als PDF")
-    pdf = bund.click("Organisation als PDF speichern")
     assert pdf.content_type == 'application/pdf'
     assert pdf.content_length
 

diff --git a/onegov/agency/views/agencies.py b/onegov/agency/views/agencies.py
@@ -190,6 +190,40 @@ def get_root_pdf(self, request):
     )
 
 
+@AgencyApp.form(
+    model=ExtendedAgencyCollection,
+    name='create-pdf',
+    template='form.pt',
+    permission=Private,
+    form=Form
+)
+def create_root_pdf(self, request, form):
+
+    if form.submitted(request):
+        request.app.root_pdf = DefaultAgencyPdf.from_agencies(
+            agencies=self.roots,
+            author=request.app.org.name,
+            title=request.app.org.name,
+            toc=True,
+            exclude=request.app.org.hidden_people_fields
+        )
+        request.success(_("PDF created"))
+        return redirect(request.link(self))
+
+    layout = AgencyCollectionLayout(self, request)
+    layout.breadcrumbs.append(Link(_("Create PDF"), '#'))
+
+    return {
+        'layout': layout,
+        'title': _("Create PDF"),
+        'helptext': _(
+            "Create a PDF of this agency and all its suborganizations. "
+            "This may take a while."
+        ),
+        'form': form
+    }
+
+
 @AgencyApp.form(
     model=ExtendedAgency,
     name='create-pdf',