fix: make BearerAuth's token_getter public

derlin · Mar 21, 2024 · b94f33c · b94f33c
1 parent 33603fb
commit b94f33c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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