Skip to content

Commit c6fefdd

Browse files
committed
feat(auth): python client support for bearer auth
1 parent 6c8f959 commit c6fefdd

File tree

1 file changed

+7
-0
lines changed
  • clients/python/src/objectstore_client

1 file changed

+7
-0
lines changed

clients/python/src/objectstore_client/client.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,7 @@ def __init__(
9797
base_url: str,
9898
metrics_backend: MetricsBackend | None = None,
9999
propagate_traces: bool = False,
100+
auth_token: str | None = None,
100101
retries: int | None = None,
101102
timeout_ms: float | None = None,
102103
connection_kwargs: Mapping[str, Any] | None = None,
@@ -125,6 +126,7 @@ def __init__(
125126
self._base_path = urlparse(base_url).path
126127
self._metrics_backend = metrics_backend or NoOpMetricsBackend()
127128
self._propagate_traces = propagate_traces
129+
self._auth_token = auth_token
128130

129131
def session(self, usecase: Usecase, **scopes: str | int | bool) -> Session:
130132
"""
@@ -177,6 +179,7 @@ def session(self, usecase: Usecase, **scopes: str | int | bool) -> Session:
177179
self._base_path,
178180
self._metrics_backend,
179181
self._propagate_traces,
182+
self._auth_token,
180183
usecase,
181184
scope_str,
182185
)
@@ -195,13 +198,15 @@ def __init__(
195198
base_path: str,
196199
metrics_backend: MetricsBackend,
197200
propagate_traces: bool,
201+
auth_token: str | None,
198202
usecase: Usecase,
199203
scope: str,
200204
):
201205
self._pool = pool
202206
self._base_path = base_path
203207
self._metrics_backend = metrics_backend
204208
self._propagate_traces = propagate_traces
209+
self._auth_token = auth_token
205210
self._usecase = usecase
206211
self._scope = scope
207212

@@ -211,6 +216,8 @@ def _make_headers(self) -> dict[str, str]:
211216
headers.update(
212217
dict(sentry_sdk.get_current_scope().iter_trace_propagation_headers())
213218
)
219+
if self._auth_token:
220+
headers['Authorization'] = f"Bearer {self._auth_token}"
214221
return headers
215222

216223
def _make_url(self, key: str | None, full: bool = False) -> str:

0 commit comments

Comments
 (0)