raw2cfrad.py

#!/usr/bin/env python
# coding: utf-8
from __future__ import absolute_import, division, print_function, unicode_literals
__metaclass__ = type

import argparse
import textwrap
from os import path
from multiprocessing import Pool, Manager
from functools import partial
import radpy
from radpy.pyart_tools import sigmet2cfrad
from pyart import config


def make_parser():
    example = textwrap.dedent('''\
        Example:
          {} ???data/*/*.raw ???data/*/*/*.RAW*
        ''')
    parser = argparse.ArgumentParser(formatter_class=argparse.RawDescriptionHelpFormatter,
                                     description='Convert raw Sigmet radar data to cfradial.')
    parser.add_argument('files', metavar='FILE', type=str, nargs='+',
                        help='a file to process')
    default_out = 'cf'
    parser.add_argument('-o', '--outdir', metavar='OUTPUT_DIR', type=str,
                        default=default_out,
                        help='output directory, default: {}'.format(default_out))
    parser.add_argument('-d', '--delete', action='store_true',
                        help='delete skipped raw files')
    parser.add_argument('-v', '--verbose', action='store_true',
                        help='increase output verbosity')
    parser.epilog = example.format(parser.prog)
    return parser


def main(filepaths, **kws):
    p = Pool()
    m = Manager()
    dirlock = m.Lock()
    p.map(partial(sigmet2cfrad, dirlock=dirlock, **kws), filepaths)


if __name__ == '__main__':
    conf = config.load_config(radpy.conf_path)
    parser = make_parser()
    args = parser.parse_args()
    main(args.files, outdir=args.outdir, verbose=args.verbose,
         filter_raw=args.delete)