diff --git a/src/delta_barth/types.py b/src/delta_barth/types.py index 48775dc..334a774 100644 --- a/src/delta_barth/types.py +++ b/src/delta_barth/types.py @@ -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