Added error handling to the resource tagger to display message to user when tag adding/deleting is not supported for a resource.

lbrendanl · Russell Hay · commit e8f4b0a04ba5 · 2017-04-11T14:27:37.000-07:00
diff --git a/tableauserverclient/server/endpoint/resource_tagger.py b/tableauserverclient/server/endpoint/resource_tagger.py
@@ -1,24 +1,38 @@
 from .endpoint import Endpoint
+from .exceptions import EndpointUnavailableError, ServerResponseError
 from .. import RequestFactory
 from ...models.tag_item import TagItem
 import logging
 import copy
 
-logger = logging.getLogger('tableau.endpoint')
+logger = logging.getLogger('tableau.endpoint.resource_tagger')
 
 
 class ResourceTagger(Endpoint):
     # Add new tags to resource
     def _add_tags(self, baseurl, resource_id, tag_set):
         url = "{0}/{1}/tags".format(baseurl, resource_id)
         add_req = RequestFactory.Tag.add_req(tag_set)
-        server_response = self.put_request(url, add_req)
+        
+        try:
+            server_response = self.put_request(url, add_req)
+        except ServerResponseError as e:
+            if e.code == "404003":
+                error = "Adding tags to this resource type is only available with REST API version 2.6 and later."
+                raise EndpointUnavailableError(error)
+        
         return TagItem.from_response(server_response.content)
 
     # Delete a resource's tag by name
     def _delete_tag(self, baseurl, resource_id, tag_name):
         url = "{0}/{1}/tags/{2}".format(baseurl, resource_id, tag_name)
-        self.delete_request(url)
+        
+        try:
+            self.delete_request(url)
+        except ServerResponseError as e:
+            if e.code == "404003":
+                error = "Deleting tags from this resource type is only available with REST API version 2.6 and later."
+                raise EndpointUnavailableError(error)
 
     # Remove and add tags to match the resource item's tag set
     def _update_tags(self, baseurl, resource_item):
