From 849964b6591586d705fd900469e0973324dfca02 Mon Sep 17 00:00:00 2001 From: Richard Bronkhorst Date: Mon, 12 Jun 2023 11:13:06 +0200 Subject: [PATCH] Update atlas_builder.py and command_line.py --- nullptr/atlas_builder.py | 6 ++++-- nullptr/command_line.py | 10 +++++++++- 2 files changed, 13 insertions(+), 3 deletions(-) diff --git a/nullptr/atlas_builder.py b/nullptr/atlas_builder.py index 97a41d8..d8c201f 100644 --- a/nullptr/atlas_builder.py +++ b/nullptr/atlas_builder.py @@ -9,7 +9,10 @@ class AtlasBuilder: self.stop_auto = False def wait_for_stop(self): - input() + try: + input() + except EOFError: + pass self.stop_auto = True print('stopping...') @@ -39,7 +42,6 @@ class AtlasBuilder: break r = self.api.list_waypoints(s) self.all_specials(r) - self.store.flush() sleep(0.5) diff --git a/nullptr/command_line.py b/nullptr/command_line.py index 55c4acb..d659584 100644 --- a/nullptr/command_line.py +++ b/nullptr/command_line.py @@ -42,6 +42,9 @@ class CommandLine: def after_cmd(self): pass + + def handle_eof(self): + print('EOF') def do_quit(self): print('byebye!') @@ -75,6 +78,11 @@ class CommandLine: def run(self): while not self.stopping and not self.reloading: - c = input(self.prompt()) + p = self.prompt() + try: + c = input(p) + except EOFError: + self.handle_eof() + break self.handle_cmd(c)