From 7d731ee7f3b41877626c6e838b4c44c83eb284b9 Mon Sep 17 00:00:00 2001
From: Benjamin Dauvergne <bdauvergne@entrouvert.com>
Date: Tue, 18 Dec 2018 15:22:19 +0100
Subject: [PATCH 3/3] manager: add ous export (#29162)

---
 src/authentic2/manager/ou_views.py             | 18 ++++++++++++++++++
 .../templates/authentic2/manager/ous.html      |  1 +
 src/authentic2/manager/urls.py                 |  3 +++
 3 files changed, 22 insertions(+)

diff --git a/src/authentic2/manager/ou_views.py b/src/authentic2/manager/ou_views.py
index 1d6e0107..ed8a001e 100644
--- a/src/authentic2/manager/ou_views.py
+++ b/src/authentic2/manager/ou_views.py
@@ -1,8 +1,12 @@
+import json
+
 from django_rbac.utils import get_ou_model
 from django.http import HttpResponseRedirect
 from django.contrib import messages
 from django.utils.translation import ugettext as _
 
+from authentic2 import data_transfer
+
 from . import tables, views, forms
 
 
@@ -13,6 +17,7 @@ class OrganizationalUnitView(views.BaseTableView):
     search_form_class = forms.NameSearchForm
     permissions = ['a2_rbac.search_organizationalunit']
     title = _('Organizational units')
+    formats = ['json']
 
 listing = OrganizationalUnitView.as_view()
 
@@ -74,3 +79,16 @@ class OrganizationalUnitDeleteView(views.BaseDeleteView):
                                                                   **kwargs)
 
 delete = OrganizationalUnitDeleteView.as_view()
+
+
+class RolesExportView(views.ExportMixin, OrganizationalUnitView):
+    def get(self, request, *args, **kwargs):
+        export = data_transfer.export_site(
+            data_transfer.ExportContext(
+                ou_qs=self.get_table_data(),
+                export_roles=False,
+                export_ous=True))
+        return self.export_response(json.dumps(export), 'application/json', 'json')
+
+
+export = RolesExportView.as_view()
diff --git a/src/authentic2/manager/templates/authentic2/manager/ous.html b/src/authentic2/manager/templates/authentic2/manager/ous.html
index 3d0e2884..e6da200c 100644
--- a/src/authentic2/manager/templates/authentic2/manager/ous.html
+++ b/src/authentic2/manager/templates/authentic2/manager/ous.html
@@ -26,4 +26,5 @@
   {% with row_link=1 %}
       {% render_table table "authentic2/manager/table.html" %}
   {% endwith %}
+  {% include "authentic2/manager/export_include.html" with export_view_name="a2-manager-ou-export" %}
 {% endblock %}
diff --git a/src/authentic2/manager/urls.py b/src/authentic2/manager/urls.py
index 0c1df367..c0646d5d 100644
--- a/src/authentic2/manager/urls.py
+++ b/src/authentic2/manager/urls.py
@@ -105,6 +105,9 @@ urlpatterns = required(
             name='a2-manager-ou-edit'),
         url(r'^organizational-units/(?P<pk>\d+)/delete/$', ou_views.delete,
             name='a2-manager-ou-delete'),
+        url(r'^organizational-units/export/(?P<format>json)/$',
+            ou_views.export,
+            name='a2-manager-ou-export'),
 
         # Services
         url(r'^services/$', service_views.listing,
-- 
2.18.0