robotframework-debug 3.5.3__tar.gz → 3.6.0__tar.gz

{robotframework-debug-3.5.3 → robotframework-debug-3.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: robotframework-debug
-Version: 3.5.3
+Version: 3.6.0
 Summary: RobotFramework debug shell
 Home-page: https://github.com/imbus/robotframework-debug/
 Author: René Rohner

{robotframework-debug-3.5.3 → robotframework-debug-3.6.0}/RobotDebug/prompttoolkitcmd.py

@@ -4,6 +4,7 @@ import re
 
 from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
 from prompt_toolkit.buffer import Buffer
+from prompt_toolkit.cursor_shapes import CursorShape
 from prompt_toolkit.history import FileHistory
 from prompt_toolkit.key_binding import KeyBindings
 from prompt_toolkit.lexers import PygmentsLexer
@@ -22,6 +23,8 @@ from pygments.lexers.robotframework import (
     _Table,
 )
 
+from .robotkeyword import get_rprompt_text
+
 
 class KeywordCall(Tokenizer):
     _tokens = (KEYWORD, ARGUMENT)
@@ -307,6 +310,9 @@ Type "help" for more information.\
         BaseCmd.__init__(self, completekey, stdin, stdout)
         self.history = FileHistory(os.path.expanduser(history_path))
 
+    def prompt_continuation(self, width, line_number, is_soft_wrap):
+        return " " * width
+
     def get_input(self):
         kwargs = {}
         if self.get_prompt_tokens:
@@ -325,6 +331,10 @@ Type "help" for more information.\
                 complete_style=CompleteStyle.COLUMN,
                 key_bindings=kb,
                 lexer=PygmentsLexer(RobotFrameworkLexer),
+                rprompt=get_rprompt_text(),
+                prompt_continuation=self.prompt_continuation,
+                mouse_support=True,
+                cursor=CursorShape.BLINKING_BEAM,
                 **kwargs
             )
         except EOFError:

{robotframework-debug-3.5.3 → robotframework-debug-3.6.0}/RobotDebug/robotkeyword.py

@@ -1,5 +1,6 @@
 import re
 import tempfile
+import time
 from pathlib import Path
 from typing import List, Tuple
 
@@ -14,6 +15,7 @@ KEYWORD_SEP = re.compile("  +|\t")
 
 _lib_keywords_cache = {}
 temp_resources = []
+last_keyword_exec_time = 0
 
 
 def parse_keyword(command) -> Tuple[List[str], str, List[str]]:
@@ -72,6 +74,8 @@ def find_keyword(keyword_name):
 
 def run_command(builtin, command: str) -> List[Tuple[str, str]]:
     """Run a command in robotframewrk environment."""
+    global last_keyword_exec_time
+    last_keyword_exec_time = 0
     if not command:
         return []
     if is_variable(command):
@@ -80,15 +84,18 @@ def run_command(builtin, command: str) -> List[Tuple[str, str]]:
     if command.startswith("***"):
         _import_resource_from_string(command)
         return []
-
     test = get_test_body_from_string(command)
     if len(test.body) > 1:
+        start = time.monotonic()
         for kw in test.body:
             kw.run(ctx)
+        last_keyword_exec_time = time.monotonic() - start
         return_val = None
     else:
         kw = test.body[0]
+        start = time.monotonic()
         return_val = kw.run(ctx)
+        last_keyword_exec_time = time.monotonic() - start
     assign = set(_get_assignments(test))
     if not assign and return_val is not None:
         return [("<", repr(return_val))]
@@ -103,6 +110,13 @@ def run_command(builtin, command: str) -> List[Tuple[str, str]]:
         return []
 
 
+def get_rprompt_text():
+    """Get text for bottom toolbar."""
+    if last_keyword_exec_time == 0:
+        return
+    return [("class:pygments.comment", f"# ΔT: {last_keyword_exec_time:.3f}s")]
+
+
 def get_test_body_from_string(command):
     if "\n" in command:
         command = "\n  ".join(command.split("\n"))

{robotframework-debug-3.5.3 → robotframework-debug-3.6.0}/RobotDebug/styles.py

@@ -1,7 +1,7 @@
 from prompt_toolkit import print_formatted_text
 from prompt_toolkit.completion import Completion
 from prompt_toolkit.formatted_text import FormattedText
-from prompt_toolkit.styles import Style, style_from_pygments_cls, merge_styles
+from prompt_toolkit.styles import Style, merge_styles, style_from_pygments_cls
 from pygments.styles import get_all_styles, get_style_by_name
 
 NORMAL_STYLE = Style.from_dict(
@@ -11,18 +11,19 @@ NORMAL_STYLE = Style.from_dict(
     }
 )
 
-ERROR_STYLE = Style.from_dict(
-    {
-        "head": "fg:red"
-    }
-)
+ERROR_STYLE = Style.from_dict({"head": "fg:red"})
 
-DEBUG_PROMPT_STYLE = merge_styles([Style.from_dict(
-    {
-        "pygments.name.function": "bold",
-        "pygments.literal.string": "italic",
-    }
-), style_from_pygments_cls(get_style_by_name("solarized-dark"))])
+DEBUG_PROMPT_STYLE = merge_styles(
+    [
+        Style.from_dict(
+            {
+                "pygments.name.function": "bold",
+                "pygments.literal.string": "italic",
+            }
+        ),
+        style_from_pygments_cls(get_style_by_name("solarized-dark")),
+    ]
+)
 
 
 def get_pygments_styles():
@@ -70,6 +71,9 @@ def _get_style_completions(text):
             start,
             display=name,
             display_meta="",
+            style=dict(style_from_pygments_cls(get_style_by_name(name)).style_rules).get(
+                "pygments.name.function"
+            ),
         )
         for name in get_pygments_styles()
         if (name.lower().strip().startswith(style_part))

robotframework-debug-3.6.0/RobotDebug/version.py

@@ -0,0 +1 @@
+VERSION = "3.6.0"

{robotframework-debug-3.5.3 → robotframework-debug-3.6.0}/robotframework_debug.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: robotframework-debug
-Version: 3.5.3
+Version: 3.6.0
 Summary: RobotFramework debug shell
 Home-page: https://github.com/imbus/robotframework-debug/
 Author: René Rohner

robotframework-debug-3.5.3/RobotDebug/version.py

@@ -1 +0,0 @@
-VERSION = "3.5.3"