CodeGen from PR 33408 in Azure/azure-rest-api-specs

Merge a2a19fe2794df4c00d7159fc4df51639c8439563 into 2c694499ee8fb4c12641bc7d4ce7e57cdd138deb

Author: SDKAuto

sdk/batch/azure-batch/MANIFEST.in

@@ -3,4 +3,4 @@ include LICENSE
 include azure/batch/py.typed
 recursive-include tests *.py
 recursive-include samples *.py *.md
-include azure/__init__.py
+include azure/__init__.py

sdk/batch/azure-batch/_meta.json

@@ -0,0 +1,6 @@
+{
+  "commit": "b5fd8cca6d08621c744477322c3fd6ecb6a2c9a9",
+  "repository_url": "https://github.com/Azure/azure-rest-api-specs",
+  "typespec_src": "specification/batch/Azure.Batch",
+  "@azure-tools/typespec-python": "0.42.2"
+}

sdk/batch/azure-batch/apiview-properties.json

@@ -2,8 +2,9 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
-# Licensed under the MIT License. See License.txt in the project root for
-# license information.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) Python Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 # pylint: disable=protected-access, broad-except
 

sdk/batch/azure-batch/azure/batch/_model_base.py

@@ -1,162 +1,15 @@
-# ------------------------------------
-# Copyright (c) Microsoft Corporation.
-# Licensed under the MIT License.
-# ------------------------------------
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------
 """Customize generated code here.
 
 Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize
 """
-import base64
-import hmac
-import hashlib
-import importlib
-from datetime import datetime
-from typing import TYPE_CHECKING, TypeVar, Any, Union
+from typing import List
 
