Inserted "self.embed" line should match passed in line number

This commit is contained in:
Grant Sanderson 2021-12-13 16:03:36 -08:00
parent 7023548ec6
commit b81f244c3c

View file

@ -185,7 +185,7 @@ def insert_embed_line(file_name, lineno):
lines = fp.readlines()
line = lines[lineno - 1]
n_spaces = len(line) - len(line.lstrip())
lines.insert(lineno - 1, " " * n_spaces + "self.embed()\n")
lines.insert(lineno, " " * n_spaces + "self.embed()\n")
alt_file = file_name.replace(".py", "_inserted_embed.py")
with open(alt_file, 'w') as fp: