From f406d5b81625a830c54b35acec6caf937c94c6da Mon Sep 17 00:00:00 2001
From: Kevin Van Brunt <kmvanbrunt@gmail.com>
Date: Wed, 19 Sep 2018 02:44:53 -0400
Subject: Wrote a prompt updater example

---
 cmd2/rl_utils.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'cmd2/rl_utils.py')

diff --git a/cmd2/rl_utils.py b/cmd2/rl_utils.py
index 634f0c5e..4717c408 100644
--- a/cmd2/rl_utils.py
+++ b/cmd2/rl_utils.py
@@ -139,3 +139,17 @@ def rl_get_point() -> int:
 
     else:  # pragma: no cover
         return 0
+
+
+# noinspection PyProtectedMember
+def rl_set_prompt(prompt: str) -> None:
+    """
+    Sets readline's prompt
+    :param prompt: the new prompt value
+    """
+    if rl_type == RlType.GNU:  # pragma: no cover
+        encoded_prompt = bytes(prompt, encoding='utf-8')
+        readline_lib.rl_set_prompt(encoded_prompt)
+
+    elif rl_type == RlType.PYREADLINE:  # pragma: no cover
+        readline.rl._set_prompt(prompt)
-- 
cgit v1.2.1