data_conversion.py

import argparse
import sys; sys.path.insert(0, './lib')

import drconvert as conv
import yaml
from os import makedirs


parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument("config", help="Options file (.yaml) for conversion")       #defines script arguments
parser.add_argument("type", help="Type of conversion to perform (shp2drf, rawSPOT2shp, rawMetO2shp)")
args = parser.parse_args()      #collects user input
config = vars(args)             #converts user input arguments into a dict

with open(config['config'], 'r') as fid:
    cfg = yaml.safe_load(fid)

if config['type'] == 'shp2drf':
    makedirs(cfg['clean_drf_folder'], exist_ok=True)
    conv.write.convert_shp_to_drf(cfg['shp_folder'], config['config'], cfg['clean_drf_folder'])
elif config['type'] == 'rawMetO2shp':
    makedirs(cfg['shp_folder'], exist_ok=True)
    drifters = conv.read.read_metocean_csv(cfg['raw_data_folders']['meto'],config['config'])
    conv.process.write_drifters_to_shp(drifters, cfg['shp_folder'], cfg['coastline_shp'])
elif config['type'] == 'rawSPOT2shp':
    makedirs(cfg['shp_folder'], exist_ok=True)
    drifters = conv.read.read_SPOT_xlsx(cfg['raw_data_folders']['spot'], cfg)
    conv.process.write_drifters_to_shp(drifters, cfg['shp_folder'], cfg['coastline_shp'])
else:
    raise ValueError('Not a valid conversion type (use shp2drf, rawSPOT2shp, rawMetO2shp)')