Skip to content

Commit fbf67f3

Browse files
committed
client: add refresh_tenant_token function as tenant tokens expire more offten
1 parent 30cf2c6 commit fbf67f3

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

onekey_client/client.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,11 @@ def use_tenant(self, tenant: m.Tenant):
159159
self._tenant_token_public_key,
160160
)
161161
self._state.raw_tenant_token = json_res["tenant_token"]
162+
self._state.tenant = tenant
163+
164+
@_tenant_required
165+
def refresh_tenant_token(self):
166+
self.use_tenant(self._state.tenant)
162167

163168
@_tenant_required
164169
def query(self, query: str, variables: Optional[Dict] = None, timeout=60):
@@ -236,3 +241,4 @@ def __init__(self):
236241
self.tenants = None
237242
self.raw_id_token = None
238243
self.raw_tenant_token = None
244+
self.tenant = None

0 commit comments

Comments
 (0)