From 557bf5ae76188335c6fa10637ff0ec54ea5618b2 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Mon, 8 May 2023 10:17:42 -0700 Subject: [PATCH] prompt for new files --- coder.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/coder.py b/coder.py index d376ae213..ca652975d 100755 --- a/coder.py +++ b/coder.py @@ -6,6 +6,7 @@ import readline import traceback import argparse from rich.console import Console +from rich.prompt import Confirm from colorama import Fore, Style from rich.live import Live from rich.text import Text @@ -312,6 +313,17 @@ class Coder: for match in self.pattern.finditer(content): _, path, _, original, updated = match.groups() + if path not in self.fnames: + if not Path(path).exists(): + question = f"[red bold]Allow creation of new file {path}?" + else: + question = f"[red bold]Allow edits to {path} which was not previously provided?" + if not Confirm.ask(question, console=self.console): + self.console.print(f"[red]Skipping edit to {path}") + continue + + self.fnames[path] = 0 + edited.add(path) if self.do_replace(path, original, updated): continue