diff --git a/rasp_grid_cfg.py b/rasp_grid_cfg.py index d51df82..368dea9 100644 --- a/rasp_grid_cfg.py +++ b/rasp_grid_cfg.py @@ -318,9 +318,15 @@ class MainWindow(QTabWidget): self.init_vars() - if 'fishgrid.cfg' in os.listdir('.'): + if os.path.exists('/media/pi/data1/fishgrid.cfg'): + cfg_file = '/media/pi/data1/fishgrid.cfg' + self.read_cfg(cfg_file) + + elif 'fishgrid.cfg' in os.listdir('.'): cfg_file = './fishgrid.cfg' self.read_cfg(cfg_file) + else: + quit() self.Grid.get_widgets() @@ -454,8 +460,10 @@ class MainWindow(QTabWidget): continue def save_cfg(self): - # if os.path.exists() - f = open('./fishgrid.cfg', 'w+') + if os.path.exists('/media/pi/data1/fishgrid.cfg'): + f = open('/media/pi/data1/fishgrid.cfg', 'w+') + else: + f = open('./fishgrid.cfg', 'w+') f.write('*FishGrid\n') f.write(' Grid &1:\n')