diff --git a/preprocessor/preprocessor/steps/output.py b/preprocessor/preprocessor/steps/output.py index 5d093c6f24b4e55c485792f42749335af2d803b6..34627278659e77c43f4ed31e844fc6feeaf638da 100644 --- a/preprocessor/preprocessor/steps/output.py +++ b/preprocessor/preprocessor/steps/output.py @@ -1,9 +1,13 @@ import os from os.path import join, basename -from glob import glob from uuid import uuid4 +from fnmatch import translate +import re from ..util import replace_ext, gdal +import logging + +logger = logging.getLogger(__name__) def output_step(source_dir: os.PathLike, target_dir: os.PathLike, options: dict=None): @@ -15,8 +19,9 @@ def output_step(source_dir: os.PathLike, target_dir: os.PathLike, options: dict= raise ValueError('Unsupported driver %s' % frmt) extension = driver.GetMetadata().get('DMD_EXTENSIONS', 'tif').split(' ')[0] # warp each individual file + match = re.compile(translate('*.tif'), re.IGNORECASE).match warped_files = [] - for filename in glob(join(source_dir, '*.tif')): + for filename in [join(source_dir, fnm) for fnm in os.listdir(source_dir) if match(fnm)]: target_filename = join(target_dir, replace_ext(basename(filename), extension)) gdal.Warp(target_filename, filename, options=gdal.WarpOptions( **options