add query-tagging attribute to connection

Author: jacalata

tableauserverclient/models/connection_item.py

@@ -17,6 +17,7 @@ def __init__(self):
         self.server_port: Optional[str] = None
         self.username: Optional[str] = None
         self.connection_credentials: Optional[ConnectionCredentials] = None
+        self.query_tagging: bool = None
 
     @property
     def datasource_id(self) -> Optional[str]:
@@ -52,6 +53,7 @@ def from_response(cls, resp, ns) -> List["ConnectionItem"]:
             connection_item.server_address = connection_xml.get("serverAddress", None)
             connection_item.server_port = connection_xml.get("serverPort", None)
             connection_item.username = connection_xml.get("userName", None)
+            connection_item.query_tagging = connection_xml.get("queryTaggingEnabled", None)
             datasource_elem = connection_xml.find(".//t:datasource", namespaces=ns)
             if datasource_elem is not None:
                 connection_item._datasource_id = datasource_elem.get("id", None)

tableauserverclient/server/request_factory.py

@@ -1002,6 +1002,8 @@ def update_req(self, xml_request: ET.Element, connection_item: "ConnectionItem")
             connection_element.attrib["password"] = connection_item.password
         if connection_item.embed_password is not None:
             connection_element.attrib["embedPassword"] = str(connection_item.embed_password).lower()
+        if connection_item.query_tagging is not None:
+            connection_element.attrib["queryTaggingEnabled"] = str(connection_item.query_tagging).lower()
 
 
 class TaskRequest(object):