Skip to content

Commit 88e2c86

Browse files
Hans Raphael (BD/TOA-SWE6)Hans Raphael (BD/TOA-SWE6)
authored andcommitted
SOCOPL-2958: initial separation of authenticator
1 parent fdaf1ff commit 88e2c86

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

atlassian/rest_client.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,8 @@ def __init__(
9494
max_backoff_retries: int = 1000,
9595
backoff_factor=1.0,
9696
backoff_jitter=1.0,
97-
retry_with_header=True,
97+
retry_with_header=True,
98+
header=None
9899
):
99100
"""
100101
init function for the AtlassianRestAPI object.
@@ -197,6 +198,8 @@ def __init__(
197198
self._create_kerberos_session(kerberos)
198199
elif cookies is not None:
199200
self._session.cookies.update(cookies)
201+
elif header is not None:
202+
self._create_header_session(header)
200203

201204
def __enter__(self) -> Self:
202205
return self
@@ -210,7 +213,10 @@ def _create_basic_session(self, username: str, password: str) -> None:
210213
def _create_token_session(self, token: str) -> None:
211214
self._update_header("Authorization", f"Bearer {token.strip()}")
212215

213-
def _create_kerberos_session(self, _: object) -> None:
216+
def _create_header_session(self, header: dict) -> None:
217+
self._session.headers.update(header)
218+
219+
def _create_kerberos_session(self, _):
214220
from requests_kerberos import OPTIONAL, HTTPKerberosAuth
215221

216222
self._session.auth = HTTPKerberosAuth(mutual_authentication=OPTIONAL)

0 commit comments

Comments
 (0)