public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Instantiate the IPython shell without a local ns

Browse source
This commit is contained in:
Pedro Fonini 2024-10-12 22:02:43 -03:00
parent d4080b8978
commit 2dd0256af6
No known key found for this signature in database
GPG key ID: B93423E4F2184BA4
1 changed files with 10 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
manimlib/scene/scene.py
View file

@ -220,14 +220,14 @@ class Scene(object):
self.save_state()
self.show_animation_progress = show_animation_progress
# Create embedded IPython terminal to be configured
shell = InteractiveShellEmbed.instance()
# Use the locals namespace of the caller
# Create embedded IPython terminal configured to have access to
# the local namespace of the caller
caller_frame = inspect.currentframe().f_back
local_ns = dict(caller_frame.f_locals)
module = get_module(caller_frame.f_globals["__file__"])
shell = InteractiveShellEmbed(user_module=module)
# Add a few custom shortcuts
# Add a few custom shortcuts to that local namespace
local_ns = dict(caller_frame.f_locals)
local_ns.update(
play=self.play,
wait=self.wait,
@ -244,6 +244,9 @@ class Scene(object):
notouch=lambda: shell.enable_gui(None),
)
# Update the shell module with the caller's locals + shortcuts
module.__dict__.update(local_ns)
# Enables gui interactions during the embed
def inputhook(context):
while not context.input_is_ready():
@ -278,13 +281,7 @@ class Scene(object):
shell.magic(f"xmode {self.embed_exception_mode}")
# Launch shell
shell(
local_ns=local_ns,
# Pretend like we're embeding in the caller function, not here
stack_depth=2,
# Specify that the present module is the caller's, not here
module=get_module(caller_frame.f_globals["__file__"])
)
shell()
# End scene when exiting an embed
if close_scene_on_exit: