cleanup connection state management class

This commit is contained in:
Florian Förster 2025-02-26 15:44:41 +01:00
parent ddafd190e6
commit 0e7778e1f4

View File

@ -33,15 +33,6 @@ class CurrentConnection:
self._session_token: str | None = None
self._logged_in: bool = False
def __getitem__(self, key: str) -> str:
return self.headers[key]
def __contains__(self, key: str) -> bool:
return key in self.headers
def as_dict(self) -> HttpContentHeaders:
return self.headers
@property
def headers(self) -> HttpContentHeaders:
return self._headers
@ -68,7 +59,7 @@ class CurrentConnection:
self._logged_in = True
def remove_session_token(self) -> None:
if "DelecoToken" in self:
if "DelecoToken" in self.headers:
del self._headers["DelecoToken"]
self._session_token = None
self._logged_in = False