Bugfix in ANSI formatted text. Correctly handle 256/true color sequences.

Author: jonathanslenders

prompt_toolkit/formatted_text/ansi.py

@@ -159,7 +159,7 @@ def _select_graphic_rendition(self, attrs):
  n = attrs.pop()
 
  # 256 colors.
- if n == 5 and len(attrs) > 1:
+ if n == 5 and len(attrs) >= 1:
  if attr == 38:
  m = attrs.pop()
  self._color = _256_colors.get(m)
@@ -168,7 +168,7 @@ def _select_graphic_rendition(self, attrs):
  self._bgcolor = _256_colors.get(m)
 
  # True colors.
- if n == 2 and len(attrs) > 3:
+ if n == 2 and len(attrs) >= 3:
  try:
  color_str = '%02x%02x%02x' % (
  attrs.pop(), attrs.pop(), attrs.pop())

tests/test_formatted_text.py

@@ -78,6 +78,12 @@ def test_ansi_formatting():
  assert isinstance(to_formatted_text(value), FormattedText)
 
 
+def test_ansi_256_color():
+ assert to_formatted_text(ANSI('\x1b[38;5;124mtest')) == [
+ ('#af0000', 't'), ('#af0000', 'e'), ('#af0000', 's'), ('#af0000', 't')
+ ]
+
+
 def test_interpolation():
  value = Template(' {} ').format(HTML('<b>hello</b>'))