Add _patch.py for aio

autorest/codegen/serializers/__init__.py

@@ -45,11 +45,8 @@ def serialize(self, code_model: CodeModel) -> None:
         )
 
         # if there was a patch file before, we keep it
-        if self._autorestapi.read_file(namespace_path / "_patch.py"):
-            self._autorestapi.write_file(
-                namespace_path / Path("_patch.py"),
-                self._autorestapi.read_file(namespace_path / "_patch.py")
-            )
+        self._keep_patch_file(namespace_path / Path("_patch.py"))
+        self._keep_patch_file(namespace_path / Path("aio") / Path("_patch.py"))
 
         self._serialize_and_write_top_level_folder(code_model=code_model, env=env, namespace_path=namespace_path)
 
@@ -72,6 +69,12 @@ def serialize(self, code_model: CodeModel) -> None:
             self._serialize_and_write_models_folder(code_model=code_model, env=env, namespace_path=namespace_path)
 
 
+
+    def _keep_patch_file(self, path_file: Path):
+        if self._autorestapi.read_file(path_file):
+            self._autorestapi.write_file(path_file, self._autorestapi.read_file(path_file))
+
+
     def _serialize_and_write_models_folder(self, code_model: CodeModel, env: Environment, namespace_path: Path) -> None:
         # Write the models folder
         models_path = namespace_path / Path("models")

autorest/codegen/templates/init.py.jinja2

@@ -8,11 +8,9 @@ from ._version import VERSION
 __version__ = VERSION
 {% endif %}
 __all__ = ['{{ code_model.class_name }}']
-{% if not async_mode %}
 
 try:
     from ._patch import patch_sdk  # type: ignore
     patch_sdk()
 except ImportError:
     pass
-{% endif %}

docs/samples/specification/azure_key_credential/generated/azure/key/credential/sample/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._auto_rest_head_test_service import AutoRestHeadTestService
 __all__ = ['AutoRestHeadTestService']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

docs/samples/specification/basic/generated/azure/basic/sample/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._auto_rest_head_test_service import AutoRestHeadTestService
 __all__ = ['AutoRestHeadTestService']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

docs/samples/specification/directives/generated/azure/directives/sample/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._polling_paging_example import PollingPagingExample
 __all__ = ['PollingPagingExample']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

docs/samples/specification/management/generated/azure/mgmt/sample/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._auto_rest_head_test_service import AutoRestHeadTestService
 __all__ = ['AutoRestHeadTestService']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

docs/samples/specification/multiapi/generated/azure/multiapi/sample/v1/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._multiapi_service_client import MultiapiServiceClient
 __all__ = ['MultiapiServiceClient']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

docs/samples/specification/multiapi/generated/azure/multiapi/sample/v2/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._multiapi_service_client import MultiapiServiceClient
 __all__ = ['MultiapiServiceClient']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

