import os import re import codecs class SourceScanner(object): """ Traverses directory tree, reads all source files, and passes their contents to the Parser. """ def ScanDir(self, srcdir, parser): """ Scans provided path and passes all found contents to the parser using parser.Parse method. """ extensions1 = tuple([".h"]) extensions2 = tuple([".cpp", ".c"]) for dirname, dirnames, filenames in os.walk(srcdir): for filename in filenames: if filename.endswith(extensions1): path = os.path.join(dirname, filename) if not self.ScanFile(path, parser): return False for filename in filenames: if filename.endswith(extensions2): path = os.path.join(dirname, filename) if not self.ScanFile(path, parser): return False return True def ScanFile(self, path, parser): """ Scans provided file and passes its contents to the parser using parser.Parse method. """ prefix = "^.*" + os.path.sep + "src" + os.path.sep scope = re.sub(prefix.replace("\\", "/"), "", os.path.dirname(os.path.relpath(path)).replace("\\", "/")) with codecs.open(path, 'r', 'utf-8') as f: try: contents = f.read() except: contents = '' print('Failed reading file: %s, skipping content.' % path) pass return parser.Parse(scope, contents)