Set terminal title.

jonathanslenders · jonathanslenders · commit dff60bd0953b · 2015-06-24T21:05:47.000+02:00
diff --git a/ptpython/entry_points/run_ptipython.py b/ptpython/entry_points/run_ptipython.py
@@ -58,10 +58,11 @@ def configure(repl):
                 run_config(repl, path)
 
         # Run interactive shell.
-        embed(vi_mode=vi_mode, 
+        embed(vi_mode=vi_mode,
               history_filename=os.path.join(config_dir, 'history'),
               configure=configure,
-              user_ns=user_ns)
+              user_ns=user_ns,
+              title='IPython REPL (ptipython)')
 
 
 if __name__ == '__main__':
diff --git a/ptpython/entry_points/run_ptpython.py b/ptpython/entry_points/run_ptpython.py
@@ -65,7 +65,8 @@ def configure(repl):
         embed(vi_mode=vi_mode,
               history_filename=os.path.join(config_dir, 'history'),
               configure=configure,
-              startup_paths=startup_paths)
+              startup_paths=startup_paths,
+              title='Python REPL (ptpython)')
 
 if __name__ == '__main__':
     run()
diff --git a/ptpython/ipython.py b/ptpython/ipython.py
@@ -169,6 +169,7 @@ def __init__(self, *a, **kw):
         vi_mode = kw.pop('vi_mode', False)
         history_filename = kw.pop('history_filename', None)
         configure = kw.pop('configure', None)
+        title = kw.pop('title', None)
 
         super(InteractiveShellEmbed, self).__init__(*a, **kw)
 
@@ -181,6 +182,9 @@ def get_globals():
             get_globals=get_globals, vi_mode=vi_mode,
             history_filename=history_filename)
 
+        if title:
+            ipython_input.terminal_title = title
+
         if configure:
             configure(ipython_input)
 
diff --git a/ptpython/python_input.py b/ptpython/python_input.py
@@ -164,6 +164,7 @@ def __init__(self,
 
         self.show_sidebar = False  # Currently show the sidebar.
         self.show_exit_confirmation = False  # Currently show 'Do you really want to exit?'
+        self.terminal_title = None  # The title to be displayed in the terminal. (None or string.)
 
         #: Load styles.
         self.code_styles = get_all_code_styles()
@@ -402,6 +403,7 @@ def create_application(self):
             on_abort=AbortAction.RETRY,
             on_exit=self._on_exit,
             get_style=lambda: self._current_style,
+            get_title=lambda: self.terminal_title,
             on_start=self._on_start,
             on_input_timeout=Callback(self._on_input_timeout))
 
diff --git a/ptpython/repl.py b/ptpython/repl.py
@@ -244,7 +244,7 @@ def enter_to_continue():
 
 
 def embed(globals=None, locals=None, configure=None,
-          vi_mode=False, history_filename=None,
+          vi_mode=False, history_filename=None, title=None,
           startup_paths=None, patch_stdout=False, return_asyncio_coroutine=False):
     """
     Call this to embed  Python shell at the current point in your program.
@@ -256,6 +256,7 @@ def embed(globals=None, locals=None, configure=None,
     :param vi_mode: Boolean. Use Vi instead of Emacs key bindings.
     :param configure: Callable that will be called with the `PythonRepl` as a first
                       argument, to trigger configuration.
+    :param title: Title to be displayed in the terminal titlebar. (None or string.)
     """
     assert configure is None or callable(configure)
 
@@ -287,6 +288,9 @@ def get_locals():
                       history_filename=history_filename,
                       startup_paths=startup_paths)
 
+    if title:
+        repl.terminal_title = title
+
     if configure:
         configure(repl)
 
