utils.py 文件源码

def get_classes(self, folder, class_suffix, selected = None):
        classes = []
        for file in glob.glob(folder + "/*.py"):
            handle = open(file, "r")
            content = handle.read()
            handle.close()

            module = folder.replace('/', '.').replace('\\', '.') + '.' + os.path.basename(file).replace('.py', '')

            regexp = "\sclass\s+([\w\d]+"+class_suffix+")\s*\(([\w\d]*)\)\s*:\s"
            for m in re.finditer(regexp, content):
                parent_class = m.group(2)
                if len(parent_class) == 0 or parent_class == 'object':
                    continue
                class_name = m.group(1)
                classes.append(module + '.' + class_name)

        return classes