switch over completely to rest requests and responses (#2605)

.chronus/changes/switch_to_rest-2024-4-30-12-59-16.md

@@ -0,0 +1,7 @@
+---
+changeKind: feature
+packages:
+  - "@autorest/python"
+---
+
+remove support for deprecated azure.core.pipeline.transport requests and responses

packages/autorest.python/autorest/codegen/models/client.py

@@ -253,18 +253,6 @@ def link_lro_initial_operations(self) -> None:
                 if isinstance(operation, (LROOperation, LROPagingOperation)):
                     operation.initial_operation = self.lookup_operation(id(operation.yaml_data["initialOperation"]))
 
-    @property
-    def need_request_converter(self) -> bool:
-        """
-        Whether we need to convert our created azure.core.rest.HttpRequest to
-        azure.core.pipeline.transport.HttpRequest
-        """
-        return (
-            self.code_model.options["show_operations"]
-            and bool(self.request_builders)
-            and not self.code_model.options["version_tolerant"]
-        )
-
     @property
     def has_abstract_operations(self) -> bool:
         """Whether there is abstract operation in any operation group."""

packages/autorest.python/autorest/codegen/models/code_model.py

@@ -120,11 +120,7 @@ def need_vendored_code(self, async_mode: bool) -> bool:
             return True
         if async_mode:
             return self.need_mixin_abc
-        return self.need_request_converter or self.need_mixin_abc or self.has_etag or self.has_form_data
-
-    @property
-    def need_request_converter(self) -> bool:
-        return any(c for c in self.clients if c.need_request_converter)
+        return self.need_mixin_abc or self.has_etag or self.has_form_data
 
     @property
     def need_mixin_abc(self) -> bool:

packages/autorest.python/autorest/codegen/models/operation.py

@@ -366,8 +366,6 @@ def imports(  # pylint: disable=too-many-branches, disable=too-many-statements
             file_import.add_import("warnings", ImportType.STDLIB)
 
         relative_path = "..." if async_mode else ".."
-        if self.code_model.need_request_converter:
-            file_import.add_submodule_import(f"{relative_path}_vendor", "_convert_request", ImportType.LOCAL)
         if self.has_etag:
             file_import.add_submodule_import(
                 "exceptions",
@@ -377,32 +375,18 @@ def imports(  # pylint: disable=too-many-branches, disable=too-many-statements
             if not async_mode:
                 file_import.add_submodule_import(f"{relative_path}_vendor", "prep_if_match", ImportType.LOCAL)
                 file_import.add_submodule_import(f"{relative_path}_vendor", "prep_if_none_match", ImportType.LOCAL)
-        if self.code_model.need_request_converter:
-            if async_mode:
-                file_import.add_submodule_import(
-                    "azure.core.pipeline.transport",
-                    "AsyncHttpResponse",
-                    ImportType.SDKCORE,
-                )
-            else:
-                file_import.add_submodule_import(
-                    "azure.core.pipeline.transport",
-                    "HttpResponse",
-                    ImportType.SDKCORE,
-                )
+        if async_mode:
+            file_import.add_submodule_import(
+                "rest",
+                "AsyncHttpResponse",
+                ImportType.SDKCORE,
+            )
         else:
-            if async_mode:
-                file_import.add_submodule_import(
-                    "rest",
-                    "AsyncHttpResponse",
-                    ImportType.SDKCORE,
-                )
-            else:
-                file_import.add_submodule_import(
-                    "rest",
-                    "HttpResponse",
-                    ImportType.SDKCORE,
-                )
+            file_import.add_submodule_import(
+                "rest",
+                "HttpResponse",
+                ImportType.SDKCORE,
+            )
         if self.code_model.options["builders_visibility"] == "embedded" and not async_mode:
             file_import.merge(self.request_builder.imports())
         file_import.add_submodule_import(

packages/autorest.python/autorest/codegen/serializers/builder_serializer.py

@@ -846,11 +846,6 @@ def _create_request_builder_call(
 
     def _postprocess_http_request(self, builder: OperationType, template_url: Optional[str] = None) -> List[str]:
         retval: List[str] = []
-        if not self.code_model.options["version_tolerant"]:
-            pass_files = ""
-            if builder.parameters.has_body and builder.parameters.body_parameter.client_name == "files":
-                pass_files = ", _files"
-            retval.append(f"_request = _convert_request(_request{pass_files})")
         if builder.parameters.path:
             retval.extend(self.serialize_path(builder))
         url_to_format = "_request.url"
@@ -969,13 +964,12 @@ def response_headers_and_deserialization(
     def handle_error_response(self, builder: OperationType) -> List[str]:
         async_await = "await " if self.async_mode else ""
         retval = [f"if response.status_code not in {str(builder.success_status_codes)}:"]
-        if not self.code_model.need_request_converter:
-            retval.extend(
-                [
-                    "    if _stream:",
-                    f"        {async_await} response.read()  # Load the body in memory and close the socket",
-                ]
-            )
+        retval.extend(
+            [
+                "    if _stream:",
+                f"        {async_await} response.read()  # Load the body in memory and close the socket",
+            ]
+        )
         type_ignore = "  # type: ignore" if _need_type_ignore(builder) else ""
         retval.append(
             f"    map_error(status_code=response.status_code, response=response, error_map=error_map){type_ignore}"

packages/autorest.python/autorest/codegen/serializers/general_serializer.py

@@ -87,13 +87,6 @@ def serialize_vendor_file(self, clients: List[Client]) -> str:
 
         # configure imports
         file_import = FileImport(self.code_model)
-        if self.code_model.need_request_converter:
-            file_import.add_submodule_import(
-                "azure.core.pipeline.transport",
-                "HttpRequest",
-                ImportType.SDKCORE,
-            )
-
         if self.code_model.need_mixin_abc:
             file_import.add_submodule_import(
                 "abc",

packages/autorest.python/autorest/codegen/templates/vendor.py.jinja2

@@ -3,14 +3,6 @@
 
 {{ imports }}
 
-{% if code_model.need_request_converter and not async_mode %}
-def _convert_request(request, files=None):
-    data = request.content if not files else None
-    request = HttpRequest(method=request.method, url=request.url, headers=request.headers, data=data)
-    if files:
-        request.set_formdata_body(files)
-    return request
-{% endif %}
 {% if code_model.need_mixin_abc %}
     {% for client in clients | selectattr("has_mixin") %}
 {% set pylint_disable = "# pylint: disable=name-too-long" if (client.name | length) + ("MixinABC" | length) > 40 else "" %}

packages/autorest.python/samples/specification/management/generated/azure/mgmt/sample/aio/operations/_http_success_operations.py

@@ -18,12 +18,10 @@
     map_error,
 )
 from azure.core.pipeline import PipelineResponse
-from azure.core.pipeline.transport import AsyncHttpResponse
-from azure.core.rest import HttpRequest
+from azure.core.rest import AsyncHttpResponse, HttpRequest
 from azure.core.tracing.decorator_async import distributed_trace_async
 from azure.mgmt.core.exceptions import ARMErrorFormat
 
-from ..._vendor import _convert_request
 from ...operations._http_success_operations import build_head200_request, build_head204_request, build_head404_request
 
 if sys.version_info >= (3, 9):
@@ -76,7 +74,6 @@ async def head200(self, **kwargs: Any) -> bool:
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -87,6 +84,8 @@ async def head200(self, **kwargs: Any) -> bool:
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 404]:
+            if _stream:
+                await response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
@@ -119,7 +118,6 @@ async def head204(self, **kwargs: Any) -> bool:
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -130,6 +128,8 @@ async def head204(self, **kwargs: Any) -> bool:
         response = pipeline_response.http_response
 
         if response.status_code not in [204, 404]:
+            if _stream:
+                await response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
@@ -162,7 +162,6 @@ async def head404(self, **kwargs: Any) -> bool:
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -173,6 +172,8 @@ async def head404(self, **kwargs: Any) -> bool:
         response = pipeline_response.http_response
 
         if response.status_code not in [204, 404]:
+            if _stream:
+                await response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 

packages/autorest.python/samples/specification/management/generated/azure/mgmt/sample/operations/_http_success_operations.py

@@ -18,13 +18,11 @@
     map_error,
 )
 from azure.core.pipeline import PipelineResponse
-from azure.core.pipeline.transport import HttpResponse
-from azure.core.rest import HttpRequest
+from azure.core.rest import HttpRequest, HttpResponse
 from azure.core.tracing.decorator import distributed_trace
 from azure.mgmt.core.exceptions import ARMErrorFormat
 
 from .._serialization import Serializer
-from .._vendor import _convert_request
 
 if sys.version_info >= (3, 9):
     from collections.abc import MutableMapping
@@ -100,7 +98,6 @@ def head200(self, **kwargs: Any) -> bool:
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -111,6 +108,8 @@ def head200(self, **kwargs: Any) -> bool:
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 404]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
@@ -143,7 +142,6 @@ def head204(self, **kwargs: Any) -> bool:
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -154,6 +152,8 @@ def head204(self, **kwargs: Any) -> bool:
         response = pipeline_response.http_response
 
         if response.status_code not in [204, 404]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
@@ -186,7 +186,6 @@ def head404(self, **kwargs: Any) -> bool:
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -197,6 +196,8 @@ def head404(self, **kwargs: Any) -> bool:
         response = pipeline_response.http_response
 
         if response.status_code not in [204, 404]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 

packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/_vendor.py

@@ -8,8 +8,6 @@
 from abc import ABC
 from typing import TYPE_CHECKING
 
-from azure.core.pipeline.transport import HttpRequest
-
 from ._configuration import MultiapiServiceClientConfiguration
 
 if TYPE_CHECKING:
@@ -19,14 +17,6 @@
     from .._serialization import Deserializer, Serializer
 
 
-def _convert_request(request, files=None):
-    data = request.content if not files else None
-    request = HttpRequest(method=request.method, url=request.url, headers=request.headers, data=data)
-    if files:
-        request.set_formdata_body(files)
-    return request
-
-
 class MultiapiServiceClientMixinABC(ABC):
     """DO NOT use this class. It is for internal typing use only."""
 

packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/aio/_vendor.py

@@ -8,8 +8,6 @@
 from abc import ABC
 from typing import TYPE_CHECKING
 
-from azure.core.pipeline.transport import HttpRequest
-
 from ._configuration import MultiapiServiceClientConfiguration
 
 if TYPE_CHECKING: