38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from rich import print
|
|
|
|
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"[bold green]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"[bold green]Deleted[/] {deleted} record(s) from collection '{collection}' "
|
|
f"where {condition}={value}."
|
|
)
|
|
return 0
|