Skip to content

Delete

Methods for deleting documents.

Delete

Methods for deleting documents.

Source code in src/scruby/mixins/delete.py
class Delete:
    """Methods for deleting documents."""

    @final
    @staticmethod
    async def _task_delete(
        filter_fn: Callable,
        db_root: str,
        hash_reduce_left: int,
        branch_number: int,
        class_model: Any,
    ) -> int:
        """Asynchronous task for find and delete documents.

        This method is for internal use.

        Returns:
            The number of deleted documents.
        """
        collection_name = class_model.__name__
        branch_number_as_hash: str = f"{branch_number:08x}"[hash_reduce_left:]
        separated_hash: str = "/".join(list(branch_number_as_hash))
        leaf_path = Path(
            *(
                db_root,
                collection_name,
                separated_hash,
                "leaf.json",
            ),
        )
        counter: int = 0
        if await leaf_path.exists():
            data_json: bytes = await leaf_path.read_bytes()
            data: dict[str, str] = orjson.loads(data_json) or {}
            new_state: dict[str, str] = {}
            for doc_name, doc_json in data.items():
                doc = class_model.model_validate_json(doc_json)
                if filter_fn(doc):
                    counter -= 1
                    match hash_reduce_left:
                        case 7:
                            del DocCache.cache[collection_name][branch_number_as_hash[0]][doc_name]
                        case 6:
                            del DocCache.cache[collection_name][branch_number_as_hash[0]][branch_number_as_hash[1]][
                                doc_name
                            ]
                        case 5:
                            del DocCache.cache[collection_name][branch_number_as_hash[0]][branch_number_as_hash[1]][
                                branch_number_as_hash[2]
                            ][doc_name]
                        case _ as unreachable:
                            assert_never(Never(unreachable))  # pyrefly: ignore[not-callable]
                else:
                    new_state[doc_name] = doc_json
            await leaf_path.write_bytes(orjson.dumps(new_state))
        return counter

    @final
    async def delete_many(
        self,
        filter_fn: Callable,
    ) -> int:
        """Asynchronous method for delete one or more documents matching the filter.

        Attention:
            - The search is based on the effect of a quantum loop.
            - The search effectiveness depends on the number of processor threads.

        Args:
            filter_fn (Callable): A function that execute the conditions of filtering.

        Returns:
            The number of deleted documents.
        """
        # Variable initialization
        hash_reduce_left: int = self._hash_reduce_left
        assert hash_reduce_left != 0, "Scruby.run(hash_reduce_left = 0) - Not valid for `delete_many` method."

        search_task_fn: Callable = self._task_delete
        branch_numbers: range = range(self._max_number_branch)
        db_root: str = self._db_root
        class_model: Any = self._class_model
        counter: int = 0

        # Run quantum loop
        with ThreadPoolExecutor(self._max_workers) as executor:
            futures: list[Future] = [
                executor.submit(
                    search_task_fn,
                    filter_fn,
                    db_root,
                    hash_reduce_left,
                    branch_number,
                    class_model,
                )
                for branch_number in branch_numbers
            ]
            for future in as_completed(futures):
                counter += await future.result()
        if counter < 0:
            await self._counter_documents(counter)
        return abs(counter)

delete_many(filter_fn) async

Asynchronous method for delete one or more documents matching the filter.

Attention
  • The search is based on the effect of a quantum loop.
  • The search effectiveness depends on the number of processor threads.

Parameters:

Name Type Description Default
filter_fn Callable

A function that execute the conditions of filtering.

required

Returns:

Type Description
int

The number of deleted documents.

Source code in src/scruby/mixins/delete.py
@final
async def delete_many(
    self,
    filter_fn: Callable,
) -> int:
    """Asynchronous method for delete one or more documents matching the filter.

    Attention:
        - The search is based on the effect of a quantum loop.
        - The search effectiveness depends on the number of processor threads.

    Args:
        filter_fn (Callable): A function that execute the conditions of filtering.

    Returns:
        The number of deleted documents.
    """
    # Variable initialization
    hash_reduce_left: int = self._hash_reduce_left
    assert hash_reduce_left != 0, "Scruby.run(hash_reduce_left = 0) - Not valid for `delete_many` method."

    search_task_fn: Callable = self._task_delete
    branch_numbers: range = range(self._max_number_branch)
    db_root: str = self._db_root
    class_model: Any = self._class_model
    counter: int = 0

    # Run quantum loop
    with ThreadPoolExecutor(self._max_workers) as executor:
        futures: list[Future] = [
            executor.submit(
                search_task_fn,
                filter_fn,
                db_root,
                hash_reduce_left,
                branch_number,
                class_model,
            )
            for branch_number in branch_numbers
        ]
        for future in as_completed(futures):
            counter += await future.result()
    if counter < 0:
        await self._counter_documents(counter)
    return abs(counter)