public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Change from get_module_with_inserted_embed_line to insert_embed_line_to_module

Browse source 
Rather than taking in a file_name and reading it in, directly take the module and edit its code.
This commit is contained in:
Grant Sanderson 2024-12-06 11:59:18 -07:00
parent fadd045fc1
commit e270f5c3d3
1 changed files with 7 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
manimlib/config.py
View file

@ -218,36 +218,31 @@ def get_indent(code_lines: list[str], line_number: int) -> str:
return "" return ""
def get_module_with_inserted_embed_line(file_name: str, line_number: int): def insert_embed_line_to_module(module: Module, line_number: int):
""" """
This is hacky, but convenient. When user includes the argument "-e", it will try This is hacky, but convenient. When user includes the argument "-e", it will try
to recreate a file that inserts the line `self.embed()` into the end of the scene's to recreate a file that inserts the line `self.embed()` into the end of the scene's
construct method. If there is an argument passed in, it will insert the line after construct method. If there is an argument passed in, it will insert the line after
the last line in the sourcefile which includes that string. the last line in the sourcefile which includes that string.
""" """
lines = Path(file_name).read_text().splitlines() lines = inspect.getsource(module).splitlines()
# Add the relevant embed line to the code # Add the relevant embed line to the code
indent = get_indent(lines, line_number) indent = get_indent(lines, line_number)
lines.insert(line_number, indent + "self.embed()") lines.insert(line_number, indent + "self.embed()")
new_code = "\n".join(lines) new_code = "\n".join(lines)
# Load the module for the original file, then exectue the new code within
# it, which should redefined the scene to have the inserted embed line
from manimlib.reload_manager import reload_manager
module = ModuleLoader.get_module(file_name, is_during_reload=reload_manager.is_reload)
code_object = compile(new_code, module.__name__, 'exec') code_object = compile(new_code, module.__name__, 'exec')
exec(code_object, module.__dict__) exec(code_object, module.__dict__)
return module return module
def get_scene_module(args: Namespace) -> Module: def get_scene_module(args: Namespace) -> Module:
if args.embed is None: from manimlib.reload_manager import reload_manager
return ModuleLoader.get_module(args.file) module = ModuleLoader.get_module(args.file, is_during_reload=reload_manager.is_reload)
else: if args.embed:
return get_module_with_inserted_embed_line(args.file, int(args.embed)) insert_embed_line_to_module(module, int(args.embed))
return module
def load_yaml(file_path: str): def load_yaml(file_path: str):
try: try: