mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
Update extract_scene.py
This commit is contained in:
parent
38432a73e7
commit
3cee3de94f
1 changed files with 12 additions and 7 deletions
|
@ -64,6 +64,8 @@ def is_child_scene(obj, module):
|
|||
return False
|
||||
if obj == Scene:
|
||||
return False
|
||||
if not obj.__module__.startswith(module.__name__):
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
|
@ -118,13 +120,16 @@ def get_scenes_to_render(scene_classes, config):
|
|||
|
||||
|
||||
def get_scene_classes_from_module(module):
|
||||
return [
|
||||
member[1]
|
||||
for member in inspect.getmembers(
|
||||
module,
|
||||
lambda x: is_child_scene(x, module)
|
||||
)
|
||||
]
|
||||
if hasattr(module, "SCENES_IN_ORDER"):
|
||||
return module.SCENES_IN_ORDER
|
||||
else:
|
||||
return [
|
||||
member[1]
|
||||
for member in inspect.getmembers(
|
||||
module,
|
||||
lambda x: is_child_scene(x, module)
|
||||
)
|
||||
]
|
||||
|
||||
|
||||
def main(config):
|
||||
|
|
Loading…
Add table
Reference in a new issue