File size: 1,076 Bytes
e8b0040 | 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 35 36 37 | import os
def get_file_name_ext(filepath):
# analyze
file_name, file_ext = os.path.splitext(filepath)
# return
return file_name, file_ext
def get_file_ext(filepath):
return get_file_name_ext(filepath)[1]
def traverse_recursively(fileroot, filepathes=[], extension='.*'):
"""Traverse all file path in specialed directory recursively.
Args:
h: crop height.
extension: e.g. '.jpg .png .bmp .webp .tif .eps'
"""
items = os.listdir(fileroot)
for item in items:
if os.path.isfile(os.path.join(fileroot, item)):
filepath = os.path.join(fileroot, item)
fileext = get_file_ext(filepath).lower()
if extension == '.*':
filepathes.append(filepath)
elif fileext in extension:
filepathes.append(filepath)
else:
pass
elif os.path.isdir(os.path.join(fileroot, item)):
traverse_recursively(os.path.join(fileroot, item), filepathes, extension)
else:
pass
|