41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
"""
|
|
tree.py — Tree object interface for Trove.
|
|
|
|
A tree is a named mapping of entries to UUIDs, serialized as UTF-8 JSON.
|
|
Corresponds to a 'tree' row in the objects table; this class handles
|
|
only the data field — storage is the caller's concern.
|
|
"""
|
|
|
|
import json
|
|
|
|
|
|
class Tree:
|
|
def __init__(self, data: bytes | None = None):
|
|
"""
|
|
Initialize a Tree. If data is provided, deserialize from UTF-8 JSON bytes.
|
|
An empty Tree is created if data is None.
|
|
"""
|
|
if not data:
|
|
self._entries: dict[str, int] = {}
|
|
else:
|
|
self._entries = json.loads(data.decode("utf-8"))
|
|
|
|
def serialize(self) -> bytes:
|
|
"""Serialize the tree to UTF-8 JSON bytes."""
|
|
return json.dumps(self._entries).encode("utf-8")
|
|
|
|
def set_entry(self, name: str, object_id: int) -> None:
|
|
"""Add or update an entry mapping name -> uuid."""
|
|
self._entries[name] = object_id
|
|
|
|
def get_entry(self, name: str) -> int:
|
|
"""Get the uuid associated with a name, or raise KeyError if not found."""
|
|
return self._entries[name]
|
|
|
|
def rm_entry(self, name: str) -> None:
|
|
"""Remove an entry by name. Raises KeyError if not found."""
|
|
del self._entries[name]
|
|
|
|
def list(self) -> dict[str, int]:
|
|
"""Return a shallow copy of all entries as {name: uuid}."""
|
|
return dict(self._entries)
|