ethanolivertroy_obsidian-markitdown/python/install_package.py

#!/usr/bin/env python3
"""Install a Python package using pip with progress output.

Usage:
    python install_package.py --package "markitdown[all]"

Output:
    Streams pip output line by line to stdout.
    Final line is JSON: {"success": true} or {"success": false, "error": "message"}
"""

import argparse
import json
import subprocess
import sys


def main():
    parser = argparse.ArgumentParser(description="Install a Python package via pip")
    parser.add_argument(
        "--package",
        required=True,
        help="Package specification (e.g., 'markitdown[all]')",
    )
    args = parser.parse_args()

    try:
        process = subprocess.Popen(
            [sys.executable, "-m", "pip", "install", args.package],
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
            text=True,
            bufsize=1,
        )

        for line in process.stdout:
            print(line, end="", flush=True)

        process.wait()

        if process.returncode == 0:
            print(json.dumps({"success": True}))
        else:
            print(
                json.dumps(
                    {
                        "success": False,
                        "error": f"pip exited with code {process.returncode}",
                    }
                )
            )
            sys.exit(1)

    except Exception as e:
        print(json.dumps({"success": False, "error": str(e)}))
        sys.exit(1)


if __name__ == "__main__":
    main()