From 135e16188deaa5698e1ecc388738cd4fbcd7191d Mon Sep 17 00:00:00 2001 From: Sander Land Date: Tue, 29 Dec 2020 13:20:21 +0100 Subject: [PATCH] sgf arg fix --- katrain/__main__.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/katrain/__main__.py b/katrain/__main__.py index 355277b..80b60b9 100644 --- a/katrain/__main__.py +++ b/katrain/__main__.py @@ -157,8 +157,13 @@ class KaTrainGui(Screen, KaTrainBase): self.board_gui.trainer_config = self.config("trainer") self.engine = KataGoEngine(self, self.config("engine")) threading.Thread(target=self._message_loop_thread, daemon=True).start() - if len(sys.argv) == 2 and os.path.isfile(sys.argv[1]) and not sys.argv[1].endswith('json'): - self.load_sgf_file(sys.argv[1], fast=True, rewind=True) + sgf_args = [ + f + for f in sys.argv[1:] + if os.path.isfile(f) and any(f.lower().endswith(ext) for ext in ["sgf", "ngf", "gib"]) + ] + if sgf_args: + self.load_sgf_file(sgf_args[0], fast=True, rewind=True) else: self._do_new_game()