Add override comments to aioboto3 Session generated methods

Author: vemel

mypy_boto3_builder/parsers/parse_wrapper_package.py

@@ -100,12 +100,14 @@ def parse_types_aioboto3_package(
             ExternalImport(Import.aiobotocore + "session", "ClientCreatorContext"),
             [method.return_type],
         )
+        method.type_ignore = "override"
         package.session_class.methods.append(method)
     for method in parser.get_session_resource_methods():
         method.return_type = TypeSubscript(
             InternalImport("ResourceCreatorContext"),
             [method.return_type],
         )
+        method.type_ignore = "override"
         package.session_class.methods.append(method)
 
     for type_annotation in package.session_class.iterate_types():

mypy_boto3_builder/stubs_static/types-aioboto3/session.pyi

@@ -27,7 +27,7 @@ class Session(Boto3Session):
         profile_name: str | None = ...,
         aws_account_id: str | None = ...,
     ) -> None:
-        self._session: AioBotocoreSession
+        self._session: AioBotocoreSession  # type: ignore[override]
         self.resource_factory: AIOBoto3ResourceFactory  # type: ignore[override]
         self._loader: Loader
 

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

@@ -11,13 +11,14 @@ from aioboto3.resources.factory import AIOBoto3ResourceFactory
 from aiobotocore.config import AioConfig
 from aiobotocore.credentials import AioCredentials
 from aiobotocore.session import AioSession as AioBotocoreSession
+from boto3.session import Session as Boto3Session
 from botocore.loaders import Loader
 
 {% for import_record in package.get_session_required_import_records() -%}
     {{ import_record -}}{{ "\n" -}}
 {% endfor -%}
 
-class Session:
+class Session(Boto3Session):
     def __init__(
         self,
         aws_access_key_id: str | None = ...,
@@ -28,10 +29,16 @@ class Session:
         profile_name: str | None = ...,
         aws_account_id: str | None = ...,
     ) -> None:
-        self._session: AioBotocoreSession
-        self.resource_factory: AIOBoto3ResourceFactory  # type: ignore [override]
+        self._session: AioBotocoreSession  # type: ignore[override]
+        self.resource_factory: AIOBoto3ResourceFactory  # type: ignore[override]
         self._loader: Loader
     def get_credentials(self) -> AioCredentials | None: ...
+    async def get_available_regions(  # type: ignore[override]
+        self,
+        service_name: str,
+        partition_name: str = ...,
+        allow_non_regional: bool = ...,
+    ) -> list[str]: ...
 
 {% filter indent(4, True) -%}
     {% for method in package.session_class.methods -%}