| |
|
|
| 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) |
|
|
