diff --git a/doc/dev/request_parameters_migration_guide.md b/doc/dev/request_parameters_migration_guide.md new file mode 100644 index 000000000000..409d08495f90 --- /dev/null +++ b/doc/dev/request_parameters_migration_guide.md @@ -0,0 +1,208 @@ +# Migration Guide: Converting `**kwargs` to `request_parameters` TypedDict + +This guide describes how to convert Azure SDK for Python service client APIs from using +untyped `**kwargs` for request configuration to using a typed `request_parameters` dictionary. + +## Overview + +Previously, service client operation methods accepted infrastructure-level parameters +(headers, API version, polling configuration, etc.) through `**kwargs`. This made it +difficult for users to discover what parameters were available and caused type-checking +tools to miss errors. + +The new pattern introduces `RequestParameters` TypedDict classes that explicitly declare +the available infrastructure parameters for each type of operation. + +## Step-by-Step Migration + +### Step 1: Create the `types` module + +Create a `types.py` file in the library's package directory (next to `__init__.py`). This +module defines TypedDict classes for request parameters. + +There are three tiers of TypedDict, reflecting the three categories of operations: + +1. **`RequestParameters`** — base parameters common to all operations (GET, simple POST, etc.) +2. **`LRORequestParameters`** — extends `RequestParameters` with long-running operation parameters +3. **`PagingRequestParameters`** — for paginated list operations (same as `RequestParameters` but + documents that `cls` callback receives page elements) + +```python +# types.py +from typing import Callable, Dict, List, Union + +from typing_extensions import TypedDict + +from azure.core.pipeline import PipelineRequest, PipelineResponse +from azure.core.polling import PollingMethod + + +class RequestParameters(TypedDict, total=False): + """Common parameters for all service client operations. + + These parameters control how the HTTP request is constructed and sent + through the Azure SDK pipeline, including retry, redirect, tracing, + logging, and transport-level behavior. + """ + + api_version: str + headers: Dict[str, str] + params: Dict[str, str] + content_type: str + request_id: str + stream: bool + decompress: bool + raw_request_hook: Callable[[PipelineRequest], None] + raw_response_hook: Callable[[PipelineResponse], None] + logging_enable: bool + retry_total: int + retry_connect: int + retry_read: int + retry_status: int + retry_backoff_factor: float + retry_backoff_max: int + retry_on_methods: List[str] + timeout: int + permit_redirects: bool + redirect_max: int + network_span_namer: Callable + tracing_attributes: Dict[str, str] + connection_timeout: float + read_timeout: float + connection_verify: Union[bool, str] + connection_cert: Union[str, tuple] + proxy: str + + +class LRORequestParameters(RequestParameters, total=False): + """Parameters for long-running operations (begin_* methods). + + Extends RequestParameters with polling and continuation support. + """ + + polling: Union[bool, PollingMethod] + polling_interval: int + continuation_token: str +``` + +### Step 2: Export types from the package + +Add the TypedDict classes to the package's `__init__.py` or ensure they are importable +from a `types` submodule. + +### Step 3: Add `request_parameters` to operation method signatures + +For each public method in the operations classes: + +**Before:** +```python +@distributed_trace +def get( + self, + resource_group_name: str, + name: str, + expand: Optional[str] = None, + **kwargs: Any +) -> Model: +``` + +**After:** +```python +@distributed_trace +def get( + self, + resource_group_name: str, + name: str, + expand: Optional[str] = None, + *, + request_parameters: Optional[RequestParameters] = None, + **kwargs: Any +) -> Model: +``` + +For LRO methods (`begin_*`), use `LRORequestParameters`: + +```python +@distributed_trace +def begin_create( + self, + resource_group_name: str, + name: str, + parameters: Union[Model, IO[bytes]], + *, + request_parameters: Optional[LRORequestParameters] = None, + **kwargs: Any +) -> LROPoller[Model]: +``` + +### Step 4: Merge `request_parameters` into kwargs at the start of each method + +Add a merge step at the beginning of each method body to combine `request_parameters` +into kwargs, giving `request_parameters` lower priority (kwargs override): + +```python +def get(self, ..., *, request_parameters=None, **kwargs): + # Merge request_parameters into kwargs (kwargs take precedence) + if request_parameters: + for key, value in request_parameters.items(): + kwargs.setdefault(key, value) + + # Existing method body continues unchanged... + error_map = { ... } + error_map.update(kwargs.pop("error_map", {}) or {}) + _headers = kwargs.pop("headers", {}) or {} + ... +``` + +This approach ensures: +- Full backward compatibility (`**kwargs` still works) +- `request_parameters` provides discoverability and type checking +- If both are specified, explicit `**kwargs` values take precedence + +**Precedence rule**: When a user passes the same parameter through both +`request_parameters` and `**kwargs`, the `**kwargs` value wins. This is because +`setdefault` only sets a key if it's not already present in `kwargs`. This behavior +is intentional — it allows `request_parameters` to provide defaults while explicit +keyword arguments can still override them. Example: + +```python +# In this call, headers from **kwargs wins: +client.vms.get("rg", "vm", request_parameters={"headers": {"X-From-RP": "1"}}, headers={"X-From-KW": "1"}) +# Result: headers = {"X-From-KW": "1"} +``` + +### Step 5: Update overloads + +If a method has `@overload` decorators, add `request_parameters` to each overload +signature as well. + +### Step 6: Update the async variant + +The `aio/operations/` module mirrors the sync operations. Apply the same changes +to async method signatures. + +### Step 7: Update `__init__.py` exports + +Make sure the `types` module types are accessible to users: + +```python +# In the package __init__.py or types module +from .types import RequestParameters, LRORequestParameters +``` + +## Checklist for Each Library + +- [ ] Create `types.py` with `RequestParameters` and `LRORequestParameters` +- [ ] Add `request_parameters` parameter to all public sync operation methods +- [ ] Add `request_parameters` parameter to all public async operation methods +- [ ] Update `@overload` signatures if present +- [ ] Export types from package `__init__.py` +- [ ] Verify backward compatibility (existing `**kwargs` usage still works) +- [ ] Run existing tests to ensure no regressions + +## Notes + +- The `request_parameters` parameter is always optional and keyword-only +- All fields in the TypedDict use `total=False` (all optional) +- The method body logic is unchanged; only the signature and a small merge step are added +- This is a purely additive, non-breaking change diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/__init__.py b/sdk/ai/azure-ai-projects/azure/ai/projects/__init__.py index 743119593f69..d6819c152d54 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/__init__.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/__init__.py @@ -14,6 +14,7 @@ from ._client import AIProjectClient # type: ignore from ._version import VERSION +from .types import LRORequestParameters, RequestParameters __version__ = VERSION @@ -26,6 +27,8 @@ __all__ = [ "AIProjectClient", + "RequestParameters", + "LRORequestParameters", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_operations.py b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_operations.py index c5bbefb67fe9..7692face15d6 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_operations.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_operations.py @@ -35,6 +35,7 @@ from ... import models as _models from ..._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._operations import ( build_agents_create_from_manifest_request, build_agents_create_request, @@ -129,7 +130,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def get(self, agent_name: str, **kwargs: Any) -> _models.AgentDetails: + async def get(self, agent_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AgentDetails: """Retrieves the agent. :param agent_name: The name of the agent to retrieve. Required. @@ -138,6 +139,10 @@ async def get(self, agent_name: str, **kwargs: Any) -> _models.AgentDetails: :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -201,7 +206,7 @@ async def create( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates the agent. @@ -234,7 +239,7 @@ async def create( @overload async def create( - self, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates the agent. @@ -250,7 +255,7 @@ async def create( @overload async def create( - self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates the agent. @@ -273,7 +278,7 @@ async def create( definition: _models.AgentDefinition = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates the agent. @@ -302,6 +307,10 @@ async def create( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -381,7 +390,7 @@ async def update( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -410,7 +419,7 @@ async def update( @overload async def update( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -429,7 +438,7 @@ async def update( @overload async def update( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -455,7 +464,7 @@ async def update( definition: _models.AgentDefinition = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -480,6 +489,10 @@ async def update( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -559,7 +572,7 @@ async def create_from_manifest( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -594,7 +607,7 @@ async def create_from_manifest( @overload async def create_from_manifest( - self, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -610,7 +623,7 @@ async def create_from_manifest( @overload async def create_from_manifest( - self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -634,7 +647,7 @@ async def create_from_manifest( parameter_values: dict[str, Any] = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -665,6 +678,10 @@ async def create_from_manifest( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -753,7 +770,7 @@ async def update_from_manifest( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -785,7 +802,7 @@ async def update_from_manifest( @overload async def update_from_manifest( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -805,7 +822,7 @@ async def update_from_manifest( @overload async def update_from_manifest( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -833,7 +850,7 @@ async def update_from_manifest( parameter_values: dict[str, Any] = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -861,6 +878,10 @@ async def update_from_manifest( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -938,7 +959,7 @@ async def update_from_manifest( return deserialized # type: ignore @distributed_trace_async - async def delete(self, agent_name: str, **kwargs: Any) -> _models.DeleteAgentResponse: + async def delete(self, agent_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DeleteAgentResponse: """Deletes an agent. :param agent_name: The name of the agent to delete. Required. @@ -947,6 +968,10 @@ async def delete(self, agent_name: str, **kwargs: Any) -> _models.DeleteAgentRes :rtype: ~azure.ai.projects.models.DeleteAgentResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1009,7 +1034,7 @@ def list( limit: Optional[int] = None, order: Optional[Union[str, _models.PageOrder]] = None, before: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.AgentDetails"]: """Returns the list of all agents. @@ -1034,6 +1059,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.AgentDetails] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -1102,7 +1131,7 @@ async def create_version( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -1135,7 +1164,7 @@ async def create_version( @overload async def create_version( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -1158,7 +1187,7 @@ async def create_version( @overload async def create_version( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -1188,7 +1217,7 @@ async def create_version( definition: _models.AgentDefinition = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -1217,6 +1246,10 @@ async def create_version( :rtype: ~azure.ai.projects.models.AgentVersionDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1296,7 +1329,7 @@ async def create_version_from_manifest( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -1331,7 +1364,7 @@ async def create_version_from_manifest( @overload async def create_version_from_manifest( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -1354,7 +1387,7 @@ async def create_version_from_manifest( @overload async def create_version_from_manifest( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -1385,7 +1418,7 @@ async def create_version_from_manifest( parameter_values: dict[str, Any] = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -1416,6 +1449,10 @@ async def create_version_from_manifest( :rtype: ~azure.ai.projects.models.AgentVersionDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1493,7 +1530,7 @@ async def create_version_from_manifest( return deserialized # type: ignore @distributed_trace_async - async def get_version(self, agent_name: str, agent_version: str, **kwargs: Any) -> _models.AgentVersionDetails: + async def get_version(self, agent_name: str, agent_version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AgentVersionDetails: """Retrieves a specific version of an agent. :param agent_name: The name of the agent to retrieve. Required. @@ -1504,6 +1541,10 @@ async def get_version(self, agent_name: str, agent_version: str, **kwargs: Any) :rtype: ~azure.ai.projects.models.AgentVersionDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1561,7 +1602,7 @@ async def get_version(self, agent_name: str, agent_version: str, **kwargs: Any) @distributed_trace_async async def delete_version( - self, agent_name: str, agent_version: str, **kwargs: Any + self, agent_name: str, agent_version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DeleteAgentVersionResponse: """Deletes a specific version of an agent. @@ -1574,6 +1615,10 @@ async def delete_version( :rtype: ~azure.ai.projects.models.DeleteAgentVersionResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1637,7 +1682,7 @@ def list_versions( limit: Optional[int] = None, order: Optional[Union[str, _models.PageOrder]] = None, before: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.AgentVersionDetails"]: """Returns the list of versions of an agent. @@ -1729,7 +1774,7 @@ async def stream_agent_container_logs( kind: Optional[Union[str, _models.ContainerLogKind]] = None, replica_name: Optional[str] = None, tail: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Container log entry streamed from the container as text chunks. Each chunk is a UTF-8 string that may be either a plain text log line @@ -1774,6 +1819,10 @@ async def stream_agent_container_logs( :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1847,7 +1896,7 @@ async def create( content_type: str = "application/json", description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -1870,7 +1919,7 @@ async def create( @overload async def create( - self, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -1886,7 +1935,7 @@ async def create( @overload async def create( - self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -1909,7 +1958,7 @@ async def create( definition: _models.MemoryStoreDefinition = _Unset, description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -1928,6 +1977,10 @@ async def create( :rtype: ~azure.ai.projects.models.MemoryStoreDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2006,7 +2059,7 @@ async def update( content_type: str = "application/json", description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -2027,7 +2080,7 @@ async def update( @overload async def update( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -2045,7 +2098,7 @@ async def update( @overload async def update( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -2069,7 +2122,7 @@ async def update( *, description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -2086,6 +2139,10 @@ async def update( :rtype: ~azure.ai.projects.models.MemoryStoreDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2154,7 +2211,7 @@ async def update( return deserialized # type: ignore @distributed_trace_async - async def get(self, name: str, **kwargs: Any) -> _models.MemoryStoreDetails: + async def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.MemoryStoreDetails: """Retrieve a memory store. :param name: The name of the memory store to retrieve. Required. @@ -2163,6 +2220,10 @@ async def get(self, name: str, **kwargs: Any) -> _models.MemoryStoreDetails: :rtype: ~azure.ai.projects.models.MemoryStoreDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2224,7 +2285,7 @@ def list( limit: Optional[int] = None, order: Optional[Union[str, _models.PageOrder]] = None, before: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.MemoryStoreDetails"]: """List all memory stores. @@ -2305,7 +2366,7 @@ async def get_next(_continuation_token=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def delete(self, name: str, **kwargs: Any) -> _models.DeleteMemoryStoreResult: + async def delete(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DeleteMemoryStoreResult: """Delete a memory store. :param name: The name of the memory store to delete. Required. @@ -2314,6 +2375,10 @@ async def delete(self, name: str, **kwargs: Any) -> _models.DeleteMemoryStoreRes :rtype: ~azure.ai.projects.models.DeleteMemoryStoreResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2378,7 +2443,7 @@ async def search_memories( items: Optional[List[_models.InputItem]] = None, previous_search_id: Optional[str] = None, options: Optional[_models.MemorySearchOptions] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -2404,7 +2469,7 @@ async def search_memories( @overload async def search_memories( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -2422,7 +2487,7 @@ async def search_memories( @overload async def search_memories( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -2448,7 +2513,7 @@ async def search_memories( items: Optional[List[_models.InputItem]] = None, previous_search_id: Optional[str] = None, options: Optional[_models.MemorySearchOptions] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -2470,6 +2535,10 @@ async def search_memories( :rtype: ~azure.ai.projects.models.MemoryStoreSearchResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2750,7 +2819,7 @@ def get_long_running_output(pipeline_response): @overload async def delete_scope( - self, name: str, *, scope: str, content_type: str = "application/json", **kwargs: Any + self, name: str, *, scope: str, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -2770,7 +2839,7 @@ async def delete_scope( @overload async def delete_scope( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -2789,7 +2858,7 @@ async def delete_scope( @overload async def delete_scope( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -2808,7 +2877,7 @@ async def delete_scope( @distributed_trace_async async def delete_scope( - self, name: str, body: Union[JSON, IO[bytes]] = _Unset, *, scope: str = _Unset, **kwargs: Any + self, name: str, body: Union[JSON, IO[bytes]] = _Unset, *, scope: str = _Unset, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -2824,6 +2893,10 @@ async def delete_scope( :rtype: ~azure.ai.projects.models.MemoryStoreDeleteScopeResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3047,7 +3120,7 @@ def list( *, connection_type: Optional[Union[str, _models.ConnectionType]] = None, default_connection: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.Connection"]: """List all connections in the project, without populating connection credentials. @@ -3062,6 +3135,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.Connection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -3157,7 +3234,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_versions(self, name: str, **kwargs: Any) -> AsyncItemPaged["_models.DatasetVersion"]: + def list_versions(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.DatasetVersion"]: """List all versions of the given DatasetVersion. :param name: The name of the resource. Required. @@ -3166,6 +3243,10 @@ def list_versions(self, name: str, **kwargs: Any) -> AsyncItemPaged["_models.Dat :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.DatasetVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -3242,13 +3323,17 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.DatasetVersion"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.DatasetVersion"]: """List the latest version of each DatasetVersion. :return: An iterator like instance of DatasetVersion :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.DatasetVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -3324,7 +3409,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, name: str, version: str, **kwargs: Any) -> _models.DatasetVersion: + async def get(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DatasetVersion: """Get the specific version of the DatasetVersion. The service returns 404 Not Found error if the DatasetVersion does not exist. @@ -3336,6 +3421,10 @@ async def get(self, name: str, version: str, **kwargs: Any) -> _models.DatasetVe :rtype: ~azure.ai.projects.models.DatasetVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3388,7 +3477,7 @@ async def get(self, name: str, version: str, **kwargs: Any) -> _models.DatasetVe return deserialized # type: ignore @distributed_trace_async - async def delete(self, name: str, version: str, **kwargs: Any) -> None: + async def delete(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete the specific version of the DatasetVersion. The service returns 204 No Content if the DatasetVersion was deleted successfully or if the DatasetVersion does not exist. @@ -3400,6 +3489,10 @@ async def delete(self, name: str, version: str, **kwargs: Any) -> None: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3447,7 +3540,7 @@ async def create_or_update( dataset_version: _models.DatasetVersion, *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -3473,7 +3566,7 @@ async def create_or_update( dataset_version: JSON, *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -3499,7 +3592,7 @@ async def create_or_update( dataset_version: IO[bytes], *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -3519,7 +3612,7 @@ async def create_or_update( @distributed_trace_async async def create_or_update( - self, name: str, version: str, dataset_version: Union[_models.DatasetVersion, JSON, IO[bytes]], **kwargs: Any + self, name: str, version: str, dataset_version: Union[_models.DatasetVersion, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -3534,6 +3627,10 @@ async def create_or_update( :rtype: ~azure.ai.projects.models.DatasetVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3603,7 +3700,7 @@ async def pending_upload( pending_upload_request: _models.PendingUploadRequest, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -3629,7 +3726,7 @@ async def pending_upload( pending_upload_request: JSON, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -3655,7 +3752,7 @@ async def pending_upload( pending_upload_request: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -3679,7 +3776,7 @@ async def pending_upload( name: str, version: str, pending_upload_request: Union[_models.PendingUploadRequest, JSON, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -3695,6 +3792,10 @@ async def pending_upload( :rtype: ~azure.ai.projects.models.PendingUploadResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3757,7 +3858,7 @@ async def pending_upload( return deserialized # type: ignore @distributed_trace_async - async def get_credentials(self, name: str, version: str, **kwargs: Any) -> _models.DatasetCredential: + async def get_credentials(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DatasetCredential: """Get the SAS credential to access the storage account associated with a Dataset version. :param name: The name of the resource. Required. @@ -3768,6 +3869,10 @@ async def get_credentials(self, name: str, version: str, **kwargs: Any) -> _mode :rtype: ~azure.ai.projects.models.DatasetCredential :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3838,7 +3943,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_versions(self, name: str, **kwargs: Any) -> AsyncItemPaged["_models.Index"]: + def list_versions(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Index"]: """List all versions of the given Index. :param name: The name of the resource. Required. @@ -3847,6 +3952,10 @@ def list_versions(self, name: str, **kwargs: Any) -> AsyncItemPaged["_models.Ind :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.Index] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -3923,13 +4032,17 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Index"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Index"]: """List the latest version of each Index. :return: An iterator like instance of Index :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.Index] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -4005,7 +4118,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, name: str, version: str, **kwargs: Any) -> _models.Index: + async def get(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Index: """Get the specific version of the Index. The service returns 404 Not Found error if the Index does not exist. @@ -4017,6 +4130,10 @@ async def get(self, name: str, version: str, **kwargs: Any) -> _models.Index: :rtype: ~azure.ai.projects.models.Index :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4069,7 +4186,7 @@ async def get(self, name: str, version: str, **kwargs: Any) -> _models.Index: return deserialized # type: ignore @distributed_trace_async - async def delete(self, name: str, version: str, **kwargs: Any) -> None: + async def delete(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete the specific version of the Index. The service returns 204 No Content if the Index was deleted successfully or if the Index does not exist. @@ -4081,6 +4198,10 @@ async def delete(self, name: str, version: str, **kwargs: Any) -> None: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4128,7 +4249,7 @@ async def create_or_update( index: _models.Index, *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -4148,7 +4269,7 @@ async def create_or_update( @overload async def create_or_update( - self, name: str, version: str, index: JSON, *, content_type: str = "application/merge-patch+json", **kwargs: Any + self, name: str, version: str, index: JSON, *, content_type: str = "application/merge-patch+json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -4174,7 +4295,7 @@ async def create_or_update( index: IO[bytes], *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -4194,7 +4315,7 @@ async def create_or_update( @distributed_trace_async async def create_or_update( - self, name: str, version: str, index: Union[_models.Index, JSON, IO[bytes]], **kwargs: Any + self, name: str, version: str, index: Union[_models.Index, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -4209,6 +4330,10 @@ async def create_or_update( :rtype: ~azure.ai.projects.models.Index :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4289,7 +4414,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def get(self, name: str, **kwargs: Any) -> _models.Deployment: + async def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Deployment: """Get a deployed model. :param name: Name of the deployment. Required. @@ -4298,6 +4423,10 @@ async def get(self, name: str, **kwargs: Any) -> _models.Deployment: :rtype: ~azure.ai.projects.models.Deployment :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4360,7 +4489,7 @@ def list( model_publisher: Optional[str] = None, model_name: Optional[str] = None, deployment_type: Optional[Union[str, _models.DeploymentType]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.Deployment"]: """List all deployed models in the project. @@ -4376,6 +4505,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.Deployment] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -4472,7 +4605,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def get(self, name: str, **kwargs: Any) -> _models.RedTeam: + async def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RedTeam: """Get a redteam by name. :param name: Identifier of the red team run. Required. @@ -4481,6 +4614,10 @@ async def get(self, name: str, **kwargs: Any) -> _models.RedTeam: :rtype: ~azure.ai.projects.models.RedTeam :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4537,13 +4674,17 @@ async def get(self, name: str, **kwargs: Any) -> _models.RedTeam: return deserialized # type: ignore @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.RedTeam"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.RedTeam"]: """List a redteam by name. :return: An iterator like instance of RedTeam :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.RedTeam] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -4620,7 +4761,7 @@ async def get_next(next_link=None): @overload async def create( - self, red_team: _models.RedTeam, *, content_type: str = "application/json", **kwargs: Any + self, red_team: _models.RedTeam, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RedTeam: """Creates a redteam run. @@ -4635,7 +4776,7 @@ async def create( """ @overload - async def create(self, red_team: JSON, *, content_type: str = "application/json", **kwargs: Any) -> _models.RedTeam: + async def create(self, red_team: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RedTeam: """Creates a redteam run. :param red_team: Redteam to be run. Required. @@ -4650,7 +4791,7 @@ async def create(self, red_team: JSON, *, content_type: str = "application/json" @overload async def create( - self, red_team: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, red_team: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RedTeam: """Creates a redteam run. @@ -4665,7 +4806,7 @@ async def create( """ @distributed_trace_async - async def create(self, red_team: Union[_models.RedTeam, JSON, IO[bytes]], **kwargs: Any) -> _models.RedTeam: + async def create(self, red_team: Union[_models.RedTeam, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RedTeam: """Creates a redteam run. :param red_team: Redteam to be run. Is one of the following types: RedTeam, JSON, IO[bytes] @@ -4675,6 +4816,10 @@ async def create(self, red_team: Union[_models.RedTeam, JSON, IO[bytes]], **kwar :rtype: ~azure.ai.projects.models.RedTeam :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4753,7 +4898,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def get(self, id: str, **kwargs: Any) -> _models.EvaluationRule: + async def get(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.EvaluationRule: """Get an evaluation rule. :param id: Unique identifier for the evaluation rule. Required. @@ -4762,6 +4907,10 @@ async def get(self, id: str, **kwargs: Any) -> _models.EvaluationRule: :rtype: ~azure.ai.projects.models.EvaluationRule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4818,7 +4967,7 @@ async def get(self, id: str, **kwargs: Any) -> _models.EvaluationRule: return deserialized # type: ignore @distributed_trace_async - async def delete(self, id: str, **kwargs: Any) -> None: + async def delete(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete an evaluation rule. :param id: Unique identifier for the evaluation rule. Required. @@ -4827,6 +4976,10 @@ async def delete(self, id: str, **kwargs: Any) -> None: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4872,7 +5025,7 @@ async def delete(self, id: str, **kwargs: Any) -> None: @overload async def create_or_update( - self, id: str, evaluation_rule: _models.EvaluationRule, *, content_type: str = "application/json", **kwargs: Any + self, id: str, evaluation_rule: _models.EvaluationRule, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -4890,7 +5043,7 @@ async def create_or_update( @overload async def create_or_update( - self, id: str, evaluation_rule: JSON, *, content_type: str = "application/json", **kwargs: Any + self, id: str, evaluation_rule: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -4908,7 +5061,7 @@ async def create_or_update( @overload async def create_or_update( - self, id: str, evaluation_rule: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, id: str, evaluation_rule: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -4926,7 +5079,7 @@ async def create_or_update( @distributed_trace_async async def create_or_update( - self, id: str, evaluation_rule: Union[_models.EvaluationRule, JSON, IO[bytes]], **kwargs: Any + self, id: str, evaluation_rule: Union[_models.EvaluationRule, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -4939,6 +5092,10 @@ async def create_or_update( :rtype: ~azure.ai.projects.models.EvaluationRule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5006,7 +5163,7 @@ def list( action_type: Optional[Union[str, _models.EvaluationRuleActionType]] = None, agent_name: Optional[str] = None, enabled: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.EvaluationRule"]: """List all evaluation rules. @@ -5021,6 +5178,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.EvaluationRule] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -5117,7 +5278,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def get(self, name: str, **kwargs: Any) -> _models.EvaluationTaxonomy: + async def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.EvaluationTaxonomy: """Get an evaluation run by name. :param name: The name of the resource. Required. @@ -5126,6 +5287,10 @@ async def get(self, name: str, **kwargs: Any) -> _models.EvaluationTaxonomy: :rtype: ~azure.ai.projects.models.EvaluationTaxonomy :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5183,7 +5348,7 @@ async def get(self, name: str, **kwargs: Any) -> _models.EvaluationTaxonomy: @distributed_trace def list( - self, *, input_name: Optional[str] = None, input_type: Optional[str] = None, **kwargs: Any + self, *, input_name: Optional[str] = None, input_type: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.EvaluationTaxonomy"]: """List evaluation taxonomies. @@ -5195,6 +5360,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.EvaluationTaxonomy] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -5272,7 +5441,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def delete(self, name: str, **kwargs: Any) -> None: + async def delete(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete an evaluation taxonomy by name. :param name: The name of the resource. Required. @@ -5281,6 +5450,10 @@ async def delete(self, name: str, **kwargs: Any) -> None: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5326,7 +5499,7 @@ async def delete(self, name: str, **kwargs: Any) -> None: @overload async def create( - self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -5344,7 +5517,7 @@ async def create( @overload async def create( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -5362,7 +5535,7 @@ async def create( @overload async def create( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -5380,7 +5553,7 @@ async def create( @distributed_trace_async async def create( - self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], **kwargs: Any + self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -5393,6 +5566,10 @@ async def create( :rtype: ~azure.ai.projects.models.EvaluationTaxonomy :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5455,7 +5632,7 @@ async def create( @overload async def update( - self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -5473,7 +5650,7 @@ async def update( @overload async def update( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -5491,7 +5668,7 @@ async def update( @overload async def update( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -5509,7 +5686,7 @@ async def update( @distributed_trace_async async def update( - self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], **kwargs: Any + self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -5522,6 +5699,10 @@ async def update( :rtype: ~azure.ai.projects.models.EvaluationTaxonomy :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5607,7 +5788,7 @@ def list_versions( *, type: Optional[Union[Literal["builtin"], Literal["custom"], Literal["all"], str]] = None, limit: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.EvaluatorVersion"]: """List all versions of the given evaluator. @@ -5624,6 +5805,10 @@ def list_versions( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.EvaluatorVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -5707,7 +5892,7 @@ def list_latest_versions( *, type: Optional[Union[Literal["builtin"], Literal["custom"], Literal["all"], str]] = None, limit: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.EvaluatorVersion"]: """List the latest version of each evaluator. @@ -5722,6 +5907,10 @@ def list_latest_versions( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.EvaluatorVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -5799,7 +5988,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get_version(self, name: str, version: str, **kwargs: Any) -> _models.EvaluatorVersion: + async def get_version(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.EvaluatorVersion: """Get the specific version of the EvaluatorVersion. The service returns 404 Not Found error if the EvaluatorVersion does not exist. @@ -5811,6 +6000,10 @@ async def get_version(self, name: str, version: str, **kwargs: Any) -> _models.E :rtype: ~azure.ai.projects.models.EvaluatorVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5863,7 +6056,7 @@ async def get_version(self, name: str, version: str, **kwargs: Any) -> _models.E return deserialized # type: ignore @distributed_trace_async - async def delete_version(self, name: str, version: str, **kwargs: Any) -> None: + async def delete_version(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete the specific version of the EvaluatorVersion. The service returns 204 No Content if the EvaluatorVersion was deleted successfully or if the EvaluatorVersion does not exist. @@ -5875,6 +6068,10 @@ async def delete_version(self, name: str, version: str, **kwargs: Any) -> None: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5921,7 +6118,7 @@ async def create_version( evaluator_version: _models.EvaluatorVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -5939,7 +6136,7 @@ async def create_version( @overload async def create_version( - self, name: str, evaluator_version: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, evaluator_version: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -5957,7 +6154,7 @@ async def create_version( @overload async def create_version( - self, name: str, evaluator_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, evaluator_version: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -5975,7 +6172,7 @@ async def create_version( @distributed_trace_async async def create_version( - self, name: str, evaluator_version: Union[_models.EvaluatorVersion, JSON, IO[bytes]], **kwargs: Any + self, name: str, evaluator_version: Union[_models.EvaluatorVersion, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -5988,6 +6185,10 @@ async def create_version( :rtype: ~azure.ai.projects.models.EvaluatorVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6056,7 +6257,7 @@ async def update_version( evaluator_version: _models.EvaluatorVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -6076,7 +6277,7 @@ async def update_version( @overload async def update_version( - self, name: str, version: str, evaluator_version: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, version: str, evaluator_version: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -6102,7 +6303,7 @@ async def update_version( evaluator_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -6126,7 +6327,7 @@ async def update_version( name: str, version: str, evaluator_version: Union[_models.EvaluatorVersion, JSON, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -6141,6 +6342,10 @@ async def update_version( :rtype: ~azure.ai.projects.models.EvaluatorVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6222,7 +6427,7 @@ def __init__(self, *args, **kwargs) -> None: @overload async def generate( - self, insight: _models.Insight, *, content_type: str = "application/json", **kwargs: Any + self, insight: _models.Insight, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Insight: """Generate Insights. @@ -6239,7 +6444,7 @@ async def generate( @overload async def generate( - self, insight: JSON, *, content_type: str = "application/json", **kwargs: Any + self, insight: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Insight: """Generate Insights. @@ -6256,7 +6461,7 @@ async def generate( @overload async def generate( - self, insight: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, insight: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Insight: """Generate Insights. @@ -6272,7 +6477,7 @@ async def generate( """ @distributed_trace_async - async def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], **kwargs: Any) -> _models.Insight: + async def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Insight: """Generate Insights. :param insight: Complete evaluation configuration including data source, evaluators, and result @@ -6282,6 +6487,10 @@ async def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], **kwa :rtype: ~azure.ai.projects.models.Insight :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6342,7 +6551,7 @@ async def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], **kwa return deserialized # type: ignore @distributed_trace_async - async def get(self, id: str, *, include_coordinates: Optional[bool] = None, **kwargs: Any) -> _models.Insight: + async def get(self, id: str, *, include_coordinates: Optional[bool] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Insight: """Get a specific insight by Id. :param id: The unique identifier for the insights report. Required. @@ -6354,6 +6563,10 @@ async def get(self, id: str, *, include_coordinates: Optional[bool] = None, **kw :rtype: ~azure.ai.projects.models.Insight :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6419,7 +6632,7 @@ def list( run_id: Optional[str] = None, agent_name: Optional[str] = None, include_coordinates: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.Insight"]: """List all insights in reverse chronological order (newest first). @@ -6537,7 +6750,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def delete(self, id: str, **kwargs: Any) -> None: + async def delete(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete a schedule. :param id: Identifier of the schedule. Required. @@ -6546,6 +6759,10 @@ async def delete(self, id: str, **kwargs: Any) -> None: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6590,7 +6807,7 @@ async def delete(self, id: str, **kwargs: Any) -> None: return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace_async - async def get(self, id: str, **kwargs: Any) -> _models.Schedule: + async def get(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Schedule: """Get a schedule by id. :param id: Identifier of the schedule. Required. @@ -6599,6 +6816,10 @@ async def get(self, id: str, **kwargs: Any) -> _models.Schedule: :rtype: ~azure.ai.projects.models.Schedule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6655,13 +6876,17 @@ async def get(self, id: str, **kwargs: Any) -> _models.Schedule: return deserialized # type: ignore @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Schedule"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Schedule"]: """List all schedules. :return: An iterator like instance of Schedule :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.ai.projects.models.Schedule] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -6738,7 +6963,7 @@ async def get_next(next_link=None): @overload async def create_or_update( - self, id: str, schedule: _models.Schedule, *, content_type: str = "application/json", **kwargs: Any + self, id: str, schedule: _models.Schedule, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -6756,7 +6981,7 @@ async def create_or_update( @overload async def create_or_update( - self, id: str, schedule: JSON, *, content_type: str = "application/json", **kwargs: Any + self, id: str, schedule: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -6774,7 +6999,7 @@ async def create_or_update( @overload async def create_or_update( - self, id: str, schedule: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, id: str, schedule: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -6792,7 +7017,7 @@ async def create_or_update( @distributed_trace_async async def create_or_update( - self, id: str, schedule: Union[_models.Schedule, JSON, IO[bytes]], **kwargs: Any + self, id: str, schedule: Union[_models.Schedule, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -6805,6 +7030,10 @@ async def create_or_update( :rtype: ~azure.ai.projects.models.Schedule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6866,7 +7095,7 @@ async def create_or_update( return deserialized # type: ignore @distributed_trace_async - async def get_run(self, schedule_id: str, run_id: str, **kwargs: Any) -> _models.ScheduleRun: + async def get_run(self, schedule_id: str, run_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.ScheduleRun: """Get a schedule run by id. :param schedule_id: Identifier of the schedule. Required. @@ -6877,6 +7106,10 @@ async def get_run(self, schedule_id: str, run_id: str, **kwargs: Any) -> _models :rtype: ~azure.ai.projects.models.ScheduleRun :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6929,7 +7162,7 @@ async def get_run(self, schedule_id: str, run_id: str, **kwargs: Any) -> _models return deserialized # type: ignore @distributed_trace - def list_runs(self, id: str, **kwargs: Any) -> AsyncItemPaged["_models.ScheduleRun"]: + def list_runs(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.ScheduleRun"]: """List all schedule runs. :param id: Identifier of the schedule. Required. diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_connections_async.py b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_connections_async.py index ecc9981f4b48..edc2cb291828 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_connections_async.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_connections_async.py @@ -13,6 +13,7 @@ from ._operations import ConnectionsOperations as ConnectionsOperationsGenerated from ...models._models import Connection from ...models._enums import ConnectionType +from ...types import RequestParameters class ConnectionsOperations(ConnectionsOperationsGenerated): @@ -26,7 +27,7 @@ class ConnectionsOperations(ConnectionsOperationsGenerated): """ @distributed_trace_async - async def get(self, name: str, *, include_credentials: Optional[bool] = False, **kwargs: Any) -> Connection: + async def get(self, name: str, *, include_credentials: Optional[bool] = False, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> Connection: """Get a connection by name. :param name: The name of the resource. Required. @@ -37,6 +38,9 @@ async def get(self, name: str, *, include_credentials: Optional[bool] = False, * :rtype: ~azure.ai.projects.models.Connection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) if include_credentials: connection = await super()._get_with_credentials(name, **kwargs) @@ -54,7 +58,7 @@ async def get(self, name: str, *, include_credentials: Optional[bool] = False, * @distributed_trace_async async def get_default( - self, connection_type: Union[str, ConnectionType], *, include_credentials: Optional[bool] = False, **kwargs: Any + self, connection_type: Union[str, ConnectionType], *, include_credentials: Optional[bool] = False, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> Connection: """Get the default connection for a given connection type. @@ -67,6 +71,10 @@ async def get_default( :raises ValueError: If no default connection is found for the given type. :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + connections = super().list(connection_type=connection_type, default_connection=True, **kwargs) async for connection in connections: return await self.get(connection.name, include_credentials=include_credentials, **kwargs) diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_datasets_async.py b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_datasets_async.py index da947cb5cf10..811e1147f29b 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_datasets_async.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_datasets_async.py @@ -17,6 +17,7 @@ from azure.core.tracing.decorator_async import distributed_trace_async from ._operations import DatasetsOperations as DatasetsOperationsGenerated +from ...types import RequestParameters from ...models._models import ( FileDatasetVersion, FolderDatasetVersion, @@ -78,7 +79,7 @@ async def _create_dataset_and_get_its_container_client( @distributed_trace_async async def upload_file( - self, *, name: str, version: str, file_path: str, connection_name: Optional[str] = None, **kwargs: Any + self, *, name: str, version: str, file_path: str, connection_name: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> FileDatasetVersion: """Upload file to a blob storage, and create a dataset that references this file. This method uses the `ContainerClient.upload_blob` method from the azure-storage-blob package @@ -97,6 +98,10 @@ async def upload_file( :rtype: ~azure.ai.projects.models.FileDatasetVersion :raises ~azure.core.exceptions.HttpResponseError: If an error occurs during the HTTP request. """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + pathlib_file_path = Path(file_path) if not pathlib_file_path.exists(): @@ -150,7 +155,7 @@ async def upload_folder( folder: str, connection_name: Optional[str] = None, file_pattern: Optional[re.Pattern] = None, - **kwargs: Any, + request_parameters: Optional[RequestParameters] = None, **kwargs: Any, ) -> FolderDatasetVersion: """Upload all files in a folder and its sub folders to a blob storage, while maintaining relative paths, and create a dataset that references this folder. @@ -173,6 +178,10 @@ async def upload_folder( :rtype: ~azure.ai.projects.models.FolderDatasetVersion :raises ~azure.core.exceptions.HttpResponseError: If an error occurs during the HTTP request. """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + path_folder = Path(folder) if not Path(path_folder).exists(): raise ValueError(f"The provided folder `{folder}` does not exist.") diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_memories_async.py b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_memories_async.py index 4f1343541de5..5171ff62a3e8 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_memories_async.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_memories_async.py @@ -12,6 +12,7 @@ from azure.core.polling import AsyncNoPolling from azure.core.utils import case_insensitive_dict from ... import models as _models +from ...types import LRORequestParameters from ...models import ( MemoryStoreOperationUsage, ResponseUsageInputTokensDetails, @@ -37,7 +38,7 @@ async def begin_update_memories( items: Optional[List[_models.InputItem]] = None, previous_update_id: Optional[str] = None, update_delay: Optional[int] = None, - **kwargs: Any, + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any, ) -> AsyncUpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -69,7 +70,7 @@ async def begin_update_memories( @overload async def begin_update_memories( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncUpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -89,7 +90,7 @@ async def begin_update_memories( @overload async def begin_update_memories( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncUpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -122,7 +123,7 @@ async def begin_update_memories( items: Optional[List[_models.InputItem]] = None, previous_update_id: Optional[str] = None, update_delay: Optional[int] = None, - **kwargs: Any, + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any, ) -> AsyncUpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -150,6 +151,10 @@ async def begin_update_memories( ~azure.ai.projects.models.AsyncUpdateMemoriesLROPoller :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_operations.py b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_operations.py index 3def1c5f7d71..a40d37418e3a 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_operations.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_operations.py @@ -37,6 +37,7 @@ from .._configuration import AIProjectClientConfiguration from .._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters JSON = MutableMapping[str, Any] _Unset: Any = object() @@ -1754,7 +1755,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def get(self, agent_name: str, **kwargs: Any) -> _models.AgentDetails: + def get(self, agent_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AgentDetails: """Retrieves the agent. :param agent_name: The name of the agent to retrieve. Required. @@ -1763,6 +1764,10 @@ def get(self, agent_name: str, **kwargs: Any) -> _models.AgentDetails: :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1826,7 +1831,7 @@ def create( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates the agent. @@ -1858,7 +1863,7 @@ def create( """ @overload - def create(self, body: JSON, *, content_type: str = "application/json", **kwargs: Any) -> _models.AgentDetails: + def create(self, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AgentDetails: """Creates the agent. :param body: Required. @@ -1872,7 +1877,7 @@ def create(self, body: JSON, *, content_type: str = "application/json", **kwargs """ @overload - def create(self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any) -> _models.AgentDetails: + def create(self, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AgentDetails: """Creates the agent. :param body: Required. @@ -1894,7 +1899,7 @@ def create( definition: _models.AgentDefinition = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates the agent. @@ -1923,6 +1928,10 @@ def create( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2002,7 +2011,7 @@ def update( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -2031,7 +2040,7 @@ def update( @overload def update( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -2050,7 +2059,7 @@ def update( @overload def update( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -2076,7 +2085,7 @@ def update( definition: _models.AgentDefinition = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent by adding a new version if there are any changes to the agent definition. If no changes, returns the existing agent version. @@ -2101,6 +2110,10 @@ def update( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2180,7 +2193,7 @@ def create_from_manifest( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -2215,7 +2228,7 @@ def create_from_manifest( @overload def create_from_manifest( - self, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -2231,7 +2244,7 @@ def create_from_manifest( @overload def create_from_manifest( - self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -2255,7 +2268,7 @@ def create_from_manifest( parameter_values: dict[str, Any] = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Creates an agent from a manifest. @@ -2286,6 +2299,10 @@ def create_from_manifest( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2374,7 +2391,7 @@ def update_from_manifest( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -2406,7 +2423,7 @@ def update_from_manifest( @overload def update_from_manifest( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -2426,7 +2443,7 @@ def update_from_manifest( @overload def update_from_manifest( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -2454,7 +2471,7 @@ def update_from_manifest( parameter_values: dict[str, Any] = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentDetails: """Updates the agent from a manifest by adding a new version if there are any changes to the agent definition. @@ -2482,6 +2499,10 @@ def update_from_manifest( :rtype: ~azure.ai.projects.models.AgentDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2559,7 +2580,7 @@ def update_from_manifest( return deserialized # type: ignore @distributed_trace - def delete(self, agent_name: str, **kwargs: Any) -> _models.DeleteAgentResponse: + def delete(self, agent_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DeleteAgentResponse: """Deletes an agent. :param agent_name: The name of the agent to delete. Required. @@ -2568,6 +2589,10 @@ def delete(self, agent_name: str, **kwargs: Any) -> _models.DeleteAgentResponse: :rtype: ~azure.ai.projects.models.DeleteAgentResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2630,7 +2655,7 @@ def list( limit: Optional[int] = None, order: Optional[Union[str, _models.PageOrder]] = None, before: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.AgentDetails"]: """Returns the list of all agents. @@ -2655,6 +2680,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.AgentDetails] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -2723,7 +2752,7 @@ def create_version( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -2756,7 +2785,7 @@ def create_version( @overload def create_version( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -2779,7 +2808,7 @@ def create_version( @overload def create_version( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -2809,7 +2838,7 @@ def create_version( definition: _models.AgentDefinition = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version. @@ -2838,6 +2867,10 @@ def create_version( :rtype: ~azure.ai.projects.models.AgentVersionDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2917,7 +2950,7 @@ def create_version_from_manifest( content_type: str = "application/json", metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -2952,7 +2985,7 @@ def create_version_from_manifest( @overload def create_version_from_manifest( - self, agent_name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -2975,7 +3008,7 @@ def create_version_from_manifest( @overload def create_version_from_manifest( - self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, agent_name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -3006,7 +3039,7 @@ def create_version_from_manifest( parameter_values: dict[str, Any] = _Unset, metadata: Optional[dict[str, str]] = None, description: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AgentVersionDetails: """Create a new agent version from a manifest. @@ -3037,6 +3070,10 @@ def create_version_from_manifest( :rtype: ~azure.ai.projects.models.AgentVersionDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3114,7 +3151,7 @@ def create_version_from_manifest( return deserialized # type: ignore @distributed_trace - def get_version(self, agent_name: str, agent_version: str, **kwargs: Any) -> _models.AgentVersionDetails: + def get_version(self, agent_name: str, agent_version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AgentVersionDetails: """Retrieves a specific version of an agent. :param agent_name: The name of the agent to retrieve. Required. @@ -3125,6 +3162,10 @@ def get_version(self, agent_name: str, agent_version: str, **kwargs: Any) -> _mo :rtype: ~azure.ai.projects.models.AgentVersionDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3181,7 +3222,7 @@ def get_version(self, agent_name: str, agent_version: str, **kwargs: Any) -> _mo return deserialized # type: ignore @distributed_trace - def delete_version(self, agent_name: str, agent_version: str, **kwargs: Any) -> _models.DeleteAgentVersionResponse: + def delete_version(self, agent_name: str, agent_version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DeleteAgentVersionResponse: """Deletes a specific version of an agent. :param agent_name: The name of the agent to delete. Required. @@ -3193,6 +3234,10 @@ def delete_version(self, agent_name: str, agent_version: str, **kwargs: Any) -> :rtype: ~azure.ai.projects.models.DeleteAgentVersionResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3256,7 +3301,7 @@ def list_versions( limit: Optional[int] = None, order: Optional[Union[str, _models.PageOrder]] = None, before: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.AgentVersionDetails"]: """Returns the list of versions of an agent. @@ -3280,6 +3325,10 @@ def list_versions( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.AgentVersionDetails] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -3348,7 +3397,7 @@ def stream_agent_container_logs( # pylint: disable=inconsistent-return-statemen kind: Optional[Union[str, _models.ContainerLogKind]] = None, replica_name: Optional[str] = None, tail: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Container log entry streamed from the container as text chunks. Each chunk is a UTF-8 string that may be either a plain text log line @@ -3393,6 +3442,10 @@ def stream_agent_container_logs( # pylint: disable=inconsistent-return-statemen :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3466,7 +3519,7 @@ def create( content_type: str = "application/json", description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -3489,7 +3542,7 @@ def create( @overload def create( - self, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -3505,7 +3558,7 @@ def create( @overload def create( - self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -3528,7 +3581,7 @@ def create( definition: _models.MemoryStoreDefinition = _Unset, description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Create a memory store. @@ -3547,6 +3600,10 @@ def create( :rtype: ~azure.ai.projects.models.MemoryStoreDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3625,7 +3682,7 @@ def update( content_type: str = "application/json", description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -3646,7 +3703,7 @@ def update( @overload def update( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -3664,7 +3721,7 @@ def update( @overload def update( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -3688,7 +3745,7 @@ def update( *, description: Optional[str] = None, metadata: Optional[dict[str, str]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDetails: """Update a memory store. @@ -3705,6 +3762,10 @@ def update( :rtype: ~azure.ai.projects.models.MemoryStoreDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3773,7 +3834,7 @@ def update( return deserialized # type: ignore @distributed_trace - def get(self, name: str, **kwargs: Any) -> _models.MemoryStoreDetails: + def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.MemoryStoreDetails: """Retrieve a memory store. :param name: The name of the memory store to retrieve. Required. @@ -3782,6 +3843,10 @@ def get(self, name: str, **kwargs: Any) -> _models.MemoryStoreDetails: :rtype: ~azure.ai.projects.models.MemoryStoreDetails :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3843,7 +3908,7 @@ def list( limit: Optional[int] = None, order: Optional[Union[str, _models.PageOrder]] = None, before: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.MemoryStoreDetails"]: """List all memory stores. @@ -3865,6 +3930,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.MemoryStoreDetails] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -3924,7 +3993,7 @@ def get_next(_continuation_token=None): return ItemPaged(get_next, extract_data) @distributed_trace - def delete(self, name: str, **kwargs: Any) -> _models.DeleteMemoryStoreResult: + def delete(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DeleteMemoryStoreResult: """Delete a memory store. :param name: The name of the memory store to delete. Required. @@ -3933,6 +4002,10 @@ def delete(self, name: str, **kwargs: Any) -> _models.DeleteMemoryStoreResult: :rtype: ~azure.ai.projects.models.DeleteMemoryStoreResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3997,7 +4070,7 @@ def search_memories( items: Optional[List[_models.InputItem]] = None, previous_search_id: Optional[str] = None, options: Optional[_models.MemorySearchOptions] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -4023,7 +4096,7 @@ def search_memories( @overload def search_memories( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -4041,7 +4114,7 @@ def search_memories( @overload def search_memories( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -4067,7 +4140,7 @@ def search_memories( items: Optional[List[_models.InputItem]] = None, previous_search_id: Optional[str] = None, options: Optional[_models.MemorySearchOptions] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreSearchResult: """Search for relevant memories from a memory store based on conversation context. @@ -4089,6 +4162,10 @@ def search_memories( :rtype: ~azure.ai.projects.models.MemoryStoreSearchResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4368,7 +4445,7 @@ def get_long_running_output(pipeline_response): @overload def delete_scope( - self, name: str, *, scope: str, content_type: str = "application/json", **kwargs: Any + self, name: str, *, scope: str, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -4388,7 +4465,7 @@ def delete_scope( @overload def delete_scope( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -4407,7 +4484,7 @@ def delete_scope( @overload def delete_scope( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -4426,7 +4503,7 @@ def delete_scope( @distributed_trace def delete_scope( - self, name: str, body: Union[JSON, IO[bytes]] = _Unset, *, scope: str = _Unset, **kwargs: Any + self, name: str, body: Union[JSON, IO[bytes]] = _Unset, *, scope: str = _Unset, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.MemoryStoreDeleteScopeResult: """Delete all memories associated with a specific scope from a memory store. @@ -4442,6 +4519,10 @@ def delete_scope( :rtype: ~azure.ai.projects.models.MemoryStoreDeleteScopeResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4665,7 +4746,7 @@ def list( *, connection_type: Optional[Union[str, _models.ConnectionType]] = None, default_connection: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.Connection"]: """List all connections in the project, without populating connection credentials. @@ -4680,6 +4761,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.Connection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -4775,7 +4860,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_versions(self, name: str, **kwargs: Any) -> ItemPaged["_models.DatasetVersion"]: + def list_versions(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.DatasetVersion"]: """List all versions of the given DatasetVersion. :param name: The name of the resource. Required. @@ -4784,6 +4869,10 @@ def list_versions(self, name: str, **kwargs: Any) -> ItemPaged["_models.DatasetV :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.DatasetVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -4860,13 +4949,17 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.DatasetVersion"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.DatasetVersion"]: """List the latest version of each DatasetVersion. :return: An iterator like instance of DatasetVersion :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.DatasetVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -4942,7 +5035,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, name: str, version: str, **kwargs: Any) -> _models.DatasetVersion: + def get(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DatasetVersion: """Get the specific version of the DatasetVersion. The service returns 404 Not Found error if the DatasetVersion does not exist. @@ -4954,6 +5047,10 @@ def get(self, name: str, version: str, **kwargs: Any) -> _models.DatasetVersion: :rtype: ~azure.ai.projects.models.DatasetVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5006,7 +5103,7 @@ def get(self, name: str, version: str, **kwargs: Any) -> _models.DatasetVersion: return deserialized # type: ignore @distributed_trace - def delete(self, name: str, version: str, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def delete(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """Delete the specific version of the DatasetVersion. The service returns 204 No Content if the DatasetVersion was deleted successfully or if the DatasetVersion does not exist. @@ -5018,6 +5115,10 @@ def delete(self, name: str, version: str, **kwargs: Any) -> None: # pylint: dis :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5065,7 +5166,7 @@ def create_or_update( dataset_version: _models.DatasetVersion, *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -5091,7 +5192,7 @@ def create_or_update( dataset_version: JSON, *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -5117,7 +5218,7 @@ def create_or_update( dataset_version: IO[bytes], *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -5137,7 +5238,7 @@ def create_or_update( @distributed_trace def create_or_update( - self, name: str, version: str, dataset_version: Union[_models.DatasetVersion, JSON, IO[bytes]], **kwargs: Any + self, name: str, version: str, dataset_version: Union[_models.DatasetVersion, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DatasetVersion: """Create a new or update an existing DatasetVersion with the given version id. @@ -5152,6 +5253,10 @@ def create_or_update( :rtype: ~azure.ai.projects.models.DatasetVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5221,7 +5326,7 @@ def pending_upload( pending_upload_request: _models.PendingUploadRequest, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -5247,7 +5352,7 @@ def pending_upload( pending_upload_request: JSON, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -5273,7 +5378,7 @@ def pending_upload( pending_upload_request: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -5297,7 +5402,7 @@ def pending_upload( name: str, version: str, pending_upload_request: Union[_models.PendingUploadRequest, JSON, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PendingUploadResponse: """Start a new or get an existing pending upload of a dataset for a specific version. @@ -5313,6 +5418,10 @@ def pending_upload( :rtype: ~azure.ai.projects.models.PendingUploadResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5375,7 +5484,7 @@ def pending_upload( return deserialized # type: ignore @distributed_trace - def get_credentials(self, name: str, version: str, **kwargs: Any) -> _models.DatasetCredential: + def get_credentials(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DatasetCredential: """Get the SAS credential to access the storage account associated with a Dataset version. :param name: The name of the resource. Required. @@ -5386,6 +5495,10 @@ def get_credentials(self, name: str, version: str, **kwargs: Any) -> _models.Dat :rtype: ~azure.ai.projects.models.DatasetCredential :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5456,7 +5569,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_versions(self, name: str, **kwargs: Any) -> ItemPaged["_models.Index"]: + def list_versions(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Index"]: """List all versions of the given Index. :param name: The name of the resource. Required. @@ -5465,6 +5578,10 @@ def list_versions(self, name: str, **kwargs: Any) -> ItemPaged["_models.Index"]: :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.Index] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -5541,13 +5658,17 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Index"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Index"]: """List the latest version of each Index. :return: An iterator like instance of Index :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.Index] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -5623,7 +5744,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, name: str, version: str, **kwargs: Any) -> _models.Index: + def get(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Index: """Get the specific version of the Index. The service returns 404 Not Found error if the Index does not exist. @@ -5635,6 +5756,10 @@ def get(self, name: str, version: str, **kwargs: Any) -> _models.Index: :rtype: ~azure.ai.projects.models.Index :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5687,7 +5812,7 @@ def get(self, name: str, version: str, **kwargs: Any) -> _models.Index: return deserialized # type: ignore @distributed_trace - def delete(self, name: str, version: str, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def delete(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """Delete the specific version of the Index. The service returns 204 No Content if the Index was deleted successfully or if the Index does not exist. @@ -5699,6 +5824,10 @@ def delete(self, name: str, version: str, **kwargs: Any) -> None: # pylint: dis :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5746,7 +5875,7 @@ def create_or_update( index: _models.Index, *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -5766,7 +5895,7 @@ def create_or_update( @overload def create_or_update( - self, name: str, version: str, index: JSON, *, content_type: str = "application/merge-patch+json", **kwargs: Any + self, name: str, version: str, index: JSON, *, content_type: str = "application/merge-patch+json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -5792,7 +5921,7 @@ def create_or_update( index: IO[bytes], *, content_type: str = "application/merge-patch+json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -5812,7 +5941,7 @@ def create_or_update( @distributed_trace def create_or_update( - self, name: str, version: str, index: Union[_models.Index, JSON, IO[bytes]], **kwargs: Any + self, name: str, version: str, index: Union[_models.Index, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Index: """Create a new or update an existing Index with the given version id. @@ -5827,6 +5956,10 @@ def create_or_update( :rtype: ~azure.ai.projects.models.Index :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5907,7 +6040,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def get(self, name: str, **kwargs: Any) -> _models.Deployment: + def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Deployment: """Get a deployed model. :param name: Name of the deployment. Required. @@ -5916,6 +6049,10 @@ def get(self, name: str, **kwargs: Any) -> _models.Deployment: :rtype: ~azure.ai.projects.models.Deployment :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -5978,7 +6115,7 @@ def list( model_publisher: Optional[str] = None, model_name: Optional[str] = None, deployment_type: Optional[Union[str, _models.DeploymentType]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.Deployment"]: """List all deployed models in the project. @@ -5994,6 +6131,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.Deployment] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -6090,7 +6231,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def get(self, name: str, **kwargs: Any) -> _models.RedTeam: + def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RedTeam: """Get a redteam by name. :param name: Identifier of the red team run. Required. @@ -6099,6 +6240,10 @@ def get(self, name: str, **kwargs: Any) -> _models.RedTeam: :rtype: ~azure.ai.projects.models.RedTeam :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6155,13 +6300,17 @@ def get(self, name: str, **kwargs: Any) -> _models.RedTeam: return deserialized # type: ignore @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.RedTeam"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.RedTeam"]: """List a redteam by name. :return: An iterator like instance of RedTeam :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.RedTeam] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -6238,7 +6387,7 @@ def get_next(next_link=None): @overload def create( - self, red_team: _models.RedTeam, *, content_type: str = "application/json", **kwargs: Any + self, red_team: _models.RedTeam, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RedTeam: """Creates a redteam run. @@ -6253,7 +6402,7 @@ def create( """ @overload - def create(self, red_team: JSON, *, content_type: str = "application/json", **kwargs: Any) -> _models.RedTeam: + def create(self, red_team: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RedTeam: """Creates a redteam run. :param red_team: Redteam to be run. Required. @@ -6267,7 +6416,7 @@ def create(self, red_team: JSON, *, content_type: str = "application/json", **kw """ @overload - def create(self, red_team: IO[bytes], *, content_type: str = "application/json", **kwargs: Any) -> _models.RedTeam: + def create(self, red_team: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RedTeam: """Creates a redteam run. :param red_team: Redteam to be run. Required. @@ -6281,7 +6430,7 @@ def create(self, red_team: IO[bytes], *, content_type: str = "application/json", """ @distributed_trace - def create(self, red_team: Union[_models.RedTeam, JSON, IO[bytes]], **kwargs: Any) -> _models.RedTeam: + def create(self, red_team: Union[_models.RedTeam, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RedTeam: """Creates a redteam run. :param red_team: Redteam to be run. Is one of the following types: RedTeam, JSON, IO[bytes] @@ -6291,6 +6440,10 @@ def create(self, red_team: Union[_models.RedTeam, JSON, IO[bytes]], **kwargs: An :rtype: ~azure.ai.projects.models.RedTeam :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6369,7 +6522,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def get(self, id: str, **kwargs: Any) -> _models.EvaluationRule: + def get(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.EvaluationRule: """Get an evaluation rule. :param id: Unique identifier for the evaluation rule. Required. @@ -6378,6 +6531,10 @@ def get(self, id: str, **kwargs: Any) -> _models.EvaluationRule: :rtype: ~azure.ai.projects.models.EvaluationRule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6434,7 +6591,7 @@ def get(self, id: str, **kwargs: Any) -> _models.EvaluationRule: return deserialized # type: ignore @distributed_trace - def delete(self, id: str, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def delete(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """Delete an evaluation rule. :param id: Unique identifier for the evaluation rule. Required. @@ -6443,6 +6600,10 @@ def delete(self, id: str, **kwargs: Any) -> None: # pylint: disable=inconsisten :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6488,7 +6649,7 @@ def delete(self, id: str, **kwargs: Any) -> None: # pylint: disable=inconsisten @overload def create_or_update( - self, id: str, evaluation_rule: _models.EvaluationRule, *, content_type: str = "application/json", **kwargs: Any + self, id: str, evaluation_rule: _models.EvaluationRule, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -6506,7 +6667,7 @@ def create_or_update( @overload def create_or_update( - self, id: str, evaluation_rule: JSON, *, content_type: str = "application/json", **kwargs: Any + self, id: str, evaluation_rule: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -6524,7 +6685,7 @@ def create_or_update( @overload def create_or_update( - self, id: str, evaluation_rule: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, id: str, evaluation_rule: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -6542,7 +6703,7 @@ def create_or_update( @distributed_trace def create_or_update( - self, id: str, evaluation_rule: Union[_models.EvaluationRule, JSON, IO[bytes]], **kwargs: Any + self, id: str, evaluation_rule: Union[_models.EvaluationRule, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationRule: """Create or update an evaluation rule. @@ -6555,6 +6716,10 @@ def create_or_update( :rtype: ~azure.ai.projects.models.EvaluationRule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6622,7 +6787,7 @@ def list( action_type: Optional[Union[str, _models.EvaluationRuleActionType]] = None, agent_name: Optional[str] = None, enabled: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.EvaluationRule"]: """List all evaluation rules. @@ -6637,6 +6802,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.EvaluationRule] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -6733,7 +6902,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def get(self, name: str, **kwargs: Any) -> _models.EvaluationTaxonomy: + def get(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.EvaluationTaxonomy: """Get an evaluation run by name. :param name: The name of the resource. Required. @@ -6742,6 +6911,10 @@ def get(self, name: str, **kwargs: Any) -> _models.EvaluationTaxonomy: :rtype: ~azure.ai.projects.models.EvaluationTaxonomy :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6799,7 +6972,7 @@ def get(self, name: str, **kwargs: Any) -> _models.EvaluationTaxonomy: @distributed_trace def list( - self, *, input_name: Optional[str] = None, input_type: Optional[str] = None, **kwargs: Any + self, *, input_name: Optional[str] = None, input_type: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.EvaluationTaxonomy"]: """List evaluation taxonomies. @@ -6811,6 +6984,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.EvaluationTaxonomy] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -6888,7 +7065,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def delete(self, name: str, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def delete(self, name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """Delete an evaluation taxonomy by name. :param name: The name of the resource. Required. @@ -6897,6 +7074,10 @@ def delete(self, name: str, **kwargs: Any) -> None: # pylint: disable=inconsist :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -6942,7 +7123,7 @@ def delete(self, name: str, **kwargs: Any) -> None: # pylint: disable=inconsist @overload def create( - self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -6960,7 +7141,7 @@ def create( @overload def create( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -6978,7 +7159,7 @@ def create( @overload def create( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -6996,7 +7177,7 @@ def create( @distributed_trace def create( - self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], **kwargs: Any + self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Create an evaluation taxonomy. @@ -7009,6 +7190,10 @@ def create( :rtype: ~azure.ai.projects.models.EvaluationTaxonomy :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -7071,7 +7256,7 @@ def create( @overload def update( - self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: _models.EvaluationTaxonomy, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -7089,7 +7274,7 @@ def update( @overload def update( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -7107,7 +7292,7 @@ def update( @overload def update( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -7125,7 +7310,7 @@ def update( @distributed_trace def update( - self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], **kwargs: Any + self, name: str, body: Union[_models.EvaluationTaxonomy, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluationTaxonomy: """Update an evaluation taxonomy. @@ -7138,6 +7323,10 @@ def update( :rtype: ~azure.ai.projects.models.EvaluationTaxonomy :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -7223,7 +7412,7 @@ def list_versions( *, type: Optional[Union[Literal["builtin"], Literal["custom"], Literal["all"], str]] = None, limit: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.EvaluatorVersion"]: """List all versions of the given evaluator. @@ -7240,6 +7429,10 @@ def list_versions( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.EvaluatorVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -7323,7 +7516,7 @@ def list_latest_versions( *, type: Optional[Union[Literal["builtin"], Literal["custom"], Literal["all"], str]] = None, limit: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.EvaluatorVersion"]: """List the latest version of each evaluator. @@ -7338,6 +7531,10 @@ def list_latest_versions( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.EvaluatorVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -7415,7 +7612,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get_version(self, name: str, version: str, **kwargs: Any) -> _models.EvaluatorVersion: + def get_version(self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.EvaluatorVersion: """Get the specific version of the EvaluatorVersion. The service returns 404 Not Found error if the EvaluatorVersion does not exist. @@ -7427,6 +7624,10 @@ def get_version(self, name: str, version: str, **kwargs: Any) -> _models.Evaluat :rtype: ~azure.ai.projects.models.EvaluatorVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -7480,7 +7681,7 @@ def get_version(self, name: str, version: str, **kwargs: Any) -> _models.Evaluat @distributed_trace def delete_version( # pylint: disable=inconsistent-return-statements - self, name: str, version: str, **kwargs: Any + self, name: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Delete the specific version of the EvaluatorVersion. The service returns 204 No Content if the EvaluatorVersion was deleted successfully or if the EvaluatorVersion does not exist. @@ -7493,6 +7694,10 @@ def delete_version( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -7539,7 +7744,7 @@ def create_version( evaluator_version: _models.EvaluatorVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -7557,7 +7762,7 @@ def create_version( @overload def create_version( - self, name: str, evaluator_version: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, evaluator_version: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -7575,7 +7780,7 @@ def create_version( @overload def create_version( - self, name: str, evaluator_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, evaluator_version: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -7593,7 +7798,7 @@ def create_version( @distributed_trace def create_version( - self, name: str, evaluator_version: Union[_models.EvaluatorVersion, JSON, IO[bytes]], **kwargs: Any + self, name: str, evaluator_version: Union[_models.EvaluatorVersion, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Create a new EvaluatorVersion with auto incremented version id. @@ -7606,6 +7811,10 @@ def create_version( :rtype: ~azure.ai.projects.models.EvaluatorVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -7674,7 +7883,7 @@ def update_version( evaluator_version: _models.EvaluatorVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -7694,7 +7903,7 @@ def update_version( @overload def update_version( - self, name: str, version: str, evaluator_version: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, version: str, evaluator_version: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -7720,7 +7929,7 @@ def update_version( evaluator_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -7744,7 +7953,7 @@ def update_version( name: str, version: str, evaluator_version: Union[_models.EvaluatorVersion, JSON, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.EvaluatorVersion: """Update an existing EvaluatorVersion with the given version id. @@ -7759,6 +7968,10 @@ def update_version( :rtype: ~azure.ai.projects.models.EvaluatorVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -7840,7 +8053,7 @@ def __init__(self, *args, **kwargs) -> None: @overload def generate( - self, insight: _models.Insight, *, content_type: str = "application/json", **kwargs: Any + self, insight: _models.Insight, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Insight: """Generate Insights. @@ -7856,7 +8069,7 @@ def generate( """ @overload - def generate(self, insight: JSON, *, content_type: str = "application/json", **kwargs: Any) -> _models.Insight: + def generate(self, insight: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Insight: """Generate Insights. :param insight: Complete evaluation configuration including data source, evaluators, and result @@ -7871,7 +8084,7 @@ def generate(self, insight: JSON, *, content_type: str = "application/json", **k """ @overload - def generate(self, insight: IO[bytes], *, content_type: str = "application/json", **kwargs: Any) -> _models.Insight: + def generate(self, insight: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Insight: """Generate Insights. :param insight: Complete evaluation configuration including data source, evaluators, and result @@ -7886,7 +8099,7 @@ def generate(self, insight: IO[bytes], *, content_type: str = "application/json" """ @distributed_trace - def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], **kwargs: Any) -> _models.Insight: + def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Insight: """Generate Insights. :param insight: Complete evaluation configuration including data source, evaluators, and result @@ -7896,6 +8109,10 @@ def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], **kwargs: A :rtype: ~azure.ai.projects.models.Insight :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -7956,7 +8173,7 @@ def generate(self, insight: Union[_models.Insight, JSON, IO[bytes]], **kwargs: A return deserialized # type: ignore @distributed_trace - def get(self, id: str, *, include_coordinates: Optional[bool] = None, **kwargs: Any) -> _models.Insight: + def get(self, id: str, *, include_coordinates: Optional[bool] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Insight: """Get a specific insight by Id. :param id: The unique identifier for the insights report. Required. @@ -7968,6 +8185,10 @@ def get(self, id: str, *, include_coordinates: Optional[bool] = None, **kwargs: :rtype: ~azure.ai.projects.models.Insight :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -8033,7 +8254,7 @@ def list( run_id: Optional[str] = None, agent_name: Optional[str] = None, include_coordinates: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.Insight"]: """List all insights in reverse chronological order (newest first). @@ -8053,6 +8274,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.Insight] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -8151,7 +8376,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def delete(self, id: str, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def delete(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """Delete a schedule. :param id: Identifier of the schedule. Required. @@ -8160,6 +8385,10 @@ def delete(self, id: str, **kwargs: Any) -> None: # pylint: disable=inconsisten :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -8204,7 +8433,7 @@ def delete(self, id: str, **kwargs: Any) -> None: # pylint: disable=inconsisten return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace - def get(self, id: str, **kwargs: Any) -> _models.Schedule: + def get(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Schedule: """Get a schedule by id. :param id: Identifier of the schedule. Required. @@ -8213,6 +8442,10 @@ def get(self, id: str, **kwargs: Any) -> _models.Schedule: :rtype: ~azure.ai.projects.models.Schedule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -8269,13 +8502,17 @@ def get(self, id: str, **kwargs: Any) -> _models.Schedule: return deserialized # type: ignore @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Schedule"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Schedule"]: """List all schedules. :return: An iterator like instance of Schedule :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.Schedule] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} @@ -8352,7 +8589,7 @@ def get_next(next_link=None): @overload def create_or_update( - self, id: str, schedule: _models.Schedule, *, content_type: str = "application/json", **kwargs: Any + self, id: str, schedule: _models.Schedule, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -8370,7 +8607,7 @@ def create_or_update( @overload def create_or_update( - self, id: str, schedule: JSON, *, content_type: str = "application/json", **kwargs: Any + self, id: str, schedule: JSON, *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -8388,7 +8625,7 @@ def create_or_update( @overload def create_or_update( - self, id: str, schedule: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, id: str, schedule: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -8406,7 +8643,7 @@ def create_or_update( @distributed_trace def create_or_update( - self, id: str, schedule: Union[_models.Schedule, JSON, IO[bytes]], **kwargs: Any + self, id: str, schedule: Union[_models.Schedule, JSON, IO[bytes]], request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Schedule: """Create or update a schedule by id. @@ -8419,6 +8656,10 @@ def create_or_update( :rtype: ~azure.ai.projects.models.Schedule :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -8480,7 +8721,7 @@ def create_or_update( return deserialized # type: ignore @distributed_trace - def get_run(self, schedule_id: str, run_id: str, **kwargs: Any) -> _models.ScheduleRun: + def get_run(self, schedule_id: str, run_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.ScheduleRun: """Get a schedule run by id. :param schedule_id: Identifier of the schedule. Required. @@ -8491,6 +8732,10 @@ def get_run(self, schedule_id: str, run_id: str, **kwargs: Any) -> _models.Sched :rtype: ~azure.ai.projects.models.ScheduleRun :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -8543,7 +8788,7 @@ def get_run(self, schedule_id: str, run_id: str, **kwargs: Any) -> _models.Sched return deserialized # type: ignore @distributed_trace - def list_runs(self, id: str, **kwargs: Any) -> ItemPaged["_models.ScheduleRun"]: + def list_runs(self, id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.ScheduleRun"]: """List all schedule runs. :param id: Identifier of the schedule. Required. @@ -8552,6 +8797,10 @@ def list_runs(self, id: str, **kwargs: Any) -> ItemPaged["_models.ScheduleRun"]: :rtype: ~azure.core.paging.ItemPaged[~azure.ai.projects.models.ScheduleRun] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_connections.py b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_connections.py index 44172818b0c1..a08709c4c34d 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_connections.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_connections.py @@ -12,6 +12,7 @@ from ._operations import ConnectionsOperations as ConnectionsOperationsGenerated from ..models._models import Connection from ..models._enums import ConnectionType +from ..types import RequestParameters class ConnectionsOperations(ConnectionsOperationsGenerated): @@ -25,7 +26,7 @@ class ConnectionsOperations(ConnectionsOperationsGenerated): """ @distributed_trace - def get(self, name: str, *, include_credentials: Optional[bool] = False, **kwargs: Any) -> Connection: + def get(self, name: str, *, include_credentials: Optional[bool] = False, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> Connection: """Get a connection by name. :param name: The name of the connection. Required. @@ -36,6 +37,10 @@ def get(self, name: str, *, include_credentials: Optional[bool] = False, **kwarg :rtype: ~azure.ai.projects.models.Connection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + if include_credentials: connection = super()._get_with_credentials(name, **kwargs) if connection.type == ConnectionType.CUSTOM: @@ -74,7 +79,7 @@ def get(self, name: str, *, include_credentials: Optional[bool] = False, **kwarg @distributed_trace def get_default( - self, connection_type: Union[str, ConnectionType], *, include_credentials: Optional[bool] = False, **kwargs: Any + self, connection_type: Union[str, ConnectionType], *, include_credentials: Optional[bool] = False, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> Connection: """Get the default connection for a given connection type. @@ -87,6 +92,10 @@ def get_default( :raises ValueError: If no default connection is found for the given type. :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + connections = super().list(connection_type=connection_type, default_connection=True, **kwargs) for connection in connections: return self.get(connection.name, include_credentials=include_credentials, **kwargs) diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_datasets.py b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_datasets.py index 4f528e416526..a27b4e4fb761 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_datasets.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_datasets.py @@ -16,6 +16,7 @@ from azure.storage.blob import ContainerClient from azure.core.tracing.decorator import distributed_trace from ._operations import DatasetsOperations as DatasetsOperationsGenerated +from ..types import RequestParameters from ..models._models import ( FileDatasetVersion, FolderDatasetVersion, @@ -77,7 +78,7 @@ def _create_dataset_and_get_its_container_client( @distributed_trace def upload_file( - self, *, name: str, version: str, file_path: str, connection_name: Optional[str] = None, **kwargs: Any + self, *, name: str, version: str, file_path: str, connection_name: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> FileDatasetVersion: """Upload file to a blob storage, and create a dataset that references this file. This method uses the `ContainerClient.upload_blob` method from the azure-storage-blob package @@ -96,6 +97,9 @@ def upload_file( :rtype: ~azure.ai.projects.models.FileDatasetVersion :raises ~azure.core.exceptions.HttpResponseError: If an error occurs during the HTTP request. """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) pathlib_file_path = Path(file_path) if not pathlib_file_path.exists(): @@ -147,6 +151,7 @@ def upload_folder( folder: str, connection_name: Optional[str] = None, file_pattern: Optional[re.Pattern] = None, + request_parameters: Optional[RequestParameters] = None, **kwargs: Any, ) -> FolderDatasetVersion: """Upload all files in a folder and its sub folders to a blob storage, while maintaining @@ -170,6 +175,10 @@ def upload_folder( :rtype: ~azure.ai.projects.models.FolderDatasetVersion :raises ~azure.core.exceptions.HttpResponseError: If an error occurs during the HTTP request. """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + path_folder = Path(folder) if not Path(path_folder).exists(): raise ValueError(f"The provided folder `{folder}` does not exist.") diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_memories.py b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_memories.py index ca25e90aa008..0127dadbabcf 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_memories.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_memories.py @@ -12,6 +12,7 @@ from azure.core.polling import NoPolling from azure.core.utils import case_insensitive_dict from .. import models as _models +from ..types import LRORequestParameters from ..models import ( MemoryStoreOperationUsage, ResponseUsageInputTokensDetails, @@ -37,6 +38,7 @@ def begin_update_memories( items: Optional[List[_models.InputItem]] = None, previous_update_id: Optional[str] = None, update_delay: Optional[int] = None, + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any, ) -> UpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -69,7 +71,7 @@ def begin_update_memories( @overload def begin_update_memories( - self, name: str, body: JSON, *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: JSON, *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> UpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -89,7 +91,7 @@ def begin_update_memories( @overload def begin_update_memories( - self, name: str, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, name: str, body: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> UpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -122,6 +124,7 @@ def begin_update_memories( items: Optional[List[_models.InputItem]] = None, previous_update_id: Optional[str] = None, update_delay: Optional[int] = None, + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any, ) -> UpdateMemoriesLROPoller: """Update memory store with conversation memories. @@ -150,6 +153,10 @@ def begin_update_memories( ~azure.ai.projects.models.UpdateMemoriesLROPoller :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/types.py b/sdk/ai/azure-ai-projects/azure/ai/projects/types.py new file mode 100644 index 000000000000..3a6fd141285b --- /dev/null +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/types.py @@ -0,0 +1,176 @@ +# --------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# --------------------------------------------------------------------------- + +"""Typed dictionaries for service client request parameters. + +These TypedDict classes provide explicit typing for the infrastructure-level +parameters that control how HTTP requests are constructed and sent through +the Azure SDK pipeline. They replace the untyped ``**kwargs`` pattern for +better discoverability and type-checking support. +""" + +from typing import Callable, Dict, List, Union + +from typing_extensions import TypedDict + +from azure.core.pipeline import PipelineRequest, PipelineResponse +from azure.core.polling import PollingMethod + + +class RequestParameters(TypedDict, total=False): + """Common parameters for all service client operations. + + These parameters control how the HTTP request is constructed and sent + through the Azure SDK pipeline, including retry, redirect, tracing, + logging, and transport-level behavior. + """ + + # -- Request construction -- + + api_version: str + """Override the API version used for this request.""" + + headers: Dict[str, str] + """Custom HTTP headers to include in the request.""" + + params: Dict[str, str] + """Custom query parameters to include in the request.""" + + content_type: str + """Override the Content-Type header for the request.""" + + request_id: str + """Custom request ID to set in the request header + (consumed by :class:`~azure.core.pipeline.policies.RequestIdPolicy`).""" + + # -- Response handling -- + + stream: bool + """Whether to stream the response body instead of loading it into memory.""" + + decompress: bool + """Whether to decompress the response body. Defaults to ``True``.""" + + # -- Hooks -- + + raw_request_hook: Callable[[PipelineRequest], None] + """Callback invoked with the :class:`~azure.core.pipeline.PipelineRequest` + before the request is sent + (consumed by :class:`~azure.core.pipeline.policies.CustomHookPolicy`).""" + + raw_response_hook: Callable[[PipelineResponse], None] + """Callback invoked with the :class:`~azure.core.pipeline.PipelineResponse` + after the response is received + (consumed by :class:`~azure.core.pipeline.policies.CustomHookPolicy`).""" + + # -- Logging -- + + logging_enable: bool + """Enable HTTP logging for this request + (consumed by :class:`~azure.core.pipeline.policies.HttpLoggingPolicy`). + Defaults to ``False``.""" + + # -- Retry policy -- + + retry_total: int + """Total number of retries to allow + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``10``.""" + + retry_connect: int + """Number of connection-related retries + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``3``.""" + + retry_read: int + """Number of retries on read errors + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``3``.""" + + retry_status: int + """Number of retries on bad status codes + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``3``.""" + + retry_backoff_factor: float + """Backoff factor for retry delays + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``0.8``.""" + + retry_backoff_max: int + """Maximum backoff time in seconds + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``120``.""" + + retry_on_methods: List[str] + """HTTP methods that are eligible for retry + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`).""" + + timeout: int + """Overall timeout for the operation in seconds, including all retries + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``604800`` (7 days).""" + + # -- Redirect policy -- + + permit_redirects: bool + """Whether to follow redirects + (consumed by :class:`~azure.core.pipeline.policies.RedirectPolicy`). + Defaults to ``True``.""" + + redirect_max: int + """Maximum number of redirects to follow + (consumed by :class:`~azure.core.pipeline.policies.RedirectPolicy`). + Defaults to ``30``.""" + + # -- Distributed tracing -- + + network_span_namer: Callable[..., str] + """Callable that receives the HTTP request and returns a span name string + (consumed by :class:`~azure.core.pipeline.policies.DistributedTracingPolicy`).""" + + tracing_attributes: Dict[str, str] + """Extra attributes to set on the tracing span + (consumed by :class:`~azure.core.pipeline.policies.DistributedTracingPolicy`).""" + + # -- Transport / connection -- + + connection_timeout: float + """Connection timeout in seconds + (consumed by the HTTP transport).""" + + read_timeout: float + """Read timeout in seconds + (consumed by the HTTP transport).""" + + connection_verify: Union[bool, str] + """SSL certificate verification. Pass ``False`` to disable, + or a path to a CA bundle + (consumed by the HTTP transport).""" + + connection_cert: Union[str, tuple] + """Client-side certificate. A path to a file or a ``(cert, key)`` tuple + (consumed by the HTTP transport).""" + + proxy: str + """Proxy URL for this request + (consumed by the HTTP transport).""" + + +class LRORequestParameters(RequestParameters, total=False): + """Parameters for long-running operations (``begin_*`` methods). + + Extends :class:`RequestParameters` with polling and continuation support. + """ + + polling: Union[bool, PollingMethod] + """Polling strategy. ``True`` for default polling, ``False`` for no polling, + or a custom :class:`~azure.core.polling.PollingMethod` instance.""" + + polling_interval: int + """Time in seconds between polling requests. Overrides the client default.""" + + continuation_token: str + """Token to resume a previously started long-running operation.""" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/__init__.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/__init__.py index ab7c46e8991d..daf42c615573 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/__init__.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/__init__.py @@ -14,6 +14,7 @@ from ._compute_management_client import ComputeManagementClient # type: ignore from ._version import VERSION +from .types import LRORequestParameters, RequestParameters __version__ = VERSION @@ -26,6 +27,8 @@ __all__ = [ "ComputeManagementClient", + "RequestParameters", + "LRORequestParameters", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_availability_sets_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_availability_sets_operations.py index 14b59641da40..10403dd6c052 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_availability_sets_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_availability_sets_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._availability_sets_operations import ( build_cancel_migration_to_virtual_machine_scale_set_request, build_convert_to_virtual_machine_scale_set_request, @@ -74,7 +75,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_subscription( - self, expand: Optional[str] = None, **kwargs: Any + self, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.AvailabilitySet"]: """Lists all availability sets in a subscription. @@ -85,6 +86,10 @@ def list_by_subscription( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -142,7 +147,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.AvailabilitySet"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -152,6 +157,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_mode :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -209,7 +218,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: + async def get(self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AvailabilitySet: """Retrieves information about an availability set. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -221,6 +230,10 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :rtype: ~azure.mgmt.compute.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -271,7 +284,7 @@ async def create_or_update( parameters: _models.AvailabilitySet, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -298,7 +311,7 @@ async def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -323,7 +336,7 @@ async def create_or_update( resource_group_name: str, availability_set_name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -339,6 +352,10 @@ async def create_or_update( :rtype: ~azure.mgmt.compute.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -401,7 +418,7 @@ async def update( parameters: _models.AvailabilitySetUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -428,7 +445,7 @@ async def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -453,7 +470,7 @@ async def update( resource_group_name: str, availability_set_name: str, parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -469,6 +486,10 @@ async def update( :rtype: ~azure.mgmt.compute.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -524,7 +545,7 @@ async def update( return deserialized # type: ignore @distributed_trace_async - async def delete(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> None: + async def delete(self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete an availability set. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -536,6 +557,10 @@ async def delete(self, resource_group_name: str, availability_set_name: str, **k :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -576,7 +601,7 @@ async def delete(self, resource_group_name: str, availability_set_name: str, **k @distributed_trace_async async def cancel_migration_to_virtual_machine_scale_set( # pylint: disable=name-too-long - self, resource_group_name: str, availability_set_name: str, **kwargs: Any + self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Cancel the migration operation on an Availability Set. @@ -589,6 +614,10 @@ async def cancel_migration_to_virtual_machine_scale_set( # pylint: disable=name :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -709,7 +738,7 @@ async def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-to parameters: Optional[_models.ConvertToVirtualMachineScaleSetInput] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Create a new Flexible Virtual Machine Scale Set and migrate all the Virtual Machines in the Availability Set. This does not trigger a downtime on the Virtual Machines. @@ -738,7 +767,7 @@ async def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-to parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Create a new Flexible Virtual Machine Scale Set and migrate all the Virtual Machines in the Availability Set. This does not trigger a downtime on the Virtual Machines. @@ -765,7 +794,7 @@ async def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-to resource_group_name: str, availability_set_name: str, parameters: Optional[Union[_models.ConvertToVirtualMachineScaleSetInput, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Create a new Flexible Virtual Machine Scale Set and migrate all the Virtual Machines in the Availability Set. This does not trigger a downtime on the Virtual Machines. @@ -782,6 +811,10 @@ async def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-to :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -836,7 +869,7 @@ async def start_migration_to_virtual_machine_scale_set( # pylint: disable=name- parameters: _models.MigrateToVirtualMachineScaleSetInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Start migration operation on an Availability Set to move its Virtual Machines to a Virtual Machine Scale Set. This should be followed by a migrate operation on each Virtual Machine that @@ -866,7 +899,7 @@ async def start_migration_to_virtual_machine_scale_set( # pylint: disable=name- parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Start migration operation on an Availability Set to move its Virtual Machines to a Virtual Machine Scale Set. This should be followed by a migrate operation on each Virtual Machine that @@ -894,7 +927,7 @@ async def start_migration_to_virtual_machine_scale_set( # pylint: disable=name- resource_group_name: str, availability_set_name: str, parameters: Union[_models.MigrateToVirtualMachineScaleSetInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Start migration operation on an Availability Set to move its Virtual Machines to a Virtual Machine Scale Set. This should be followed by a migrate operation on each Virtual Machine that @@ -912,6 +945,10 @@ async def start_migration_to_virtual_machine_scale_set( # pylint: disable=name- :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -970,7 +1007,7 @@ async def validate_migration_to_virtual_machine_scale_set( # pylint: disable=na parameters: _models.MigrateToVirtualMachineScaleSetInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Validates that the Virtual Machines in the Availability Set can be migrated to the provided Virtual Machine Scale Set. @@ -999,7 +1036,7 @@ async def validate_migration_to_virtual_machine_scale_set( # pylint: disable=na parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Validates that the Virtual Machines in the Availability Set can be migrated to the provided Virtual Machine Scale Set. @@ -1026,7 +1063,7 @@ async def validate_migration_to_virtual_machine_scale_set( # pylint: disable=na resource_group_name: str, availability_set_name: str, parameters: Union[_models.MigrateToVirtualMachineScaleSetInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Validates that the Virtual Machines in the Availability Set can be migrated to the provided Virtual Machine Scale Set. @@ -1043,6 +1080,10 @@ async def validate_migration_to_virtual_machine_scale_set( # pylint: disable=na :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1095,7 +1136,7 @@ async def validate_migration_to_virtual_machine_scale_set( # pylint: disable=na @distributed_trace def list_available_sizes( - self, resource_group_name: str, availability_set_name: str, **kwargs: Any + self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachineSize"]: """Lists all available virtual machine sizes that can be used to create a new virtual machine in an existing availability set. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservation_groups_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservation_groups_operations.py index 116d17c7d3ec..711a22ff4a72 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservation_groups_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservation_groups_operations.py @@ -28,6 +28,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._capacity_reservation_groups_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def list_by_subscription( self, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, resource_ids_only: Optional[Union[str, _models.ResourceIdOptionsForGetCapacityReservationGroups]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.CapacityReservationGroup"]: """Lists all of the capacity reservation groups in the subscription. Use the nextLink property in the response to get the next page of capacity reservation groups. @@ -92,6 +93,10 @@ def list_by_subscription( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CapacityReservationGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -154,7 +159,7 @@ def list_by_resource_group( self, resource_group_name: str, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.CapacityReservationGroup"]: """Lists all of the capacity reservation groups in the specified resource group. Use the nextLink property in the response to get the next page of capacity reservation groups. @@ -173,6 +178,10 @@ def list_by_resource_group( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CapacityReservationGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -236,7 +245,7 @@ async def get( resource_group_name: str, capacity_reservation_group_name: str, expand: Optional[Union[str, _models.CapacityReservationGroupInstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation that retrieves information about a capacity reservation group. @@ -255,6 +264,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -306,7 +319,7 @@ async def create_or_update( parameters: _models.CapacityReservationGroup, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. Please refer to @@ -335,7 +348,7 @@ async def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. Please refer to @@ -362,7 +375,7 @@ async def create_or_update( resource_group_name: str, capacity_reservation_group_name: str, parameters: Union[_models.CapacityReservationGroup, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. Please refer to @@ -380,6 +393,10 @@ async def create_or_update( :rtype: ~azure.mgmt.compute.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -442,7 +459,7 @@ async def update( parameters: _models.CapacityReservationGroupUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. @@ -471,7 +488,7 @@ async def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. @@ -498,7 +515,7 @@ async def update( resource_group_name: str, capacity_reservation_group_name: str, parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. @@ -515,6 +532,10 @@ async def update( :rtype: ~azure.mgmt.compute.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -570,7 +591,7 @@ async def update( return deserialized # type: ignore @distributed_trace_async - async def delete(self, resource_group_name: str, capacity_reservation_group_name: str, **kwargs: Any) -> None: + async def delete(self, resource_group_name: str, capacity_reservation_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """The operation to delete a capacity reservation group. This operation is allowed only if all the associated resources are disassociated from the reservation group and all capacity reservations under the reservation group have also been deleted. Please refer to @@ -585,6 +606,10 @@ async def delete(self, resource_group_name: str, capacity_reservation_group_name :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservations_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservations_operations.py index fca2399bd3ee..1ed2b070eee4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservations_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_capacity_reservations_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._capacity_reservations_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_capacity_reservation_group( - self, resource_group_name: str, capacity_reservation_group_name: str, **kwargs: Any + self, resource_group_name: str, capacity_reservation_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.CapacityReservation"]: """Lists all of the capacity reservations in the specified capacity reservation group. Use the nextLink property in the response to get the next page of capacity reservations. @@ -81,6 +82,10 @@ def list_by_capacity_reservation_group( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CapacityReservation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -145,7 +150,7 @@ async def get( capacity_reservation_group_name: str, capacity_reservation_name: str, expand: Optional[Union[str, _models.CapacityReservationInstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservation: """The operation that retrieves information about the capacity reservation. @@ -164,6 +169,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -290,7 +299,7 @@ async def begin_create_or_update( parameters: _models.CapacityReservation, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be set only during capacity reservation creation. Please refer to @@ -323,7 +332,7 @@ async def begin_create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be set only during capacity reservation creation. Please refer to @@ -354,7 +363,7 @@ async def begin_create_or_update( capacity_reservation_group_name: str, capacity_reservation_name: str, parameters: Union[_models.CapacityReservation, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be set only during capacity reservation creation. Please refer to @@ -375,6 +384,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.CapacityReservation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -507,7 +520,7 @@ async def begin_update( parameters: _models.CapacityReservationUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -538,7 +551,7 @@ async def begin_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -567,7 +580,7 @@ async def begin_update( capacity_reservation_group_name: str, capacity_reservation_name: str, parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -586,6 +599,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.CapacityReservation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -702,7 +719,7 @@ async def begin_delete( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete a capacity reservation. This operation is allowed only when all the associated resources are disassociated from the capacity reservation. Please refer to @@ -720,6 +737,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_operating_systems_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_operating_systems_operations.py index 983b01d22ced..2965bd032013 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_operating_systems_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_operating_systems_operations.py @@ -27,6 +27,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._cloud_service_operating_systems_operations import ( build_get_os_family_request, build_get_os_version_request, @@ -60,7 +61,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> _models.OSVersion: + async def get_os_version(self, location: str, os_version_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.OSVersion: """Gets properties of a guest operating system version that can be specified in the XML service configuration (.cscfg) for a cloud service. @@ -72,6 +73,10 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An :rtype: ~azure.mgmt.compute.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -115,7 +120,7 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An return deserialized # type: ignore @distributed_trace - def list_os_versions(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.OSVersion"]: + def list_os_versions(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.OSVersion"]: """Gets a list of all guest operating system versions available to be specified in the XML service configuration (.cscfg) for a cloud service. Use nextLink property in the response to get the next page of OS versions. Do this till nextLink is null to fetch all the OS versions. @@ -126,6 +131,10 @@ def list_os_versions(self, location: str, **kwargs: Any) -> AsyncItemPaged["_mod :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.OSVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -183,7 +192,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: + async def get_os_family(self, location: str, os_family_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service configuration (.cscfg) for a cloud service. @@ -195,6 +204,10 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) :rtype: ~azure.mgmt.compute.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -238,7 +251,7 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) return deserialized # type: ignore @distributed_trace - def list_os_families(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.OSFamily"]: + def list_os_families(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.OSFamily"]: """Gets a list of all guest operating system families available to be specified in the XML service configuration (.cscfg) for a cloud service. Use nextLink property in the response to get the next page of OS Families. Do this till nextLink is null to fetch all the OS Families. @@ -249,6 +262,10 @@ def list_os_families(self, location: str, **kwargs: Any) -> AsyncItemPaged["_mod :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.OSFamily] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_role_instances_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_role_instances_operations.py index 0c418a0a9a24..a25dd95d908b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_role_instances_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_role_instances_operations.py @@ -31,6 +31,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._cloud_service_role_instances_operations import ( build_delete_request, build_get_instance_view_request, @@ -120,7 +121,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a role instance from a cloud service. @@ -134,6 +135,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -182,7 +187,7 @@ async def get( resource_group_name: str, cloud_service_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RoleInstance: """Gets a role instance from a cloud service. @@ -200,6 +205,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -246,7 +255,7 @@ async def get( @distributed_trace_async async def get_instance_view( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RoleInstanceView: """Retrieves information about the run-time state of a role instance in a cloud service. @@ -260,6 +269,10 @@ async def get_instance_view( :rtype: ~azure.mgmt.compute.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -309,7 +322,7 @@ def list( resource_group_name: str, cloud_service_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.RoleInstance"]: """Gets the list of all role instances in a cloud service. Use nextLink property in the response to get the next page of role instances. Do this till nextLink is null to fetch all the role @@ -327,6 +340,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.RoleInstance] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -438,7 +455,7 @@ async def _restart_initial( @distributed_trace_async async def begin_restart( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The Reboot Role Instance asynchronous operation requests a reboot of a role instance in the cloud service. @@ -453,6 +470,10 @@ async def begin_restart( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -547,7 +568,7 @@ async def _reimage_initial( @distributed_trace_async async def begin_reimage( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The Reimage Role Instance asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -562,6 +583,10 @@ async def begin_reimage( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -656,7 +681,7 @@ async def _rebuild_initial( @distributed_trace_async async def begin_rebuild( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The Rebuild Role Instance asynchronous operation reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If @@ -672,6 +697,10 @@ async def begin_rebuild( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -715,7 +744,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace_async async def get_remote_desktop_file( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncIterator[bytes]: """Gets a remote desktop file for a role instance in a cloud service. @@ -729,6 +758,10 @@ async def get_remote_desktop_file( :rtype: AsyncIterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_roles_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_roles_operations.py index 1784c8731157..2daaa7b3e4c1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_roles_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_service_roles_operations.py @@ -27,6 +27,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._cloud_service_roles_operations import build_get_request, build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -56,7 +57,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace_async async def get( - self, role_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CloudServiceRole: """Gets a role from a cloud service. @@ -70,6 +71,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -115,7 +120,7 @@ async def get( @distributed_trace def list( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.CloudServiceRole"]: """Gets a list of all roles in a cloud service. Use nextLink property in the response to get the next page of roles. Do this till nextLink is null to fetch all the roles. @@ -128,6 +133,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CloudServiceRole] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_operations.py index 9f68999b8e60..840ee72ff186 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._cloud_services_operations import ( build_create_or_update_request, build_delete_instances_request, @@ -152,7 +153,7 @@ async def begin_create_or_update( parameters: Optional[_models.CloudService] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud service creation. @@ -180,7 +181,7 @@ async def begin_create_or_update( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud service creation. @@ -206,7 +207,7 @@ async def begin_create_or_update( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud service creation. @@ -223,6 +224,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -349,7 +354,7 @@ async def begin_update( parameters: Optional[_models.CloudServiceUpdate] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Update a cloud service. @@ -376,7 +381,7 @@ async def begin_update( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Update a cloud service. @@ -401,7 +406,7 @@ async def begin_update( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Update a cloud service. @@ -417,6 +422,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -517,7 +526,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a cloud service. @@ -529,6 +538,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -570,7 +583,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: + async def get(self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.CloudService: """Display information about a cloud service. :param resource_group_name: Name of the resource group. Required. @@ -581,6 +594,10 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: :rtype: ~azure.mgmt.compute.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -625,7 +642,7 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: @distributed_trace_async async def get_instance_view( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CloudServiceInstanceView: """Gets the status of a cloud service. @@ -637,6 +654,10 @@ async def get_instance_view( :rtype: ~azure.mgmt.compute.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -680,7 +701,7 @@ async def get_instance_view( return deserialized # type: ignore @distributed_trace - def list_all(self, **kwargs: Any) -> AsyncItemPaged["_models.CloudService"]: + def list_all(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.CloudService"]: """Gets a list of all cloud services in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. @@ -689,6 +710,10 @@ def list_all(self, **kwargs: Any) -> AsyncItemPaged["_models.CloudService"]: :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -745,7 +770,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.CloudService"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. @@ -756,6 +781,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_mode :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -864,7 +893,7 @@ async def _start_initial( @distributed_trace_async async def begin_start( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts the cloud service. @@ -876,6 +905,10 @@ async def begin_start( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -968,7 +1001,7 @@ async def _power_off_initial( @distributed_trace_async async def begin_power_off( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off the cloud service. Note that resources are still attached and you are getting charged for the resources. @@ -981,6 +1014,10 @@ async def begin_power_off( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1099,7 +1136,7 @@ async def begin_restart( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -1125,7 +1162,7 @@ async def begin_restart( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -1149,7 +1186,7 @@ async def begin_restart( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -1164,6 +1201,10 @@ async def begin_restart( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1286,7 +1327,7 @@ async def begin_reimage( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -1313,7 +1354,7 @@ async def begin_reimage( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -1338,7 +1379,7 @@ async def begin_reimage( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -1354,6 +1395,10 @@ async def begin_reimage( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1476,7 +1521,7 @@ async def begin_rebuild( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If you do not want to @@ -1504,7 +1549,7 @@ async def begin_rebuild( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If you do not want to @@ -1530,7 +1575,7 @@ async def begin_rebuild( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If you do not want to @@ -1547,6 +1592,10 @@ async def begin_rebuild( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1669,7 +1718,7 @@ async def begin_delete_instances( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes role instances in a cloud service. @@ -1695,7 +1744,7 @@ async def begin_delete_instances( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes role instances in a cloud service. @@ -1719,7 +1768,7 @@ async def begin_delete_instances( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes role instances in a cloud service. @@ -1734,6 +1783,10 @@ async def begin_delete_instances( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_update_domain_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_update_domain_operations.py index 0dd3ac13ac51..d4832365de9e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_update_domain_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_cloud_services_update_domain_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._cloud_services_update_domain_operations import ( build_get_update_domain_request, build_list_update_domains_request, @@ -144,7 +145,7 @@ async def begin_walk_update_domain( parameters: Optional[_models.UpdateDomain] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Updates the role instances in the specified update domain. @@ -175,7 +176,7 @@ async def begin_walk_update_domain( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Updates the role instances in the specified update domain. @@ -204,7 +205,7 @@ async def begin_walk_update_domain( cloud_service_name: str, update_domain: int, parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Updates the role instances in the specified update domain. @@ -223,6 +224,10 @@ async def begin_walk_update_domain( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -270,7 +275,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace_async async def get_update_domain( - self, resource_group_name: str, cloud_service_name: str, update_domain: int, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, update_domain: int, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.UpdateDomain: """Gets the specified update domain of a cloud service. Use nextLink property in the response to get the next page of update domains. Do this till nextLink is null to fetch all the update @@ -288,6 +293,10 @@ async def get_update_domain( :rtype: ~azure.mgmt.compute.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -333,7 +342,7 @@ async def get_update_domain( @distributed_trace def list_update_domains( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.UpdateDomain"]: """Gets a list of all update domains in a cloud service. @@ -345,6 +354,10 @@ def list_update_domains( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.UpdateDomain] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_galleries_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_galleries_operations.py index 91f6921b85bc..c123f823381d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_galleries_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_galleries_operations.py @@ -25,6 +25,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._community_galleries_operations import build_get_request from .._configuration import ComputeManagementClientConfiguration @@ -53,7 +54,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace_async - async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models.CommunityGallery: + async def get(self, location: str, public_gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.CommunityGallery: """Get a community gallery by gallery public name. :param location: The name of Azure region. Required. @@ -64,6 +65,10 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ :rtype: ~azure.mgmt.compute.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_image_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_image_versions_operations.py index ee19f01d2ad1..09a02e600fb8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_image_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_image_versions_operations.py @@ -27,6 +27,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._community_gallery_image_versions_operations import build_get_request, build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -56,7 +57,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, location: str, public_gallery_name: str, gallery_image_name: str, **kwargs: Any + self, location: str, public_gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.CommunityGalleryImageVersion"]: """List community gallery image versions inside an image. @@ -72,6 +73,10 @@ def list( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CommunityGalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -137,7 +142,7 @@ async def get( public_gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CommunityGalleryImageVersion: """Get a community gallery image version. @@ -156,6 +161,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_images_operations.py index 4632db83d3b7..91ba3e7c1866 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_community_gallery_images_operations.py @@ -27,6 +27,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._community_gallery_images_operations import build_get_request, build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -56,7 +57,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, location: str, public_gallery_name: str, **kwargs: Any + self, location: str, public_gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.CommunityGalleryImage"]: """List community gallery images inside a gallery. @@ -70,6 +71,10 @@ def list( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.CommunityGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -129,7 +134,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, location: str, public_gallery_name: str, gallery_image_name: str, **kwargs: Any + self, location: str, public_gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CommunityGalleryImage: """Get a community gallery image. @@ -143,6 +148,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_host_groups_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_host_groups_operations.py index e02d26432bcb..643270a911ce 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_host_groups_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_host_groups_operations.py @@ -28,6 +28,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._dedicated_host_groups_operations import ( build_create_or_update_request, build_delete_request, @@ -63,7 +64,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.DedicatedHostGroup"]: + def list_by_subscription(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. @@ -71,6 +72,10 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.Dedicat :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -128,7 +133,7 @@ async def get_next(next_link=None): @distributed_trace def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any + self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the specified resource group. Use the nextLink property in the response to get the next page of dedicated host groups. @@ -140,6 +145,10 @@ def list_by_resource_group( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -202,7 +211,7 @@ async def get( resource_group_name: str, host_group_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Retrieves information about a dedicated host group. @@ -220,6 +229,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -271,7 +284,7 @@ async def create_or_update( parameters: _models.DedicatedHostGroup, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host Groups please see [Dedicated Host Documentation] @@ -300,7 +313,7 @@ async def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host Groups please see [Dedicated Host Documentation] @@ -327,7 +340,7 @@ async def create_or_update( resource_group_name: str, host_group_name: str, parameters: Union[_models.DedicatedHostGroup, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host Groups please see [Dedicated Host Documentation] @@ -345,6 +358,10 @@ async def create_or_update( :rtype: ~azure.mgmt.compute.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -407,7 +424,7 @@ async def update( parameters: _models.DedicatedHostGroupUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -434,7 +451,7 @@ async def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -459,7 +476,7 @@ async def update( resource_group_name: str, host_group_name: str, parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -475,6 +492,10 @@ async def update( :rtype: ~azure.mgmt.compute.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -530,7 +551,7 @@ async def update( return deserialized # type: ignore @distributed_trace_async - async def delete(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> None: + async def delete(self, resource_group_name: str, host_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete a dedicated host group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -542,6 +563,10 @@ async def delete(self, resource_group_name: str, host_group_name: str, **kwargs: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_hosts_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_hosts_operations.py index 3b221393166b..9a8e490b1652 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_hosts_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_dedicated_hosts_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._dedicated_hosts_operations import ( build_create_or_update_request, build_delete_request, @@ -71,7 +72,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_host_group( - self, resource_group_name: str, host_group_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.DedicatedHost"]: """Lists all of the dedicated hosts in the specified dedicated host group. Use the nextLink property in the response to get the next page of dedicated hosts. @@ -85,6 +86,10 @@ def list_by_host_group( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -149,7 +154,7 @@ async def get( host_group_name: str, host_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHost: """Retrieves information about a dedicated host. @@ -169,6 +174,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -295,7 +304,7 @@ async def begin_create_or_update( parameters: _models.DedicatedHost, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -326,7 +335,7 @@ async def begin_create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -355,7 +364,7 @@ async def begin_create_or_update( host_group_name: str, host_name: str, parameters: Union[_models.DedicatedHost, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -374,6 +383,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -505,7 +518,7 @@ async def begin_update( parameters: _models.DedicatedHostUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -536,7 +549,7 @@ async def begin_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -565,7 +578,7 @@ async def begin_update( host_group_name: str, host_name: str, parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -584,6 +597,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -697,7 +714,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a dedicated host. @@ -712,6 +729,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -757,7 +778,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_available_sizes( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged[str]: """Lists all available dedicated host sizes to which the specified dedicated host can be resized. NOTE: The dedicated host sizes provided can be used to only scale up the existing dedicated @@ -774,6 +795,10 @@ def list_available_sizes( :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -889,7 +914,7 @@ async def _redeploy_initial( @distributed_trace_async async def begin_redeploy( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Redeploy the dedicated host. The operation will complete successfully once the dedicated host has migrated to a new node and is running. To determine the health of VMs deployed on the @@ -908,6 +933,10 @@ async def begin_redeploy( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1008,7 +1037,7 @@ async def _restart_initial( @distributed_trace_async async def begin_restart( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restart the dedicated host. The operation will complete successfully once the dedicated host has restarted and is running. To determine the health of VMs deployed on the dedicated host @@ -1026,6 +1055,10 @@ async def begin_restart( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_accesses_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_accesses_operations.py index c9f9d5080013..a3bca96db3f7 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_accesses_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_accesses_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._disk_accesses_operations import ( build_create_or_update_request, build_delete_a_private_endpoint_connection_request, @@ -73,13 +74,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.DiskAccess"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -136,7 +141,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.DiskAccess"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.DiskAccess"]: """Lists all the disk access resources under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -146,6 +151,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -203,7 +212,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: + async def get(self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DiskAccess: """Gets information about a disk access resource. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -217,6 +226,10 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A :rtype: ~azure.mgmt.compute.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -338,7 +351,7 @@ async def begin_create_or_update( disk_access: _models.DiskAccess, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -369,7 +382,7 @@ async def begin_create_or_update( disk_access: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -398,7 +411,7 @@ async def begin_create_or_update( resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -417,6 +430,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -545,7 +562,7 @@ async def begin_update( disk_access: _models.DiskAccessUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -576,7 +593,7 @@ async def begin_update( disk_access: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -605,7 +622,7 @@ async def begin_update( resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -624,6 +641,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -730,7 +751,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, disk_access_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a disk access resource. @@ -745,6 +766,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -789,7 +814,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_private_endpoint_connections( - self, resource_group_name: str, disk_access_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.PrivateEndpointConnection"]: """List information about private endpoint connections under a disk access resource. @@ -806,6 +831,10 @@ def list_private_endpoint_connections( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.PrivateEndpointConnection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -865,7 +894,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get_a_private_endpoint_connection( - self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PrivateEndpointConnection: """Gets information about a private endpoint connection under a disk access resource. @@ -882,6 +911,10 @@ async def get_a_private_endpoint_connection( :rtype: ~azure.mgmt.compute.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1007,7 +1040,7 @@ async def begin_update_a_private_endpoint_connection( # pylint: disable=name-to private_endpoint_connection: _models.PrivateEndpointConnection, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used to create a new private endpoint connection. @@ -1043,7 +1076,7 @@ async def begin_update_a_private_endpoint_connection( # pylint: disable=name-to private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used to create a new private endpoint connection. @@ -1077,7 +1110,7 @@ async def begin_update_a_private_endpoint_connection( # pylint: disable=name-to disk_access_name: str, private_endpoint_connection_name: str, private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used to create a new private endpoint connection. @@ -1102,6 +1135,10 @@ async def begin_update_a_private_endpoint_connection( # pylint: disable=name-to ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.PrivateEndpointConnection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1210,7 +1247,7 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=name @distributed_trace_async async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long - self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -1227,6 +1264,10 @@ async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-to :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1272,7 +1313,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace_async async def get_private_link_resources( - self, resource_group_name: str, disk_access_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PrivateLinkResourceListResult: """Gets the private link resources possible under disk access resource. @@ -1287,6 +1328,10 @@ async def get_private_link_resources( :rtype: ~azure.mgmt.compute.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_encryption_sets_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_encryption_sets_operations.py index c135d4d0fdf2..d9f4a6c2b648 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_encryption_sets_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_encryption_sets_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._disk_encryption_sets_operations import ( build_create_or_update_request, build_delete_request, @@ -68,13 +69,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.DiskEncryptionSet"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -132,7 +137,7 @@ async def get_next(next_link=None): @distributed_trace def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any + self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a resource group. @@ -143,6 +148,10 @@ def list_by_resource_group( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -201,7 +210,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any + self, resource_group_name: str, disk_encryption_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DiskEncryptionSet: """Gets information about a disk encryption set. @@ -216,6 +225,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -337,7 +350,7 @@ async def begin_create_or_update( disk_encryption_set: _models.DiskEncryptionSet, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -368,7 +381,7 @@ async def begin_create_or_update( disk_encryption_set: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -397,7 +410,7 @@ async def begin_create_or_update( resource_group_name: str, disk_encryption_set_name: str, disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -416,6 +429,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -544,7 +561,7 @@ async def begin_update( disk_encryption_set: _models.DiskEncryptionSetUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -575,7 +592,7 @@ async def begin_update( disk_encryption_set: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -604,7 +621,7 @@ async def begin_update( resource_group_name: str, disk_encryption_set_name: str, disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -624,6 +641,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -730,7 +751,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any + self, resource_group_name: str, disk_encryption_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a disk encryption set. @@ -745,6 +766,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -789,7 +814,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_associated_resources( - self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any + self, resource_group_name: str, disk_encryption_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged[str]: """Lists all resources that are encrypted with this disk encryption set. @@ -804,6 +829,10 @@ def list_associated_resources( :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_restore_point_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_restore_point_operations.py index 3ad112e7da66..eb36f5483e48 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_restore_point_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disk_restore_point_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._disk_restore_point_operations import ( build_get_request, build_grant_access_request, @@ -66,7 +67,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_restore_point( - self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, **kwargs: Any + self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.DiskRestorePoint"]: """Lists diskRestorePoints under a vmRestorePoint. @@ -83,6 +84,10 @@ def list_by_restore_point( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -148,7 +153,7 @@ async def get( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DiskRestorePoint: """Get disk restorePoint resource. @@ -167,6 +172,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -296,7 +305,7 @@ async def begin_grant_access( grant_access_data: _models.GrantAccessData, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -333,7 +342,7 @@ async def begin_grant_access( grant_access_data: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -368,7 +377,7 @@ async def begin_grant_access( vm_restore_point_name: str, disk_restore_point_name: str, grant_access_data: Union[_models.GrantAccessData, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -391,6 +400,10 @@ async def begin_grant_access( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -511,7 +524,7 @@ async def begin_revoke_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Revokes access to a diskRestorePoint. @@ -530,6 +543,10 @@ async def begin_revoke_access( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disks_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disks_operations.py index b2f1bd4ed240..8d7401fe59bd 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disks_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_disks_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._disks_operations import ( build_create_or_update_request, build_delete_request, @@ -70,13 +71,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Disk"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Disk"]: """Lists all the disks under a subscription. :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -133,7 +138,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.Disk"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Disk"]: """Lists all the disks under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -143,6 +148,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -200,7 +209,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: + async def get(self, resource_group_name: str, disk_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Disk: """Gets information about a disk. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -214,6 +223,10 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> :rtype: ~azure.mgmt.compute.models.Disk :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -331,7 +344,7 @@ async def begin_create_or_update( disk: _models.Disk, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -360,7 +373,7 @@ async def begin_create_or_update( disk: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -383,7 +396,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -401,6 +414,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -525,7 +542,7 @@ async def begin_update( disk: _models.DiskUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -554,7 +571,7 @@ async def begin_update( disk: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -577,7 +594,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -595,6 +612,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -698,7 +719,7 @@ async def _delete_initial(self, resource_group_name: str, disk_name: str, **kwar return deserialized # type: ignore @distributed_trace_async - async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_delete(self, resource_group_name: str, disk_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """Deletes a disk. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -712,6 +733,10 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -833,7 +858,7 @@ async def begin_grant_access( grant_access_data: _models.GrantAccessData, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -864,7 +889,7 @@ async def begin_grant_access( grant_access_data: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -893,7 +918,7 @@ async def begin_grant_access( resource_group_name: str, disk_name: str, grant_access_data: Union[_models.GrantAccessData, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -912,6 +937,10 @@ async def begin_grant_access( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1018,7 +1047,7 @@ async def _revoke_access_initial( @distributed_trace_async async def begin_revoke_access( - self, resource_group_name: str, disk_name: str, **kwargs: Any + self, resource_group_name: str, disk_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Revokes access to a disk. @@ -1033,6 +1062,10 @@ async def begin_revoke_access( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_galleries_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_galleries_operations.py index 66bd47f9a305..fba5e2345c2f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_galleries_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_galleries_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._galleries_operations import ( build_create_or_update_request, build_delete_request, @@ -67,13 +68,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Gallery"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Gallery"]: """List galleries under a subscription. :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -130,7 +135,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.Gallery"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Gallery"]: """List galleries under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -140,6 +145,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -203,7 +212,7 @@ async def get( gallery_name: str, select: Optional[Union[str, _models.SelectPermissions]] = None, expand: Optional[Union[str, _models.GalleryExpandParams]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Gallery: """Retrieves information about a Shared Image Gallery. @@ -222,6 +231,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -345,7 +358,7 @@ async def begin_create_or_update( gallery: _models.Gallery, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -374,7 +387,7 @@ async def begin_create_or_update( gallery: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -397,7 +410,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -414,6 +427,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -541,7 +558,7 @@ async def begin_update( gallery: _models.GalleryUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -569,7 +586,7 @@ async def begin_update( gallery: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -595,7 +612,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -612,6 +629,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -720,7 +741,7 @@ async def _delete_initial(self, resource_group_name: str, gallery_name: str, **k return deserialized # type: ignore @distributed_trace_async - async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_delete(self, resource_group_name: str, gallery_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """Delete a Shared Image Gallery. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -732,6 +753,10 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_application_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_application_versions_operations.py index 8db00316577a..7666cfcc0016 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_application_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_application_versions_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_application_versions_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_application( - self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_application_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryApplicationVersion"]: """List gallery Application Versions in a gallery Application Definition. @@ -85,6 +86,10 @@ def list_by_gallery_application( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryApplicationVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -151,7 +156,7 @@ async def get( gallery_application_name: str, gallery_application_version_name: str, expand: Optional[Union[str, _models.ReplicationStatusTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryApplicationVersion: """Retrieves information about a gallery Application Version. @@ -173,6 +178,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -307,7 +316,7 @@ async def begin_create_or_update( gallery_application_version: _models.GalleryApplicationVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -345,7 +354,7 @@ async def begin_create_or_update( gallery_application_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -381,7 +390,7 @@ async def begin_create_or_update( gallery_application_name: str, gallery_application_version_name: str, gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -407,6 +416,10 @@ async def begin_create_or_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryApplicationVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -542,7 +555,7 @@ async def begin_update( gallery_application_version: _models.GalleryApplicationVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -580,7 +593,7 @@ async def begin_update( gallery_application_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -616,7 +629,7 @@ async def begin_update( gallery_application_name: str, gallery_application_version_name: str, gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -642,6 +655,10 @@ async def begin_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryApplicationVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -767,7 +784,7 @@ async def begin_delete( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery Application Version. @@ -786,6 +803,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_applications_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_applications_operations.py index 83676f0d8cd2..9bc2df79a752 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_applications_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_applications_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_applications_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryApplication"]: """List gallery Application Definitions in a gallery. @@ -80,6 +81,10 @@ def list_by_gallery( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -139,7 +144,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_application_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryApplication: """Retrieves information about a gallery Application Definition. @@ -155,6 +160,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -284,7 +293,7 @@ async def begin_create_or_update( gallery_application: _models.GalleryApplication, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -317,7 +326,7 @@ async def begin_create_or_update( gallery_application: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -348,7 +357,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application: Union[_models.GalleryApplication, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -368,6 +377,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -499,7 +512,7 @@ async def begin_update( gallery_application: _models.GalleryApplicationUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -532,7 +545,7 @@ async def begin_update( gallery_application: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -563,7 +576,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -583,6 +596,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -696,7 +713,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_application_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery Application. @@ -712,6 +729,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_image_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_image_versions_operations.py index 53f5e4c5b42e..878321573180 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_image_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_image_versions_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_image_versions_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_image( - self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryImageVersion"]: """List gallery image versions in a gallery image definition. @@ -82,6 +83,10 @@ def list_by_gallery_image( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -148,7 +153,7 @@ async def get( gallery_image_name: str, gallery_image_version_name: str, expand: Optional[Union[str, _models.ReplicationStatusTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryImageVersion: """Retrieves information about a gallery image version. @@ -169,6 +174,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -303,7 +312,7 @@ async def begin_create_or_update( gallery_image_version: _models.GalleryImageVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -339,7 +348,7 @@ async def begin_create_or_update( gallery_image_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -373,7 +382,7 @@ async def begin_create_or_update( gallery_image_name: str, gallery_image_version_name: str, gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -395,6 +404,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -530,7 +543,7 @@ async def begin_update( gallery_image_version: _models.GalleryImageVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -566,7 +579,7 @@ async def begin_update( gallery_image_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -600,7 +613,7 @@ async def begin_update( gallery_image_name: str, gallery_image_version_name: str, gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -622,6 +635,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -747,7 +764,7 @@ async def begin_delete( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery image version. @@ -765,6 +782,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_images_operations.py index 52122aeeddc2..7016b8015fc4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_images_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_images_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryImage"]: """List gallery image definitions in a gallery. @@ -80,6 +81,10 @@ def list_by_gallery( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -139,7 +144,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryImage: """Retrieves information about a gallery image definition. @@ -154,6 +159,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -283,7 +292,7 @@ async def begin_create_or_update( gallery_image: _models.GalleryImage, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -315,7 +324,7 @@ async def begin_create_or_update( gallery_image: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -345,7 +354,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image: Union[_models.GalleryImage, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -364,6 +373,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -495,7 +508,7 @@ async def begin_update( gallery_image: _models.GalleryImageUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -526,7 +539,7 @@ async def begin_update( gallery_image: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -555,7 +568,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -574,6 +587,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -687,7 +704,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_image_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery image. @@ -702,6 +719,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profile_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profile_versions_operations.py index 0bb5ce07a754..f5eb5667eb1e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profile_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profile_versions_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_in_vm_access_control_profile_versions_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_in_vm_access_control_profile( # pylint: disable=name-too-long - self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryInVMAccessControlProfileVersion"]: """List gallery inVMAccessControlProfile versions in a gallery inVMAccessControlProfile. @@ -85,6 +86,10 @@ def list_by_gallery_in_vm_access_control_profile( # pylint: disable=name-too-lo ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -150,7 +155,7 @@ async def get( gallery_name: str, in_vm_access_control_profile_name: str, in_vm_access_control_profile_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryInVMAccessControlProfileVersion: """Retrieves information about a gallery inVMAccessControlProfile version. @@ -169,6 +174,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -300,7 +309,7 @@ async def begin_create_or_update( gallery_in_vm_access_control_profile_version: _models.GalleryInVMAccessControlProfileVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Create or update a gallery inVMAccessControlProfile version. @@ -339,7 +348,7 @@ async def begin_create_or_update( gallery_in_vm_access_control_profile_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Create or update a gallery inVMAccessControlProfile version. @@ -375,7 +384,7 @@ async def begin_create_or_update( in_vm_access_control_profile_name: str, in_vm_access_control_profile_version_name: str, gallery_in_vm_access_control_profile_version: Union[_models.GalleryInVMAccessControlProfileVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Create or update a gallery inVMAccessControlProfile version. @@ -401,6 +410,10 @@ async def begin_create_or_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -540,7 +553,7 @@ async def begin_update( gallery_in_vm_access_control_profile_version: _models.GalleryInVMAccessControlProfileVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Update a gallery inVMAccessControlProfile version. @@ -579,7 +592,7 @@ async def begin_update( gallery_in_vm_access_control_profile_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Update a gallery inVMAccessControlProfile version. @@ -617,7 +630,7 @@ async def begin_update( gallery_in_vm_access_control_profile_version: Union[ _models.GalleryInVMAccessControlProfileVersionUpdate, IO[bytes] ], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Update a gallery inVMAccessControlProfile version. @@ -643,6 +656,10 @@ async def begin_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -770,7 +787,7 @@ async def begin_delete( gallery_name: str, in_vm_access_control_profile_name: str, in_vm_access_control_profile_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery inVMAccessControlProfile version. @@ -789,6 +806,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profiles_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profiles_operations.py index e50e47d6276b..10d09e85b3e0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profiles_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_in_vm_access_control_profiles_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_in_vm_access_control_profiles_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryInVMAccessControlProfile"]: """List gallery inVMAccessControlProfiles in a gallery. @@ -82,6 +83,10 @@ def list_by_gallery( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryInVMAccessControlProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -141,7 +146,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryInVMAccessControlProfile: """Retrieves information about a gallery inVMAccessControlProfile. @@ -157,6 +162,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryInVMAccessControlProfile :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -282,7 +291,7 @@ async def begin_create_or_update( gallery_in_vm_access_control_profile: _models.GalleryInVMAccessControlProfile, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfile]: """Create or update a gallery inVMAccessControlProfile. @@ -317,7 +326,7 @@ async def begin_create_or_update( gallery_in_vm_access_control_profile: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfile]: """Create or update a gallery inVMAccessControlProfile. @@ -349,7 +358,7 @@ async def begin_create_or_update( gallery_name: str, in_vm_access_control_profile_name: str, gallery_in_vm_access_control_profile: Union[_models.GalleryInVMAccessControlProfile, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfile]: """Create or update a gallery inVMAccessControlProfile. @@ -372,6 +381,10 @@ async def begin_create_or_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -503,7 +516,7 @@ async def begin_update( gallery_in_vm_access_control_profile: _models.GalleryInVMAccessControlProfileUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfile]: """Update a gallery inVMAccessControlProfile. @@ -538,7 +551,7 @@ async def begin_update( gallery_in_vm_access_control_profile: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfile]: """Update a gallery inVMAccessControlProfile. @@ -570,7 +583,7 @@ async def begin_update( gallery_name: str, in_vm_access_control_profile_name: str, gallery_in_vm_access_control_profile: Union[_models.GalleryInVMAccessControlProfileUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryInVMAccessControlProfile]: """Update a gallery inVMAccessControlProfile. @@ -593,6 +606,10 @@ async def begin_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -708,7 +725,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery inVMAccessControlProfile. @@ -724,6 +741,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_script_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_script_versions_operations.py index 3d8e8c923a15..b6656d59277a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_script_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_script_versions_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_script_versions_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_script( - self, resource_group_name: str, gallery_name: str, gallery_script_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_script_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryScriptVersion"]: """List gallery Script Versions in a gallery Script Definition. @@ -85,6 +86,10 @@ def list_by_gallery_script( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryScriptVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -150,7 +155,7 @@ async def get( gallery_name: str, gallery_script_name: str, gallery_script_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryScriptVersion: """Custom ArmResourceRead operation template with CloudError as Error. @@ -169,6 +174,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryScriptVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -301,7 +310,7 @@ async def begin_create_or_update( gallery_script_version: _models.GalleryScriptVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScriptVersion]: """Create or update a gallery Script Version. @@ -338,7 +347,7 @@ async def begin_create_or_update( gallery_script_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScriptVersion]: """Create or update a gallery Script Version. @@ -373,7 +382,7 @@ async def begin_create_or_update( gallery_script_name: str, gallery_script_version_name: str, gallery_script_version: Union[_models.GalleryScriptVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScriptVersion]: """Create or update a gallery Script Version. @@ -396,6 +405,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryScriptVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -536,7 +549,7 @@ async def begin_update( gallery_script_version: _models.GalleryScriptVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScriptVersion]: """Update a gallery Script Version. @@ -573,7 +586,7 @@ async def begin_update( gallery_script_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScriptVersion]: """Update a gallery Script Version. @@ -608,7 +621,7 @@ async def begin_update( gallery_script_name: str, gallery_script_version_name: str, gallery_script_version: Union[_models.GalleryScriptVersionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScriptVersion]: """Update a gallery Script Version. @@ -632,6 +645,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryScriptVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -756,7 +773,7 @@ async def begin_delete( gallery_name: str, gallery_script_name: str, gallery_script_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery Script Version. @@ -775,6 +792,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_scripts_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_scripts_operations.py index 92567b33dbc4..fe136a6adf88 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_scripts_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_scripts_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_scripts_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GalleryScript"]: """List gallery Script Definitions in a gallery. @@ -80,6 +81,10 @@ def list_by_gallery( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GalleryScript] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -139,7 +144,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, resource_group_name: str, gallery_name: str, gallery_script_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_script_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryScript: """Retrieves information about a gallery script definition. @@ -155,6 +160,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.GalleryScript :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -280,7 +289,7 @@ async def begin_create_or_update( gallery_script: _models.GalleryScript, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScript]: """Create or update a Gallery Script Definition. Gallery scripts allow the storage, sharing and reuse of common scripts. @@ -314,7 +323,7 @@ async def begin_create_or_update( gallery_script: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScript]: """Create or update a Gallery Script Definition. Gallery scripts allow the storage, sharing and reuse of common scripts. @@ -346,7 +355,7 @@ async def begin_create_or_update( gallery_name: str, gallery_script_name: str, gallery_script: Union[_models.GalleryScript, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScript]: """Create or update a Gallery Script Definition. Gallery scripts allow the storage, sharing and reuse of common scripts. @@ -367,6 +376,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryScript] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -499,7 +512,7 @@ async def begin_update( gallery_script: _models.GalleryScriptUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScript]: """Update a gallery Script Definition. @@ -531,7 +544,7 @@ async def begin_update( gallery_script: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScript]: """Update a gallery Script Definition. @@ -561,7 +574,7 @@ async def begin_update( gallery_name: str, gallery_script_name: str, gallery_script: Union[_models.GalleryScriptUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.GalleryScript]: """Update a gallery Script Definition. @@ -581,6 +594,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.GalleryScript] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -689,7 +706,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, gallery_name: str, gallery_script_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_script_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Delete a gallery Script Definition. @@ -705,6 +722,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_sharing_profile_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_sharing_profile_operations.py index c279d37b2cb4..4e6c30f7459c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_sharing_profile_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_gallery_sharing_profile_operations.py @@ -30,6 +30,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._gallery_sharing_profile_operations import build_update_request from .._configuration import ComputeManagementClientConfiguration @@ -136,7 +137,7 @@ async def begin_update( sharing_update: _models.SharingUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -164,7 +165,7 @@ async def begin_update( sharing_update: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -190,7 +191,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, sharing_update: Union[_models.SharingUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -207,6 +208,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.SharingUpdate] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_images_operations.py index 86fdd4595487..40fd419cd5db 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_images_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._images_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Image"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. @@ -75,6 +76,10 @@ def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Image"]: :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -131,7 +136,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.Image"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Image"]: """Gets the list of images under a resource group. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. @@ -142,6 +147,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -200,7 +209,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, resource_group_name: str, image_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, image_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Image: """Gets an image. @@ -215,6 +224,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.Image :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -333,7 +346,7 @@ async def begin_create_or_update( parameters: _models.Image, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -360,7 +373,7 @@ async def begin_create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -381,7 +394,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -397,6 +410,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -525,7 +542,7 @@ async def begin_update( parameters: _models.ImageUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -552,7 +569,7 @@ async def begin_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -577,7 +594,7 @@ async def begin_update( resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -593,6 +610,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -696,7 +717,7 @@ async def _delete_initial(self, resource_group_name: str, image_name: str, **kwa return deserialized # type: ignore @distributed_trace_async - async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_delete(self, resource_group_name: str, image_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """Deletes an Image. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -708,6 +729,10 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_log_analytics_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_log_analytics_operations.py index ae1591a6b38e..8f90dac2443e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_log_analytics_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_log_analytics_operations.py @@ -30,6 +30,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._log_analytics_operations import ( build_export_request_rate_by_interval_request, build_export_throttled_requests_request, @@ -135,7 +136,7 @@ async def begin_export_request_rate_by_interval( parameters: _models.RequestRateByIntervalInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -157,7 +158,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -179,7 +180,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -195,6 +196,10 @@ async def begin_export_request_rate_by_interval( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.LogAnalyticsOperationResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -319,7 +324,7 @@ async def begin_export_throttled_requests( parameters: _models.ThrottledRequestsInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -340,7 +345,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -361,7 +366,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -377,6 +382,10 @@ async def begin_export_throttled_requests( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.LogAnalyticsOperationResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_operations.py index f3bb26d06769..123b47784427 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_operations.py @@ -26,6 +26,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._operations import build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -54,13 +55,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Operation"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Operation"]: """List the operations for the provider. :return: An iterator like instance of either Operation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Operation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_proximity_placement_groups_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_proximity_placement_groups_operations.py index c22b1865f31c..a622c7249041 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_proximity_placement_groups_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_proximity_placement_groups_operations.py @@ -28,6 +28,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._proximity_placement_groups_operations import ( build_create_or_update_request, build_delete_request, @@ -63,7 +64,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.ProximityPlacementGroup"]: + def list_by_subscription(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. :return: An iterator like instance of either ProximityPlacementGroup or the result of @@ -72,6 +73,10 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.Proximi ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.ProximityPlacementGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -129,7 +134,7 @@ async def get_next(next_link=None): @distributed_trace def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any + self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a resource group. @@ -142,6 +147,10 @@ def list_by_resource_group( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.ProximityPlacementGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -204,7 +213,7 @@ async def get( resource_group_name: str, proximity_placement_group_name: str, include_colocation_status: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Retrieves information about a proximity placement group . @@ -220,6 +229,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -271,7 +284,7 @@ async def create_or_update( parameters: _models.ProximityPlacementGroup, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -299,7 +312,7 @@ async def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -325,7 +338,7 @@ async def create_or_update( resource_group_name: str, proximity_placement_group_name: str, parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -341,6 +354,10 @@ async def create_or_update( :rtype: ~azure.mgmt.compute.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -403,7 +420,7 @@ async def update( parameters: _models.ProximityPlacementGroupUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -431,7 +448,7 @@ async def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -457,7 +474,7 @@ async def update( resource_group_name: str, proximity_placement_group_name: str, parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -473,6 +490,10 @@ async def update( :rtype: ~azure.mgmt.compute.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -528,7 +549,7 @@ async def update( return deserialized # type: ignore @distributed_trace_async - async def delete(self, resource_group_name: str, proximity_placement_group_name: str, **kwargs: Any) -> None: + async def delete(self, resource_group_name: str, proximity_placement_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete a proximity placement group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -540,6 +561,10 @@ async def delete(self, resource_group_name: str, proximity_placement_group_name: :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_resource_skus_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_resource_skus_operations.py index db03a012a131..01ebdfa5b5fe 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_resource_skus_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_resource_skus_operations.py @@ -26,6 +26,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._resource_skus_operations import build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -55,7 +56,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, filter: Optional[str] = None, include_extended_locations: Optional[str] = None, **kwargs: Any + self, filter: Optional[str] = None, include_extended_locations: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.ResourceSku"]: """Gets the list of Microsoft.Compute SKUs available for your Subscription. @@ -69,6 +70,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.ResourceSku] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_point_collections_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_point_collections_operations.py index fbbe59c6cd24..39ff3f8b5932 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_point_collections_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_point_collections_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._restore_point_collections_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_all(self, **kwargs: Any) -> AsyncItemPaged["_models.RestorePointCollection"]: + def list_all(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. @@ -78,6 +79,10 @@ def list_all(self, **kwargs: Any) -> AsyncItemPaged["_models.RestorePointCollect ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.RestorePointCollection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -134,7 +139,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.RestorePointCollection"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.RestorePointCollection"]: """Gets the list of restore point collections in a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -146,6 +151,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_mode ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.RestorePointCollection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -208,7 +217,7 @@ async def get( resource_group_name: str, restore_point_collection_name: str, expand: Optional[Union[str, _models.RestorePointCollectionExpandOptions]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to get the restore point collection. @@ -225,6 +234,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -276,7 +289,7 @@ async def create_or_update( parameters: _models.RestorePointCollection, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to https://aka.ms/RestorePoints for more details. When updating a restore point collection, only @@ -306,7 +319,7 @@ async def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to https://aka.ms/RestorePoints for more details. When updating a restore point collection, only @@ -334,7 +347,7 @@ async def create_or_update( resource_group_name: str, restore_point_collection_name: str, parameters: Union[_models.RestorePointCollection, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to https://aka.ms/RestorePoints for more details. When updating a restore point collection, only @@ -352,6 +365,10 @@ async def create_or_update( :rtype: ~azure.mgmt.compute.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -414,7 +431,7 @@ async def update( parameters: _models.RestorePointCollectionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -442,7 +459,7 @@ async def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -468,7 +485,7 @@ async def update( resource_group_name: str, restore_point_collection_name: str, parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -484,6 +501,10 @@ async def update( :rtype: ~azure.mgmt.compute.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -595,7 +616,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any + self, resource_group_name: str, restore_point_collection_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete the restore point collection. This operation will also delete all the contained restore points. @@ -609,6 +630,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_points_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_points_operations.py index 7e654e2febae..35dd11d00571 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_points_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_restore_points_operations.py @@ -30,6 +30,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._restore_points_operations import build_create_request, build_delete_request, build_get_request from .._configuration import ComputeManagementClientConfiguration @@ -64,7 +65,7 @@ async def get( restore_point_collection_name: str, restore_point_name: str, expand: Optional[Union[str, _models.RestorePointExpandOptions]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePoint: """The operation to get the restore point. @@ -82,6 +83,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -207,7 +212,7 @@ async def begin_create( parameters: _models.RestorePoint, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is not allowed. @@ -239,7 +244,7 @@ async def begin_create( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is not allowed. @@ -269,7 +274,7 @@ async def begin_create( restore_point_collection_name: str, restore_point_name: str, parameters: Union[_models.RestorePoint, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is not allowed. @@ -289,6 +294,10 @@ async def begin_create( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.RestorePoint] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -402,7 +411,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any + self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete the restore point. @@ -417,6 +426,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_galleries_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_galleries_operations.py index 99b9963a5e94..c5cd9c95e8ed 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_galleries_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_galleries_operations.py @@ -27,6 +27,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._shared_galleries_operations import build_get_request, build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -56,7 +57,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, location: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, **kwargs: Any + self, location: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.SharedGallery"]: """List shared galleries by subscription id or tenant id. @@ -69,6 +70,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -127,7 +132,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: + async def get(self, location: str, gallery_unique_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. :param location: The name of Azure region. Required. @@ -138,6 +143,10 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ :rtype: ~azure.mgmt.compute.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_image_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_image_versions_operations.py index 4b3757f7afd7..7bfe0b094a37 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_image_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_image_versions_operations.py @@ -27,6 +27,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._shared_gallery_image_versions_operations import build_get_request, build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -61,7 +62,7 @@ def list( gallery_unique_name: str, gallery_image_name: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.SharedGalleryImageVersion"]: """List shared gallery image versions by subscription id or tenant id. @@ -81,6 +82,10 @@ def list( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.SharedGalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -147,7 +152,7 @@ async def get( gallery_unique_name: str, gallery_image_name: str, gallery_image_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SharedGalleryImageVersion: """Get a shared gallery image version by subscription id or tenant id. @@ -167,6 +172,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_images_operations.py index 8ef5eaeacc84..2d4a01d28dbe 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_shared_gallery_images_operations.py @@ -27,6 +27,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._shared_gallery_images_operations import build_get_request, build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -60,7 +61,7 @@ def list( location: str, gallery_unique_name: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.SharedGalleryImage"]: """List shared gallery images by subscription id or tenant id. @@ -75,6 +76,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -135,7 +140,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any + self, location: str, gallery_unique_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SharedGalleryImage: """Get a shared gallery image by subscription id or tenant id. @@ -150,6 +155,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_snapshots_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_snapshots_operations.py index 45ccd92b4008..a426db9803bc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_snapshots_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_snapshots_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._snapshots_operations import ( build_create_or_update_request, build_delete_request, @@ -70,13 +71,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Snapshot"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Snapshot"]: """Lists snapshots under a subscription. :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -133,7 +138,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.Snapshot"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Snapshot"]: """Lists snapshots under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -143,6 +148,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -200,7 +209,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: + async def get(self, resource_group_name: str, snapshot_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Snapshot: """Gets information about a snapshot. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -214,6 +223,10 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) :rtype: ~azure.mgmt.compute.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -331,7 +344,7 @@ async def begin_create_or_update( snapshot: _models.Snapshot, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -361,7 +374,7 @@ async def begin_create_or_update( snapshot: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -385,7 +398,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -404,6 +417,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -532,7 +549,7 @@ async def begin_update( snapshot: _models.SnapshotUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -563,7 +580,7 @@ async def begin_update( snapshot: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -592,7 +609,7 @@ async def begin_update( resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -611,6 +628,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -716,7 +737,7 @@ async def _delete_initial( return deserialized # type: ignore @distributed_trace_async - async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_delete(self, resource_group_name: str, snapshot_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """Deletes a snapshot. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -730,6 +751,10 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -851,7 +876,7 @@ async def begin_grant_access( grant_access_data: _models.GrantAccessData, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -882,7 +907,7 @@ async def begin_grant_access( grant_access_data: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -911,7 +936,7 @@ async def begin_grant_access( resource_group_name: str, snapshot_name: str, grant_access_data: Union[_models.GrantAccessData, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -930,6 +955,10 @@ async def begin_grant_access( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1036,7 +1065,7 @@ async def _revoke_access_initial( @distributed_trace_async async def begin_revoke_access( - self, resource_group_name: str, snapshot_name: str, **kwargs: Any + self, resource_group_name: str, snapshot_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Revokes access to a snapshot. @@ -1051,6 +1080,10 @@ async def begin_revoke_access( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_soft_deleted_resource_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_soft_deleted_resource_operations.py index 05a5ca85eae9..7babcbc4a965 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_soft_deleted_resource_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_soft_deleted_resource_operations.py @@ -26,6 +26,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._soft_deleted_resource_operations import build_list_by_artifact_name_request from .._configuration import ComputeManagementClientConfiguration @@ -55,7 +56,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_artifact_name( - self, resource_group_name: str, gallery_name: str, artifact_type: str, artifact_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, artifact_type: str, artifact_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.GallerySoftDeletedResource"]: """List soft-deleted resources of an artifact in the gallery, such as soft-deleted gallery image version of an image. @@ -77,6 +78,10 @@ def list_by_artifact_name( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.GallerySoftDeletedResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_ssh_public_keys_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_ssh_public_keys_operations.py index 1150e4e22cb6..1d9774359674 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_ssh_public_keys_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_ssh_public_keys_operations.py @@ -28,6 +28,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._ssh_public_keys_operations import ( build_create_request, build_delete_request, @@ -64,7 +65,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.SshPublicKeyResource"]: + def list_by_subscription(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.SshPublicKeyResource"]: """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. @@ -74,6 +75,10 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.SshPubl ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.SshPublicKeyResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -131,7 +136,7 @@ async def get_next(next_link=None): @distributed_trace def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any + self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.SshPublicKeyResource"]: """Lists all of the SSH public keys in the specified resource group. Use the nextLink property in the response to get the next page of SSH public keys. @@ -145,6 +150,10 @@ def list_by_resource_group( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.SshPublicKeyResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -203,7 +212,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get( - self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any + self, resource_group_name: str, ssh_public_key_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Retrieves information about an SSH public key. @@ -216,6 +225,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -266,7 +279,7 @@ async def create( parameters: _models.SshPublicKeyResource, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -293,7 +306,7 @@ async def create( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -318,7 +331,7 @@ async def create( resource_group_name: str, ssh_public_key_name: str, parameters: Union[_models.SshPublicKeyResource, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -334,6 +347,10 @@ async def create( :rtype: ~azure.mgmt.compute.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -396,7 +413,7 @@ async def update( parameters: _models.SshPublicKeyUpdateResource, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -423,7 +440,7 @@ async def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -448,7 +465,7 @@ async def update( resource_group_name: str, ssh_public_key_name: str, parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -464,6 +481,10 @@ async def update( :rtype: ~azure.mgmt.compute.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -519,7 +540,7 @@ async def update( return deserialized # type: ignore @distributed_trace_async - async def delete(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> None: + async def delete(self, resource_group_name: str, ssh_public_key_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Delete an SSH public key. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -531,6 +552,10 @@ async def delete(self, resource_group_name: str, ssh_public_key_name: str, **kwa :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -577,7 +602,7 @@ async def generate_key_pair( parameters: Optional[_models.SshGenerateKeyPairInputParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with the public key. The length of the key will be 3072 bits. This operation can only be performed @@ -606,7 +631,7 @@ async def generate_key_pair( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with the public key. The length of the key will be 3072 bits. This operation can only be performed @@ -633,7 +658,7 @@ async def generate_key_pair( resource_group_name: str, ssh_public_key_name: str, parameters: Optional[Union[_models.SshGenerateKeyPairInputParameters, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with the public key. The length of the key will be 3072 bits. This operation can only be performed @@ -651,6 +676,10 @@ async def generate_key_pair( :rtype: ~azure.mgmt.compute.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_usage_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_usage_operations.py index 6dbf95b15ed7..ac5b1c086c34 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_usage_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_usage_operations.py @@ -26,6 +26,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._usage_operations import build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -54,7 +55,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.Usage"]: + def list(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.Usage"]: """Gets, for the specified location, the current compute resource usage information as well as the limits for compute resources under the subscription. @@ -64,6 +65,10 @@ def list(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.Usage"]: :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extension_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extension_images_operations.py index 4efa690856e6..984dfac0aca4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extension_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extension_images_operations.py @@ -25,6 +25,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_extension_images_operations import ( build_get_request, build_list_types_request, @@ -58,7 +59,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace_async async def list_types( - self, location: str, publisher_name: str, **kwargs: Any + self, location: str, publisher_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineExtensionImage]: """Gets a list of virtual machine extension image types. @@ -70,6 +71,10 @@ async def list_types( :rtype: list[~azure.mgmt.compute.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -121,7 +126,7 @@ async def list_versions( filter: Optional[str] = None, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineExtensionImage]: """Gets a list of virtual machine extension image versions. @@ -141,6 +146,10 @@ async def list_versions( :rtype: list[~azure.mgmt.compute.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -189,7 +198,7 @@ async def list_versions( @distributed_trace_async async def get( - self, location: str, publisher_name: str, type: str, version: str, **kwargs: Any + self, location: str, publisher_name: str, type: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineExtensionImage: """Gets a virtual machine extension image. @@ -205,6 +214,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extensions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extensions_operations.py index 34c879fd6281..44349d060ad6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extensions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_extensions_operations.py @@ -30,6 +30,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_extensions_operations import ( build_create_or_update_request, build_delete_request, @@ -65,7 +66,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace_async async def list( - self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineExtensionsListResult: """The operation to get all extensions of a Virtual Machine. @@ -80,6 +81,10 @@ async def list( :rtype: ~azure.mgmt.compute.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -130,7 +135,7 @@ async def get( vm_name: str, vm_extension_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineExtension: """The operation to get the extension. @@ -147,6 +152,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -273,7 +282,7 @@ async def begin_create_or_update( extension_parameters: _models.VirtualMachineExtension, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -305,7 +314,7 @@ async def begin_create_or_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -335,7 +344,7 @@ async def begin_create_or_update( vm_name: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -354,6 +363,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -485,7 +498,7 @@ async def begin_update( extension_parameters: _models.VirtualMachineExtensionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -517,7 +530,7 @@ async def begin_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -547,7 +560,7 @@ async def begin_update( vm_name: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -567,6 +580,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -680,7 +697,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, vm_extension_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete the extension. @@ -695,6 +712,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_edge_zone_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_edge_zone_operations.py index c9bb9aa28aaa..52f797d07067 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_edge_zone_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_edge_zone_operations.py @@ -25,6 +25,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_images_edge_zone_operations import ( build_get_request, build_list_offers_request, @@ -60,7 +61,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace_async async def list_publishers( - self, location: str, edge_zone: str, **kwargs: Any + self, location: str, edge_zone: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image publishers for the specified Azure location and edge zone. @@ -72,6 +73,10 @@ async def list_publishers( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -116,7 +121,7 @@ async def list_publishers( @distributed_trace_async async def list_offers( - self, location: str, edge_zone: str, publisher_name: str, **kwargs: Any + self, location: str, edge_zone: str, publisher_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image offers for the specified location, edge zone and publisher. @@ -131,6 +136,10 @@ async def list_offers( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -176,7 +185,7 @@ async def list_offers( @distributed_trace_async async def list_skus( - self, location: str, edge_zone: str, publisher_name: str, offer: str, **kwargs: Any + self, location: str, edge_zone: str, publisher_name: str, offer: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image SKUs for the specified location, edge zone, publisher, and offer. @@ -193,6 +202,10 @@ async def list_skus( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -248,7 +261,7 @@ async def list( expand: Optional[str] = None, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of all virtual machine image versions for the specified location, edge zone, publisher, offer, and SKU. @@ -275,6 +288,10 @@ async def list( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -325,7 +342,7 @@ async def list( @distributed_trace_async async def get( - self, location: str, edge_zone: str, publisher_name: str, offer: str, skus: str, version: str, **kwargs: Any + self, location: str, edge_zone: str, publisher_name: str, offer: str, skus: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineImage: """Gets a virtual machine image in an edge zone. @@ -345,6 +362,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_operations.py index e29b4464cc90..0b5871f8a9ba 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_images_operations.py @@ -25,6 +25,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_images_operations import ( build_get_request, build_list_by_edge_zone_request, @@ -62,7 +63,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace_async async def list_by_edge_zone( - self, location: str, edge_zone: str, **kwargs: Any + self, location: str, edge_zone: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VmImagesInEdgeZoneListResult: """Gets a list of all virtual machine image versions for the specified edge zone. @@ -74,6 +75,10 @@ async def list_by_edge_zone( :rtype: ~azure.mgmt.compute.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -117,7 +122,7 @@ async def list_by_edge_zone( return deserialized # type: ignore @distributed_trace_async - async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: + async def list_publishers(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image publishers for the specified Azure location. :param location: The name of Azure region. Required. @@ -126,6 +131,10 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -169,7 +178,7 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi @distributed_trace_async async def list_offers( - self, location: str, publisher_name: str, **kwargs: Any + self, location: str, publisher_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image offers for the specified location and publisher. @@ -181,6 +190,10 @@ async def list_offers( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -225,7 +238,7 @@ async def list_offers( @distributed_trace_async async def list_skus( - self, location: str, publisher_name: str, offer: str, **kwargs: Any + self, location: str, publisher_name: str, offer: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image SKUs for the specified location, publisher, and offer. @@ -239,6 +252,10 @@ async def list_skus( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -292,7 +309,7 @@ async def list( expand: Optional[str] = None, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of all virtual machine image versions for the specified location, publisher, offer, and SKU. @@ -315,6 +332,10 @@ async def list( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -364,7 +385,7 @@ async def list( @distributed_trace_async async def get( - self, location: str, publisher_name: str, offer: str, skus: str, version: str, **kwargs: Any + self, location: str, publisher_name: str, offer: str, skus: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineImage: """Gets a virtual machine image. @@ -382,6 +403,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -437,7 +462,7 @@ async def list_with_properties( expand: str, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImage]: """list_with_properties. @@ -459,6 +484,10 @@ async def list_with_properties( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_run_commands_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_run_commands_operations.py index 71e38d7bb3b2..91dc8b50a14f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_run_commands_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_run_commands_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_run_commands_operations import ( build_create_or_update_request, build_delete_request, @@ -68,7 +69,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.RunCommandDocumentBase"]: + def list(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.RunCommandDocumentBase"]: """Lists all available run commands for a subscription in a location. :param location: The name of Azure region. Required. @@ -79,6 +80,10 @@ def list(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.RunComma ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.RunCommandDocumentBase] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -136,7 +141,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace_async - async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: + async def get(self, location: str, command_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. :param location: The name of Azure region. Required. @@ -149,6 +154,10 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :rtype: ~azure.mgmt.compute.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -193,7 +202,7 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru @distributed_trace def list_by_virtual_machine( - self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachineRunCommand"]: """The operation to get all run commands of a Virtual Machine. @@ -210,6 +219,10 @@ def list_by_virtual_machine( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -270,7 +283,7 @@ async def get_next(next_link=None): @distributed_trace_async async def get_by_virtual_machine( - self, resource_group_name: str, vm_name: str, run_command_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, run_command_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineRunCommand: """The operation to get the run command. @@ -287,6 +300,10 @@ async def get_by_virtual_machine( :rtype: ~azure.mgmt.compute.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -413,7 +430,7 @@ async def begin_create_or_update( run_command: _models.VirtualMachineRunCommand, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -445,7 +462,7 @@ async def begin_create_or_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -475,7 +492,7 @@ async def begin_create_or_update( vm_name: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -494,6 +511,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -625,7 +646,7 @@ async def begin_update( run_command: _models.VirtualMachineRunCommandUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -657,7 +678,7 @@ async def begin_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -687,7 +708,7 @@ async def begin_update( vm_name: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -706,6 +727,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -819,7 +844,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, run_command_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete the run command. @@ -834,6 +859,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_extensions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_extensions_operations.py index 4479d6eddff6..8676e784204f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_extensions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_extensions_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_scale_set_extensions_operations import ( build_create_or_update_request, build_delete_request, @@ -67,7 +68,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachineScaleSetExtension"]: """Gets a list of all extensions in a VM scale set. @@ -82,6 +83,10 @@ def list( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSetExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -146,7 +151,7 @@ async def get( vm_scale_set_name: str, vmss_extension_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: """The operation to get the extension. @@ -163,6 +168,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -289,7 +298,7 @@ async def begin_create_or_update( extension_parameters: _models.VirtualMachineScaleSetExtension, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -322,7 +331,7 @@ async def begin_create_or_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -353,7 +362,7 @@ async def begin_create_or_update( vm_scale_set_name: str, vmss_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -374,6 +383,10 @@ async def begin_create_or_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -506,7 +519,7 @@ async def begin_update( extension_parameters: _models.VirtualMachineScaleSetExtensionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -539,7 +552,7 @@ async def begin_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -570,7 +583,7 @@ async def begin_update( vm_scale_set_name: str, vmss_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -592,6 +605,10 @@ async def begin_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -700,7 +717,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete the extension. @@ -715,6 +732,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py index 26c843af12a7..77cf0eb8cf73 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py @@ -29,6 +29,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_scale_set_rolling_upgrades_operations import ( build_cancel_request, build_get_latest_request, @@ -118,7 +119,7 @@ async def _start_extension_upgrade_initial( @distributed_trace_async async def begin_start_extension_upgrade( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts a rolling upgrade to move all extensions for all virtual machine scale set instances to the latest available extension version. Instances which are already running the latest @@ -133,6 +134,10 @@ async def begin_start_extension_upgrade( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -232,7 +237,7 @@ async def _start_os_upgrade_initial( @distributed_trace_async async def begin_start_os_upgrade( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts a rolling upgrade to move all virtual machine scale set instances to the latest available Platform Image OS version. Instances which are already running the latest available @@ -247,6 +252,10 @@ async def begin_start_os_upgrade( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -346,7 +355,7 @@ async def _cancel_initial( @distributed_trace_async async def begin_cancel( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Cancels the current virtual machine scale set rolling upgrade. @@ -359,6 +368,10 @@ async def begin_cancel( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -403,7 +416,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace_async async def get_latest( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RollingUpgradeStatusInfo: """Gets the status of the latest virtual machine scale set rolling upgrade. @@ -416,6 +429,10 @@ async def get_latest( :rtype: ~azure.mgmt.compute.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_extensions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_extensions_operations.py index 0b3526fb03ee..3987ba285020 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_extensions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_extensions_operations.py @@ -30,6 +30,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_scale_set_vm_extensions_operations import ( build_create_or_update_request, build_delete_request, @@ -70,7 +71,7 @@ async def list( vm_scale_set_name: str, instance_id: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtensionsListResult: """The operation to get all extensions of an instance in Virtual Machine Scaleset. @@ -87,6 +88,10 @@ async def list( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -141,7 +146,7 @@ async def get( instance_id: str, vm_extension_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: """The operation to get the VMSS VM extension. @@ -160,6 +165,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -290,7 +299,7 @@ async def begin_create_or_update( extension_parameters: _models.VirtualMachineScaleSetVMExtension, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -326,7 +335,7 @@ async def begin_create_or_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -360,7 +369,7 @@ async def begin_create_or_update( instance_id: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -383,6 +392,10 @@ async def begin_create_or_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -518,7 +531,7 @@ async def begin_update( extension_parameters: _models.VirtualMachineScaleSetVMExtensionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -554,7 +567,7 @@ async def begin_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -588,7 +601,7 @@ async def begin_update( instance_id: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -612,6 +625,10 @@ async def begin_update( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -727,7 +744,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete the VMSS VM extension. @@ -744,6 +761,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_run_commands_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_run_commands_operations.py index 0e0ad3884a05..380165bf7401 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_run_commands_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vm_run_commands_operations.py @@ -32,6 +32,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_scale_set_vm_run_commands_operations import ( build_create_or_update_request, build_delete_request, @@ -72,7 +73,7 @@ def list( vm_scale_set_name: str, instance_id: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachineRunCommand"]: """The operation to get all run commands of an instance in Virtual Machine Scaleset. @@ -91,6 +92,10 @@ def list( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -158,7 +163,7 @@ async def get( instance_id: str, run_command_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineRunCommand: """The operation to get the VMSS VM run command. @@ -177,6 +182,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -307,7 +316,7 @@ async def begin_create_or_update( run_command: _models.VirtualMachineRunCommand, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -342,7 +351,7 @@ async def begin_create_or_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -375,7 +384,7 @@ async def begin_create_or_update( instance_id: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -396,6 +405,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -531,7 +544,7 @@ async def begin_update( run_command: _models.VirtualMachineRunCommandUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -565,7 +578,7 @@ async def begin_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -597,7 +610,7 @@ async def begin_update( instance_id: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -618,6 +631,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -733,7 +750,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete the VMSS VM run command. @@ -750,6 +767,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vms_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vms_operations.py index aa67e4809e3d..0865f21f90ac 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vms_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_set_vms_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_scale_set_vms_operations import ( build_approve_rolling_upgrade_request, build_attach_detach_data_disks_request, @@ -87,7 +88,7 @@ def list( filter: Optional[str] = None, select: Optional[str] = None, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachineScaleSetVM"]: """Gets a list of all virtual machines in a VM scale sets. @@ -112,6 +113,10 @@ def list( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSetVM] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -179,7 +184,7 @@ async def get( vm_scale_set_name: str, instance_id: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: """Gets a virtual machine from a VM scale set. @@ -199,6 +204,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -331,7 +340,7 @@ async def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -373,7 +382,7 @@ async def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -413,7 +422,7 @@ async def begin_update( parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -440,6 +449,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetVM] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -561,7 +574,7 @@ async def begin_delete( vm_scale_set_name: str, instance_id: str, force_deletion: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a virtual machine from a VM scale set. @@ -579,6 +592,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -680,7 +697,7 @@ async def _approve_rolling_upgrade_initial( @distributed_trace_async async def begin_approve_rolling_upgrade( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Approve upgrade on deferred rolling upgrade for OS disk on a VM scale set instance. @@ -695,6 +712,10 @@ async def begin_approve_rolling_upgrade( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -820,7 +841,7 @@ async def begin_attach_detach_data_disks( parameters: _models.AttachDetachDataDisksRequest, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -852,7 +873,7 @@ async def begin_attach_detach_data_disks( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -882,7 +903,7 @@ async def begin_attach_detach_data_disks( vm_scale_set_name: str, instance_id: str, parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -902,6 +923,10 @@ async def begin_attach_detach_data_disks( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.StorageProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1010,7 +1035,7 @@ async def _deallocate_initial( @distributed_trace_async async def begin_deallocate( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates a specific virtual machine in a VM scale set. Shuts down the virtual machine and releases the compute resources it uses. You are not billed for the compute resources of this @@ -1027,6 +1052,10 @@ async def begin_deallocate( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1072,7 +1101,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace_async async def get_instance_view( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVMInstanceView: """Gets the status of a virtual machine from a VM scale set. @@ -1087,6 +1116,10 @@ async def get_instance_view( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1188,7 +1221,7 @@ async def _perform_maintenance_initial( @distributed_trace_async async def begin_perform_maintenance( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Performs maintenance on a virtual machine in a VM scale set. @@ -1203,6 +1236,10 @@ async def begin_perform_maintenance( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1315,7 +1352,7 @@ async def begin_power_off( vm_scale_set_name: str, instance_id: str, skip_shutdown: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off (stop) a virtual machine in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release resources and @@ -1336,6 +1373,10 @@ async def begin_power_off( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1438,7 +1479,7 @@ async def _redeploy_initial( @distributed_trace_async async def begin_redeploy( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down the virtual machine in the virtual machine scale set, moves it to a new node, and powers it back on. @@ -1454,6 +1495,10 @@ async def begin_redeploy( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1585,7 +1630,7 @@ async def begin_reimage( vm_scale_set_vm_reimage_input: Optional[_models.VirtualMachineScaleSetVMReimageParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -1617,7 +1662,7 @@ async def begin_reimage( vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -1648,7 +1693,7 @@ async def begin_reimage( vm_scale_set_vm_reimage_input: Optional[ Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] ] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -1668,6 +1713,10 @@ async def begin_reimage( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1773,7 +1822,7 @@ async def _reimage_all_initial( @distributed_trace_async async def begin_reimage_all( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Allows you to re-image all the disks ( including data disks ) in the a VM scale set instance. This operation is only supported for managed disks. @@ -1789,6 +1838,10 @@ async def begin_reimage_all( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1890,7 +1943,7 @@ async def _restart_initial( @distributed_trace_async async def begin_restart( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts a virtual machine in a VM scale set. @@ -1905,6 +1958,10 @@ async def begin_restart( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1955,7 +2012,7 @@ async def retrieve_boot_diagnostics_data( vm_scale_set_name: str, instance_id: str, sas_uri_expiration_time_in_minutes: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RetrieveBootDiagnosticsDataResult: """The operation to retrieve SAS URIs of boot diagnostic logs for a virtual machine in a VM scale set. @@ -1975,6 +2032,10 @@ async def retrieve_boot_diagnostics_data( :rtype: ~azure.mgmt.compute.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2101,7 +2162,7 @@ async def begin_run_command( parameters: _models.RunCommandInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -2132,7 +2193,7 @@ async def begin_run_command( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -2161,7 +2222,7 @@ async def begin_run_command( vm_scale_set_name: str, instance_id: str, parameters: Union[_models.RunCommandInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -2180,6 +2241,10 @@ async def begin_run_command( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.RunCommandResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2232,7 +2297,7 @@ def get_long_running_output(pipeline_response): @distributed_trace_async async def simulate_eviction( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """The operation to simulate the eviction of spot virtual machine in a VM scale set. @@ -2247,6 +2312,10 @@ async def simulate_eviction( :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2344,7 +2413,7 @@ async def _start_initial( @distributed_trace_async async def begin_start( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts a virtual machine in a VM scale set. @@ -2359,6 +2428,10 @@ async def begin_start( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_sets_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_sets_operations.py index 78c9b2c00986..1844199a8828 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_sets_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_scale_sets_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_scale_sets_operations import ( build_approve_rolling_upgrade_request, build_convert_to_single_placement_group_request, @@ -88,7 +89,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_location(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineScaleSet"]: + def list_by_location(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineScaleSet"]: """Gets all the VM scale sets under the specified subscription for the specified location. :param location: The name of Azure region. Required. @@ -99,6 +100,10 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncItemPaged["_mod ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -156,7 +161,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list_all(self, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineScaleSet"]: + def list_all(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. @@ -167,6 +172,10 @@ def list_all(self, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineScale ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -223,7 +232,7 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineScaleSet"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineScaleSet"]: """Gets a list of all VM scale sets under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -235,6 +244,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_mode ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -297,7 +310,7 @@ async def get( resource_group_name: str, vm_scale_set_name: str, expand: Optional[Union[str, _models.ExpandTypesForGetVMScaleSets]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSet: """Display information about a virtual machine scale set. @@ -314,6 +327,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -442,7 +459,7 @@ async def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -480,7 +497,7 @@ async def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -516,7 +533,7 @@ async def begin_create_or_update( parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -541,6 +558,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -676,7 +697,7 @@ async def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -714,7 +735,7 @@ async def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -750,7 +771,7 @@ async def begin_update( parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -775,6 +796,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -889,7 +914,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, force_deletion: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, force_deletion: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a VM scale set. @@ -905,6 +930,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1030,7 +1059,7 @@ async def begin_approve_rolling_upgrade( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale set. @@ -1059,7 +1088,7 @@ async def begin_approve_rolling_upgrade( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale set. @@ -1086,7 +1115,7 @@ async def begin_approve_rolling_upgrade( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale set. @@ -1104,6 +1133,10 @@ async def begin_approve_rolling_upgrade( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1158,7 +1191,7 @@ async def convert_to_single_placement_group( parameters: _models.VMScaleSetConvertToSinglePlacementGroupInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -1185,7 +1218,7 @@ async def convert_to_single_placement_group( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -1210,7 +1243,7 @@ async def convert_to_single_placement_group( resource_group_name: str, vm_scale_set_name: str, parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -1227,6 +1260,10 @@ async def convert_to_single_placement_group( :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1363,7 +1400,7 @@ async def begin_deallocate( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and releases the compute resources. You are not billed for the compute resources that this virtual @@ -1398,7 +1435,7 @@ async def begin_deallocate( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and releases the compute resources. You are not billed for the compute resources that this virtual @@ -1431,7 +1468,7 @@ async def begin_deallocate( vm_scale_set_name: str, hibernate: Optional[bool] = None, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and releases the compute resources. You are not billed for the compute resources that this virtual @@ -1454,6 +1491,10 @@ async def begin_deallocate( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1583,7 +1624,7 @@ async def begin_delete_instances( force_deletion: Optional[bool] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -1615,7 +1656,7 @@ async def begin_delete_instances( force_deletion: Optional[bool] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -1645,7 +1686,7 @@ async def begin_delete_instances( vm_scale_set_name: str, vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], force_deletion: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -1665,6 +1706,10 @@ async def begin_delete_instances( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1719,7 +1764,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: platform_update_domain: int, zone: Optional[str] = None, placement_group_id: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual machine scale set. @@ -1742,6 +1787,10 @@ async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: :rtype: ~azure.mgmt.compute.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1789,7 +1838,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: @distributed_trace_async async def get_instance_view( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetInstanceView: """Gets the status of a VM scale set instance. @@ -1802,6 +1851,10 @@ async def get_instance_view( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1923,7 +1976,7 @@ async def begin_update_instances( vm_instance_i_ds: _models.VirtualMachineScaleSetVMInstanceRequiredIDs, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -1951,7 +2004,7 @@ async def begin_update_instances( vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -1977,7 +2030,7 @@ async def begin_update_instances( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -1994,6 +2047,10 @@ async def begin_update_instances( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2041,7 +2098,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def get_os_upgrade_history( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.UpgradeOperationHistoricalStatusInfo"]: """Gets list of OS upgrades on a VM scale set instance. @@ -2056,6 +2113,10 @@ def get_os_upgrade_history( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.UpgradeOperationHistoricalStatusInfo] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2196,7 +2257,7 @@ async def begin_perform_maintenance( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances which are not eligible for perform maintenance will be failed. Please refer to best practices @@ -2227,7 +2288,7 @@ async def begin_perform_maintenance( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances which are not eligible for perform maintenance will be failed. Please refer to best practices @@ -2256,7 +2317,7 @@ async def begin_perform_maintenance( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances which are not eligible for perform maintenance will be failed. Please refer to best practices @@ -2276,6 +2337,10 @@ async def begin_perform_maintenance( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2408,7 +2473,7 @@ async def begin_power_off( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release @@ -2443,7 +2508,7 @@ async def begin_power_off( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release @@ -2476,7 +2541,7 @@ async def begin_power_off( vm_scale_set_name: str, skip_shutdown: Optional[bool] = None, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release @@ -2499,6 +2564,10 @@ async def begin_power_off( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2603,7 +2672,7 @@ async def _reapply_initial( @distributed_trace_async async def begin_reapply( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reapplies the Virtual Machine Scale Set Virtual Machine Profile to the Virtual Machine Instances. @@ -2617,6 +2686,10 @@ async def begin_reapply( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2742,7 +2815,7 @@ async def begin_redeploy( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, and powers them back on. @@ -2771,7 +2844,7 @@ async def begin_redeploy( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, and powers them back on. @@ -2798,7 +2871,7 @@ async def begin_redeploy( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, and powers them back on. @@ -2816,6 +2889,10 @@ async def begin_redeploy( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2945,7 +3022,7 @@ async def begin_reimage( vm_scale_set_reimage_input: Optional[_models.VirtualMachineScaleSetReimageParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual @@ -2975,7 +3052,7 @@ async def begin_reimage( vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual @@ -3002,7 +3079,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual @@ -3021,6 +3098,10 @@ async def begin_reimage( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3150,7 +3231,7 @@ async def begin_reimage_all( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This operation is only supported for managed disks. @@ -3179,7 +3260,7 @@ async def begin_reimage_all( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This operation is only supported for managed disks. @@ -3206,7 +3287,7 @@ async def begin_reimage_all( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This operation is only supported for managed disks. @@ -3224,6 +3305,10 @@ async def begin_reimage_all( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3353,7 +3438,7 @@ async def begin_restart( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -3381,7 +3466,7 @@ async def begin_restart( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -3407,7 +3492,7 @@ async def begin_restart( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -3424,6 +3509,10 @@ async def begin_restart( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3549,7 +3638,7 @@ async def begin_scale_out( parameters: _models.VMScaleSetScaleOutInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Scales out one or more virtual machines in a VM scale set. @@ -3576,7 +3665,7 @@ async def begin_scale_out( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Scales out one or more virtual machines in a VM scale set. @@ -3601,7 +3690,7 @@ async def begin_scale_out( resource_group_name: str, vm_scale_set_name: str, parameters: Union[_models.VMScaleSetScaleOutInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Scales out one or more virtual machines in a VM scale set. @@ -3617,6 +3706,10 @@ async def begin_scale_out( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3741,7 +3834,7 @@ async def begin_set_orchestration_service_state( parameters: _models.OrchestrationServiceStateInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -3768,7 +3861,7 @@ async def begin_set_orchestration_service_state( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -3793,7 +3886,7 @@ async def begin_set_orchestration_service_state( resource_group_name: str, vm_scale_set_name: str, parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -3809,6 +3902,10 @@ async def begin_set_orchestration_service_state( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3856,7 +3953,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_skus( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachineScaleSetSku"]: """Gets a list of SKUs available for your VM scale set, including the minimum and maximum VM instances allowed for each SKU. @@ -3872,6 +3969,10 @@ def list_skus( ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSetSku] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4012,7 +4113,7 @@ async def begin_start( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -4040,7 +4141,7 @@ async def begin_start( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -4066,7 +4167,7 @@ async def begin_start( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -4083,6 +4184,10 @@ async def begin_start( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_sizes_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_sizes_operations.py index 3bfe04e07bd3..f0569e06bdba 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_sizes_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machine_sizes_operations.py @@ -26,6 +26,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machine_sizes_operations import build_list_request from .._configuration import ComputeManagementClientConfiguration @@ -54,7 +55,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineSize"]: + def list(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachineSize"]: """This API is deprecated. Use `Resources Skus `_. @@ -64,6 +65,10 @@ def list(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.VirtualM :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machines_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machines_operations.py index cf66998a4be7..b6a4e5afa95a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machines_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/operations/_virtual_machines_operations.py @@ -33,6 +33,7 @@ from ... import models as _models from ..._utils.serialization import Deserializer, Serializer +from ...types import LRORequestParameters, RequestParameters from ...operations._virtual_machines_operations import ( build_assess_patches_request, build_attach_detach_data_disks_request, @@ -89,7 +90,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_location(self, location: str, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachine"]: + def list_by_location(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> AsyncItemPaged["_models.VirtualMachine"]: """Gets all the virtual machines under the specified subscription for the specified location. :param location: The name of Azure region. Required. @@ -98,6 +99,10 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncItemPaged["_mod :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -160,7 +165,7 @@ def list_all( status_only: Optional[str] = None, filter: Optional[str] = None, expand: Optional[Union[str, _models.ExpandTypesForListVMs]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. @@ -181,6 +186,10 @@ def list_all( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -245,7 +254,7 @@ def list( resource_group_name: str, filter: Optional[str] = None, expand: Optional[Union[str, _models.ExpandTypeForListVMs]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified resource group. Use the nextLink property in the response to get the next page of virtual machines. @@ -266,6 +275,10 @@ def list( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -330,7 +343,7 @@ async def get( resource_group_name: str, vm_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachine: """Retrieves information about the model view or the instance view of a virtual machine. @@ -349,6 +362,10 @@ async def get( :rtype: ~azure.mgmt.compute.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -477,7 +494,7 @@ async def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -516,7 +533,7 @@ async def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -553,7 +570,7 @@ async def begin_create_or_update( parameters: Union[_models.VirtualMachine, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -579,6 +596,10 @@ async def begin_create_or_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -714,7 +735,7 @@ async def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -752,7 +773,7 @@ async def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -788,7 +809,7 @@ async def begin_update( parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -813,6 +834,10 @@ async def begin_update( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -927,7 +952,7 @@ async def _delete_initial( @distributed_trace_async async def begin_delete( - self, resource_group_name: str, vm_name: str, force_deletion: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, force_deletion: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to delete a virtual machine. @@ -943,6 +968,10 @@ async def begin_delete( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1043,7 +1072,7 @@ async def _assess_patches_initial( @distributed_trace_async async def begin_assess_patches( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]: """Assess patches on the VM. @@ -1058,6 +1087,10 @@ async def begin_assess_patches( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineAssessPatchesResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1183,7 +1216,7 @@ async def begin_attach_detach_data_disks( parameters: _models.AttachDetachDataDisksRequest, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -1212,7 +1245,7 @@ async def begin_attach_detach_data_disks( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -1239,7 +1272,7 @@ async def begin_attach_detach_data_disks( resource_group_name: str, vm_name: str, parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -1256,6 +1289,10 @@ async def begin_attach_detach_data_disks( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.StorageProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1384,7 +1421,7 @@ async def begin_capture( parameters: _models.VirtualMachineCaptureParameters, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used to create similar VMs. @@ -1414,7 +1451,7 @@ async def begin_capture( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used to create similar VMs. @@ -1442,7 +1479,7 @@ async def begin_capture( resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used to create similar VMs. @@ -1461,6 +1498,10 @@ async def begin_capture( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineCaptureResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1567,7 +1608,7 @@ async def _convert_to_managed_disks_initial( @distributed_trace_async async def begin_convert_to_managed_disks( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Converts virtual machine disks from blob-based to managed disks. Virtual machine must be stop-deallocated before invoking this operation. @@ -1581,6 +1622,10 @@ async def begin_convert_to_managed_disks( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1681,7 +1726,7 @@ async def _deallocate_initial( @distributed_trace_async async def begin_deallocate( - self, resource_group_name: str, vm_name: str, hibernate: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, hibernate: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down the virtual machine and releases the compute resources. You are not billed for the compute resources that this virtual machine uses. @@ -1697,6 +1742,10 @@ async def begin_deallocate( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1741,7 +1790,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> None: + async def generalize(self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """Sets the OS state of the virtual machine to generalized. It is recommended to sysprep the virtual machine before performing this operation. For Windows, please refer to `Create a managed image of a generalized VM in Azure @@ -1758,6 +1807,10 @@ async def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1875,7 +1928,7 @@ async def begin_install_patches( install_patches_input: _models.VirtualMachineInstallPatchesParameters, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -1905,7 +1958,7 @@ async def begin_install_patches( install_patches_input: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -1933,7 +1986,7 @@ async def begin_install_patches( resource_group_name: str, vm_name: str, install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -1952,6 +2005,10 @@ async def begin_install_patches( ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.VirtualMachineInstallPatchesResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2003,7 +2060,7 @@ def get_long_running_output(pipeline_response): @distributed_trace_async async def instance_view( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineInstanceView: """Retrieves information about the run-time state of a virtual machine. @@ -2016,6 +2073,10 @@ async def instance_view( :rtype: ~azure.mgmt.compute.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2140,7 +2201,7 @@ async def begin_migrate_to_vm_scale_set( parameters: Optional[_models.MigrateVMToVirtualMachineScaleSetInput] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Migrate a virtual machine from availability set to Flexible Virtual Machine Scale Set. @@ -2168,7 +2229,7 @@ async def begin_migrate_to_vm_scale_set( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Migrate a virtual machine from availability set to Flexible Virtual Machine Scale Set. @@ -2194,7 +2255,7 @@ async def begin_migrate_to_vm_scale_set( resource_group_name: str, vm_name: str, parameters: Optional[Union[_models.MigrateVMToVirtualMachineScaleSetInput, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Migrate a virtual machine from availability set to Flexible Virtual Machine Scale Set. @@ -2211,6 +2272,10 @@ async def begin_migrate_to_vm_scale_set( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2314,7 +2379,7 @@ async def _perform_maintenance_initial( @distributed_trace_async async def begin_perform_maintenance( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to perform maintenance on a virtual machine. @@ -2327,6 +2392,10 @@ async def begin_perform_maintenance( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2427,7 +2496,7 @@ async def _power_off_initial( @distributed_trace_async async def begin_power_off( - self, resource_group_name: str, vm_name: str, skip_shutdown: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, skip_shutdown: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """The operation to power off (stop) a virtual machine. The virtual machine can be restarted with the same provisioned resources. You are still charged for this virtual machine. @@ -2445,6 +2514,10 @@ async def begin_power_off( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2542,7 +2615,7 @@ async def _reapply_initial(self, resource_group_name: str, vm_name: str, **kwarg return deserialized # type: ignore @distributed_trace_async - async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_reapply(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """The operation to reapply a virtual machine's state. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -2554,6 +2627,10 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2650,7 +2727,7 @@ async def _redeploy_initial(self, resource_group_name: str, vm_name: str, **kwar return deserialized # type: ignore @distributed_trace_async - async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_redeploy(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """Shuts down the virtual machine, moves it to a new node, and powers it back on. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -2662,6 +2739,10 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2787,7 +2868,7 @@ async def begin_reimage( parameters: Optional[_models.VirtualMachineReimageParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual machine is reset to initial @@ -2820,7 +2901,7 @@ async def begin_reimage( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual machine is reset to initial @@ -2851,7 +2932,7 @@ async def begin_reimage( resource_group_name: str, vm_name: str, parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual machine is reset to initial @@ -2872,6 +2953,10 @@ async def begin_reimage( :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2972,7 +3057,7 @@ async def _restart_initial(self, resource_group_name: str, vm_name: str, **kwarg return deserialized # type: ignore @distributed_trace_async - async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_restart(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """The operation to restart a virtual machine. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -2984,6 +3069,10 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3032,7 +3121,7 @@ async def retrieve_boot_diagnostics_data( resource_group_name: str, vm_name: str, sas_uri_expiration_time_in_minutes: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RetrieveBootDiagnosticsDataResult: """The operation to retrieve SAS URIs for a virtual machine's boot diagnostic logs. @@ -3049,6 +3138,10 @@ async def retrieve_boot_diagnostics_data( :rtype: ~azure.mgmt.compute.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3171,7 +3264,7 @@ async def begin_run_command( parameters: _models.RunCommandInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -3199,7 +3292,7 @@ async def begin_run_command( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -3225,7 +3318,7 @@ async def begin_run_command( resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -3242,6 +3335,10 @@ async def begin_run_command( :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.models.RunCommandResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3292,7 +3389,7 @@ def get_long_running_output(pipeline_response): ) @distributed_trace_async - async def simulate_eviction(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> None: + async def simulate_eviction(self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> None: """The operation to simulate the eviction of spot virtual machine. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -3304,6 +3401,10 @@ async def simulate_eviction(self, resource_group_name: str, vm_name: str, **kwar :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3396,7 +3497,7 @@ async def _start_initial(self, resource_group_name: str, vm_name: str, **kwargs: return deserialized # type: ignore @distributed_trace_async - async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + async def begin_start(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> AsyncLROPoller[None]: """The operation to start a virtual machine. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -3408,6 +3509,10 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3452,7 +3557,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_available_sizes( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> AsyncItemPaged["_models.VirtualMachineSize"]: """Lists all available virtual machine sizes to which the specified virtual machine can be resized. @@ -3466,6 +3571,10 @@ def list_available_sizes( :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_availability_sets_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_availability_sets_operations.py index 351cebb3dc99..40cf4451d9b7 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_availability_sets_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_availability_sets_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -432,7 +433,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_subscription(self, expand: Optional[str] = None, **kwargs: Any) -> ItemPaged["_models.AvailabilitySet"]: + def list_by_subscription(self, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.AvailabilitySet"]: """Lists all availability sets in a subscription. :param expand: The expand expression to apply to the operation. Allowed values are @@ -442,6 +443,10 @@ def list_by_subscription(self, expand: Optional[str] = None, **kwargs: Any) -> I :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -499,7 +504,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.AvailabilitySet"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -509,6 +514,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Av :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -566,7 +575,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: + def get(self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.AvailabilitySet: """Retrieves information about an availability set. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -578,6 +587,10 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :rtype: ~azure.mgmt.compute.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -628,7 +641,7 @@ def create_or_update( parameters: _models.AvailabilitySet, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -655,7 +668,7 @@ def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -680,7 +693,7 @@ def create_or_update( resource_group_name: str, availability_set_name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -696,6 +709,10 @@ def create_or_update( :rtype: ~azure.mgmt.compute.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -758,7 +775,7 @@ def update( parameters: _models.AvailabilitySetUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -785,7 +802,7 @@ def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -810,7 +827,7 @@ def update( resource_group_name: str, availability_set_name: str, parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -826,6 +843,10 @@ def update( :rtype: ~azure.mgmt.compute.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -882,7 +903,7 @@ def update( @distributed_trace def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, availability_set_name: str, **kwargs: Any + self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Delete an availability set. @@ -895,6 +916,10 @@ def delete( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -935,7 +960,7 @@ def delete( # pylint: disable=inconsistent-return-statements @distributed_trace def cancel_migration_to_virtual_machine_scale_set( # pylint: disable=inconsistent-return-statements,name-too-long - self, resource_group_name: str, availability_set_name: str, **kwargs: Any + self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Cancel the migration operation on an Availability Set. @@ -948,6 +973,10 @@ def cancel_migration_to_virtual_machine_scale_set( # pylint: disable=inconsiste :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1068,7 +1097,7 @@ def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-too-long parameters: Optional[_models.ConvertToVirtualMachineScaleSetInput] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Create a new Flexible Virtual Machine Scale Set and migrate all the Virtual Machines in the Availability Set. This does not trigger a downtime on the Virtual Machines. @@ -1097,7 +1126,7 @@ def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-too-long parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Create a new Flexible Virtual Machine Scale Set and migrate all the Virtual Machines in the Availability Set. This does not trigger a downtime on the Virtual Machines. @@ -1124,7 +1153,7 @@ def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, parameters: Optional[Union[_models.ConvertToVirtualMachineScaleSetInput, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Create a new Flexible Virtual Machine Scale Set and migrate all the Virtual Machines in the Availability Set. This does not trigger a downtime on the Virtual Machines. @@ -1141,6 +1170,10 @@ def begin_convert_to_virtual_machine_scale_set( # pylint: disable=name-too-long :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1195,7 +1228,7 @@ def start_migration_to_virtual_machine_scale_set( # pylint: disable=name-too-lo parameters: _models.MigrateToVirtualMachineScaleSetInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Start migration operation on an Availability Set to move its Virtual Machines to a Virtual Machine Scale Set. This should be followed by a migrate operation on each Virtual Machine that @@ -1225,7 +1258,7 @@ def start_migration_to_virtual_machine_scale_set( # pylint: disable=name-too-lo parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Start migration operation on an Availability Set to move its Virtual Machines to a Virtual Machine Scale Set. This should be followed by a migrate operation on each Virtual Machine that @@ -1253,7 +1286,7 @@ def start_migration_to_virtual_machine_scale_set( # pylint: disable=inconsisten resource_group_name: str, availability_set_name: str, parameters: Union[_models.MigrateToVirtualMachineScaleSetInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Start migration operation on an Availability Set to move its Virtual Machines to a Virtual Machine Scale Set. This should be followed by a migrate operation on each Virtual Machine that @@ -1271,6 +1304,10 @@ def start_migration_to_virtual_machine_scale_set( # pylint: disable=inconsisten :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1329,7 +1366,7 @@ def validate_migration_to_virtual_machine_scale_set( # pylint: disable=name-too parameters: _models.MigrateToVirtualMachineScaleSetInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Validates that the Virtual Machines in the Availability Set can be migrated to the provided Virtual Machine Scale Set. @@ -1358,7 +1395,7 @@ def validate_migration_to_virtual_machine_scale_set( # pylint: disable=name-too parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Validates that the Virtual Machines in the Availability Set can be migrated to the provided Virtual Machine Scale Set. @@ -1385,7 +1422,7 @@ def validate_migration_to_virtual_machine_scale_set( # pylint: disable=inconsis resource_group_name: str, availability_set_name: str, parameters: Union[_models.MigrateToVirtualMachineScaleSetInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Validates that the Virtual Machines in the Availability Set can be migrated to the provided Virtual Machine Scale Set. @@ -1402,6 +1439,10 @@ def validate_migration_to_virtual_machine_scale_set( # pylint: disable=inconsis :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1454,7 +1495,7 @@ def validate_migration_to_virtual_machine_scale_set( # pylint: disable=inconsis @distributed_trace def list_available_sizes( - self, resource_group_name: str, availability_set_name: str, **kwargs: Any + self, resource_group_name: str, availability_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachineSize"]: """Lists all available virtual machine sizes that can be used to create a new virtual machine in an existing availability set. @@ -1468,6 +1509,10 @@ def list_available_sizes( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservation_groups_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservation_groups_operations.py index 5c7bd968a7af..a33ba1f94aa6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservation_groups_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservation_groups_operations.py @@ -29,6 +29,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -289,7 +290,7 @@ def list_by_subscription( self, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, resource_ids_only: Optional[Union[str, _models.ResourceIdOptionsForGetCapacityReservationGroups]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.CapacityReservationGroup"]: """Lists all of the capacity reservation groups in the subscription. Use the nextLink property in the response to get the next page of capacity reservation groups. @@ -313,6 +314,10 @@ def list_by_subscription( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CapacityReservationGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -375,7 +380,7 @@ def list_by_resource_group( self, resource_group_name: str, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.CapacityReservationGroup"]: """Lists all of the capacity reservation groups in the specified resource group. Use the nextLink property in the response to get the next page of capacity reservation groups. @@ -393,6 +398,10 @@ def list_by_resource_group( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CapacityReservationGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -456,7 +465,7 @@ def get( resource_group_name: str, capacity_reservation_group_name: str, expand: Optional[Union[str, _models.CapacityReservationGroupInstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation that retrieves information about a capacity reservation group. @@ -475,6 +484,10 @@ def get( :rtype: ~azure.mgmt.compute.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -526,7 +539,7 @@ def create_or_update( parameters: _models.CapacityReservationGroup, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. Please refer to @@ -555,7 +568,7 @@ def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. Please refer to @@ -582,7 +595,7 @@ def create_or_update( resource_group_name: str, capacity_reservation_group_name: str, parameters: Union[_models.CapacityReservationGroup, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. Please refer to @@ -600,6 +613,10 @@ def create_or_update( :rtype: ~azure.mgmt.compute.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -662,7 +679,7 @@ def update( parameters: _models.CapacityReservationGroupUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. @@ -691,7 +708,7 @@ def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. @@ -718,7 +735,7 @@ def update( resource_group_name: str, capacity_reservation_group_name: str, parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation group, only tags and sharing profile may be modified. @@ -735,6 +752,10 @@ def update( :rtype: ~azure.mgmt.compute.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -791,7 +812,7 @@ def update( @distributed_trace def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, capacity_reservation_group_name: str, **kwargs: Any + self, resource_group_name: str, capacity_reservation_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """The operation to delete a capacity reservation group. This operation is allowed only if all the associated resources are disassociated from the reservation group and all capacity reservations @@ -807,6 +828,10 @@ def delete( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservations_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservations_operations.py index 6fb46647f82d..5ffbac91cbb5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservations_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_capacity_reservations_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -268,7 +269,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_capacity_reservation_group( - self, resource_group_name: str, capacity_reservation_group_name: str, **kwargs: Any + self, resource_group_name: str, capacity_reservation_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.CapacityReservation"]: """Lists all of the capacity reservations in the specified capacity reservation group. Use the nextLink property in the response to get the next page of capacity reservations. @@ -282,6 +283,10 @@ def list_by_capacity_reservation_group( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CapacityReservation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -346,7 +351,7 @@ def get( capacity_reservation_group_name: str, capacity_reservation_name: str, expand: Optional[Union[str, _models.CapacityReservationInstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CapacityReservation: """The operation that retrieves information about the capacity reservation. @@ -365,6 +370,10 @@ def get( :rtype: ~azure.mgmt.compute.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -491,7 +500,7 @@ def begin_create_or_update( parameters: _models.CapacityReservation, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be set only during capacity reservation creation. Please refer to @@ -524,7 +533,7 @@ def begin_create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be set only during capacity reservation creation. Please refer to @@ -555,7 +564,7 @@ def begin_create_or_update( capacity_reservation_group_name: str, capacity_reservation_name: str, parameters: Union[_models.CapacityReservation, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be set only during capacity reservation creation. Please refer to @@ -576,6 +585,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.CapacityReservation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -708,7 +721,7 @@ def begin_update( parameters: _models.CapacityReservationUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -739,7 +752,7 @@ def begin_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -768,7 +781,7 @@ def begin_update( capacity_reservation_group_name: str, capacity_reservation_name: str, parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -787,6 +800,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.CapacityReservation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -903,7 +920,7 @@ def begin_delete( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete a capacity reservation. This operation is allowed only when all the associated resources are disassociated from the capacity reservation. Please refer to @@ -921,6 +938,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_operating_systems_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_operating_systems_operations.py index d2b1aba2d047..0c30c0c7586d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_operating_systems_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_operating_systems_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -173,7 +174,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> _models.OSVersion: + def get_os_version(self, location: str, os_version_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.OSVersion: """Gets properties of a guest operating system version that can be specified in the XML service configuration (.cscfg) for a cloud service. @@ -185,6 +186,10 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> :rtype: ~azure.mgmt.compute.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -228,7 +233,7 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> return deserialized # type: ignore @distributed_trace - def list_os_versions(self, location: str, **kwargs: Any) -> ItemPaged["_models.OSVersion"]: + def list_os_versions(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.OSVersion"]: """Gets a list of all guest operating system versions available to be specified in the XML service configuration (.cscfg) for a cloud service. Use nextLink property in the response to get the next page of OS versions. Do this till nextLink is null to fetch all the OS versions. @@ -239,6 +244,10 @@ def list_os_versions(self, location: str, **kwargs: Any) -> ItemPaged["_models.O :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.OSVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -296,7 +305,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: + def get_os_family(self, location: str, os_family_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service configuration (.cscfg) for a cloud service. @@ -308,6 +317,10 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m :rtype: ~azure.mgmt.compute.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -351,7 +364,7 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m return deserialized # type: ignore @distributed_trace - def list_os_families(self, location: str, **kwargs: Any) -> ItemPaged["_models.OSFamily"]: + def list_os_families(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.OSFamily"]: """Gets a list of all guest operating system families available to be specified in the XML service configuration (.cscfg) for a cloud service. Use nextLink property in the response to get the next page of OS Families. Do this till nextLink is null to fetch all the OS Families. @@ -362,6 +375,10 @@ def list_os_families(self, location: str, **kwargs: Any) -> ItemPaged["_models.O :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.OSFamily] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_role_instances_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_role_instances_operations.py index eb1f7c67c43a..96dce5cab591 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_role_instances_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_role_instances_operations.py @@ -32,6 +32,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -383,7 +384,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes a role instance from a cloud service. @@ -397,6 +398,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -445,7 +450,7 @@ def get( resource_group_name: str, cloud_service_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RoleInstance: """Gets a role instance from a cloud service. @@ -463,6 +468,10 @@ def get( :rtype: ~azure.mgmt.compute.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -509,7 +518,7 @@ def get( @distributed_trace def get_instance_view( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RoleInstanceView: """Retrieves information about the run-time state of a role instance in a cloud service. @@ -523,6 +532,10 @@ def get_instance_view( :rtype: ~azure.mgmt.compute.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -572,7 +585,7 @@ def list( resource_group_name: str, cloud_service_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.RoleInstance"]: """Gets the list of all role instances in a cloud service. Use nextLink property in the response to get the next page of role instances. Do this till nextLink is null to fetch all the role @@ -590,6 +603,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.RoleInstance] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -701,7 +718,7 @@ def _restart_initial( @distributed_trace def begin_restart( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The Reboot Role Instance asynchronous operation requests a reboot of a role instance in the cloud service. @@ -716,6 +733,10 @@ def begin_restart( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -810,7 +831,7 @@ def _reimage_initial( @distributed_trace def begin_reimage( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The Reimage Role Instance asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -825,6 +846,10 @@ def begin_reimage( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -919,7 +944,7 @@ def _rebuild_initial( @distributed_trace def begin_rebuild( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The Rebuild Role Instance asynchronous operation reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If @@ -935,6 +960,10 @@ def begin_rebuild( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -978,7 +1007,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def get_remote_desktop_file( - self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> Iterator[bytes]: """Gets a remote desktop file for a role instance in a cloud service. @@ -992,6 +1021,10 @@ def get_remote_desktop_file( :rtype: Iterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_roles_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_roles_operations.py index 0ea127486dca..9f27536699b9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_roles_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_service_roles_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -121,7 +122,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def get( - self, role_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, role_name: str, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CloudServiceRole: """Gets a role from a cloud service. @@ -135,6 +136,10 @@ def get( :rtype: ~azure.mgmt.compute.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -180,7 +185,7 @@ def get( @distributed_trace def list( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.CloudServiceRole"]: """Gets a list of all roles in a cloud service. Use nextLink property in the response to get the next page of roles. Do this till nextLink is null to fetch all the roles. @@ -193,6 +198,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CloudServiceRole] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_operations.py index 8fb653b359b7..2baa65a9afa1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -550,7 +551,7 @@ def begin_create_or_update( parameters: Optional[_models.CloudService] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud service creation. @@ -578,7 +579,7 @@ def begin_create_or_update( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud service creation. @@ -604,7 +605,7 @@ def begin_create_or_update( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud service creation. @@ -621,6 +622,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -747,7 +752,7 @@ def begin_update( parameters: Optional[_models.CloudServiceUpdate] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Update a cloud service. @@ -774,7 +779,7 @@ def begin_update( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Update a cloud service. @@ -799,7 +804,7 @@ def begin_update( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Update a cloud service. @@ -815,6 +820,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -912,7 +921,7 @@ def _delete_initial(self, resource_group_name: str, cloud_service_name: str, **k return deserialized # type: ignore @distributed_trace - def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_delete(self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Deletes a cloud service. :param resource_group_name: Name of the resource group. Required. @@ -923,6 +932,10 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -964,7 +977,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: + def get(self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.CloudService: """Display information about a cloud service. :param resource_group_name: Name of the resource group. Required. @@ -975,6 +988,10 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) :rtype: ~azure.mgmt.compute.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1019,7 +1036,7 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) @distributed_trace def get_instance_view( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CloudServiceInstanceView: """Gets the status of a cloud service. @@ -1031,6 +1048,10 @@ def get_instance_view( :rtype: ~azure.mgmt.compute.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1074,7 +1095,7 @@ def get_instance_view( return deserialized # type: ignore @distributed_trace - def list_all(self, **kwargs: Any) -> ItemPaged["_models.CloudService"]: + def list_all(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.CloudService"]: """Gets a list of all cloud services in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. @@ -1083,6 +1104,10 @@ def list_all(self, **kwargs: Any) -> ItemPaged["_models.CloudService"]: :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1139,7 +1164,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.CloudService"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. @@ -1150,6 +1175,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Cl :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1255,7 +1284,7 @@ def _start_initial(self, resource_group_name: str, cloud_service_name: str, **kw return deserialized # type: ignore @distributed_trace - def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_start(self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Starts the cloud service. :param resource_group_name: Name of the resource group. Required. @@ -1266,6 +1295,10 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1355,7 +1388,7 @@ def _power_off_initial(self, resource_group_name: str, cloud_service_name: str, return deserialized # type: ignore @distributed_trace - def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_power_off(self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Power off the cloud service. Note that resources are still attached and you are getting charged for the resources. @@ -1367,6 +1400,10 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1485,7 +1522,7 @@ def begin_restart( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -1511,7 +1548,7 @@ def begin_restart( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -1535,7 +1572,7 @@ def begin_restart( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -1550,6 +1587,10 @@ def begin_restart( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1672,7 +1713,7 @@ def begin_reimage( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -1699,7 +1740,7 @@ def begin_reimage( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -1724,7 +1765,7 @@ def begin_reimage( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or worker roles. @@ -1740,6 +1781,10 @@ def begin_reimage( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1862,7 +1907,7 @@ def begin_rebuild( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If you do not want to @@ -1890,7 +1935,7 @@ def begin_rebuild( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If you do not want to @@ -1916,7 +1961,7 @@ def begin_rebuild( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker roles and initializes the storage resources that are used by them. If you do not want to @@ -1933,6 +1978,10 @@ def begin_rebuild( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2055,7 +2104,7 @@ def begin_delete_instances( parameters: Optional[_models.RoleInstances] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes role instances in a cloud service. @@ -2081,7 +2130,7 @@ def begin_delete_instances( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes role instances in a cloud service. @@ -2105,7 +2154,7 @@ def begin_delete_instances( resource_group_name: str, cloud_service_name: str, parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes role instances in a cloud service. @@ -2120,6 +2169,10 @@ def begin_delete_instances( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_update_domain_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_update_domain_operations.py index 4a19924c2f71..5f5c43a251bd 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_update_domain_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_cloud_services_update_domain_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -240,7 +241,7 @@ def begin_walk_update_domain( parameters: Optional[_models.UpdateDomain] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Updates the role instances in the specified update domain. @@ -271,7 +272,7 @@ def begin_walk_update_domain( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Updates the role instances in the specified update domain. @@ -300,7 +301,7 @@ def begin_walk_update_domain( cloud_service_name: str, update_domain: int, parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Updates the role instances in the specified update domain. @@ -319,6 +320,10 @@ def begin_walk_update_domain( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -366,7 +371,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def get_update_domain( - self, resource_group_name: str, cloud_service_name: str, update_domain: int, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, update_domain: int, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.UpdateDomain: """Gets the specified update domain of a cloud service. Use nextLink property in the response to get the next page of update domains. Do this till nextLink is null to fetch all the update @@ -384,6 +389,10 @@ def get_update_domain( :rtype: ~azure.mgmt.compute.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -429,7 +438,7 @@ def get_update_domain( @distributed_trace def list_update_domains( - self, resource_group_name: str, cloud_service_name: str, **kwargs: Any + self, resource_group_name: str, cloud_service_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.UpdateDomain"]: """Gets a list of all update domains in a cloud service. @@ -441,6 +450,10 @@ def list_update_domains( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.UpdateDomain] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_galleries_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_galleries_operations.py index 475927ae4330..a95eb63dcea2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_galleries_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_galleries_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -85,7 +86,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models.CommunityGallery: + def get(self, location: str, public_gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.CommunityGallery: """Get a community gallery by gallery public name. :param location: The name of Azure region. Required. @@ -96,6 +97,10 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models :rtype: ~azure.mgmt.compute.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_image_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_image_versions_operations.py index c6bcf6636957..202fd18ea597 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_image_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_image_versions_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -128,7 +129,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, location: str, public_gallery_name: str, gallery_image_name: str, **kwargs: Any + self, location: str, public_gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.CommunityGalleryImageVersion"]: """List community gallery image versions inside an image. @@ -143,6 +144,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CommunityGalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -208,7 +213,7 @@ def get( public_gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CommunityGalleryImageVersion: """Get a community gallery image version. @@ -227,6 +232,10 @@ def get( :rtype: ~azure.mgmt.compute.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_images_operations.py index f0cf726b1170..6ef25c7ed5c3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_community_gallery_images_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -119,7 +120,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, location: str, public_gallery_name: str, **kwargs: Any + self, location: str, public_gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.CommunityGalleryImage"]: """List community gallery images inside a gallery. @@ -132,6 +133,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.CommunityGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -191,7 +196,7 @@ def get_next(next_link=None): @distributed_trace def get( - self, location: str, public_gallery_name: str, gallery_image_name: str, **kwargs: Any + self, location: str, public_gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.CommunityGalleryImage: """Get a community gallery image. @@ -205,6 +210,10 @@ def get( :rtype: ~azure.mgmt.compute.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_host_groups_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_host_groups_operations.py index c743fe43f236..ff4a183a440c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_host_groups_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_host_groups_operations.py @@ -29,6 +29,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -257,7 +258,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.DedicatedHostGroup"]: + def list_by_subscription(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. @@ -265,6 +266,10 @@ def list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.DedicatedHos :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -322,7 +327,7 @@ def get_next(next_link=None): @distributed_trace def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any + self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the specified resource group. Use the nextLink property in the response to get the next page of dedicated host groups. @@ -334,6 +339,10 @@ def list_by_resource_group( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -396,7 +405,7 @@ def get( resource_group_name: str, host_group_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Retrieves information about a dedicated host group. @@ -414,6 +423,10 @@ def get( :rtype: ~azure.mgmt.compute.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -465,7 +478,7 @@ def create_or_update( parameters: _models.DedicatedHostGroup, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host Groups please see [Dedicated Host Documentation] @@ -494,7 +507,7 @@ def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host Groups please see [Dedicated Host Documentation] @@ -521,7 +534,7 @@ def create_or_update( resource_group_name: str, host_group_name: str, parameters: Union[_models.DedicatedHostGroup, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host Groups please see [Dedicated Host Documentation] @@ -539,6 +552,10 @@ def create_or_update( :rtype: ~azure.mgmt.compute.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -601,7 +618,7 @@ def update( parameters: _models.DedicatedHostGroupUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -628,7 +645,7 @@ def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -653,7 +670,7 @@ def update( resource_group_name: str, host_group_name: str, parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -669,6 +686,10 @@ def update( :rtype: ~azure.mgmt.compute.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -725,7 +746,7 @@ def update( @distributed_trace def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, host_group_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Delete a dedicated host group. @@ -738,6 +759,10 @@ def delete( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_hosts_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_hosts_operations.py index ddeb37d67b63..b3bfc3ec2d63 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_hosts_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_dedicated_hosts_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -348,7 +349,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_host_group( - self, resource_group_name: str, host_group_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.DedicatedHost"]: """Lists all of the dedicated hosts in the specified dedicated host group. Use the nextLink property in the response to get the next page of dedicated hosts. @@ -362,6 +363,10 @@ def list_by_host_group( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -426,7 +431,7 @@ def get( host_group_name: str, host_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DedicatedHost: """Retrieves information about a dedicated host. @@ -446,6 +451,10 @@ def get( :rtype: ~azure.mgmt.compute.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -572,7 +581,7 @@ def begin_create_or_update( parameters: _models.DedicatedHost, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -603,7 +612,7 @@ def begin_create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -632,7 +641,7 @@ def begin_create_or_update( host_group_name: str, host_name: str, parameters: Union[_models.DedicatedHost, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -651,6 +660,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -782,7 +795,7 @@ def begin_update( parameters: _models.DedicatedHostUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -813,7 +826,7 @@ def begin_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -842,7 +855,7 @@ def begin_update( host_group_name: str, host_name: str, parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -861,6 +874,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -974,7 +991,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a dedicated host. @@ -989,6 +1006,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1034,7 +1055,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_available_sizes( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged[str]: """Lists all available dedicated host sizes to which the specified dedicated host can be resized. NOTE: The dedicated host sizes provided can be used to only scale up the existing dedicated @@ -1051,6 +1072,10 @@ def list_available_sizes( :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1166,7 +1191,7 @@ def _redeploy_initial( @distributed_trace def begin_redeploy( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Redeploy the dedicated host. The operation will complete successfully once the dedicated host has migrated to a new node and is running. To determine the health of VMs deployed on the @@ -1185,6 +1210,10 @@ def begin_redeploy( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1285,7 +1314,7 @@ def _restart_initial( @distributed_trace def begin_restart( - self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any + self, resource_group_name: str, host_group_name: str, host_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restart the dedicated host. The operation will complete successfully once the dedicated host has restarted and is running. To determine the health of VMs deployed on the dedicated host @@ -1303,6 +1332,10 @@ def begin_restart( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_accesses_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_accesses_operations.py index 281bf0cf2b08..bdeff6eedc32 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_accesses_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_accesses_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -443,13 +444,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.DiskAccess"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -506,7 +511,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.DiskAccess"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.DiskAccess"]: """Lists all the disk access resources under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -516,6 +521,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -573,7 +582,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: + def get(self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DiskAccess: """Gets information about a disk access resource. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -587,6 +596,10 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> :rtype: ~azure.mgmt.compute.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -708,7 +721,7 @@ def begin_create_or_update( disk_access: _models.DiskAccess, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -738,7 +751,7 @@ def begin_create_or_update( disk_access: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -766,7 +779,7 @@ def begin_create_or_update( resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -784,6 +797,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -912,7 +929,7 @@ def begin_update( disk_access: _models.DiskAccessUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -942,7 +959,7 @@ def begin_update( disk_access: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -970,7 +987,7 @@ def begin_update( resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -988,6 +1005,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1091,7 +1112,7 @@ def _delete_initial(self, resource_group_name: str, disk_access_name: str, **kwa return deserialized # type: ignore @distributed_trace - def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_delete(self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Deletes a disk access resource. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -1105,6 +1126,10 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1149,7 +1174,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_private_endpoint_connections( - self, resource_group_name: str, disk_access_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.PrivateEndpointConnection"]: """List information about private endpoint connections under a disk access resource. @@ -1165,6 +1190,10 @@ def list_private_endpoint_connections( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.PrivateEndpointConnection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1224,7 +1253,7 @@ def get_next(next_link=None): @distributed_trace def get_a_private_endpoint_connection( - self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PrivateEndpointConnection: """Gets information about a private endpoint connection under a disk access resource. @@ -1241,6 +1270,10 @@ def get_a_private_endpoint_connection( :rtype: ~azure.mgmt.compute.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1366,7 +1399,7 @@ def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long private_endpoint_connection: _models.PrivateEndpointConnection, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used to create a new private endpoint connection. @@ -1401,7 +1434,7 @@ def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used to create a new private endpoint connection. @@ -1434,7 +1467,7 @@ def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long disk_access_name: str, private_endpoint_connection_name: str, private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used to create a new private endpoint connection. @@ -1458,6 +1491,10 @@ def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.PrivateEndpointConnection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1566,7 +1603,7 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=name-too-l @distributed_trace def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long - self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -1583,6 +1620,10 @@ def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1628,7 +1669,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def get_private_link_resources( - self, resource_group_name: str, disk_access_name: str, **kwargs: Any + self, resource_group_name: str, disk_access_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.PrivateLinkResourceListResult: """Gets the private link resources possible under disk access resource. @@ -1643,6 +1684,10 @@ def get_private_link_resources( :rtype: ~azure.mgmt.compute.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_encryption_sets_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_encryption_sets_operations.py index 9e19ec5f096a..a787ce2466c4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_encryption_sets_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_encryption_sets_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -287,13 +288,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.DiskEncryptionSet"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -350,7 +355,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.DiskEncryptionSet"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -360,6 +365,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -417,7 +426,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: + def get(self, resource_group_name: str, disk_encryption_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.DiskEncryptionSet: """Gets information about a disk encryption set. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -431,6 +440,10 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: :rtype: ~azure.mgmt.compute.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -552,7 +565,7 @@ def begin_create_or_update( disk_encryption_set: _models.DiskEncryptionSet, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -583,7 +596,7 @@ def begin_create_or_update( disk_encryption_set: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -612,7 +625,7 @@ def begin_create_or_update( resource_group_name: str, disk_encryption_set_name: str, disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -631,6 +644,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -759,7 +776,7 @@ def begin_update( disk_encryption_set: _models.DiskEncryptionSetUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -790,7 +807,7 @@ def begin_update( disk_encryption_set: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -819,7 +836,7 @@ def begin_update( resource_group_name: str, disk_encryption_set_name: str, disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -839,6 +856,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -944,7 +965,7 @@ def _delete_initial( return deserialized # type: ignore @distributed_trace - def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Deletes a disk encryption set. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -958,6 +979,10 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1002,7 +1027,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_associated_resources( - self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any + self, resource_group_name: str, disk_encryption_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged[str]: """Lists all resources that are encrypted with this disk encryption set. @@ -1017,6 +1042,10 @@ def list_associated_resources( :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_restore_point_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_restore_point_operations.py index 6aead15e7370..fd41231db9d5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_restore_point_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disk_restore_point_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -232,7 +233,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_restore_point( - self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, **kwargs: Any + self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.DiskRestorePoint"]: """Lists diskRestorePoints under a vmRestorePoint. @@ -249,6 +250,10 @@ def list_by_restore_point( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -314,7 +319,7 @@ def get( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.DiskRestorePoint: """Get disk restorePoint resource. @@ -333,6 +338,10 @@ def get( :rtype: ~azure.mgmt.compute.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -462,7 +471,7 @@ def begin_grant_access( grant_access_data: _models.GrantAccessData, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -498,7 +507,7 @@ def begin_grant_access( grant_access_data: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -532,7 +541,7 @@ def begin_grant_access( vm_restore_point_name: str, disk_restore_point_name: str, grant_access_data: Union[_models.GrantAccessData, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -554,6 +563,10 @@ def begin_grant_access( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -674,7 +687,7 @@ def begin_revoke_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Revokes access to a diskRestorePoint. @@ -693,6 +706,10 @@ def begin_revoke_access( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disks_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disks_operations.py index 3e8b1b36ff32..7bd8d177bf9e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disks_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_disks_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -317,13 +318,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Disk"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Disk"]: """Lists all the disks under a subscription. :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -380,7 +385,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Disk"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Disk"]: """Lists all the disks under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -390,6 +395,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -447,7 +456,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: + def get(self, resource_group_name: str, disk_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Disk: """Gets information about a disk. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -461,6 +470,10 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model :rtype: ~azure.mgmt.compute.models.Disk :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -578,7 +591,7 @@ def begin_create_or_update( disk: _models.Disk, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -607,7 +620,7 @@ def begin_create_or_update( disk: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -630,7 +643,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -648,6 +661,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -772,7 +789,7 @@ def begin_update( disk: _models.DiskUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -801,7 +818,7 @@ def begin_update( disk: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -824,7 +841,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -842,6 +859,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -945,7 +966,7 @@ def _delete_initial(self, resource_group_name: str, disk_name: str, **kwargs: An return deserialized # type: ignore @distributed_trace - def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_delete(self, resource_group_name: str, disk_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Deletes a disk. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -959,6 +980,10 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1080,7 +1105,7 @@ def begin_grant_access( grant_access_data: _models.GrantAccessData, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1110,7 +1135,7 @@ def begin_grant_access( grant_access_data: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1138,7 +1163,7 @@ def begin_grant_access( resource_group_name: str, disk_name: str, grant_access_data: Union[_models.GrantAccessData, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1156,6 +1181,10 @@ def begin_grant_access( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1259,7 +1288,7 @@ def _revoke_access_initial(self, resource_group_name: str, disk_name: str, **kwa return deserialized # type: ignore @distributed_trace - def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_revoke_access(self, resource_group_name: str, disk_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Revokes access to a disk. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -1273,6 +1302,10 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_galleries_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_galleries_operations.py index 96a7a4d4598a..8d5131299859 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_galleries_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_galleries_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -264,13 +265,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Gallery"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Gallery"]: """List galleries under a subscription. :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -327,7 +332,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Gallery"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Gallery"]: """List galleries under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -337,6 +342,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -400,7 +409,7 @@ def get( gallery_name: str, select: Optional[Union[str, _models.SelectPermissions]] = None, expand: Optional[Union[str, _models.GalleryExpandParams]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Gallery: """Retrieves information about a Shared Image Gallery. @@ -419,6 +428,10 @@ def get( :rtype: ~azure.mgmt.compute.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -542,7 +555,7 @@ def begin_create_or_update( gallery: _models.Gallery, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -570,7 +583,7 @@ def begin_create_or_update( gallery: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -592,7 +605,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -608,6 +621,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -735,7 +752,7 @@ def begin_update( gallery: _models.GalleryUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -762,7 +779,7 @@ def begin_update( gallery: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -787,7 +804,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -803,6 +820,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -911,7 +932,7 @@ def _delete_initial(self, resource_group_name: str, gallery_name: str, **kwargs: return deserialized # type: ignore @distributed_trace - def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_delete(self, resource_group_name: str, gallery_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Delete a Shared Image Gallery. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -923,6 +944,10 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_application_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_application_versions_operations.py index 76a96a75d9dc..c77e031ea2e9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_application_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_application_versions_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -275,7 +276,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_application( - self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_application_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryApplicationVersion"]: """List gallery Application Versions in a gallery Application Definition. @@ -292,6 +293,10 @@ def list_by_gallery_application( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryApplicationVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -358,7 +363,7 @@ def get( gallery_application_name: str, gallery_application_version_name: str, expand: Optional[Union[str, _models.ReplicationStatusTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryApplicationVersion: """Retrieves information about a gallery Application Version. @@ -380,6 +385,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -514,7 +523,7 @@ def begin_create_or_update( gallery_application_version: _models.GalleryApplicationVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -551,7 +560,7 @@ def begin_create_or_update( gallery_application_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -586,7 +595,7 @@ def begin_create_or_update( gallery_application_name: str, gallery_application_version_name: str, gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -611,6 +620,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryApplicationVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -746,7 +759,7 @@ def begin_update( gallery_application_version: _models.GalleryApplicationVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -783,7 +796,7 @@ def begin_update( gallery_application_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -818,7 +831,7 @@ def begin_update( gallery_application_name: str, gallery_application_version_name: str, gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -843,6 +856,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryApplicationVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -968,7 +985,7 @@ def begin_delete( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery Application Version. @@ -987,6 +1004,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_applications_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_applications_operations.py index 1933282d518b..3d637d6267fc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_applications_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_applications_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -238,7 +239,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryApplication"]: """List gallery Application Definitions in a gallery. @@ -251,6 +252,10 @@ def list_by_gallery( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -310,7 +315,7 @@ def get_next(next_link=None): @distributed_trace def get( - self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_application_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryApplication: """Retrieves information about a gallery Application Definition. @@ -326,6 +331,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -455,7 +464,7 @@ def begin_create_or_update( gallery_application: _models.GalleryApplication, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -488,7 +497,7 @@ def begin_create_or_update( gallery_application: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -519,7 +528,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application: Union[_models.GalleryApplication, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -539,6 +548,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -670,7 +683,7 @@ def begin_update( gallery_application: _models.GalleryApplicationUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -703,7 +716,7 @@ def begin_update( gallery_application: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -734,7 +747,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -754,6 +767,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -867,7 +884,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_application_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery Application. @@ -883,6 +900,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_image_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_image_versions_operations.py index 7c5aaba51feb..e6b6d0ed35ae 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_image_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_image_versions_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -267,7 +268,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_image( - self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryImageVersion"]: """List gallery image versions in a gallery image definition. @@ -282,6 +283,10 @@ def list_by_gallery_image( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -348,7 +353,7 @@ def get( gallery_image_name: str, gallery_image_version_name: str, expand: Optional[Union[str, _models.ReplicationStatusTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryImageVersion: """Retrieves information about a gallery image version. @@ -369,6 +374,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -503,7 +512,7 @@ def begin_create_or_update( gallery_image_version: _models.GalleryImageVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -539,7 +548,7 @@ def begin_create_or_update( gallery_image_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -573,7 +582,7 @@ def begin_create_or_update( gallery_image_name: str, gallery_image_version_name: str, gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -595,6 +604,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -730,7 +743,7 @@ def begin_update( gallery_image_version: _models.GalleryImageVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -766,7 +779,7 @@ def begin_update( gallery_image_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -800,7 +813,7 @@ def begin_update( gallery_image_name: str, gallery_image_version_name: str, gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -822,6 +835,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -947,7 +964,7 @@ def begin_delete( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery image version. @@ -965,6 +982,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_images_operations.py index a94a9eb97656..98f3cbb1f2d5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_images_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -238,7 +239,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryImage"]: """List gallery image definitions in a gallery. @@ -251,6 +252,10 @@ def list_by_gallery( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -310,7 +315,7 @@ def get_next(next_link=None): @distributed_trace def get( - self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryImage: """Retrieves information about a gallery image definition. @@ -325,6 +330,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -454,7 +463,7 @@ def begin_create_or_update( gallery_image: _models.GalleryImage, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -486,7 +495,7 @@ def begin_create_or_update( gallery_image: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -516,7 +525,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image: Union[_models.GalleryImage, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -535,6 +544,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -666,7 +679,7 @@ def begin_update( gallery_image: _models.GalleryImageUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -697,7 +710,7 @@ def begin_update( gallery_image: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -726,7 +739,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -745,6 +758,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -858,7 +875,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_image_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery image. @@ -873,6 +890,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profile_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profile_versions_operations.py index aa054c43b03c..dcd7eb6e0ca8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profile_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profile_versions_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -285,7 +286,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_in_vm_access_control_profile( # pylint: disable=name-too-long - self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryInVMAccessControlProfileVersion"]: """List gallery inVMAccessControlProfile versions in a gallery inVMAccessControlProfile. @@ -303,6 +304,10 @@ def list_by_gallery_in_vm_access_control_profile( # pylint: disable=name-too-lo ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -368,7 +373,7 @@ def get( gallery_name: str, in_vm_access_control_profile_name: str, in_vm_access_control_profile_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryInVMAccessControlProfileVersion: """Retrieves information about a gallery inVMAccessControlProfile version. @@ -387,6 +392,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -518,7 +527,7 @@ def begin_create_or_update( gallery_in_vm_access_control_profile_version: _models.GalleryInVMAccessControlProfileVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Create or update a gallery inVMAccessControlProfile version. @@ -557,7 +566,7 @@ def begin_create_or_update( gallery_in_vm_access_control_profile_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Create or update a gallery inVMAccessControlProfile version. @@ -593,7 +602,7 @@ def begin_create_or_update( in_vm_access_control_profile_name: str, in_vm_access_control_profile_version_name: str, gallery_in_vm_access_control_profile_version: Union[_models.GalleryInVMAccessControlProfileVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Create or update a gallery inVMAccessControlProfile version. @@ -619,6 +628,10 @@ def begin_create_or_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -758,7 +771,7 @@ def begin_update( gallery_in_vm_access_control_profile_version: _models.GalleryInVMAccessControlProfileVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Update a gallery inVMAccessControlProfile version. @@ -797,7 +810,7 @@ def begin_update( gallery_in_vm_access_control_profile_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Update a gallery inVMAccessControlProfile version. @@ -835,7 +848,7 @@ def begin_update( gallery_in_vm_access_control_profile_version: Union[ _models.GalleryInVMAccessControlProfileVersionUpdate, IO[bytes] ], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfileVersion]: """Update a gallery inVMAccessControlProfile version. @@ -861,6 +874,10 @@ def begin_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfileVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -988,7 +1005,7 @@ def begin_delete( gallery_name: str, in_vm_access_control_profile_name: str, in_vm_access_control_profile_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery inVMAccessControlProfile version. @@ -1007,6 +1024,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profiles_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profiles_operations.py index e9f40ded7017..ea6f0221518b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profiles_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_in_vm_access_control_profiles_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -262,7 +263,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryInVMAccessControlProfile"]: """List gallery inVMAccessControlProfiles in a gallery. @@ -277,6 +278,10 @@ def list_by_gallery( ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryInVMAccessControlProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -336,7 +341,7 @@ def get_next(next_link=None): @distributed_trace def get( - self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryInVMAccessControlProfile: """Retrieves information about a gallery inVMAccessControlProfile. @@ -352,6 +357,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryInVMAccessControlProfile :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -477,7 +486,7 @@ def begin_create_or_update( gallery_in_vm_access_control_profile: _models.GalleryInVMAccessControlProfile, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfile]: """Create or update a gallery inVMAccessControlProfile. @@ -512,7 +521,7 @@ def begin_create_or_update( gallery_in_vm_access_control_profile: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfile]: """Create or update a gallery inVMAccessControlProfile. @@ -544,7 +553,7 @@ def begin_create_or_update( gallery_name: str, in_vm_access_control_profile_name: str, gallery_in_vm_access_control_profile: Union[_models.GalleryInVMAccessControlProfile, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfile]: """Create or update a gallery inVMAccessControlProfile. @@ -567,6 +576,10 @@ def begin_create_or_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -698,7 +711,7 @@ def begin_update( gallery_in_vm_access_control_profile: _models.GalleryInVMAccessControlProfileUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfile]: """Update a gallery inVMAccessControlProfile. @@ -733,7 +746,7 @@ def begin_update( gallery_in_vm_access_control_profile: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfile]: """Update a gallery inVMAccessControlProfile. @@ -765,7 +778,7 @@ def begin_update( gallery_name: str, in_vm_access_control_profile_name: str, gallery_in_vm_access_control_profile: Union[_models.GalleryInVMAccessControlProfileUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryInVMAccessControlProfile]: """Update a gallery inVMAccessControlProfile. @@ -788,6 +801,10 @@ def begin_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryInVMAccessControlProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -903,7 +920,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, in_vm_access_control_profile_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery inVMAccessControlProfile. @@ -919,6 +936,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_script_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_script_versions_operations.py index 827e8dc50260..cdff3219bc26 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_script_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_script_versions_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -281,7 +282,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery_script( - self, resource_group_name: str, gallery_name: str, gallery_script_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_script_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryScriptVersion"]: """List gallery Script Versions in a gallery Script Definition. @@ -298,6 +299,10 @@ def list_by_gallery_script( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryScriptVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -363,7 +368,7 @@ def get( gallery_name: str, gallery_script_name: str, gallery_script_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryScriptVersion: """Custom ArmResourceRead operation template with CloudError as Error. @@ -382,6 +387,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryScriptVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -514,7 +523,7 @@ def begin_create_or_update( gallery_script_version: _models.GalleryScriptVersion, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScriptVersion]: """Create or update a gallery Script Version. @@ -551,7 +560,7 @@ def begin_create_or_update( gallery_script_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScriptVersion]: """Create or update a gallery Script Version. @@ -586,7 +595,7 @@ def begin_create_or_update( gallery_script_name: str, gallery_script_version_name: str, gallery_script_version: Union[_models.GalleryScriptVersion, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScriptVersion]: """Create or update a gallery Script Version. @@ -609,6 +618,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryScriptVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -748,7 +761,7 @@ def begin_update( gallery_script_version: _models.GalleryScriptVersionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScriptVersion]: """Update a gallery Script Version. @@ -785,7 +798,7 @@ def begin_update( gallery_script_version: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScriptVersion]: """Update a gallery Script Version. @@ -820,7 +833,7 @@ def begin_update( gallery_script_name: str, gallery_script_version_name: str, gallery_script_version: Union[_models.GalleryScriptVersionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScriptVersion]: """Update a gallery Script Version. @@ -844,6 +857,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryScriptVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -967,7 +984,7 @@ def begin_delete( gallery_name: str, gallery_script_name: str, gallery_script_version_name: str, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery Script Version. @@ -986,6 +1003,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_scripts_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_scripts_operations.py index ab4d334d5998..c7f6ae374699 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_scripts_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_scripts_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -246,7 +247,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_gallery( - self, resource_group_name: str, gallery_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GalleryScript"]: """List gallery Script Definitions in a gallery. @@ -259,6 +260,10 @@ def list_by_gallery( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GalleryScript] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -318,7 +323,7 @@ def get_next(next_link=None): @distributed_trace def get( - self, resource_group_name: str, gallery_name: str, gallery_script_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_script_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.GalleryScript: """Retrieves information about a gallery script definition. @@ -334,6 +339,10 @@ def get( :rtype: ~azure.mgmt.compute.models.GalleryScript :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -459,7 +468,7 @@ def begin_create_or_update( gallery_script: _models.GalleryScript, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScript]: """Create or update a Gallery Script Definition. Gallery scripts allow the storage, sharing and reuse of common scripts. @@ -493,7 +502,7 @@ def begin_create_or_update( gallery_script: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScript]: """Create or update a Gallery Script Definition. Gallery scripts allow the storage, sharing and reuse of common scripts. @@ -525,7 +534,7 @@ def begin_create_or_update( gallery_name: str, gallery_script_name: str, gallery_script: Union[_models.GalleryScript, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScript]: """Create or update a Gallery Script Definition. Gallery scripts allow the storage, sharing and reuse of common scripts. @@ -546,6 +555,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryScript] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -678,7 +691,7 @@ def begin_update( gallery_script: _models.GalleryScriptUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScript]: """Update a gallery Script Definition. @@ -710,7 +723,7 @@ def begin_update( gallery_script: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScript]: """Update a gallery Script Definition. @@ -740,7 +753,7 @@ def begin_update( gallery_name: str, gallery_script_name: str, gallery_script: Union[_models.GalleryScriptUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.GalleryScript]: """Update a gallery Script Definition. @@ -760,6 +773,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.GalleryScript] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -868,7 +885,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, gallery_name: str, gallery_script_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery_script_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Delete a gallery Script Definition. @@ -884,6 +901,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_sharing_profile_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_sharing_profile_operations.py index d02d6da61ba9..1a845b9372b8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_sharing_profile_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_gallery_sharing_profile_operations.py @@ -32,6 +32,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -175,7 +176,7 @@ def begin_update( sharing_update: _models.SharingUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -203,7 +204,7 @@ def begin_update( sharing_update: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -229,7 +230,7 @@ def begin_update( resource_group_name: str, gallery_name: str, sharing_update: Union[_models.SharingUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -246,6 +247,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.SharingUpdate] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_images_operations.py index f717f077e5ca..213ee4db5fb0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_images_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -252,7 +253,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Image"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. @@ -260,6 +261,10 @@ def list(self, **kwargs: Any) -> ItemPaged["_models.Image"]: :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -316,7 +321,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Image"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Image"]: """Gets the list of images under a resource group. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. @@ -327,6 +332,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -385,7 +394,7 @@ def get_next(next_link=None): @distributed_trace def get( - self, resource_group_name: str, image_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, image_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.Image: """Gets an image. @@ -400,6 +409,10 @@ def get( :rtype: ~azure.mgmt.compute.models.Image :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -518,7 +531,7 @@ def begin_create_or_update( parameters: _models.Image, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -545,7 +558,7 @@ def begin_create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -566,7 +579,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -582,6 +595,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -710,7 +727,7 @@ def begin_update( parameters: _models.ImageUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -737,7 +754,7 @@ def begin_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -762,7 +779,7 @@ def begin_update( resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -778,6 +795,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Image] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -881,7 +902,7 @@ def _delete_initial(self, resource_group_name: str, image_name: str, **kwargs: A return deserialized # type: ignore @distributed_trace - def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_delete(self, resource_group_name: str, image_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Deletes an Image. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -893,6 +914,10 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_log_analytics_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_log_analytics_operations.py index 41ced928791a..60b51b04ff11 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_log_analytics_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_log_analytics_operations.py @@ -32,6 +32,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -199,7 +200,7 @@ def begin_export_request_rate_by_interval( parameters: _models.RequestRateByIntervalInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -220,7 +221,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -241,7 +242,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -256,6 +257,10 @@ def begin_export_request_rate_by_interval( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.LogAnalyticsOperationResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -379,7 +384,7 @@ def begin_export_throttled_requests( parameters: _models.ThrottledRequestsInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -399,7 +404,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -419,7 +424,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -434,6 +439,10 @@ def begin_export_throttled_requests( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.LogAnalyticsOperationResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_operations.py index 84468c6bad8b..175f82ef1229 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -75,13 +76,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Operation"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Operation"]: """List the operations for the provider. :return: An iterator like instance of either Operation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Operation] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_proximity_placement_groups_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_proximity_placement_groups_operations.py index 1f5332fdc1ff..353b08b202f1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_proximity_placement_groups_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_proximity_placement_groups_operations.py @@ -29,6 +29,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -269,7 +270,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.ProximityPlacementGroup"]: + def list_by_subscription(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. :return: An iterator like instance of either ProximityPlacementGroup or the result of @@ -277,6 +278,10 @@ def list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.ProximityPla :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.ProximityPlacementGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -334,7 +339,7 @@ def get_next(next_link=None): @distributed_trace def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any + self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a resource group. @@ -346,6 +351,10 @@ def list_by_resource_group( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.ProximityPlacementGroup] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -408,7 +417,7 @@ def get( resource_group_name: str, proximity_placement_group_name: str, include_colocation_status: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Retrieves information about a proximity placement group . @@ -424,6 +433,10 @@ def get( :rtype: ~azure.mgmt.compute.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -475,7 +488,7 @@ def create_or_update( parameters: _models.ProximityPlacementGroup, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -503,7 +516,7 @@ def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -529,7 +542,7 @@ def create_or_update( resource_group_name: str, proximity_placement_group_name: str, parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -545,6 +558,10 @@ def create_or_update( :rtype: ~azure.mgmt.compute.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -607,7 +624,7 @@ def update( parameters: _models.ProximityPlacementGroupUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -635,7 +652,7 @@ def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -661,7 +678,7 @@ def update( resource_group_name: str, proximity_placement_group_name: str, parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -677,6 +694,10 @@ def update( :rtype: ~azure.mgmt.compute.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -733,7 +754,7 @@ def update( @distributed_trace def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, proximity_placement_group_name: str, **kwargs: Any + self, resource_group_name: str, proximity_placement_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Delete a proximity placement group. @@ -746,6 +767,10 @@ def delete( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_resource_skus_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_resource_skus_operations.py index 14927fc97dcf..0ee62aeae807 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_resource_skus_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_resource_skus_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -93,7 +94,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, filter: Optional[str] = None, include_extended_locations: Optional[str] = None, **kwargs: Any + self, filter: Optional[str] = None, include_extended_locations: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.ResourceSku"]: """Gets the list of Microsoft.Compute SKUs available for your Subscription. @@ -107,6 +108,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.ResourceSku] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_point_collections_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_point_collections_operations.py index c9719e875fa9..38af34533255 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_point_collections_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_point_collections_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -271,7 +272,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_all(self, **kwargs: Any) -> ItemPaged["_models.RestorePointCollection"]: + def list_all(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. @@ -281,6 +282,10 @@ def list_all(self, **kwargs: Any) -> ItemPaged["_models.RestorePointCollection"] :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.RestorePointCollection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -337,7 +342,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.RestorePointCollection"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.RestorePointCollection"]: """Gets the list of restore point collections in a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -348,6 +353,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Re :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.RestorePointCollection] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -410,7 +419,7 @@ def get( resource_group_name: str, restore_point_collection_name: str, expand: Optional[Union[str, _models.RestorePointCollectionExpandOptions]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to get the restore point collection. @@ -427,6 +436,10 @@ def get( :rtype: ~azure.mgmt.compute.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -478,7 +491,7 @@ def create_or_update( parameters: _models.RestorePointCollection, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to https://aka.ms/RestorePoints for more details. When updating a restore point collection, only @@ -508,7 +521,7 @@ def create_or_update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to https://aka.ms/RestorePoints for more details. When updating a restore point collection, only @@ -536,7 +549,7 @@ def create_or_update( resource_group_name: str, restore_point_collection_name: str, parameters: Union[_models.RestorePointCollection, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to https://aka.ms/RestorePoints for more details. When updating a restore point collection, only @@ -554,6 +567,10 @@ def create_or_update( :rtype: ~azure.mgmt.compute.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -616,7 +633,7 @@ def update( parameters: _models.RestorePointCollectionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -644,7 +661,7 @@ def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -670,7 +687,7 @@ def update( resource_group_name: str, restore_point_collection_name: str, parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -686,6 +703,10 @@ def update( :rtype: ~azure.mgmt.compute.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -797,7 +818,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any + self, resource_group_name: str, restore_point_collection_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete the restore point collection. This operation will also delete all the contained restore points. @@ -811,6 +832,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_points_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_points_operations.py index 34521c70cb22..f561ab808e19 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_points_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_restore_points_operations.py @@ -32,6 +32,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -194,7 +195,7 @@ def get( restore_point_collection_name: str, restore_point_name: str, expand: Optional[Union[str, _models.RestorePointExpandOptions]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RestorePoint: """The operation to get the restore point. @@ -212,6 +213,10 @@ def get( :rtype: ~azure.mgmt.compute.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -337,7 +342,7 @@ def begin_create( parameters: _models.RestorePoint, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is not allowed. @@ -369,7 +374,7 @@ def begin_create( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is not allowed. @@ -399,7 +404,7 @@ def begin_create( restore_point_collection_name: str, restore_point_name: str, parameters: Union[_models.RestorePoint, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is not allowed. @@ -419,6 +424,10 @@ def begin_create( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.RestorePoint] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -532,7 +541,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any + self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete the restore point. @@ -547,6 +556,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_galleries_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_galleries_operations.py index 294a531c78aa..5f8d8dc62cba 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_galleries_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_galleries_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -123,7 +124,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, location: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, **kwargs: Any + self, location: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.SharedGallery"]: """List shared galleries by subscription id or tenant id. @@ -136,6 +137,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -194,7 +199,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: + def get(self, location: str, gallery_unique_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. :param location: The name of Azure region. Required. @@ -205,6 +210,10 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models :rtype: ~azure.mgmt.compute.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_image_versions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_image_versions_operations.py index a85f3bb3bab2..a4f04c3c3118 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_image_versions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_image_versions_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -141,7 +142,7 @@ def list( gallery_unique_name: str, gallery_image_name: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.SharedGalleryImageVersion"]: """List shared gallery image versions by subscription id or tenant id. @@ -160,6 +161,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.SharedGalleryImageVersion] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -226,7 +231,7 @@ def get( gallery_unique_name: str, gallery_image_name: str, gallery_image_version_name: str, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SharedGalleryImageVersion: """Get a shared gallery image version by subscription id or tenant id. @@ -246,6 +251,10 @@ def get( :rtype: ~azure.mgmt.compute.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_images_operations.py index c1e9393a26cb..ab17d7f0e944 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_shared_gallery_images_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -132,7 +133,7 @@ def list( location: str, gallery_unique_name: str, shared_to: Optional[Union[str, _models.SharedToValues]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.SharedGalleryImage"]: """List shared gallery images by subscription id or tenant id. @@ -147,6 +148,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -207,7 +212,7 @@ def get_next(next_link=None): @distributed_trace def get( - self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any + self, location: str, gallery_unique_name: str, gallery_image_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SharedGalleryImage: """Get a shared gallery image by subscription id or tenant id. @@ -222,6 +227,10 @@ def get( :rtype: ~azure.mgmt.compute.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_snapshots_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_snapshots_operations.py index 62d80b6db608..371b5a231ec9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_snapshots_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_snapshots_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -321,13 +322,17 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> ItemPaged["_models.Snapshot"]: + def list(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Snapshot"]: """Lists snapshots under a subscription. :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -384,7 +389,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Snapshot"]: + def list_by_resource_group(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Snapshot"]: """Lists snapshots under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -394,6 +399,10 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -451,7 +460,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: + def get(self, resource_group_name: str, snapshot_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.Snapshot: """Gets information about a snapshot. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -465,6 +474,10 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m :rtype: ~azure.mgmt.compute.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -582,7 +595,7 @@ def begin_create_or_update( snapshot: _models.Snapshot, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -611,7 +624,7 @@ def begin_create_or_update( snapshot: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -634,7 +647,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -652,6 +665,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -780,7 +797,7 @@ def begin_update( snapshot: _models.SnapshotUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -810,7 +827,7 @@ def begin_update( snapshot: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -838,7 +855,7 @@ def begin_update( resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -856,6 +873,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -959,7 +980,7 @@ def _delete_initial(self, resource_group_name: str, snapshot_name: str, **kwargs return deserialized # type: ignore @distributed_trace - def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_delete(self, resource_group_name: str, snapshot_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Deletes a snapshot. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -973,6 +994,10 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1094,7 +1119,7 @@ def begin_grant_access( grant_access_data: _models.GrantAccessData, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -1124,7 +1149,7 @@ def begin_grant_access( grant_access_data: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -1152,7 +1177,7 @@ def begin_grant_access( resource_group_name: str, snapshot_name: str, grant_access_data: Union[_models.GrantAccessData, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -1170,6 +1195,10 @@ def begin_grant_access( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1273,7 +1302,7 @@ def _revoke_access_initial(self, resource_group_name: str, snapshot_name: str, * return deserialized # type: ignore @distributed_trace - def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Revokes access to a snapshot. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -1287,6 +1316,10 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_soft_deleted_resource_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_soft_deleted_resource_operations.py index 26393d30bb01..5b38e53f73b0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_soft_deleted_resource_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_soft_deleted_resource_operations.py @@ -28,6 +28,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -102,7 +103,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_by_artifact_name( - self, resource_group_name: str, gallery_name: str, artifact_type: str, artifact_name: str, **kwargs: Any + self, resource_group_name: str, gallery_name: str, artifact_type: str, artifact_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.GallerySoftDeletedResource"]: """List soft-deleted resources of an artifact in the gallery, such as soft-deleted gallery image version of an image. @@ -123,6 +124,10 @@ def list_by_artifact_name( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.GallerySoftDeletedResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_ssh_public_keys_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_ssh_public_keys_operations.py index 7b151bd4a88b..4cec9c5065d0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_ssh_public_keys_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_ssh_public_keys_operations.py @@ -29,6 +29,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -286,7 +287,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.SshPublicKeyResource"]: + def list_by_subscription(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.SshPublicKeyResource"]: """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. @@ -295,6 +296,10 @@ def list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.SshPublicKey :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.SshPublicKeyResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -352,7 +357,7 @@ def get_next(next_link=None): @distributed_trace def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any + self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.SshPublicKeyResource"]: """Lists all of the SSH public keys in the specified resource group. Use the nextLink property in the response to get the next page of SSH public keys. @@ -365,6 +370,10 @@ def list_by_resource_group( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.SshPublicKeyResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -422,7 +431,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: + def get(self, resource_group_name: str, ssh_public_key_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.SshPublicKeyResource: """Retrieves information about an SSH public key. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -434,6 +443,10 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :rtype: ~azure.mgmt.compute.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -484,7 +497,7 @@ def create( parameters: _models.SshPublicKeyResource, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -511,7 +524,7 @@ def create( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -536,7 +549,7 @@ def create( resource_group_name: str, ssh_public_key_name: str, parameters: Union[_models.SshPublicKeyResource, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -552,6 +565,10 @@ def create( :rtype: ~azure.mgmt.compute.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -614,7 +631,7 @@ def update( parameters: _models.SshPublicKeyUpdateResource, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -641,7 +658,7 @@ def update( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -666,7 +683,7 @@ def update( resource_group_name: str, ssh_public_key_name: str, parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -682,6 +699,10 @@ def update( :rtype: ~azure.mgmt.compute.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -738,7 +759,7 @@ def update( @distributed_trace def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any + self, resource_group_name: str, ssh_public_key_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Delete an SSH public key. @@ -751,6 +772,10 @@ def delete( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -797,7 +822,7 @@ def generate_key_pair( parameters: Optional[_models.SshGenerateKeyPairInputParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with the public key. The length of the key will be 3072 bits. This operation can only be performed @@ -826,7 +851,7 @@ def generate_key_pair( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with the public key. The length of the key will be 3072 bits. This operation can only be performed @@ -853,7 +878,7 @@ def generate_key_pair( resource_group_name: str, ssh_public_key_name: str, parameters: Optional[Union[_models.SshGenerateKeyPairInputParameters, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with the public key. The length of the key will be 3072 bits. This operation can only be performed @@ -871,6 +896,10 @@ def generate_key_pair( :rtype: ~azure.mgmt.compute.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_usage_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_usage_operations.py index c678205a22e5..726fb085cec8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_usage_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_usage_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -83,7 +84,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, location: str, **kwargs: Any) -> ItemPaged["_models.Usage"]: + def list(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.Usage"]: """Gets, for the specified location, the current compute resource usage information as well as the limits for compute resources under the subscription. @@ -93,6 +94,10 @@ def list(self, location: str, **kwargs: Any) -> ItemPaged["_models.Usage"]: :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extension_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extension_images_operations.py index 45f36345bb9d..4b187e287941 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extension_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extension_images_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -165,7 +166,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_types( - self, location: str, publisher_name: str, **kwargs: Any + self, location: str, publisher_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineExtensionImage]: """Gets a list of virtual machine extension image types. @@ -177,6 +178,10 @@ def list_types( :rtype: list[~azure.mgmt.compute.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -228,7 +233,7 @@ def list_versions( filter: Optional[str] = None, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineExtensionImage]: """Gets a list of virtual machine extension image versions. @@ -248,6 +253,10 @@ def list_versions( :rtype: list[~azure.mgmt.compute.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -296,7 +305,7 @@ def list_versions( @distributed_trace def get( - self, location: str, publisher_name: str, type: str, version: str, **kwargs: Any + self, location: str, publisher_name: str, type: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineExtensionImage: """Gets a virtual machine extension image. @@ -312,6 +321,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extensions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extensions_operations.py index 11665d43b8c9..3075c2ce2474 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extensions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_extensions_operations.py @@ -32,6 +32,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -247,7 +248,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineExtensionsListResult: """The operation to get all extensions of a Virtual Machine. @@ -262,6 +263,10 @@ def list( :rtype: ~azure.mgmt.compute.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -312,7 +317,7 @@ def get( vm_name: str, vm_extension_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineExtension: """The operation to get the extension. @@ -329,6 +334,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -455,7 +464,7 @@ def begin_create_or_update( extension_parameters: _models.VirtualMachineExtension, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -487,7 +496,7 @@ def begin_create_or_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -517,7 +526,7 @@ def begin_create_or_update( vm_name: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -536,6 +545,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -667,7 +680,7 @@ def begin_update( extension_parameters: _models.VirtualMachineExtensionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -699,7 +712,7 @@ def begin_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -729,7 +742,7 @@ def begin_update( vm_name: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -749,6 +762,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -862,7 +879,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, vm_extension_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete the extension. @@ -877,6 +894,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_edge_zone_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_edge_zone_operations.py index 66d3fd62caf9..842f98e594c3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_edge_zone_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_edge_zone_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -243,7 +244,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list_publishers( - self, location: str, edge_zone: str, **kwargs: Any + self, location: str, edge_zone: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image publishers for the specified Azure location and edge zone. @@ -255,6 +256,10 @@ def list_publishers( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -299,7 +304,7 @@ def list_publishers( @distributed_trace def list_offers( - self, location: str, edge_zone: str, publisher_name: str, **kwargs: Any + self, location: str, edge_zone: str, publisher_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image offers for the specified location, edge zone and publisher. @@ -314,6 +319,10 @@ def list_offers( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -359,7 +368,7 @@ def list_offers( @distributed_trace def list_skus( - self, location: str, edge_zone: str, publisher_name: str, offer: str, **kwargs: Any + self, location: str, edge_zone: str, publisher_name: str, offer: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image SKUs for the specified location, edge zone, publisher, and offer. @@ -376,6 +385,10 @@ def list_skus( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -431,7 +444,7 @@ def list( expand: Optional[str] = None, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of all virtual machine image versions for the specified location, edge zone, publisher, offer, and SKU. @@ -458,6 +471,10 @@ def list( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -508,7 +525,7 @@ def list( @distributed_trace def get( - self, location: str, edge_zone: str, publisher_name: str, offer: str, skus: str, version: str, **kwargs: Any + self, location: str, edge_zone: str, publisher_name: str, offer: str, skus: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineImage: """Gets a virtual machine image in an edge zone. @@ -528,6 +545,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_operations.py index 1e10fba23718..e21d84d9cc26 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_images_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -302,7 +303,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: + def list_by_edge_zone(self, location: str, edge_zone: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: """Gets a list of all virtual machine image versions for the specified edge zone. :param location: The name of Azure region. Required. @@ -313,6 +314,10 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo :rtype: ~azure.mgmt.compute.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -356,7 +361,7 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo return deserialized # type: ignore @distributed_trace - def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: + def list_publishers(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image publishers for the specified Azure location. :param location: The name of Azure region. Required. @@ -365,6 +370,10 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -408,7 +417,7 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM @distributed_trace def list_offers( - self, location: str, publisher_name: str, **kwargs: Any + self, location: str, publisher_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image offers for the specified location and publisher. @@ -420,6 +429,10 @@ def list_offers( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -464,7 +477,7 @@ def list_offers( @distributed_trace def list_skus( - self, location: str, publisher_name: str, offer: str, **kwargs: Any + self, location: str, publisher_name: str, offer: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of virtual machine image SKUs for the specified location, publisher, and offer. @@ -478,6 +491,10 @@ def list_skus( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -531,7 +548,7 @@ def list( expand: Optional[str] = None, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImageResource]: """Gets a list of all virtual machine image versions for the specified location, publisher, offer, and SKU. @@ -554,6 +571,10 @@ def list( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -603,7 +624,7 @@ def list( @distributed_trace def get( - self, location: str, publisher_name: str, offer: str, skus: str, version: str, **kwargs: Any + self, location: str, publisher_name: str, offer: str, skus: str, version: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineImage: """Gets a virtual machine image. @@ -621,6 +642,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -676,7 +701,7 @@ def list_with_properties( expand: str, top: Optional[int] = None, orderby: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> List[_models.VirtualMachineImage]: """list_with_properties. @@ -698,6 +723,10 @@ def list_with_properties( :rtype: list[~azure.mgmt.compute.models.VirtualMachineImage] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_run_commands_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_run_commands_operations.py index c4f8d9c7eb99..85a9e6c590d4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_run_commands_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_run_commands_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -303,7 +304,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, location: str, **kwargs: Any) -> ItemPaged["_models.RunCommandDocumentBase"]: + def list(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.RunCommandDocumentBase"]: """Lists all available run commands for a subscription in a location. :param location: The name of Azure region. Required. @@ -313,6 +314,10 @@ def list(self, location: str, **kwargs: Any) -> ItemPaged["_models.RunCommandDoc :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.RunCommandDocumentBase] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -370,7 +375,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: + def get(self, location: str, command_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. :param location: The name of Azure region. Required. @@ -383,6 +388,10 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :rtype: ~azure.mgmt.compute.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -427,7 +436,7 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma @distributed_trace def list_by_virtual_machine( - self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachineRunCommand"]: """The operation to get all run commands of a Virtual Machine. @@ -443,6 +452,10 @@ def list_by_virtual_machine( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -503,7 +516,7 @@ def get_next(next_link=None): @distributed_trace def get_by_virtual_machine( - self, resource_group_name: str, vm_name: str, run_command_name: str, expand: Optional[str] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, run_command_name: str, expand: Optional[str] = None, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineRunCommand: """The operation to get the run command. @@ -520,6 +533,10 @@ def get_by_virtual_machine( :rtype: ~azure.mgmt.compute.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -646,7 +663,7 @@ def begin_create_or_update( run_command: _models.VirtualMachineRunCommand, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -678,7 +695,7 @@ def begin_create_or_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -708,7 +725,7 @@ def begin_create_or_update( vm_name: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -727,6 +744,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -858,7 +879,7 @@ def begin_update( run_command: _models.VirtualMachineRunCommandUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -890,7 +911,7 @@ def begin_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -920,7 +941,7 @@ def begin_update( vm_name: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -939,6 +960,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1052,7 +1077,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, run_command_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete the run command. @@ -1067,6 +1092,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_extensions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_extensions_operations.py index e0df57a0f89d..518807cb09a2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_extensions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_extensions_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -246,7 +247,7 @@ def __init__(self, *args, **kwargs) -> None: @distributed_trace def list( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachineScaleSetExtension"]: """Gets a list of all extensions in a VM scale set. @@ -261,6 +262,10 @@ def list( ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSetExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -325,7 +330,7 @@ def get( vm_scale_set_name: str, vmss_extension_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: """The operation to get the extension. @@ -342,6 +347,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -468,7 +477,7 @@ def begin_create_or_update( extension_parameters: _models.VirtualMachineScaleSetExtension, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -501,7 +510,7 @@ def begin_create_or_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -532,7 +541,7 @@ def begin_create_or_update( vm_scale_set_name: str, vmss_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -553,6 +562,10 @@ def begin_create_or_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -685,7 +698,7 @@ def begin_update( extension_parameters: _models.VirtualMachineScaleSetExtensionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -718,7 +731,7 @@ def begin_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -749,7 +762,7 @@ def begin_update( vm_scale_set_name: str, vmss_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -771,6 +784,10 @@ def begin_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -879,7 +896,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete the extension. @@ -894,6 +911,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py index 24a9984e9114..07eab9d2bd95 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_rolling_upgrades_operations.py @@ -31,6 +31,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -248,7 +249,7 @@ def _start_extension_upgrade_initial( @distributed_trace def begin_start_extension_upgrade( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Starts a rolling upgrade to move all extensions for all virtual machine scale set instances to the latest available extension version. Instances which are already running the latest @@ -263,6 +264,10 @@ def begin_start_extension_upgrade( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -362,7 +367,7 @@ def _start_os_upgrade_initial( @distributed_trace def begin_start_os_upgrade( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Starts a rolling upgrade to move all virtual machine scale set instances to the latest available Platform Image OS version. Instances which are already running the latest available @@ -377,6 +382,10 @@ def begin_start_os_upgrade( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -473,7 +482,7 @@ def _cancel_initial(self, resource_group_name: str, vm_scale_set_name: str, **kw return deserialized # type: ignore @distributed_trace - def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Cancels the current virtual machine scale set rolling upgrade. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -485,6 +494,10 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -529,7 +542,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def get_latest( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RollingUpgradeStatusInfo: """Gets the status of the latest virtual machine scale set rolling upgrade. @@ -542,6 +555,10 @@ def get_latest( :rtype: ~azure.mgmt.compute.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_extensions_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_extensions_operations.py index 82cd8f30057c..713e1af48b25 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_extensions_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_extensions_operations.py @@ -32,6 +32,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -279,7 +280,7 @@ def list( vm_scale_set_name: str, instance_id: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtensionsListResult: """The operation to get all extensions of an instance in Virtual Machine Scaleset. @@ -296,6 +297,10 @@ def list( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -350,7 +355,7 @@ def get( instance_id: str, vm_extension_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: """The operation to get the VMSS VM extension. @@ -369,6 +374,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -499,7 +508,7 @@ def begin_create_or_update( extension_parameters: _models.VirtualMachineScaleSetVMExtension, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -535,7 +544,7 @@ def begin_create_or_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -569,7 +578,7 @@ def begin_create_or_update( instance_id: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -592,6 +601,10 @@ def begin_create_or_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -727,7 +740,7 @@ def begin_update( extension_parameters: _models.VirtualMachineScaleSetVMExtensionUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -763,7 +776,7 @@ def begin_update( extension_parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -797,7 +810,7 @@ def begin_update( instance_id: str, vm_extension_name: str, extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -821,6 +834,10 @@ def begin_update( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetVMExtension] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -936,7 +953,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete the VMSS VM extension. @@ -953,6 +970,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_run_commands_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_run_commands_operations.py index 7db24a6425d4..9a75124d30bd 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_run_commands_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vm_run_commands_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -280,7 +281,7 @@ def list( vm_scale_set_name: str, instance_id: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachineRunCommand"]: """The operation to get all run commands of an instance in Virtual Machine Scaleset. @@ -298,6 +299,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -365,7 +370,7 @@ def get( instance_id: str, run_command_name: str, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineRunCommand: """The operation to get the VMSS VM run command. @@ -384,6 +389,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -514,7 +523,7 @@ def begin_create_or_update( run_command: _models.VirtualMachineRunCommand, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -549,7 +558,7 @@ def begin_create_or_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -582,7 +591,7 @@ def begin_create_or_update( instance_id: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -603,6 +612,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -738,7 +751,7 @@ def begin_update( run_command: _models.VirtualMachineRunCommandUpdate, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -772,7 +785,7 @@ def begin_update( run_command: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -804,7 +817,7 @@ def begin_update( instance_id: str, run_command_name: str, run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -825,6 +838,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineRunCommand] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -940,7 +957,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete the VMSS VM run command. @@ -957,6 +974,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vms_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vms_operations.py index 93f4f3a70fca..9352739bdd23 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vms_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_set_vms_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -752,7 +753,7 @@ def list( filter: Optional[str] = None, select: Optional[str] = None, expand: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachineScaleSetVM"]: """Gets a list of all virtual machines in a VM scale sets. @@ -776,6 +777,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSetVM] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -843,7 +848,7 @@ def get( vm_scale_set_name: str, instance_id: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: """Gets a virtual machine from a VM scale set. @@ -863,6 +868,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -995,7 +1004,7 @@ def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -1037,7 +1046,7 @@ def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -1077,7 +1086,7 @@ def begin_update( parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -1104,6 +1113,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSetVM] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1225,7 +1238,7 @@ def begin_delete( vm_scale_set_name: str, instance_id: str, force_deletion: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes a virtual machine from a VM scale set. @@ -1243,6 +1256,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1344,7 +1361,7 @@ def _approve_rolling_upgrade_initial( @distributed_trace def begin_approve_rolling_upgrade( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Approve upgrade on deferred rolling upgrade for OS disk on a VM scale set instance. @@ -1359,6 +1376,10 @@ def begin_approve_rolling_upgrade( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1484,7 +1505,7 @@ def begin_attach_detach_data_disks( parameters: _models.AttachDetachDataDisksRequest, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -1516,7 +1537,7 @@ def begin_attach_detach_data_disks( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -1546,7 +1567,7 @@ def begin_attach_detach_data_disks( vm_scale_set_name: str, instance_id: str, parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -1566,6 +1587,10 @@ def begin_attach_detach_data_disks( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.StorageProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1674,7 +1699,7 @@ def _deallocate_initial( @distributed_trace def begin_deallocate( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates a specific virtual machine in a VM scale set. Shuts down the virtual machine and releases the compute resources it uses. You are not billed for the compute resources of this @@ -1691,6 +1716,10 @@ def begin_deallocate( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1736,7 +1765,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def get_instance_view( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetVMInstanceView: """Gets the status of a virtual machine from a VM scale set. @@ -1751,6 +1780,10 @@ def get_instance_view( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1852,7 +1885,7 @@ def _perform_maintenance_initial( @distributed_trace def begin_perform_maintenance( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Performs maintenance on a virtual machine in a VM scale set. @@ -1867,6 +1900,10 @@ def begin_perform_maintenance( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1979,7 +2016,7 @@ def begin_power_off( vm_scale_set_name: str, instance_id: str, skip_shutdown: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Power off (stop) a virtual machine in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release resources and @@ -2000,6 +2037,10 @@ def begin_power_off( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2102,7 +2143,7 @@ def _redeploy_initial( @distributed_trace def begin_redeploy( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down the virtual machine in the virtual machine scale set, moves it to a new node, and powers it back on. @@ -2118,6 +2159,10 @@ def begin_redeploy( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2249,7 +2294,7 @@ def begin_reimage( vm_scale_set_vm_reimage_input: Optional[_models.VirtualMachineScaleSetVMReimageParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -2281,7 +2326,7 @@ def begin_reimage( vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -2312,7 +2357,7 @@ def begin_reimage( vm_scale_set_vm_reimage_input: Optional[ Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] ] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -2332,6 +2377,10 @@ def begin_reimage( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2437,7 +2486,7 @@ def _reimage_all_initial( @distributed_trace def begin_reimage_all( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Allows you to re-image all the disks ( including data disks ) in the a VM scale set instance. This operation is only supported for managed disks. @@ -2453,6 +2502,10 @@ def begin_reimage_all( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2554,7 +2607,7 @@ def _restart_initial( @distributed_trace def begin_restart( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts a virtual machine in a VM scale set. @@ -2569,6 +2622,10 @@ def begin_restart( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2619,7 +2676,7 @@ def retrieve_boot_diagnostics_data( vm_scale_set_name: str, instance_id: str, sas_uri_expiration_time_in_minutes: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RetrieveBootDiagnosticsDataResult: """The operation to retrieve SAS URIs of boot diagnostic logs for a virtual machine in a VM scale set. @@ -2639,6 +2696,10 @@ def retrieve_boot_diagnostics_data( :rtype: ~azure.mgmt.compute.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2765,7 +2826,7 @@ def begin_run_command( parameters: _models.RunCommandInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -2796,7 +2857,7 @@ def begin_run_command( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -2825,7 +2886,7 @@ def begin_run_command( vm_scale_set_name: str, instance_id: str, parameters: Union[_models.RunCommandInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -2844,6 +2905,10 @@ def begin_run_command( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.RunCommandResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2896,7 +2961,7 @@ def get_long_running_output(pipeline_response): @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """The operation to simulate the eviction of spot virtual machine in a VM scale set. @@ -2911,6 +2976,10 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3008,7 +3077,7 @@ def _start_initial( @distributed_trace def begin_start( - self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Starts a virtual machine in a VM scale set. @@ -3023,6 +3092,10 @@ def begin_start( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_sets_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_sets_operations.py index a3be666bf27a..100623548082 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_sets_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_scale_sets_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -1020,7 +1021,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_location(self, location: str, **kwargs: Any) -> ItemPaged["_models.VirtualMachineScaleSet"]: + def list_by_location(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.VirtualMachineScaleSet"]: """Gets all the VM scale sets under the specified subscription for the specified location. :param location: The name of Azure region. Required. @@ -1030,6 +1031,10 @@ def list_by_location(self, location: str, **kwargs: Any) -> ItemPaged["_models.V :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1087,7 +1092,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list_all(self, **kwargs: Any) -> ItemPaged["_models.VirtualMachineScaleSet"]: + def list_all(self, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. @@ -1097,6 +1102,10 @@ def list_all(self, **kwargs: Any) -> ItemPaged["_models.VirtualMachineScaleSet"] :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1153,7 +1162,7 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) @distributed_trace - def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.VirtualMachineScaleSet"]: + def list(self, resource_group_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.VirtualMachineScaleSet"]: """Gets a list of all VM scale sets under a resource group. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -1164,6 +1173,10 @@ def list(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Vi :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1226,7 +1239,7 @@ def get( resource_group_name: str, vm_scale_set_name: str, expand: Optional[Union[str, _models.ExpandTypesForGetVMScaleSets]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSet: """Display information about a virtual machine scale set. @@ -1243,6 +1256,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1371,7 +1388,7 @@ def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -1409,7 +1426,7 @@ def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -1445,7 +1462,7 @@ def begin_create_or_update( parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -1470,6 +1487,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1605,7 +1626,7 @@ def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1643,7 +1664,7 @@ def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1679,7 +1700,7 @@ def begin_update( parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1704,6 +1725,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineScaleSet] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1818,7 +1843,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, vm_scale_set_name: str, force_deletion: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, force_deletion: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes a VM scale set. @@ -1834,6 +1859,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1959,7 +1988,7 @@ def begin_approve_rolling_upgrade( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale set. @@ -1988,7 +2017,7 @@ def begin_approve_rolling_upgrade( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale set. @@ -2015,7 +2044,7 @@ def begin_approve_rolling_upgrade( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale set. @@ -2033,6 +2062,10 @@ def begin_approve_rolling_upgrade( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2087,7 +2120,7 @@ def convert_to_single_placement_group( parameters: _models.VMScaleSetConvertToSinglePlacementGroupInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -2114,7 +2147,7 @@ def convert_to_single_placement_group( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -2139,7 +2172,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st resource_group_name: str, vm_scale_set_name: str, parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -2156,6 +2189,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2292,7 +2329,7 @@ def begin_deallocate( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and releases the compute resources. You are not billed for the compute resources that this virtual @@ -2327,7 +2364,7 @@ def begin_deallocate( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and releases the compute resources. You are not billed for the compute resources that this virtual @@ -2360,7 +2397,7 @@ def begin_deallocate( vm_scale_set_name: str, hibernate: Optional[bool] = None, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and releases the compute resources. You are not billed for the compute resources that this virtual @@ -2383,6 +2420,10 @@ def begin_deallocate( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2512,7 +2553,7 @@ def begin_delete_instances( force_deletion: Optional[bool] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -2544,7 +2585,7 @@ def begin_delete_instances( force_deletion: Optional[bool] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -2574,7 +2615,7 @@ def begin_delete_instances( vm_scale_set_name: str, vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], force_deletion: Optional[bool] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -2594,6 +2635,10 @@ def begin_delete_instances( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2648,7 +2693,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disabl platform_update_domain: int, zone: Optional[str] = None, placement_group_id: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual machine scale set. @@ -2671,6 +2716,10 @@ def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disabl :rtype: ~azure.mgmt.compute.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2718,7 +2767,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disabl @distributed_trace def get_instance_view( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineScaleSetInstanceView: """Gets the status of a VM scale set instance. @@ -2731,6 +2780,10 @@ def get_instance_view( :rtype: ~azure.mgmt.compute.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2852,7 +2905,7 @@ def begin_update_instances( vm_instance_i_ds: _models.VirtualMachineScaleSetVMInstanceRequiredIDs, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -2880,7 +2933,7 @@ def begin_update_instances( vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -2906,7 +2959,7 @@ def begin_update_instances( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -2923,6 +2976,10 @@ def begin_update_instances( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2970,7 +3027,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def get_os_upgrade_history( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.UpgradeOperationHistoricalStatusInfo"]: """Gets list of OS upgrades on a VM scale set instance. @@ -2985,6 +3042,10 @@ def get_os_upgrade_history( ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.UpgradeOperationHistoricalStatusInfo] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3125,7 +3186,7 @@ def begin_perform_maintenance( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances which are not eligible for perform maintenance will be failed. Please refer to best practices @@ -3156,7 +3217,7 @@ def begin_perform_maintenance( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances which are not eligible for perform maintenance will be failed. Please refer to best practices @@ -3185,7 +3246,7 @@ def begin_perform_maintenance( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances which are not eligible for perform maintenance will be failed. Please refer to best practices @@ -3205,6 +3266,10 @@ def begin_perform_maintenance( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3337,7 +3402,7 @@ def begin_power_off( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release @@ -3372,7 +3437,7 @@ def begin_power_off( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release @@ -3405,7 +3470,7 @@ def begin_power_off( vm_scale_set_name: str, skip_shutdown: Optional[bool] = None, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still attached and you are getting charged for the resources. Instead, use deallocate to release @@ -3428,6 +3493,10 @@ def begin_power_off( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3529,7 +3598,7 @@ def _reapply_initial(self, resource_group_name: str, vm_scale_set_name: str, **k return deserialized # type: ignore @distributed_trace - def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Reapplies the Virtual Machine Scale Set Virtual Machine Profile to the Virtual Machine Instances. @@ -3542,6 +3611,10 @@ def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwar :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3667,7 +3740,7 @@ def begin_redeploy( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, and powers them back on. @@ -3696,7 +3769,7 @@ def begin_redeploy( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, and powers them back on. @@ -3723,7 +3796,7 @@ def begin_redeploy( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, and powers them back on. @@ -3741,6 +3814,10 @@ def begin_redeploy( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3870,7 +3947,7 @@ def begin_reimage( vm_scale_set_reimage_input: Optional[_models.VirtualMachineScaleSetReimageParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual @@ -3900,7 +3977,7 @@ def begin_reimage( vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual @@ -3927,7 +4004,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual @@ -3946,6 +4023,10 @@ def begin_reimage( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4075,7 +4156,7 @@ def begin_reimage_all( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This operation is only supported for managed disks. @@ -4104,7 +4185,7 @@ def begin_reimage_all( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This operation is only supported for managed disks. @@ -4131,7 +4212,7 @@ def begin_reimage_all( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This operation is only supported for managed disks. @@ -4149,6 +4230,10 @@ def begin_reimage_all( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4278,7 +4363,7 @@ def begin_restart( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -4306,7 +4391,7 @@ def begin_restart( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -4332,7 +4417,7 @@ def begin_restart( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -4349,6 +4434,10 @@ def begin_restart( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4474,7 +4563,7 @@ def begin_scale_out( parameters: _models.VMScaleSetScaleOutInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Scales out one or more virtual machines in a VM scale set. @@ -4501,7 +4590,7 @@ def begin_scale_out( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Scales out one or more virtual machines in a VM scale set. @@ -4526,7 +4615,7 @@ def begin_scale_out( resource_group_name: str, vm_scale_set_name: str, parameters: Union[_models.VMScaleSetScaleOutInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Scales out one or more virtual machines in a VM scale set. @@ -4542,6 +4631,10 @@ def begin_scale_out( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4666,7 +4759,7 @@ def begin_set_orchestration_service_state( parameters: _models.OrchestrationServiceStateInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -4693,7 +4786,7 @@ def begin_set_orchestration_service_state( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -4718,7 +4811,7 @@ def begin_set_orchestration_service_state( resource_group_name: str, vm_scale_set_name: str, parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -4734,6 +4827,10 @@ def begin_set_orchestration_service_state( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4781,7 +4878,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_skus( - self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any + self, resource_group_name: str, vm_scale_set_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachineScaleSetSku"]: """Gets a list of SKUs available for your VM scale set, including the minimum and maximum VM instances allowed for each SKU. @@ -4796,6 +4893,10 @@ def list_skus( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineScaleSetSku] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4936,7 +5037,7 @@ def begin_start( vm_instance_i_ds: Optional[_models.VirtualMachineScaleSetVMInstanceIDs] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -4964,7 +5065,7 @@ def begin_start( vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -4990,7 +5091,7 @@ def begin_start( resource_group_name: str, vm_scale_set_name: str, vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -5007,6 +5108,10 @@ def begin_start( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_sizes_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_sizes_operations.py index 9baafdfd05c3..757551b3fcfe 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_sizes_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machine_sizes_operations.py @@ -27,6 +27,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -83,7 +84,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, location: str, **kwargs: Any) -> ItemPaged["_models.VirtualMachineSize"]: + def list(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.VirtualMachineSize"]: """This API is deprecated. Use `Resources Skus `_. @@ -93,6 +94,10 @@ def list(self, location: str, **kwargs: Any) -> ItemPaged["_models.VirtualMachin :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machines_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machines_operations.py index 057e4e4f1ae1..8aa2cd7a35f8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machines_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/operations/_virtual_machines_operations.py @@ -33,6 +33,7 @@ from .. import models as _models from .._configuration import ComputeManagementClientConfiguration from .._utils.serialization import Deserializer, Serializer +from ..types import LRORequestParameters, RequestParameters T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] @@ -1019,7 +1020,7 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list_by_location(self, location: str, **kwargs: Any) -> ItemPaged["_models.VirtualMachine"]: + def list_by_location(self, location: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any) -> ItemPaged["_models.VirtualMachine"]: """Gets all the virtual machines under the specified subscription for the specified location. :param location: The name of Azure region. Required. @@ -1028,6 +1029,10 @@ def list_by_location(self, location: str, **kwargs: Any) -> ItemPaged["_models.V :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1090,7 +1095,7 @@ def list_all( status_only: Optional[str] = None, filter: Optional[str] = None, expand: Optional[Union[str, _models.ExpandTypesForListVMs]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. @@ -1111,6 +1116,10 @@ def list_all( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1175,7 +1184,7 @@ def list( resource_group_name: str, filter: Optional[str] = None, expand: Optional[Union[str, _models.ExpandTypeForListVMs]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified resource group. Use the nextLink property in the response to get the next page of virtual machines. @@ -1196,6 +1205,10 @@ def list( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1260,7 +1273,7 @@ def get( resource_group_name: str, vm_name: str, expand: Optional[Union[str, _models.InstanceViewTypes]] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachine: """Retrieves information about the model view or the instance view of a virtual machine. @@ -1279,6 +1292,10 @@ def get( :rtype: ~azure.mgmt.compute.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -1407,7 +1424,7 @@ def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -1446,7 +1463,7 @@ def begin_create_or_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -1483,7 +1500,7 @@ def begin_create_or_update( parameters: Union[_models.VirtualMachine, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -1509,6 +1526,10 @@ def begin_create_or_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1644,7 +1665,7 @@ def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -1682,7 +1703,7 @@ def begin_update( if_none_match: Optional[str] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -1718,7 +1739,7 @@ def begin_update( parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], if_match: Optional[str] = None, if_none_match: Optional[str] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -1743,6 +1764,10 @@ def begin_update( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1857,7 +1882,7 @@ def _delete_initial( @distributed_trace def begin_delete( - self, resource_group_name: str, vm_name: str, force_deletion: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, force_deletion: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to delete a virtual machine. @@ -1873,6 +1898,10 @@ def begin_delete( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1971,7 +2000,7 @@ def _assess_patches_initial(self, resource_group_name: str, vm_name: str, **kwar @distributed_trace def begin_assess_patches( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineAssessPatchesResult]: """Assess patches on the VM. @@ -1986,6 +2015,10 @@ def begin_assess_patches( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineAssessPatchesResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2111,7 +2144,7 @@ def begin_attach_detach_data_disks( parameters: _models.AttachDetachDataDisksRequest, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -2140,7 +2173,7 @@ def begin_attach_detach_data_disks( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -2167,7 +2200,7 @@ def begin_attach_detach_data_disks( resource_group_name: str, vm_name: str, parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -2184,6 +2217,10 @@ def begin_attach_detach_data_disks( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.StorageProfile] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2312,7 +2349,7 @@ def begin_capture( parameters: _models.VirtualMachineCaptureParameters, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used to create similar VMs. @@ -2341,7 +2378,7 @@ def begin_capture( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used to create similar VMs. @@ -2368,7 +2405,7 @@ def begin_capture( resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used to create similar VMs. @@ -2386,6 +2423,10 @@ def begin_capture( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineCaptureResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2491,7 +2532,7 @@ def _convert_to_managed_disks_initial( return deserialized # type: ignore @distributed_trace - def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Converts virtual machine disks from blob-based to managed disks. Virtual machine must be stop-deallocated before invoking this operation. @@ -2504,6 +2545,10 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2604,7 +2649,7 @@ def _deallocate_initial( @distributed_trace def begin_deallocate( - self, resource_group_name: str, vm_name: str, hibernate: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, hibernate: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down the virtual machine and releases the compute resources. You are not billed for the compute resources that this virtual machine uses. @@ -2620,6 +2665,10 @@ def begin_deallocate( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2665,7 +2714,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """Sets the OS state of the virtual machine to generalized. It is recommended to sysprep the virtual machine before performing this operation. For Windows, please refer to `Create a @@ -2683,6 +2732,10 @@ def generalize( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -2800,7 +2853,7 @@ def begin_install_patches( install_patches_input: _models.VirtualMachineInstallPatchesParameters, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -2830,7 +2883,7 @@ def begin_install_patches( install_patches_input: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -2858,7 +2911,7 @@ def begin_install_patches( resource_group_name: str, vm_name: str, install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -2877,6 +2930,10 @@ def begin_install_patches( ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.VirtualMachineInstallPatchesResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2928,7 +2985,7 @@ def get_long_running_output(pipeline_response): @distributed_trace def instance_view( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.VirtualMachineInstanceView: """Retrieves information about the run-time state of a virtual machine. @@ -2941,6 +2998,10 @@ def instance_view( :rtype: ~azure.mgmt.compute.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -3065,7 +3126,7 @@ def begin_migrate_to_vm_scale_set( parameters: Optional[_models.MigrateVMToVirtualMachineScaleSetInput] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Migrate a virtual machine from availability set to Flexible Virtual Machine Scale Set. @@ -3093,7 +3154,7 @@ def begin_migrate_to_vm_scale_set( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Migrate a virtual machine from availability set to Flexible Virtual Machine Scale Set. @@ -3119,7 +3180,7 @@ def begin_migrate_to_vm_scale_set( resource_group_name: str, vm_name: str, parameters: Optional[Union[_models.MigrateVMToVirtualMachineScaleSetInput, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Migrate a virtual machine from availability set to Flexible Virtual Machine Scale Set. @@ -3136,6 +3197,10 @@ def begin_migrate_to_vm_scale_set( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3236,7 +3301,7 @@ def _perform_maintenance_initial(self, resource_group_name: str, vm_name: str, * return deserialized # type: ignore @distributed_trace - def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """The operation to perform maintenance on a virtual machine. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -3248,6 +3313,10 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3348,7 +3417,7 @@ def _power_off_initial( @distributed_trace def begin_power_off( - self, resource_group_name: str, vm_name: str, skip_shutdown: Optional[bool] = None, **kwargs: Any + self, resource_group_name: str, vm_name: str, skip_shutdown: Optional[bool] = None, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """The operation to power off (stop) a virtual machine. The virtual machine can be restarted with the same provisioned resources. You are still charged for this virtual machine. @@ -3366,6 +3435,10 @@ def begin_power_off( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3463,7 +3536,7 @@ def _reapply_initial(self, resource_group_name: str, vm_name: str, **kwargs: Any return deserialized # type: ignore @distributed_trace - def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_reapply(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """The operation to reapply a virtual machine's state. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -3475,6 +3548,10 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3571,7 +3648,7 @@ def _redeploy_initial(self, resource_group_name: str, vm_name: str, **kwargs: An return deserialized # type: ignore @distributed_trace - def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_redeploy(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """Shuts down the virtual machine, moves it to a new node, and powers it back on. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -3583,6 +3660,10 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3708,7 +3789,7 @@ def begin_reimage( parameters: Optional[_models.VirtualMachineReimageParameters] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual machine is reset to initial @@ -3741,7 +3822,7 @@ def begin_reimage( parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual machine is reset to initial @@ -3772,7 +3853,7 @@ def begin_reimage( resource_group_name: str, vm_name: str, parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual machine is reset to initial @@ -3793,6 +3874,10 @@ def begin_reimage( :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3893,7 +3978,7 @@ def _restart_initial(self, resource_group_name: str, vm_name: str, **kwargs: Any return deserialized # type: ignore @distributed_trace - def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_restart(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """The operation to restart a virtual machine. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -3905,6 +3990,10 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3953,7 +4042,7 @@ def retrieve_boot_diagnostics_data( resource_group_name: str, vm_name: str, sas_uri_expiration_time_in_minutes: Optional[int] = None, - **kwargs: Any + request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> _models.RetrieveBootDiagnosticsDataResult: """The operation to retrieve SAS URIs for a virtual machine's boot diagnostic logs. @@ -3970,6 +4059,10 @@ def retrieve_boot_diagnostics_data( :rtype: ~azure.mgmt.compute.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4092,7 +4185,7 @@ def begin_run_command( parameters: _models.RunCommandInput, *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -4120,7 +4213,7 @@ def begin_run_command( parameters: IO[bytes], *, content_type: str = "application/json", - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -4146,7 +4239,7 @@ def begin_run_command( resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO[bytes]], - **kwargs: Any + request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -4163,6 +4256,10 @@ def begin_run_command( :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.models.RunCommandResult] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4214,7 +4311,7 @@ def get_long_running_output(pipeline_response): @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> None: """The operation to simulate the eviction of spot virtual machine. @@ -4227,6 +4324,10 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :rtype: None :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -4319,7 +4420,7 @@ def _start_initial(self, resource_group_name: str, vm_name: str, **kwargs: Any) return deserialized # type: ignore @distributed_trace - def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: + def begin_start(self, resource_group_name: str, vm_name: str, request_parameters: Optional[LRORequestParameters] = None, **kwargs: Any) -> LROPoller[None]: """The operation to start a virtual machine. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -4331,6 +4432,10 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4375,7 +4480,7 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- @distributed_trace def list_available_sizes( - self, resource_group_name: str, vm_name: str, **kwargs: Any + self, resource_group_name: str, vm_name: str, request_parameters: Optional[RequestParameters] = None, **kwargs: Any ) -> ItemPaged["_models.VirtualMachineSize"]: """Lists all available virtual machine sizes to which the specified virtual machine can be resized. @@ -4389,6 +4494,10 @@ def list_available_sizes( :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: """ + if request_parameters: + for _rp_key, _rp_value in request_parameters.items(): + kwargs.setdefault(_rp_key, _rp_value) + _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/types.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/types.py new file mode 100644 index 000000000000..3a6fd141285b --- /dev/null +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/types.py @@ -0,0 +1,176 @@ +# --------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# --------------------------------------------------------------------------- + +"""Typed dictionaries for service client request parameters. + +These TypedDict classes provide explicit typing for the infrastructure-level +parameters that control how HTTP requests are constructed and sent through +the Azure SDK pipeline. They replace the untyped ``**kwargs`` pattern for +better discoverability and type-checking support. +""" + +from typing import Callable, Dict, List, Union + +from typing_extensions import TypedDict + +from azure.core.pipeline import PipelineRequest, PipelineResponse +from azure.core.polling import PollingMethod + + +class RequestParameters(TypedDict, total=False): + """Common parameters for all service client operations. + + These parameters control how the HTTP request is constructed and sent + through the Azure SDK pipeline, including retry, redirect, tracing, + logging, and transport-level behavior. + """ + + # -- Request construction -- + + api_version: str + """Override the API version used for this request.""" + + headers: Dict[str, str] + """Custom HTTP headers to include in the request.""" + + params: Dict[str, str] + """Custom query parameters to include in the request.""" + + content_type: str + """Override the Content-Type header for the request.""" + + request_id: str + """Custom request ID to set in the request header + (consumed by :class:`~azure.core.pipeline.policies.RequestIdPolicy`).""" + + # -- Response handling -- + + stream: bool + """Whether to stream the response body instead of loading it into memory.""" + + decompress: bool + """Whether to decompress the response body. Defaults to ``True``.""" + + # -- Hooks -- + + raw_request_hook: Callable[[PipelineRequest], None] + """Callback invoked with the :class:`~azure.core.pipeline.PipelineRequest` + before the request is sent + (consumed by :class:`~azure.core.pipeline.policies.CustomHookPolicy`).""" + + raw_response_hook: Callable[[PipelineResponse], None] + """Callback invoked with the :class:`~azure.core.pipeline.PipelineResponse` + after the response is received + (consumed by :class:`~azure.core.pipeline.policies.CustomHookPolicy`).""" + + # -- Logging -- + + logging_enable: bool + """Enable HTTP logging for this request + (consumed by :class:`~azure.core.pipeline.policies.HttpLoggingPolicy`). + Defaults to ``False``.""" + + # -- Retry policy -- + + retry_total: int + """Total number of retries to allow + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``10``.""" + + retry_connect: int + """Number of connection-related retries + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``3``.""" + + retry_read: int + """Number of retries on read errors + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``3``.""" + + retry_status: int + """Number of retries on bad status codes + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``3``.""" + + retry_backoff_factor: float + """Backoff factor for retry delays + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``0.8``.""" + + retry_backoff_max: int + """Maximum backoff time in seconds + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``120``.""" + + retry_on_methods: List[str] + """HTTP methods that are eligible for retry + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`).""" + + timeout: int + """Overall timeout for the operation in seconds, including all retries + (consumed by :class:`~azure.core.pipeline.policies.RetryPolicy`). + Defaults to ``604800`` (7 days).""" + + # -- Redirect policy -- + + permit_redirects: bool + """Whether to follow redirects + (consumed by :class:`~azure.core.pipeline.policies.RedirectPolicy`). + Defaults to ``True``.""" + + redirect_max: int + """Maximum number of redirects to follow + (consumed by :class:`~azure.core.pipeline.policies.RedirectPolicy`). + Defaults to ``30``.""" + + # -- Distributed tracing -- + + network_span_namer: Callable[..., str] + """Callable that receives the HTTP request and returns a span name string + (consumed by :class:`~azure.core.pipeline.policies.DistributedTracingPolicy`).""" + + tracing_attributes: Dict[str, str] + """Extra attributes to set on the tracing span + (consumed by :class:`~azure.core.pipeline.policies.DistributedTracingPolicy`).""" + + # -- Transport / connection -- + + connection_timeout: float + """Connection timeout in seconds + (consumed by the HTTP transport).""" + + read_timeout: float + """Read timeout in seconds + (consumed by the HTTP transport).""" + + connection_verify: Union[bool, str] + """SSL certificate verification. Pass ``False`` to disable, + or a path to a CA bundle + (consumed by the HTTP transport).""" + + connection_cert: Union[str, tuple] + """Client-side certificate. A path to a file or a ``(cert, key)`` tuple + (consumed by the HTTP transport).""" + + proxy: str + """Proxy URL for this request + (consumed by the HTTP transport).""" + + +class LRORequestParameters(RequestParameters, total=False): + """Parameters for long-running operations (``begin_*`` methods). + + Extends :class:`RequestParameters` with polling and continuation support. + """ + + polling: Union[bool, PollingMethod] + """Polling strategy. ``True`` for default polling, ``False`` for no polling, + or a custom :class:`~azure.core.polling.PollingMethod` instance.""" + + polling_interval: int + """Time in seconds between polling requests. Overrides the client default.""" + + continuation_token: str + """Token to resume a previously started long-running operation."""