Skip to content

Commit

Permalink
Fix issue where session auth was not cleared when logging out
Browse files Browse the repository at this point in the history
  • Loading branch information
Josh5 committed Mar 16, 2024
1 parent 41346cb commit cf962be
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions unmanic/libs/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def __fetch_installation_data(self):
self.__update_session_auth(access_token=current_installation.user_access_token,
session_cookies=current_installation.session_cookies)

def __store_installation_data(self):
def __store_installation_data(self, force_save_access_token=False):
"""
Store installation data in DB to persist reboot
Expand All @@ -207,9 +207,9 @@ def __store_installation_data(self):
db_installation.name = self.name
db_installation.email = self.email
db_installation.created = self.created
if self.user_access_token:
if self.user_access_token or force_save_access_token:
db_installation.user_access_token = self.user_access_token
if self.session_cookies:
if self.session_cookies or force_save_access_token:
db_installation.session_cookies = self.session_cookies
db_installation.save()

Expand All @@ -226,7 +226,8 @@ def __reset_session_installation_data(self):
self.email = ''
self.created = time.time()
self.user_access_token = None
self.__store_installation_data()
self.__store_installation_data(force_save_access_token=True)
self.__clear_session_auth()

def __update_session_auth(self, access_token=None, session_cookies=None):
# Update session headers
Expand All @@ -241,6 +242,10 @@ def __update_session_auth(self, access_token=None, session_cookies=None):
except Exception as e:
self.logger.error('Error trying to reload session cookies - %s', str(e))

def __clear_session_auth(self):
self.requests_session.cookies.clear()
self.requests_session.headers.update({'Authorization': ''})

def get_installation_uuid(self):
"""
Returns the installation UUID as a string.
Expand Down

0 comments on commit cf962be

Please sign in to comment.