WIP: Add prompt_toolkit style and use it in prompt.

This commit is contained in:
Paul Gauthier 2023-05-09 09:54:50 -07:00
parent 240d1fa727
commit 85fb17710f

View file

@ -7,9 +7,11 @@ import traceback
from prompt_toolkit import prompt from prompt_toolkit import prompt
from prompt_toolkit.completion import Completer, Completion from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.history import FileHistory from prompt_toolkit.history import FileHistory
from prompt_toolkit.styles import Style
from rich.console import Console from rich.console import Console
from rich.prompt import Confirm, Prompt from rich.prompt import Confirm, Prompt
from colorama import Style
from rich.live import Live from rich.live import Live
from rich.text import Text from rich.text import Text
from rich.markdown import Markdown from rich.markdown import Markdown
@ -148,17 +150,21 @@ class Coder:
inp = "" inp = ""
multiline_input = False multiline_input = False
style = Style.from_dict({'': 'green'})
while True: while True:
try:
completer_instance = FileContentCompleter(self.fnames) completer_instance = FileContentCompleter(self.fnames)
if multiline_input: if multiline_input:
show = ". " show = ". "
else: else:
show = "> " show = "> "
try:
line = prompt( line = prompt(
show, show,
completer=completer_instance, completer=completer_instance,
history=FileHistory(self.history_file), history=FileHistory(self.history_file),
style=style,
) )
except EOFError: except EOFError:
return return
@ -209,9 +215,6 @@ class Coder:
break break
self.console.print("[bold red]^C again to quit") self.console.print("[bold red]^C again to quit")
if self.pretty:
print(Style.RESET_ALL)
def run_loop(self): def run_loop(self):
inp = self.get_input() inp = self.get_input()
if inp is None: if inp is None: