fix: Move Spinner class to top level and remove elapsed time display

This commit is contained in:
Paul Gauthier (aider) 2024-08-05 17:25:55 -03:00
parent 17f1f2862d
commit 44f3ed6433

View file

@ -1,25 +1,10 @@
import base64
import itertools
import os
import time
from collections import defaultdict
from datetime import datetime
from pathlib import Path
class Spinner:
def __init__(self, io, text):
self.io = io
self.text = text
print(f"{self.text} {next(self.io.spinner_chars)}", end="\r", flush=True)
def step(self):
print(f"{self.text} {next(self.io.spinner_chars)}", end="\r", flush=True)
def end(self):
print(f"{self.text} Done!")
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.enums import EditingMode
from prompt_toolkit.history import FileHistory
@ -37,6 +22,19 @@ from .dump import dump # noqa: F401
from .utils import is_image_file
class Spinner:
def __init__(self, io, text):
self.io = io
self.text = text
print(f"{self.text} {next(self.io.spinner_chars)}", end="\r", flush=True)
def step(self):
print(f"{self.text} {next(self.io.spinner_chars)}", end="\r", flush=True)
def end(self):
print(f"{self.text} Done!")
class AutoCompleter(Completer):
def __init__(self, root, rel_fnames, addable_rel_fnames, commands, encoding):
self.addable_rel_fnames = addable_rel_fnames