apify 1.7.1b1__py3-none-any.whl → 2.2.1b1__py3-none-any.whl

apify/_memory_storage/resource_clients/dataset.py

@@ -1,452 +0,0 @@
-from __future__ import annotations
-
-import asyncio
-import json
-import os
-from datetime import datetime, timezone
-from typing import TYPE_CHECKING, Any, AsyncIterator
-
-import aioshutil
-from apify_shared.models import ListPage
-from apify_shared.utils import ignore_docs
-
-from apify._crypto import crypto_random_object_id
-from apify._memory_storage.file_storage_utils import _update_dataset_items, update_metadata
-from apify._memory_storage.resource_clients.base_resource_client import BaseResourceClient
-from apify._utils import force_rename, raise_on_duplicate_storage, raise_on_non_existing_storage
-from apify.consts import StorageTypes
-
-if TYPE_CHECKING:
-    from apify_shared.types import JSONSerializable
-
-    from apify._memory_storage.memory_storage_client import MemoryStorageClient
-
-# This is what API returns in the x-apify-pagination-limit
-# header when no limit query parameter is used.
-LIST_ITEMS_LIMIT = 999_999_999_999
-
-# Number of characters of the dataset item file names.
-# E.g.: 000000019.json - 9 digits
-LOCAL_ENTRY_NAME_DIGITS = 9
-
-
-@ignore_docs
-class DatasetClient(BaseResourceClient):
-    """Sub-client for manipulating a single dataset."""
-
-    _id: str
-    _resource_directory: str
-    _memory_storage_client: MemoryStorageClient
-    _name: str | None
-    _dataset_entries: dict[str, dict]
-    _created_at: datetime
-    _accessed_at: datetime
-    _modified_at: datetime
-    _item_count = 0
-    _file_operation_lock: asyncio.Lock
-
-    def __init__(
-        self: DatasetClient,
-        *,
-        base_storage_directory: str,
-        memory_storage_client: MemoryStorageClient,
-        id: str | None = None,  # noqa: A002
-        name: str | None = None,
-    ) -> None:
-        """Initialize the DatasetClient."""
-        self._id = id or crypto_random_object_id()
-        self._resource_directory = os.path.join(base_storage_directory, name or self._id)
-        self._memory_storage_client = memory_storage_client
-        self._name = name
-        self._dataset_entries = {}
-        self._created_at = datetime.now(timezone.utc)
-        self._accessed_at = datetime.now(timezone.utc)
-        self._modified_at = datetime.now(timezone.utc)
-        self._file_operation_lock = asyncio.Lock()
-
-    async def get(self: DatasetClient) -> dict | None:
-        """Retrieve the dataset.
-
-        Returns:
-            dict, optional: The retrieved dataset, or None, if it does not exist
-        """
-        found = self._find_or_create_client_by_id_or_name(memory_storage_client=self._memory_storage_client, id=self._id, name=self._name)
-
-        if found:
-            async with found._file_operation_lock:
-                await found._update_timestamps(has_been_modified=False)
-                return found._to_resource_info()
-
-        return None
-
-    async def update(self: DatasetClient, *, name: str | None = None) -> dict:
-        """Update the dataset with specified fields.
-
-        Args:
-            name (str, optional): The new name for the dataset
-
-        Returns:
-            dict: The updated dataset
-        """
-        # Check by id
-        existing_dataset_by_id = self._find_or_create_client_by_id_or_name(
-            memory_storage_client=self._memory_storage_client,
-            id=self._id,
-            name=self._name,
-        )
-
-        if existing_dataset_by_id is None:
-            raise_on_non_existing_storage(StorageTypes.DATASET, self._id)
-
-        # Skip if no changes
-        if name is None:
-            return existing_dataset_by_id._to_resource_info()
-
-        async with existing_dataset_by_id._file_operation_lock:
-            # Check that name is not in use already
-            existing_dataset_by_name = next(
-                (dataset for dataset in self._memory_storage_client._datasets_handled if dataset._name and dataset._name.lower() == name.lower()),
-                None,
-            )
-
-            if existing_dataset_by_name is not None:
-                raise_on_duplicate_storage(StorageTypes.DATASET, 'name', name)
-
-            existing_dataset_by_id._name = name
-
-            previous_dir = existing_dataset_by_id._resource_directory
-
-            existing_dataset_by_id._resource_directory = os.path.join(self._memory_storage_client._datasets_directory, name)
-
-            await force_rename(previous_dir, existing_dataset_by_id._resource_directory)
-
-            # Update timestamps
-            await existing_dataset_by_id._update_timestamps(has_been_modified=True)
-
-        return existing_dataset_by_id._to_resource_info()
-
-    async def delete(self: DatasetClient) -> None:
-        """Delete the dataset."""
-        dataset = next((dataset for dataset in self._memory_storage_client._datasets_handled if dataset._id == self._id), None)
-
-        if dataset is not None:
-            async with dataset._file_operation_lock:
-                self._memory_storage_client._datasets_handled.remove(dataset)
-                dataset._item_count = 0
-                dataset._dataset_entries.clear()
-
-                if os.path.exists(dataset._resource_directory):
-                    await aioshutil.rmtree(dataset._resource_directory)
-
-    async def list_items(
-        self: DatasetClient,
-        *,
-        offset: int | None = 0,
-        limit: int | None = LIST_ITEMS_LIMIT,
-        clean: bool | None = None,  # noqa: ARG002
-        desc: bool | None = None,
-        fields: list[str] | None = None,  # noqa: ARG002
-        omit: list[str] | None = None,  # noqa: ARG002
-        unwind: str | None = None,  # noqa: ARG002
-        skip_empty: bool | None = None,  # noqa: ARG002
-        skip_hidden: bool | None = None,  # noqa: ARG002
-        flatten: list[str] | None = None,  # noqa: ARG002
-        view: str | None = None,  # noqa: ARG002
-    ) -> ListPage:
-        """List the items of the dataset.
-
-        Args:
-            offset (int, optional): Number of items that should be skipped at the start. The default value is 0
-            limit (int, optional): Maximum number of items to return. By default there is no limit.
-            desc (bool, optional): By default, results are returned in the same order as they were stored.
-                To reverse the order, set this parameter to True.
-            clean (bool, optional): If True, returns only non-empty items and skips hidden fields (i.e. fields starting with the # character).
-                The clean parameter is just a shortcut for skip_hidden=True and skip_empty=True parameters.
-                Note that since some objects might be skipped from the output, that the result might contain less items than the limit value.
-            fields (list of str, optional): A list of fields which should be picked from the items,
-                only these fields will remain in the resulting record objects.
-                Note that the fields in the outputted items are sorted the same way as they are specified in the fields parameter.
-                You can use this feature to effectively fix the output format.
-            omit (list of str, optional): A list of fields which should be omitted from the items.
-            unwind (str, optional): Name of a field which should be unwound.
-                If the field is an array then every element of the array will become a separate record and merged with parent object.
-                If the unwound field is an object then it is merged with the parent object.
-                If the unwound field is missing or its value is neither an array nor an object and therefore cannot be merged with a parent object,
-                then the item gets preserved as it is. Note that the unwound items ignore the desc parameter.
-            skip_empty (bool, optional): If True, then empty items are skipped from the output.
-                Note that if used, the results might contain less items than the limit value.
-            skip_hidden (bool, optional): If True, then hidden fields are skipped from the output, i.e. fields starting with the # character.
-            flatten (list of str, optional): A list of fields that should be flattened
-            view (str, optional): Name of the dataset view to be used
-
-        Returns:
-            ListPage: A page of the list of dataset items according to the specified filters.
-        """
-        # Check by id
-        existing_dataset_by_id = self._find_or_create_client_by_id_or_name(
-            memory_storage_client=self._memory_storage_client,
-            id=self._id,
-            name=self._name,
-        )
-
-        if existing_dataset_by_id is None:
-            raise_on_non_existing_storage(StorageTypes.DATASET, self._id)
-
-        async with existing_dataset_by_id._file_operation_lock:
-            start, end = existing_dataset_by_id._get_start_and_end_indexes(
-                max(existing_dataset_by_id._item_count - (offset or 0) - (limit or LIST_ITEMS_LIMIT), 0) if desc else offset or 0,
-                limit,
-            )
-
-            items = []
-
-            for idx in range(start, end):
-                entry_number = self._generate_local_entry_name(idx)
-                items.append(existing_dataset_by_id._dataset_entries[entry_number])
-
-            await existing_dataset_by_id._update_timestamps(has_been_modified=False)
-
-            if desc:
-                items.reverse()
-
-            return ListPage(
-                {
-                    'count': len(items),
-                    'desc': desc or False,
-                    'items': items,
-                    'limit': limit or LIST_ITEMS_LIMIT,
-                    'offset': offset or 0,
-                    'total': existing_dataset_by_id._item_count,
-                }
-            )
-
-    async def iterate_items(
-        self: DatasetClient,
-        *,
-        offset: int = 0,
-        limit: int | None = None,
-        clean: bool | None = None,  # noqa: ARG002
-        desc: bool | None = None,
-        fields: list[str] | None = None,  # noqa: ARG002
-        omit: list[str] | None = None,  # noqa: ARG002
-        unwind: str | None = None,  # noqa: ARG002
-        skip_empty: bool | None = None,  # noqa: ARG002
-        skip_hidden: bool | None = None,  # noqa: ARG002
-    ) -> AsyncIterator[dict]:
-        """Iterate over the items in the dataset.
-
-        Args:
-            offset (int, optional): Number of items that should be skipped at the start. The default value is 0
-            limit (int, optional): Maximum number of items to return. By default there is no limit.
-            desc (bool, optional): By default, results are returned in the same order as they were stored.
-                To reverse the order, set this parameter to True.
-            clean (bool, optional): If True, returns only non-empty items and skips hidden fields (i.e. fields starting with the # character).
-                The clean parameter is just a shortcut for skip_hidden=True and skip_empty=True parameters.
-                Note that since some objects might be skipped from the output, that the result might contain less items than the limit value.
-            fields (list of str, optional): A list of fields which should be picked from the items,
-                only these fields will remain in the resulting record objects.
-                Note that the fields in the outputted items are sorted the same way as they are specified in the fields parameter.
-                You can use this feature to effectively fix the output format.
-            omit (list of str, optional): A list of fields which should be omitted from the items.
-            unwind (str, optional): Name of a field which should be unwound.
-                If the field is an array then every element of the array will become a separate record and merged with parent object.
-                If the unwound field is an object then it is merged with the parent object.
-                If the unwound field is missing or its value is neither an array nor an object and therefore cannot be merged with a parent object,
-                then the item gets preserved as it is. Note that the unwound items ignore the desc parameter.
-            skip_empty (bool, optional): If True, then empty items are skipped from the output.
-                Note that if used, the results might contain less items than the limit value.
-            skip_hidden (bool, optional): If True, then hidden fields are skipped from the output, i.e. fields starting with the # character.
-
-        Yields:
-            dict: An item from the dataset
-        """
-        cache_size = 1000
-        first_item = offset
-
-        # If there is no limit, set last_item to None until we get the total from the first API response
-        last_item = None if limit is None else offset + limit
-
-        current_offset = first_item
-        while last_item is None or current_offset < last_item:
-            current_limit = cache_size if last_item is None else min(cache_size, last_item - current_offset)
-
-            current_items_page = await self.list_items(
-                offset=current_offset,
-                limit=current_limit,
-                desc=desc,
-            )
-
-            current_offset += current_items_page.count
-            if last_item is None or current_items_page.total < last_item:
-                last_item = current_items_page.total
-
-            for item in current_items_page.items:
-                yield item
-
-    async def get_items_as_bytes(self: DatasetClient, *_args: Any, **_kwargs: Any) -> bytes:
-        raise NotImplementedError('This method is not supported in local memory storage.')
-
-    async def stream_items(self: DatasetClient, *_args: Any, **_kwargs: Any) -> AsyncIterator:
-        raise NotImplementedError('This method is not supported in local memory storage.')
-
-    async def push_items(self: DatasetClient, items: JSONSerializable) -> None:
-        """Push items to the dataset.
-
-        Args:
-            items: The items which to push in the dataset. Either a stringified JSON, a dictionary, or a list of strings or dictionaries.
-        """
-        # Check by id
-        existing_dataset_by_id = self._find_or_create_client_by_id_or_name(
-            memory_storage_client=self._memory_storage_client, id=self._id, name=self._name
-        )
-
-        if existing_dataset_by_id is None:
-            raise_on_non_existing_storage(StorageTypes.DATASET, self._id)
-
-        normalized = self._normalize_items(items)
-
-        added_ids: list[str] = []
-        for entry in normalized:
-            existing_dataset_by_id._item_count += 1
-            idx = self._generate_local_entry_name(existing_dataset_by_id._item_count)
-
-            existing_dataset_by_id._dataset_entries[idx] = entry
-            added_ids.append(idx)
-
-        data_entries = [(id, existing_dataset_by_id._dataset_entries[id]) for id in added_ids]  # noqa: A001
-
-        async with existing_dataset_by_id._file_operation_lock:
-            await existing_dataset_by_id._update_timestamps(has_been_modified=True)
-
-            await _update_dataset_items(
-                data=data_entries,
-                entity_directory=existing_dataset_by_id._resource_directory,
-                persist_storage=self._memory_storage_client._persist_storage,
-            )
-
-    def _to_resource_info(self: DatasetClient) -> dict:
-        """Retrieve the dataset info."""
-        return {
-            'id': self._id,
-            'name': self._name,
-            'itemCount': self._item_count,
-            'accessedAt': self._accessed_at,
-            'createdAt': self._created_at,
-            'modifiedAt': self._modified_at,
-        }
-
-    async def _update_timestamps(self: DatasetClient, has_been_modified: bool) -> None:  # noqa: FBT001
-        """Update the timestamps of the dataset."""
-        self._accessed_at = datetime.now(timezone.utc)
-
-        if has_been_modified:
-            self._modified_at = datetime.now(timezone.utc)
-
-        dataset_info = self._to_resource_info()
-        await update_metadata(
-            data=dataset_info,
-            entity_directory=self._resource_directory,
-            write_metadata=self._memory_storage_client._write_metadata,
-        )
-
-    def _get_start_and_end_indexes(self: DatasetClient, offset: int, limit: int | None = None) -> tuple[int, int]:
-        actual_limit = limit or self._item_count
-        start = offset + 1
-        end = min(offset + actual_limit, self._item_count) + 1
-        return (start, end)
-
-    def _generate_local_entry_name(self: DatasetClient, idx: int) -> str:
-        return str(idx).zfill(LOCAL_ENTRY_NAME_DIGITS)
-
-    def _normalize_items(self: DatasetClient, items: JSONSerializable) -> list[dict]:
-        def normalize_item(item: Any) -> dict | None:
-            if isinstance(item, str):
-                item = json.loads(item)
-
-            if isinstance(item, list):
-                received = ',\n'.join(item)
-                raise TypeError(f'Each dataset item can only be a single JSON object, not an array. Received: [{received}]')
-
-            if (not isinstance(item, dict)) and item is not None:
-                raise TypeError(f'Each dataset item must be a JSON object. Received: {item}')
-
-            return item
-
-        if isinstance(items, str):
-            items = json.loads(items)
-
-        result = list(map(normalize_item, items)) if isinstance(items, list) else [normalize_item(items)]
-        # filter(None, ..) returns items that are True
-        return list(filter(None, result))
-
-    @classmethod
-    def _get_storages_dir(cls: type[DatasetClient], memory_storage_client: MemoryStorageClient) -> str:
-        return memory_storage_client._datasets_directory
-
-    @classmethod
-    def _get_storage_client_cache(
-        cls: type[DatasetClient],
-        memory_storage_client: MemoryStorageClient,
-    ) -> list[DatasetClient]:
-        return memory_storage_client._datasets_handled
-
-    @classmethod
-    def _create_from_directory(
-        cls: type[DatasetClient],
-        storage_directory: str,
-        memory_storage_client: MemoryStorageClient,
-        id: str | None = None,  # noqa: A002
-        name: str | None = None,
-    ) -> DatasetClient:
-        item_count = 0
-        created_at = datetime.now(timezone.utc)
-        accessed_at = datetime.now(timezone.utc)
-        modified_at = datetime.now(timezone.utc)
-        entries: dict[str, dict] = {}
-
-        has_seen_metadata_file = False
-
-        # Access the dataset folder
-        for entry in os.scandir(storage_directory):
-            if entry.is_file():
-                if entry.name == '__metadata__.json':
-                    has_seen_metadata_file = True
-
-                    # We have found the dataset's metadata file, build out information based on it
-                    with open(os.path.join(storage_directory, entry.name), encoding='utf-8') as f:
-                        metadata = json.load(f)
-                    id = metadata['id']  # noqa: A001
-                    name = metadata['name']
-                    item_count = metadata['itemCount']
-                    created_at = datetime.fromisoformat(metadata['createdAt'])
-                    accessed_at = datetime.fromisoformat(metadata['accessedAt'])
-                    modified_at = datetime.fromisoformat(metadata['modifiedAt'])
-
-                    continue
-
-                with open(os.path.join(storage_directory, entry.name), encoding='utf-8') as f:
-                    entry_content = json.load(f)
-                entry_name = entry.name.split('.')[0]
-
-                entries[entry_name] = entry_content
-
-                if not has_seen_metadata_file:
-                    item_count += 1
-
-        new_client = DatasetClient(
-            base_storage_directory=memory_storage_client._datasets_directory,
-            memory_storage_client=memory_storage_client,
-            id=id,
-            name=name,
-        )
-
-        # Overwrite properties
-        new_client._accessed_at = accessed_at
-        new_client._created_at = created_at
-        new_client._modified_at = modified_at
-        new_client._item_count = item_count
-
-        for entry_id, content in entries.items():
-            new_client._dataset_entries[entry_id] = content
-
-        return new_client

apify/_memory_storage/resource_clients/dataset_collection.py

@@ -1,48 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from apify_shared.utils import ignore_docs
-
-from apify._memory_storage.resource_clients.base_resource_collection_client import BaseResourceCollectionClient
-from apify._memory_storage.resource_clients.dataset import DatasetClient
-
-if TYPE_CHECKING:
-    from apify_shared.models import ListPage
-
-
-@ignore_docs
-class DatasetCollectionClient(BaseResourceCollectionClient):
-    """Sub-client for manipulating datasets."""
-
-    def _get_storage_client_cache(self: DatasetCollectionClient) -> list[DatasetClient]:
-        return self._memory_storage_client._datasets_handled
-
-    def _get_resource_client_class(self: DatasetCollectionClient) -> type[DatasetClient]:
-        return DatasetClient
-
-    async def list(self: DatasetCollectionClient) -> ListPage:
-        """List the available datasets.
-
-        Returns:
-            ListPage: The list of available datasets matching the specified filters.
-        """
-        return await super().list()
-
-    async def get_or_create(
-        self: DatasetCollectionClient,
-        *,
-        name: str | None = None,
-        schema: dict | None = None,
-        _id: str | None = None,
-    ) -> dict:
-        """Retrieve a named dataset, or create a new one when it doesn't exist.
-
-        Args:
-            name (str, optional): The name of the dataset to retrieve or create.
-            schema (dict, optional): The schema of the dataset
-
-        Returns:
-            dict: The retrieved or newly-created dataset.
-        """
-        return await super().get_or_create(name=name, schema=schema, _id=_id)