Update extract_scene.py

This commit is contained in:
zavden 2019-07-24 18:01:12 -05:00 committed by GitHub
parent 38432a73e7
commit 3cee3de94f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -64,6 +64,8 @@ def is_child_scene(obj, module):
return False return False
if obj == Scene: if obj == Scene:
return False return False
if not obj.__module__.startswith(module.__name__):
return False
return True return True
@ -118,13 +120,16 @@ def get_scenes_to_render(scene_classes, config):
def get_scene_classes_from_module(module): def get_scene_classes_from_module(module):
return [ if hasattr(module, "SCENES_IN_ORDER"):
member[1] return module.SCENES_IN_ORDER
for member in inspect.getmembers( else:
module, return [
lambda x: is_child_scene(x, module) member[1]
) for member in inspect.getmembers(
] module,
lambda x: is_child_scene(x, module)
)
]
def main(config): def main(config):