dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 01:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

refactor: ensure _get_style always returns a dictionary and simplify calling code

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-09-27 19:44:14 -07:00
parent 1cd5b11b5c
commit d2e0d3c06d
1 changed files with 10 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

29
aider/io.py
View file

@ -252,26 +252,19 @@ class InputOutput:
self.console = Console(force_terminal=False, no_color=True) # non-pretty
def _get_style(self, style_dict=None):
if style_dict is None:
style_dict = {}
if self.pretty and self.user_input_color:
if style_dict is None:
style_dict = {"": self.user_input_color}
else:
style_dict.setdefault("", self.user_input_color)
# Add the completion menu styles
style_dict.update(
{
"completion-menu": (
f"bg:{self.completion_menu_bg_color} {self.completion_menu_color}"
),
"completion-menu.completion.current": (
f"bg:{self.completion_menu_current_bg_color} "
f"{self.completion_menu_current_color}"
),
}
)
style_dict.setdefault("", self.user_input_color)
style_dict.update({
"completion-menu": f"bg:{self.completion_menu_bg_color} {self.completion_menu_color}",
"completion-menu.completion.current": (
f"bg:{self.completion_menu_current_bg_color} {self.completion_menu_current_color}"
),
})
return Style.from_dict(style_dict)
else:
return None
return {}
def read_image(self, filename):
try:
@ -494,8 +487,6 @@ class InputOutput:
self.tool_output(subject, bold=True)
style = self._get_style()
if style is None:
style = {}
def is_valid_response(text):
if not text: