@autorest/python 6.7.0 → 6.7.2

package/autorest/multiapi/__init__.py

@@ -6,7 +6,6 @@
 import sys
 import logging
 import json
-import shutil
 
 from collections import defaultdict
 from pathlib import Path
@@ -174,10 +173,6 @@ class MultiAPI(ReaderAndWriter):  # pylint: disable=abstract-method
             user_specified_default_api=self.user_specified_default_api,
         )
 
-        # In case we are transitioning from a single api generation, clean old folders
-        shutil.rmtree(str(self.output_folder / "operations"), ignore_errors=True)
-        shutil.rmtree(str(self.output_folder / "models"), ignore_errors=True)
-
         multiapi_serializer = self.serializer
         multiapi_serializer.serialize(code_model, self.no_async)
 

package/autorest/preprocess/__init__.py

@@ -176,6 +176,37 @@ def update_paging_response(yaml_data: Dict[str, Any]) -> None:
     )
 
 
+HEADERS_HIDE_IN_METHOD = (
+    "repeatability-request-id",
+    "repeatability-first-sent",
+    "x-ms-client-request-id",
+    "client-request-id",
+    "return-client-request-id",
+)
+HEADERS_CONVERT_IN_METHOD = {
+    "if-match": {
+        "clientName": "etag",
+        "wireName": "etag",
+        "description": "check if resource is changed. Set None to skip checking etag.",
+    },
+    "if-none-match": {
+        "clientName": "match_condition",
+        "wireName": "match-condition",
+        "description": "The match condition to use upon the etag.",
+        "type": {
+            "type": "azurecore",
+            "name": "MatchConditions",
+        },
+    },
+}
+
+
+def headers_convert(yaml_data: Dict[str, Any], replace_data: Any) -> None:
+    if isinstance(replace_data, dict):
+        for k, v in replace_data.items():
+            yaml_data[k] = v
+
+
 class PreProcessPlugin(YamlUpdatePlugin):  # pylint: disable=abstract-method
     """Add Python naming information."""
 
@@ -242,6 +273,11 @@ class PreProcessPlugin(YamlUpdatePlugin):  # pylint: disable=abstract-method
                         )
                 type["values"].extend(values_to_add)
 
+        # add type for reference
+        for v in HEADERS_CONVERT_IN_METHOD.values():
+            if isinstance(v, dict) and "type" in v:
+                yaml_data.append(v["type"])
+
     def update_client(self, yaml_data: Dict[str, Any]) -> None:
         yaml_data["description"] = update_description(
             yaml_data["description"], default_description=yaml_data["name"]
@@ -253,6 +289,21 @@ class PreProcessPlugin(YamlUpdatePlugin):  # pylint: disable=abstract-method
         if prop_name.endswith("Client"):
             prop_name = prop_name[: len(prop_name) - len("Client")]
         yaml_data["builderPadName"] = to_snake_case(prop_name)
+        for og in yaml_data["operationGroups"]:
+            for o in og["operations"]:
+                for p in o["parameters"]:
+                    if (
+                        p["location"] == "header"
+                        and p["wireName"] == "client-request-id"
+                    ):
+                        yaml_data["requestIdHeaderName"] = p["wireName"]
+                    if (
+                        self.version_tolerant
+                        and p["location"] == "header"
+                        and p["clientName"] in ("if_match", "if_none_match")
+                    ):
+                        o["hasEtag"] = True
+                        yaml_data["hasEtag"] = True
 
     def get_operation_updater(
         self, yaml_data: Dict[str, Any]
@@ -282,6 +333,18 @@ class PreProcessPlugin(YamlUpdatePlugin):  # pylint: disable=abstract-method
                 .lower()
                 for prop, param_name in yaml_data["propertyToParameterName"].items()
             }
+        wire_name_lower = (yaml_data.get("wireName") or "").lower()
+        if (
+            yaml_data["location"] == "header"
+            and wire_name_lower in HEADERS_HIDE_IN_METHOD
+        ):
+            yaml_data["hideInMethod"] = True
+        if (
+            self.version_tolerant
+            and yaml_data["location"] == "header"
+            and wire_name_lower in HEADERS_CONVERT_IN_METHOD
+        ):
+            headers_convert(yaml_data, HEADERS_CONVERT_IN_METHOD[wire_name_lower])
 
     def update_operation(
         self,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autorest/python",
-  "version": "6.7.0",
+  "version": "6.7.2",
   "description": "The Python extension for generators in AutoRest.",
   "main": "index.js",
   "repository": {