check IGNORE_FILES on get_files

This commit is contained in:
idning 2013-11-03 18:08:08 +08:00
commit 7fadbd682a

View file

@ -6,6 +6,7 @@ import math
import random import random
import logging import logging
import shutil import shutil
import fnmatch
from codecs import open from codecs import open
from collections import defaultdict from collections import defaultdict
@ -96,6 +97,12 @@ class Generator(object):
if extensions is None: if extensions is None:
extensions = tuple(self.readers.extensions) extensions = tuple(self.readers.extensions)
basename = os.path.basename(path) basename = os.path.basename(path)
#check IGNORE_FILES
ignores = self.settings['IGNORE_FILES']
if any(fnmatch.fnmatch(basename, ignore) for ignore in ignores):
return False
if extensions is False or basename.endswith(extensions): if extensions is False or basename.endswith(extensions):
return True return True
return False return False