This commit is contained in:
Paul Gauthier 2023-06-24 14:53:19 -07:00
parent 82e19fcf1d
commit 1754e574fd

View file

@ -32,8 +32,6 @@ class WholeFileCoder(Coder):
output = [] output = []
lines = content.splitlines(keepends=True) lines = content.splitlines(keepends=True)
allowed_to_edit = False
fname = None fname = None
new_lines = [] new_lines = []
for i, line in enumerate(lines): for i, line in enumerate(lines):
@ -53,7 +51,7 @@ class WholeFileCoder(Coder):
).splitlines() ).splitlines()
output += show_diff output += show_diff
else: else:
if allowed_to_edit: if self.allowed_to_edit(fname):
edited.add(fname) edited.add(fname)
if not self.dry_run: if not self.dry_run:
new_lines = "".join(new_lines) new_lines = "".join(new_lines)
@ -72,13 +70,6 @@ class WholeFileCoder(Coder):
raise ValueError("No filename provided before ``` block") raise ValueError("No filename provided before ``` block")
else: else:
fname = lines[i - 1].strip() fname = lines[i - 1].strip()
if mode == "update":
if self.allowed_to_edit(fname):
allowed_to_edit = True
else:
allowed_to_edit = False
elif fname: elif fname:
new_lines.append(line) new_lines.append(line)
else: else:
@ -101,7 +92,7 @@ class WholeFileCoder(Coder):
return "\n".join(output) return "\n".join(output)
if fname and allowed_to_edit: if fname and self.allowed_to_edit(fname):
edited.add(fname) edited.add(fname)
if not self.dry_run: if not self.dry_run:
new_lines = "".join(new_lines) new_lines = "".join(new_lines)