Chromy/chromy/handlers/delete_collection.py

from __future__ import annotations

from chromy.chroma_functions import delete_collection, delete_data


def _parse_where_clause(where_clause: str) -> dict[str, str]:
    condition, separator, value = where_clause.partition("=")

    if separator == "":
        raise ValueError("Invalid --where value. Expected <condition>=<value>.")

    condition = condition.strip()
    value = value.strip()

    if not condition or not value:
        raise ValueError("Invalid --where value. Expected <condition>=<value>.")

    return {condition: value}


def handle_delete_collection(collection: str) -> int:
    delete_collection(collection)
    print(f"Deleted collection '{collection}'.")
    return 0


def handle_delete_records(collection: str, where_clause: str) -> int:
    where = _parse_where_clause(where_clause)
    deleted = delete_data(collection, where)
    condition, value = next(iter(where.items()))
    print(
        f"Deleted {deleted} record(s) from collection '{collection}' "
        f"where {condition}={value}."
    )
    return 0