oceano.py

import argparse
import sys
import PySimpleGUI as sg
# import PySimpleGUIQt as sg
import toml
import logging
from file_extractor import FileExtractor
import pathlib

# usage:
# > python oceano.py data/cnv/dfr2900[1-3].cnv -d
# > python oceano.py data/cnv/dfr2900[1-3].cnv -k PRES TEMP PSAL DOX2 DENS
# > python oceano.py data/cnv/dfr29*.cnv -d
parser = argparse.ArgumentParser(
    description='This program read multiple ASCII file, extract physical parameter \
            following ROSCOP codification at the given column, fill arrays, write header file',
    epilog='J. Grelet IRD US191 - March 2019')
parser.add_argument('-d', '--debug', help='display debug informations',
                    action='store_true')
parser.add_argument('-c', '--config', help="toml configuration file, (default: %(default)s)",
                    default='tests/test.toml')
parser.add_argument('-k', '--key', nargs='+', default=['PRES', 'TEMP', 'PSAL'],
                    help='display dictionary for key(s), (default: %(default)s)')
parser.add_argument('-g', '--gui', action='store_true',
                    help='use GUI interface')
parser.add_argument('files', nargs='*',
                    help='cnv file(s) to parse, (default: data/cnv/dfr29*.cnv)')
args = parser.parse_args()

# set looging mode if debug
if args.debug:
    logging.basicConfig(
        format='%(levelname)s:%(message)s', level=logging.DEBUG)

# read config Toml file and get the physical parameter, Roscop code
cfg = toml.load(args.config)
keys = cfg['split']['ctd'].keys()

# test arguements from sys.argv, args is never to None with default option set
if args.gui or args.debug or len(sys.argv) == 1:

    # change look and feel color scheme
    sg.ChangeLookAndFeel('SandyBeach')

    # define GUI layout
    layout = ([[sg.Text('File(s) to read and convert')],
               [sg.FilesBrowse(key='_FILE',
                               initial_folder='data/cnv', file_types=(("cnv files", "*.cnv"),))],
               *[[sg.Checkbox(k, key=k)] for k in keys],
               [sg.CloseButton('Run'), sg.CloseButton('Cancel')]])

    # create the main windows
    event, values = sg.Window('Oceano converter').Layout(layout).Read()

    # debug return values from GUI
    logging.debug("Event: {}, Values: {}".format(event, values))

    # extract parameters selected (true) from dict values
    new_values = values.copy()
    for k in values.keys():
        if k[0] == '_' or values[k] == False:
            del new_values[k]
    args.key = new_values.keys()

    # values['_FILE'] is a string with files separated by ';' and fileExtractor need a list
    files = values['_FILE'].split(';')
    args.files = files

else:
    # in line command mode
    args.files

# check if no file selected or cancel button pressed
logging.debug("File(s): {}, Config: {}, Keys: {}".format(
    args.files, args.config, args.key))
if not all(args.files):
    sg.Popup("Cancel", "No filename supplied")
    raise SystemExit("Cancelling: no filename supplied")

# fileExtractor
fe = FileExtractor(args.files)

cfg = toml.load(args.config)
[n, m] = fe.firstPass()
fe.secondPass(args.key, cfg, 'ctd')
# fe.secondPass(['PRES', 'TEMP', 'PSAL', 'DOX2'], cdf, 'ctd')
fe.disp(args.key)
# fe.disp(['PRES', 'TEMP', 'PSAL', 'DOX2'])