Enhancement tableau#117: Add support for the oAuth flag

hugoboos · Russell Hay · commit 01235eaba040 · 2017-01-11T11:44:17.000-08:00
* Add support for the oAuth flag when publishing workbooks and data sources
diff --git a/tableauserverclient/models/connection_credentials.py b/tableauserverclient/models/connection_credentials.py
@@ -9,10 +9,11 @@ class ConnectionCredentials(object):
 
     """
 
-    def __init__(self, name, password, embed=True):
+    def __init__(self, name, password, embed=True, oauth=False):
         self.name = name
         self.password = password
         self.embed = embed
+        self.oauth = oauth
 
     @property
     def embed(self):
@@ -22,3 +23,12 @@ def embed(self):
     @property_is_boolean
     def embed(self, value):
         self._embed = value
+
+    @property
+    def oauth(self):
+        return self._oauth
+
+    @oauth.setter
+    @property_is_boolean
+    def oauth(self, value):
+        self._oauth = value
diff --git a/tableauserverclient/server/request_factory.py b/tableauserverclient/server/request_factory.py
@@ -42,6 +42,9 @@ def _generate_xml(self, datasource_item, connection_credentials=None):
             credentials_element.attrib['name'] = connection_credentials.name
             credentials_element.attrib['password'] = connection_credentials.password
             credentials_element.attrib['embed'] = 'true' if connection_credentials.embed else 'false'
+
+            if connection_credentials.oauth:
+                credentials_element.attrib['oAuth'] = 'true'
         return ET.tostring(xml_request)
 
     def update_req(self, datasource_item):
@@ -279,6 +282,9 @@ def _generate_xml(self, workbook_item, connection_credentials=None):
             credentials_element.attrib['name'] = connection_credentials.name
             credentials_element.attrib['password'] = connection_credentials.password
             credentials_element.attrib['embed'] = 'true' if connection_credentials.embed else 'false'
+
+            if connection_credentials.oauth:
+                credentials_element.attrib['oAuth'] = 'true'
         return ET.tostring(xml_request)
 
     def update_req(self, workbook_item):
