Skip to content

Commit 2dc672c

Browse files
cabhishekgabeiglio
authored andcommitted
Ability to override session adapter and auth when session object is created
1 parent 640c592 commit 2dc672c

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

pyiceberg/catalog/rest/__init__.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -274,6 +274,14 @@ def _create_session(self) -> Session:
274274
if property_as_bool(self.properties, SIGV4, False):
275275
self._init_sigv4(session)
276276

277+
# Mount custom adapters
278+
if session_adapters := self.properties.get("session_adapters"):
279+
for prefix, adapter in session_adapters.items(): # type: ignore
280+
session.mount(prefix, adapter)
281+
# Add custom auth
282+
if session_auth := self.properties.get("session_auth"):
283+
session.auth = session_auth # type: ignore
284+
277285
return session
278286

279287
def _create_legacy_oauth2_auth_manager(self, session: Session) -> AuthManager:

0 commit comments

Comments
 (0)