kevinheyer/bootstrap-vz
1
0
Fork 0
mirror of https://github.com/kevingruesser/bootstrap-vz.git synced 2025-08-22 09:50:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix error mentioned in #152 (Pyro4 shouldn't be a dependency)

Browse source
This commit is contained in:
Anders Ingemann 2015-04-12 09:40:09 +02:00
parent d9adb293eb
commit f1e4d34eb9
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
bootstrapvz/base/tasklist.py
View file

@ -117,7 +117,8 @@ def get_all_tasks():
# Get a generator that returns all classes in the package
import os.path
pkg_path = os.path.normpath(os.path.join(os.path.dirname(__file__), '..'))
classes = get_all_classes(pkg_path, 'bootstrapvz.')
exclude_pkgs = ['bootstrapvz.base', 'bootstrapvz.remote']
classes = get_all_classes(pkg_path, 'bootstrapvz.', exclude_pkgs)
# lambda function to check whether a class is a task (excluding the superclass Task)
def is_task(obj):
@ -126,11 +127,12 @@ def get_all_tasks():
return filter(is_task, classes) # Only return classes that are tasks
def get_all_classes(path=None, prefix=''):
def get_all_classes(path=None, prefix='', excludes=[]):
""" Given a path to a package, this function retrieves all the classes in it
:param str path: Path to the package
:param str prefix: Name of the package followed by a dot
:param list excludes: List of str matching module names that should be ignored
:return: A generator that yields classes
:rtype: generator
:raises Exception: If a module cannot be inspected.
@ -139,10 +141,13 @@ def get_all_classes(path=None, prefix=''):
import importlib
import inspect
def walk_error(module):
raise Exception('Unable to inspect module ' + module)
def walk_error(module_name):
if not any(map(lambda excl: module_name.startswith(excl), excludes)):
raise Exception('Unable to inspect module ' + module_name)
walker = pkgutil.walk_packages([path], prefix, walk_error)
for _, module_name, _ in walker:
if any(map(lambda excl: module_name.startswith(excl), excludes)):
continue
module = importlib.import_module(module_name)
classes = inspect.getmembers(module, inspect.isclass)
for class_name, obj in classes: