Create a type for StorageOwnership

Author: nmanerikar

src/apify_client/_resource_clients/dataset_collection.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Literal
+from typing import TYPE_CHECKING, Any
 
 from apify_client._docs import docs_group
 from apify_client._models import Dataset, DatasetResponse, ListOfDatasets, ListOfDatasetsResponse
 from apify_client._resource_clients._resource_client import ResourceClient, ResourceClientAsync
 
 if TYPE_CHECKING:
-    from apify_client._types import Timeout
+    from apify_client._types import StorageOwnership, Timeout
 
 
 @docs_group('Resource clients')
@@ -36,7 +36,7 @@ def list(
         limit: int | None = None,
         offset: int | None = None,
         desc: bool | None = None,
-        ownership: Literal['ownedByMe', 'sharedWithMe'] | None = None,
+        ownership: StorageOwnership | None = None,
         timeout: Timeout = 'medium',
     ) -> ListOfDatasets:
         """List the available datasets.
@@ -109,7 +109,7 @@ async def list(
         limit: int | None = None,
         offset: int | None = None,
         desc: bool | None = None,
-        ownership: Literal['ownedByMe', 'sharedWithMe'] | None = None,
+        ownership: StorageOwnership | None = None,
         timeout: Timeout = 'medium',
     ) -> ListOfDatasets:
         """List the available datasets.

src/apify_client/_resource_clients/key_value_store_collection.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Literal
+from typing import TYPE_CHECKING, Any
 
 from apify_client._docs import docs_group
 from apify_client._models import (
@@ -12,7 +12,7 @@
 from apify_client._resource_clients._resource_client import ResourceClient, ResourceClientAsync
 
 if TYPE_CHECKING:
-    from apify_client._types import Timeout
+    from apify_client._types import StorageOwnership, Timeout
 
 
 @docs_group('Resource clients')
@@ -41,7 +41,7 @@ def list(
         limit: int | None = None,
         offset: int | None = None,
         desc: bool | None = None,
-        ownership: Literal['ownedByMe', 'sharedWithMe'] | None = None,
+        ownership: StorageOwnership | None = None,
         timeout: Timeout = 'medium',
     ) -> ListOfKeyValueStores:
         """List the available key-value stores.
@@ -114,7 +114,7 @@ async def list(
         limit: int | None = None,
         offset: int | None = None,
         desc: bool | None = None,
-        ownership: Literal['ownedByMe', 'sharedWithMe'] | None = None,
+        ownership: StorageOwnership | None = None,
         timeout: Timeout = 'medium',
     ) -> ListOfKeyValueStores:
         """List the available key-value stores.

src/apify_client/_resource_clients/request_queue_collection.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Literal
+from typing import TYPE_CHECKING, Any
 
 from apify_client._docs import docs_group
 from apify_client._models import (
@@ -12,7 +12,7 @@
 from apify_client._resource_clients._resource_client import ResourceClient, ResourceClientAsync
 
 if TYPE_CHECKING:
-    from apify_client._types import Timeout
+    from apify_client._types import StorageOwnership, Timeout
 
 
 @docs_group('Resource clients')
@@ -41,7 +41,7 @@ def list(
         limit: int | None = None,
         offset: int | None = None,
         desc: bool | None = None,
-        ownership: Literal['ownedByMe', 'sharedWithMe'] | None = None,
+        ownership: StorageOwnership | None = None,
         timeout: Timeout = 'medium',
     ) -> ListOfRequestQueues:
         """List the available request queues.
@@ -112,7 +112,7 @@ async def list(
         limit: int | None = None,
         offset: int | None = None,
         desc: bool | None = None,
-        ownership: Literal['ownedByMe', 'sharedWithMe'] | None = None,
+        ownership: StorageOwnership | None = None,
         timeout: Timeout = 'medium',
     ) -> ListOfRequestQueues:
         """List the available request queues.

src/apify_client/_types.py

@@ -9,6 +9,9 @@
 
 from apify_client._models import ActorJobStatus, WebhookCreate  # noqa: TC001
 
+StorageOwnership = Literal['ownedByMe', 'sharedWithMe']
+"""Filter for storage listing methods to return only storages owned by the user or shared with the user."""
+
 Timeout = timedelta | Literal['no_timeout', 'short', 'medium', 'long']
 """Type for the `timeout` parameter on resource client methods.