-
Notifications
You must be signed in to change notification settings - Fork 11
/
prepare_data.py
34 lines (30 loc) · 1.02 KB
/
prepare_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import os
import sys
import glob
from shutil import copyfile
#print(sys.argv[1])
if(len(sys.argv) < 2): # No extra arguments, use default input and output files
raise Exception('Input file or folder not specified.')
curdir = sys.argv[1]
file_list = sys.argv[2]
wasdir = 0
# Check that input exists
if(os.path.isdir(file_list)):
fileList = sorted(glob.glob(file_list + '*.wav'))
wasdir = 1
if(len(fileList) == 0):
raise Exception('Provided directory contains no .wav files')
elif(file_list.endswith('.wav')):
fileList = list()
fileList.append(file_list)
if not(os.path.isfile(fileList[0])):
raise Exception('Provided input file does not exist.')
else:
if(os.path.isfile(file_list)):
fileList = list(filter(bool,[line.rstrip('\n') for line in open(file_list)]))
else:
raise Exception('Provided input file list does not exist.')
# Copy target files to a folder
for file in fileList:
path, filename = os.path.split(file)
copyfile(file,curdir + '/tmp_data/' + filename)