only packages, read only mode, full reset

Author: Peter-Sh

src/redis_release/bht/args.py

@@ -10,4 +10,4 @@ class ReleaseArgs(BaseModel):
 
     release_tag: str
     force_rebuild: List[str] = Field(default_factory=list)
-
+    only_packages: List[str] = Field(default_factory=list)

src/redis_release/bht/state.py

@@ -245,6 +245,7 @@ def __init__(
         storage: StateStorage,
         config: Config,
         args: "ReleaseArgs",
+        read_only: bool = False,
     ):
         self.tag = args.release_tag
         self.storage = storage
@@ -253,8 +254,11 @@ def __init__(
         self.last_dump: Optional[str] = None
         self._state: Optional[ReleaseState] = None
         self._lock_acquired = False
+        self.read_only = read_only
 
     def __enter__(self) -> "StateSyncer":
+        if self.read_only:
+            return self
         """Acquire lock when entering context."""
         if not self.storage.acquire_lock(self.tag):
             raise RuntimeError(f"Failed to acquire lock for tag: {self.tag}")
@@ -263,6 +267,8 @@ def __enter__(self) -> "StateSyncer":
         return self
 
     def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        if self.read_only:
+            return
         """Release lock when exiting context."""
         if self._lock_acquired:
             self.storage.release_lock(self.tag)
@@ -273,7 +279,15 @@ def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
     @property
     def state(self) -> ReleaseState:
         if self._state is None:
-            loaded = self.load()
+            loaded = None
+            if self.args.force_rebuild and "all" in self.args.force_rebuild:
+                logger.info(
+                    "Force rebuild 'all' enabled, using default state based on config"
+                )
+                loaded = self.default_state()
+            else:
+                loaded = self.load()
+
             if loaded is None:
                 self._state = self.default_state()
             else:
@@ -315,6 +329,8 @@ def load(self) -> Optional[ReleaseState]:
 
     def sync(self) -> None:
         """Save state to storage backend if changed since last sync."""
+        if self.read_only:
+            raise RuntimeError("Cannot sync read-only state")
         current_dump = self.state.model_dump_json(indent=2)
 
         if current_dump != self.last_dump:

src/redis_release/bht/tree.py

@@ -11,7 +11,6 @@
 from py_trees.display import unicode_tree
 from py_trees.trees import BehaviourTree
 from py_trees.visitors import SnapshotVisitor
-from rich.pretty import pretty_repr
 from rich.text import Text
 
 from ..config import Config
@@ -52,6 +51,125 @@
 logger = logging.getLogger(__name__)
 
 
+class TreeInspector:
+    """Inspector for creating and inspecting behavior tree branches and PPAs."""
+
+    # List of available branch/PPA names
+    AVAILABLE_NAMES = [
+        "workflow_success",
+        "workflow_completion",
+        "find_workflow",
+        "trigger_workflow",
+        "identify_target_ref",
+        "download_artifacts",
+        "extract_artifact_result",
+        "workflow_complete_branch",
+        "workflow_with_result_branch",
+        "publish_workflow_branch",
+        "build_workflow_branch",
+        "demo_sequence",
+        "demo_selector",
+    ]
+
+    def __init__(self, release_tag: str):
+        """Initialize TreeInspector.
+
+        Args:
+            release_tag: Release tag for creating mock ReleaseMeta
+        """
+        self.release_tag = release_tag
+
+    def get_names(self) -> List[str]:
+        """Get list of available branch/PPA names.
+
+        Returns:
+            List of available names that can be passed to create_by_name()
+        """
+        return self.AVAILABLE_NAMES.copy()
+
+    def create_by_name(self, name: str) -> Union[Selector, Sequence, Behaviour]:
+        """Create a branch or PPA by name.
+
+        Args:
+            name: Name of the branch or PPA to create
+
+        Returns:
+            The created behavior tree branch or PPA
+
+        Raises:
+            ValueError: If the name is not found in the available branches
+        """
+        if name not in self.AVAILABLE_NAMES:
+            available = ", ".join(self.get_names())
+            raise ValueError(f"Unknown name '{name}'. Available options: {available}")
+
+        # Create mock objects for PPA/branch creation
+        workflow = Workflow(workflow_file="test.yml", inputs={})
+        package_meta = PackageMeta(repo="redis/redis", ref="main")
+        release_meta = ReleaseMeta(tag=self.release_tag)
+        github_client = GitHubClientAsync(token="dummy")
+        package = Package(
+            meta=package_meta,
+            build=workflow,
+            publish=Workflow(workflow_file="publish.yml", inputs={}),
+        )
+        log_prefix = "test"
+
+        # Create and return the requested branch/PPA
+        if name == "workflow_success":
+            return create_workflow_success_ppa(workflow, log_prefix)
+        elif name == "workflow_completion":
+            return create_workflow_completion_ppa(
+                workflow, package_meta, github_client, log_prefix
+            )
+        elif name == "find_workflow":
+            return create_find_workflow_by_uuid_ppa(
+                workflow, package_meta, github_client, log_prefix
+            )
+        elif name == "trigger_workflow":
+            return create_trigger_workflow_ppa(
+                workflow, package_meta, release_meta, github_client, log_prefix
+            )
+        elif name == "identify_target_ref":
+            return create_identify_target_ref_ppa(
+                package_meta, release_meta, github_client, log_prefix
+            )
+        elif name == "download_artifacts":
+            return create_download_artifacts_ppa(
+                workflow, package_meta, github_client, log_prefix
+            )
+        elif name == "extract_artifact_result":
+            return create_extract_artifact_result_ppa(
+                "test-artifact", workflow, package_meta, github_client, log_prefix
+            )
+        elif name == "workflow_complete_branch":
+            return create_workflow_complete_tree_branch(
+                workflow, package_meta, release_meta, github_client, ""
+            )
+        elif name == "workflow_with_result_branch":
+            return create_workflow_with_result_tree_branch(
+                "artifact", workflow, package_meta, release_meta, github_client, ""
+            )
+        elif name == "publish_workflow_branch":
+            return create_publish_workflow_tree_branch(
+                workflow,
+                workflow,
+                package_meta,
+                release_meta,
+                workflow,
+                github_client,
+                "",
+            )
+        elif name == "build_workflow_branch":
+            return create_build_workflow_tree_branch(
+                package, release_meta, package, github_client, ""
+            )
+        elif name == "demo_sequence":
+            return create_sequence_branch()
+        else:  # name == "demo_selector"
+            return create_selector_branch()
+
+
 async def async_tick_tock(tree: BehaviourTree, cutoff: int = 100) -> None:
     """Drive Behaviour tree using async event loop
 
@@ -98,6 +216,7 @@ def initialize_tree_and_state(
     config: Config,
     args: ReleaseArgs,
     storage: Optional[StateStorage] = None,
+    read_only: bool = False,
 ) -> Iterator[Tuple[BehaviourTree, StateSyncer]]:
     github_client = GitHubClientAsync(token=os.getenv("GITHUB_TOKEN"))
 
@@ -109,9 +228,13 @@ def initialize_tree_and_state(
         storage=storage,
         config=config,
         args=args,
+        read_only=read_only,
     ) as state_syncer:
         root = create_root_node(
-            state_syncer.state, state_syncer.default_state(), github_client
+            state_syncer.state,
+            state_syncer.default_state(),
+            github_client,
+            args.only_packages,
         )
         tree = BehaviourTree(root)
 
@@ -149,14 +272,20 @@ def log_tree_state_with_markup(tree: BehaviourTree) -> None:
 
 
 def create_root_node(
-    state: ReleaseState, default_state: ReleaseState, github_client: GitHubClientAsync
+    state: ReleaseState,
+    default_state: ReleaseState,
+    github_client: GitHubClientAsync,
+    only_packages: Optional[List[str]] = None,
 ) -> Behaviour:
 
     root = ParallelBarrier(
         "Redis Release",
         children=[],
     )
     for package_name, package in state.packages.items():
+        if only_packages and package_name not in only_packages:
+            logger.info(f"Skipping package {package_name} as it's not in only_packages")
+            continue
         root.add_child(
             create_package_release_tree_branch(
                 package,