From 76070d7c2e834efe235b59931c2f786200c849e0 Mon Sep 17 00:00:00 2001 From: Qudix <17361645+Qudix@users.noreply.github.com> Date: Fri, 1 Oct 2021 21:54:36 -0500 Subject: [PATCH] Fix save name edge case --- games/stalkeranomaly/XRSave.py | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/games/stalkeranomaly/XRSave.py b/games/stalkeranomaly/XRSave.py index 9166013..f4a23e8 100644 --- a/games/stalkeranomaly/XRSave.py +++ b/games/stalkeranomaly/XRSave.py @@ -82,18 +82,24 @@ class XRSave: self.readObject(stream) def fetchInfo(self): - save_clean = self.filepath.name.split(".scop", 1)[0] - save_split = save_clean.split(" - ", 1) - self.user = save_split[0] - save_end = save_split[1].split("_", 1) - if len(save_end) > 1: - self.save_fmt = f"{save_end[0]} (#{save_end[1]})".title() - else: - self.save_fmt = f"{save_end[0]}".title() + self.splitInfo() self.time = self.filepath.stat().st_mtime self.time_date = datetime.fromtimestamp(self.time) self.time_fmt = self.time_date.strftime("%I:%M %m/%d/%Y") + def splitInfo(self): + save_clean = self.filepath.name.split(".scop", 1)[0] + save_split = save_clean.split(" - ", 1) + self.user = save_split[0] + if len(save_split) > 1: + save_end = save_split[1].split("_", 1) + if len(save_end) > 1: + self.save_fmt = f"{save_end[0]} (#{save_end[1]})".title() + else: + self.save_fmt = f"{save_end[0]}".title() + else: + self.save_fmt = "Unknown" + def readFile(self, file) -> Optional[XRStream]: size = self.filepath.stat().st_size if size < 8: