Replace halo spinner with rich

CHANGELOG.md

@@ -7,6 +7,7 @@
 
 - add function in capa/helpers to load plain and compressed JSON reports #1883 @Rohit1123
 - document Antivirus warnings and VirusTotal false positive detections #2028 @RionEV @mr-tz
+- replace Halo spinner with Rich #2086 @s-ff
 
 ### Breaking Changes
 

capa/loader.py

@@ -11,7 +11,7 @@
 from typing import Set, Dict, List, Optional
 from pathlib import Path
 
-import halo
+from rich.console import Console
 from typing_extensions import assert_never
 
 import capa.perf
@@ -176,6 +176,10 @@ def get_extractor(
  UnsupportedArchError
  UnsupportedOSError
  """
+
+ # stderr=True is used here to redirect the spinner banner to stderr, so that users can redirect capa's output.
+ console = Console(stderr=True, quiet=disable_progress)
+
  if backend == BACKEND_CAPE:
  import capa.features.extractors.cape.extractor
 
@@ -222,7 +226,7 @@ def get_extractor(
  if os_ == OS_AUTO and not is_supported_os(input_path):
  raise UnsupportedOSError()
 
- with halo.Halo(text="analyzing program", spinner="simpleDots", stream=sys.stderr, enabled=not disable_progress):
+ with console.status("Analyzing program...", spinner="dots"):
  bv: BinaryView = binaryninja.load(str(input_path))
  if bv is None:
  raise RuntimeError(f"Binary Ninja cannot open file {input_path}")
@@ -247,7 +251,7 @@ def get_extractor(
  if os_ == OS_AUTO and not is_supported_os(input_path):
  raise UnsupportedOSError()
 
- with halo.Halo(text="analyzing program", spinner="simpleDots", stream=sys.stderr, enabled=not disable_progress):
+ with console.status("Analyzing program...", spinner="dots"):
  vw = get_workspace(input_path, input_format, sigpaths)
 
  if should_save_workspace:

pyproject.toml

@@ -40,7 +40,6 @@ dependencies = [
  "wcwidth==0.2.13",
  "ida-settings==2.1.0",
  "viv-utils[flirt]==0.7.9",
- "halo==0.0.31",
  "networkx==3.1",
  "ruamel.yaml==0.18.6",
  "vivisect==1.1.1",
@@ -50,6 +49,7 @@ dependencies = [
  "dncil==1.0.2",
  "pydantic==2.7.1",
  "protobuf==5.26.1",
+ "rich==13.4.2"
 ]
 dynamic = ["version"]