-from azure.core.pipeline.policies import SansIOHTTPPolicy
-from azure.core.credentials import AzureNamedKeyCredential, TokenCredential
-from azure.core.pipeline import PipelineResponse, PipelineRequest
-from azure.core.pipeline.transport import HttpResponse
-from azure.core.rest import HttpRequest
-
-from ._client import BatchClient as GenerateBatchClient
-from ._serialization import (
-    Serializer,
-    TZ_UTC,
-)
-
-try:
-    from urlparse import urlparse, parse_qs
-except ImportError:
-    from urllib.parse import urlparse, parse_qs
-__all__ = [
-    "BatchClient",
-]  # Add all objects you want publicly available to users at this package level
-
-if TYPE_CHECKING:
-    # pylint: disable=unused-import,ungrouped-imports
-    from typing import Any, Callable, Dict, Optional, TypeVar, Union
-
-    from azure.core.credentials import TokenCredential
-    from azure.core.pipeline import PipelineRequest
-
-    ClientType = TypeVar("ClientType", bound="BatchClient")
-    T = TypeVar("T")
-    ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]]
-
-
-class BatchSharedKeyAuthPolicy(SansIOHTTPPolicy):
-
-    headers_to_sign = [
-        "content-encoding",
-        "content-language",
-        "content-length",
-        "content-md5",
-        "content-type",
-        "date",
-        "if-modified-since",
-        "if-match",
-        "if-none-match",
-        "if-unmodified-since",
-        "range",
-    ]
-
-    def __init__(self, credential: AzureNamedKeyCredential):
-        super(BatchSharedKeyAuthPolicy, self).__init__()
-        self._account_name = credential.named_key.name
-        self._key = credential.named_key.key
-
-    def on_request(self, request: PipelineRequest):
-        if not request.http_request.headers.get("ocp-date"):
-            now = datetime.utcnow()
-            now = now.replace(tzinfo=TZ_UTC)
-            request.http_request.headers["ocp-date"] = Serializer.serialize_rfc(now)
-        url = urlparse(request.http_request.url)
-        uri_path = url.path
-
-        # method to sign
-        string_to_sign = request.http_request.method + "\n"
-
-        # get headers to sign
-        request_header_dict = {key.lower(): val for key, val in request.http_request.headers.items() if val}
-
-        if request.http_request.method not in ["GET", "HEAD"]:
-            if "content-length" not in request_header_dict:
-                request_header_dict["content-length"] = "0"
-
-        request_headers = [str(request_header_dict.get(x, "")) for x in self.headers_to_sign]
-
-        string_to_sign += "\n".join(request_headers) + "\n"
-
-        # get ocp- header to sign
-        ocp_headers = []
-        for name, value in request.http_request.headers.items():
-            if "ocp-" in name and value:
-                ocp_headers.append((name.lower(), value))
-        for name, value in sorted(ocp_headers):
-            string_to_sign += "{}:{}\n".format(name, value)
-        # get account_name and uri path to sign
-        string_to_sign += "/{}{}".format(self._account_name, uri_path)
-
-        # get query string to sign if it is not table service
-        query_to_sign = parse_qs(url.query)
-
-        for name in sorted(query_to_sign.keys()):
-            value = query_to_sign[name][0]
-            if value:
-                string_to_sign += "\n{}:{}".format(name, value)
-        # sign the request
-        auth_string = "SharedKey {}:{}".format(self._account_name, self._sign_string(string_to_sign))
-
-        request.http_request.headers["Authorization"] = auth_string
-
-        return super().on_request(request)
-
-    def _sign_string(self, string_to_sign):
-
-        _key = self._key.encode("utf-8")
-        string_to_sign = string_to_sign.encode("utf-8")
-
-        try:
-            key = base64.b64decode(_key)
-        except TypeError:
-            raise ValueError("Invalid key value: {}".format(self._key))
-        signed_hmac_sha256 = hmac.HMAC(key, string_to_sign, hashlib.sha256)
-        digest = signed_hmac_sha256.digest()
-
-        return base64.b64encode(digest).decode("utf-8")
-
-
-class BatchClient(GenerateBatchClient):
-    """BatchClient.
-
-    :param endpoint: HTTP or HTTPS endpoint for the Web PubSub service instance.
-    :type endpoint: str
-    :param hub: Target hub name, which should start with alphabetic characters and only contain
-     alpha-numeric characters or underscore.
-    :type hub: str
-    :param credentials: Credential needed for the client to connect to Azure.
-    :type credentials: ~azure.identity.ClientSecretCredential, ~azure.core.credentials.AzureNamedKeyCredential,
-     or ~azure.identity.TokenCredentials
-    :keyword api_version: Api Version. The default value is "2021-10-01". Note that overriding this
-     default value may result in unsupported behavior.
-    :paramtype api_version: str
-    """
-
-    def __init__(self, endpoint: str, credential: Union[AzureNamedKeyCredential, TokenCredential], **kwargs):
-        super().__init__(
-            endpoint=endpoint,
-            credential=credential, # type: ignore
-            authentication_policy=kwargs.pop(
-                "authentication_policy", self._format_shared_key_credential("", credential)
-            ),
-            **kwargs
-        )
-
-    def _format_shared_key_credential(self, account_name, credential):
-        if isinstance(credential, AzureNamedKeyCredential):
-            return BatchSharedKeyAuthPolicy(credential)
-        return None
+__all__: List[str] = []  # Add all objects you want publicly available to users at this package level
 
 
 def patch_sdk():

sdk/batch/azure-batch/azure/batch/_operations/_operations.py

@@ -1,28 +1,10 @@
 # pylint: disable=line-too-long,useless-suppression,too-many-lines
+# coding=utf-8
 # --------------------------------------------------------------------------
-#
 # Copyright (c) Microsoft Corporation. All rights reserved.
-#
-# The MIT License (MIT)
-#
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the ""Software""), to
-# deal in the Software without restriction, including without limitation the
-# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
-# sell copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-#
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
-# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
-# IN THE SOFTWARE.
-#
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) Python Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
 # pyright: reportUnnecessaryTypeIgnoreComment=false

sdk/batch/azure-batch/azure/batch/_operations/_patch.py

@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "15.0.0b2"
+VERSION = "1.0.0b1"