kmeans_rs/py/kmeans_rs/__init__.py

import os
import sys
from importlib.metadata import version
from pathlib import Path

os.environ["RUST_BACKTRACE"] = "full"
os.environ["COLORBT_SHOW_HIDDEN"] = "1"

from .kmeans_rs import *  # noqa

try:
    __version__ = version(Path(__file__).parent.name)
except (Exception,):
    __version__ = "unknown"

try:
    with open(Path(__file__).parent.parent / ".git" / "HEAD") as g:
        head = g.read().split(":")[1].strip()
    with open(Path(__file__).parent.parent / ".git" / head) as h:
        __git_commit_hash__ = h.read().rstrip("\n")
except (Exception,):
    __git_commit_hash__ = "unknown"


def kmeans_generate_stub():
    if len(sys.argv) > 1:
        path = Path(sys.argv[1]).resolve()
    else:
        path = Path.cwd().resolve()
    if (path / "py" / "kmeans_rs" / "__init__.py").exists():
        generate_stub(str(path))  # noqa
    else:
        raise ModuleNotFoundError(str(path / "py" / "kmeans_rs" / "__init__.py"))