From b94f33cd26a489265fc741f18c6d9fd3cedb4e7a Mon Sep 17 00:00:00 2001 From: Lucy Linder Date: Thu, 21 Mar 2024 17:19:18 +0100 Subject: [PATCH] fix: make BearerAuth's token_getter public --- mantelo/client.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mantelo/client.py b/mantelo/client.py index 58b4d2b..395353d 100644 --- a/mantelo/client.py +++ b/mantelo/client.py @@ -2,6 +2,7 @@ import requests import slumber +from attrs import define from slumber.exceptions import SlumberHttpBaseException from .connection import ( @@ -27,12 +28,12 @@ def _request(self, *args, **kwargs): raise HttpException.from_slumber_exception(ex) from None +@define class BearerAuth(requests.auth.AuthBase): - def __init__(self, token_getter: Callable[[], str]): - self._token_getter = token_getter + token_getter: Callable[[], str] def __call__(self, r): - r.headers["Authorization"] = f"Bearer {self._token_getter()}" + r.headers["Authorization"] = f"Bearer {self.token_getter()}" return r