Skip to content

Commit 6daceb4

Browse files
committed
Added target token chose for token deletion handler
1 parent 18f6be7 commit 6daceb4

2 files changed

Lines changed: 14 additions & 4 deletions

File tree

bugout/app.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -146,10 +146,13 @@ def create_token_restricted(
146146
return self.user.create_token_restricted(token=token)
147147

148148
def revoke_token(
149-
self, token: Union[str, uuid.UUID], timeout: float = REQUESTS_TIMEOUT
149+
self,
150+
token: Union[str, uuid.UUID],
151+
target_token: Optional[Union[str, uuid.UUID]] = None,
152+
timeout: float = REQUESTS_TIMEOUT,
150153
) -> uuid.UUID:
151154
self.user.timeout = timeout
152-
return self.user.revoke_token(token=token)
155+
return self.user.revoke_token(token=token, target_token=target_token)
153156

154157
def revoke_token_by_id(
155158
self, token: Union[str, uuid.UUID], timeout: float = REQUESTS_TIMEOUT

bugout/user.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,13 +177,20 @@ def create_token_restricted(self, token: Union[str, uuid.UUID]) -> BugoutToken:
177177
result = self._call(method=Method.post, path=create_token_path, headers=headers)
178178
return BugoutToken(**result)
179179

180-
def revoke_token(self, token: Union[str, uuid.UUID]) -> uuid.UUID:
180+
def revoke_token(
181+
self,
182+
token: Union[str, uuid.UUID],
183+
target_token: Optional[Union[str, uuid.UUID]] = None,
184+
) -> uuid.UUID:
181185
revoke_token_path = "token"
182186
headers = {
183187
"Authorization": f"Bearer {token}",
184188
}
189+
data = {}
190+
if target_token is not None:
191+
data.update({"target_token": target_token})
185192
result = self._call(
186-
method=Method.delete, path=revoke_token_path, headers=headers
193+
method=Method.delete, path=revoke_token_path, headers=headers, data=data
187194
)
188195
return result
189196

0 commit comments

Comments
 (0)