filter_files_by_ext.py

import fnmatch
import os

MY_DIR = 'Q:\\my_dir\\sub_dir'
FILE_EXT = 'xml'

def filter_files_by_ext(dir, filter_ext):
    files = [file for file in os.listdir(dir) if fnmatch.fnmatch(file, '*.%s' % filter_ext)]
    return files


def execute():
    print(filter_files_by_ext(MY_DIR, FILE_EXT))


if __name__ == "__main__":
    execute()