Add missing CRTTransferManager stubs

youtype · Aug 27, 2024 · 3e78319 · 3e78319
1 parent 898c99d
commit 3e78319
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 1 deletion.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1,3 +1,2 @@
 s3transfer-stubs/py.typed
 s3transfer-stubs/*.pyi
-s3transfer-stubs/*/*.pyi
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,6 +31,7 @@ config-settings = { editable-mode = "strict" }
 dev-dependencies = [
     "setuptools",
     "pre-commit",
+    "awscrt",
     "s3transfer",
     "botocore",
     "mypy",

diff --git a/s3transfer-stubs/crt.pyi b/s3transfer-stubs/crt.pyi
@@ -6,6 +6,7 @@ from typing import (
     Callable,
     Dict,
     Iterator,
+    List,
     Mapping,
     Optional,
     Sequence,
@@ -26,6 +27,7 @@ from s3transfer.subscribers import BaseSubscriber
 from s3transfer.utils import CallArgs as CallArgs
 from s3transfer.utils import OSUtils as OSUtils
 from s3transfer.utils import get_callbacks as get_callbacks
+from s3transfer.manager import TransferManager
 
 _R = TypeVar("_R")
 
@@ -44,6 +46,11 @@ def create_s3_crt_client(
 ) -> S3Client: ...
 
 class CRTTransferManager:
+    ALLOWED_DOWNLOAD_ARGS: List[str] = ...
+    ALLOWED_UPLOAD_ARGS: List[str] = ...
+    ALLOWED_DELETE_ARGS: List[str] = ...
+    VALIDATE_SUPPORTED_BUCKET_VALUES: bool = ...
+
     def __init__(
         self,
         crt_s3_client: S3Client,  # type: ignore

diff --git a/uv.lock b/uv.lock