10000 feat(exceptions): separate failed signin error · tableau/server-client-python@74f7797 · GitHub
[go: up one dir, main page]
Skip to content

Commit 74f7797

Browse files
jorwoodsjorwoods
committed
feat(exceptions): separate failed signin error
Closes #1472 This makes sign in failures their own class of exceptions, while still inheriting from NotSignedInException to not break backwards compatability for any existing client code. This should allow users to get out more specific exceptions more easily on what failed with their authentication request.
1 parent 6ec632e commit 74f7797

File tree

5 files changed

+29
-9
lines changed

5 files changed

+29
-9
lines changed

‎tableauserverclient/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
PDFRequestOptions,
5757
RequestOptions,
5858
MissingRequiredFieldError,
59+
FailedSignInError,
5960
NotSignedInError,
6061
ServerResponseError,
6162
Filter,
@@ -79,6 +80,7 @@
7980
"DatabaseItem",
8081
"DataFreshnessPolicyItem",
8182
"DatasourceItem",
83+
"FailedSignInError",
8284
"FavoriteItem",
8385
"FlowItem",
8486
"FlowRunItem",

‎tableauserverclient/server/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from tableauserverclient.server.sort import Sort
1212
from tableauserverclient.server.server import Server
1313
from tableauserverclient.server.pager import Pager
14-
from tableauserverclient.server.endpoint.exceptions import NotSignedInError
14+
from tableauserverclient.server.endpoint.exceptions import FailedSignInError, NotSignedInError
1515

1616
from tableauserverclient.server.endpoint import (
1717
Auth,
@@ -57,6 +57,7 @@
5757
"Sort",
5858
"Server",
5959
"Pager",
60+
"FailedSignInError",
6061
"NotSignedInError",
6162
"Auth",
6263
"CustomViews",

‎tableauserverclient/server/endpoint/endpoint.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from tableauserverclient.server.request_options import RequestOptions
2020

2121
from tableauserverclient.server.endpoint.exceptions import (
22+
FailedSignInError,
2223
ServerResponseError,
2324
InternalServerError,
2425
NonXMLResponseError,
@@ -160,7 +161,7 @@ def _check_status(self, server_response: "Response", url: Optional[str] = None):
160161
try:
161162
if server_response.status_code == 401:
162163
# TODO: catch this in server.py and attempt to sign in again, in case it's a session expiry
163-
raise NotSignedInError(server_response.content, url)
164+
raise FailedSignInError.from_response(server_response.content, self.parent_srv.namespace, url)
164165

165166
raise ServerResponseError.from_response(server_response.content, self.parent_srv.namespace, url)
166167
except ParseError:

‎tableauserverclient/server/endpoint/exceptions.py

Lines changed: 20 additions & 4 deletions
< B41A /tr>
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,20 @@
11
from defusedxml.ElementTree import fromstring
2-
from typing import Optional
2+
from typing import Mapping, Optional, TypeVar
3+
4+
5+
def split_pascal_case(s: str) -> str:
6+
return "".join([f" {c}" if c.isupper() else c for c in s]).strip()
37

48

59
class TableauError(Exception):
610
pass
711

812

9-
class ServerResponseError(TableauError):
10-
def __init__(self, code, summary, detail, url=None):
13+
T = TypeVar("T")
14+
15+
16+
class XMLError(TableauError):
17+
def __init__(self, code: str, summary: str, detail: str, url: Optional[str] = None) -> None:
1118
self.code = code
1219
self.summary = summary
1320
self.detail = detail
@@ -18,7 +25,7 @@ def __str__(self):
1825
return f"\n\n\t{self.code}: {self.summary}\n\t\t{self.detail}"
1926

2027
@classmethod
21-
def from_response(cls, resp, ns, url=None):
28+
def from_response(cls, resp, ns, url):
2229
# Check elements exist before .text
2330
parsed_response = fromstring(resp)
2431
try:
@@ -33,6 +40,10 @@ def from_response(cls, resp, ns, url=None):
3340
return error_response
3441

3542

43+
class ServerResponseError(XMLError):
44+
pass
45+
46+
3647
class InternalServerError(TableauError):
3748
def __init__(self, server_response, request_url: Optional[str] = None):
3849
self.code = server_response.status_code
@@ -51,6 +62,11 @@ class NotSignedInError(TableauError):
5162
pass
5263

5364

65+
class FailedSignInError(XMLError, NotSignedInError):
66+
def __str__(self):
67+
return f"{split_pascal_case(self.__class__.__name__)}: {super().__str__()}"
68+
69+
5470
class ItemTypeNotAllowed(TableauError):
5571
pass
5672

‎test/test_auth.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,23 +63,23 @@ def test_sign_in_error(self):
6363
with requests_mock.mock() as m:
6464
m.post(self.baseurl + "/signin", text=response_xml, status_code=401)
6565
tableau_auth = TSC.TableauAuth("testuser", "wrongpassword")
66-
self.assertRaises(TSC.NotSignedInError, self.server.auth.sign_in, tableau_auth)
66+
self.assertRaises(TSC.FailedSignInError, self.server.auth.sign_in, tableau_auth)
6767

6868
def test_sign_in_invalid_token(self):
6969
with open(SIGN_IN_ERROR_XML, "rb") as f:
7070
response_xml = f.read().decode("utf-8")
7171
with requests_mock.mock() as m:
7272
m.post(self.baseurl + "/signin", text=response_xml, status_code=401)
7373
tableau_auth = TSC.PersonalAccessTokenAuth(token_name="mytoken", personal_access_token="invalid")
74-
self.assertRaises(TSC.NotSignedInError, self.server.auth.sign_in, tableau_auth)
74+
self.assertRaises(TSC.FailedSignInError, self.server.auth.sign_in, tableau_auth)
7575

7676
def test_sign_in_without_auth(self):
7777
with open(SIGN_IN_ERROR_XML, "rb") as f:
7878
response_xml = f.read().decode("utf-8")
7979
with requests_mock.mock() as m:
8080
m.post(self.baseurl + "/signin", text=response_xml, status_code=401)
8181
tableau_auth = TSC.TableauAuth("", "")
82-
self.assertRaises(TSC.NotSignedInError, self.server.auth.sign_in, tableau_auth)
82+
self.assertRaises(TSC.FailedSignInError, self.server.auth.sign_in, tableau_auth)
8383

8484
def test_sign_out(self):
8585
with open(SIGN_IN_XML, "rb") as f:

0 commit comments

Comments
 (0)
0