docs/samples/specification/multiapi/generated/azure/multiapi/sample/v3/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._multiapi_service_client import MultiapiServiceClient
 __all__ = ['MultiapiServiceClient']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/bodyduration/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_duration_test_service import AutoRestDurationTestService
 
 __all__ = ["AutoRestDurationTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/AzureParameterGrouping/azureparametergrouping/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_parameter_grouping_test_service import AutoRestParameterGroupingTestService
 
 __all__ = ["AutoRestParameterGroupingTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/AzureReport/azurereport/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_report_service_for_azure import AutoRestReportServiceForAzure
 
 __all__ = ["AutoRestReportServiceForAzure"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/AzureSpecials/azurespecialproperties/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_azure_special_parameters_test_client import AutoRestAzureSpecialParametersTestClient
 
 __all__ = ["AutoRestAzureSpecialParametersTestClient"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/custombaseurl/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_parameterized_host_test_client import AutoRestParameterizedHostTestClient
 
 __all__ = ["AutoRestParameterizedHostTestClient"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/aio/__init__.py

@@ -8,3 +8,9 @@
 
 from ._auto_rest_paging_test_service import AutoRestPagingTestService
 __all__ = ['AutoRestPagingTestService']
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_parameterized_host_test_paging_client import AutoRestParameterizedHostTestPagingClient
 
 __all__ = ["AutoRestParameterizedHostTestPagingClient"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/Head/head/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_head_test_service import AutoRestHeadTestService
 
 __all__ = ["AutoRestHeadTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/headexceptions/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_head_exception_test_service import AutoRestHeadExceptionTestService
 
 __all__ = ["AutoRestHeadExceptionTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/headwithazurekeycredentialpolicy/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_head_test_service import AutoRestHeadTestService
 
 __all__ = ["AutoRestHeadTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/Lro/lro/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_long_running_operation_test_service import AutoRestLongRunningOperationTestService
 
 __all__ = ["AutoRestLongRunningOperationTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/LroWithParameterizedEndpoints/lrowithparameterizedendpoints/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._lro_with_paramaterized_endpoints import LROWithParamaterizedEndpoints
 
 __all__ = ["LROWithParamaterizedEndpoints"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/Paging/paging/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_paging_test_service import AutoRestPagingTestService
 
 __all__ = ["AutoRestPagingTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._storage_management_client import StorageManagementClient
 
 __all__ = ["StorageManagementClient"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/legacy/Expected/AcceptanceTests/SubscriptionIdApiVersion/subscriptionidapiversion/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._microsoft_azure_test_url import MicrosoftAzureTestUrl
 
 __all__ = ["MicrosoftAzureTestUrl"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/AzureBodyDurationLowLevel/bodydurationlowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_duration_test_service import AutoRestDurationTestService
 
 __all__ = ["AutoRestDurationTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/AzureParameterGroupingLowLevel/azureparametergroupinglowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_parameter_grouping_test_service import AutoRestParameterGroupingTestService
 
 __all__ = ["AutoRestParameterGroupingTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/AzureReportLowLevel/azurereportlowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_report_service_for_azure import AutoRestReportServiceForAzure
 
 __all__ = ["AutoRestReportServiceForAzure"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/AzureSpecialsLowLevel/azurespecialpropertieslowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_azure_special_parameters_test_client import AutoRestAzureSpecialParametersTestClient
 
 __all__ = ["AutoRestAzureSpecialParametersTestClient"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/CustomBaseUriLowLevel/custombaseurllowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_parameterized_host_test_client import AutoRestParameterizedHostTestClient
 
 __all__ = ["AutoRestParameterizedHostTestClient"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/CustomUrlPagingLowLevel/custombaseurlpaginglowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_parameterized_host_test_paging_client import AutoRestParameterizedHostTestPagingClient
 
 __all__ = ["AutoRestParameterizedHostTestPagingClient"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/HeadExceptionsLowLevel/headexceptionslowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_head_exception_test_service import AutoRestHeadExceptionTestService
 
 __all__ = ["AutoRestHeadExceptionTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/HeadLowLevel/headlowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_head_test_service import AutoRestHeadTestService
 
 __all__ = ["AutoRestHeadTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/LroLowLevel/lrolowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_long_running_operation_test_service import AutoRestLongRunningOperationTestService
 
 __all__ = ["AutoRestLongRunningOperationTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/LroWithParameterizedEndpointsLowLevel/lrowithparameterizedendpointslowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._lro_with_paramaterized_endpoints import LROWithParamaterizedEndpoints
 
 __all__ = ["LROWithParamaterizedEndpoints"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass

test/azure/low-level/Expected/AcceptanceTests/PagingLowLevel/paginglowlevel/aio/__init__.py

@@ -9,3 +9,10 @@
 from ._auto_rest_paging_test_service import AutoRestPagingTestService
 
 __all__ = ["AutoRestPagingTestService"]
+
+try:
+    from ._patch import patch_sdk  # type: ignore
+
+    patch_sdk()
+except ImportError:
+    pass