Pull static and add aws_account_id argument to Session

Author: vemel

mypy_boto3_builder/stubs_static/types-aioboto3/session.pyi

@@ -25,6 +25,7 @@ class Session(Boto3Session):
         region_name: str | None = ...,
         botocore_session: AioBotocoreSession | None = ...,
         profile_name: str | None = ...,
+        aws_account_id: str | None = ...,
     ) -> None:
         self._session: AioBotocoreSession
         self.resource_factory: AIOBoto3ResourceFactory  # type: ignore[override]
@@ -43,6 +44,7 @@ class Session(Boto3Session):
         aws_secret_access_key: str | None = ...,
         aws_session_token: str | None = ...,
         config: AioConfig | None = ...,
+        aws_account_id: str | None = ...,
     ) -> ClientCreatorContext[Any]: ...
     def resource(  # type: ignore[override]
         self,

mypy_boto3_builder/stubs_static/types-aiobotocore/credentials.pyi

@@ -56,21 +56,17 @@ def create_mfa_serial_refresher(actual_refresh: Callable[[], dict[str, Any]]) ->
 def create_aio_mfa_serial_refresher(actual_refresh: Callable[[], dict[str, Any]]) -> Any: ...
 
 class AioCredentials(Credentials):
+    def get_account_id(self) -> str | None: ...
+    def get_access_key(self) -> str: ...
+    def get_secret_key(self) -> str: ...
+    def get_token(self) -> str | None: ...
     async def get_frozen_credentials(self) -> ReadOnlyCredentials: ...  # type: ignore[override]
 
 class AioRefreshableCredentials(RefreshableCredentials):
-    @property  # type: ignore[override]
-    def access_key(self) -> str: ...
-    @access_key.setter
-    def access_key(self, value: str) -> None: ...
-    @property  # type: ignore[override]
-    def secret_key(self) -> str: ...
-    @secret_key.setter
-    def secret_key(self, value: str) -> None: ...
-    @property  # type: ignore[override]
-    def token(self) -> str: ...
-    @token.setter
-    def token(self, value: str) -> None: ...
+    async def get_account_id(self) -> str | None: ...
+    async def get_access_key(self) -> str: ...
+    async def get_secret_key(self) -> str: ...
+    async def get_token(self) -> str: ...
     async def get_frozen_credentials(self) -> ReadOnlyCredentials: ...  # type: ignore[override]
 
 class AioDeferredRefreshableCredentials(AioRefreshableCredentials):

mypy_boto3_builder/stubs_static/types-aiobotocore/session.pyi

@@ -59,8 +59,12 @@ class AioSession(Session):
         config: Config | None = ...,
     ) -> ClientCreatorContext[Any]: ...
     async def get_credentials(self) -> AioCredentials | None: ...  # type: ignore[override]
-    def set_credentials(
-        self, access_key: str, secret_key: str, token: Any | None = ...
+    def set_credentials(  # type: ignore[override]
+        self,
+        access_key: str,
+        secret_key: str,
+        token: Any | None = ...,
+        account_id: str | None = ...,
     ) -> None: ...
     async def get_service_model(  # type: ignore[override]
         self, service_name: str, api_version: Any | None = ...

mypy_boto3_builder/templates/types-aioboto3/aioboto3-stubs/session.pyi.jinja2

@@ -26,6 +26,7 @@ class Session:
         region_name: str | None = ...,
         botocore_session: AioBotocoreSession | None = ...,
         profile_name: str | None = ...,
+        aws_account_id: str | None = ...,
     ) -> None:
         self._session: AioBotocoreSession
         self.resource_factory: AIOBoto3ResourceFactory  # type: ignore [override]

mypy_boto3_builder/templates/types-boto3/boto3-stubs/session.pyi.jinja2

@@ -33,6 +33,7 @@ class Session:
         region_name: str | None = ...,
         botocore_session: BotocoreSession | None = ...,
         profile_name: str | None = ...,
+        aws_account_id: str | None = ...,
     ) -> None:
         self._session: BotocoreSession
         self.resource_factory: ResourceFactory