From 8bf7ab3ce2348881dae7896615c91d854cba9eb5 Mon Sep 17 00:00:00 2001 From: "Jimi (Dimitris) Charalampidis" Date: Tue, 11 Mar 2025 17:53:52 +0200 Subject: [PATCH] Fix cookie management flow --- streamlit_authenticator/models/cookie_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/streamlit_authenticator/models/cookie_model.py b/streamlit_authenticator/models/cookie_model.py index 56e1c8bc..9f199e02 100644 --- a/streamlit_authenticator/models/cookie_model.py +++ b/streamlit_authenticator/models/cookie_model.py @@ -76,12 +76,12 @@ def get_cookie(self) -> Optional[Dict[str, Any]]: If valid, returns a dictionary containing the cookie's data. Returns None if the cookie is expired or invalid. """ - if st.session_state['logout']: - return False - # self.token = self.cookie_manager.get(self.cookie_name) + if st.session_state.get('logout'): + return None self.token = st.context.cookies[self.cookie_name] if self.cookie_name in \ st.context.cookies else None if self.token is not None: + self.cookie_manager.set(self.cookie_name, self.token) self.token = self._token_decode() if (self.token is not False and 'username' in self.token and self.token['exp_date'] > datetime.now().timestamp()):