From e270f5c3d3cb545f1721533241ae1859d418f663 Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Fri, 6 Dec 2024 11:59:18 -0700
Subject: [PATCH] Change from get_module_with_inserted_embed_line to
 insert_embed_line_to_module

Rather than taking in a file_name and reading it in, directly take the module and edit its code.
---
 manimlib/config.py | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/manimlib/config.py b/manimlib/config.py
index 80b5c2f7..f28bdfe0 100644
--- a/manimlib/config.py
+++ b/manimlib/config.py
@@ -218,36 +218,31 @@ def get_indent(code_lines: list[str], line_number: int) -> str:
     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
     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
     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
     indent = get_indent(lines, line_number)
     lines.insert(line_number, indent + "self.embed()")
     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')
     exec(code_object, module.__dict__)
     return module
 
 
 def get_scene_module(args: Namespace) -> Module:
-    if args.embed is None:
-        return ModuleLoader.get_module(args.file)
-    else:
-        return get_module_with_inserted_embed_line(args.file, int(args.embed))
-
+    from manimlib.reload_manager import reload_manager
+    module = ModuleLoader.get_module(args.file, is_during_reload=reload_manager.is_reload)
+    if args.embed:
+        insert_embed_line_to_module(module, int(args.embed))
+    return module
 
 def load_yaml(file_path: str):
     try: