Use os.path.relpath in place of string.replace

There is a standard library function for that!

Change-Id: If1de6e38c35c9c5f0fa28b756724e91daf12a776
This commit is contained in:
Radomir Dopieralski 2016-11-30 11:40:03 +01:00
parent 2394397bfd
commit 255de3d361

View File

@ -28,7 +28,7 @@ def discover_files(base_path, sub_path='', ext='', trim_base_path=False):
file_list = []
for root, dirs, files in walk(path.join(base_path, sub_path)):
if trim_base_path:
root = root.replace(base_path, '', 1)
root = path.relpath(root, base_path)
file_list.extend([path.join(root, file_name)
for file_name in files
if file_name.endswith(ext)])