WIP: Refactor Coder class to use set instead of dict for fnames.

This commit is contained in:
Paul Gauthier 2023-05-09 10:15:33 -07:00
parent 82cf653d73
commit c6197fd121

View file

@ -29,7 +29,7 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
class Coder: class Coder:
fnames = dict() fnames = set()
last_modified = 0 last_modified = 0
repo = None repo = None
@ -55,7 +55,7 @@ class Coder:
else: else:
self.console.print(f"[red]Loading {fname}") self.console.print(f"[red]Loading {fname}")
self.fnames[str(fname)] = fname.stat().st_mtime self.fnames.add(str(fname))
self.set_repo() self.set_repo()
if not self.repo: if not self.repo:
@ -333,7 +333,7 @@ class Coder:
self.console.print(f"[red]Skipping edit to {path}") self.console.print(f"[red]Skipping edit to {path}")
continue continue
self.fnames[path] = 0 self.fnames.add(path)
edited.add(path) edited.add(path)
if utils.do_replace(path, original, updated): if utils.do_replace(path, original, updated):