Session is inherited from Boto3Session

Author: vemel

.vscode/settings.json

@@ -2,6 +2,15 @@
     "[python]": {
         "editor.formatOnSave": true,
         "editor.defaultFormatter": "charliermarsh.ruff",
-        "ruff.nativeServer": "on",
+        "editor.codeActionsOnSave": {
+            "source.organizeImports": "explicit"
+        }
+    },
+    "files.exclude": {
+        ".venv": true,
+        ".mypy_cache": true,
+        ".ruff_cache": true,
+        "*.egg-info": true,
+        "build": true,
     }
 }

aioboto3-stubs/session.pyi

@@ -12,11 +12,12 @@ from aioboto3.resources.factory import AIOBoto3ResourceFactory
 from aiobotocore.client import AioBaseClient
 from aiobotocore.config import AioConfig
 from aiobotocore.credentials import AioCredentials
-from botocore.hooks import BaseEventHooks
+from aiobotocore.session import ClientCreatorContext
+from boto3.session import Session as Boto3Session
 from botocore.loaders import Loader
 from botocore.session import Session as BotocoreSession
 
-class Session:
+class Session(Boto3Session):
     def __init__(
         self,
         aws_access_key_id: str | None = ...,
@@ -30,26 +31,7 @@ class Session:
         self.resource_factory: AIOBoto3ResourceFactory
         self._loader: Loader
 
-    @property
-    def profile_name(self) -> str: ...
-    @property
-    def region_name(self) -> str: ...
-    @property
-    def events(self) -> BaseEventHooks: ...
-    @property
-    def available_profiles(self) -> list[str]: ...
-    def _setup_loader(self) -> None: ...
-    def get_available_services(self) -> list[str]: ...
-    def get_available_resources(self) -> list[str]: ...
-    def get_available_partitions(self) -> list[str]: ...
-    def get_available_regions(
-        self,
-        service_name: str,
-        partition_name: str = ...,
-        allow_non_regional: bool = ...,
-    ) -> list[str]: ...
     def get_credentials(self) -> AioCredentials | None: ...
-    def _register_default_handlers(self) -> None: ...
     def client(  # type: ignore [override]
         self,
         service_name: str,
@@ -62,8 +44,8 @@ class Session:
         aws_secret_access_key: str | None = ...,
         aws_session_token: str | None = ...,
         config: AioConfig | None = ...,
-    ) -> AioBaseClient: ...
-    def resource(
+    ) -> ClientCreatorContext[AioBaseClient]: ...
+    def resource(  # type: ignore [override]
         self,
         service_name: str,
         region_name: str | None = ...,

test.py

@@ -4,6 +4,11 @@
 
 from aioboto3.session import Session
 
-session = Session("aws_access_key_id")
-session = Session(None)
-session = Session(123)
+
+async def main() -> None:
+    session = Session("aws_access_key_id")
+    session = Session(None)
+    session = Session(123)
+
+    async with session.client("s3") as client:
+        client.create_bucket(Bucket="bucket")