Skip to content

Commit 70e79c5

Browse files
authored
Login update ramsons (#41)
* update login scheme * update login scheme * fix indentation
1 parent 256cf03 commit 70e79c5

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

app/main.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ async def read_item(
128128

129129

130130
def _okapi_login():
131-
url = f"{os.getenv('OKAPI_URL')}/authn/login"
131+
url = f"{os.getenv('OKAPI_URL')}/authn/login-with-expiry"
132132
headers = {
133133
"X-Okapi-Tenant": os.getenv("OKAPI_TENANT"),
134134
}
@@ -139,10 +139,14 @@ def _okapi_login():
139139
r = requests.post(url, json=data, headers=headers)
140140
r.raise_for_status()
141141
if r.status_code == 201:
142-
return r.headers["X-Okapi-Token"]
142+
cookies = r.headers.get("Set-Cookie")
143+
if cookies:
144+
for cookie in cookies.split(';'):
145+
if cookie.startswith("folioAccessToken="):
146+
r.headers["X-Okapi-Token"] = cookie.split("=")[1].split(";")[0]
147+
return r.headers["X-Okapi-Token"]
143148
return None
144149

145-
146150
def _reps_to_regex(replacements: List, field: str):
147151
return [
148152
(fr"^{rep['string']}$", f"{rep['replacement']}")

0 commit comments

Comments
 (0)