name:
src/python/etc/util.py
-rw-r--r--
2210
1from argparse import ArgumentParser
2
3def get_file_name(path):
4 """
5 Separates a path name of the format "/whatever/dir/file.wav" into "file".
6 :param path: to parse into the primary component of the file.
7 :return: the separated file name.
8 """
9 return path.split("/")[-1:][0].split('.')[0]
10
11def get_args_for_input_directory(description):
12 """
13 Simple args parser for a single input directory.
14 :description: default description to output
15 :return: args parsed from command line
16 """
17 parser = ArgumentParser(description=description)
18 parser.add_argument('-d', '--directory', type=str, help='Directory of input files', required=True)
19 args = parser.parse_args()
20 return args
21
22def get_args_positive_negative_out(description):
23 """
24 Simple args parser for a single input directory, single output directory.
25 :description: default description
26 :return: args parsed from command line
27 """
28 parser = ArgumentParser(description=description)
29 parser.add_argument('-p', '--positive', type=str, help='Directory of input files, positive match', required=True)
30 parser.add_argument('-n', '--negative', type=str, help='Directory of input files, negative match', required=True)
31 parser.add_argument('-o', '--output', type=str, help='Directory of output files', required=True)
32 args = parser.parse_args()
33 return args
34
35def get_args_for_input_output_directories(description):
36 """
37 Simple args parser for a single input directory, single output directory.
38 :description: default description
39 :return: args parsed from command line
40 """
41 parser = ArgumentParser(description=description)
42 parser.add_argument('-d', '--directory', type=str, help='Directory of input files', required=True)
43 parser.add_argument('-o', '--output', type=str, help='Directory of output files', required=True)
44 args = parser.parse_args()
45 return args
46
47def get_args_for_input_file(description):
48 """
49 Simple args parser for a single input file.
50 :description: default description to output
51 :return: args parsed from command line
52 """
53 parser = ArgumentParser(description=description)
54 parser.add_argument('-f', '--file', type=str, help='Input file', required=True)
55 args = parser.parse_args()
56 return args