| | |
| |
|
| | import argparse |
| | import functools |
| | import subprocess |
| |
|
| |
|
| | def read_file(fname): |
| | with open(fname, encoding='utf-8') as f: |
| | return f.read() |
| |
|
| |
|
| | def write_file(fname, content, mode='w'): |
| | with open(fname, mode, encoding='utf-8') as f: |
| | return f.write(content) |
| |
|
| |
|
| | def get_filename_args(has_infile=False, default_outfile=None): |
| | parser = argparse.ArgumentParser() |
| | if has_infile: |
| | parser.add_argument('infile', help='Input file') |
| | kwargs = {'nargs': '?', 'default': default_outfile} if default_outfile else {} |
| | parser.add_argument('outfile', **kwargs, help='Output file') |
| |
|
| | opts = parser.parse_args() |
| | if has_infile: |
| | return opts.infile, opts.outfile |
| | return opts.outfile |
| |
|
| |
|
| | def compose_functions(*functions): |
| | return lambda x: functools.reduce(lambda y, f: f(y), functions, x) |
| |
|
| |
|
| | def run_process(*args, **kwargs): |
| | kwargs.setdefault('text', True) |
| | kwargs.setdefault('check', True) |
| | kwargs.setdefault('capture_output', True) |
| | if kwargs['text']: |
| | kwargs.setdefault('encoding', 'utf-8') |
| | kwargs.setdefault('errors', 'replace') |
| | return subprocess.run(args, **kwargs) |
| |
|
