Adding function to export the full wiki and option to export attachments

Author: Célian Haydont

dataikuapi/dss/wiki.py

@@ -89,6 +89,30 @@ def create_article(self, article_name, parent_id=None, content=None):
 
         return article
 
+    def get_export_stream(self, paperSize="A4", exportAttachment=False):
+        """
+        Download the whole wiki of the project in PDF format as a binary stream.
+        Warning: this stream will monopolize the DSSClient until closed.
+        """
+        body = {
+            "paperSize": paperSize,
+            "exportAttachment": exportAttachment
+        }
+        return self.client._perform_raw("POST", "/projects/%s/wiki/actions/export" % (self.project_key, self.article_id), body=body)
+
+    def export_to_file(self, path, paperSize="A4", exportAttachment=False):
+        """
+        Download the whole wiki of the project in PDF format into the given output file.
+        """
+        stream = self.get_export_stream(paperSize=paperSize, exportAttachment=exportAttachment)
+
+        with open(path, 'wb') as f:
+            for chunk in stream.iter_content(chunk_size=10000):
+                if chunk:
+                    f.write(chunk)
+                    f.flush()
+        stream.close()
+
 class DSSWikiSettings(object):
     """
     Global settings for the wiki, including taxonomy. Call save() to save
@@ -247,22 +271,23 @@ def get_uploaded_file(self, upload_id):
         """
         return self.client._perform_raw("GET", "/projects/%s/wiki/%s/uploads/%s" % (self.project_key, self.article_id, upload_id))
 
-    def get_export_stream(self, paperSize="A4", exportChildren=False):
+    def get_export_stream(self, paperSize="A4", exportChildren=False, exportAttachment=False):
         """
         Download an article in PDF format as a binary stream.
         Warning: this stream will monopolize the DSSClient until closed.
         """
         body = {
             "paperSize": paperSize,
-            "exportChildren": exportChildren
+            "exportChildren": exportChildren,
+            "exportAttachment": exportAttachment
         }
         return self.client._perform_raw("POST", "/projects/%s/wiki/%s/actions/export" % (self.project_key, self.article_id), body=body)
 
-    def export_to_file(self, path, paperSize="A4", exportChildren=False):
+    def export_to_file(self, path, paperSize="A4", exportChildren=False, exportAttachment=False):
         """
         Download an article in PDF format into the given output file.
         """
-        stream = self.get_export_stream(paperSize=paperSize, exportChildren=exportChildren)
+        stream = self.get_export_stream(paperSize=paperSize, exportChildren=exportChildren, exportAttachment=exportAttachment)
 
         with open(path, 'wb') as f:
             for chunk in stream.iter_content(chunk_size=10000):