Class BPlusTreeMap

python/bplustree/__init__.py:19–120

Wrapper around the C extension to provide a consistent API.

else:

    class BPlusTreeMap(_c_ext.BPlusTree):
        """Wrapper around the C extension to provide a consistent API."""

        def __init__(self, capacity=None):
            """Initialize BPlusTreeMap with optional capacity."""
            if capacity is None:
                super().__init__()
            else:
                super().__init__(capacity=capacity)

        def get(self, key, default=None):
            """Get value with default."""
            try:
                return self[key]
            except KeyError:
                return default

        def values(self):
            """Return iterator over values."""
            for key, value in self.items():
                yield value

        def clear(self):
            """Remove all items from the tree."""
            # C extension doesn't have clear method, so remove keys one by one
            # Use while loop to avoid issues with iterator invalidation
            while len(self) > 0:
                # Get first key and delete it
                for key in self.keys():
                    del self[key]
                    break

        def pop(self, key, *args):
            """Remove and return value for key with optional default."""
            if len(args) > 1:
                raise TypeError(
                    f"pop expected at most 2 arguments, got {len(args) + 1}"
                )
            try:
                value = self[key]
                del self[key]
                return value
            except KeyError:
                if args:
                    return args[0]
                raise

        def popitem(self):
            """Remove and return an arbitrary (key, value) pair."""
            try:
                # Get the first key-value pair
                for key, value in self.items():
                    del self[key]
                    return (key, value)
            except:
                pass
            raise KeyError("popitem(): tree is empty")