diff options
Diffstat (limited to 'src/test_strings.py')
| -rw-r--r-- | src/test_strings.py | 23 |
1 files changed, 11 insertions, 12 deletions
diff --git a/src/test_strings.py b/src/test_strings.py index c6bf75d..9aaad08 100644 --- a/src/test_strings.py +++ b/src/test_strings.py @@ -85,20 +85,19 @@ class Checker: else: return os.path.isfile(path) - def visit(self, result, dirname, names): - for name, path in [(n, os.path.join(dirname, n)) for n in names]: - if self.remove_this(name, path): - names.remove(name) - elif self.search_this(path): - body = open(path, 'r').read() - for expr in self.expressions: - if not expr.search(body): - msg = '%s: missing %s' % (path, repr(expr.pattern)) - result.append(msg) - def find_missing(self): result = [] - os.path.walk(self.directory, self.visit, result) + for dirpath, dirnames, filenames in os.walk(self.directory): + for fname in filenames: + fpath = os.path.join(dirpath, fname) + if self.remove_this(fname, fpath): + result.remove(fname) + elif self.search_this(fpath): + body = open(path, 'r').read() + for expr in self.expressions: + if not expr.search(body): + msg = '%s: missing %s' % (path, repr(expr.pattern)) + result.append(msg) return result class CheckUnexpandedStrings(Checker): |