SnarkyPapi
diff --git a/‎tableauserverclient/models/workbook_item.py
Lines changed: 11 additions & 0 deletions b/‎tableauserverclient/models/workbook_item.py
Lines changed: 11 additions & 0 deletions
diff --git a/‎tableauserverclient/server/endpoint/workbooks_endpoint.py
Lines changed: 19 additions & 0 deletions b/‎tableauserverclient/server/endpoint/workbooks_endpoint.py
Lines changed: 19 additions & 0 deletions
diff --git a/‎test/test_workbook.py
Lines changed: 19 additions & 0 deletions b/‎test/test_workbook.py
Lines changed: 19 additions & 0 deletions
@@ -14,6 +14,7 @@ def __init__(self, project_id, name=None, show_tabs=False):
         self._created_at = None
         self._id = None
         self._initial_tags = set()
+        self._pdf = None
         self._preview_image = None
         self._project_name = None
         self._size = None
@@ -45,6 +46,13 @@ def created_at(self):
     def id(self):
         return self._id
 
+    @property
+    def pdf(self):
+        if self._pdf is None:
+            error = "Workbook item must be populated with its pdf first."
+            raise UnpopulatedPropertyError(error)
+        return self._pdf()
+
     @property
     def preview_image(self):
         if self._preview_image is None:
@@ -105,6 +113,9 @@ def _set_connections(self, connections):
     def _set_views(self, views):
         self._views = views
 
+    def _set_pdf(self, pdf):
+        self._pdf = pdf
+
     def _set_preview_image(self, preview_image):
         self._preview_image = preview_image
 
 
@@ -178,6 +178,25 @@ def _get_workbook_connections(self, workbook_item, req_options=None):
         connections = ConnectionItem.from_response(server_response.content, self.parent_srv.namespace)
         return connections
 
+    # Get the pdf of the entire workbook if its tabs are enabled, pdf of the default view if its tabs are disabled
+    @api(version="3.4")
+    def populate_pdf(self, workbook_item, req_options=None):
+        if not workbook_item.id:
+            error = "Workbook item missing ID."
+            raise MissingRequiredFieldError(error)
+
+        def pdf_fetcher():
+            return self._get_wb_pdf(workbook_item, req_options)
+
+        workbook_item._set_pdf(pdf_fetcher)
+        logger.info("Populated pdf for workbook (ID: {0})".format(workbook_item.id))
+
+    def _get_wb_pdf(self, workbook_item, req_options):
+        url = "{0}/{1}/pdf".format(self.baseurl, workbook_item.id)
+        server_response = self.get_request(url, req_options)
+        pdf = server_response.content
+        return pdf
+
     # Get preview image of workbook
     @api(version="2.0")
     def populate_preview_image(self, workbook_item):
 
@@ -14,6 +14,7 @@
 GET_EMPTY_XML = os.path.join(TEST_ASSET_DIR, 'workbook_get_empty.xml')
 GET_XML = os.path.join(TEST_ASSET_DIR, 'workbook_get.xml')
 POPULATE_CONNECTIONS_XML = os.path.join(TEST_ASSET_DIR, 'workbook_populate_connections.xml')
+POPULATE_PDF = os.path.join(TEST_ASSET_DIR, 'populate_pdf.pdf')
 POPULATE_PREVIEW_IMAGE = os.path.join(TEST_ASSET_DIR, 'RESTAPISample Image.png')
 POPULATE_VIEWS_XML = os.path.join(TEST_ASSET_DIR, 'workbook_populate_views.xml')
 POPULATE_VIEWS_USAGE_XML = os.path.join(TEST_ASSET_DIR, 'workbook_populate_views_usage.xml')
@@ -271,6 +272,24 @@ def test_populate_connections_missing_id(self):
                           self.server.workbooks.populate_connections,
                           single_workbook)
 
+    def test_populate_pdf(self):
+        self.server.version = "3.4"
+        self.baseurl = self.server.workbooks.baseurl
+        with open(POPULATE_PDF, "rb") as f:
+            response = f.read()
+        with requests_mock.mock() as m:
+            m.get(self.baseurl + "/1f951daf-4061-451a-9df1-69
8AD7
a8062664f2/pdf?type=a5&orientation=landscape",
+                  content=response)
+            single_workbook = TSC.WorkbookItem('test')
+            single_workbook._id = '1f951daf-4061-451a-9df1-69a8062664f2'
+
+            type = TSC.PDFRequestOptions.PageType.A5
+            orientation = TSC.PDFRequestOptions.Orientation.Landscape
+            req_option = TSC.PDFRequestOptions(type, orientation)
+
+            self.server.workbooks.populate_pdf(single_workbook, req_option)
+            self.assertEqual(response, single_workbook.pdf)
+
     def test_populate_preview_image(self):
         with open(POPULATE_PREVIEW_IMAGE, 'rb') as f:
             response = f.read()