From d59bf2a89fdf46b272173b276c3d94c612563581 Mon Sep 17 00:00:00 2001 From: SDKAuto Date: Mon, 10 Apr 2023 20:51:09 +0000 Subject: [PATCH] CodeGen from PR 22789 in Azure/azure-rest-api-specs Merge 1bb6878b3227f270495cc4f0ecc15442b3d5eafc into 1f22d4dbd99b0fe347ad79e79d4eb1ed44a87291 --- .../azure-mgmt-policyinsights/_meta.json | 6 +- .../policyinsights/_policy_insights_client.py | 4 +- .../mgmt/policyinsights/_serialization.py | 93 ++++-- .../azure/mgmt/policyinsights/_vendor.py | 5 +- .../azure/mgmt/policyinsights/_version.py | 2 +- .../aio/_policy_insights_client.py | 4 +- .../operations/_attestations_operations.py | 80 ++--- .../_component_policy_states_operations.py | 30 +- .../aio/operations/_operations.py | 10 +- .../operations/_policy_events_operations.py | 40 ++- .../operations/_policy_metadata_operations.py | 15 +- .../_policy_restrictions_operations.py | 27 +- .../operations/_policy_states_operations.py | 94 +++--- .../_policy_tracked_resources_operations.py | 28 +- .../operations/_remediations_operations.py | 132 ++++---- .../mgmt/policyinsights/models/__init__.py | 4 + .../mgmt/policyinsights/models/_models_py3.py | 292 ++++++++++++------ .../models/_policy_insights_client_enums.py | 20 +- .../operations/_attestations_operations.py | 104 ++++--- .../_component_policy_states_operations.py | 42 +-- .../policyinsights/operations/_operations.py | 12 +- .../operations/_policy_events_operations.py | 72 +++-- .../operations/_policy_metadata_operations.py | 19 +- .../_policy_restrictions_operations.py | 33 +- .../operations/_policy_states_operations.py | 146 +++++---- .../_policy_tracked_resources_operations.py | 44 +-- .../operations/_remediations_operations.py | 180 ++++++----- ...ictions_check_at_management_group_scope.py | 2 +- ...trictions_check_at_resource_group_scope.py | 2 +- ...source_group_scope_include_audit_effect.py | 52 ++++ ...estrictions_check_at_subscription_scope.py | 2 +- ...subscription_scope_include_audit_effect.py | 51 +++ 32 files changed, 1004 insertions(+), 643 deletions(-) create mode 100644 sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope_include_audit_effect.py create mode 100644 sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope_include_audit_effect.py diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/_meta.json b/sdk/policyinsights/azure-mgmt-policyinsights/_meta.json index 0839a738a4e9..4664461e7cb3 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/_meta.json +++ b/sdk/policyinsights/azure-mgmt-policyinsights/_meta.json @@ -1,11 +1,11 @@ { - "commit": "f38115ac455af89493b0a0719d9a987404560dda", + "commit": "025ca2395b5eb78212fd89d6470ddf6c96415d1d", "repository_url": "https://github.com/Azure/azure-rest-api-specs", "autorest": "3.9.2", "use": [ - "@autorest/python@6.2.7", + "@autorest/python@6.4.7", "@autorest/modelerfour@4.24.3" ], - "autorest_command": "autorest specification/policyinsights/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.2.7 --use=@autorest/modelerfour@4.24.3 --version=3.9.2 --version-tolerant=False", + "autorest_command": "autorest specification/policyinsights/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/mnt/vss/_work/1/s/azure-sdk-for-python/sdk --use=@autorest/python@6.4.7 --use=@autorest/modelerfour@4.24.3 --version=3.9.2 --version-tolerant=False", "readme": "specification/policyinsights/resource-manager/readme.md" } \ No newline at end of file diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_policy_insights_client.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_policy_insights_client.py index 70b828c54000..61220fc11884 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_policy_insights_client.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_policy_insights_client.py @@ -75,7 +75,7 @@ def __init__( self._config = PolicyInsightsClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client = ARMPipelineClient(base_url=base_url, config=self._config, **kwargs) + self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **kwargs) client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) @@ -126,5 +126,5 @@ def __enter__(self) -> "PolicyInsightsClient": self._client.__enter__() return self - def __exit__(self, *exc_details) -> None: + def __exit__(self, *exc_details: Any) -> None: self._client.__exit__(*exc_details) diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_serialization.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_serialization.py index 2c170e28dbca..842ae727fbbc 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_serialization.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_serialization.py @@ -38,7 +38,22 @@ import re import sys import codecs -from typing import Optional, Union, AnyStr, IO, Mapping +from typing import ( + Dict, + Any, + cast, + Optional, + Union, + AnyStr, + IO, + Mapping, + Callable, + TypeVar, + MutableMapping, + Type, + List, + Mapping, +) try: from urllib import quote # type: ignore @@ -48,12 +63,14 @@ import isodate # type: ignore -from typing import Dict, Any, cast - from azure.core.exceptions import DeserializationError, SerializationError, raise_with_traceback +from azure.core.serialization import NULL as AzureCoreNull _BOM = codecs.BOM_UTF8.decode(encoding="utf-8") +ModelType = TypeVar("ModelType", bound="Model") +JSON = MutableMapping[str, Any] + class RawDeserializer: @@ -277,8 +294,8 @@ class Model(object): _attribute_map: Dict[str, Dict[str, Any]] = {} _validation: Dict[str, Dict[str, Any]] = {} - def __init__(self, **kwargs): - self.additional_properties = {} + def __init__(self, **kwargs: Any) -> None: + self.additional_properties: Dict[str, Any] = {} for k in kwargs: if k not in self._attribute_map: _LOGGER.warning("%s is not a known attribute of class %s and will be ignored", k, self.__class__) @@ -287,25 +304,25 @@ def __init__(self, **kwargs): else: setattr(self, k, kwargs[k]) - def __eq__(self, other): + def __eq__(self, other: Any) -> bool: """Compare objects by comparing all attributes.""" if isinstance(other, self.__class__): return self.__dict__ == other.__dict__ return False - def __ne__(self, other): + def __ne__(self, other: Any) -> bool: """Compare objects by comparing all attributes.""" return not self.__eq__(other) - def __str__(self): + def __str__(self) -> str: return str(self.__dict__) @classmethod - def enable_additional_properties_sending(cls): + def enable_additional_properties_sending(cls) -> None: cls._attribute_map["additional_properties"] = {"key": "", "type": "{object}"} @classmethod - def is_xml_model(cls): + def is_xml_model(cls) -> bool: try: cls._xml_map # type: ignore except AttributeError: @@ -322,7 +339,7 @@ def _create_xml_node(cls): return _create_xml_node(xml_map.get("name", cls.__name__), xml_map.get("prefix", None), xml_map.get("ns", None)) - def serialize(self, keep_readonly=False, **kwargs): + def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: """Return the JSON that would be sent to azure from this model. This is an alias to `as_dict(full_restapi_key_transformer, keep_readonly=False)`. @@ -336,8 +353,13 @@ def serialize(self, keep_readonly=False, **kwargs): serializer = Serializer(self._infer_class_models()) return serializer._serialize(self, keep_readonly=keep_readonly, **kwargs) - def as_dict(self, keep_readonly=True, key_transformer=attribute_transformer, **kwargs): - """Return a dict that can be JSONify using json.dump. + def as_dict( + self, + keep_readonly: bool = True, + key_transformer: Callable[[str, Dict[str, Any], Any], Any] = attribute_transformer, + **kwargs: Any + ) -> JSON: + """Return a dict that can be serialized using json.dump. Advanced usage might optionally use a callback as parameter: @@ -384,7 +406,7 @@ def _infer_class_models(cls): return client_models @classmethod - def deserialize(cls, data, content_type=None): + def deserialize(cls: Type[ModelType], data: Any, content_type: Optional[str] = None) -> ModelType: """Parse a str using the RestAPI syntax and return a model. :param str data: A str using RestAPI structure. JSON by default. @@ -396,7 +418,12 @@ def deserialize(cls, data, content_type=None): return deserializer(cls.__name__, data, content_type=content_type) @classmethod - def from_dict(cls, data, key_extractors=None, content_type=None): + def from_dict( + cls: Type[ModelType], + data: Any, + key_extractors: Optional[Callable[[str, Dict[str, Any], Any], Any]] = None, + content_type: Optional[str] = None, + ) -> ModelType: """Parse a dict using given key extractor return a model. By default consider key @@ -409,8 +436,8 @@ def from_dict(cls, data, key_extractors=None, content_type=None): :raises: DeserializationError if something went wrong """ deserializer = Deserializer(cls._infer_class_models()) - deserializer.key_extractors = ( - [ + deserializer.key_extractors = ( # type: ignore + [ # type: ignore attribute_key_case_insensitive_extractor, rest_key_case_insensitive_extractor, last_rest_key_case_insensitive_extractor, @@ -518,7 +545,7 @@ class Serializer(object): "multiple": lambda x, y: x % y != 0, } - def __init__(self, classes=None): + def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]] = None): self.serialize_type = { "iso-8601": Serializer.serialize_iso, "rfc-1123": Serializer.serialize_rfc, @@ -534,7 +561,7 @@ def __init__(self, classes=None): "[]": self.serialize_iter, "{}": self.serialize_dict, } - self.dependencies = dict(classes) if classes else {} + self.dependencies: Dict[str, Type[ModelType]] = dict(classes) if classes else {} self.key_transformer = full_restapi_key_transformer self.client_side_validation = True @@ -602,7 +629,7 @@ def _serialize(self, target_obj, data_type=None, **kwargs): if xml_desc.get("attr", False): if xml_ns: ET.register_namespace(xml_prefix, xml_ns) - xml_name = "{}{}".format(xml_ns, xml_name) + xml_name = "{{{}}}{}".format(xml_ns, xml_name) serialized.set(xml_name, new_attr) # type: ignore continue if xml_desc.get("text", False): @@ -626,8 +653,7 @@ def _serialize(self, target_obj, data_type=None, **kwargs): serialized.append(local_node) # type: ignore else: # JSON for k in reversed(keys): # type: ignore - unflattened = {k: new_attr} - new_attr = unflattened + new_attr = {k: new_attr} _new_attr = new_attr _serialized = serialized @@ -656,8 +682,8 @@ def body(self, data, data_type, **kwargs): """ # Just in case this is a dict - internal_data_type = data_type.strip("[]{}") - internal_data_type = self.dependencies.get(internal_data_type, None) + internal_data_type_str = data_type.strip("[]{}") + internal_data_type = self.dependencies.get(internal_data_type_str, None) try: is_xml_model_serialization = kwargs["is_xml"] except KeyError: @@ -777,6 +803,8 @@ def serialize_data(self, data, data_type, **kwargs): raise ValueError("No value for given attribute") try: + if data is AzureCoreNull: + return None if data_type in self.basic_types.values(): return self.serialize_basic(data, data_type, **kwargs) @@ -1161,7 +1189,8 @@ def rest_key_extractor(attr, attr_desc, data): working_data = data while "." in key: - dict_keys = _FLATTEN.split(key) + # Need the cast, as for some reasons "split" is typed as list[str | Any] + dict_keys = cast(List[str], _FLATTEN.split(key)) if len(dict_keys) == 1: key = _decode_attribute_map_key(dict_keys[0]) break @@ -1242,7 +1271,7 @@ def _extract_name_from_internal_type(internal_type): xml_name = internal_type_xml_map.get("name", internal_type.__name__) xml_ns = internal_type_xml_map.get("ns", None) if xml_ns: - xml_name = "{}{}".format(xml_ns, xml_name) + xml_name = "{{{}}}{}".format(xml_ns, xml_name) return xml_name @@ -1266,7 +1295,7 @@ def xml_key_extractor(attr, attr_desc, data): # Integrate namespace if necessary xml_ns = xml_desc.get("ns", internal_type_xml_map.get("ns", None)) if xml_ns: - xml_name = "{}{}".format(xml_ns, xml_name) + xml_name = "{{{}}}{}".format(xml_ns, xml_name) # If it's an attribute, that's simple if xml_desc.get("attr", False): @@ -1332,7 +1361,7 @@ class Deserializer(object): valid_date = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}" r"\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") - def __init__(self, classes=None): + def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]] = None): self.deserialize_type = { "iso-8601": Deserializer.deserialize_iso, "rfc-1123": Deserializer.deserialize_rfc, @@ -1352,7 +1381,7 @@ def __init__(self, classes=None): "duration": (isodate.Duration, datetime.timedelta), "iso-8601": (datetime.datetime), } - self.dependencies = dict(classes) if classes else {} + self.dependencies: Dict[str, Type[ModelType]] = dict(classes) if classes else {} self.key_extractors = [rest_key_extractor, xml_key_extractor] # Additional properties only works if the "rest_key_extractor" is used to # extract the keys. Making it to work whatever the key extractor is too much @@ -1471,7 +1500,7 @@ def _classify_target(self, target, data): Once classification has been determined, initialize object. :param str target: The target object type to deserialize to. - :param str/dict data: The response data to deseralize. + :param str/dict data: The response data to deserialize. """ if target is None: return None, None @@ -1486,7 +1515,7 @@ def _classify_target(self, target, data): target = target._classify(data, self.dependencies) except AttributeError: pass # Target is not a Model, no classify - return target, target.__class__.__name__ + return target, target.__class__.__name__ # type: ignore def failsafe_deserialize(self, target_obj, data, content_type=None): """Ignores any errors encountered in deserialization, @@ -1496,7 +1525,7 @@ def failsafe_deserialize(self, target_obj, data, content_type=None): a deserialization error. :param str target_obj: The target object type to deserialize to. - :param str/dict data: The response data to deseralize. + :param str/dict data: The response data to deserialize. :param str content_type: Swagger "produces" if available. """ try: diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_vendor.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_vendor.py index 9aad73fc743e..bd0df84f5319 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_vendor.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_vendor.py @@ -5,6 +5,8 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import List, cast + from azure.core.pipeline.transport import HttpRequest @@ -22,6 +24,7 @@ def _format_url_section(template, **kwargs): try: return template.format(**kwargs) except KeyError as key: - formatted_components = template.split("/") + # Need the cast, as for some reasons "split" is typed as list[str | Any] + formatted_components = cast(List[str], template.split("/")) components = [c for c in formatted_components if "{}".format(key.args[0]) not in c] template = "/".join(components) diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_version.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_version.py index a741dc7b69b5..e5754a47ce68 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_version.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/_version.py @@ -6,4 +6,4 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "1.1.0b4" +VERSION = "1.0.0b1" diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/_policy_insights_client.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/_policy_insights_client.py index cbbc2695b40e..7050a61a2c3c 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/_policy_insights_client.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/_policy_insights_client.py @@ -76,7 +76,7 @@ def __init__( self._config = PolicyInsightsClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client = AsyncARMPipelineClient(base_url=base_url, config=self._config, **kwargs) + self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **kwargs) client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) @@ -127,5 +127,5 @@ async def __aenter__(self) -> "PolicyInsightsClient": await self._client.__aenter__() return self - async def __aexit__(self, *exc_details) -> None: + async def __aexit__(self, *exc_details: Any) -> None: await self._client.__aexit__(*exc_details) diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_attestations_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_attestations_operations.py index 1e9c5d0d72b7..a99977f3acb7 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_attestations_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_attestations_operations.py @@ -6,7 +6,6 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys from typing import Any, AsyncIterable, Callable, Dict, IO, Optional, TypeVar, Union, cast, overload from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -45,10 +44,6 @@ build_list_for_subscription_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -88,7 +83,7 @@ def list_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.AttestationListResult] = kwargs.pop("cls", None) error_map = { @@ -136,8 +131,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -168,7 +164,7 @@ async def _create_or_update_at_subscription_initial( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) @@ -194,8 +190,9 @@ async def _create_or_update_at_subscription_initial( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -287,7 +284,8 @@ async def begin_create_or_update_at_subscription( :param attestation_name: The name of the attestation. Required. :type attestation_name: str - :param parameters: The attestation parameters. Is either a model type or a IO type. Required. + :param parameters: The attestation parameters. Is either a Attestation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Attestation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -308,7 +306,7 @@ async def begin_create_or_update_at_subscription( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) @@ -374,7 +372,7 @@ async def get_at_subscription(self, attestation_name: str, **kwargs: Any) -> _mo _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) request = build_get_at_subscription_request( @@ -388,8 +386,9 @@ async def get_at_subscription(self, attestation_name: str, **kwargs: Any) -> _mo request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -434,7 +433,7 @@ async def delete_at_subscription( # pylint: disable=inconsistent-return-stateme _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_delete_at_subscription_request( @@ -448,8 +447,9 @@ async def delete_at_subscription( # pylint: disable=inconsistent-return-stateme request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -485,7 +485,7 @@ def list_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.AttestationListResult] = kwargs.pop("cls", None) error_map = { @@ -534,8 +534,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -566,7 +567,7 @@ async def _create_or_update_at_resource_group_initial( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) @@ -593,8 +594,9 @@ async def _create_or_update_at_resource_group_initial( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -702,7 +704,8 @@ async def begin_create_or_update_at_resource_group( :type resource_group_name: str :param attestation_name: The name of the attestation. Required. :type attestation_name: str - :param parameters: The attestation parameters. Is either a model type or a IO type. Required. + :param parameters: The attestation parameters. Is either a Attestation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Attestation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -723,7 +726,7 @@ async def begin_create_or_update_at_resource_group( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) @@ -795,7 +798,7 @@ async def get_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) request = build_get_at_resource_group_request( @@ -810,8 +813,9 @@ async def get_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -859,7 +863,7 @@ async def delete_at_resource_group( # pylint: disable=inconsistent-return-state _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_delete_at_resource_group_request( @@ -874,8 +878,9 @@ async def delete_at_resource_group( # pylint: disable=inconsistent-return-state request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -910,7 +915,7 @@ def list_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.AttestationListResult] = kwargs.pop("cls", None) error_map = { @@ -958,8 +963,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -988,7 +994,7 @@ async def _create_or_update_at_resource_initial( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) @@ -1014,8 +1020,9 @@ async def _create_or_update_at_resource_initial( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1120,7 +1127,8 @@ async def begin_create_or_update_at_resource( :type resource_id: str :param attestation_name: The name of the attestation. Required. :type attestation_name: str - :param parameters: The attestation parameters. Is either a model type or a IO type. Required. + :param parameters: The attestation parameters. Is either a Attestation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Attestation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -1141,7 +1149,7 @@ async def begin_create_or_update_at_resource( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) @@ -1210,7 +1218,7 @@ async def get_at_resource(self, resource_id: str, attestation_name: str, **kwarg _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) request = build_get_at_resource_request( @@ -1224,8 +1232,9 @@ async def get_at_resource(self, resource_id: str, attestation_name: str, **kwarg request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1272,7 +1281,7 @@ async def delete_at_resource( # pylint: disable=inconsistent-return-statements _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_delete_at_resource_request( @@ -1286,8 +1295,9 @@ async def delete_at_resource( # pylint: disable=inconsistent-return-statements request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_component_policy_states_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_component_policy_states_operations.py index 5fe586f6804f..0abe04aa3b0b 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_component_policy_states_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_component_policy_states_operations.py @@ -121,7 +121,7 @@ async def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_subscription_request( @@ -142,8 +142,9 @@ async def list_query_results_for_subscription( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -225,7 +226,7 @@ async def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_resource_group_request( @@ -247,8 +248,9 @@ async def list_query_results_for_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -330,7 +332,7 @@ async def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_resource_request( @@ -352,8 +354,9 @@ async def list_query_results_for_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -442,7 +445,7 @@ async def list_query_results_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_policy_definition_request( @@ -465,8 +468,9 @@ async def list_query_results_for_policy_definition( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -555,7 +559,7 @@ async def list_query_results_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_subscription_level_policy_assignment_request( @@ -578,8 +582,9 @@ async def list_query_results_for_subscription_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -671,7 +676,7 @@ async def list_query_results_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_resource_group_level_policy_assignment_request( @@ -695,8 +700,9 @@ async def list_query_results_for_resource_group_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_operations.py index d0b493314b59..6375cbccb399 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_operations.py @@ -6,7 +6,6 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( @@ -28,10 +27,6 @@ from ..._vendor import _convert_request from ...operations._operations import build_list_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -75,7 +70,7 @@ async def list(self, **kwargs: Any) -> _models.OperationsListResults: _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.OperationsListResults] = kwargs.pop("cls", None) request = build_list_request( @@ -87,8 +82,9 @@ async def list(self, **kwargs: Any) -> _models.OperationsListResults: request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_events_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_events_operations.py index 2ca681465d66..320a274fc9eb 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_events_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_events_operations.py @@ -98,7 +98,7 @@ def list_query_results_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -191,8 +191,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -234,7 +235,7 @@ def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -326,8 +327,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -372,7 +374,7 @@ def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -465,8 +467,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -508,7 +511,7 @@ def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -605,8 +608,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -658,7 +662,7 @@ def list_query_results_for_policy_set_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -752,8 +756,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -805,7 +810,7 @@ def list_query_results_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -899,8 +904,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -952,7 +958,7 @@ def list_query_results_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1046,8 +1052,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1102,7 +1109,7 @@ def list_query_results_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1197,8 +1204,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_metadata_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_metadata_operations.py index 7815466c8bd8..61595d8d12e2 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_metadata_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_metadata_operations.py @@ -6,7 +6,6 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -30,10 +29,6 @@ from ..._vendor import _convert_request from ...operations._policy_metadata_operations import build_get_resource_request, build_list_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -79,7 +74,7 @@ async def get_resource(self, resource_name: str, **kwargs: Any) -> _models.Polic _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyMetadata] = kwargs.pop("cls", None) request = build_get_resource_request( @@ -92,8 +87,9 @@ async def get_resource(self, resource_name: str, **kwargs: Any) -> _models.Polic request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -129,7 +125,7 @@ def list( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyMetadataCollection] = kwargs.pop("cls", None) error_map = { @@ -173,8 +169,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_restrictions_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_restrictions_operations.py index 513f60ca926c..a9ab87d791be 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_restrictions_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_restrictions_operations.py @@ -99,8 +99,8 @@ async def check_at_subscription_scope( ) -> _models.CheckRestrictionsResult: """Checks what restrictions Azure Policy will place on a resource within a subscription. - :param parameters: The check policy restrictions parameters. Is either a model type or a IO - type. Required. + :param parameters: The check policy restrictions parameters. Is either a + CheckRestrictionsRequest type or a IO type. Required. :type parameters: ~azure.mgmt.policyinsights.models.CheckRestrictionsRequest or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -121,7 +121,7 @@ async def check_at_subscription_scope( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.CheckRestrictionsResult] = kwargs.pop("cls", None) @@ -146,8 +146,9 @@ async def check_at_subscription_scope( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -225,8 +226,8 @@ async def check_at_resource_group_scope( :param resource_group_name: The name of the resource group. The name is case insensitive. Required. :type resource_group_name: str - :param parameters: The check policy restrictions parameters. Is either a model type or a IO - type. Required. + :param parameters: The check policy restrictions parameters. Is either a + CheckRestrictionsRequest type or a IO type. Required. :type parameters: ~azure.mgmt.policyinsights.models.CheckRestrictionsRequest or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -247,7 +248,7 @@ async def check_at_resource_group_scope( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.CheckRestrictionsResult] = kwargs.pop("cls", None) @@ -273,8 +274,9 @@ async def check_at_resource_group_scope( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -357,8 +359,8 @@ async def check_at_management_group_scope( :param management_group_id: Management group ID. Required. :type management_group_id: str - :param parameters: The check policy restrictions parameters. Is either a model type or a IO - type. Required. + :param parameters: The check policy restrictions parameters. Is either a + CheckManagementGroupRestrictionsRequest type or a IO type. Required. :type parameters: ~azure.mgmt.policyinsights.models.CheckManagementGroupRestrictionsRequest or IO :keyword management_groups_namespace: The namespace for Microsoft Management RP; only @@ -387,7 +389,7 @@ async def check_at_management_group_scope( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.CheckRestrictionsResult] = kwargs.pop("cls", None) @@ -413,8 +415,9 @@ async def check_at_management_group_scope( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_states_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_states_operations.py index 4bcf5cfe0b96..2a6ec8ff1ffe 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_states_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_states_operations.py @@ -112,7 +112,7 @@ def list_query_results_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -205,8 +205,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -265,7 +266,7 @@ async def summarize_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -294,8 +295,9 @@ async def summarize_for_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -342,7 +344,7 @@ def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -434,8 +436,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -487,7 +490,7 @@ async def summarize_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -515,8 +518,9 @@ async def summarize_for_subscription( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -566,7 +570,7 @@ def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -659,8 +663,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -715,7 +720,7 @@ async def summarize_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -744,8 +749,9 @@ async def summarize_for_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -792,7 +798,7 @@ def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -889,8 +895,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -942,7 +949,7 @@ async def summarize_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -970,8 +977,9 @@ async def summarize_for_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1006,7 +1014,7 @@ async def _trigger_subscription_evaluation_initial( # pylint: disable=inconsist _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_trigger_subscription_evaluation_request( @@ -1019,8 +1027,9 @@ async def _trigger_subscription_evaluation_initial( # pylint: disable=inconsist request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1058,7 +1067,7 @@ async def begin_trigger_subscription_evaluation(self, subscription_id: str, **kw _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) @@ -1113,7 +1122,7 @@ async def _trigger_resource_group_evaluation_initial( # pylint: disable=inconsi _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_trigger_resource_group_evaluation_request( @@ -1127,8 +1136,9 @@ async def _trigger_resource_group_evaluation_initial( # pylint: disable=inconsi request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1170,7 +1180,7 @@ async def begin_trigger_resource_group_evaluation( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) @@ -1248,7 +1258,7 @@ def list_query_results_for_policy_set_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1342,8 +1352,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1405,7 +1416,7 @@ async def summarize_for_policy_set_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -1435,8 +1446,9 @@ async def summarize_for_policy_set_definition( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1493,7 +1505,7 @@ def list_query_results_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1587,8 +1599,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1650,7 +1663,7 @@ async def summarize_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -1680,8 +1693,9 @@ async def summarize_for_policy_definition( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1738,7 +1752,7 @@ def list_query_results_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1832,8 +1846,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1895,7 +1910,7 @@ async def summarize_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -1925,8 +1940,9 @@ async def summarize_for_subscription_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1986,7 +2002,7 @@ def list_query_results_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -2081,8 +2097,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2147,7 +2164,7 @@ async def summarize_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -2178,8 +2195,9 @@ async def summarize_for_resource_group_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_tracked_resources_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_tracked_resources_operations.py index 32e7309c298e..779066762230 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_tracked_resources_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_policy_tracked_resources_operations.py @@ -96,9 +96,7 @@ def list_query_results_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -148,8 +146,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -191,9 +190,7 @@ def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -242,8 +239,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -288,9 +286,7 @@ def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -340,8 +336,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -386,9 +383,7 @@ def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -437,8 +432,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_remediations_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_remediations_operations.py index 5057a838e275..151fa448480b 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_remediations_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/aio/operations/_remediations_operations.py @@ -115,7 +115,7 @@ def list_deployments_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -162,8 +162,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -213,7 +214,7 @@ async def cancel_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_management_group_request( @@ -228,8 +229,9 @@ async def cancel_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -275,7 +277,7 @@ def list_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -324,8 +326,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -414,7 +417,8 @@ async def create_or_update_at_management_group( :type management_group_id: str :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword management_groups_namespace: The namespace for Microsoft Management RP; only "Microsoft.Management" is allowed. Default value is "Microsoft.Management". Note that @@ -442,7 +446,7 @@ async def create_or_update_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -469,8 +473,9 @@ async def create_or_update_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -528,7 +533,7 @@ async def get_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_management_group_request( @@ -543,8 +548,9 @@ async def get_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -598,7 +604,7 @@ async def delete_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_management_group_request( @@ -613,8 +619,9 @@ async def delete_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -657,7 +664,7 @@ def list_deployments_at_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -703,8 +710,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -743,7 +751,7 @@ async def cancel_at_subscription(self, remediation_name: str, **kwargs: Any) -> _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_subscription_request( @@ -757,8 +765,9 @@ async def cancel_at_subscription(self, remediation_name: str, **kwargs: Any) -> request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -795,7 +804,7 @@ def list_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -843,8 +852,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -912,7 +922,8 @@ async def create_or_update_at_subscription( :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -933,7 +944,7 @@ async def create_or_update_at_subscription( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -959,8 +970,9 @@ async def create_or_update_at_subscription( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1007,7 +1019,7 @@ async def get_at_subscription(self, remediation_name: str, **kwargs: Any) -> _mo _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_subscription_request( @@ -1021,8 +1033,9 @@ async def get_at_subscription(self, remediation_name: str, **kwargs: Any) -> _mo request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1065,7 +1078,7 @@ async def delete_at_subscription(self, remediation_name: str, **kwargs: Any) -> _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_subscription_request( @@ -1079,8 +1092,9 @@ async def delete_at_subscription(self, remediation_name: str, **kwargs: Any) -> request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1129,7 +1143,7 @@ def list_deployments_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -1176,8 +1190,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1220,7 +1235,7 @@ async def cancel_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_resource_group_request( @@ -1235,8 +1250,9 @@ async def cancel_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1275,7 +1291,7 @@ def list_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -1324,8 +1340,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1406,7 +1423,8 @@ async def create_or_update_at_resource_group( :type resource_group_name: str :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -1427,7 +1445,7 @@ async def create_or_update_at_resource_group( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -1454,8 +1472,9 @@ async def create_or_update_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1506,7 +1525,7 @@ async def get_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_resource_group_request( @@ -1521,8 +1540,9 @@ async def get_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1569,7 +1589,7 @@ async def delete_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_resource_group_request( @@ -1584,8 +1604,9 @@ async def delete_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1634,7 +1655,7 @@ def list_deployments_at_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -1680,8 +1701,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1722,7 +1744,7 @@ async def cancel_at_resource(self, resource_id: str, remediation_name: str, **kw _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_resource_request( @@ -1736,8 +1758,9 @@ async def cancel_at_resource(self, resource_id: str, remediation_name: str, **kw request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1776,7 +1799,7 @@ def list_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -1824,8 +1847,9 @@ async def extract_data(pipeline_response): async def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1904,7 +1928,8 @@ async def create_or_update_at_resource( :type resource_id: str :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -1925,7 +1950,7 @@ async def create_or_update_at_resource( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -1951,8 +1976,9 @@ async def create_or_update_at_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2001,7 +2027,7 @@ async def get_at_resource(self, resource_id: str, remediation_name: str, **kwarg _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_resource_request( @@ -2015,8 +2041,9 @@ async def get_at_resource(self, resource_id: str, remediation_name: str, **kwarg request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2063,7 +2090,7 @@ async def delete_at_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_resource_request( @@ -2077,8 +2104,9 @@ async def delete_at_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/__init__.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/__init__.py index 00583df241b2..123c9affb52a 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/__init__.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/__init__.py @@ -10,6 +10,7 @@ from ._models_py3 import AttestationEvidence from ._models_py3 import AttestationListResult from ._models_py3 import CheckManagementGroupRestrictionsRequest +from ._models_py3 import CheckRestrictionEvaluationDetails from ._models_py3 import CheckRestrictionsRequest from ._models_py3 import CheckRestrictionsResourceDetails from ._models_py3 import CheckRestrictionsResult @@ -38,6 +39,7 @@ from ._models_py3 import PolicyAssignmentSummary from ._models_py3 import PolicyDefinitionSummary from ._models_py3 import PolicyDetails +from ._models_py3 import PolicyEffectDetails from ._models_py3 import PolicyEvaluationDetails from ._models_py3 import PolicyEvaluationResult from ._models_py3 import PolicyEvent @@ -89,6 +91,7 @@ "AttestationEvidence", "AttestationListResult", "CheckManagementGroupRestrictionsRequest", + "CheckRestrictionEvaluationDetails", "CheckRestrictionsRequest", "CheckRestrictionsResourceDetails", "CheckRestrictionsResult", @@ -117,6 +120,7 @@ "PolicyAssignmentSummary", "PolicyDefinitionSummary", "PolicyDetails", + "PolicyEffectDetails", "PolicyEvaluationDetails", "PolicyEvaluationResult", "PolicyEvent", diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_models_py3.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_models_py3.py index 4e688b25229a..4e8dd5c223a1 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_models_py3.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_models_py3.py @@ -51,7 +51,7 @@ class Resource(_serialization.Model): "type": {"key": "type", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.id = None @@ -148,8 +148,8 @@ def __init__( evidence: Optional[List["_models.AttestationEvidence"]] = None, assessment_date: Optional[datetime.datetime] = None, metadata: Optional[JSON] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword policy_assignment_id: The resource ID of the policy assignment that the attestation is setting the state for. Required. @@ -205,7 +205,7 @@ class AttestationEvidence(_serialization.Model): "source_uri": {"key": "sourceUri", "type": "str"}, } - def __init__(self, *, description: Optional[str] = None, source_uri: Optional[str] = None, **kwargs): + def __init__(self, *, description: Optional[str] = None, source_uri: Optional[str] = None, **kwargs: Any) -> None: """ :keyword description: The description for this piece of evidence. :paramtype description: str @@ -238,7 +238,7 @@ class AttestationListResult(_serialization.Model): "next_link": {"key": "nextLink", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.value = None @@ -265,8 +265,8 @@ def __init__( *, resource_details: Optional["_models.CheckRestrictionsResourceDetails"] = None, pending_fields: Optional[List["_models.PendingField"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword resource_details: The information about the resource that will be evaluated. :paramtype resource_details: ~azure.mgmt.policyinsights.models.CheckRestrictionsResourceDetails @@ -279,6 +279,51 @@ def __init__( self.pending_fields = pending_fields +class CheckRestrictionEvaluationDetails(_serialization.Model): + """Policy evaluation details. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar evaluated_expressions: Details of the evaluated expressions. + :vartype evaluated_expressions: + list[~azure.mgmt.policyinsights.models.ExpressionEvaluationDetails] + :ivar if_not_exists_details: Evaluation details of IfNotExists effect. + :vartype if_not_exists_details: ~azure.mgmt.policyinsights.models.IfNotExistsEvaluationDetails + :ivar reason: The reason for the evaluation result. + :vartype reason: str + """ + + _validation = { + "reason": {"readonly": True}, + } + + _attribute_map = { + "evaluated_expressions": {"key": "evaluatedExpressions", "type": "[ExpressionEvaluationDetails]"}, + "if_not_exists_details": {"key": "ifNotExistsDetails", "type": "IfNotExistsEvaluationDetails"}, + "reason": {"key": "reason", "type": "str"}, + } + + def __init__( + self, + *, + evaluated_expressions: Optional[List["_models.ExpressionEvaluationDetails"]] = None, + if_not_exists_details: Optional["_models.IfNotExistsEvaluationDetails"] = None, + **kwargs: Any + ) -> None: + """ + :keyword evaluated_expressions: Details of the evaluated expressions. + :paramtype evaluated_expressions: + list[~azure.mgmt.policyinsights.models.ExpressionEvaluationDetails] + :keyword if_not_exists_details: Evaluation details of IfNotExists effect. + :paramtype if_not_exists_details: + ~azure.mgmt.policyinsights.models.IfNotExistsEvaluationDetails + """ + super().__init__(**kwargs) + self.evaluated_expressions = evaluated_expressions + self.if_not_exists_details = if_not_exists_details + self.reason = None + + class CheckRestrictionsRequest(_serialization.Model): """The check policy restrictions parameters describing the resource that is being evaluated. @@ -289,6 +334,9 @@ class CheckRestrictionsRequest(_serialization.Model): :ivar pending_fields: The list of fields and values that should be evaluated for potential restrictions. :vartype pending_fields: list[~azure.mgmt.policyinsights.models.PendingField] + :ivar include_audit_effect: Whether to include policies with the 'audit' effect in the results. + Defaults to false. + :vartype include_audit_effect: bool """ _validation = { @@ -298,6 +346,7 @@ class CheckRestrictionsRequest(_serialization.Model): _attribute_map = { "resource_details": {"key": "resourceDetails", "type": "CheckRestrictionsResourceDetails"}, "pending_fields": {"key": "pendingFields", "type": "[PendingField]"}, + "include_audit_effect": {"key": "includeAuditEffect", "type": "bool"}, } def __init__( @@ -305,18 +354,23 @@ def __init__( *, resource_details: "_models.CheckRestrictionsResourceDetails", pending_fields: Optional[List["_models.PendingField"]] = None, - **kwargs - ): + include_audit_effect: bool = False, + **kwargs: Any + ) -> None: """ :keyword resource_details: The information about the resource that will be evaluated. Required. :paramtype resource_details: ~azure.mgmt.policyinsights.models.CheckRestrictionsResourceDetails :keyword pending_fields: The list of fields and values that should be evaluated for potential restrictions. :paramtype pending_fields: list[~azure.mgmt.policyinsights.models.PendingField] + :keyword include_audit_effect: Whether to include policies with the 'audit' effect in the + results. Defaults to false. + :paramtype include_audit_effect: bool """ super().__init__(**kwargs) self.resource_details = resource_details self.pending_fields = pending_fields + self.include_audit_effect = include_audit_effect class CheckRestrictionsResourceDetails(_serialization.Model): @@ -345,8 +399,8 @@ class CheckRestrictionsResourceDetails(_serialization.Model): } def __init__( - self, *, resource_content: JSON, api_version: Optional[str] = None, scope: Optional[str] = None, **kwargs - ): + self, *, resource_content: JSON, api_version: Optional[str] = None, scope: Optional[str] = None, **kwargs: Any + ) -> None: """ :keyword resource_content: The resource content. This should include whatever properties are already known and can be a partial set of all resource properties. Required. @@ -389,7 +443,7 @@ class CheckRestrictionsResult(_serialization.Model): }, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.field_restrictions = None @@ -408,7 +462,9 @@ class CheckRestrictionsResultContentEvaluationResult(_serialization.Model): "policy_evaluations": {"key": "policyEvaluations", "type": "[PolicyEvaluationResult]"}, } - def __init__(self, *, policy_evaluations: Optional[List["_models.PolicyEvaluationResult"]] = None, **kwargs): + def __init__( + self, *, policy_evaluations: Optional[List["_models.PolicyEvaluationResult"]] = None, **kwargs: Any + ) -> None: """ :keyword policy_evaluations: Policy evaluation results against the given resource content. This will indicate if the partial content that was provided will be denied as-is. @@ -432,7 +488,7 @@ class ComplianceDetail(_serialization.Model): "count": {"key": "count", "type": "int"}, } - def __init__(self, *, compliance_state: Optional[str] = None, count: Optional[int] = None, **kwargs): + def __init__(self, *, compliance_state: Optional[str] = None, count: Optional[int] = None, **kwargs: Any) -> None: """ :keyword compliance_state: The compliance state. :paramtype compliance_state: str @@ -489,8 +545,8 @@ def __init__( tenant_id: Optional[str] = None, principal_oid: Optional[str] = None, policy_definition_action: Optional[str] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword additional_properties: Unmatched properties from the message are deserialized to this collection. @@ -562,7 +618,7 @@ class ComponentExpressionEvaluationDetails(_serialization.Model): "operator": {"key": "operator", "type": "str"}, } - def __init__(self, *, result: Optional[str] = None, **kwargs): + def __init__(self, *, result: Optional[str] = None, **kwargs: Any) -> None: """ :keyword result: Evaluation result. :paramtype result: str @@ -598,7 +654,7 @@ class ComponentPolicyEvaluationDetails(_serialization.Model): "reason": {"key": "reason", "type": "str"}, } - def __init__(self, *, reason: Optional[str] = None, **kwargs): + def __init__(self, *, reason: Optional[str] = None, **kwargs: Any) -> None: """ :keyword reason: Additional textual reason for the evaluation outcome. :paramtype reason: str @@ -767,8 +823,8 @@ def __init__( # pylint: disable=too-many-locals *, additional_properties: Optional[Dict[str, Any]] = None, policy_evaluation_details: Optional["_models.ComponentPolicyEvaluationDetails"] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword additional_properties: Unmatched properties from the message are deserialized to this collection. @@ -842,8 +898,8 @@ def __init__( odata_context: Optional[str] = None, odata_count: Optional[int] = None, value: Optional[List["_models.ComponentPolicyState"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword odata_context: OData context string; used by OData clients to resolve type information based on metadata. @@ -896,8 +952,8 @@ def __init__( name: Optional[str] = None, timestamp: Optional[datetime.datetime] = None, compliance_state: Optional[str] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword additional_properties: Unmatched properties from the message are deserialized to this collection. @@ -955,7 +1011,7 @@ class ErrorDefinition(_serialization.Model): "additional_info": {"key": "additionalInfo", "type": "[TypedErrorInfo]"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.code = None @@ -998,7 +1054,7 @@ class ErrorDefinitionAutoGenerated(_serialization.Model): "additional_info": {"key": "additionalInfo", "type": "[TypedErrorInfo]"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.code = None @@ -1041,7 +1097,7 @@ class ErrorDefinitionAutoGenerated2(_serialization.Model): "additional_info": {"key": "additionalInfo", "type": "[TypedErrorInfo]"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.code = None @@ -1062,7 +1118,7 @@ class ErrorResponse(_serialization.Model): "error": {"key": "error", "type": "ErrorDefinition"}, } - def __init__(self, *, error: Optional["_models.ErrorDefinition"] = None, **kwargs): + def __init__(self, *, error: Optional["_models.ErrorDefinition"] = None, **kwargs: Any) -> None: """ :keyword error: The error details. :paramtype error: ~azure.mgmt.policyinsights.models.ErrorDefinition @@ -1082,7 +1138,7 @@ class ErrorResponseAutoGenerated(_serialization.Model): "error": {"key": "error", "type": "ErrorDefinitionAutoGenerated"}, } - def __init__(self, *, error: Optional["_models.ErrorDefinitionAutoGenerated"] = None, **kwargs): + def __init__(self, *, error: Optional["_models.ErrorDefinitionAutoGenerated"] = None, **kwargs: Any) -> None: """ :keyword error: The error details. :paramtype error: ~azure.mgmt.policyinsights.models.ErrorDefinitionAutoGenerated @@ -1102,7 +1158,7 @@ class ErrorResponseAutoGenerated2(_serialization.Model): "error": {"key": "error", "type": "ErrorDefinitionAutoGenerated2"}, } - def __init__(self, *, error: Optional["_models.ErrorDefinitionAutoGenerated2"] = None, **kwargs): + def __init__(self, *, error: Optional["_models.ErrorDefinitionAutoGenerated2"] = None, **kwargs: Any) -> None: """ :keyword error: The error details. :paramtype error: ~azure.mgmt.policyinsights.models.ErrorDefinitionAutoGenerated2 @@ -1155,8 +1211,8 @@ def __init__( expression_value: Optional[JSON] = None, target_value: Optional[JSON] = None, operator: Optional[str] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword result: Evaluation result. :paramtype result: str @@ -1187,7 +1243,7 @@ class FieldRestriction(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. :ivar result: The type of restriction that is imposed on the field. Known values are: - "Required", "Removed", and "Deny". + "Required", "Removed", "Deny", and "Audit". :vartype result: str or ~azure.mgmt.policyinsights.models.FieldRestrictionResult :ivar default_value: The value that policy will set for the field if the user does not provide a value. @@ -1196,6 +1252,11 @@ class FieldRestriction(_serialization.Model): :vartype values: list[str] :ivar policy: The details of the policy that is causing the field restriction. :vartype policy: ~azure.mgmt.policyinsights.models.PolicyReference + :ivar policy_effect: The effect of the policy that is causing the field restriction. + http://aka.ms/policyeffects. + :vartype policy_effect: str + :ivar reason: The reason for the restriction. + :vartype reason: str """ _validation = { @@ -1203,6 +1264,8 @@ class FieldRestriction(_serialization.Model): "default_value": {"readonly": True}, "values": {"readonly": True}, "policy": {"readonly": True}, + "policy_effect": {"readonly": True}, + "reason": {"readonly": True}, } _attribute_map = { @@ -1210,15 +1273,19 @@ class FieldRestriction(_serialization.Model): "default_value": {"key": "defaultValue", "type": "str"}, "values": {"key": "values", "type": "[str]"}, "policy": {"key": "policy", "type": "PolicyReference"}, + "policy_effect": {"key": "policyEffect", "type": "str"}, + "reason": {"key": "reason", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.result = None self.default_value = None self.values = None self.policy = None + self.policy_effect = None + self.reason = None class FieldRestrictions(_serialization.Model): @@ -1242,7 +1309,7 @@ class FieldRestrictions(_serialization.Model): "restrictions": {"key": "restrictions", "type": "[FieldRestriction]"}, } - def __init__(self, *, restrictions: Optional[List["_models.FieldRestriction"]] = None, **kwargs): + def __init__(self, *, restrictions: Optional[List["_models.FieldRestriction"]] = None, **kwargs: Any) -> None: """ :keyword restrictions: The restrictions placed on that field by policy. :paramtype restrictions: list[~azure.mgmt.policyinsights.models.FieldRestriction] @@ -1267,7 +1334,9 @@ class IfNotExistsEvaluationDetails(_serialization.Model): "total_resources": {"key": "totalResources", "type": "int"}, } - def __init__(self, *, resource_id: Optional[str] = None, total_resources: Optional[int] = None, **kwargs): + def __init__( + self, *, resource_id: Optional[str] = None, total_resources: Optional[int] = None, **kwargs: Any + ) -> None: """ :keyword resource_id: ID of the last evaluated resource for IfNotExists effect. :paramtype resource_id: str @@ -1303,8 +1372,8 @@ def __init__( name: Optional[str] = None, is_data_action: Optional[bool] = None, display: Optional["_models.OperationDisplay"] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword name: Operation name. :paramtype name: str @@ -1346,8 +1415,8 @@ def __init__( resource: Optional[str] = None, operation: Optional[str] = None, description: Optional[str] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword provider: Resource provider name. :paramtype provider: str @@ -1384,8 +1453,8 @@ class OperationsListResults(_serialization.Model): } def __init__( - self, *, odata_count: Optional[int] = None, value: Optional[List["_models.Operation"]] = None, **kwargs - ): + self, *, odata_count: Optional[int] = None, value: Optional[List["_models.Operation"]] = None, **kwargs: Any + ) -> None: """ :keyword odata_count: OData entity count; represents the number of operations returned. :paramtype odata_count: int @@ -1419,7 +1488,7 @@ class PendingField(_serialization.Model): "values": {"key": "values", "type": "[str]"}, } - def __init__(self, *, field: str, values: Optional[List[str]] = None, **kwargs): + def __init__(self, *, field: str, values: Optional[List[str]] = None, **kwargs: Any) -> None: """ :keyword field: The name of the field. This can be a top-level property like 'name' or 'type' or an Azure Policy field alias. Required. @@ -1465,8 +1534,8 @@ def __init__( results: Optional["_models.SummaryResults"] = None, policy_definitions: Optional[List["_models.PolicyDefinitionSummary"]] = None, policy_groups: Optional[List["_models.PolicyGroupSummary"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword policy_assignment_id: Policy assignment ID. :paramtype policy_assignment_id: str @@ -1519,8 +1588,8 @@ def __init__( policy_definition_group_names: Optional[List[str]] = None, effect: Optional[str] = None, results: Optional["_models.SummaryResults"] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword policy_definition_id: Policy definition ID. :paramtype policy_definition_id: str @@ -1579,7 +1648,7 @@ class PolicyDetails(_serialization.Model): "policy_definition_reference_id": {"key": "policyDefinitionReferenceId", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.policy_definition_id = None @@ -1590,6 +1659,29 @@ def __init__(self, **kwargs): self.policy_definition_reference_id = None +class PolicyEffectDetails(_serialization.Model): + """The details of the effect that was applied to the resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar policy_effect: The effect that was applied to the resource. http://aka.ms/policyeffects. + :vartype policy_effect: str + """ + + _validation = { + "policy_effect": {"readonly": True}, + } + + _attribute_map = { + "policy_effect": {"key": "policyEffect", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.policy_effect = None + + class PolicyEvaluationDetails(_serialization.Model): """Policy evaluation details. @@ -1610,8 +1702,8 @@ def __init__( *, evaluated_expressions: Optional[List["_models.ExpressionEvaluationDetails"]] = None, if_not_exists_details: Optional["_models.IfNotExistsEvaluationDetails"] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword evaluated_expressions: Details of the evaluated expressions. :paramtype evaluated_expressions: @@ -1637,27 +1729,33 @@ class PolicyEvaluationResult(_serialization.Model): :vartype evaluation_result: str :ivar evaluation_details: The detailed results of the policy expressions and values that were evaluated. - :vartype evaluation_details: ~azure.mgmt.policyinsights.models.PolicyEvaluationDetails + :vartype evaluation_details: + ~azure.mgmt.policyinsights.models.CheckRestrictionEvaluationDetails + :ivar effect_details: The details of the effect that was applied to the resource. + :vartype effect_details: ~azure.mgmt.policyinsights.models.PolicyEffectDetails """ _validation = { "policy_info": {"readonly": True}, "evaluation_result": {"readonly": True}, "evaluation_details": {"readonly": True}, + "effect_details": {"readonly": True}, } _attribute_map = { "policy_info": {"key": "policyInfo", "type": "PolicyReference"}, "evaluation_result": {"key": "evaluationResult", "type": "str"}, - "evaluation_details": {"key": "evaluationDetails", "type": "PolicyEvaluationDetails"}, + "evaluation_details": {"key": "evaluationDetails", "type": "CheckRestrictionEvaluationDetails"}, + "effect_details": {"key": "effectDetails", "type": "PolicyEffectDetails"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.policy_info = None self.evaluation_result = None self.evaluation_details = None + self.effect_details = None class PolicyEvent(_serialization.Model): # pylint: disable=too-many-instance-attributes @@ -1812,8 +1910,8 @@ def __init__( # pylint: disable=too-many-locals tenant_id: Optional[str] = None, principal_oid: Optional[str] = None, components: Optional[List["_models.ComponentEventDetails"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword additional_properties: Unmatched properties from the message are deserialized to this collection. @@ -1960,8 +2058,8 @@ def __init__( odata_count: Optional[int] = None, odata_next_link: Optional[str] = None, value: Optional[List["_models.PolicyEvent"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword odata_context: OData context string; used by OData clients to resolve type information based on metadata. @@ -1996,8 +2094,12 @@ class PolicyGroupSummary(_serialization.Model): } def __init__( - self, *, policy_group_name: Optional[str] = None, results: Optional["_models.SummaryResults"] = None, **kwargs - ): + self, + *, + policy_group_name: Optional[str] = None, + results: Optional["_models.SummaryResults"] = None, + **kwargs: Any + ) -> None: """ :keyword policy_group_name: Policy group name. :paramtype policy_group_name: str @@ -2066,7 +2168,7 @@ class PolicyMetadata(_serialization.Model): # pylint: disable=too-many-instance "requirements": {"key": "properties.requirements", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.id = None @@ -2103,7 +2205,7 @@ class PolicyMetadataCollection(_serialization.Model): "next_link": {"key": "nextLink", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.value = None @@ -2147,7 +2249,7 @@ class PolicyMetadataSlimProperties(_serialization.Model): "metadata": {"key": "metadata", "type": "object"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.metadata_id = None @@ -2203,7 +2305,7 @@ class PolicyMetadataProperties(PolicyMetadataSlimProperties): "requirements": {"key": "requirements", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.description = None @@ -2240,7 +2342,7 @@ class PolicyReference(_serialization.Model): "policy_assignment_id": {"key": "policyAssignmentId", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.policy_definition_id = None @@ -2418,8 +2520,8 @@ def __init__( # pylint: disable=too-many-locals policy_evaluation_details: Optional["_models.PolicyEvaluationDetails"] = None, policy_definition_group_names: Optional[List[str]] = None, components: Optional[List["_models.ComponentStateDetails"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword additional_properties: Unmatched properties from the message are deserialized to this collection. @@ -2569,8 +2671,8 @@ def __init__( odata_count: Optional[int] = None, odata_next_link: Optional[str] = None, value: Optional[List["_models.PolicyState"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword odata_context: OData context string; used by OData clients to resolve type information based on metadata. @@ -2625,7 +2727,7 @@ class PolicyTrackedResource(_serialization.Model): "last_update_utc": {"key": "lastUpdateUtc", "type": "iso-8601"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.tracked_resource_id = None @@ -2656,7 +2758,7 @@ class PolicyTrackedResourcesQueryResults(_serialization.Model): "next_link": {"key": "nextLink", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.value = None @@ -2674,7 +2776,7 @@ class QueryFailure(_serialization.Model): "error": {"key": "error", "type": "QueryFailureError"}, } - def __init__(self, *, error: Optional["_models.QueryFailureError"] = None, **kwargs): + def __init__(self, *, error: Optional["_models.QueryFailureError"] = None, **kwargs: Any) -> None: """ :keyword error: Error definition. :paramtype error: ~azure.mgmt.policyinsights.models.QueryFailureError @@ -2704,7 +2806,7 @@ class QueryFailureError(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.code = None @@ -2769,8 +2871,8 @@ def __init__( apply: Optional[str] = None, skip_token: Optional[str] = None, expand: Optional[str] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword top: Maximum number of records to return. :paramtype top: int @@ -2835,7 +2937,9 @@ class Remediation(_serialization.Model): # pylint: disable=too-many-instance-at "ReEvaluateCompliance". :vartype resource_discovery_mode: str or ~azure.mgmt.policyinsights.models.ResourceDiscoveryMode - :ivar provisioning_state: The status of the remediation. + :ivar provisioning_state: The status of the remediation. This refers to the entire remediation + task, not individual deployments. Allowed values are Evaluating, Canceled, Cancelling, Failed, + Complete, or Succeeded. :vartype provisioning_state: str :ivar created_on: The time at which the remediation was created. :vartype created_on: ~datetime.datetime @@ -2907,8 +3011,8 @@ def __init__( resource_count: Optional[int] = None, parallel_deployments: Optional[int] = None, failure_threshold: Optional["_models.RemediationPropertiesFailureThreshold"] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword policy_assignment_id: The resource ID of the policy assignment that should be remediated. @@ -2997,7 +3101,7 @@ class RemediationDeployment(_serialization.Model): "last_updated_on": {"key": "lastUpdatedOn", "type": "iso-8601"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.remediated_resource_id = None @@ -3030,7 +3134,7 @@ class RemediationDeploymentsListResult(_serialization.Model): "next_link": {"key": "nextLink", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.value = None @@ -3064,7 +3168,7 @@ class RemediationDeploymentSummary(_serialization.Model): "failed_deployments": {"key": "failedDeployments", "type": "int"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.total_deployments = None @@ -3083,7 +3187,7 @@ class RemediationFilters(_serialization.Model): "locations": {"key": "locations", "type": "[str]"}, } - def __init__(self, *, locations: Optional[List[str]] = None, **kwargs): + def __init__(self, *, locations: Optional[List[str]] = None, **kwargs: Any) -> None: """ :keyword locations: The resource locations that will be remediated. :paramtype locations: list[str] @@ -3113,7 +3217,7 @@ class RemediationListResult(_serialization.Model): "next_link": {"key": "nextLink", "type": "str"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.value = None @@ -3133,7 +3237,7 @@ class RemediationPropertiesFailureThreshold(_serialization.Model): "percentage": {"key": "percentage", "type": "float"}, } - def __init__(self, *, percentage: Optional[float] = None, **kwargs): + def __init__(self, *, percentage: Optional[float] = None, **kwargs: Any) -> None: """ :keyword percentage: A number between 0.0 to 1.0 representing the percentage failure threshold. The remediation will fail if the percentage of failed remediation operations (i.e. failed @@ -3193,7 +3297,7 @@ class SlimPolicyMetadata(_serialization.Model): "metadata": {"key": "properties.metadata", "type": "object"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.id = None @@ -3236,8 +3340,8 @@ def __init__( odata_context: Optional[str] = None, odata_count: Optional[int] = None, value: Optional[List["_models.Summary"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword odata_context: OData context string; used by OData clients to resolve type information based on metadata. @@ -3282,8 +3386,8 @@ def __init__( odata_context: Optional[str] = None, results: Optional["_models.SummaryResults"] = None, policy_assignments: Optional[List["_models.PolicyAssignmentSummary"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword odata_id: OData entity ID; always set to null since summaries do not have an entity ID. @@ -3347,8 +3451,8 @@ def __init__( resource_details: Optional[List["_models.ComplianceDetail"]] = None, policy_details: Optional[List["_models.ComplianceDetail"]] = None, policy_group_details: Optional[List["_models.ComplianceDetail"]] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword query_results_uri: HTTP POST URI for queryResults action on Microsoft.PolicyInsights to retrieve raw results for the compliance summary. This property will not be available by @@ -3413,8 +3517,8 @@ def __init__( last_modified_by: Optional[str] = None, last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, last_modified_at: Optional[datetime.datetime] = None, - **kwargs - ): + **kwargs: Any + ) -> None: """ :keyword created_by: The identity that created the resource. :paramtype created_by: str @@ -3466,7 +3570,7 @@ class TrackedResourceModificationDetails(_serialization.Model): "deployment_time": {"key": "deploymentTime", "type": "iso-8601"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.policy_details = None @@ -3495,7 +3599,7 @@ class TypedErrorInfo(_serialization.Model): "info": {"key": "info", "type": "object"}, } - def __init__(self, **kwargs): + def __init__(self, **kwargs: Any) -> None: """ """ super().__init__(**kwargs) self.type = None diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_policy_insights_client_enums.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_policy_insights_client_enums.py index c327e4726fe2..8bbc193201a8 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_policy_insights_client_enums.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/models/_policy_insights_client_enums.py @@ -13,12 +13,12 @@ class ComplianceState(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The compliance state that should be set on the resource.""" - #: The resource is in compliance with the policy. COMPLIANT = "Compliant" - #: The resource is not in compliance with the policy. + """The resource is in compliance with the policy.""" NON_COMPLIANT = "NonCompliant" - #: The compliance state of the resource is not known. + """The resource is not in compliance with the policy.""" UNKNOWN = "Unknown" + """The compliance state of the resource is not known.""" class ComponentPolicyStatesResource(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -39,12 +39,14 @@ class CreatedByType(str, Enum, metaclass=CaseInsensitiveEnumMeta): class FieldRestrictionResult(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The type of restriction that is imposed on the field.""" - #: The field and/or values are required by policy. REQUIRED = "Required" - #: The field will be removed by policy. + """The field and/or values are required by policy.""" REMOVED = "Removed" - #: The field and/or values will be denied by policy. + """The field will be removed by policy.""" DENY = "Deny" + """The field and/or values will be denied by policy.""" + AUDIT = "Audit" + """The field and/or values will be audited by policy.""" class PolicyEventsResourceType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -77,8 +79,8 @@ class ResourceDiscoveryMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): specified. """ - #: Remediate resources that are already known to be non-compliant. EXISTING_NON_COMPLIANT = "ExistingNonCompliant" - #: Re-evaluate the compliance state of resources and then remediate the resources found to be - #: non-compliant. + """Remediate resources that are already known to be non-compliant.""" RE_EVALUATE_COMPLIANCE = "ReEvaluateCompliance" + """Re-evaluate the compliance state of resources and then remediate the resources found to be + #: non-compliant.""" diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_attestations_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_attestations_operations.py index dec1033e723e..d91dd81eea8f 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_attestations_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_attestations_operations.py @@ -6,7 +6,6 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, cast, overload from azure.core.exceptions import ( @@ -31,10 +30,6 @@ from .._serialization import Serializer from .._vendor import _convert_request, _format_url_section -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -48,7 +43,7 @@ def build_list_for_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -78,7 +73,7 @@ def build_create_or_update_at_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -109,7 +104,7 @@ def build_get_at_subscription_request(attestation_name: str, subscription_id: st _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -137,7 +132,7 @@ def build_delete_at_subscription_request(attestation_name: str, subscription_id: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -172,7 +167,7 @@ def build_list_for_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -208,7 +203,7 @@ def build_create_or_update_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -244,7 +239,7 @@ def build_get_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -277,7 +272,7 @@ def build_delete_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -310,7 +305,7 @@ def build_list_for_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -338,7 +333,7 @@ def build_create_or_update_at_resource_request(resource_id: str, attestation_nam _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -366,7 +361,7 @@ def build_get_at_resource_request(resource_id: str, attestation_name: str, **kwa _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -391,7 +386,7 @@ def build_delete_at_resource_request(resource_id: str, attestation_name: str, ** _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -447,7 +442,7 @@ def list_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.AttestationListResult] = kwargs.pop("cls", None) error_map = { @@ -495,8 +490,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -527,7 +523,7 @@ def _create_or_update_at_subscription_initial( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) @@ -553,8 +549,9 @@ def _create_or_update_at_subscription_initial( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -646,7 +643,8 @@ def begin_create_or_update_at_subscription( :param attestation_name: The name of the attestation. Required. :type attestation_name: str - :param parameters: The attestation parameters. Is either a model type or a IO type. Required. + :param parameters: The attestation parameters. Is either a Attestation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Attestation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -667,7 +665,7 @@ def begin_create_or_update_at_subscription( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) @@ -733,7 +731,7 @@ def get_at_subscription(self, attestation_name: str, **kwargs: Any) -> _models.A _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) request = build_get_at_subscription_request( @@ -747,8 +745,9 @@ def get_at_subscription(self, attestation_name: str, **kwargs: Any) -> _models.A request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -793,7 +792,7 @@ def delete_at_subscription( # pylint: disable=inconsistent-return-statements _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_delete_at_subscription_request( @@ -807,8 +806,9 @@ def delete_at_subscription( # pylint: disable=inconsistent-return-statements request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -844,7 +844,7 @@ def list_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.AttestationListResult] = kwargs.pop("cls", None) error_map = { @@ -893,8 +893,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -925,7 +926,7 @@ def _create_or_update_at_resource_group_initial( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) @@ -952,8 +953,9 @@ def _create_or_update_at_resource_group_initial( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1061,7 +1063,8 @@ def begin_create_or_update_at_resource_group( :type resource_group_name: str :param attestation_name: The name of the attestation. Required. :type attestation_name: str - :param parameters: The attestation parameters. Is either a model type or a IO type. Required. + :param parameters: The attestation parameters. Is either a Attestation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Attestation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -1082,7 +1085,7 @@ def begin_create_or_update_at_resource_group( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) @@ -1154,7 +1157,7 @@ def get_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) request = build_get_at_resource_group_request( @@ -1169,8 +1172,9 @@ def get_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1218,7 +1222,7 @@ def delete_at_resource_group( # pylint: disable=inconsistent-return-statements _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_delete_at_resource_group_request( @@ -1233,8 +1237,9 @@ def delete_at_resource_group( # pylint: disable=inconsistent-return-statements request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1269,7 +1274,7 @@ def list_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.AttestationListResult] = kwargs.pop("cls", None) error_map = { @@ -1317,8 +1322,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1347,7 +1353,7 @@ def _create_or_update_at_resource_initial( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) @@ -1373,8 +1379,9 @@ def _create_or_update_at_resource_initial( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1479,7 +1486,8 @@ def begin_create_or_update_at_resource( :type resource_id: str :param attestation_name: The name of the attestation. Required. :type attestation_name: str - :param parameters: The attestation parameters. Is either a model type or a IO type. Required. + :param parameters: The attestation parameters. Is either a Attestation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Attestation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -1500,7 +1508,7 @@ def begin_create_or_update_at_resource( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) @@ -1569,7 +1577,7 @@ def get_at_resource(self, resource_id: str, attestation_name: str, **kwargs: Any _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[_models.Attestation] = kwargs.pop("cls", None) request = build_get_at_resource_request( @@ -1583,8 +1591,9 @@ def get_at_resource(self, resource_id: str, attestation_name: str, **kwargs: Any request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1631,7 +1640,7 @@ def delete_at_resource( # pylint: disable=inconsistent-return-statements _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-09-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_delete_at_resource_request( @@ -1645,8 +1654,9 @@ def delete_at_resource( # pylint: disable=inconsistent-return-statements request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_component_policy_states_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_component_policy_states_operations.py index f3c51adf7680..beeb7abe594b 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_component_policy_states_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_component_policy_states_operations.py @@ -56,7 +56,7 @@ def build_list_query_results_for_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -113,7 +113,7 @@ def build_list_query_results_for_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -171,7 +171,7 @@ def build_list_query_results_for_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -233,7 +233,7 @@ def build_list_query_results_for_policy_definition_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -302,7 +302,7 @@ def build_list_query_results_for_subscription_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -372,7 +372,7 @@ def build_list_query_results_for_resource_group_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -499,7 +499,7 @@ def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_subscription_request( @@ -520,8 +520,9 @@ def list_query_results_for_subscription( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -603,7 +604,7 @@ def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_resource_group_request( @@ -625,8 +626,9 @@ def list_query_results_for_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -708,7 +710,7 @@ def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_resource_request( @@ -730,8 +732,9 @@ def list_query_results_for_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -820,7 +823,7 @@ def list_query_results_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_policy_definition_request( @@ -843,8 +846,9 @@ def list_query_results_for_policy_definition( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -933,7 +937,7 @@ def list_query_results_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_subscription_level_policy_assignment_request( @@ -956,8 +960,9 @@ def list_query_results_for_subscription_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1049,7 +1054,7 @@ def list_query_results_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.ComponentPolicyStatesQueryResults] = kwargs.pop("cls", None) request = build_list_query_results_for_resource_group_level_policy_assignment_request( @@ -1073,8 +1078,9 @@ def list_query_results_for_resource_group_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_operations.py index 3c8442baf3f1..309c194f59eb 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_operations.py @@ -6,7 +6,6 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( @@ -28,10 +27,6 @@ from .._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -43,7 +38,7 @@ def build_list_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -97,7 +92,7 @@ def list(self, **kwargs: Any) -> _models.OperationsListResults: _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-04-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2022-04-01")) cls: ClsType[_models.OperationsListResults] = kwargs.pop("cls", None) request = build_list_request( @@ -109,8 +104,9 @@ def list(self, **kwargs: Any) -> _models.OperationsListResults: request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_events_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_events_operations.py index 4543dcf2d14b..3fd5252db62b 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_events_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_events_operations.py @@ -61,7 +61,7 @@ def build_list_query_results_for_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -106,7 +106,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -145,7 +145,7 @@ def build_list_query_results_for_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -189,7 +189,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -229,7 +229,7 @@ def build_list_query_results_for_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -274,7 +274,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -314,7 +314,7 @@ def build_list_query_results_for_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -360,7 +360,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -403,7 +403,7 @@ def build_list_query_results_for_policy_set_definition_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -449,7 +449,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -492,7 +492,7 @@ def build_list_query_results_for_policy_definition_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -538,7 +538,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -581,7 +581,7 @@ def build_list_query_results_for_subscription_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -627,7 +627,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -671,7 +671,7 @@ def build_list_query_results_for_resource_group_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -718,7 +718,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -791,7 +791,7 @@ def list_query_results_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -884,8 +884,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -927,7 +928,7 @@ def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1019,8 +1020,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1065,7 +1067,7 @@ def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1158,8 +1160,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1201,7 +1204,7 @@ def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1298,8 +1301,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1351,7 +1355,7 @@ def list_query_results_for_policy_set_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1445,8 +1449,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1498,7 +1503,7 @@ def list_query_results_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1592,8 +1597,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1645,7 +1651,7 @@ def list_query_results_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1739,8 +1745,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1795,7 +1802,7 @@ def list_query_results_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyEventsQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1890,8 +1897,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_metadata_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_metadata_operations.py index d90da636026f..2f91e27f3a80 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_metadata_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_metadata_operations.py @@ -6,7 +6,6 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys from typing import Any, Callable, Dict, Iterable, Optional, TypeVar from azure.core.exceptions import ( @@ -29,10 +28,6 @@ from .._serialization import Serializer from .._vendor import _convert_request, _format_url_section -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -44,7 +39,7 @@ def build_get_resource_request(resource_name: str, **kwargs: Any) -> HttpRequest _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -68,7 +63,7 @@ def build_list_request(*, top: Optional[int] = None, **kwargs: Any) -> HttpReque _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -126,7 +121,7 @@ def get_resource(self, resource_name: str, **kwargs: Any) -> _models.PolicyMetad _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyMetadata] = kwargs.pop("cls", None) request = build_get_resource_request( @@ -139,8 +134,9 @@ def get_resource(self, resource_name: str, **kwargs: Any) -> _models.PolicyMetad request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -175,7 +171,7 @@ def list( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyMetadataCollection] = kwargs.pop("cls", None) error_map = { @@ -219,8 +215,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_restrictions_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_restrictions_operations.py index c362a36a53ff..37b0f2bce5f2 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_restrictions_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_restrictions_operations.py @@ -43,7 +43,7 @@ def build_check_at_subscription_scope_request(subscription_id: str, **kwargs: An _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -74,7 +74,7 @@ def build_check_at_resource_group_scope_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -110,7 +110,7 @@ def build_check_at_management_group_scope_request(management_group_id: str, **kw management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -196,8 +196,8 @@ def check_at_subscription_scope( ) -> _models.CheckRestrictionsResult: """Checks what restrictions Azure Policy will place on a resource within a subscription. - :param parameters: The check policy restrictions parameters. Is either a model type or a IO - type. Required. + :param parameters: The check policy restrictions parameters. Is either a + CheckRestrictionsRequest type or a IO type. Required. :type parameters: ~azure.mgmt.policyinsights.models.CheckRestrictionsRequest or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -218,7 +218,7 @@ def check_at_subscription_scope( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.CheckRestrictionsResult] = kwargs.pop("cls", None) @@ -243,8 +243,9 @@ def check_at_subscription_scope( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -322,8 +323,8 @@ def check_at_resource_group_scope( :param resource_group_name: The name of the resource group. The name is case insensitive. Required. :type resource_group_name: str - :param parameters: The check policy restrictions parameters. Is either a model type or a IO - type. Required. + :param parameters: The check policy restrictions parameters. Is either a + CheckRestrictionsRequest type or a IO type. Required. :type parameters: ~azure.mgmt.policyinsights.models.CheckRestrictionsRequest or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -344,7 +345,7 @@ def check_at_resource_group_scope( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.CheckRestrictionsResult] = kwargs.pop("cls", None) @@ -370,8 +371,9 @@ def check_at_resource_group_scope( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -454,8 +456,8 @@ def check_at_management_group_scope( :param management_group_id: Management group ID. Required. :type management_group_id: str - :param parameters: The check policy restrictions parameters. Is either a model type or a IO - type. Required. + :param parameters: The check policy restrictions parameters. Is either a + CheckManagementGroupRestrictionsRequest type or a IO type. Required. :type parameters: ~azure.mgmt.policyinsights.models.CheckManagementGroupRestrictionsRequest or IO :keyword management_groups_namespace: The namespace for Microsoft Management RP; only @@ -484,7 +486,7 @@ def check_at_management_group_scope( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2022-03-01"] = kwargs.pop("api_version", _params.pop("api-version", "2022-03-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2023-03-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.CheckRestrictionsResult] = kwargs.pop("cls", None) @@ -510,8 +512,9 @@ def check_at_management_group_scope( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_states_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_states_operations.py index 4fd4449053ba..c6eb0e97caca 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_states_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_states_operations.py @@ -63,7 +63,7 @@ def build_list_query_results_for_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -108,7 +108,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -146,7 +146,7 @@ def build_summarize_for_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -198,7 +198,7 @@ def build_list_query_results_for_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -242,7 +242,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -277,7 +277,7 @@ def build_summarize_for_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -329,7 +329,7 @@ def build_list_query_results_for_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -374,7 +374,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -410,7 +410,7 @@ def build_summarize_for_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -463,7 +463,7 @@ def build_list_query_results_for_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -509,7 +509,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -544,7 +544,7 @@ def build_summarize_for_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -582,7 +582,7 @@ def build_trigger_subscription_evaluation_request(subscription_id: str, **kwargs _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -611,7 +611,7 @@ def build_trigger_resource_group_evaluation_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -656,7 +656,7 @@ def build_list_query_results_for_policy_set_definition_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -702,7 +702,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -741,7 +741,7 @@ def build_summarize_for_policy_set_definition_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -798,7 +798,7 @@ def build_list_query_results_for_policy_definition_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -844,7 +844,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -883,7 +883,7 @@ def build_summarize_for_policy_definition_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -940,7 +940,7 @@ def build_list_query_results_for_subscription_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -986,7 +986,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -1025,7 +1025,7 @@ def build_summarize_for_subscription_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -1083,7 +1083,7 @@ def build_list_query_results_for_resource_group_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -1130,7 +1130,7 @@ def build_next_link_request(next_link: str, *, skip_token: Optional[str] = None, _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -1170,7 +1170,7 @@ def build_summarize_for_resource_group_level_policy_assignment_request( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -1259,7 +1259,7 @@ def list_query_results_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1352,8 +1352,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1412,7 +1413,7 @@ def summarize_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -1441,8 +1442,9 @@ def summarize_for_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1489,7 +1491,7 @@ def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1581,8 +1583,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1634,7 +1637,7 @@ def summarize_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -1662,8 +1665,9 @@ def summarize_for_subscription( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1713,7 +1717,7 @@ def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -1806,8 +1810,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1862,7 +1867,7 @@ def summarize_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -1891,8 +1896,9 @@ def summarize_for_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1939,7 +1945,7 @@ def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -2036,8 +2042,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2089,7 +2096,7 @@ def summarize_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -2117,8 +2124,9 @@ def summarize_for_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2153,7 +2161,7 @@ def _trigger_subscription_evaluation_initial( # pylint: disable=inconsistent-re _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_trigger_subscription_evaluation_request( @@ -2166,8 +2174,9 @@ def _trigger_subscription_evaluation_initial( # pylint: disable=inconsistent-re request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2205,7 +2214,7 @@ def begin_trigger_subscription_evaluation(self, subscription_id: str, **kwargs: _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) @@ -2260,7 +2269,7 @@ def _trigger_resource_group_evaluation_initial( # pylint: disable=inconsistent- _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) request = build_trigger_resource_group_evaluation_request( @@ -2274,8 +2283,9 @@ def _trigger_resource_group_evaluation_initial( # pylint: disable=inconsistent- request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2317,7 +2327,7 @@ def begin_trigger_resource_group_evaluation( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) @@ -2395,7 +2405,7 @@ def list_query_results_for_policy_set_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -2489,8 +2499,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2552,7 +2563,7 @@ def summarize_for_policy_set_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -2582,8 +2593,9 @@ def summarize_for_policy_set_definition( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2640,7 +2652,7 @@ def list_query_results_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -2734,8 +2746,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2797,7 +2810,7 @@ def summarize_for_policy_definition( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -2827,8 +2840,9 @@ def summarize_for_policy_definition( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2885,7 +2899,7 @@ def list_query_results_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -2979,8 +2993,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3042,7 +3057,7 @@ def summarize_for_subscription_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -3072,8 +3087,9 @@ def summarize_for_subscription_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3133,7 +3149,7 @@ def list_query_results_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.PolicyStatesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -3228,8 +3244,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3294,7 +3311,7 @@ def summarize_for_resource_group_level_policy_assignment( authorization_namespace: Literal["Microsoft.Authorization"] = kwargs.pop( "authorization_namespace", "Microsoft.Authorization" ) - api_version: Literal["2019-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2019-10-01")) cls: ClsType[_models.SummarizeResults] = kwargs.pop("cls", None) _top = None @@ -3325,8 +3342,9 @@ def summarize_for_resource_group_level_policy_assignment( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_tracked_resources_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_tracked_resources_operations.py index e59da858511c..c240fc668b82 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_tracked_resources_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_policy_tracked_resources_operations.py @@ -54,9 +54,7 @@ def build_list_query_results_for_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -98,9 +96,7 @@ def build_list_query_results_for_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -142,9 +138,7 @@ def build_list_query_results_for_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -186,9 +180,7 @@ def build_list_query_results_for_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -271,9 +263,7 @@ def list_query_results_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -323,8 +313,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -365,9 +356,7 @@ def list_query_results_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -416,8 +405,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -461,9 +451,7 @@ def list_query_results_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -513,8 +501,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -558,9 +547,7 @@ def list_query_results_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2018-07-01-preview"] = kwargs.pop( - "api_version", _params.pop("api-version", "2018-07-01-preview") - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2018-07-01-preview")) cls: ClsType[_models.PolicyTrackedResourcesQueryResults] = kwargs.pop("cls", None) error_map = { @@ -609,8 +596,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_remediations_operations.py b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_remediations_operations.py index 2f56a35ecd62..e9d6612da002 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_remediations_operations.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/azure/mgmt/policyinsights/operations/_remediations_operations.py @@ -49,7 +49,7 @@ def build_list_deployments_at_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -85,7 +85,7 @@ def build_cancel_at_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -119,7 +119,7 @@ def build_list_for_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -156,7 +156,7 @@ def build_create_or_update_at_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -193,7 +193,7 @@ def build_get_at_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -227,7 +227,7 @@ def build_delete_at_management_group_request( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -258,7 +258,7 @@ def build_list_deployments_at_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -288,7 +288,7 @@ def build_cancel_at_subscription_request(remediation_name: str, subscription_id: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -318,7 +318,7 @@ def build_list_for_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -348,7 +348,7 @@ def build_create_or_update_at_subscription_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -379,7 +379,7 @@ def build_get_at_subscription_request(remediation_name: str, subscription_id: st _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -407,7 +407,7 @@ def build_delete_at_subscription_request(remediation_name: str, subscription_id: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -437,7 +437,7 @@ def build_list_deployments_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -470,7 +470,7 @@ def build_cancel_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -506,7 +506,7 @@ def build_list_for_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -540,7 +540,7 @@ def build_create_or_update_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -574,7 +574,7 @@ def build_get_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -605,7 +605,7 @@ def build_delete_at_resource_group_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -636,7 +636,7 @@ def build_list_deployments_at_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -666,7 +666,7 @@ def build_cancel_at_resource_request(resource_id: str, remediation_name: str, ** _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -695,7 +695,7 @@ def build_list_for_resource_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -723,7 +723,7 @@ def build_create_or_update_at_resource_request(resource_id: str, remediation_nam _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") @@ -751,7 +751,7 @@ def build_get_at_resource_request(resource_id: str, remediation_name: str, **kwa _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -776,7 +776,7 @@ def build_delete_at_resource_request(resource_id: str, remediation_name: str, ** _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) accept = _headers.pop("Accept", "application/json") # Construct URL @@ -848,7 +848,7 @@ def list_deployments_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -895,8 +895,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -946,7 +947,7 @@ def cancel_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_management_group_request( @@ -961,8 +962,9 @@ def cancel_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1008,7 +1010,7 @@ def list_for_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -1057,8 +1059,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1147,7 +1150,8 @@ def create_or_update_at_management_group( :type management_group_id: str :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword management_groups_namespace: The namespace for Microsoft Management RP; only "Microsoft.Management" is allowed. Default value is "Microsoft.Management". Note that @@ -1175,7 +1179,7 @@ def create_or_update_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -1202,8 +1206,9 @@ def create_or_update_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1261,7 +1266,7 @@ def get_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_management_group_request( @@ -1276,8 +1281,9 @@ def get_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1331,7 +1337,7 @@ def delete_at_management_group( management_groups_namespace: Literal["Microsoft.Management"] = kwargs.pop( "management_groups_namespace", "Microsoft.Management" ) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_management_group_request( @@ -1346,8 +1352,9 @@ def delete_at_management_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1389,7 +1396,7 @@ def list_deployments_at_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -1435,8 +1442,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1475,7 +1483,7 @@ def cancel_at_subscription(self, remediation_name: str, **kwargs: Any) -> _model _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_subscription_request( @@ -1489,8 +1497,9 @@ def cancel_at_subscription(self, remediation_name: str, **kwargs: Any) -> _model request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1527,7 +1536,7 @@ def list_for_subscription( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -1575,8 +1584,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1644,7 +1654,8 @@ def create_or_update_at_subscription( :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -1665,7 +1676,7 @@ def create_or_update_at_subscription( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -1691,8 +1702,9 @@ def create_or_update_at_subscription( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1739,7 +1751,7 @@ def get_at_subscription(self, remediation_name: str, **kwargs: Any) -> _models.R _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_subscription_request( @@ -1753,8 +1765,9 @@ def get_at_subscription(self, remediation_name: str, **kwargs: Any) -> _models.R request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1797,7 +1810,7 @@ def delete_at_subscription(self, remediation_name: str, **kwargs: Any) -> Option _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_subscription_request( @@ -1811,8 +1824,9 @@ def delete_at_subscription(self, remediation_name: str, **kwargs: Any) -> Option request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1860,7 +1874,7 @@ def list_deployments_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -1907,8 +1921,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1951,7 +1966,7 @@ def cancel_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_resource_group_request( @@ -1966,8 +1981,9 @@ def cancel_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2006,7 +2022,7 @@ def list_for_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -2055,8 +2071,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2137,7 +2154,8 @@ def create_or_update_at_resource_group( :type resource_group_name: str :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -2158,7 +2176,7 @@ def create_or_update_at_resource_group( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -2185,8 +2203,9 @@ def create_or_update_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2237,7 +2256,7 @@ def get_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_resource_group_request( @@ -2252,8 +2271,9 @@ def get_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2300,7 +2320,7 @@ def delete_at_resource_group( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_resource_group_request( @@ -2315,8 +2335,9 @@ def delete_at_resource_group( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2364,7 +2385,7 @@ def list_deployments_at_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationDeploymentsListResult] = kwargs.pop("cls", None) error_map = { @@ -2410,8 +2431,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2452,7 +2474,7 @@ def cancel_at_resource(self, resource_id: str, remediation_name: str, **kwargs: _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_cancel_at_resource_request( @@ -2466,8 +2488,9 @@ def cancel_at_resource(self, resource_id: str, remediation_name: str, **kwargs: request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2506,7 +2529,7 @@ def list_for_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.RemediationListResult] = kwargs.pop("cls", None) error_map = { @@ -2554,8 +2577,9 @@ def extract_data(pipeline_response): def get_next(next_link=None): request = prepare_request(next_link) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2634,7 +2658,8 @@ def create_or_update_at_resource( :type resource_id: str :param remediation_name: The name of the remediation. Required. :type remediation_name: str - :param parameters: The remediation parameters. Is either a model type or a IO type. Required. + :param parameters: The remediation parameters. Is either a Remediation type or a IO type. + Required. :type parameters: ~azure.mgmt.policyinsights.models.Remediation or IO :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. Default value is None. @@ -2655,7 +2680,7 @@ def create_or_update_at_resource( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) @@ -2681,8 +2706,9 @@ def create_or_update_at_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2731,7 +2757,7 @@ def get_at_resource(self, resource_id: str, remediation_name: str, **kwargs: Any _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[_models.Remediation] = kwargs.pop("cls", None) request = build_get_at_resource_request( @@ -2745,8 +2771,9 @@ def get_at_resource(self, resource_id: str, remediation_name: str, **kwargs: Any request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2793,7 +2820,7 @@ def delete_at_resource( _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: Literal["2021-10-01"] = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2021-10-01")) cls: ClsType[Optional[_models.Remediation]] = kwargs.pop("cls", None) request = build_delete_at_resource_request( @@ -2807,8 +2834,9 @@ def delete_at_resource( request = _convert_request(request) request.url = self._client.format_url(request.url) + _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=False, **kwargs + request, stream=_stream, **kwargs ) response = pipeline_response.http_response diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_management_group_scope.py b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_management_group_scope.py index f6dfccaa2af6..92c22bfc8a1d 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_management_group_scope.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_management_group_scope.py @@ -36,6 +36,6 @@ def main(): print(response) -# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2022-03-01/examples/PolicyRestrictions_CheckAtManagementGroupScope.json +# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2023-03-01/examples/PolicyRestrictions_CheckAtManagementGroupScope.json if __name__ == "__main__": main() diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope.py b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope.py index e941240fe97d..ea91494e62c5 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope.py @@ -46,6 +46,6 @@ def main(): print(response) -# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2022-03-01/examples/PolicyRestrictions_CheckAtResourceGroupScope.json +# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2023-03-01/examples/PolicyRestrictions_CheckAtResourceGroupScope.json if __name__ == "__main__": main() diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope_include_audit_effect.py b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope_include_audit_effect.py new file mode 100644 index 000000000000..3eb4ee0441c2 --- /dev/null +++ b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_resource_group_scope_include_audit_effect.py @@ -0,0 +1,52 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential +from azure.mgmt.policyinsights import PolicyInsightsClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-policyinsights +# USAGE + python policy_restrictions_check_at_resource_group_scope_include_audit_effect.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = PolicyInsightsClient( + credential=DefaultAzureCredential(), + subscription_id="35ee058e-5fa0-414c-8145-3ebb8d09b6e2", + ) + + response = client.policy_restrictions.check_at_resource_group_scope( + resource_group_name="vmRg", + parameters={ + "includeAuditEffect": True, + "pendingFields": [ + {"field": "name", "values": ["myVMName"]}, + {"field": "location", "values": ["eastus", "westus", "westus2", "westeurope"]}, + {"field": "tags"}, + ], + "resourceDetails": { + "apiVersion": "2019-12-01", + "resourceContent": {"properties": {"priority": "Spot"}, "type": "Microsoft.Compute/virtualMachines"}, + }, + }, + ) + print(response) + + +# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2023-03-01/examples/PolicyRestrictions_CheckAtResourceGroupScopeIncludeAuditEffect.json +if __name__ == "__main__": + main() diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope.py b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope.py index 1a5168c00e25..3e3edc6b4d1f 100644 --- a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope.py +++ b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope.py @@ -45,6 +45,6 @@ def main(): print(response) -# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2022-03-01/examples/PolicyRestrictions_CheckAtSubscriptionScope.json +# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2023-03-01/examples/PolicyRestrictions_CheckAtSubscriptionScope.json if __name__ == "__main__": main() diff --git a/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope_include_audit_effect.py b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope_include_audit_effect.py new file mode 100644 index 000000000000..314d9c5e7e50 --- /dev/null +++ b/sdk/policyinsights/azure-mgmt-policyinsights/generated_samples/policy_restrictions_check_at_subscription_scope_include_audit_effect.py @@ -0,0 +1,51 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential +from azure.mgmt.policyinsights import PolicyInsightsClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-policyinsights +# USAGE + python policy_restrictions_check_at_subscription_scope_include_audit_effect.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = PolicyInsightsClient( + credential=DefaultAzureCredential(), + subscription_id="35ee058e-5fa0-414c-8145-3ebb8d09b6e2", + ) + + response = client.policy_restrictions.check_at_subscription_scope( + parameters={ + "includeAuditEffect": True, + "pendingFields": [ + {"field": "name", "values": ["myVMName"]}, + {"field": "location", "values": ["eastus", "westus", "westus2", "westeurope"]}, + {"field": "tags"}, + ], + "resourceDetails": { + "apiVersion": "2019-12-01", + "resourceContent": {"properties": {"priority": "Spot"}, "type": "Microsoft.Compute/virtualMachines"}, + }, + }, + ) + print(response) + + +# x-ms-original-file: specification/policyinsights/resource-manager/Microsoft.PolicyInsights/stable/2023-03-01/examples/PolicyRestrictions_CheckAtSubscriptionScopeIncludeAuditEffect.json +if __name__ == "__main__": + main()