janito 1.10.0__py3-none-any.whl → 1.11.1__py3-none-any.whl

janito/agent/tools/get_file_outline/python_outline.py

@@ -2,114 +2,158 @@ import re
 from typing import List
 
 
-def parse_python_outline(lines: List[str]):
-    class_pat = re.compile(r"^(\s*)class\s+(\w+)")
-    func_pat = re.compile(r"^(\s*)def\s+(\w+)")
-    assign_pat = re.compile(r"^(\s*)([A-Za-z_][A-Za-z0-9_]*)\s*=.*")
-    main_pat = re.compile(r"^\s*if\s+__name__\s*==\s*[\'\"]__main__[\'\"]\s*:")
+def handle_assignment(idx, assign_match, outline):
+    var_name = assign_match.group(2)
+    var_type = "const" if var_name.isupper() else "var"
+    outline.append(
+        {
+            "type": var_type,
+            "name": var_name,
+            "start": idx + 1,
+            "end": idx + 1,
+            "parent": "",
+            "docstring": "",
+        }
+    )
+
+
+def handle_main(idx, outline):
+    outline.append(
+        {
+            "type": "main",
+            "name": "__main__",
+            "start": idx + 1,
+            "end": idx + 1,
+            "parent": "",
+            "docstring": "",
+        }
+    )
+
+
+def close_stack_objects(idx, indent, stack, obj_ranges):
+    while stack and indent < stack[-1][2]:
+        popped = stack.pop()
+        obj_ranges.append((popped[0], popped[1], popped[3], idx, popped[4], popped[2]))
+
+
+def close_last_top_obj(idx, last_top_obj, stack, obj_ranges):
+    if last_top_obj and last_top_obj in stack:
+        stack.remove(last_top_obj)
+        obj_ranges.append(
+            (
+                last_top_obj[0],
+                last_top_obj[1],
+                last_top_obj[3],
+                idx,
+                last_top_obj[4],
+                last_top_obj[2],
+            )
+        )
+        return None
+    return last_top_obj
+
+
+def handle_class(idx, class_match, indent, stack, last_top_obj):
+    name = class_match.group(2)
+    parent = stack[-1][1] if stack and stack[-1][0] == "class" else ""
+    obj = ("class", name, indent, idx + 1, parent)
+    stack.append(obj)
+    if indent == 0:
+        last_top_obj = obj
+    return last_top_obj
+
+
+def handle_function(idx, func_match, indent, stack, last_top_obj):
+    name = func_match.group(2)
+    parent = ""
+    for s in reversed(stack):
+        if s[0] == "class" and indent > s[2]:
+            parent = s[1]
+            break
+    obj = ("function", name, indent, idx + 1, parent)
+    stack.append(obj)
+    if indent == 0:
+        last_top_obj = obj
+    return last_top_obj
+
+
+def process_line(idx, line, regexes, stack, obj_ranges, outline, last_top_obj):
+    class_pat, func_pat, assign_pat, main_pat = regexes
+    class_match = class_pat.match(line)
+    func_match = func_pat.match(line)
+    assign_match = assign_pat.match(line)
+    indent = len(line) - len(line.lstrip())
+    # If a new top-level class or function starts, close the previous one
+    if (class_match or func_match) and indent == 0 and last_top_obj:
+        last_top_obj = close_last_top_obj(idx, last_top_obj, stack, obj_ranges)
+    if class_match:
+        last_top_obj = handle_class(idx, class_match, indent, stack, last_top_obj)
+    elif func_match:
+        last_top_obj = handle_function(idx, func_match, indent, stack, last_top_obj)
+    elif assign_match and indent == 0:
+        handle_assignment(idx, assign_match, outline)
+    main_match = main_pat.match(line)
+    if main_match:
+        handle_main(idx, outline)
+    close_stack_objects(idx, indent, stack, obj_ranges)
+    return last_top_obj
+
+
+def build_outline_entry(obj, lines, outline):
+    obj_type, name, start, end, parent, indent = obj
+    # Determine if this is a method
+    if obj_type == "function" and parent:
+        outline_type = "method"
+    elif obj_type == "function":
+        outline_type = "function"
+    else:
+        outline_type = obj_type
+    docstring = extract_docstring(lines, start, end)
+    outline.append(
+        {
+            "type": outline_type,
+            "name": name,
+            "start": start,
+            "end": end,
+            "parent": parent,
+            "docstring": docstring,
+        }
+    )
+
+
+def process_lines(lines, regexes):
     outline = []
-    stack = []  # (type, name, indent, start, parent)
-    obj_ranges = []  # (type, name, start, end, parent, indent)
+    stack = []
+    obj_ranges = []
     last_top_obj = None
     for idx, line in enumerate(lines):
-        class_match = class_pat.match(line)
-        func_match = func_pat.match(line)
-        assign_match = assign_pat.match(line)
-        indent = len(line) - len(line.lstrip())
-        # If a new top-level class or function starts, close the previous one
-        if (class_match or func_match) and indent == 0 and last_top_obj:
-            # Only close if still open
-            if last_top_obj in stack:
-                stack.remove(last_top_obj)
-                obj_ranges.append(
-                    (
-                        last_top_obj[0],
-                        last_top_obj[1],
-                        last_top_obj[3],
-                        idx,
-                        last_top_obj[4],
-                        last_top_obj[2],
-                    )
-                )
-            last_top_obj = None
-        if class_match:
-            name = class_match.group(2)
-            parent = stack[-1][1] if stack and stack[-1][0] == "class" else ""
-            obj = ("class", name, indent, idx + 1, parent)
-            stack.append(obj)
-            if indent == 0:
-                last_top_obj = obj
-        elif func_match:
-            name = func_match.group(2)
-            parent = ""
-            for s in reversed(stack):
-                if s[0] == "class" and indent > s[2]:
-                    parent = s[1]
-                    break
-            obj = ("function", name, indent, idx + 1, parent)
-            stack.append(obj)
-            if indent == 0:
-                last_top_obj = obj
-        elif assign_match and indent == 0:
-            var_name = assign_match.group(2)
-            var_type = "const" if var_name.isupper() else "var"
-            outline.append(
-                {
-                    "type": var_type,
-                    "name": var_name,
-                    "start": idx + 1,
-                    "end": idx + 1,
-                    "parent": "",
-                    "docstring": "",
-                }
-            )
-        main_match = main_pat.match(line)
-        if main_match:
-            outline.append(
-                {
-                    "type": "main",
-                    "name": "__main__",
-                    "start": idx + 1,
-                    "end": idx + 1,
-                    "parent": "",
-                    "docstring": "",
-                }
-            )
-        while stack and indent < stack[-1][2]:
-            popped = stack.pop()
-            obj_ranges.append(
-                (popped[0], popped[1], popped[3], idx, popped[4], popped[2])
-            )
+        last_top_obj = process_line(
+            idx, line, regexes, stack, obj_ranges, outline, last_top_obj
+        )
     # Close any remaining open objects
     for popped in stack:
         obj_ranges.append(
             (popped[0], popped[1], popped[3], len(lines), popped[4], popped[2])
         )
+    return outline, obj_ranges
+
 
-    # Now, extract docstrings for classes, functions, and methods
+def build_outline(obj_ranges, lines, outline):
     for obj in obj_ranges:
-        obj_type, name, start, end, parent, indent = obj
-        # Determine if this is a method
-        if obj_type == "function" and parent:
-            outline_type = "method"
-        elif obj_type == "function":
-            outline_type = "function"
-        else:
-            outline_type = obj_type
-        docstring = extract_docstring(lines, start, end)
-        outline.append(
-            {
-                "type": outline_type,
-                "name": name,
-                "start": start,
-                "end": end,
-                "parent": parent,
-                "docstring": docstring,
-            }
-        )
+        build_outline_entry(obj, lines, outline)
     return outline
 
 
+def parse_python_outline(lines: List[str]):
+    class_pat = re.compile(r"^(\s*)class\s+(\w+)")
+    func_pat = re.compile(r"^(\s*)def\s+(\w+)")
+    assign_pat = re.compile(r"^(\s*)([A-Za-z_][A-Za-z0-9_]*)\s*=.*")
+    main_pat = re.compile(r"^\s*if\s+__name__\s*==\s*[\'\"]__main__[\'\"]\s*:")
+    regexes = (class_pat, func_pat, assign_pat, main_pat)
+    outline, obj_ranges = process_lines(lines, regexes)
+    return build_outline(obj_ranges, lines, outline)
+
+
 def extract_docstring(lines, start_idx, end_idx):
     """Extracts a docstring from lines[start_idx:end_idx] if present."""
     for i in range(start_idx, min(end_idx, len(lines))):

janito/agent/tools/get_lines.py

@@ -22,18 +22,34 @@ class GetLinesTool(ToolBase):
             - "---\nFile: /path/to/file.py | Lines: 1-10 (of 100)\n---\n<lines...>"
             - "---\nFile: /path/to/file.py | All lines (total: 100 (all))\n---\n<all lines...>"
             - "Error reading file: <error message>"
-            - "❗ not found"
+            - "\u2757 not found"
     """
 
     def run(self, file_path: str, from_line: int = None, to_line: int = None) -> str:
         from janito.agent.tools_utils.utils import display_path
 
         disp_path = display_path(file_path)
+        self._report_read_info(disp_path, from_line, to_line)
+        try:
+            lines = self._read_file_lines(file_path)
+            selected, selected_len, total_lines = self._select_lines(
+                lines, from_line, to_line
+            )
+            self._report_success(selected_len, from_line, to_line, total_lines)
+            header = self._format_header(
+                disp_path, from_line, to_line, selected_len, total_lines
+            )
+            return header + "".join(selected)
+        except Exception as e:
+            return self._handle_read_error(e)
+
+    def _report_read_info(self, disp_path, from_line, to_line):
+        """Report the info message for reading lines."""
         if from_line and to_line:
             self.report_info(
                 ActionType.READ,
                 tr(
-                    "📖 Reading file '{disp_path}' {from_line}-{to_line}",
+                    "📖 Read file '{disp_path}' {from_line}-{to_line}",
                     disp_path=disp_path,
                     from_line=from_line,
                     to_line=to_line,
@@ -42,79 +58,92 @@ class GetLinesTool(ToolBase):
         else:
             self.report_info(
                 ActionType.READ,
-                tr("📖 Reading file '{disp_path}'", disp_path=disp_path),
+                tr("📖 Read file '{disp_path}'", disp_path=disp_path),
             )
-        try:
-            with open(file_path, "r", encoding="utf-8", errors="replace") as f:
-                lines = f.readlines()
-            selected = lines[
-                (from_line - 1 if from_line else 0) : (to_line if to_line else None)
-            ]
-            selected_len = len(selected)
-            total_lines = len(lines)
-            at_end = False
-            if from_line and to_line:
-                requested = to_line - from_line + 1
-                if to_line >= total_lines or selected_len < requested:
-                    at_end = True
-                if at_end:
-                    self.report_success(
-                        tr(
-                            " ✅ {selected_len} {line_word} (end)",
-                            selected_len=selected_len,
-                            line_word=pluralize("line", selected_len),
-                        )
-                    )
-                elif to_line < total_lines:
-                    self.report_success(
-                        tr(
-                            " ✅ {selected_len} {line_word} ({remaining} to end)",
-                            selected_len=selected_len,
-                            line_word=pluralize("line", selected_len),
-                            remaining=total_lines - to_line,
-                        )
-                    )
-            else:
+
+    def _read_file_lines(self, file_path):
+        """Read all lines from the file."""
+        with open(file_path, "r", encoding="utf-8", errors="replace") as f:
+            return f.readlines()
+
+    def _select_lines(self, lines, from_line, to_line):
+        """Select the requested lines and return them with their count and total lines."""
+        selected = lines[
+            (from_line - 1 if from_line else 0) : (to_line if to_line else None)
+        ]
+        selected_len = len(selected)
+        total_lines = len(lines)
+        return selected, selected_len, total_lines
+
+    def _report_success(self, selected_len, from_line, to_line, total_lines):
+        """Report the success message after reading lines."""
+        if from_line and to_line:
+            requested = to_line - from_line + 1
+            at_end = to_line >= total_lines or selected_len < requested
+            if at_end:
                 self.report_success(
                     tr(
-                        " ✅ {selected_len} {line_word} (all)",
+                        " \u2705 {selected_len} {line_word} (end)",
                         selected_len=selected_len,
                         line_word=pluralize("line", selected_len),
                     )
                 )
-            if from_line and to_line:
-                if to_line >= total_lines or selected_len < (to_line - from_line + 1):
-                    header = tr(
-                        "---\n{disp_path} {from_line}-{to_line} (end)\n---\n",
-                        disp_path=disp_path,
-                        from_line=from_line,
-                        to_line=to_line,
-                    )
-                else:
-                    header = tr(
-                        "---\n{disp_path} {from_line}-{to_line} (of {total_lines})\n---\n",
-                        disp_path=disp_path,
-                        from_line=from_line,
-                        to_line=to_line,
-                        total_lines=total_lines,
+            elif to_line < total_lines:
+                self.report_success(
+                    tr(
+                        " \u2705 {selected_len} {line_word} ({remaining} to end)",
+                        selected_len=selected_len,
+                        line_word=pluralize("line", selected_len),
+                        remaining=total_lines - to_line,
                     )
-            elif from_line:
-                header = tr(
-                    "---\n{disp_path} {from_line}-END (of {total_lines})\n---\n",
+                )
+        else:
+            self.report_success(
+                tr(
+                    " \u2705 {selected_len} {line_word} (all)",
+                    selected_len=selected_len,
+                    line_word=pluralize("line", selected_len),
+                )
+            )
+
+    def _format_header(self, disp_path, from_line, to_line, selected_len, total_lines):
+        """Format the header for the output."""
+        if from_line and to_line:
+            requested = to_line - from_line + 1
+            at_end = selected_len < requested or to_line >= total_lines
+            if at_end:
+                return tr(
+                    "---\n{disp_path} {from_line}-{to_line} (end)\n---\n",
                     disp_path=disp_path,
                     from_line=from_line,
-                    total_lines=total_lines,
+                    to_line=to_line,
                 )
             else:
-                header = tr(
-                    "---\n{disp_path} All lines (total: {total_lines} (all))\n---\n",
+                return tr(
+                    "---\n{disp_path} {from_line}-{to_line} (of {total_lines})\n---\n",
                     disp_path=disp_path,
+                    from_line=from_line,
+                    to_line=to_line,
                     total_lines=total_lines,
                 )
-            return header + "".join(selected)
-        except Exception as e:
-            if isinstance(e, FileNotFoundError):
-                self.report_error(tr("❗ not found"))
-                return tr("❗ not found")
-            self.report_error(tr(" ❌ Error: {error}", error=e))
-            return tr("Error reading file: {error}", error=e)
+        elif from_line:
+            return tr(
+                "---\n{disp_path} {from_line}-END (of {total_lines})\n---\n",
+                disp_path=disp_path,
+                from_line=from_line,
+                total_lines=total_lines,
+            )
+        else:
+            return tr(
+                "---\n{disp_path} All lines (total: {total_lines} (all))\n---\n",
+                disp_path=disp_path,
+                total_lines=total_lines,
+            )
+
+    def _handle_read_error(self, e):
+        """Handle file read errors and report appropriately."""
+        if isinstance(e, FileNotFoundError):
+            self.report_error(tr("\u2757 not found"))
+            return tr("\u2757 not found")
+        self.report_error(tr(" \u274c Error: {error}", error=e))
+        return tr("Error reading file: {error}", error=e)

janito/agent/tools/move_file.py

@@ -1,8 +1,6 @@
 import os
 import shutil
 from janito.agent.tool_registry import register_tool
-
-# from janito.agent.tools_utils.expand_path import expand_path
 from janito.agent.tools_utils.utils import display_path
 from janito.agent.tool_base import ToolBase
 from janito.agent.tools_utils.action_type import ActionType
@@ -32,16 +30,55 @@ class MoveFileTool(ToolBase):
     ) -> str:
         original_src = src_path
         original_dest = dest_path
-        src = src_path  # Using src_path as is
-        dest = dest_path  # Using dest_path as is
+        src = src_path
+        dest = dest_path
         disp_src = display_path(original_src)
         disp_dest = display_path(original_dest)
         backup_path = None
+
+        valid, is_src_file, is_src_dir, err_msg = self._validate_source(src, disp_src)
+        if not valid:
+            return err_msg
+
+        dest_result = self._handle_destination(dest, disp_dest, overwrite, backup)
+        if dest_result is not None:
+            backup_path, err_msg = dest_result
+            if err_msg:
+                return err_msg
+
+        try:
+            self.report_info(
+                ActionType.WRITE,
+                tr(
+                    "📝 Moving from '{disp_src}' to '{disp_dest}' ...",
+                    disp_src=disp_src,
+                    disp_dest=disp_dest,
+                ),
+            )
+            shutil.move(src, dest)
+            self.report_success(tr("✅ Move complete."))
+            msg = tr("✅ Move complete.")
+            if backup_path:
+                msg += tr(
+                    " (backup at {backup_disp})",
+                    backup_disp=display_path(backup_path),
+                )
+            return msg
+        except Exception as e:
+            self.report_error(tr("❌ Error moving: {error}", error=e))
+            return tr("❌ Error moving: {error}", error=e)
+
+    def _validate_source(self, src, disp_src):
         if not os.path.exists(src):
             self.report_error(
                 tr("❌ Source '{disp_src}' does not exist.", disp_src=disp_src)
             )
-            return tr("❌ Source '{disp_src}' does not exist.", disp_src=disp_src)
+            return (
+                False,
+                False,
+                False,
+                tr("❌ Source '{disp_src}' does not exist.", disp_src=disp_src),
+            )
         is_src_file = os.path.isfile(src)
         is_src_dir = os.path.isdir(src)
         if not (is_src_file or is_src_dir):
@@ -51,10 +88,19 @@ class MoveFileTool(ToolBase):
                     disp_src=disp_src,
                 )
             )
-            return tr(
-                "❌ Source path '{disp_src}' is neither a file nor a directory.",
-                disp_src=disp_src,
+            return (
+                False,
+                False,
+                False,
+                tr(
+                    "❌ Source path '{disp_src}' is neither a file nor a directory.",
+                    disp_src=disp_src,
+                ),
             )
+        return True, is_src_file, is_src_dir, None
+
+    def _handle_destination(self, dest, disp_dest, overwrite, backup):
+        backup_path = None
         if os.path.exists(dest):
             if not overwrite:
                 self.report_error(
@@ -63,7 +109,7 @@ class MoveFileTool(ToolBase):
                         disp_dest=disp_dest,
                     )
                 )
-                return tr(
+                return None, tr(
                     "❗ Destination '{disp_dest}' already exists and overwrite is False.",
                     disp_dest=disp_dest,
                 )
@@ -83,27 +129,7 @@ class MoveFileTool(ToolBase):
                 self.report_error(
                     tr("❌ Error removing destination before move: {error}", error=e)
                 )
-                return tr("❌ Error removing destination before move: {error}", error=e)
-        try:
-            self.report_info(
-                ActionType.WRITE,
-                tr(
-                    "📝 Moving from '{disp_src}' to '{disp_dest}' ...",
-                    disp_src=disp_src,
-                    disp_dest=disp_dest,
-                ),
-            )
-            shutil.move(src, dest)
-            self.report_success(tr("✅ Move complete."))
-            msg = tr("✅ Move complete.")
-            if backup_path:
-                msg += tr(
-                    " (backup at {backup_disp})",
-                    backup_disp=display_path(
-                        original_dest + (".bak" if is_src_file else ".bak.zip")
-                    ),
+                return None, tr(
+                    "❌ Error removing destination before move: {error}", error=e
                 )
-            return msg
-        except Exception as e:
-            self.report_error(tr("❌ Error moving: {error}", error=e))
-            return tr("❌ Error moving: {error}", error=e)
+        return backup_path, None

janito/agent/tools/open_url.py

@@ -0,0 +1,31 @@
+import webbrowser
+from janito.agent.tool_registry import register_tool
+from janito.agent.tool_base import ToolBase
+from janito.agent.tools_utils.action_type import ActionType
+from janito.i18n import tr
+
+
+@register_tool(name="open_url")
+class OpenUrlTool(ToolBase):
+    """
+    Open the supplied URL in the default web browser.
+    Args:
+        url (str): The URL to open.
+    Returns:
+        str: Status message indicating the result.
+    """
+
+    def run(self, url: str) -> str:
+        if not url.strip():
+            self.report_warning(tr("ℹ️ Empty URL provided."))
+            return tr("Warning: Empty URL provided. Operation skipped.")
+        self.report_info(ActionType.READ, tr("🌐 Opening URL '{url}' ...", url=url))
+        try:
+            webbrowser.open(url)
+        except Exception as err:
+            self.report_error(
+                tr("❗ Error opening URL: {url}: {err}", url=url, err=str(err))
+            )
+            return tr("Warning: Error opening URL: {url}: {err}", url=url, err=str(err))
+        self.report_success(tr("✅ URL opened in browser: {url}", url=url))
+        return tr("URL opened in browser: {url}", url=url)