Skip to content

Commit 6e668fa

Browse files
GTGT
authored andcommitted
ccsrftoken to new format ccsrftoken_<port>
1 parent 92b95a3 commit 6e668fa

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

fortigate_api/fortigate.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -155,12 +155,13 @@ def login(self) -> None:
155155
verify=self.verify)
156156
except Exception as ex:
157157
raise self._hide_secret_ex(ex)
158+
csrf_token_name = f"ccsrftoken_{self.port}"
158159
for cookie in session.cookies:
159-
if cookie.name == "ccsrftoken" and isinstance(cookie.value, str):
160+
if cookie.name == csrf_token_name and isinstance(cookie.value, str):
160161
session.headers.update({"X-CSRFTOKEN": cookie.value[1:-1]})
161162
break
162163
else:
163-
raise ValueError("invalid login credentials, absent cookie ccsrftoken")
164+
raise ValueError(f"invalid login credentials, absent cookie {csrf_token_name}")
164165
response: Response = session.get(url=f"{self.url}/api/v2/cmdb/system/vdom")
165166
response.raise_for_status()
166167
self._session = session

0 commit comments

Comments
 (0)