misc: include object url in XML exports (#51899)

tests/admin_pages/test_form.py

@@ -1018,6 +1018,7 @@ def test_form_export(pub):
 
     resp = resp.click(href='export')
     xml_export = resp.text
+    assert ET.fromstring(xml_export).attrib['url'] == 'http://example.net/backoffice/forms/1/'
 
     fd = io.StringIO(xml_export)
     formdef2 = FormDef.import_from_xml(fd)

tests/admin_pages/test_workflow.py

@@ -350,7 +350,7 @@ def test_workflows_export_import(pub):
     app = login(get_app(pub))
     resp = app.get('/backoffice/workflows/1/')
     resp = resp.click('Export')
-    assert resp.content_type == 'application/x-wcs-form'
+    assert resp.content_type == 'application/x-wcs-workflow'
     wf_export = resp.body
 
     resp = app.get('/backoffice/workflows/')
@@ -381,7 +381,7 @@ def test_workflows_import_from_url(pub):
     app = login(get_app(pub))
     resp = app.get('/backoffice/workflows/1/')
     resp = resp.click('Export')
-    assert resp.content_type == 'application/x-wcs-form'
+    assert resp.content_type == 'application/x-wcs-workflow'
     wf_export = resp.body
 
     resp = app.get('/backoffice/workflows/')
@@ -421,7 +421,7 @@ def test_workflows_export_import_create_role(pub):
     app = login(get_app(pub))
     resp = app.get('/backoffice/workflows/1/')
     resp = resp.click('Export')
-    assert resp.content_type == 'application/x-wcs-form'
+    assert resp.content_type == 'application/x-wcs-workflow'
     wf_export = resp.body
 
     resp = app.get('/backoffice/workflows/')

wcs/admin/blocks.py

@@ -14,8 +14,6 @@
 # You should have received a copy of the GNU General Public License
 # along with this program; if not, see <http://www.gnu.org/licenses/>.
 
-import xml.etree.ElementTree as ET
-
 from quixote import get_publisher, get_response, get_session, redirect
 from quixote.directory import Directory
 from quixote.html import TemplateIO, htmltext
@@ -128,12 +126,11 @@ class BlockDirectory(FieldsDirectory):
             return redirect('..')
 
     def export(self):
-        x = self.objectdef.export_to_xml(include_id=True)
-        misc.indent_xml(x)
-        response = get_response()
-        response.set_content_type('application/x-wcs-form')
-        response.set_header('content-disposition', 'attachment; filename=block-%s.wcs' % self.objectdef.slug)
-        return '<?xml version="1.0"?>\n' + ET.tostring(x).decode('utf-8')
+        return misc.xml_response(
+            self.objectdef,
+            filename='block-%s.wcs' % self.objectdef.slug,
+            content_type='application/x-wcs-form',
+        )
 
     def settings(self):
         get_response().breadcrumb.append(('settings', _('Settings')))

wcs/admin/data_sources.py

@@ -14,8 +14,6 @@
 # You should have received a copy of the GNU General Public License
 # along with this program; if not, see <http://www.gnu.org/licenses/>.
 
-import xml.etree.ElementTree as ET
-
 from quixote import get_publisher, redirect
 from quixote.directory import Directory
 from quixote.html import TemplateIO, htmltext
@@ -31,7 +29,7 @@ from wcs.data_sources import (
     has_chrono,
 )
 from wcs.formdef import get_formdefs_of_all_kinds
-from wcs.qommon import _, errors, force_str, misc, template
+from wcs.qommon import _, errors, misc, template
 from wcs.qommon.backoffice.menu import html_top
 from wcs.qommon.form import (
     CheckboxWidget,
@@ -371,14 +369,11 @@ class NamedDataSourcePage(Directory):
             return redirect('..')
 
     def export(self):
-        x = self.datasource.export_to_xml(include_id=True)
-        misc.indent_xml(x)
-        response = get_response()
-        response.set_content_type('application/x-wcs-datasource')
-        response.set_header(
-            'content-disposition', 'attachment; filename=datasource-%s.wcs' % self.datasource.slug
+        return misc.xml_response(
+            self.datasource,
+            filename='datasource-%s.wcs' % self.datasource.slug,
+            content_type='application/x-wcs-datasource',
         )
-        return '<?xml version="1.0"?>\n' + force_str(ET.tostring(x))
 
 
 class NamedDataSourcesDirectory(Directory):

wcs/admin/forms.py

@@ -1466,15 +1466,11 @@ class FormDefPage(Directory):
         return r.getvalue()
 
     def export(self):
-        x = self.formdef.export_to_xml(include_id=True)
-        misc.indent_xml(x)
-        response = get_response()
-        response.set_content_type('application/x-wcs-form')
-        response.set_header(
-            'content-disposition',
-            'attachment; filename=%s-%s.wcs' % (self.formdef_export_prefix, self.formdef.url_name),
+        return misc.xml_response(
+            self.formdef,
+            filename='%s-%s.wcs' % (self.formdef_export_prefix, self.formdef.url_name),
+            content_type='application/x-wcs-form',
         )
-        return '<?xml version="1.0"?>\n' + force_str(ET.tostring(x))
 
     def archive(self):
         if get_publisher().is_using_postgresql():

wcs/admin/mail_templates.py

@@ -14,14 +14,12 @@
 # You should have received a copy of the GNU General Public License
 # along with this program; if not, see <http://www.gnu.org/licenses/>.
 
-import xml.etree.ElementTree as ET
-
 from quixote import get_publisher, get_response, redirect
 from quixote.directory import Directory
 from quixote.html import TemplateIO, htmltext
 
 from wcs.mail_templates import MailTemplate
-from wcs.qommon import _, errors, force_str, misc, template
+from wcs.qommon import _, errors, misc, template
 from wcs.qommon.backoffice.menu import html_top
 from wcs.qommon.form import (
     ComputedExpressionWidget,
@@ -279,11 +277,8 @@ class MailTemplatePage(Directory):
             return redirect('..')
 
     def export(self):
-        x = self.mail_template.export_to_xml(include_id=True)
-        misc.indent_xml(x)
-        response = get_response()
-        response.set_content_type('application/x-wcs-mail-template')
-        response.set_header(
-            'content-disposition', 'attachment; filename=mail-template-%s.wcs' % self.mail_template.slug
+        return misc.xml_response(
+            self.mail_template,
+            filename='mail-template-%s.wcs' % self.mail_template.slug,
+            content_type='application/x-wcs-mail-template',
         )
-        return '<?xml version="1.0"?>\n' + force_str(ET.tostring(x))

wcs/admin/workflows.py

@@ -1696,14 +1696,11 @@ class WorkflowPage(Directory):
         )
 
     def export(self):
-        x = self.workflow.export_to_xml(include_id=True)
-        misc.indent_xml(x)
-        response = get_response()
-        response.set_content_type('application/x-wcs-form')
-        response.set_header(
-            'content-disposition', 'attachment; filename=workflow-%s.wcs' % misc.simplify(self.workflow.name)
+        return misc.xml_response(
+            self.workflow,
+            filename='workflow-%s.wcs' % misc.simplify(self.workflow.name),
+            content_type='application/x-wcs-workflow',
         )
-        return '<?xml version="1.0"?>\n' + force_str(ET.tostring(x))
 
     def get_new_status_form(self):
         r = TemplateIO(html=True)

wcs/admin/wscalls.py

@@ -14,15 +14,13 @@
 # You should have received a copy of the GNU General Public License
 # along with this program; if not, see <http://www.gnu.org/licenses/>.
 
-import xml.etree.ElementTree as ET
-
 from quixote import get_publisher, get_response, get_session, redirect
 from quixote.directory import Directory
 from quixote.html import TemplateIO, htmltext
 
 from wcs.admin import utils
 from wcs.backoffice.snapshots import SnapshotsDirectory
-from wcs.qommon import _, errors, force_str, misc, template
+from wcs.qommon import _, errors, misc, template
 from wcs.qommon.backoffice.menu import html_top
 from wcs.qommon.form import CheckboxWidget, FileWidget, Form, HtmlWidget, StringWidget, TextWidget
 from wcs.wscalls import NamedWsCall, WsCallRequestWidget
@@ -177,12 +175,9 @@ class NamedWsCallPage(Directory):
             return redirect('..')
 
     def export(self):
-        x = self.wscall.export_to_xml(include_id=True)
-        misc.indent_xml(x)
-        response = get_response()
-        response.set_content_type('application/x-wcs-wscall')
-        response.set_header('content-disposition', 'attachment; filename=wscall-%s.wcs' % self.wscall.slug)
-        return '<?xml version="1.0"?>\n' + force_str(ET.tostring(x))
+        return misc.xml_response(
+            self.wscall, filename='wscall-%s.wcs' % self.wscall.slug, content_type='application/x-wcs-wscall'
+        )
 
 
 class NamedWsCallsDirectory(Directory):

wcs/qommon/misc.py

@@ -28,6 +28,7 @@ import subprocess
 import time
 import unicodedata
 import urllib.parse
+import xml.etree.ElementTree as ET
 
 import requests
 from requests.adapters import HTTPAdapter
@@ -1011,3 +1012,14 @@ def get_document_type_value_options(current_document_type):
     options = [({}, '---', '')]
     options += [(doc_type, doc_type['label'], key) for key, doc_type in document_types.items()]
     return options
+
+
+def xml_response(obj, filename, content_type='text/xml'):
+    etree = obj.export_to_xml(include_id=True)
+    if hasattr(obj, 'get_admin_url'):
+        etree.attrib['url'] = obj.get_admin_url()
+    indent_xml(etree)
+    response = get_response()
+    response.set_content_type(content_type)
+    response.set_header('content-disposition', 'attachment; filename=%s' % filename)
+    return '<?xml version="1.0"?>\n' + ET.tostring(etree).decode('utf-8')