summaryrefslogtreecommitdiff
path: root/cmd2/rl_utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'cmd2/rl_utils.py')
-rw-r--r--cmd2/rl_utils.py14
1 files changed, 14 insertions, 0 deletions
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)