Skip to content

Commit 93ae1b8

Browse files
committed
added UAT support to TSC JWT Login
1 parent fd187ba commit 93ae1b8

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tableauserverclient/models/tableau_auth.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -198,19 +198,20 @@ class JWTAuth(Credentials):
198198
199199
"""
200200

201-
def __init__(self, jwt: str, site_id: Optional[str] = None, user_id_to_impersonate: Optional[str] = None) -> None:
201+
def __init__(self, jwt: str, isUat: Optional[bool] = False, site_id: Optional[str] = None, user_id_to_impersonate: Optional[str] = None) -> None:
202202
if jwt is None:
203203
raise TabError("Must provide a JWT token when using JWT authentication")
204204
super().__init__(site_id, user_id_to_impersonate)
205205
self.jwt = jwt
206+
self.isUat = isUat
206207

207208
@property
208209
def credentials(self) -> dict[str, str]:
209-
return {"jwt": self.jwt}
210+
return {"jwt": self.jwt, "isUat": str(self.isUat).lower()}
210211

211212
def __repr__(self):
212213
if self.user_id_to_impersonate:
213214
uid = f", user_id_to_impersonate=f{self.user_id_to_impersonate}"
214215
else:
215216
uid = ""
216-
return f"<{self.__class__.__qualname__} jwt={self.jwt[:5]}... (site={self.site_id}{uid})>"
217+
return f"<{self.__class__.__qualname__} jwt={self.jwt[:5]}... isUat={self.isUat} (site={self.site_id}{uid})>"

0 commit comments

Comments
 (0)