dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 18:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

refactor: simplify _get_style method in InputOutput class

Browse source
This commit is contained in:
Paul Gauthier 2024-09-27 19:47:40 -07:00 • committed by Paul Gauthier (aider)
parent d3908ca971
commit ede3cbb372
1 changed files with 20 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

34
aider/io.py
View file

@ -251,26 +251,32 @@ class InputOutput:
else: else:
self.console = Console(force_terminal=False, no_color=True) # non-pretty self.console = Console(force_terminal=False, no_color=True) # non-pretty
def _get_style(self, style_dict=None): def _get_style(self):
if style_dict is None: style_dict = {}
style_dict = {} if not self.pretty:
if self.pretty and self.user_input_color: return Style.from_dict(style_dict)
if self.user_input_color:
style_dict.setdefault("", self.user_input_color) style_dict.setdefault("", self.user_input_color)
style_dict.update( style_dict.update(
{ {
"pygments.literal.string": f"bold italic {self.user_input_color}", "pygments.literal.string": f"bold italic {self.user_input_color}",
"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}"
),
} }
) )
return Style.from_dict(style_dict)
else: style_dict.update(
return {} {
"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}"
),
}
)
return Style.from_dict(style_dict)
def read_image(self, filename): def read_image(self, filename):
try: try: