Function get_dependency_graph

packages/pytest_changed/__init__.py:101–151

Get dependency graph from ruff analyze graph. Args: repo_root: Root of the repository direction: Either "dependencies" or "dependents" Returns: Dictionary mapping file paths to list of related file paths

(
    repo_root: Path, direction: str = "dependents"
)

def get_dependency_graph(
    repo_root: Path, direction: str = "dependents"
) -> dict[str, list[str]]:
    """
    Get dependency graph from ruff analyze graph.

    Args:
        repo_root: Root of the repository
        direction: Either "dependencies" or "dependents"

    Returns:
        Dictionary mapping file paths to list of related file paths
    """
    try:
        result = subprocess.run(
            [
                "uvx",
                # uv notes `analyze graph` is experimental,
                # so we fix the ruff version for now
                # Follow project's UV_EXCLUDE_NEWER setting to avoid resolution issues
                "ruff@0.15.18",
                "analyze",
                "graph",
                "--detect-string-imports",
                "--direction",
                direction,
                ".",
            ],
            cwd=repo_root,
            capture_output=True,
            text=True,
            check=True,
        )

        # Parse JSON output
        graph: dict[str, list[str]] = json.loads(result.stdout)

        # Convert to absolute paths
        absolute_graph = {}
        for file_path, dependencies in graph.items():
            abs_path = str(repo_root / file_path)
            abs_deps = [str(repo_root / dep) for dep in dependencies]
            absolute_graph[abs_path] = abs_deps

        return absolute_graph
    except subprocess.CalledProcessError as e:
        pytest.exit(f"Failed to run ruff analyze graph: {e}", returncode=1)
    except json.JSONDecodeError as e:
        pytest.exit(f"Failed to parse ruff output: {e}", returncode=1)

    raise UnreachableError("Failed to run ruff analyze graph")


def find_affected_files(