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

Have -e write over original source file, then correct

Browse source
This commit is contained in:
Grant Sanderson 2022-04-23 10:16:11 -07:00
parent 0fd8491c51
commit d733687834
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
manimlib/config.py
View file

@ -243,15 +243,15 @@ def insert_embed_line(file_name: str, scene_name: str, line_marker: str):
# Insert and write new file # Insert and write new file
if n_spaces is None: if n_spaces is None:
n_spaces = get_indent(lines[prev_line_num]) n_spaces = get_indent(lines[prev_line_num])
lines.insert(prev_line_num + 1, " " * n_spaces + "self.embed()\n") new_lines = list(lines)
alt_file = file_name.replace(".py", "_inserted_embed.py") new_lines.insert(prev_line_num + 1, " " * n_spaces + "self.embed()\n")
with open(alt_file, 'w') as fp: with open(file_name, 'w') as fp:
fp.writelines(lines) fp.writelines(new_lines)
try: try:
yield alt_file yield file_name
finally: finally:
os.remove(alt_file) with open(file_name, 'w') as fp:
fp.writelines(lines)
def get_custom_config(): def get_custom_config():