Skip to content

Commit cba111a

Browse files
authored
Add UAT (unified access token) support to JWT login (#1671)
1 parent fd187ba commit cba111a

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

tableauserverclient/models/tableau_auth.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -198,19 +198,26 @@ 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__(
202+
self,
203+
jwt: str,
204+
isUat: bool = False,
205+
site_id: Optional[str] = None,
206+
user_id_to_impersonate: Optional[str] = None,
207+
) -> None:
202208
if jwt is None:
203209
raise TabError("Must provide a JWT token when using JWT authentication")
204210
super().__init__(site_id, user_id_to_impersonate)
205211
self.jwt = jwt
212+
self.isUat = isUat
206213

207214
@property
208215
def credentials(self) -> dict[str, str]:
209-
return {"jwt": self.jwt}
216+
return {"jwt": self.jwt, "isUat": str(self.isUat).lower()}
210217

211218
def __repr__(self):
212219
if self.user_id_to_impersonate:
213220
uid = f", user_id_to_impersonate=f{self.user_id_to_impersonate}"
214221
else:
215222
uid = ""
216-
return f"<{self.__class__.__qualname__} jwt={self.jwt[:5]}... (site={self.site_id}{uid})>"
223+
return f"<{self.__class__.__qualname__} jwt={self.jwt[:5]}... isUat={self.isUat} (site={self.site_id}{uid})>"

0 commit comments

Comments
 (0)