diff --git a/installer/client/cli/helper.py b/installer/client/cli/helper.py index afde5ec..3f5db8a 100644 --- a/installer/client/cli/helper.py +++ b/installer/client/cli/helper.py @@ -60,12 +60,16 @@ class Session: def list_sessions(self): sessionlist = os.listdir(self.sessions_folder) - most_recent = self.find_most_recent_file().split("/")[-1] - for session in sessionlist: - with open(os.path.join(self.sessions_folder, session), "r") as f: - firstline = f.readline().strip() - secondline = f.readline().strip() - if session == most_recent: - print(f"{session} **default** \"{firstline}\n{secondline}\n\"") - else: - print(f"{session} \"{firstline}\n{secondline}\n\"") + find_most_recent_file_result = self.find_most_recent_file() + if find_most_recent_file_result is not None: + most_recent = find_most_recent_file_result.split("/")[-1] + for session in sessionlist: + with open(os.path.join(self.sessions_folder, session), "r") as f: + firstline = f.readline().strip() + secondline = f.readline().strip() + if session == most_recent: + print(f"{session} **default** \"{firstline}\n{secondline}\n\"") + else: + print(f"{session} \"{firstline}\n{secondline}\n\"") + else: + print('No files present in sessions directory')