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

Edited insert_embed_line to write over existing file rather than creating a new one

Browse source 
Otherwise behavior of things like __file__ is not as expected.
This commit is contained in:
Grant Sanderson 2022-12-18 10:50:17 -08:00
parent 07feb33cbb
commit 99dbf6b8c3
1 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
manimlib/config.py
View file

@ -216,8 +216,9 @@ def insert_embed_line(file_name: str, scene_name: str, line_marker: str):
)
except StopIteration:
log.error(f"No scene {scene_name}")
return
prev_line_num = None
prev_line_num = -1
n_spaces = None
if len(line_marker) == 0:
# Find the end of the construct method
@ -228,10 +229,10 @@ def insert_embed_line(file_name: str, scene_name: str, line_marker: str):
in_construct = True
n_spaces = get_indent(line) + 4
elif in_construct:
if len(line.strip()) > 0 and get_indent(line) < n_spaces:
if len(line.strip()) > 0 and get_indent(line) < (n_spaces or 0):
prev_line_num = index - 1
break
if prev_line_num is None:
if prev_line_num < 0:
prev_line_num = len(lines) - 1
elif line_marker.isdigit():
# Treat the argument as a line number
@ -248,18 +249,20 @@ def insert_embed_line(file_name: str, scene_name: str, line_marker: str):
log.error(f"No lines matching {line_marker}")
sys.exit(2)
# Insert and write new file
# Insert the embed line, rewrite file, then write it back when done
if n_spaces is None:
n_spaces = get_indent(lines[prev_line_num])
inserted_line = " " * n_spaces + "self.embed()\n"
new_lines = list(lines)
new_lines.insert(prev_line_num + 1, " " * n_spaces + "self.embed()\n")
alt_file = file_name.replace(".py", "_insert_embed.py")
with open(alt_file, 'w') as fp:
new_lines.insert(prev_line_num + 1, inserted_line)
with open(file_name, 'w') as fp:
fp.writelines(new_lines)
try:
yield alt_file
yield file_name
finally:
os.remove(alt_file)
with open(file_name, 'w') as fp:
fp.writelines(lines)
def get_custom_config():