add the vizql initial support

Author: heshaojian

samples/vds.py

@@ -0,0 +1,72 @@
+####
+# Getting Started Part Three of Three
+# This script demonstrates all the different types of 'content' a server contains
+#
+# To make it easy to run, it doesn't take any arguments - you need to edit the code with your info
+####
+
+from typing import Dict, Any
+from rich import print
+
+import getpass
+import tableauserverclient as TSC
+
+
+def main():
+    # 1 - replace with your server domain: stop at the slash
+    server_url = ""
+
+    # 2 - optional - change to false **for testing only** if you get a certificate error
+    use_ssl = True
+
+    server = TSC.Server(server_url, use_server_version=True, http_options={"verify": use_ssl})
+    print(f"Connected to {server.server_info.baseurl}")
+
+    # 3 - replace with your site name exactly as it looks in the url
+    # e.g https://my-server/#/site/this-is-your-site-url-name/not-this-part
+    site_url_name = ""  # leave empty if there is no site name in the url (you are on the default site)
+
+    # 4 - replace with your username.
+    # REMEMBER: if you are using Tableau Online, your username is the entire email address
+    username = ""
+    password = "" #getpass.getpass("Enter your password:")  # so you don't save it in this file
+    tableau_auth = TSC.TableauAuth(username, password, site_id=site_url_name)
+
+    # OR instead of username+password, use a Personal Access Token (PAT) (required by Tableau Cloud)
+    # token_name = "your-token-name"
+    # token_value = "your-token-value-long-random-string"
+    # tableau_auth = TSC.PersonalAccessTokenAuth(token_name, token_value, site_id=site_url_name)
+
+    with server.auth.sign_in(tableau_auth):
+        # schema - may be used to understand the schema of the VDS API
+        print(server.vizql.VizQL_Schema)
+        # query
+        query_dict: Dict[str, Any] = {
+            "fields": [{"fieldCaption": "School Code"}],
+            "filters": [
+                {
+                    "field": {"fieldCaption": "Enrollment (K-12)", "function": "SUM"},
+                    "filterType": "QUANTITATIVE_NUMERICAL",
+                    "quantitativeFilterType": "MIN",
+                    "min": 500
+                }
+            ]
+        }
+        datasource_id = "e4d75fd4-af1c-4fb8-a049-058e3fef57bc"
+        
+        # query metadata
+        vds_metadata = server.vizql.query_vds_metadata(
+            datasource_id=datasource_id
+        )
+        if vds_metadata:
+            print(vds_metadata)
+            
+        # query data
+        vds_data = server.vizql.query_vds_data(
+            query=query_dict,
+            datasource_id=datasource_id
+        )
+        if vds_data:
+            print(vds_data)
+if __name__ == "__main__":
+    main()

setup.cfg

@@ -0,0 +1,6 @@
+[versioneer]
+VCS = git
+style = pep440-pre
+versionfile_source = tableauserverclient/bin/_version.py
+versionfile_build = tableauserverclient/bin/_version.py
+tag_prefix = v 

tableauserverclient/server/__init__.py

@@ -45,6 +45,7 @@
     Views,
     Webhooks,
     Workbooks,
+    VizQL,
 )
 
 __all__ = [
@@ -91,4 +92,5 @@
     "Views",
     "Webhooks",
     "Workbooks",
+    "VizQL",
 ]

tableauserverclient/server/endpoint/__init__.py

@@ -30,7 +30,7 @@
 from tableauserverclient.server.endpoint.virtual_connections_endpoint import VirtualConnections
 from tableauserverclient.server.endpoint.webhooks_endpoint import Webhooks
 from tableauserverclient.server.endpoint.workbooks_endpoint import Workbooks
-
+from tableauserverclient.server.endpoint.vizql_endpoint import VizQL
 __all__ = [
     "Auth",
     "CustomViews",
@@ -66,4 +66,5 @@
     "VirtualConnections",
     "Webhooks",
     "Workbooks",
+    "VizQL",
 ]

tableauserverclient/server/endpoint/endpoint.py

@@ -66,7 +66,7 @@ def set_parameters(http_options, auth_token, content, content_type, parameters)
             parameters["headers"][CONTENT_TYPE_HEADER] = content_type
 
         Endpoint.set_user_agent(parameters)
-        if content is not None:
+        if content is not None and content_type != JSON_CONTENT_TYPE:
             parameters["data"] = content
         return parameters or {}
 
@@ -86,21 +86,20 @@ def set_user_agent(parameters):
         # return explicitly for testing only
         return parameters
 
-    def _blocking_request(self, method, url, parameters={}) -> Optional[Union["Response", Exception]]:
+    def _blocking_request(self, method, url, parameters={}, content=None, content_type=None) -> Optional[Union["Response", Exception]]:
         response = None
         logger.debug(f"[{datetime.timestamp()}] Begin blocking request to {url}")
         try:
-            response = method(url, **parameters)
+            if content and content_type == JSON_CONTENT_TYPE:
+                response = method(url, json=content, **parameters)
+            else:
+                response = method(url, **parameters)
             logger.debug(f"[{datetime.timestamp()}] Call finished")
         except Exception as e:
             logger.debug(f"Error making request to server: {e}")
             raise e
         return response
 
-    def send_request_while_show_progress_threaded(
-        self, method, url, parameters={}, request_timeout=None
-    ) -> Optional[Union["Response", Exception]]:
-        return self._blocking_request(method, url, parameters)
 
     def _make_request(
         self,
@@ -116,17 +115,19 @@ def _make_request(
         )
 
         logger.debug(f"request method {method.__name__}, url: {url}")
-        if content:
-            redacted = helpers.strings.redact_xml(content[:200])
-            # this needs to be under a trace or something, it's a LOT
-            # logger.debug("request content: {}".format(redacted))
 
         # a request can, for stuff like publishing, spin for ages waiting for a response.
         # we need some user-facing activity so they know it's not dead.
-        request_timeout = self.parent_srv.http_options.get("timeout") or 0
-        server_response: Optional[Union["Response", Exception]] = self.send_request_while_show_progress_threaded(
-            method, url, parameters, request_timeout
-        )
+        
+        if content and content_type == JSON_CONTENT_TYPE:
+            server_response: Optional[Union["Response", Exception]] = self._blocking_request(
+                method, url, parameters, content, content_type
+             )
+        else:
+            server_response: Optional[Union["Response", Exception]] = self._blocking_request(
+                method, url, parameters
+             )
+         
         logger.debug(f"[{datetime.timestamp()}] Async request returned: received {server_response}")
         # is this blocking retry really necessary? I guess if it was just the threading messing it up?
         if server_response is None: