import argparse
import re
from pathlib import Path


def update_version(file_path: str, version: str) -> None:
    """Update the version string in the given __init__.py file."""
    file_path = Path(file_path)

    # Read the file and replace the version
    file_contents = file_path.read_text(encoding="utf-8")
    new_contents = re.sub(r'version = "0\.0\.0"', f'version = "{version}"', file_contents)

    # Write the updated contents back to the file
    file_path.write_text(new_contents, encoding="utf-8")

    print(f"Updated version in {file_path} to {version}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Update the version string in a Python file.")
    parser.add_argument(
        "version",
        type=str,
        help="The version string to replace '0.0.0' with (e.g., '1.2.3')."
    )
    parser.add_argument(
        "file",
        type=str,
        help="The path to the __init__.py file where the version will be updated."
    )

    args = parser.parse_args()
    update_version(args.file, args.version)