Support for 24bit true color.

Author: jonathanslenders

examples/ptpython_config/config.py

@@ -97,6 +97,10 @@ def configure(repl):
     # Use this colorscheme for the code.
     repl.use_code_colorscheme('pastie')
 
+    # Enable 24bit True color. (Not all terminals support this. -- maybe check
+    # $TERM before changing.)
+    repl.true_color = False
+
     # Install custom colorscheme named 'my-colorscheme' and use it.
     """
     repl.install_ui_colorscheme('my-colorscheme', _custom_ui_colorscheme)

ptpython/python_input.py

@@ -19,6 +19,7 @@
 from prompt_toolkit.interface import CommandLineInterface, Application, AcceptAction
 from prompt_toolkit.key_binding.manager import KeyBindingManager
 from prompt_toolkit.layout.lexers import PygmentsLexer
+from prompt_toolkit.shortcuts import create_output
 from prompt_toolkit.styles import DynamicStyle
 from prompt_toolkit.utils import Callback, is_windows
 from prompt_toolkit.validation import ConditionalValidator
@@ -120,7 +121,7 @@ class PythonInput(object):
 
         python_input = PythonInput(...)
         application = python_input.create_application()
-        cli = CommandLineInterface(application=application)
+        cli = PythonCommandLineInterface(application=application)
         python_code = cli.run()
     """
     def __init__(self,
@@ -210,6 +211,7 @@ def __init__(self,
             self._current_code_style_name = 'win32'
 
         self._current_style = self._generate_style()
+        self.true_color = False
 
         # Options to be configurable from the sidebar.
         self.options = self._create_options()
@@ -479,6 +481,9 @@ def get_values():
                        get_values=lambda: dict(
                             (name, partial(self.use_ui_colorscheme, name)) for name in self.ui_styles)
                        ),
+                simple_option(title='True color (24 bit)',
+                              description='Use 24 bit colors instead of 265 colors',
+                              field_name='true_color'),
             ]),
         ]
 
@@ -613,8 +618,15 @@ def on_reset(self, cli):
 
 
 class PythonCommandLineInterface(CommandLineInterface):
-    def __init__(self, eventloop=None, input=None, output=None):
-        python_input = PythonInput()
+    def __init__(self, eventloop=None, python_input=None, input=None, output=None):
+        assert python_input is None or isinstance(python_input, PythonInput)
+
+        python_input = python_input or PythonInput()
+
+        # Make sure that the prompt_toolkit 'renderer' knows about the
+        # 'true_color' property of PythonInput.
+        if output is None:
+            output=create_output(true_color=Condition(lambda: python_input.true_color))
 
         super(PythonCommandLineInterface, self).__init__(
             application=python_input.create_application(),

ptpython/repl.py

@@ -13,14 +13,14 @@
 from pygments.styles.default import DefaultStyle
 
 from prompt_toolkit.application import AbortAction
+from prompt_toolkit.enums import DEFAULT_BUFFER
 from prompt_toolkit.interface import AcceptAction, CommandLineInterface
 from prompt_toolkit.layout.utils import token_list_width
 from prompt_toolkit.shortcuts import create_asyncio_eventloop
-from prompt_toolkit.utils import DummyContext, Callback
-from prompt_toolkit.enums import DEFAULT_BUFFER
 from prompt_toolkit.styles import PygmentsStyle
+from prompt_toolkit.utils import DummyContext, Callback
 
-from .python_input import PythonInput
+from .python_input import PythonInput, PythonCommandLineInterface
 from .eventloop import create_eventloop
 
 import os
@@ -291,7 +291,7 @@ def get_locals():
     if configure:
         configure(repl)
 
-    cli = CommandLineInterface(application=repl.create_application(), eventloop=eventloop)
+    cli = PythonCommandLineInterface(python_input=repl, eventloop=eventloop)
 
     # Start repl.
     patch_context = cli.patch_stdout_context() if patch_stdout else DummyContext()