forktown/Av1an
1
0
Fork 0
mirror of https://github.com/master-of-zen/Av1an.git synced 2024-11-25 02:29:40 +00:00
Code Issues Projects Releases Packages Wiki Activity

input: to be able to have several directories and files

Browse source
This commit is contained in:
sybilin 2020-12-13 16:40:13 +01:00
parent 60e972c643
commit d00f1513ac
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
Startup/file_validation.py
View file

@ -8,16 +8,19 @@ def process_inputs(inputs):
print('No input file')
exit()
if inputs[0].is_dir():
inputs = [x for x in inputs[0].iterdir() if x.suffix in (".mkv", ".mp4", ".mov", ".avi", ".flv", ".m2ts")]
input_list = []
valid = np.array([i.exists() for i in inputs])
for item in inputs:
if item.is_dir():
new_inputs = [x for x in item.iterdir() if x.suffix in (".mkv", ".mp4", ".mov", ".avi", ".flv", ".m2ts")]
input_list.extend(new_inputs)
else:
input_list.append(item)
valid = np.array([i.exists() for i in input_list])
if not all(valid):
print(f'File(s) do not exist: {", ".join([str(inputs[i]) for i in np.where(not valid)[0]])}')
print(f'File(s) do not exist: {", ".join([str(input_list[i]) for i in np.where(not valid)[0]])}')
exit()
return inputs
return input_list