From 19abde0a606a0dc57c2ee266614a0dace2c24308 Mon Sep 17 00:00:00 2001 From: SDKAuto Date: Wed, 5 Jan 2022 15:02:08 +0000 Subject: [PATCH] CodeGen from PR 17107 in Azure/azure-rest-api-specs [DO NOT MERGE ] Introduce new scanners resource type (#17107) * introduce new scanners resource type * fix error * fixing errors * fixing errors * fix error * fixing errors * fixing comments * adding new tag --- sdk/security/azure-mgmt-security/_meta.json | 10 +- .../azure/mgmt/security/__init__.py | 9 +- .../azure/mgmt/security/_configuration.py | 21 +- .../azure/mgmt/security/_metadata.json | 29 +- .../azure/mgmt/security/_patch.py | 31 + .../azure/mgmt/security/_security_center.py | 349 +- .../azure/mgmt/security/_vendor.py | 27 + .../azure/mgmt/security/_version.py | 2 +- .../azure/mgmt/security/aio/__init__.py | 5 + .../azure/mgmt/security/aio/_configuration.py | 6 +- .../azure/mgmt/security/aio/_patch.py | 31 + .../mgmt/security/aio/_security_center.py | 344 +- .../mgmt/security/aio/operations/__init__.py | 8 + ...daptive_application_controls_operations.py | 141 +- ..._adaptive_network_hardenings_operations.py | 180 +- .../_advanced_threat_protection_operations.py | 77 +- .../aio/operations/_alerts_operations.py | 494 +- .../_alerts_suppression_rules_operations.py | 154 +- .../_allowed_connections_operations.py | 139 +- .../_assessments_metadata_operations.py | 258 +- .../aio/operations/_assessments_operations.py | 175 +- .../_auto_provisioning_settings_operations.py | 122 +- .../aio/operations/_automations_operations.py | 233 +- .../_compliance_results_operations.py | 85 +- .../aio/operations/_compliances_operations.py | 79 +- .../aio/operations/_connectors_operations.py | 150 +- ...ustom_assessment_automations_operations.py | 360 + ...tom_entity_store_assignments_operations.py | 362 + .../_device_security_groups_operations.py | 150 +- ...iscovered_security_solutions_operations.py | 139 +- ..._external_security_solutions_operations.py | 139 +- ...ormation_protection_policies_operations.py | 130 +- .../_ingestion_settings_operations.py | 208 +- ..._security_solution_analytics_operations.py | 70 +- .../_iot_security_solution_operations.py | 256 +- ...s_analytics_aggregated_alert_operations.py | 131 +- ...ons_analytics_recommendation_operations.py | 99 +- ..._jit_network_access_policies_operations.py | 358 +- .../aio/operations/_locations_operations.py | 79 +- .../operations/_mde_onboardings_operations.py | 131 + .../security/aio/operations/_operations.py | 42 +- .../aio/operations/_pricings_operations.py | 101 +- ...atory_compliance_assessments_operations.py | 99 +- ...gulatory_compliance_controls_operations.py | 94 +- ...ulatory_compliance_standards_operations.py | 89 +- ...re_score_control_definitions_operations.py | 95 +- .../_secure_score_controls_operations.py | 114 +- .../operations/_secure_scores_operations.py | 79 +- .../_security_connectors_operations.py | 410 + .../_security_contacts_operations.py | 184 +- .../_security_solutions_operations.py | 89 +- ...ity_solutions_reference_data_operations.py | 64 +- ...ver_vulnerability_assessment_operations.py | 179 +- .../aio/operations/_settings_operations.py | 124 +- .../_software_inventories_operations.py | 140 +- ...ty_assessment_baseline_rules_operations.py | 182 +- ...lity_assessment_scan_results_operations.py | 72 +- ...lnerability_assessment_scans_operations.py | 68 +- .../operations/_sub_assessments_operations.py | 137 +- .../aio/operations/_tasks_operations.py | 291 +- .../aio/operations/_topology_operations.py | 127 +- .../_workspace_settings_operations.py | 187 +- .../azure/mgmt/security/models/__init__.py | 794 +- .../azure/mgmt/security/models/_models.py | 9600 -------------- .../azure/mgmt/security/models/_models_py3.py | 10528 ++++++++++------ .../security/models/_security_center_enums.py | 336 +- .../mgmt/security/operations/__init__.py | 8 + ...daptive_application_controls_operations.py | 338 +- ..._adaptive_network_hardenings_operations.py | 383 +- .../_advanced_threat_protection_operations.py | 181 +- .../security/operations/_alerts_operations.py | 1070 +- .../_alerts_suppression_rules_operations.py | 336 +- .../_allowed_connections_operations.py | 264 +- .../_assessments_metadata_operations.py | 493 +- .../operations/_assessments_operations.py | 361 +- .../_auto_provisioning_settings_operations.py | 258 +- .../operations/_automations_operations.py | 519 +- .../_compliance_results_operations.py | 173 +- .../operations/_compliances_operations.py | 170 +- .../operations/_connectors_operations.py | 326 +- ...ustom_assessment_automations_operations.py | 540 + ...tom_entity_store_assignments_operations.py | 542 + .../_device_security_groups_operations.py | 334 +- ...iscovered_security_solutions_operations.py | 270 +- ..._external_security_solutions_operations.py | 264 +- ...ormation_protection_policies_operations.py | 271 +- .../_ingestion_settings_operations.py | 464 +- ..._security_solution_analytics_operations.py | 170 +- .../_iot_security_solution_operations.py | 548 +- ...s_analytics_aggregated_alert_operations.py | 288 +- ...ons_analytics_recommendation_operations.py | 204 +- ..._jit_network_access_policies_operations.py | 722 +- .../operations/_locations_operations.py | 164 +- .../operations/_mde_onboardings_operations.py | 195 + .../mgmt/security/operations/_operations.py | 83 +- .../operations/_pricings_operations.py | 241 +- ...atory_compliance_assessments_operations.py | 204 +- ...gulatory_compliance_controls_operations.py | 195 +- ...ulatory_compliance_standards_operations.py | 182 +- ...re_score_control_definitions_operations.py | 172 +- .../_secure_score_controls_operations.py | 203 +- .../operations/_secure_scores_operations.py | 166 +- .../_security_connectors_operations.py | 632 + .../_security_contacts_operations.py | 411 +- .../_security_solutions_operations.py | 179 +- ...ity_solutions_reference_data_operations.py | 150 +- ...ver_vulnerability_assessment_operations.py | 418 +- .../operations/_settings_operations.py | 259 +- .../_software_inventories_operations.py | 295 +- ...ty_assessment_baseline_rules_operations.py | 442 +- ...lity_assessment_scan_results_operations.py | 182 +- ...lnerability_assessment_scans_operations.py | 170 +- .../operations/_sub_assessments_operations.py | 268 +- .../security/operations/_tasks_operations.py | 616 +- .../operations/_topology_operations.py | 258 +- .../_workspace_settings_operations.py | 414 +- 116 files changed, 23560 insertions(+), 22432 deletions(-) create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/_patch.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/_vendor.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_patch.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_assessment_automations_operations.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_entity_store_assignments_operations.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_mde_onboardings_operations.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_connectors_operations.py delete mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_assessment_automations_operations.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_entity_store_assignments_operations.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_mde_onboardings_operations.py create mode 100644 sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_connectors_operations.py diff --git a/sdk/security/azure-mgmt-security/_meta.json b/sdk/security/azure-mgmt-security/_meta.json index 4fcd76c66d92d..50989ec7d78e7 100644 --- a/sdk/security/azure-mgmt-security/_meta.json +++ b/sdk/security/azure-mgmt-security/_meta.json @@ -1,11 +1,11 @@ { - "autorest": "3.4.5", + "autorest": "3.7.2", "use": [ - "@autorest/python@5.8.4", - "@autorest/modelerfour@4.19.2" + "@autorest/python@5.12.0", + "@autorest/modelerfour@4.19.3" ], - "commit": "a8719243647b7c6d06ed287483d788808de7ecab", + "commit": "30fca88fa23112e4296322f38610e02da522d390", "repository_url": "https://github.com/Azure/azure-rest-api-specs", - "autorest_command": "autorest specification/security/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --track2 --use=@autorest/python@5.8.4 --use=@autorest/modelerfour@4.19.2 --version=3.4.5", + "autorest_command": "autorest specification/security/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --python3-only --track2 --use=@autorest/python@5.12.0 --use=@autorest/modelerfour@4.19.3 --version=3.7.2", "readme": "specification/security/resource-manager/readme.md" } \ No newline at end of file diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/__init__.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/__init__.py index 48195ae40f80f..e0ed98854a42a 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/__init__.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/__init__.py @@ -12,8 +12,7 @@ __version__ = VERSION __all__ = ['SecurityCenter'] -try: - from ._patch import patch_sdk # type: ignore - patch_sdk() -except ImportError: - pass +# `._patch.py` is used for handwritten extensions to the generated code +# Example: https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md +from ._patch import patch_sdk +patch_sdk() diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/_configuration.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/_configuration.py index 9e9ac18611d0f..dec320ff2538d 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/_configuration.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/_configuration.py @@ -6,18 +6,16 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +from typing import Any, TYPE_CHECKING from azure.core.configuration import Configuration from azure.core.pipeline import policies -from azure.mgmt.core.policies import ARMHttpLoggingPolicy +from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy from ._version import VERSION if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports - from typing import Any - from azure.core.credentials import TokenCredential @@ -37,19 +35,18 @@ class SecurityCenterConfiguration(Configuration): def __init__( self, - credential, # type: "TokenCredential" - subscription_id, # type: str - asc_location, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + credential: "TokenCredential", + subscription_id: str, + asc_location: str, + **kwargs: Any + ) -> None: + super(SecurityCenterConfiguration, self).__init__(**kwargs) if credential is None: raise ValueError("Parameter 'credential' must not be None.") if subscription_id is None: raise ValueError("Parameter 'subscription_id' must not be None.") if asc_location is None: raise ValueError("Parameter 'asc_location' must not be None.") - super(SecurityCenterConfiguration, self).__init__(**kwargs) self.credential = credential self.subscription_id = subscription_id @@ -73,4 +70,4 @@ def _configure( self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/_metadata.json b/sdk/security/azure-mgmt-security/azure/mgmt/security/_metadata.json index 57c07b73f3a86..d5a17b09eee66 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/_metadata.json +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/_metadata.json @@ -1,17 +1,17 @@ { "chosen_version": "", - "total_api_version_list": ["2015-06-01-preview", "2017-08-01", "2017-08-01-preview", "2018-06-01", "2019-01-01", "2019-01-01-preview", "2019-08-01", "2020-01-01", "2020-01-01-preview", "2020-07-01-preview", "2021-01-01", "2021-01-15-preview", "2021-05-01-preview", "2021-07-01"], + "total_api_version_list": ["2015-06-01-preview", "2017-08-01", "2017-08-01-preview", "2018-06-01", "2019-01-01", "2019-01-01-preview", "2019-08-01", "2020-01-01", "2020-01-01-preview", "2020-07-01-preview", "2021-01-01", "2021-01-15-preview", "2021-05-01-preview", "2021-06-01", "2021-07-01", "2021-07-01-preview", "2021-10-01-preview"], "client": { "name": "SecurityCenter", "filename": "_security_center", "description": "API spec for Microsoft.Security (Azure Security Center) resource provider.", - "base_url": "\u0027https://management.azure.com\u0027", - "custom_base_url": null, + "host_value": "\"https://management.azure.com\"", + "parameterized_host_template": null, "azure_arm": true, "has_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}, \"regular\": {\"azurecore\": {\"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"], \"msrest\": [\"Deserializer\", \"Serializer\"], \"azure.mgmt.core\": [\"ARMPipelineClient\"]}, \"local\": {\"._configuration\": [\"SecurityCenterConfiguration\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}, \"azurecore\": {\"azure.core.pipeline.transport\": [\"HttpRequest\", \"HttpResponse\"]}}}", - "async_imports": "{\"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}, \"regular\": {\"azurecore\": {\"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"], \"msrest\": [\"Deserializer\", \"Serializer\"], \"azure.mgmt.core\": [\"AsyncARMPipelineClient\"]}, \"local\": {\"._configuration\": [\"SecurityCenterConfiguration\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}, \"azurecore\": {\"azure.core.pipeline.transport\": [\"AsyncHttpResponse\", \"HttpRequest\"]}}}" + "sync_imports": "{\"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}, \"regular\": {\"azurecore\": {\"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"], \"msrest\": [\"Deserializer\", \"Serializer\"], \"azure.mgmt.core\": [\"ARMPipelineClient\"]}, \"local\": {\"._configuration\": [\"SecurityCenterConfiguration\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}}", + "async_imports": "{\"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"], \"azure.core.credentials\": [\"TokenCredential\"]}}, \"regular\": {\"azurecore\": {\"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"], \"msrest\": [\"Deserializer\", \"Serializer\"], \"azure.mgmt.core\": [\"AsyncARMPipelineClient\"]}, \"local\": {\"._configuration\": [\"SecurityCenterConfiguration\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}}" }, "global_parameters": { "sync": { @@ -66,7 +66,7 @@ "required": false }, "base_url": { - "signature": "base_url=None, # type: Optional[str]", + "signature": "base_url=\"https://management.azure.com\", # type: str", "description": "Service URL", "docstring_type": "str", "required": false @@ -86,7 +86,7 @@ "required": false }, "base_url": { - "signature": "base_url: Optional[str] = None,", + "signature": "base_url: str = \"https://management.azure.com\",", "description": "Service URL", "docstring_type": "str", "required": false @@ -103,13 +103,15 @@ "config": { "credential": true, "credential_scopes": ["https://management.azure.com/.default"], - "credential_default_policy_type": "BearerTokenCredentialPolicy", - "credential_default_policy_type_has_async_version": true, - "credential_key_header_name": null, - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", + "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", + "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { + "mde_onboardings": "MdeOnboardingsOperations", + "custom_assessment_automations": "CustomAssessmentAutomationsOperations", + "custom_entity_store_assignments": "CustomEntityStoreAssignmentsOperations", "compliance_results": "ComplianceResultsOperations", "pricings": "PricingsOperations", "advanced_threat_protection": "AdvancedThreatProtectionOperations", @@ -154,6 +156,7 @@ "alerts": "AlertsOperations", "settings": "SettingsOperations", "ingestion_settings": "IngestionSettingsOperations", - "software_inventories": "SoftwareInventoriesOperations" + "software_inventories": "SoftwareInventoriesOperations", + "security_connectors": "SecurityConnectorsOperations" } } \ No newline at end of file diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/_patch.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/_patch.py new file mode 100644 index 0000000000000..74e48ecd07cf3 --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/_patch.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# +# Copyright (c) Microsoft Corporation. All rights reserved. +# +# The MIT License (MIT) +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the ""Software""), to +# deal in the Software without restriction, including without limitation the +# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or +# sell copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING +# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS +# IN THE SOFTWARE. +# +# -------------------------------------------------------------------------- + +# This file is used for handwritten extensions to the generated code. Example: +# https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md +def patch_sdk(): + pass \ No newline at end of file diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/_security_center.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/_security_center.py index 0c411e391bbb8..51d9fdbe823b2 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/_security_center.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/_security_center.py @@ -6,86 +6,54 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +from copy import deepcopy +from typing import Any, Optional, TYPE_CHECKING +from azure.core.rest import HttpRequest, HttpResponse from azure.mgmt.core import ARMPipelineClient from msrest import Deserializer, Serializer +from . import models +from ._configuration import SecurityCenterConfiguration +from .operations import AdaptiveApplicationControlsOperations, AdaptiveNetworkHardeningsOperations, AdvancedThreatProtectionOperations, AlertsOperations, AlertsSuppressionRulesOperations, AllowedConnectionsOperations, AssessmentsMetadataOperations, AssessmentsOperations, AutoProvisioningSettingsOperations, AutomationsOperations, ComplianceResultsOperations, CompliancesOperations, ConnectorsOperations, CustomAssessmentAutomationsOperations, CustomEntityStoreAssignmentsOperations, DeviceSecurityGroupsOperations, DiscoveredSecuritySolutionsOperations, ExternalSecuritySolutionsOperations, InformationProtectionPoliciesOperations, IngestionSettingsOperations, IotSecuritySolutionAnalyticsOperations, IotSecuritySolutionOperations, IotSecuritySolutionsAnalyticsAggregatedAlertOperations, IotSecuritySolutionsAnalyticsRecommendationOperations, JitNetworkAccessPoliciesOperations, LocationsOperations, MdeOnboardingsOperations, Operations, PricingsOperations, RegulatoryComplianceAssessmentsOperations, RegulatoryComplianceControlsOperations, RegulatoryComplianceStandardsOperations, SecureScoreControlDefinitionsOperations, SecureScoreControlsOperations, SecureScoresOperations, SecurityConnectorsOperations, SecurityContactsOperations, SecuritySolutionsOperations, SecuritySolutionsReferenceDataOperations, ServerVulnerabilityAssessmentOperations, SettingsOperations, SoftwareInventoriesOperations, SqlVulnerabilityAssessmentBaselineRulesOperations, SqlVulnerabilityAssessmentScanResultsOperations, SqlVulnerabilityAssessmentScansOperations, SubAssessmentsOperations, TasksOperations, TopologyOperations, WorkspaceSettingsOperations + if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Optional - from azure.core.credentials import TokenCredential - from azure.core.pipeline.transport import HttpRequest, HttpResponse - -from ._configuration import SecurityCenterConfiguration -from .operations import ComplianceResultsOperations -from .operations import PricingsOperations -from .operations import AdvancedThreatProtectionOperations -from .operations import DeviceSecurityGroupsOperations -from .operations import IotSecuritySolutionOperations -from .operations import IotSecuritySolutionAnalyticsOperations -from .operations import IotSecuritySolutionsAnalyticsAggregatedAlertOperations -from .operations import IotSecuritySolutionsAnalyticsRecommendationOperations -from .operations import LocationsOperations -from .operations import Operations -from .operations import TasksOperations -from .operations import AutoProvisioningSettingsOperations -from .operations import CompliancesOperations -from .operations import InformationProtectionPoliciesOperations -from .operations import SecurityContactsOperations -from .operations import WorkspaceSettingsOperations -from .operations import RegulatoryComplianceStandardsOperations -from .operations import RegulatoryComplianceControlsOperations -from .operations import RegulatoryComplianceAssessmentsOperations -from .operations import SubAssessmentsOperations -from .operations import AutomationsOperations -from .operations import AlertsSuppressionRulesOperations -from .operations import ServerVulnerabilityAssessmentOperations -from .operations import AssessmentsMetadataOperations -from .operations import AssessmentsOperations -from .operations import AdaptiveApplicationControlsOperations -from .operations import AdaptiveNetworkHardeningsOperations -from .operations import AllowedConnectionsOperations -from .operations import TopologyOperations -from .operations import JitNetworkAccessPoliciesOperations -from .operations import DiscoveredSecuritySolutionsOperations -from .operations import SecuritySolutionsReferenceDataOperations -from .operations import ExternalSecuritySolutionsOperations -from .operations import SecureScoresOperations -from .operations import SecureScoreControlsOperations -from .operations import SecureScoreControlDefinitionsOperations -from .operations import SecuritySolutionsOperations -from .operations import ConnectorsOperations -from .operations import SqlVulnerabilityAssessmentScansOperations -from .operations import SqlVulnerabilityAssessmentScanResultsOperations -from .operations import SqlVulnerabilityAssessmentBaselineRulesOperations -from .operations import AlertsOperations -from .operations import SettingsOperations -from .operations import IngestionSettingsOperations -from .operations import SoftwareInventoriesOperations -from . import models - -class SecurityCenter(object): +class SecurityCenter: """API spec for Microsoft.Security (Azure Security Center) resource provider. + :ivar mde_onboardings: MdeOnboardingsOperations operations + :vartype mde_onboardings: azure.mgmt.security.operations.MdeOnboardingsOperations + :ivar custom_assessment_automations: CustomAssessmentAutomationsOperations operations + :vartype custom_assessment_automations: + azure.mgmt.security.operations.CustomAssessmentAutomationsOperations + :ivar custom_entity_store_assignments: CustomEntityStoreAssignmentsOperations operations + :vartype custom_entity_store_assignments: + azure.mgmt.security.operations.CustomEntityStoreAssignmentsOperations :ivar compliance_results: ComplianceResultsOperations operations :vartype compliance_results: azure.mgmt.security.operations.ComplianceResultsOperations :ivar pricings: PricingsOperations operations :vartype pricings: azure.mgmt.security.operations.PricingsOperations :ivar advanced_threat_protection: AdvancedThreatProtectionOperations operations - :vartype advanced_threat_protection: azure.mgmt.security.operations.AdvancedThreatProtectionOperations + :vartype advanced_threat_protection: + azure.mgmt.security.operations.AdvancedThreatProtectionOperations :ivar device_security_groups: DeviceSecurityGroupsOperations operations :vartype device_security_groups: azure.mgmt.security.operations.DeviceSecurityGroupsOperations :ivar iot_security_solution: IotSecuritySolutionOperations operations :vartype iot_security_solution: azure.mgmt.security.operations.IotSecuritySolutionOperations :ivar iot_security_solution_analytics: IotSecuritySolutionAnalyticsOperations operations - :vartype iot_security_solution_analytics: azure.mgmt.security.operations.IotSecuritySolutionAnalyticsOperations - :ivar iot_security_solutions_analytics_aggregated_alert: IotSecuritySolutionsAnalyticsAggregatedAlertOperations operations - :vartype iot_security_solutions_analytics_aggregated_alert: azure.mgmt.security.operations.IotSecuritySolutionsAnalyticsAggregatedAlertOperations - :ivar iot_security_solutions_analytics_recommendation: IotSecuritySolutionsAnalyticsRecommendationOperations operations - :vartype iot_security_solutions_analytics_recommendation: azure.mgmt.security.operations.IotSecuritySolutionsAnalyticsRecommendationOperations + :vartype iot_security_solution_analytics: + azure.mgmt.security.operations.IotSecuritySolutionAnalyticsOperations + :ivar iot_security_solutions_analytics_aggregated_alert: + IotSecuritySolutionsAnalyticsAggregatedAlertOperations operations + :vartype iot_security_solutions_analytics_aggregated_alert: + azure.mgmt.security.operations.IotSecuritySolutionsAnalyticsAggregatedAlertOperations + :ivar iot_security_solutions_analytics_recommendation: + IotSecuritySolutionsAnalyticsRecommendationOperations operations + :vartype iot_security_solutions_analytics_recommendation: + azure.mgmt.security.operations.IotSecuritySolutionsAnalyticsRecommendationOperations :ivar locations: LocationsOperations operations :vartype locations: azure.mgmt.security.operations.LocationsOperations :ivar operations: Operations operations @@ -93,65 +61,84 @@ class SecurityCenter(object): :ivar tasks: TasksOperations operations :vartype tasks: azure.mgmt.security.operations.TasksOperations :ivar auto_provisioning_settings: AutoProvisioningSettingsOperations operations - :vartype auto_provisioning_settings: azure.mgmt.security.operations.AutoProvisioningSettingsOperations + :vartype auto_provisioning_settings: + azure.mgmt.security.operations.AutoProvisioningSettingsOperations :ivar compliances: CompliancesOperations operations :vartype compliances: azure.mgmt.security.operations.CompliancesOperations :ivar information_protection_policies: InformationProtectionPoliciesOperations operations - :vartype information_protection_policies: azure.mgmt.security.operations.InformationProtectionPoliciesOperations + :vartype information_protection_policies: + azure.mgmt.security.operations.InformationProtectionPoliciesOperations :ivar security_contacts: SecurityContactsOperations operations :vartype security_contacts: azure.mgmt.security.operations.SecurityContactsOperations :ivar workspace_settings: WorkspaceSettingsOperations operations :vartype workspace_settings: azure.mgmt.security.operations.WorkspaceSettingsOperations :ivar regulatory_compliance_standards: RegulatoryComplianceStandardsOperations operations - :vartype regulatory_compliance_standards: azure.mgmt.security.operations.RegulatoryComplianceStandardsOperations + :vartype regulatory_compliance_standards: + azure.mgmt.security.operations.RegulatoryComplianceStandardsOperations :ivar regulatory_compliance_controls: RegulatoryComplianceControlsOperations operations - :vartype regulatory_compliance_controls: azure.mgmt.security.operations.RegulatoryComplianceControlsOperations + :vartype regulatory_compliance_controls: + azure.mgmt.security.operations.RegulatoryComplianceControlsOperations :ivar regulatory_compliance_assessments: RegulatoryComplianceAssessmentsOperations operations - :vartype regulatory_compliance_assessments: azure.mgmt.security.operations.RegulatoryComplianceAssessmentsOperations + :vartype regulatory_compliance_assessments: + azure.mgmt.security.operations.RegulatoryComplianceAssessmentsOperations :ivar sub_assessments: SubAssessmentsOperations operations :vartype sub_assessments: azure.mgmt.security.operations.SubAssessmentsOperations :ivar automations: AutomationsOperations operations :vartype automations: azure.mgmt.security.operations.AutomationsOperations :ivar alerts_suppression_rules: AlertsSuppressionRulesOperations operations - :vartype alerts_suppression_rules: azure.mgmt.security.operations.AlertsSuppressionRulesOperations + :vartype alerts_suppression_rules: + azure.mgmt.security.operations.AlertsSuppressionRulesOperations :ivar server_vulnerability_assessment: ServerVulnerabilityAssessmentOperations operations - :vartype server_vulnerability_assessment: azure.mgmt.security.operations.ServerVulnerabilityAssessmentOperations + :vartype server_vulnerability_assessment: + azure.mgmt.security.operations.ServerVulnerabilityAssessmentOperations :ivar assessments_metadata: AssessmentsMetadataOperations operations :vartype assessments_metadata: azure.mgmt.security.operations.AssessmentsMetadataOperations :ivar assessments: AssessmentsOperations operations :vartype assessments: azure.mgmt.security.operations.AssessmentsOperations :ivar adaptive_application_controls: AdaptiveApplicationControlsOperations operations - :vartype adaptive_application_controls: azure.mgmt.security.operations.AdaptiveApplicationControlsOperations + :vartype adaptive_application_controls: + azure.mgmt.security.operations.AdaptiveApplicationControlsOperations :ivar adaptive_network_hardenings: AdaptiveNetworkHardeningsOperations operations - :vartype adaptive_network_hardenings: azure.mgmt.security.operations.AdaptiveNetworkHardeningsOperations + :vartype adaptive_network_hardenings: + azure.mgmt.security.operations.AdaptiveNetworkHardeningsOperations :ivar allowed_connections: AllowedConnectionsOperations operations :vartype allowed_connections: azure.mgmt.security.operations.AllowedConnectionsOperations :ivar topology: TopologyOperations operations :vartype topology: azure.mgmt.security.operations.TopologyOperations :ivar jit_network_access_policies: JitNetworkAccessPoliciesOperations operations - :vartype jit_network_access_policies: azure.mgmt.security.operations.JitNetworkAccessPoliciesOperations + :vartype jit_network_access_policies: + azure.mgmt.security.operations.JitNetworkAccessPoliciesOperations :ivar discovered_security_solutions: DiscoveredSecuritySolutionsOperations operations - :vartype discovered_security_solutions: azure.mgmt.security.operations.DiscoveredSecuritySolutionsOperations + :vartype discovered_security_solutions: + azure.mgmt.security.operations.DiscoveredSecuritySolutionsOperations :ivar security_solutions_reference_data: SecuritySolutionsReferenceDataOperations operations - :vartype security_solutions_reference_data: azure.mgmt.security.operations.SecuritySolutionsReferenceDataOperations + :vartype security_solutions_reference_data: + azure.mgmt.security.operations.SecuritySolutionsReferenceDataOperations :ivar external_security_solutions: ExternalSecuritySolutionsOperations operations - :vartype external_security_solutions: azure.mgmt.security.operations.ExternalSecuritySolutionsOperations + :vartype external_security_solutions: + azure.mgmt.security.operations.ExternalSecuritySolutionsOperations :ivar secure_scores: SecureScoresOperations operations :vartype secure_scores: azure.mgmt.security.operations.SecureScoresOperations :ivar secure_score_controls: SecureScoreControlsOperations operations :vartype secure_score_controls: azure.mgmt.security.operations.SecureScoreControlsOperations :ivar secure_score_control_definitions: SecureScoreControlDefinitionsOperations operations - :vartype secure_score_control_definitions: azure.mgmt.security.operations.SecureScoreControlDefinitionsOperations + :vartype secure_score_control_definitions: + azure.mgmt.security.operations.SecureScoreControlDefinitionsOperations :ivar security_solutions: SecuritySolutionsOperations operations :vartype security_solutions: azure.mgmt.security.operations.SecuritySolutionsOperations :ivar connectors: ConnectorsOperations operations :vartype connectors: azure.mgmt.security.operations.ConnectorsOperations :ivar sql_vulnerability_assessment_scans: SqlVulnerabilityAssessmentScansOperations operations - :vartype sql_vulnerability_assessment_scans: azure.mgmt.security.operations.SqlVulnerabilityAssessmentScansOperations - :ivar sql_vulnerability_assessment_scan_results: SqlVulnerabilityAssessmentScanResultsOperations operations - :vartype sql_vulnerability_assessment_scan_results: azure.mgmt.security.operations.SqlVulnerabilityAssessmentScanResultsOperations - :ivar sql_vulnerability_assessment_baseline_rules: SqlVulnerabilityAssessmentBaselineRulesOperations operations - :vartype sql_vulnerability_assessment_baseline_rules: azure.mgmt.security.operations.SqlVulnerabilityAssessmentBaselineRulesOperations + :vartype sql_vulnerability_assessment_scans: + azure.mgmt.security.operations.SqlVulnerabilityAssessmentScansOperations + :ivar sql_vulnerability_assessment_scan_results: + SqlVulnerabilityAssessmentScanResultsOperations operations + :vartype sql_vulnerability_assessment_scan_results: + azure.mgmt.security.operations.SqlVulnerabilityAssessmentScanResultsOperations + :ivar sql_vulnerability_assessment_baseline_rules: + SqlVulnerabilityAssessmentBaselineRulesOperations operations + :vartype sql_vulnerability_assessment_baseline_rules: + azure.mgmt.security.operations.SqlVulnerabilityAssessmentBaselineRulesOperations :ivar alerts: AlertsOperations operations :vartype alerts: azure.mgmt.security.operations.AlertsOperations :ivar settings: SettingsOperations operations @@ -160,144 +147,112 @@ class SecurityCenter(object): :vartype ingestion_settings: azure.mgmt.security.operations.IngestionSettingsOperations :ivar software_inventories: SoftwareInventoriesOperations operations :vartype software_inventories: azure.mgmt.security.operations.SoftwareInventoriesOperations + :ivar security_connectors: SecurityConnectorsOperations operations + :vartype security_connectors: azure.mgmt.security.operations.SecurityConnectorsOperations :param credential: Credential needed for the client to connect to Azure. :type credential: ~azure.core.credentials.TokenCredential :param subscription_id: Azure subscription ID. :type subscription_id: str - :param asc_location: The location where ASC stores the data of the subscription. can be retrieved from Get locations. + :param asc_location: The location where ASC stores the data of the subscription. can be + retrieved from Get locations. :type asc_location: str - :param str base_url: Service URL - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :param base_url: Service URL. Default value is 'https://management.azure.com'. + :type base_url: str + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. """ def __init__( self, - credential, # type: "TokenCredential" - subscription_id, # type: str - asc_location, # type: str - base_url=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> None - if not base_url: - base_url = 'https://management.azure.com' - self._config = SecurityCenterConfiguration(credential, subscription_id, asc_location, **kwargs) + credential: "TokenCredential", + subscription_id: str, + asc_location: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: + self._config = SecurityCenterConfiguration(credential=credential, subscription_id=subscription_id, asc_location=asc_location, **kwargs) self._client = 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) - self._serialize.client_side_validation = False self._deserialize = Deserializer(client_models) + self._serialize.client_side_validation = False + self.mde_onboardings = MdeOnboardingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.custom_assessment_automations = CustomAssessmentAutomationsOperations(self._client, self._config, self._serialize, self._deserialize) + self.custom_entity_store_assignments = CustomEntityStoreAssignmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.compliance_results = ComplianceResultsOperations(self._client, self._config, self._serialize, self._deserialize) + self.pricings = PricingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.advanced_threat_protection = AdvancedThreatProtectionOperations(self._client, self._config, self._serialize, self._deserialize) + self.device_security_groups = DeviceSecurityGroupsOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solution = IotSecuritySolutionOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solution_analytics = IotSecuritySolutionAnalyticsOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solutions_analytics_aggregated_alert = IotSecuritySolutionsAnalyticsAggregatedAlertOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solutions_analytics_recommendation = IotSecuritySolutionsAnalyticsRecommendationOperations(self._client, self._config, self._serialize, self._deserialize) + self.locations = LocationsOperations(self._client, self._config, self._serialize, self._deserialize) + self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) + self.tasks = TasksOperations(self._client, self._config, self._serialize, self._deserialize) + self.auto_provisioning_settings = AutoProvisioningSettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.compliances = CompliancesOperations(self._client, self._config, self._serialize, self._deserialize) + self.information_protection_policies = InformationProtectionPoliciesOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_contacts = SecurityContactsOperations(self._client, self._config, self._serialize, self._deserialize) + self.workspace_settings = WorkspaceSettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.regulatory_compliance_standards = RegulatoryComplianceStandardsOperations(self._client, self._config, self._serialize, self._deserialize) + self.regulatory_compliance_controls = RegulatoryComplianceControlsOperations(self._client, self._config, self._serialize, self._deserialize) + self.regulatory_compliance_assessments = RegulatoryComplianceAssessmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.sub_assessments = SubAssessmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.automations = AutomationsOperations(self._client, self._config, self._serialize, self._deserialize) + self.alerts_suppression_rules = AlertsSuppressionRulesOperations(self._client, self._config, self._serialize, self._deserialize) + self.server_vulnerability_assessment = ServerVulnerabilityAssessmentOperations(self._client, self._config, self._serialize, self._deserialize) + self.assessments_metadata = AssessmentsMetadataOperations(self._client, self._config, self._serialize, self._deserialize) + self.assessments = AssessmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.adaptive_application_controls = AdaptiveApplicationControlsOperations(self._client, self._config, self._serialize, self._deserialize) + self.adaptive_network_hardenings = AdaptiveNetworkHardeningsOperations(self._client, self._config, self._serialize, self._deserialize) + self.allowed_connections = AllowedConnectionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.topology = TopologyOperations(self._client, self._config, self._serialize, self._deserialize) + self.jit_network_access_policies = JitNetworkAccessPoliciesOperations(self._client, self._config, self._serialize, self._deserialize) + self.discovered_security_solutions = DiscoveredSecuritySolutionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_solutions_reference_data = SecuritySolutionsReferenceDataOperations(self._client, self._config, self._serialize, self._deserialize) + self.external_security_solutions = ExternalSecuritySolutionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.secure_scores = SecureScoresOperations(self._client, self._config, self._serialize, self._deserialize) + self.secure_score_controls = SecureScoreControlsOperations(self._client, self._config, self._serialize, self._deserialize) + self.secure_score_control_definitions = SecureScoreControlDefinitionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_solutions = SecuritySolutionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.connectors = ConnectorsOperations(self._client, self._config, self._serialize, self._deserialize) + self.sql_vulnerability_assessment_scans = SqlVulnerabilityAssessmentScansOperations(self._client, self._config, self._serialize, self._deserialize) + self.sql_vulnerability_assessment_scan_results = SqlVulnerabilityAssessmentScanResultsOperations(self._client, self._config, self._serialize, self._deserialize) + self.sql_vulnerability_assessment_baseline_rules = SqlVulnerabilityAssessmentBaselineRulesOperations(self._client, self._config, self._serialize, self._deserialize) + self.alerts = AlertsOperations(self._client, self._config, self._serialize, self._deserialize) + self.settings = SettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.ingestion_settings = IngestionSettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.software_inventories = SoftwareInventoriesOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_connectors = SecurityConnectorsOperations(self._client, self._config, self._serialize, self._deserialize) - self.compliance_results = ComplianceResultsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.pricings = PricingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.advanced_threat_protection = AdvancedThreatProtectionOperations( - self._client, self._config, self._serialize, self._deserialize) - self.device_security_groups = DeviceSecurityGroupsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solution = IotSecuritySolutionOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solution_analytics = IotSecuritySolutionAnalyticsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solutions_analytics_aggregated_alert = IotSecuritySolutionsAnalyticsAggregatedAlertOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solutions_analytics_recommendation = IotSecuritySolutionsAnalyticsRecommendationOperations( - self._client, self._config, self._serialize, self._deserialize) - self.locations = LocationsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.operations = Operations( - self._client, self._config, self._serialize, self._deserialize) - self.tasks = TasksOperations( - self._client, self._config, self._serialize, self._deserialize) - self.auto_provisioning_settings = AutoProvisioningSettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.compliances = CompliancesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.information_protection_policies = InformationProtectionPoliciesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.security_contacts = SecurityContactsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.workspace_settings = WorkspaceSettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.regulatory_compliance_standards = RegulatoryComplianceStandardsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.regulatory_compliance_controls = RegulatoryComplianceControlsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.regulatory_compliance_assessments = RegulatoryComplianceAssessmentsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sub_assessments = SubAssessmentsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.automations = AutomationsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.alerts_suppression_rules = AlertsSuppressionRulesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.server_vulnerability_assessment = ServerVulnerabilityAssessmentOperations( - self._client, self._config, self._serialize, self._deserialize) - self.assessments_metadata = AssessmentsMetadataOperations( - self._client, self._config, self._serialize, self._deserialize) - self.assessments = AssessmentsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.adaptive_application_controls = AdaptiveApplicationControlsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.adaptive_network_hardenings = AdaptiveNetworkHardeningsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.allowed_connections = AllowedConnectionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.topology = TopologyOperations( - self._client, self._config, self._serialize, self._deserialize) - self.jit_network_access_policies = JitNetworkAccessPoliciesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.discovered_security_solutions = DiscoveredSecuritySolutionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.security_solutions_reference_data = SecuritySolutionsReferenceDataOperations( - self._client, self._config, self._serialize, self._deserialize) - self.external_security_solutions = ExternalSecuritySolutionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.secure_scores = SecureScoresOperations( - self._client, self._config, self._serialize, self._deserialize) - self.secure_score_controls = SecureScoreControlsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.secure_score_control_definitions = SecureScoreControlDefinitionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.security_solutions = SecuritySolutionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.connectors = ConnectorsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sql_vulnerability_assessment_scans = SqlVulnerabilityAssessmentScansOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sql_vulnerability_assessment_scan_results = SqlVulnerabilityAssessmentScanResultsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sql_vulnerability_assessment_baseline_rules = SqlVulnerabilityAssessmentBaselineRulesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.alerts = AlertsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.settings = SettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.ingestion_settings = IngestionSettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.software_inventories = SoftwareInventoriesOperations( - self._client, self._config, self._serialize, self._deserialize) - def _send_request(self, http_request, **kwargs): - # type: (HttpRequest, Any) -> HttpResponse + def _send_request( + self, + request, # type: HttpRequest + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. - :param http_request: The network request you want to make. Required. - :type http_request: ~azure.core.pipeline.transport.HttpRequest - :keyword bool stream: Whether the response payload will be streamed. Defaults to True. + >>> from azure.core.rest import HttpRequest + >>> request = HttpRequest("GET", "https://www.example.org/") + + >>> response = client._send_request(request) + + + For more information on this code flow, see https://aka.ms/azsdk/python/protocol/quickstart + + :param request: The network request you want to make. Required. + :type request: ~azure.core.rest.HttpRequest + :keyword bool stream: Whether the response payload will be streamed. Defaults to False. :return: The response of your network call. Does not do error handling on your response. - :rtype: ~azure.core.pipeline.transport.HttpResponse + :rtype: ~azure.core.rest.HttpResponse """ - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - http_request.url = self._client.format_url(http_request.url, **path_format_arguments) - stream = kwargs.pop("stream", True) - pipeline_response = self._client._pipeline.run(http_request, stream=stream, **kwargs) - return pipeline_response.http_response + + request_copy = deepcopy(request) + request_copy.url = self._client.format_url(request_copy.url) + return self._client.send_request(request_copy, **kwargs) def close(self): # type: () -> None diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/_vendor.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/_vendor.py new file mode 100644 index 0000000000000..138f663c53a4e --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/_vendor.py @@ -0,0 +1,27 @@ +# -------------------------------------------------------------------------- +# 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.core.pipeline.transport import HttpRequest + +def _convert_request(request, files=None): + data = request.content if not files else None + request = HttpRequest(method=request.method, url=request.url, headers=request.headers, data=data) + if files: + request.set_formdata_body(files) + return request + +def _format_url_section(template, **kwargs): + components = template.split("/") + while components: + try: + return template.format(**kwargs) + except KeyError as key: + formatted_components = template.split("/") + components = [ + c for c in formatted_components if "{}".format(key.args[0]) not in c + ] + template = "/".join(components) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/_version.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/_version.py index e32dc6ec42180..e5754a47ce68f 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/_version.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/_version.py @@ -6,4 +6,4 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "2.0.0b1" +VERSION = "1.0.0b1" diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/__init__.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/__init__.py index caee127a80812..857b4e1a9ddb6 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/__init__.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/__init__.py @@ -8,3 +8,8 @@ from ._security_center import SecurityCenter __all__ = ['SecurityCenter'] + +# `._patch.py` is used for handwritten extensions to the generated code +# Example: https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md +from ._patch import patch_sdk +patch_sdk() diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_configuration.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_configuration.py index 133510e3bacef..247eec48abc9f 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_configuration.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_configuration.py @@ -10,7 +10,7 @@ from azure.core.configuration import Configuration from azure.core.pipeline import policies -from azure.mgmt.core.policies import ARMHttpLoggingPolicy +from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy from .._version import VERSION @@ -40,13 +40,13 @@ def __init__( asc_location: str, **kwargs: Any ) -> None: + super(SecurityCenterConfiguration, self).__init__(**kwargs) if credential is None: raise ValueError("Parameter 'credential' must not be None.") if subscription_id is None: raise ValueError("Parameter 'subscription_id' must not be None.") if asc_location is None: raise ValueError("Parameter 'asc_location' must not be None.") - super(SecurityCenterConfiguration, self).__init__(**kwargs) self.credential = credential self.subscription_id = subscription_id @@ -69,4 +69,4 @@ def _configure( self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_patch.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_patch.py new file mode 100644 index 0000000000000..74e48ecd07cf3 --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_patch.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# +# Copyright (c) Microsoft Corporation. All rights reserved. +# +# The MIT License (MIT) +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the ""Software""), to +# deal in the Software without restriction, including without limitation the +# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or +# sell copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING +# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS +# IN THE SOFTWARE. +# +# -------------------------------------------------------------------------- + +# This file is used for handwritten extensions to the generated code. Example: +# https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md +def patch_sdk(): + pass \ No newline at end of file diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_security_center.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_security_center.py index b238b910a8bb0..64de70c2758d1 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_security_center.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/_security_center.py @@ -6,84 +6,56 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Optional, TYPE_CHECKING +from copy import deepcopy +from typing import Any, Awaitable, Optional, TYPE_CHECKING -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.mgmt.core import AsyncARMPipelineClient from msrest import Deserializer, Serializer +from .. import models +from ._configuration import SecurityCenterConfiguration +from .operations import AdaptiveApplicationControlsOperations, AdaptiveNetworkHardeningsOperations, AdvancedThreatProtectionOperations, AlertsOperations, AlertsSuppressionRulesOperations, AllowedConnectionsOperations, AssessmentsMetadataOperations, AssessmentsOperations, AutoProvisioningSettingsOperations, AutomationsOperations, ComplianceResultsOperations, CompliancesOperations, ConnectorsOperations, CustomAssessmentAutomationsOperations, CustomEntityStoreAssignmentsOperations, DeviceSecurityGroupsOperations, DiscoveredSecuritySolutionsOperations, ExternalSecuritySolutionsOperations, InformationProtectionPoliciesOperations, IngestionSettingsOperations, IotSecuritySolutionAnalyticsOperations, IotSecuritySolutionOperations, IotSecuritySolutionsAnalyticsAggregatedAlertOperations, IotSecuritySolutionsAnalyticsRecommendationOperations, JitNetworkAccessPoliciesOperations, LocationsOperations, MdeOnboardingsOperations, Operations, PricingsOperations, RegulatoryComplianceAssessmentsOperations, RegulatoryComplianceControlsOperations, RegulatoryComplianceStandardsOperations, SecureScoreControlDefinitionsOperations, SecureScoreControlsOperations, SecureScoresOperations, SecurityConnectorsOperations, SecurityContactsOperations, SecuritySolutionsOperations, SecuritySolutionsReferenceDataOperations, ServerVulnerabilityAssessmentOperations, SettingsOperations, SoftwareInventoriesOperations, SqlVulnerabilityAssessmentBaselineRulesOperations, SqlVulnerabilityAssessmentScanResultsOperations, SqlVulnerabilityAssessmentScansOperations, SubAssessmentsOperations, TasksOperations, TopologyOperations, WorkspaceSettingsOperations + if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports from azure.core.credentials_async import AsyncTokenCredential -from ._configuration import SecurityCenterConfiguration -from .operations import ComplianceResultsOperations -from .operations import PricingsOperations -from .operations import AdvancedThreatProtectionOperations -from .operations import DeviceSecurityGroupsOperations -from .operations import IotSecuritySolutionOperations -from .operations import IotSecuritySolutionAnalyticsOperations -from .operations import IotSecuritySolutionsAnalyticsAggregatedAlertOperations -from .operations import IotSecuritySolutionsAnalyticsRecommendationOperations -from .operations import LocationsOperations -from .operations import Operations -from .operations import TasksOperations -from .operations import AutoProvisioningSettingsOperations -from .operations import CompliancesOperations -from .operations import InformationProtectionPoliciesOperations -from .operations import SecurityContactsOperations -from .operations import WorkspaceSettingsOperations -from .operations import RegulatoryComplianceStandardsOperations -from .operations import RegulatoryComplianceControlsOperations -from .operations import RegulatoryComplianceAssessmentsOperations -from .operations import SubAssessmentsOperations -from .operations import AutomationsOperations -from .operations import AlertsSuppressionRulesOperations -from .operations import ServerVulnerabilityAssessmentOperations -from .operations import AssessmentsMetadataOperations -from .operations import AssessmentsOperations -from .operations import AdaptiveApplicationControlsOperations -from .operations import AdaptiveNetworkHardeningsOperations -from .operations import AllowedConnectionsOperations -from .operations import TopologyOperations -from .operations import JitNetworkAccessPoliciesOperations -from .operations import DiscoveredSecuritySolutionsOperations -from .operations import SecuritySolutionsReferenceDataOperations -from .operations import ExternalSecuritySolutionsOperations -from .operations import SecureScoresOperations -from .operations import SecureScoreControlsOperations -from .operations import SecureScoreControlDefinitionsOperations -from .operations import SecuritySolutionsOperations -from .operations import ConnectorsOperations -from .operations import SqlVulnerabilityAssessmentScansOperations -from .operations import SqlVulnerabilityAssessmentScanResultsOperations -from .operations import SqlVulnerabilityAssessmentBaselineRulesOperations -from .operations import AlertsOperations -from .operations import SettingsOperations -from .operations import IngestionSettingsOperations -from .operations import SoftwareInventoriesOperations -from .. import models - - -class SecurityCenter(object): +class SecurityCenter: """API spec for Microsoft.Security (Azure Security Center) resource provider. + :ivar mde_onboardings: MdeOnboardingsOperations operations + :vartype mde_onboardings: azure.mgmt.security.aio.operations.MdeOnboardingsOperations + :ivar custom_assessment_automations: CustomAssessmentAutomationsOperations operations + :vartype custom_assessment_automations: + azure.mgmt.security.aio.operations.CustomAssessmentAutomationsOperations + :ivar custom_entity_store_assignments: CustomEntityStoreAssignmentsOperations operations + :vartype custom_entity_store_assignments: + azure.mgmt.security.aio.operations.CustomEntityStoreAssignmentsOperations :ivar compliance_results: ComplianceResultsOperations operations :vartype compliance_results: azure.mgmt.security.aio.operations.ComplianceResultsOperations :ivar pricings: PricingsOperations operations :vartype pricings: azure.mgmt.security.aio.operations.PricingsOperations :ivar advanced_threat_protection: AdvancedThreatProtectionOperations operations - :vartype advanced_threat_protection: azure.mgmt.security.aio.operations.AdvancedThreatProtectionOperations + :vartype advanced_threat_protection: + azure.mgmt.security.aio.operations.AdvancedThreatProtectionOperations :ivar device_security_groups: DeviceSecurityGroupsOperations operations - :vartype device_security_groups: azure.mgmt.security.aio.operations.DeviceSecurityGroupsOperations + :vartype device_security_groups: + azure.mgmt.security.aio.operations.DeviceSecurityGroupsOperations :ivar iot_security_solution: IotSecuritySolutionOperations operations - :vartype iot_security_solution: azure.mgmt.security.aio.operations.IotSecuritySolutionOperations + :vartype iot_security_solution: + azure.mgmt.security.aio.operations.IotSecuritySolutionOperations :ivar iot_security_solution_analytics: IotSecuritySolutionAnalyticsOperations operations - :vartype iot_security_solution_analytics: azure.mgmt.security.aio.operations.IotSecuritySolutionAnalyticsOperations - :ivar iot_security_solutions_analytics_aggregated_alert: IotSecuritySolutionsAnalyticsAggregatedAlertOperations operations - :vartype iot_security_solutions_analytics_aggregated_alert: azure.mgmt.security.aio.operations.IotSecuritySolutionsAnalyticsAggregatedAlertOperations - :ivar iot_security_solutions_analytics_recommendation: IotSecuritySolutionsAnalyticsRecommendationOperations operations - :vartype iot_security_solutions_analytics_recommendation: azure.mgmt.security.aio.operations.IotSecuritySolutionsAnalyticsRecommendationOperations + :vartype iot_security_solution_analytics: + azure.mgmt.security.aio.operations.IotSecuritySolutionAnalyticsOperations + :ivar iot_security_solutions_analytics_aggregated_alert: + IotSecuritySolutionsAnalyticsAggregatedAlertOperations operations + :vartype iot_security_solutions_analytics_aggregated_alert: + azure.mgmt.security.aio.operations.IotSecuritySolutionsAnalyticsAggregatedAlertOperations + :ivar iot_security_solutions_analytics_recommendation: + IotSecuritySolutionsAnalyticsRecommendationOperations operations + :vartype iot_security_solutions_analytics_recommendation: + azure.mgmt.security.aio.operations.IotSecuritySolutionsAnalyticsRecommendationOperations :ivar locations: LocationsOperations operations :vartype locations: azure.mgmt.security.aio.operations.LocationsOperations :ivar operations: Operations operations @@ -91,65 +63,85 @@ class SecurityCenter(object): :ivar tasks: TasksOperations operations :vartype tasks: azure.mgmt.security.aio.operations.TasksOperations :ivar auto_provisioning_settings: AutoProvisioningSettingsOperations operations - :vartype auto_provisioning_settings: azure.mgmt.security.aio.operations.AutoProvisioningSettingsOperations + :vartype auto_provisioning_settings: + azure.mgmt.security.aio.operations.AutoProvisioningSettingsOperations :ivar compliances: CompliancesOperations operations :vartype compliances: azure.mgmt.security.aio.operations.CompliancesOperations :ivar information_protection_policies: InformationProtectionPoliciesOperations operations - :vartype information_protection_policies: azure.mgmt.security.aio.operations.InformationProtectionPoliciesOperations + :vartype information_protection_policies: + azure.mgmt.security.aio.operations.InformationProtectionPoliciesOperations :ivar security_contacts: SecurityContactsOperations operations :vartype security_contacts: azure.mgmt.security.aio.operations.SecurityContactsOperations :ivar workspace_settings: WorkspaceSettingsOperations operations :vartype workspace_settings: azure.mgmt.security.aio.operations.WorkspaceSettingsOperations :ivar regulatory_compliance_standards: RegulatoryComplianceStandardsOperations operations - :vartype regulatory_compliance_standards: azure.mgmt.security.aio.operations.RegulatoryComplianceStandardsOperations + :vartype regulatory_compliance_standards: + azure.mgmt.security.aio.operations.RegulatoryComplianceStandardsOperations :ivar regulatory_compliance_controls: RegulatoryComplianceControlsOperations operations - :vartype regulatory_compliance_controls: azure.mgmt.security.aio.operations.RegulatoryComplianceControlsOperations + :vartype regulatory_compliance_controls: + azure.mgmt.security.aio.operations.RegulatoryComplianceControlsOperations :ivar regulatory_compliance_assessments: RegulatoryComplianceAssessmentsOperations operations - :vartype regulatory_compliance_assessments: azure.mgmt.security.aio.operations.RegulatoryComplianceAssessmentsOperations + :vartype regulatory_compliance_assessments: + azure.mgmt.security.aio.operations.RegulatoryComplianceAssessmentsOperations :ivar sub_assessments: SubAssessmentsOperations operations :vartype sub_assessments: azure.mgmt.security.aio.operations.SubAssessmentsOperations :ivar automations: AutomationsOperations operations :vartype automations: azure.mgmt.security.aio.operations.AutomationsOperations :ivar alerts_suppression_rules: AlertsSuppressionRulesOperations operations - :vartype alerts_suppression_rules: azure.mgmt.security.aio.operations.AlertsSuppressionRulesOperations + :vartype alerts_suppression_rules: + azure.mgmt.security.aio.operations.AlertsSuppressionRulesOperations :ivar server_vulnerability_assessment: ServerVulnerabilityAssessmentOperations operations - :vartype server_vulnerability_assessment: azure.mgmt.security.aio.operations.ServerVulnerabilityAssessmentOperations + :vartype server_vulnerability_assessment: + azure.mgmt.security.aio.operations.ServerVulnerabilityAssessmentOperations :ivar assessments_metadata: AssessmentsMetadataOperations operations :vartype assessments_metadata: azure.mgmt.security.aio.operations.AssessmentsMetadataOperations :ivar assessments: AssessmentsOperations operations :vartype assessments: azure.mgmt.security.aio.operations.AssessmentsOperations :ivar adaptive_application_controls: AdaptiveApplicationControlsOperations operations - :vartype adaptive_application_controls: azure.mgmt.security.aio.operations.AdaptiveApplicationControlsOperations + :vartype adaptive_application_controls: + azure.mgmt.security.aio.operations.AdaptiveApplicationControlsOperations :ivar adaptive_network_hardenings: AdaptiveNetworkHardeningsOperations operations - :vartype adaptive_network_hardenings: azure.mgmt.security.aio.operations.AdaptiveNetworkHardeningsOperations + :vartype adaptive_network_hardenings: + azure.mgmt.security.aio.operations.AdaptiveNetworkHardeningsOperations :ivar allowed_connections: AllowedConnectionsOperations operations :vartype allowed_connections: azure.mgmt.security.aio.operations.AllowedConnectionsOperations :ivar topology: TopologyOperations operations :vartype topology: azure.mgmt.security.aio.operations.TopologyOperations :ivar jit_network_access_policies: JitNetworkAccessPoliciesOperations operations - :vartype jit_network_access_policies: azure.mgmt.security.aio.operations.JitNetworkAccessPoliciesOperations + :vartype jit_network_access_policies: + azure.mgmt.security.aio.operations.JitNetworkAccessPoliciesOperations :ivar discovered_security_solutions: DiscoveredSecuritySolutionsOperations operations - :vartype discovered_security_solutions: azure.mgmt.security.aio.operations.DiscoveredSecuritySolutionsOperations + :vartype discovered_security_solutions: + azure.mgmt.security.aio.operations.DiscoveredSecuritySolutionsOperations :ivar security_solutions_reference_data: SecuritySolutionsReferenceDataOperations operations - :vartype security_solutions_reference_data: azure.mgmt.security.aio.operations.SecuritySolutionsReferenceDataOperations + :vartype security_solutions_reference_data: + azure.mgmt.security.aio.operations.SecuritySolutionsReferenceDataOperations :ivar external_security_solutions: ExternalSecuritySolutionsOperations operations - :vartype external_security_solutions: azure.mgmt.security.aio.operations.ExternalSecuritySolutionsOperations + :vartype external_security_solutions: + azure.mgmt.security.aio.operations.ExternalSecuritySolutionsOperations :ivar secure_scores: SecureScoresOperations operations :vartype secure_scores: azure.mgmt.security.aio.operations.SecureScoresOperations :ivar secure_score_controls: SecureScoreControlsOperations operations - :vartype secure_score_controls: azure.mgmt.security.aio.operations.SecureScoreControlsOperations + :vartype secure_score_controls: + azure.mgmt.security.aio.operations.SecureScoreControlsOperations :ivar secure_score_control_definitions: SecureScoreControlDefinitionsOperations operations - :vartype secure_score_control_definitions: azure.mgmt.security.aio.operations.SecureScoreControlDefinitionsOperations + :vartype secure_score_control_definitions: + azure.mgmt.security.aio.operations.SecureScoreControlDefinitionsOperations :ivar security_solutions: SecuritySolutionsOperations operations :vartype security_solutions: azure.mgmt.security.aio.operations.SecuritySolutionsOperations :ivar connectors: ConnectorsOperations operations :vartype connectors: azure.mgmt.security.aio.operations.ConnectorsOperations :ivar sql_vulnerability_assessment_scans: SqlVulnerabilityAssessmentScansOperations operations - :vartype sql_vulnerability_assessment_scans: azure.mgmt.security.aio.operations.SqlVulnerabilityAssessmentScansOperations - :ivar sql_vulnerability_assessment_scan_results: SqlVulnerabilityAssessmentScanResultsOperations operations - :vartype sql_vulnerability_assessment_scan_results: azure.mgmt.security.aio.operations.SqlVulnerabilityAssessmentScanResultsOperations - :ivar sql_vulnerability_assessment_baseline_rules: SqlVulnerabilityAssessmentBaselineRulesOperations operations - :vartype sql_vulnerability_assessment_baseline_rules: azure.mgmt.security.aio.operations.SqlVulnerabilityAssessmentBaselineRulesOperations + :vartype sql_vulnerability_assessment_scans: + azure.mgmt.security.aio.operations.SqlVulnerabilityAssessmentScansOperations + :ivar sql_vulnerability_assessment_scan_results: + SqlVulnerabilityAssessmentScanResultsOperations operations + :vartype sql_vulnerability_assessment_scan_results: + azure.mgmt.security.aio.operations.SqlVulnerabilityAssessmentScanResultsOperations + :ivar sql_vulnerability_assessment_baseline_rules: + SqlVulnerabilityAssessmentBaselineRulesOperations operations + :vartype sql_vulnerability_assessment_baseline_rules: + azure.mgmt.security.aio.operations.SqlVulnerabilityAssessmentBaselineRulesOperations :ivar alerts: AlertsOperations operations :vartype alerts: azure.mgmt.security.aio.operations.AlertsOperations :ivar settings: SettingsOperations operations @@ -158,14 +150,19 @@ class SecurityCenter(object): :vartype ingestion_settings: azure.mgmt.security.aio.operations.IngestionSettingsOperations :ivar software_inventories: SoftwareInventoriesOperations operations :vartype software_inventories: azure.mgmt.security.aio.operations.SoftwareInventoriesOperations + :ivar security_connectors: SecurityConnectorsOperations operations + :vartype security_connectors: azure.mgmt.security.aio.operations.SecurityConnectorsOperations :param credential: Credential needed for the client to connect to Azure. :type credential: ~azure.core.credentials_async.AsyncTokenCredential :param subscription_id: Azure subscription ID. :type subscription_id: str - :param asc_location: The location where ASC stores the data of the subscription. can be retrieved from Get locations. + :param asc_location: The location where ASC stores the data of the subscription. can be + retrieved from Get locations. :type asc_location: str - :param str base_url: Service URL - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :param base_url: Service URL. Default value is 'https://management.azure.com'. + :type base_url: str + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. """ def __init__( @@ -173,127 +170,92 @@ def __init__( credential: "AsyncTokenCredential", subscription_id: str, asc_location: str, - base_url: Optional[str] = None, + base_url: str = "https://management.azure.com", **kwargs: Any ) -> None: - if not base_url: - base_url = 'https://management.azure.com' - self._config = SecurityCenterConfiguration(credential, subscription_id, asc_location, **kwargs) + self._config = SecurityCenterConfiguration(credential=credential, subscription_id=subscription_id, asc_location=asc_location, **kwargs) self._client = 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) - self._serialize.client_side_validation = False self._deserialize = Deserializer(client_models) + self._serialize.client_side_validation = False + self.mde_onboardings = MdeOnboardingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.custom_assessment_automations = CustomAssessmentAutomationsOperations(self._client, self._config, self._serialize, self._deserialize) + self.custom_entity_store_assignments = CustomEntityStoreAssignmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.compliance_results = ComplianceResultsOperations(self._client, self._config, self._serialize, self._deserialize) + self.pricings = PricingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.advanced_threat_protection = AdvancedThreatProtectionOperations(self._client, self._config, self._serialize, self._deserialize) + self.device_security_groups = DeviceSecurityGroupsOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solution = IotSecuritySolutionOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solution_analytics = IotSecuritySolutionAnalyticsOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solutions_analytics_aggregated_alert = IotSecuritySolutionsAnalyticsAggregatedAlertOperations(self._client, self._config, self._serialize, self._deserialize) + self.iot_security_solutions_analytics_recommendation = IotSecuritySolutionsAnalyticsRecommendationOperations(self._client, self._config, self._serialize, self._deserialize) + self.locations = LocationsOperations(self._client, self._config, self._serialize, self._deserialize) + self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) + self.tasks = TasksOperations(self._client, self._config, self._serialize, self._deserialize) + self.auto_provisioning_settings = AutoProvisioningSettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.compliances = CompliancesOperations(self._client, self._config, self._serialize, self._deserialize) + self.information_protection_policies = InformationProtectionPoliciesOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_contacts = SecurityContactsOperations(self._client, self._config, self._serialize, self._deserialize) + self.workspace_settings = WorkspaceSettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.regulatory_compliance_standards = RegulatoryComplianceStandardsOperations(self._client, self._config, self._serialize, self._deserialize) + self.regulatory_compliance_controls = RegulatoryComplianceControlsOperations(self._client, self._config, self._serialize, self._deserialize) + self.regulatory_compliance_assessments = RegulatoryComplianceAssessmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.sub_assessments = SubAssessmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.automations = AutomationsOperations(self._client, self._config, self._serialize, self._deserialize) + self.alerts_suppression_rules = AlertsSuppressionRulesOperations(self._client, self._config, self._serialize, self._deserialize) + self.server_vulnerability_assessment = ServerVulnerabilityAssessmentOperations(self._client, self._config, self._serialize, self._deserialize) + self.assessments_metadata = AssessmentsMetadataOperations(self._client, self._config, self._serialize, self._deserialize) + self.assessments = AssessmentsOperations(self._client, self._config, self._serialize, self._deserialize) + self.adaptive_application_controls = AdaptiveApplicationControlsOperations(self._client, self._config, self._serialize, self._deserialize) + self.adaptive_network_hardenings = AdaptiveNetworkHardeningsOperations(self._client, self._config, self._serialize, self._deserialize) + self.allowed_connections = AllowedConnectionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.topology = TopologyOperations(self._client, self._config, self._serialize, self._deserialize) + self.jit_network_access_policies = JitNetworkAccessPoliciesOperations(self._client, self._config, self._serialize, self._deserialize) + self.discovered_security_solutions = DiscoveredSecuritySolutionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_solutions_reference_data = SecuritySolutionsReferenceDataOperations(self._client, self._config, self._serialize, self._deserialize) + self.external_security_solutions = ExternalSecuritySolutionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.secure_scores = SecureScoresOperations(self._client, self._config, self._serialize, self._deserialize) + self.secure_score_controls = SecureScoreControlsOperations(self._client, self._config, self._serialize, self._deserialize) + self.secure_score_control_definitions = SecureScoreControlDefinitionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_solutions = SecuritySolutionsOperations(self._client, self._config, self._serialize, self._deserialize) + self.connectors = ConnectorsOperations(self._client, self._config, self._serialize, self._deserialize) + self.sql_vulnerability_assessment_scans = SqlVulnerabilityAssessmentScansOperations(self._client, self._config, self._serialize, self._deserialize) + self.sql_vulnerability_assessment_scan_results = SqlVulnerabilityAssessmentScanResultsOperations(self._client, self._config, self._serialize, self._deserialize) + self.sql_vulnerability_assessment_baseline_rules = SqlVulnerabilityAssessmentBaselineRulesOperations(self._client, self._config, self._serialize, self._deserialize) + self.alerts = AlertsOperations(self._client, self._config, self._serialize, self._deserialize) + self.settings = SettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.ingestion_settings = IngestionSettingsOperations(self._client, self._config, self._serialize, self._deserialize) + self.software_inventories = SoftwareInventoriesOperations(self._client, self._config, self._serialize, self._deserialize) + self.security_connectors = SecurityConnectorsOperations(self._client, self._config, self._serialize, self._deserialize) - self.compliance_results = ComplianceResultsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.pricings = PricingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.advanced_threat_protection = AdvancedThreatProtectionOperations( - self._client, self._config, self._serialize, self._deserialize) - self.device_security_groups = DeviceSecurityGroupsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solution = IotSecuritySolutionOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solution_analytics = IotSecuritySolutionAnalyticsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solutions_analytics_aggregated_alert = IotSecuritySolutionsAnalyticsAggregatedAlertOperations( - self._client, self._config, self._serialize, self._deserialize) - self.iot_security_solutions_analytics_recommendation = IotSecuritySolutionsAnalyticsRecommendationOperations( - self._client, self._config, self._serialize, self._deserialize) - self.locations = LocationsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.operations = Operations( - self._client, self._config, self._serialize, self._deserialize) - self.tasks = TasksOperations( - self._client, self._config, self._serialize, self._deserialize) - self.auto_provisioning_settings = AutoProvisioningSettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.compliances = CompliancesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.information_protection_policies = InformationProtectionPoliciesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.security_contacts = SecurityContactsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.workspace_settings = WorkspaceSettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.regulatory_compliance_standards = RegulatoryComplianceStandardsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.regulatory_compliance_controls = RegulatoryComplianceControlsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.regulatory_compliance_assessments = RegulatoryComplianceAssessmentsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sub_assessments = SubAssessmentsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.automations = AutomationsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.alerts_suppression_rules = AlertsSuppressionRulesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.server_vulnerability_assessment = ServerVulnerabilityAssessmentOperations( - self._client, self._config, self._serialize, self._deserialize) - self.assessments_metadata = AssessmentsMetadataOperations( - self._client, self._config, self._serialize, self._deserialize) - self.assessments = AssessmentsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.adaptive_application_controls = AdaptiveApplicationControlsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.adaptive_network_hardenings = AdaptiveNetworkHardeningsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.allowed_connections = AllowedConnectionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.topology = TopologyOperations( - self._client, self._config, self._serialize, self._deserialize) - self.jit_network_access_policies = JitNetworkAccessPoliciesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.discovered_security_solutions = DiscoveredSecuritySolutionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.security_solutions_reference_data = SecuritySolutionsReferenceDataOperations( - self._client, self._config, self._serialize, self._deserialize) - self.external_security_solutions = ExternalSecuritySolutionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.secure_scores = SecureScoresOperations( - self._client, self._config, self._serialize, self._deserialize) - self.secure_score_controls = SecureScoreControlsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.secure_score_control_definitions = SecureScoreControlDefinitionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.security_solutions = SecuritySolutionsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.connectors = ConnectorsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sql_vulnerability_assessment_scans = SqlVulnerabilityAssessmentScansOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sql_vulnerability_assessment_scan_results = SqlVulnerabilityAssessmentScanResultsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.sql_vulnerability_assessment_baseline_rules = SqlVulnerabilityAssessmentBaselineRulesOperations( - self._client, self._config, self._serialize, self._deserialize) - self.alerts = AlertsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.settings = SettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.ingestion_settings = IngestionSettingsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.software_inventories = SoftwareInventoriesOperations( - self._client, self._config, self._serialize, self._deserialize) - async def _send_request(self, http_request: HttpRequest, **kwargs: Any) -> AsyncHttpResponse: + def _send_request( + self, + request: HttpRequest, + **kwargs: Any + ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. - :param http_request: The network request you want to make. Required. - :type http_request: ~azure.core.pipeline.transport.HttpRequest - :keyword bool stream: Whether the response payload will be streamed. Defaults to True. + >>> from azure.core.rest import HttpRequest + >>> request = HttpRequest("GET", "https://www.example.org/") + + >>> response = await client._send_request(request) + + + For more information on this code flow, see https://aka.ms/azsdk/python/protocol/quickstart + + :param request: The network request you want to make. Required. + :type request: ~azure.core.rest.HttpRequest + :keyword bool stream: Whether the response payload will be streamed. Defaults to False. :return: The response of your network call. Does not do error handling on your response. - :rtype: ~azure.core.pipeline.transport.AsyncHttpResponse + :rtype: ~azure.core.rest.AsyncHttpResponse """ - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - http_request.url = self._client.format_url(http_request.url, **path_format_arguments) - stream = kwargs.pop("stream", True) - pipeline_response = await self._client._pipeline.run(http_request, stream=stream, **kwargs) - return pipeline_response.http_response + + request_copy = deepcopy(request) + request_copy.url = self._client.format_url(request_copy.url) + return self._client.send_request(request_copy, **kwargs) async def close(self) -> None: await self._client.close() diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/__init__.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/__init__.py index 65df4fa5ef359..e0a22181ca006 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/__init__.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/__init__.py @@ -6,6 +6,9 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from ._mde_onboardings_operations import MdeOnboardingsOperations +from ._custom_assessment_automations_operations import CustomAssessmentAutomationsOperations +from ._custom_entity_store_assignments_operations import CustomEntityStoreAssignmentsOperations from ._compliance_results_operations import ComplianceResultsOperations from ._pricings_operations import PricingsOperations from ._advanced_threat_protection_operations import AdvancedThreatProtectionOperations @@ -51,8 +54,12 @@ from ._settings_operations import SettingsOperations from ._ingestion_settings_operations import IngestionSettingsOperations from ._software_inventories_operations import SoftwareInventoriesOperations +from ._security_connectors_operations import SecurityConnectorsOperations __all__ = [ + 'MdeOnboardingsOperations', + 'CustomAssessmentAutomationsOperations', + 'CustomEntityStoreAssignmentsOperations', 'ComplianceResultsOperations', 'PricingsOperations', 'AdvancedThreatProtectionOperations', @@ -98,4 +105,5 @@ 'SettingsOperations', 'IngestionSettingsOperations', 'SoftwareInventoriesOperations', + 'SecurityConnectorsOperations', ] diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_application_controls_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_application_controls_operations.py index 14db3cdfd4143..66e94363ef964 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_application_controls_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_application_controls_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._adaptive_application_controls_operations import build_delete_request, build_get_request, build_list_request, build_put_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def list( self, include_path_recommendations: Optional[bool] = None, @@ -62,29 +67,17 @@ async def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if include_path_recommendations is not None: - query_parameters['includePathRecommendations'] = self._serialize.query("include_path_recommendations", include_path_recommendations, 'bool') - if summary is not None: - query_parameters['summary'] = self._serialize.query("summary", summary, 'bool') + + request = build_list_request( + subscription_id=self._config.subscription_id, + include_path_recommendations=include_path_recommendations, + summary=summary, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -98,8 +91,11 @@ async def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/applicationWhitelistings'} # type: ignore + + @distributed_trace_async async def get( self, group_name: str, @@ -119,27 +115,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'groupName': self._serialize.url("group_name", group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + group_name=group_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -153,8 +139,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}'} # type: ignore + + @distributed_trace_async async def put( self, group_name: str, @@ -177,32 +166,22 @@ async def put( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.put.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'groupName': self._serialize.url("group_name", group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'AdaptiveApplicationControlGroup') + + request = build_put_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + group_name=group_name, + content_type=content_type, + json=_json, + template_url=self.put.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'AdaptiveApplicationControlGroup') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -216,8 +195,11 @@ async def put( return cls(pipeline_response, deserialized, {}) return deserialized + put.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}'} # type: ignore + + @distributed_trace_async async def delete( self, group_name: str, @@ -237,27 +219,17 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'groupName': self._serialize.url("group_name", group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + group_name=group_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -269,3 +241,4 @@ async def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_network_hardenings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_network_hardenings_operations.py index 3a66706140ae9..013e0cba6fada 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_network_hardenings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_adaptive_network_hardenings_operations.py @@ -5,19 +5,24 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._adaptive_network_hardenings_operations import build_enforce_request_initial, build_get_request, build_list_by_extended_resource_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -43,6 +48,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list_by_extended_resource( self, resource_group_name: str, @@ -63,8 +69,10 @@ def list_by_extended_resource( :param resource_name: Name of the resource. :type resource_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AdaptiveNetworkHardeningsList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AdaptiveNetworkHardeningsList] + :return: An iterator like instance of either AdaptiveNetworkHardeningsList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AdaptiveNetworkHardeningsList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.AdaptiveNetworkHardeningsList"] @@ -72,38 +80,37 @@ def list_by_extended_resource( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_extended_resource.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.list_by_extended_resource.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AdaptiveNetworkHardeningsList', pipeline_response) + deserialized = self._deserialize("AdaptiveNetworkHardeningsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -121,11 +128,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_extended_resource.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -159,30 +168,20 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'adaptiveNetworkHardeningResourceName': self._serialize.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + adaptive_network_hardening_resource_name=adaptive_network_hardening_resource_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -196,8 +195,10 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}'} # type: ignore + async def _enforce_initial( self, resource_group_name: str, @@ -213,37 +214,25 @@ async def _enforce_initial( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - adaptive_network_hardening_enforce_action = "enforce" - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._enforce_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'adaptiveNetworkHardeningResourceName': self._serialize.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), - 'adaptiveNetworkHardeningEnforceAction': self._serialize.url("adaptive_network_hardening_enforce_action", adaptive_network_hardening_enforce_action, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'AdaptiveNetworkHardeningEnforceRequest') + + request = build_enforce_request_initial( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + adaptive_network_hardening_resource_name=adaptive_network_hardening_resource_name, + content_type=content_type, + json=_json, + template_url=self._enforce_initial.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'AdaptiveNetworkHardeningEnforceRequest') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -256,6 +245,8 @@ async def _enforce_initial( _enforce_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}/{adaptiveNetworkHardeningEnforceAction}'} # type: ignore + + @distributed_trace_async async def begin_enforce( self, resource_group_name: str, @@ -284,15 +275,18 @@ async def begin_enforce( :type body: ~azure.mgmt.security.models.AdaptiveNetworkHardeningEnforceRequest :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. - Pass in False for this operation to not poll, or pass in your own initialized polling object for a personal polling strategy. + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: + :raises: ~azure.core.exceptions.HttpResponseError """ - polling = kwargs.pop('polling', True) # type: Union[bool, AsyncPollingMethod] + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + polling = kwargs.pop('polling', True) # type: Union[bool, azure.core.polling.AsyncPollingMethod] cls = kwargs.pop('cls', None) # type: ClsType[None] lro_delay = kwargs.pop( 'polling_interval', @@ -307,29 +301,18 @@ async def begin_enforce( resource_name=resource_name, adaptive_network_hardening_resource_name=adaptive_network_hardening_resource_name, body=body, + content_type=content_type, cls=lambda x,y,z: x, **kwargs ) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) def get_long_running_output(pipeline_response): if cls: return cls(pipeline_response, None, {}) - adaptive_network_hardening_enforce_action = "enforce" - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'adaptiveNetworkHardeningResourceName': self._serialize.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), - 'adaptiveNetworkHardeningEnforceAction': self._serialize.url("adaptive_network_hardening_enforce_action", adaptive_network_hardening_enforce_action, 'str'), - } - if polling is True: polling_method = AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs) elif polling is False: polling_method = AsyncNoPolling() else: polling_method = polling if cont_token: @@ -341,4 +324,5 @@ def get_long_running_output(pipeline_response): ) else: return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) + begin_enforce.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}/{adaptiveNetworkHardeningEnforceAction}'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_advanced_threat_protection_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_advanced_threat_protection_operations.py index 9e1c93fc00a1b..f1ff12986a28f 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_advanced_threat_protection_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_advanced_threat_protection_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._advanced_threat_protection_operations import build_create_request, build_get_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def get( self, resource_id: str, @@ -59,27 +64,15 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01" - setting_name = "current" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + resource_id=resource_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -93,8 +86,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/advancedThreatProtectionSettings/{settingName}'} # type: ignore + + @distributed_trace_async async def create( self, resource_id: str, @@ -106,7 +102,8 @@ async def create( :param resource_id: The identifier of the resource. :type resource_id: str :param advanced_threat_protection_setting: Advanced Threat Protection Settings. - :type advanced_threat_protection_setting: ~azure.mgmt.security.models.AdvancedThreatProtectionSetting + :type advanced_threat_protection_setting: + ~azure.mgmt.security.models.AdvancedThreatProtectionSetting :keyword callable cls: A custom type or function that will be passed the direct response :return: AdvancedThreatProtectionSetting, or the result of cls(response) :rtype: ~azure.mgmt.security.models.AdvancedThreatProtectionSetting @@ -117,32 +114,20 @@ async def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01" - setting_name = "current" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(advanced_threat_protection_setting, 'AdvancedThreatProtectionSetting') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_request( + resource_id=resource_id, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(advanced_threat_protection_setting, 'AdvancedThreatProtectionSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -156,4 +141,6 @@ async def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/advancedThreatProtectionSettings/{settingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_operations.py index 4374722f99e79..2af27fd406310 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_operations.py @@ -5,19 +5,24 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._alerts_operations import build_get_resource_group_level_request, build_get_subscription_level_request, build_list_by_resource_group_request, build_list_request, build_list_resource_group_level_by_region_request, build_list_subscription_level_by_region_request, build_simulate_request_initial, build_update_resource_group_level_state_to_activate_request, build_update_resource_group_level_state_to_dismiss_request, build_update_resource_group_level_state_to_resolve_request, build_update_subscription_level_state_to_activate_request, build_update_subscription_level_state_to_dismiss_request, build_update_subscription_level_state_to_resolve_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -43,6 +48,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -59,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -104,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alerts'} # type: ignore + @distributed_trace def list_by_resource_group( self, resource_group_name: str, @@ -129,35 +132,31 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -175,11 +174,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/alerts'} # type: ignore + @distributed_trace def list_subscription_level_by_region( self, **kwargs: Any @@ -197,35 +198,31 @@ def list_subscription_level_by_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_subscription_level_by_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_subscription_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_subscription_level_by_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_subscription_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -243,11 +240,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_subscription_level_by_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts'} # type: ignore + @distributed_trace def list_resource_group_level_by_region( self, resource_group_name: str, @@ -269,36 +268,33 @@ def list_resource_group_level_by_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_resource_group_level_by_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_resource_group_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + resource_group_name=resource_group_name, + template_url=self.list_resource_group_level_by_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_resource_group_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -316,11 +312,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_resource_group_level_by_region.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts'} # type: ignore + @distributed_trace_async async def get_subscription_level( self, alert_name: str, @@ -340,27 +338,17 @@ async def get_subscription_level( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.get_subscription_level.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_subscription_level_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.get_subscription_level.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -374,8 +362,11 @@ async def get_subscription_level( return cls(pipeline_response, deserialized, {}) return deserialized + get_subscription_level.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}'} # type: ignore + + @distributed_trace_async async def get_resource_group_level( self, alert_name: str, @@ -399,28 +390,18 @@ async def get_resource_group_level( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.get_resource_group_level.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_resource_group_level_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.get_resource_group_level.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -434,8 +415,11 @@ async def get_resource_group_level( return cls(pipeline_response, deserialized, {}) return deserialized + get_resource_group_level.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}'} # type: ignore + + @distributed_trace_async async def update_subscription_level_state_to_dismiss( self, alert_name: str, @@ -455,27 +439,17 @@ async def update_subscription_level_state_to_dismiss( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_state_to_dismiss.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_state_to_dismiss_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.update_subscription_level_state_to_dismiss.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -488,6 +462,8 @@ async def update_subscription_level_state_to_dismiss( update_subscription_level_state_to_dismiss.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/dismiss'} # type: ignore + + @distributed_trace_async async def update_subscription_level_state_to_resolve( self, alert_name: str, @@ -507,27 +483,17 @@ async def update_subscription_level_state_to_resolve( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_state_to_resolve.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_state_to_resolve_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.update_subscription_level_state_to_resolve.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -540,6 +506,8 @@ async def update_subscription_level_state_to_resolve( update_subscription_level_state_to_resolve.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/resolve'} # type: ignore + + @distributed_trace_async async def update_subscription_level_state_to_activate( self, alert_name: str, @@ -559,27 +527,17 @@ async def update_subscription_level_state_to_activate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_state_to_activate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_state_to_activate_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.update_subscription_level_state_to_activate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -592,6 +550,8 @@ async def update_subscription_level_state_to_activate( update_subscription_level_state_to_activate.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/activate'} # type: ignore + + @distributed_trace_async async def update_resource_group_level_state_to_resolve( self, alert_name: str, @@ -615,28 +575,18 @@ async def update_resource_group_level_state_to_resolve( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_state_to_resolve.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_state_to_resolve_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.update_resource_group_level_state_to_resolve.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -649,6 +599,8 @@ async def update_resource_group_level_state_to_resolve( update_resource_group_level_state_to_resolve.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/resolve'} # type: ignore + + @distributed_trace_async async def update_resource_group_level_state_to_dismiss( self, alert_name: str, @@ -672,28 +624,18 @@ async def update_resource_group_level_state_to_dismiss( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_state_to_dismiss.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_state_to_dismiss_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.update_resource_group_level_state_to_dismiss.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -706,6 +648,8 @@ async def update_resource_group_level_state_to_dismiss( update_resource_group_level_state_to_dismiss.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/dismiss'} # type: ignore + + @distributed_trace_async async def update_resource_group_level_state_to_activate( self, alert_name: str, @@ -729,28 +673,18 @@ async def update_resource_group_level_state_to_activate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_state_to_activate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_state_to_activate_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.update_resource_group_level_state_to_activate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -763,6 +697,7 @@ async def update_resource_group_level_state_to_activate( update_resource_group_level_state_to_activate.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/activate'} # type: ignore + async def _simulate_initial( self, alert_simulator_request_body: "_models.AlertSimulatorRequestBody", @@ -773,31 +708,21 @@ async def _simulate_initial( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._simulate_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(alert_simulator_request_body, 'AlertSimulatorRequestBody') + + request = build_simulate_request_initial( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + content_type=content_type, + json=_json, + template_url=self._simulate_initial.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(alert_simulator_request_body, 'AlertSimulatorRequestBody') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -810,6 +735,8 @@ async def _simulate_initial( _simulate_initial.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/default/simulate'} # type: ignore + + @distributed_trace_async async def begin_simulate( self, alert_simulator_request_body: "_models.AlertSimulatorRequestBody", @@ -821,15 +748,18 @@ async def begin_simulate( :type alert_simulator_request_body: ~azure.mgmt.security.models.AlertSimulatorRequestBody :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. - Pass in False for this operation to not poll, or pass in your own initialized polling object for a personal polling strategy. + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: + :raises: ~azure.core.exceptions.HttpResponseError """ - polling = kwargs.pop('polling', True) # type: Union[bool, AsyncPollingMethod] + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + polling = kwargs.pop('polling', True) # type: Union[bool, azure.core.polling.AsyncPollingMethod] cls = kwargs.pop('cls', None) # type: ClsType[None] lro_delay = kwargs.pop( 'polling_interval', @@ -839,23 +769,18 @@ async def begin_simulate( if cont_token is None: raw_result = await self._simulate_initial( alert_simulator_request_body=alert_simulator_request_body, + content_type=content_type, cls=lambda x,y,z: x, **kwargs ) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) def get_long_running_output(pipeline_response): if cls: return cls(pipeline_response, None, {}) - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - if polling is True: polling_method = AsyncARMPolling(lro_delay, lro_options={'final-state-via': 'original-uri'}, path_format_arguments=path_format_arguments, **kwargs) + if polling is True: polling_method = AsyncARMPolling(lro_delay, lro_options={'final-state-via': 'original-uri'}, **kwargs) elif polling is False: polling_method = AsyncNoPolling() else: polling_method = polling if cont_token: @@ -867,4 +792,5 @@ def get_long_running_output(pipeline_response): ) else: return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) + begin_simulate.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/default/simulate'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_suppression_rules_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_suppression_rules_operations.py index 24ad78f67198a..11c6ba1f20501 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_suppression_rules_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_alerts_suppression_rules_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._alerts_suppression_rules_operations import build_delete_request, build_get_request, build_list_request, build_update_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, alert_type: Optional[str] = None, @@ -51,8 +57,10 @@ def list( :param alert_type: Type of the alert to get rules for. :type alert_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AlertsSuppressionRulesList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AlertsSuppressionRulesList] + :return: An iterator like instance of either AlertsSuppressionRulesList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AlertsSuppressionRulesList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.AlertsSuppressionRulesList"] @@ -60,36 +68,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if alert_type is not None: - query_parameters['AlertType'] = self._serialize.query("alert_type", alert_type, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + alert_type=alert_type, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + alert_type=alert_type, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AlertsSuppressionRulesList', pipeline_response) + deserialized = self._deserialize("AlertsSuppressionRulesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,11 +110,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules'} # type: ignore + @distributed_trace_async async def get( self, alerts_suppression_rule_name: str, @@ -131,26 +136,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'alertsSuppressionRuleName': self._serialize.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + alerts_suppression_rule_name=alerts_suppression_rule_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -164,8 +159,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}'} # type: ignore + + @distributed_trace_async async def update( self, alerts_suppression_rule_name: str, @@ -188,31 +186,21 @@ async def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'alertsSuppressionRuleName': self._serialize.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(alerts_suppression_rule, 'AlertsSuppressionRule') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_update_request( + subscription_id=self._config.subscription_id, + alerts_suppression_rule_name=alerts_suppression_rule_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(alerts_suppression_rule, 'AlertsSuppressionRule') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -226,8 +214,11 @@ async def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}'} # type: ignore + + @distributed_trace_async async def delete( self, alerts_suppression_rule_name: str, @@ -247,26 +238,16 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'alertsSuppressionRuleName': self._serialize.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + alerts_suppression_rule_name=alerts_suppression_rule_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -278,3 +259,4 @@ async def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_allowed_connections_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_allowed_connections_operations.py index 86f634a6af276..6b9ee0056aa9f 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_allowed_connections_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_allowed_connections_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._allowed_connections_operations import build_get_request, build_list_by_home_region_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """Gets the list of all possible traffic between resources for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AllowedConnectionsList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AllowedConnectionsList] + :return: An iterator like instance of either AllowedConnectionsList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AllowedConnectionsList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.AllowedConnectionsList"] @@ -57,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AllowedConnectionsList', pipeline_response) + deserialized = self._deserialize("AllowedConnectionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/allowedConnections'} # type: ignore + @distributed_trace def list_by_home_region( self, **kwargs: Any @@ -114,8 +119,10 @@ def list_by_home_region( """Gets the list of all possible traffic between resources for the subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AllowedConnectionsList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AllowedConnectionsList] + :return: An iterator like instance of either AllowedConnectionsList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AllowedConnectionsList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.AllowedConnectionsList"] @@ -123,35 +130,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AllowedConnectionsList', pipeline_response) + deserialized = self._deserialize("AllowedConnectionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -169,11 +172,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/allowedConnections'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -198,28 +203,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'connectionType': self._serialize.url("connection_type", connection_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + connection_type=connection_type, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -233,4 +228,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/allowedConnections/{connectionType}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_metadata_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_metadata_operations.py index df9555e68ef32..0f646ce467837 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_metadata_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_metadata_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._assessments_metadata_operations import build_create_in_subscription_request, build_delete_in_subscription_request, build_get_in_subscription_request, build_get_request, build_list_by_subscription_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,46 +46,46 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any - ) -> AsyncIterable["_models.SecurityAssessmentMetadataList"]: + ) -> AsyncIterable["_models.SecurityAssessmentMetadataResponseList"]: """Get metadata information on all assessment types. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecurityAssessmentMetadataList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataList] + :return: An iterator like instance of either SecurityAssessmentMetadataResponseList or the + result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataResponseList] :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataList"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponseList"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityAssessmentMetadataList', pipeline_response) + deserialized = self._deserialize("SecurityAssessmentMetadataResponseList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -98,49 +103,41 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/providers/Microsoft.Security/assessmentMetadata'} # type: ignore + @distributed_trace_async async def get( self, assessment_metadata_name: str, **kwargs: Any - ) -> "_models.SecurityAssessmentMetadata": + ) -> "_models.SecurityAssessmentMetadataResponse": """Get metadata information on an assessment type. :param assessment_metadata_name: The Assessment Key - Unique key for the assessment type. :type assessment_metadata_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessmentMetadata, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :return: SecurityAssessmentMetadataResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadata"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + assessment_metadata_name=assessment_metadata_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -148,58 +145,58 @@ async def get( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessmentMetadata', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentMetadataResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + + @distributed_trace def list_by_subscription( self, **kwargs: Any - ) -> AsyncIterable["_models.SecurityAssessmentMetadataList"]: + ) -> AsyncIterable["_models.SecurityAssessmentMetadataResponseList"]: """Get metadata information on all assessment types in a specific subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecurityAssessmentMetadataList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataList] + :return: An iterator like instance of either SecurityAssessmentMetadataResponseList or the + result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataResponseList] :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataList"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponseList"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityAssessmentMetadataList', pipeline_response) + deserialized = self._deserialize("SecurityAssessmentMetadataResponseList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -217,50 +214,42 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata'} # type: ignore + @distributed_trace_async async def get_in_subscription( self, assessment_metadata_name: str, **kwargs: Any - ) -> "_models.SecurityAssessmentMetadata": + ) -> "_models.SecurityAssessmentMetadataResponse": """Get metadata information on an assessment type in a specific subscription. :param assessment_metadata_name: The Assessment Key - Unique key for the assessment type. :type assessment_metadata_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessmentMetadata, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :return: SecurityAssessmentMetadataResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadata"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get_in_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_in_subscription_request( + assessment_metadata_name=assessment_metadata_name, + subscription_id=self._config.subscription_id, + template_url=self.get_in_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -268,61 +257,54 @@ async def get_in_subscription( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessmentMetadata', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentMetadataResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + get_in_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + + @distributed_trace_async async def create_in_subscription( self, assessment_metadata_name: str, - assessment_metadata: "_models.SecurityAssessmentMetadata", + assessment_metadata: "_models.SecurityAssessmentMetadataResponse", **kwargs: Any - ) -> "_models.SecurityAssessmentMetadata": + ) -> "_models.SecurityAssessmentMetadataResponse": """Create metadata information on an assessment type in a specific subscription. :param assessment_metadata_name: The Assessment Key - Unique key for the assessment type. :type assessment_metadata_name: str :param assessment_metadata: AssessmentMetadata object. - :type assessment_metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :type assessment_metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessmentMetadata, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :return: SecurityAssessmentMetadataResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadata"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_in_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(assessment_metadata, 'SecurityAssessmentMetadataResponse') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_in_subscription_request( + assessment_metadata_name=assessment_metadata_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + json=_json, + template_url=self.create_in_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(assessment_metadata, 'SecurityAssessmentMetadata') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -330,14 +312,17 @@ async def create_in_subscription( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessmentMetadata', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentMetadataResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + create_in_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + + @distributed_trace_async async def delete_in_subscription( self, assessment_metadata_name: str, @@ -358,26 +343,16 @@ async def delete_in_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete_in_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_in_subscription_request( + assessment_metadata_name=assessment_metadata_name, + subscription_id=self._config.subscription_id, + template_url=self.delete_in_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -389,3 +364,4 @@ async def delete_in_subscription( return cls(pipeline_response, None, {}) delete_in_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_operations.py index 306648b7d29e5..daa92fe211ac7 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_assessments_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._assessments_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, scope: str, @@ -53,8 +59,10 @@ def list( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecurityAssessmentList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityAssessmentList] + :return: An iterator like instance of either SecurityAssessmentList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentList"] @@ -62,34 +70,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityAssessmentList', pipeline_response) + deserialized = self._deserialize("SecurityAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,18 +110,20 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/assessments'} # type: ignore + @distributed_trace_async async def get( self, resource_id: str, assessment_name: str, expand: Optional[Union[str, "_models.ExpandEnum"]] = None, **kwargs: Any - ) -> "_models.SecurityAssessment": + ) -> "_models.SecurityAssessmentResponse": """Get a security assessment on your scanned resource. :param resource_id: The identifier of the resource. @@ -128,37 +133,26 @@ async def get( :param expand: OData expand. Optional. :type expand: str or ~azure.mgmt.security.models.ExpandEnum :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessment, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessment + :return: SecurityAssessmentResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessment"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if expand is not None: - query_parameters['$expand'] = self._serialize.query("expand", expand, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + resource_id=resource_id, + assessment_name=assessment_name, + expand=expand, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -166,21 +160,24 @@ async def get( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessment', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, resource_id: str, assessment_name: str, assessment: "_models.SecurityAssessment", **kwargs: Any - ) -> "_models.SecurityAssessment": + ) -> "_models.SecurityAssessmentResponse": """Create a security assessment on your resource. An assessment metadata that describes this assessment must be predefined with the same name before inserting the assessment result. @@ -191,40 +188,30 @@ async def create_or_update( :param assessment: Calculated assessment on a pre-defined assessment metadata. :type assessment: ~azure.mgmt.security.models.SecurityAssessment :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessment, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessment + :return: SecurityAssessmentResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessment"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(assessment, 'SecurityAssessment') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_or_update_request( + resource_id=resource_id, + assessment_name=assessment_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(assessment, 'SecurityAssessment') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -233,17 +220,20 @@ async def create_or_update( raise HttpResponseError(response=response, error_format=ARMErrorFormat) if response.status_code == 200: - deserialized = self._deserialize('SecurityAssessment', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentResponse', pipeline_response) if response.status_code == 201: - deserialized = self._deserialize('SecurityAssessment', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}'} # type: ignore + + @distributed_trace_async async def delete( self, resource_id: str, @@ -267,26 +257,16 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + resource_id=resource_id, + assessment_name=assessment_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -298,3 +278,4 @@ async def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_auto_provisioning_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_auto_provisioning_settings_operations.py index ab252d8978e65..f7e242eb58e06 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_auto_provisioning_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_auto_provisioning_settings_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._auto_provisioning_settings_operations import build_create_request, build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """Exposes the auto provisioning settings of the subscriptions. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AutoProvisioningSettingList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AutoProvisioningSettingList] + :return: An iterator like instance of either AutoProvisioningSettingList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.AutoProvisioningSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.AutoProvisioningSettingList"] @@ -57,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AutoProvisioningSettingList', pipeline_response) + deserialized = self._deserialize("AutoProvisioningSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings'} # type: ignore + @distributed_trace_async async def get( self, setting_name: str, @@ -126,26 +131,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -159,8 +154,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings/{settingName}'} # type: ignore + + @distributed_trace_async async def create( self, setting_name: str, @@ -183,31 +181,21 @@ async def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(setting, 'AutoProvisioningSetting') + + request = build_create_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(setting, 'AutoProvisioningSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -221,4 +209,6 @@ async def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings/{settingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_automations_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_automations_operations.py index b2f197c45478e..90b09f8da05d5 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_automations_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_automations_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._automations_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_by_resource_group_request, build_list_request, build_validate_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -58,34 +64,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AutomationList', pipeline_response) + deserialized = self._deserialize("AutomationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -103,11 +104,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/automations'} # type: ignore + @distributed_trace def list_by_resource_group( self, resource_group_name: str, @@ -129,35 +132,31 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AutomationList', pipeline_response) + deserialized = self._deserialize("AutomationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -175,11 +174,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -203,27 +204,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -237,8 +228,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, resource_group_name: str, @@ -266,32 +260,22 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(automation, 'Automation') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(automation, 'Automation') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -309,8 +293,11 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}'} # type: ignore + + @distributed_trace_async async def delete( self, resource_group_name: str, @@ -334,27 +321,17 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -367,6 +344,8 @@ async def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}'} # type: ignore + + @distributed_trace_async async def validate( self, resource_group_name: str, @@ -394,32 +373,22 @@ async def validate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.validate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(automation, 'Automation') + + request = build_validate_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + content_type=content_type, + json=_json, + template_url=self.validate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(automation, 'Automation') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -433,4 +402,6 @@ async def validate( return cls(pipeline_response, deserialized, {}) return deserialized + validate.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}/validate'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliance_results_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliance_results_operations.py index 791a6cba37364..154988ad37230 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliance_results_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliance_results_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._compliance_results_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, scope: str, @@ -53,8 +59,10 @@ def list( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ComplianceResultList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ComplianceResultList] + :return: An iterator like instance of either ComplianceResultList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ComplianceResultList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.ComplianceResultList"] @@ -62,34 +70,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('ComplianceResultList', pipeline_response) + deserialized = self._deserialize("ComplianceResultList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,11 +110,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/complianceResults'} # type: ignore + @distributed_trace_async async def get( self, resource_id: str, @@ -134,26 +139,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'complianceResultName': self._serialize.url("compliance_result_name", compliance_result_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + resource_id=resource_id, + compliance_result_name=compliance_result_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -167,4 +162,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/complianceResults/{complianceResultName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliances_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliances_operations.py index 6e8444354d248..17b84be2bccb9 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliances_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_compliances_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._compliances_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, scope: str, @@ -62,34 +68,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('ComplianceList', pipeline_response) + deserialized = self._deserialize("ComplianceList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,11 +108,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/compliances'} # type: ignore + @distributed_trace_async async def get( self, scope: str, @@ -136,26 +139,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'complianceName': self._serialize.url("compliance_name", compliance_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scope=scope, + compliance_name=compliance_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -169,4 +162,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{scope}/providers/Microsoft.Security/compliances/{complianceName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_connectors_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_connectors_operations.py index 5df7e742ad66b..40fbbdd65c4dd 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_connectors_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_connectors_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._connectors_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """Cloud accounts connectors of a subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ConnectorSettingList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ConnectorSettingList] + :return: An iterator like instance of either ConnectorSettingList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ConnectorSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.ConnectorSettingList"] @@ -57,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('ConnectorSettingList', pipeline_response) + deserialized = self._deserialize("ConnectorSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors'} # type: ignore + @distributed_trace_async async def get( self, connector_name: str, @@ -126,26 +131,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'connectorName': self._serialize.url("connector_name", connector_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + connector_name=connector_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -159,8 +154,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, connector_name: str, @@ -185,31 +183,21 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'connectorName': self._serialize.url("connector_name", connector_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(connector_setting, 'ConnectorSetting') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + connector_name=connector_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(connector_setting, 'ConnectorSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -223,8 +211,11 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}'} # type: ignore + + @distributed_trace_async async def delete( self, connector_name: str, @@ -244,26 +235,16 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'connectorName': self._serialize.url("connector_name", connector_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + connector_name=connector_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -275,3 +256,4 @@ async def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_assessment_automations_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_assessment_automations_operations.py new file mode 100644 index 0000000000000..ba3ece22716a4 --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_assessment_automations_operations.py @@ -0,0 +1,360 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union +import warnings + +from azure.core.async_paging import AsyncItemPaged, AsyncList +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.mgmt.core.exceptions import ARMErrorFormat + +from ... import models as _models +from ..._vendor import _convert_request +from ...operations._custom_assessment_automations_operations import build_create_request, build_delete_request, build_get_request, build_list_by_resource_group_request, build_list_by_subscription_request +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + +class CustomAssessmentAutomationsOperations: + """CustomAssessmentAutomationsOperations async operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer) -> None: + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + custom_assessment_automation_name: str, + **kwargs: Any + ) -> "_models.CustomAssessmentAutomation": + """Gets a custom assessment automation. + + Gets a single custom assessment automation by name for the provided subscription and resource + group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_assessment_automation_name: Name of the Custom Assessment Automation. + :type custom_assessment_automation_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomAssessmentAutomation, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomAssessmentAutomation + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomation"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_assessment_automation_name=custom_assessment_automation_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('CustomAssessmentAutomation', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}'} # type: ignore + + + @distributed_trace_async + async def create( + self, + resource_group_name: str, + custom_assessment_automation_name: str, + custom_assessment_automation_body: "_models.CustomAssessmentAutomationRequest", + **kwargs: Any + ) -> "_models.CustomAssessmentAutomation": + """Creates a custom assessment automation. + + Creates or updates a custom assessment automation for the provided subscription. Please note + that providing an existing custom assessment automation will replace the existing record. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_assessment_automation_name: Name of the Custom Assessment Automation. + :type custom_assessment_automation_name: str + :param custom_assessment_automation_body: Custom Assessment Automation body. + :type custom_assessment_automation_body: + ~azure.mgmt.security.models.CustomAssessmentAutomationRequest + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomAssessmentAutomation, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomAssessmentAutomation + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomation"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(custom_assessment_automation_body, 'CustomAssessmentAutomationRequest') + + request = build_create_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_assessment_automation_name=custom_assessment_automation_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if response.status_code == 200: + deserialized = self._deserialize('CustomAssessmentAutomation', pipeline_response) + + if response.status_code == 201: + deserialized = self._deserialize('CustomAssessmentAutomation', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + create.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}'} # type: ignore + + + @distributed_trace_async + async def delete( + self, + resource_group_name: str, + custom_assessment_automation_name: str, + **kwargs: Any + ) -> None: + """Deletes a custom assessment automation. + + Deletes a custom assessment automation by name for a provided subscription. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_assessment_automation_name: Name of the Custom Assessment Automation. + :type custom_assessment_automation_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: None, or the result of cls(response) + :rtype: None + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType[None] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_assessment_automation_name=custom_assessment_automation_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) + + delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}'} # type: ignore + + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> AsyncIterable["_models.CustomAssessmentAutomationsListResult"]: + """List custom assessment automations in a subscription and a resource group. + + List custom assessment automations by provided subscription and resource group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomAssessmentAutomationsListResult or the + result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.CustomAssessmentAutomationsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomationsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + async def extract_data(pipeline_response): + deserialized = self._deserialize("CustomAssessmentAutomationsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return AsyncItemPaged( + get_next, extract_data + ) + list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations'} # type: ignore + + @distributed_trace + def list_by_subscription( + self, + **kwargs: Any + ) -> AsyncIterable["_models.CustomAssessmentAutomationsListResult"]: + """List custom assessment automations in a subscription. + + List custom assessment automations by provided subscription. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomAssessmentAutomationsListResult or the + result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.CustomAssessmentAutomationsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomationsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + async def extract_data(pipeline_response): + deserialized = self._deserialize("CustomAssessmentAutomationsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return AsyncItemPaged( + get_next, extract_data + ) + list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/customAssessmentAutomations'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_entity_store_assignments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_entity_store_assignments_operations.py new file mode 100644 index 0000000000000..3acb99309acee --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_custom_entity_store_assignments_operations.py @@ -0,0 +1,362 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union +import warnings + +from azure.core.async_paging import AsyncItemPaged, AsyncList +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.mgmt.core.exceptions import ARMErrorFormat + +from ... import models as _models +from ..._vendor import _convert_request +from ...operations._custom_entity_store_assignments_operations import build_create_request, build_delete_request, build_get_request, build_list_by_resource_group_request, build_list_by_subscription_request +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + +class CustomEntityStoreAssignmentsOperations: + """CustomEntityStoreAssignmentsOperations async operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer) -> None: + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + custom_entity_store_assignment_name: str, + **kwargs: Any + ) -> "_models.CustomEntityStoreAssignment": + """Gets a custom entity store assignment. + + Gets a single custom entity store assignment by name for the provided subscription and resource + group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_entity_store_assignment_name: Name of the custom entity store assignment. + Generated name is GUID. + :type custom_entity_store_assignment_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomEntityStoreAssignment, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomEntityStoreAssignment + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignment"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_entity_store_assignment_name=custom_entity_store_assignment_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('CustomEntityStoreAssignment', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}'} # type: ignore + + + @distributed_trace_async + async def create( + self, + resource_group_name: str, + custom_entity_store_assignment_name: str, + custom_entity_store_assignment_request_body: "_models.CustomEntityStoreAssignmentRequest", + **kwargs: Any + ) -> "_models.CustomEntityStoreAssignment": + """Creates a custom entity store assignment. + + Creates a custom entity store assignment for the provided subscription, if not already exists. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_entity_store_assignment_name: Name of the custom entity store assignment. + Generated name is GUID. + :type custom_entity_store_assignment_name: str + :param custom_entity_store_assignment_request_body: Custom entity store assignment body. + :type custom_entity_store_assignment_request_body: + ~azure.mgmt.security.models.CustomEntityStoreAssignmentRequest + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomEntityStoreAssignment, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomEntityStoreAssignment + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignment"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(custom_entity_store_assignment_request_body, 'CustomEntityStoreAssignmentRequest') + + request = build_create_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_entity_store_assignment_name=custom_entity_store_assignment_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if response.status_code == 200: + deserialized = self._deserialize('CustomEntityStoreAssignment', pipeline_response) + + if response.status_code == 201: + deserialized = self._deserialize('CustomEntityStoreAssignment', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + create.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}'} # type: ignore + + + @distributed_trace_async + async def delete( + self, + resource_group_name: str, + custom_entity_store_assignment_name: str, + **kwargs: Any + ) -> None: + """Deleted a custom entity store assignment. + + Delete a custom entity store assignment by name for a provided subscription. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_entity_store_assignment_name: Name of the custom entity store assignment. + Generated name is GUID. + :type custom_entity_store_assignment_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: None, or the result of cls(response) + :rtype: None + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType[None] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_entity_store_assignment_name=custom_entity_store_assignment_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) + + delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}'} # type: ignore + + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> AsyncIterable["_models.CustomEntityStoreAssignmentsListResult"]: + """List custom entity store assignments in a subscription and a resource group. + + List custom entity store assignments by a provided subscription and resource group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomEntityStoreAssignmentsListResult or the + result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.CustomEntityStoreAssignmentsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignmentsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + async def extract_data(pipeline_response): + deserialized = self._deserialize("CustomEntityStoreAssignmentsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return AsyncItemPaged( + get_next, extract_data + ) + list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments'} # type: ignore + + @distributed_trace + def list_by_subscription( + self, + **kwargs: Any + ) -> AsyncIterable["_models.CustomEntityStoreAssignmentsListResult"]: + """List custom entity store assignments in a subscription. + + List custom entity store assignments by provided subscription. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomEntityStoreAssignmentsListResult or the + result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.CustomEntityStoreAssignmentsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignmentsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + async def extract_data(pipeline_response): + deserialized = self._deserialize("CustomEntityStoreAssignmentsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return AsyncItemPaged( + get_next, extract_data + ) + list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/customEntityStoreAssignments'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_device_security_groups_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_device_security_groups_operations.py index 25828e3acbb71..154565d37deca 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_device_security_groups_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_device_security_groups_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._device_security_groups_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, resource_id: str, @@ -51,8 +57,10 @@ def list( :param resource_id: The identifier of the resource. :type resource_id: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DeviceSecurityGroupList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.DeviceSecurityGroupList] + :return: An iterator like instance of either DeviceSecurityGroupList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.DeviceSecurityGroupList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.DeviceSecurityGroupList"] @@ -60,34 +68,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + resource_id=resource_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + resource_id=resource_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('DeviceSecurityGroupList', pipeline_response) + deserialized = self._deserialize("DeviceSecurityGroupList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -105,11 +108,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups'} # type: ignore + @distributed_trace_async async def get( self, resource_id: str, @@ -133,26 +138,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'deviceSecurityGroupName': self._serialize.url("device_security_group_name", device_security_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + resource_id=resource_id, + device_security_group_name=device_security_group_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -166,8 +161,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, resource_id: str, @@ -195,31 +193,21 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'deviceSecurityGroupName': self._serialize.url("device_security_group_name", device_security_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(device_security_group, 'DeviceSecurityGroup') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_or_update_request( + resource_id=resource_id, + device_security_group_name=device_security_group_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(device_security_group, 'DeviceSecurityGroup') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -237,8 +225,11 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}'} # type: ignore + + @distributed_trace_async async def delete( self, resource_id: str, @@ -262,26 +253,16 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'deviceSecurityGroupName': self._serialize.url("device_security_group_name", device_security_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + resource_id=resource_id, + device_security_group_name=device_security_group_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -293,3 +274,4 @@ async def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_discovered_security_solutions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_discovered_security_solutions_operations.py index 7fac89302664d..95cfd53123710 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_discovered_security_solutions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_discovered_security_solutions_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._discovered_security_solutions_operations import build_get_request, build_list_by_home_region_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """Gets a list of discovered Security Solutions for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] + :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.DiscoveredSecuritySolutionList"] @@ -57,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('DiscoveredSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("DiscoveredSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/discoveredSecuritySolutions'} # type: ignore + @distributed_trace def list_by_home_region( self, **kwargs: Any @@ -114,8 +119,10 @@ def list_by_home_region( """Gets a list of discovered Security Solutions for the subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] + :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.DiscoveredSecuritySolutionList"] @@ -123,35 +130,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('DiscoveredSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("DiscoveredSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -169,11 +172,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/discoveredSecuritySolutions'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -197,28 +202,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'discoveredSecuritySolutionName': self._serialize.url("discovered_security_solution_name", discovered_security_solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + discovered_security_solution_name=discovered_security_solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -232,4 +227,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/discoveredSecuritySolutions/{discoveredSecuritySolutionName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_external_security_solutions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_external_security_solutions_operations.py index 5734e6ff6fce7..47f409d137bbc 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_external_security_solutions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_external_security_solutions_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._external_security_solutions_operations import build_get_request, build_list_by_home_region_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """Gets a list of external security solutions for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ExternalSecuritySolutionList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ExternalSecuritySolutionList] + :return: An iterator like instance of either ExternalSecuritySolutionList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ExternalSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.ExternalSecuritySolutionList"] @@ -57,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('ExternalSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("ExternalSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/externalSecuritySolutions'} # type: ignore + @distributed_trace def list_by_home_region( self, **kwargs: Any @@ -114,8 +119,10 @@ def list_by_home_region( """Gets a list of external Security Solutions for the subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ExternalSecuritySolutionList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ExternalSecuritySolutionList] + :return: An iterator like instance of either ExternalSecuritySolutionList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.ExternalSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.ExternalSecuritySolutionList"] @@ -123,35 +130,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('ExternalSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("ExternalSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -169,11 +172,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/ExternalSecuritySolutions'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -197,28 +202,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'externalSecuritySolutionsName': self._serialize.url("external_security_solutions_name", external_security_solutions_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + external_security_solutions_name=external_security_solutions_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -232,4 +227,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/ExternalSecuritySolutions/{externalSecuritySolutionsName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_information_protection_policies_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_information_protection_policies_operations.py index 821084dba2219..8e3f55214dbec 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_information_protection_policies_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_information_protection_policies_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._information_protection_policies_operations import build_create_or_update_request, build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,10 +46,11 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def get( self, scope: str, - information_protection_policy_name: Union[str, "_models.Enum15"], + information_protection_policy_name: Union[str, "_models.Enum17"], **kwargs: Any ) -> "_models.InformationProtectionPolicy": """Details of the information protection policy. @@ -54,7 +60,7 @@ async def get( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :param information_protection_policy_name: Name of the information protection policy. - :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum15 + :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum17 :keyword callable cls: A custom type or function that will be passed the direct response :return: InformationProtectionPolicy, or the result of cls(response) :rtype: ~azure.mgmt.security.models.InformationProtectionPolicy @@ -65,26 +71,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'informationProtectionPolicyName': self._serialize.url("information_protection_policy_name", information_protection_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scope=scope, + information_protection_policy_name=information_protection_policy_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -98,12 +94,15 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{scope}/providers/Microsoft.Security/informationProtectionPolicies/{informationProtectionPolicyName}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, scope: str, - information_protection_policy_name: Union[str, "_models.Enum15"], + information_protection_policy_name: Union[str, "_models.Enum17"], information_protection_policy: "_models.InformationProtectionPolicy", **kwargs: Any ) -> "_models.InformationProtectionPolicy": @@ -114,7 +113,7 @@ async def create_or_update( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :param information_protection_policy_name: Name of the information protection policy. - :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum15 + :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum17 :param information_protection_policy: Information protection policy. :type information_protection_policy: ~azure.mgmt.security.models.InformationProtectionPolicy :keyword callable cls: A custom type or function that will be passed the direct response @@ -127,31 +126,21 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'informationProtectionPolicyName': self._serialize.url("information_protection_policy_name", information_protection_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(information_protection_policy, 'InformationProtectionPolicy') + + request = build_create_or_update_request( + scope=scope, + information_protection_policy_name=information_protection_policy_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(information_protection_policy, 'InformationProtectionPolicy') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -169,8 +158,11 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{scope}/providers/Microsoft.Security/informationProtectionPolicies/{informationProtectionPolicyName}'} # type: ignore + + @distributed_trace def list( self, scope: str, @@ -183,8 +175,10 @@ def list( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either InformationProtectionPolicyList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.InformationProtectionPolicyList] + :return: An iterator like instance of either InformationProtectionPolicyList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.InformationProtectionPolicyList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.InformationProtectionPolicyList"] @@ -192,34 +186,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('InformationProtectionPolicyList', pipeline_response) + deserialized = self._deserialize("InformationProtectionPolicyList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -237,6 +226,7 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_ingestion_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_ingestion_settings_operations.py index d601b6ff1714d..3a0fbc853f7af 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_ingestion_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_ingestion_settings_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._ingestion_settings_operations import build_create_request, build_delete_request, build_get_request, build_list_connection_strings_request, build_list_request, build_list_tokens_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -49,8 +55,10 @@ def list( subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IngestionSettingList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IngestionSettingList] + :return: An iterator like instance of either IngestionSettingList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IngestionSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.IngestionSettingList"] @@ -58,34 +66,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('IngestionSettingList', pipeline_response) + deserialized = self._deserialize("IngestionSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -103,11 +106,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings'} # type: ignore + @distributed_trace_async async def get( self, ingestion_setting_name: str, @@ -128,26 +133,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -161,8 +156,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}'} # type: ignore + + @distributed_trace_async async def create( self, ingestion_setting_name: str, @@ -186,31 +184,21 @@ async def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(ingestion_setting, 'IngestionSetting') + + request = build_create_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(ingestion_setting, 'IngestionSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -224,8 +212,11 @@ async def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}'} # type: ignore + + @distributed_trace_async async def delete( self, ingestion_setting_name: str, @@ -245,26 +236,16 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -277,6 +258,8 @@ async def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}'} # type: ignore + + @distributed_trace_async async def list_tokens( self, ingestion_setting_name: str, @@ -297,26 +280,16 @@ async def list_tokens( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.list_tokens.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_tokens_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.list_tokens.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -330,8 +303,11 @@ async def list_tokens( return cls(pipeline_response, deserialized, {}) return deserialized + list_tokens.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}/listTokens'} # type: ignore + + @distributed_trace_async async def list_connection_strings( self, ingestion_setting_name: str, @@ -351,26 +327,16 @@ async def list_connection_strings( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.list_connection_strings.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_connection_strings_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.list_connection_strings.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -384,4 +350,6 @@ async def list_connection_strings( return cls(pipeline_response, deserialized, {}) return deserialized + list_connection_strings.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}/listConnectionStrings'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_analytics_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_analytics_operations.py index ec6b2c993fdea..d73d0040cf2ad 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_analytics_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_analytics_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._iot_security_solution_analytics_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def list( self, resource_group_name: str, @@ -63,27 +68,17 @@ async def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -97,8 +92,11 @@ async def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels'} # type: ignore + + @distributed_trace_async async def get( self, resource_group_name: str, @@ -122,27 +120,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -156,4 +144,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_operations.py index 0caa63196f556..4304f86fb9238 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solution_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._iot_security_solution_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_by_resource_group_request, build_list_by_subscription_request, build_update_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list_by_subscription( self, filter: Optional[str] = None, @@ -52,8 +58,10 @@ def list_by_subscription( iotHubs. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecuritySolutionsList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecuritySolutionsList] + :return: An iterator like instance of either IoTSecuritySolutionsList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecuritySolutionsList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.IoTSecuritySolutionsList"] @@ -61,36 +69,31 @@ def list_by_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecuritySolutionsList', pipeline_response) + deserialized = self._deserialize("IoTSecuritySolutionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -108,11 +111,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/iotSecuritySolutions'} # type: ignore + @distributed_trace def list_by_resource_group( self, resource_group_name: str, @@ -128,8 +133,10 @@ def list_by_resource_group( iotHubs. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecuritySolutionsList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecuritySolutionsList] + :return: An iterator like instance of either IoTSecuritySolutionsList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecuritySolutionsList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.IoTSecuritySolutionsList"] @@ -137,37 +144,33 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + filter=filter, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecuritySolutionsList', pipeline_response) + deserialized = self._deserialize("IoTSecuritySolutionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -185,11 +188,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -213,27 +218,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -247,8 +242,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, resource_group_name: str, @@ -275,32 +273,22 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(iot_security_solution_data, 'IoTSecuritySolutionModel') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(iot_security_solution_data, 'IoTSecuritySolutionModel') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -318,8 +306,11 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + + @distributed_trace_async async def update( self, resource_group_name: str, @@ -336,7 +327,8 @@ async def update( :param solution_name: The name of the IoT Security solution. :type solution_name: str :param update_iot_security_solution_data: The security solution data. - :type update_iot_security_solution_data: ~azure.mgmt.security.models.UpdateIotSecuritySolutionData + :type update_iot_security_solution_data: + ~azure.mgmt.security.models.UpdateIotSecuritySolutionData :keyword callable cls: A custom type or function that will be passed the direct response :return: IoTSecuritySolutionModel, or the result of cls(response) :rtype: ~azure.mgmt.security.models.IoTSecuritySolutionModel @@ -347,32 +339,22 @@ async def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(update_iot_security_solution_data, 'UpdateIotSecuritySolutionData') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(update_iot_security_solution_data, 'UpdateIotSecuritySolutionData') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -386,8 +368,11 @@ async def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + + @distributed_trace_async async def delete( self, resource_group_name: str, @@ -411,27 +396,17 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -443,3 +418,4 @@ async def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py index 80c3c47f7db1d..8a45c52a65757 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._iot_security_solutions_analytics_aggregated_alert_operations import build_dismiss_request, build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, resource_group_name: str, @@ -58,8 +64,10 @@ def list( :param top: Number of results to retrieve. :type top: int :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecurityAggregatedAlertList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedAlertList] + :return: An iterator like instance of either IoTSecurityAggregatedAlertList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedAlertList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.IoTSecurityAggregatedAlertList"] @@ -67,38 +75,35 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecurityAggregatedAlertList', pipeline_response) + deserialized = self._deserialize("IoTSecurityAggregatedAlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -116,11 +121,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -148,28 +155,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - 'aggregatedAlertName': self._serialize.url("aggregated_alert_name", aggregated_alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + aggregated_alert_name=aggregated_alert_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -183,8 +180,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts/{aggregatedAlertName}'} # type: ignore + + @distributed_trace_async async def dismiss( self, resource_group_name: str, @@ -211,28 +211,18 @@ async def dismiss( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.dismiss.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - 'aggregatedAlertName': self._serialize.url("aggregated_alert_name", aggregated_alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_dismiss_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + aggregated_alert_name=aggregated_alert_name, + template_url=self.dismiss.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -244,3 +234,4 @@ async def dismiss( return cls(pipeline_response, None, {}) dismiss.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts/{aggregatedAlertName}/dismiss'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_recommendation_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_recommendation_operations.py index 771b12ab1ca2d..39a2feece76cb 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_recommendation_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_iot_security_solutions_analytics_recommendation_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._iot_security_solutions_analytics_recommendation_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def get( self, resource_group_name: str, @@ -68,28 +74,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - 'aggregatedRecommendationName': self._serialize.url("aggregated_recommendation_name", aggregated_recommendation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + aggregated_recommendation_name=aggregated_recommendation_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -103,8 +99,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedRecommendations/{aggregatedRecommendationName}'} # type: ignore + + @distributed_trace def list( self, resource_group_name: str, @@ -123,8 +122,10 @@ def list( :param top: Number of results to retrieve. :type top: int :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecurityAggregatedRecommendationList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendationList] + :return: An iterator like instance of either IoTSecurityAggregatedRecommendationList or the + result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendationList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.IoTSecurityAggregatedRecommendationList"] @@ -132,38 +133,35 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecurityAggregatedRecommendationList', pipeline_response) + deserialized = self._deserialize("IoTSecurityAggregatedRecommendationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -181,6 +179,7 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_jit_network_access_policies_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_jit_network_access_policies_operations.py index bda90ec34063e..149001d54090e 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_jit_network_access_policies_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_jit_network_access_policies_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._jit_network_access_policies_operations import build_create_or_update_request, build_delete_request, build_get_request, build_initiate_request, build_list_by_region_request, build_list_by_resource_group_and_region_request, build_list_by_resource_group_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """Policies for protecting resources using Just-in-Time access control. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.JitNetworkAccessPoliciesList"] @@ -57,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace def list_by_region( self, **kwargs: Any @@ -115,8 +120,10 @@ def list_by_region( location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.JitNetworkAccessPoliciesList"] @@ -124,35 +131,31 @@ def list_by_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -170,11 +173,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace def list_by_resource_group( self, resource_group_name: str, @@ -187,8 +192,10 @@ def list_by_resource_group( name is case insensitive. :type resource_group_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.JitNetworkAccessPoliciesList"] @@ -196,35 +203,31 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -242,11 +245,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace def list_by_resource_group_and_region( self, resource_group_name: str, @@ -259,8 +264,10 @@ def list_by_resource_group_and_region( name is case insensitive. :type resource_group_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.JitNetworkAccessPoliciesList"] @@ -268,36 +275,33 @@ def list_by_resource_group_and_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group_and_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_and_region_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + template_url=self.list_by_resource_group_and_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_and_region_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -315,11 +319,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_resource_group_and_region.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -344,28 +350,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -379,8 +375,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, resource_group_name: str, @@ -407,33 +406,23 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'JitNetworkAccessPolicy') + + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'JitNetworkAccessPolicy') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -447,8 +436,11 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}'} # type: ignore + + @distributed_trace_async async def delete( self, resource_group_name: str, @@ -472,28 +464,18 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -506,6 +488,8 @@ async def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}'} # type: ignore + + @distributed_trace_async async def initiate( self, resource_group_name: str, @@ -532,35 +516,23 @@ async def initiate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - jit_network_access_policy_initiate_type = "initiate" - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.initiate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - 'jitNetworkAccessPolicyInitiateType': self._serialize.url("jit_network_access_policy_initiate_type", jit_network_access_policy_initiate_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'JitNetworkAccessPolicyInitiateRequest') + + request = build_initiate_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + content_type=content_type, + json=_json, + template_url=self.initiate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'JitNetworkAccessPolicyInitiateRequest') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -574,4 +546,6 @@ async def initiate( return cls(pipeline_response, deserialized, {}) return deserialized + initiate.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}/{jitNetworkAccessPolicyInitiateType}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_locations_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_locations_operations.py index 0597ef7bfed60..aaa5db5e68814 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_locations_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_locations_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._locations_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -59,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('AscLocationList', pipeline_response) + deserialized = self._deserialize("AscLocationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -104,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations'} # type: ignore + @distributed_trace_async async def get( self, **kwargs: Any @@ -125,26 +128,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -158,4 +151,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_mde_onboardings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_mde_onboardings_operations.py new file mode 100644 index 0000000000000..d68c0681cd631 --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_mde_onboardings_operations.py @@ -0,0 +1,131 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar +import warnings + +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.mgmt.core.exceptions import ARMErrorFormat + +from ... import models as _models +from ..._vendor import _convert_request +from ...operations._mde_onboardings_operations import build_get_request, build_list_request +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + +class MdeOnboardingsOperations: + """MdeOnboardingsOperations async operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer) -> None: + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace_async + async def list( + self, + **kwargs: Any + ) -> "_models.MdeOnboardingDataList": + """The configuration or data needed to onboard the machine to MDE. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: MdeOnboardingDataList, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.MdeOnboardingDataList + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.MdeOnboardingDataList"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('MdeOnboardingDataList', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/mdeOnboardings'} # type: ignore + + + @distributed_trace_async + async def get( + self, + **kwargs: Any + ) -> "_models.MdeOnboardingData": + """The default configuration or data needed to onboard the machine to MDE. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: MdeOnboardingData, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.MdeOnboardingData + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.MdeOnboardingData"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('MdeOnboardingData', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/mdeOnboardings/default'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_operations.py index 81c86cb0e66b9..9b8b1fbd8ff4c 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._operations import build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -57,30 +63,27 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('OperationList', pipeline_response) + deserialized = self._deserialize("OperationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -98,6 +101,7 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_pricings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_pricings_operations.py index 2f5c1427124af..ee0ff48cd2c93 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_pricings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_pricings_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._pricings_operations import build_get_request, build_list_request, build_update_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def list( self, **kwargs: Any @@ -56,25 +61,15 @@ async def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2018-06-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -88,8 +83,11 @@ async def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings'} # type: ignore + + @distributed_trace_async async def get( self, pricing_name: str, @@ -109,26 +107,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2018-06-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'pricingName': self._serialize.url("pricing_name", pricing_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + pricing_name=pricing_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -142,8 +130,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings/{pricingName}'} # type: ignore + + @distributed_trace_async async def update( self, pricing_name: str, @@ -166,31 +157,21 @@ async def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2018-06-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'pricingName': self._serialize.url("pricing_name", pricing_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(pricing, 'Pricing') + + request = build_update_request( + subscription_id=self._config.subscription_id, + pricing_name=pricing_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(pricing, 'Pricing') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -204,4 +185,6 @@ async def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings/{pricingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_assessments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_assessments_operations.py index 32614aec3d996..0099cae607772 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_assessments_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_assessments_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._regulatory_compliance_assessments_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, regulatory_compliance_standard_name: str, @@ -57,8 +63,10 @@ def list( :param filter: OData filter. Optional. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either RegulatoryComplianceAssessmentList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.RegulatoryComplianceAssessmentList] + :return: An iterator like instance of either RegulatoryComplianceAssessmentList or the result + of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.RegulatoryComplianceAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.RegulatoryComplianceAssessmentList"] @@ -66,38 +74,35 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - 'regulatoryComplianceControlName': self._serialize.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('RegulatoryComplianceAssessmentList', pipeline_response) + deserialized = self._deserialize("RegulatoryComplianceAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -115,11 +120,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}/regulatoryComplianceAssessments'} # type: ignore + @distributed_trace_async async def get( self, regulatory_compliance_standard_name: str, @@ -146,28 +153,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - 'regulatoryComplianceControlName': self._serialize.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), - 'regulatoryComplianceAssessmentName': self._serialize.url("regulatory_compliance_assessment_name", regulatory_compliance_assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + regulatory_compliance_assessment_name=regulatory_compliance_assessment_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -181,4 +178,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}/regulatoryComplianceAssessments/{regulatoryComplianceAssessmentName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_controls_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_controls_operations.py index 2c64a381af9cc..ba9e29bb16f9c 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_controls_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_controls_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._regulatory_compliance_controls_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, regulatory_compliance_standard_name: str, @@ -54,8 +60,10 @@ def list( :param filter: OData filter. Optional. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either RegulatoryComplianceControlList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.RegulatoryComplianceControlList] + :return: An iterator like instance of either RegulatoryComplianceControlList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.RegulatoryComplianceControlList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.RegulatoryComplianceControlList"] @@ -63,37 +71,33 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('RegulatoryComplianceControlList', pipeline_response) + deserialized = self._deserialize("RegulatoryComplianceControlList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -111,11 +115,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls'} # type: ignore + @distributed_trace_async async def get( self, regulatory_compliance_standard_name: str, @@ -138,27 +144,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - 'regulatoryComplianceControlName': self._serialize.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -172,4 +168,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_standards_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_standards_operations.py index 5dbcd284f0d0b..06214a9d7050d 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_standards_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_regulatory_compliance_standards_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._regulatory_compliance_standards_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, filter: Optional[str] = None, @@ -51,8 +57,10 @@ def list( :param filter: OData filter. Optional. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either RegulatoryComplianceStandardList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.RegulatoryComplianceStandardList] + :return: An iterator like instance of either RegulatoryComplianceStandardList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.RegulatoryComplianceStandardList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.RegulatoryComplianceStandardList"] @@ -60,36 +68,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('RegulatoryComplianceStandardList', pipeline_response) + deserialized = self._deserialize("RegulatoryComplianceStandardList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,11 +110,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards'} # type: ignore + @distributed_trace_async async def get( self, regulatory_compliance_standard_name: str, @@ -131,26 +136,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -164,4 +159,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_control_definitions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_control_definitions_operations.py index e5df37ba2e9b9..1d9805ff2ee57 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_control_definitions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_control_definitions_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._secure_score_control_definitions_operations import build_list_by_subscription_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """List the available security controls, their assessments, and the max score. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] + :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecureScoreControlDefinitionList"] @@ -57,30 +65,27 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlDefinitionList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlDefinitionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -98,11 +103,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/providers/Microsoft.Security/secureScoreControlDefinitions'} # type: ignore + @distributed_trace def list_by_subscription( self, **kwargs: Any @@ -111,8 +118,10 @@ def list_by_subscription( max score. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] + :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecureScoreControlDefinitionList"] @@ -120,34 +129,29 @@ def list_by_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlDefinitionList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlDefinitionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -165,6 +169,7 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_controls_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_controls_operations.py index 07f3d874ea7e5..5403d7a1bd365 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_controls_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_score_controls_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._secure_score_controls_operations import build_list_by_secure_score_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list_by_secure_score( self, secure_score_name: str, @@ -55,8 +61,10 @@ def list_by_secure_score( :param expand: OData expand. Optional. :type expand: str or ~azure.mgmt.security.models.ExpandControlsEnum :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlList] + :return: An iterator like instance of either SecureScoreControlList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecureScoreControlList"] @@ -64,37 +72,33 @@ def list_by_secure_score( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_secure_score.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'secureScoreName': self._serialize.url("secure_score_name", secure_score_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if expand is not None: - query_parameters['$expand'] = self._serialize.query("expand", expand, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_secure_score_request( + subscription_id=self._config.subscription_id, + secure_score_name=secure_score_name, + expand=expand, + template_url=self.list_by_secure_score.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_secure_score_request( + subscription_id=self._config.subscription_id, + secure_score_name=secure_score_name, + expand=expand, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,11 +116,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_secure_score.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores/{secureScoreName}/secureScoreControls'} # type: ignore + @distributed_trace def list( self, expand: Optional[Union[str, "_models.ExpandControlsEnum"]] = None, @@ -127,8 +133,10 @@ def list( :param expand: OData expand. Optional. :type expand: str or ~azure.mgmt.security.models.ExpandControlsEnum :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlList] + :return: An iterator like instance of either SecureScoreControlList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecureScoreControlList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecureScoreControlList"] @@ -136,36 +144,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if expand is not None: - query_parameters['$expand'] = self._serialize.query("expand", expand, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + expand=expand, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + expand=expand, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -183,6 +186,7 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_scores_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_scores_operations.py index 90e1ceb3501bd..3862eb8a155d4 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_scores_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_secure_scores_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._secure_scores_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -57,34 +63,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoresList', pipeline_response) + deserialized = self._deserialize("SecureScoresList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +103,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores'} # type: ignore + @distributed_trace_async async def get( self, secure_score_name: str, @@ -128,26 +131,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'secureScoreName': self._serialize.url("secure_score_name", secure_score_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + secure_score_name=secure_score_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -161,4 +154,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores/{secureScoreName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_connectors_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_connectors_operations.py new file mode 100644 index 0000000000000..50aabd96f0879 --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_connectors_operations.py @@ -0,0 +1,410 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union +import warnings + +from azure.core.async_paging import AsyncItemPaged, AsyncList +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.mgmt.core.exceptions import ARMErrorFormat + +from ... import models as _models +from ..._vendor import _convert_request +from ...operations._security_connectors_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_by_resource_group_request, build_list_request, build_update_request +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + +class SecurityConnectorsOperations: + """SecurityConnectorsOperations async operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer) -> None: + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace + def list( + self, + **kwargs: Any + ) -> AsyncIterable["_models.SecurityConnectorsList"]: + """Lists all the security connectors in the specified subscription. Use the 'nextLink' property in + the response to get the next page of security connectors for the specified subscription. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either SecurityConnectorsList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityConnectorsList] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnectorsList"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + async def extract_data(pipeline_response): + deserialized = self._deserialize("SecurityConnectorsList", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return AsyncItemPaged( + get_next, extract_data + ) + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityConnectors'} # type: ignore + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> AsyncIterable["_models.SecurityConnectorsList"]: + """Lists all the security connectors in the specified resource group. Use the 'nextLink' property + in the response to get the next page of security connectors for the specified resource group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either SecurityConnectorsList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityConnectorsList] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnectorsList"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + async def extract_data(pipeline_response): + deserialized = self._deserialize("SecurityConnectorsList", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return AsyncItemPaged( + get_next, extract_data + ) + list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors'} # type: ignore + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + security_connector_name: str, + **kwargs: Any + ) -> "_models.SecurityConnector": + """Retrieves details of a specific security connector. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: SecurityConnector, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityConnector + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnector"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + security_connector_name: str, + security_connector: "_models.SecurityConnector", + **kwargs: Any + ) -> "_models.SecurityConnector": + """Creates or updates a security connector. If a security connector is already created and a + subsequent request is issued for the same security connector id, then it will be updated. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :param security_connector: The security connector resource. + :type security_connector: ~azure.mgmt.security.models.SecurityConnector + :keyword callable cls: A custom type or function that will be passed the direct response + :return: SecurityConnector, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityConnector + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnector"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(security_connector, 'SecurityConnector') + + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if response.status_code == 200: + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if response.status_code == 201: + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + security_connector_name: str, + security_connector: "_models.SecurityConnector", + **kwargs: Any + ) -> "_models.SecurityConnector": + """Updates a security connector. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :param security_connector: The security connector resource. + :type security_connector: ~azure.mgmt.security.models.SecurityConnector + :keyword callable cls: A custom type or function that will be passed the direct response + :return: SecurityConnector, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityConnector + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnector"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(security_connector, 'SecurityConnector') + + request = build_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + + + @distributed_trace_async + async def delete( + self, + resource_group_name: str, + security_connector_name: str, + **kwargs: Any + ) -> None: + """Deletes a security connector. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: None, or the result of cls(response) + :rtype: None + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType[None] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) + + delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_contacts_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_contacts_operations.py index d2466e69b609f..f9922be6b08ba 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_contacts_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_contacts_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._security_contacts_operations import build_create_request, build_delete_request, build_get_request, build_list_request, build_update_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -49,7 +55,8 @@ def list( :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SecurityContactList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityContactList] + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecurityContactList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityContactList"] @@ -57,34 +64,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityContactList', pipeline_response) + deserialized = self._deserialize("SecurityContactList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +104,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts'} # type: ignore + @distributed_trace_async async def get( self, security_contact_name: str, @@ -126,26 +130,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -159,8 +153,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + + @distributed_trace_async async def create( self, security_contact_name: str, @@ -183,31 +180,21 @@ async def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(security_contact, 'SecurityContact') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(security_contact, 'SecurityContact') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -221,8 +208,11 @@ async def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + + @distributed_trace_async async def delete( self, security_contact_name: str, @@ -242,26 +232,16 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -274,6 +254,8 @@ async def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + + @distributed_trace_async async def update( self, security_contact_name: str, @@ -296,31 +278,21 @@ async def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(security_contact, 'SecurityContact') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_update_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(security_contact, 'SecurityContact') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -334,4 +306,6 @@ async def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_operations.py index 24b5cc9fe67bf..5902a44cf1534 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._security_solutions_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -48,8 +54,10 @@ def list( """Gets a list of Security Solutions for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecuritySolutionList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecuritySolutionList] + :return: An iterator like instance of either SecuritySolutionList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecuritySolutionList"] @@ -57,34 +65,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecuritySolutionList', pipeline_response) + deserialized = self._deserialize("SecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +105,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securitySolutions'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -130,28 +135,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'securitySolutionName': self._serialize.url("security_solution_name", security_solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + security_solution_name=security_solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -165,4 +160,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/securitySolutions/{securitySolutionName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_reference_data_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_reference_data_operations.py index 944d446ef9e1a..8334616b667ae 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_reference_data_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_security_solutions_reference_data_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._security_solutions_reference_data_operations import build_list_by_home_region_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def list( self, **kwargs: Any @@ -56,25 +61,15 @@ async def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -88,8 +83,11 @@ async def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securitySolutionsReferenceData'} # type: ignore + + @distributed_trace_async async def list_by_home_region( self, **kwargs: Any @@ -106,26 +104,16 @@ async def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -139,4 +127,6 @@ async def list_by_home_region( return cls(pipeline_response, deserialized, {}) return deserialized + list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/securitySolutionsReferenceData'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_server_vulnerability_assessment_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_server_vulnerability_assessment_operations.py index f66bf89789203..9cb6a62fa3221 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_server_vulnerability_assessment_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_server_vulnerability_assessment_operations.py @@ -5,18 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._server_vulnerability_assessment_operations import build_create_or_update_request, build_delete_request_initial, build_get_request, build_list_by_extended_resource_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -42,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def list_by_extended_resource( self, resource_group_name: str, @@ -71,29 +76,19 @@ async def list_by_extended_resource( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list_by_extended_resource.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.list_by_extended_resource.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -107,8 +102,11 @@ async def list_by_extended_resource( return cls(pipeline_response, deserialized, {}) return deserialized + list_by_extended_resource.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments'} # type: ignore + + @distributed_trace_async async def get( self, resource_group_name: str, @@ -138,31 +136,19 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - server_vulnerability_assessment = "default" - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -176,8 +162,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore + + @distributed_trace_async async def create_or_update( self, resource_group_name: str, @@ -208,31 +197,19 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - server_vulnerability_assessment = "default" - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.put(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -246,8 +223,10 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore + async def _delete_initial( self, resource_group_name: str, @@ -261,31 +240,19 @@ async def _delete_initial( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - server_vulnerability_assessment = "default" - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self._delete_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request_initial( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self._delete_initial.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -298,6 +265,8 @@ async def _delete_initial( _delete_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore + + @distributed_trace_async async def begin_delete( self, resource_group_name: str, @@ -319,15 +288,17 @@ async def begin_delete( :type resource_name: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. - Pass in False for this operation to not poll, or pass in your own initialized polling object for a personal polling strategy. + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: + :raises: ~azure.core.exceptions.HttpResponseError """ - polling = kwargs.pop('polling', True) # type: Union[bool, AsyncPollingMethod] + polling = kwargs.pop('polling', True) # type: Union[bool, azure.core.polling.AsyncPollingMethod] cls = kwargs.pop('cls', None) # type: ClsType[None] lro_delay = kwargs.pop( 'polling_interval', @@ -343,25 +314,14 @@ async def begin_delete( cls=lambda x,y,z: x, **kwargs ) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) def get_long_running_output(pipeline_response): if cls: return cls(pipeline_response, None, {}) - server_vulnerability_assessment = "default" - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - if polling is True: polling_method = AsyncARMPolling(lro_delay, lro_options={'final-state-via': 'location'}, path_format_arguments=path_format_arguments, **kwargs) + if polling is True: polling_method = AsyncARMPolling(lro_delay, lro_options={'final-state-via': 'location'}, **kwargs) elif polling is False: polling_method = AsyncNoPolling() else: polling_method = polling if cont_token: @@ -373,4 +333,5 @@ def get_long_running_output(pipeline_response): ) else: return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) + begin_delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_settings_operations.py index bd641d4d7e726..019f3a6a18120 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_settings_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._settings_operations import build_get_request, build_list_request, build_update_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -57,34 +63,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-07-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SettingsList', pipeline_response) + deserialized = self._deserialize("SettingsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,20 +103,22 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings'} # type: ignore + @distributed_trace_async async def get( self, - setting_name: Union[str, "_models.Enum69"], + setting_name: Union[str, "_models.Enum73"], **kwargs: Any ) -> "_models.Setting": """Settings of different configurations in security center. :param setting_name: The name of the setting. - :type setting_name: str or ~azure.mgmt.security.models.Enum69 + :type setting_name: str or ~azure.mgmt.security.models.Enum73 :keyword callable cls: A custom type or function that will be passed the direct response :return: Setting, or the result of cls(response) :rtype: ~azure.mgmt.security.models.Setting @@ -126,26 +129,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-07-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -159,18 +152,21 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings/{settingName}'} # type: ignore + + @distributed_trace_async async def update( self, - setting_name: Union[str, "_models.Enum69"], + setting_name: Union[str, "_models.Enum73"], setting: "_models.Setting", **kwargs: Any ) -> "_models.Setting": """updating settings about different configurations in security center. :param setting_name: The name of the setting. - :type setting_name: str or ~azure.mgmt.security.models.Enum69 + :type setting_name: str or ~azure.mgmt.security.models.Enum73 :param setting: Setting object. :type setting: ~azure.mgmt.security.models.Setting :keyword callable cls: A custom type or function that will be passed the direct response @@ -183,31 +179,21 @@ async def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-07-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(setting, 'Setting') + + request = build_update_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(setting, 'Setting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -221,4 +207,6 @@ async def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings/{settingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_software_inventories_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_software_inventories_operations.py index d38fc050a78a5..4c569c93d1903 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_software_inventories_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_software_inventories_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._software_inventories_operations import build_get_request, build_list_by_extended_resource_request, build_list_by_subscription_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list_by_extended_resource( self, resource_group_name: str, @@ -70,38 +76,37 @@ def list_by_extended_resource( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-05-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_extended_resource.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.list_by_extended_resource.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SoftwaresList', pipeline_response) + deserialized = self._deserialize("SoftwaresList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -119,11 +124,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_extended_resource.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/softwareInventories'} # type: ignore + @distributed_trace def list_by_subscription( self, **kwargs: Any @@ -140,34 +147,29 @@ def list_by_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-05-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SoftwaresList', pipeline_response) + deserialized = self._deserialize("SoftwaresList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -185,11 +187,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/softwareInventories'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -222,30 +226,20 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-05-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'softwareName': self._serialize.url("software_name", software_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + software_name=software_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -259,4 +253,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/softwareInventories/{softwareName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_baseline_rules_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_baseline_rules_operations.py index 025272f789c96..19b968123936d 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_baseline_rules_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_baseline_rules_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._sql_vulnerability_assessment_baseline_rules_operations import build_add_request, build_create_or_update_request, build_delete_request, build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def create_or_update( self, rule_id: str, @@ -70,35 +75,25 @@ async def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'ruleId': self._serialize.url("rule_id", rule_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - body_content_kwargs = {} # type: Dict[str, Any] if body is not None: - body_content = self._serialize.body(body, 'RuleResultsInput') + _json = self._serialize.body(body, 'RuleResultsInput') else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) + _json = None + + request = build_create_or_update_request( + rule_id=rule_id, + resource_id=resource_id, + content_type=content_type, + workspace_id=workspace_id, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -112,8 +107,11 @@ async def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'} # type: ignore + + @distributed_trace_async async def get( self, rule_id: str, @@ -141,27 +139,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'ruleId': self._serialize.url("rule_id", rule_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + rule_id=rule_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -175,8 +163,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'} # type: ignore + + @distributed_trace_async async def delete( self, rule_id: str, @@ -204,27 +195,17 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'ruleId': self._serialize.url("rule_id", rule_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + rule_id=rule_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -237,6 +218,8 @@ async def delete( delete.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'} # type: ignore + + @distributed_trace_async async def list( self, workspace_id: str, @@ -261,26 +244,16 @@ async def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -294,8 +267,11 @@ async def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules'} # type: ignore + + @distributed_trace_async async def add( self, workspace_id: str, @@ -323,34 +299,24 @@ async def add( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.add.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - body_content_kwargs = {} # type: Dict[str, Any] if body is not None: - body_content = self._serialize.body(body, 'RulesResultsInput') + _json = self._serialize.body(body, 'RulesResultsInput') else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) + _json = None + + request = build_add_request( + resource_id=resource_id, + content_type=content_type, + workspace_id=workspace_id, + json=_json, + template_url=self.add.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -364,4 +330,6 @@ async def add( return cls(pipeline_response, deserialized, {}) return deserialized + add.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scan_results_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scan_results_operations.py index 0ff8ebb2bbe3d..cd94725aceb18 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scan_results_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scan_results_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._sql_vulnerability_assessment_scan_results_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def get( self, scan_id: str, @@ -70,28 +75,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scanId': self._serialize.url("scan_id", scan_id, 'str'), - 'scanResultId': self._serialize.url("scan_result_id", scan_result_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scan_id=scan_id, + scan_result_id=scan_result_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -105,8 +100,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}/scanResults/{scanResultId}'} # type: ignore + + @distributed_trace_async async def list( self, scan_id: str, @@ -134,27 +132,17 @@ async def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scanId': self._serialize.url("scan_id", scan_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + scan_id=scan_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -168,4 +156,6 @@ async def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}/scanResults'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scans_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scans_operations.py index dc4e1348504bf..8a188cf2513a3 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scans_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sql_vulnerability_assessment_scans_operations.py @@ -5,16 +5,20 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._sql_vulnerability_assessment_scans_operations import build_get_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -40,6 +44,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace_async async def get( self, scan_id: str, @@ -67,27 +72,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scanId': self._serialize.url("scan_id", scan_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scan_id=scan_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -101,8 +96,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}'} # type: ignore + + @distributed_trace_async async def list( self, workspace_id: str, @@ -127,26 +125,16 @@ async def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -160,4 +148,6 @@ async def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sub_assessments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sub_assessments_operations.py index 36f5025511636..fe589cd32d6eb 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sub_assessments_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_sub_assessments_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._sub_assessments_operations import build_get_request, build_list_all_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list_all( self, scope: str, @@ -53,8 +59,10 @@ def list_all( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecuritySubAssessmentList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecuritySubAssessmentList] + :return: An iterator like instance of either SecuritySubAssessmentList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecuritySubAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecuritySubAssessmentList"] @@ -62,34 +70,29 @@ def list_all( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_all.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_all_request( + scope=scope, + template_url=self.list_all.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_all_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecuritySubAssessmentList', pipeline_response) + deserialized = self._deserialize("SecuritySubAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,11 +110,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_all.metadata = {'url': '/{scope}/providers/Microsoft.Security/subAssessments'} # type: ignore + @distributed_trace def list( self, scope: str, @@ -127,8 +132,10 @@ def list( :param assessment_name: The Assessment Key - Unique key for the assessment type. :type assessment_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecuritySubAssessmentList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecuritySubAssessmentList] + :return: An iterator like instance of either SecuritySubAssessmentList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.SecuritySubAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecuritySubAssessmentList"] @@ -136,35 +143,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + assessment_name=assessment_name, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + assessment_name=assessment_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecuritySubAssessmentList', pipeline_response) + deserialized = self._deserialize("SecuritySubAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -182,11 +185,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/assessments/{assessmentName}/subAssessments'} # type: ignore + @distributed_trace_async async def get( self, scope: str, @@ -214,27 +219,17 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - 'subAssessmentName': self._serialize.url("sub_assessment_name", sub_assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scope=scope, + assessment_name=assessment_name, + sub_assessment_name=sub_assessment_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -248,4 +243,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{scope}/providers/Microsoft.Security/assessments/{assessmentName}/subAssessments/{subAssessmentName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_tasks_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_tasks_operations.py index db7bbd5d59afd..3fed4e9c6c6be 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_tasks_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_tasks_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._tasks_operations import build_get_resource_group_level_task_request, build_get_subscription_level_task_request, build_list_by_home_region_request, build_list_by_resource_group_request, build_list_request, build_update_resource_group_level_task_state_request, build_update_subscription_level_task_state_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, filter: Optional[str] = None, @@ -60,36 +66,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityTaskList', pipeline_response) + deserialized = self._deserialize("SecurityTaskList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,11 +108,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/tasks'} # type: ignore + @distributed_trace def list_by_home_region( self, filter: Optional[str] = None, @@ -131,37 +134,33 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + filter=filter, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityTaskList', pipeline_response) + deserialized = self._deserialize("SecurityTaskList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -179,11 +178,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks'} # type: ignore + @distributed_trace_async async def get_subscription_level_task( self, task_name: str, @@ -203,27 +204,17 @@ async def get_subscription_level_task( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.get_subscription_level_task.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_subscription_level_task_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + task_name=task_name, + template_url=self.get_subscription_level_task.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -237,12 +228,15 @@ async def get_subscription_level_task( return cls(pipeline_response, deserialized, {}) return deserialized + get_subscription_level_task.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}'} # type: ignore + + @distributed_trace_async async def update_subscription_level_task_state( self, task_name: str, - task_update_action_type: Union[str, "_models.Enum13"], + task_update_action_type: Union[str, "_models.Enum15"], **kwargs: Any ) -> None: """Recommended tasks that will help improve the security of the subscription proactively. @@ -250,7 +244,7 @@ async def update_subscription_level_task_state( :param task_name: Name of the task object, will be a GUID. :type task_name: str :param task_update_action_type: Type of the action to do on the task. - :type task_update_action_type: str or ~azure.mgmt.security.models.Enum13 + :type task_update_action_type: str or ~azure.mgmt.security.models.Enum15 :keyword callable cls: A custom type or function that will be passed the direct response :return: None, or the result of cls(response) :rtype: None @@ -261,28 +255,18 @@ async def update_subscription_level_task_state( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_task_state.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - 'taskUpdateActionType': self._serialize.url("task_update_action_type", task_update_action_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_task_state_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + task_name=task_name, + task_update_action_type=task_update_action_type, + template_url=self.update_subscription_level_task_state.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -295,6 +279,8 @@ async def update_subscription_level_task_state( update_subscription_level_task_state.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}/{taskUpdateActionType}'} # type: ignore + + @distributed_trace def list_by_resource_group( self, resource_group_name: str, @@ -318,38 +304,35 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + filter=filter, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityTaskList', pipeline_response) + deserialized = self._deserialize("SecurityTaskList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -367,11 +350,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks'} # type: ignore + @distributed_trace_async async def get_resource_group_level_task( self, resource_group_name: str, @@ -395,28 +380,18 @@ async def get_resource_group_level_task( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.get_resource_group_level_task.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_resource_group_level_task_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + task_name=task_name, + template_url=self.get_resource_group_level_task.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -430,13 +405,16 @@ async def get_resource_group_level_task( return cls(pipeline_response, deserialized, {}) return deserialized + get_resource_group_level_task.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}'} # type: ignore + + @distributed_trace_async async def update_resource_group_level_task_state( self, resource_group_name: str, task_name: str, - task_update_action_type: Union[str, "_models.Enum13"], + task_update_action_type: Union[str, "_models.Enum15"], **kwargs: Any ) -> None: """Recommended tasks that will help improve the security of the subscription proactively. @@ -447,7 +425,7 @@ async def update_resource_group_level_task_state( :param task_name: Name of the task object, will be a GUID. :type task_name: str :param task_update_action_type: Type of the action to do on the task. - :type task_update_action_type: str or ~azure.mgmt.security.models.Enum13 + :type task_update_action_type: str or ~azure.mgmt.security.models.Enum15 :keyword callable cls: A custom type or function that will be passed the direct response :return: None, or the result of cls(response) :rtype: None @@ -458,29 +436,19 @@ async def update_resource_group_level_task_state( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_task_state.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - 'taskUpdateActionType': self._serialize.url("task_update_action_type", task_update_action_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_task_state_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + task_name=task_name, + task_update_action_type=task_update_action_type, + template_url=self.update_resource_group_level_task_state.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -492,3 +460,4 @@ async def update_resource_group_level_task_state( return cls(pipeline_response, None, {}) update_resource_group_level_task_state.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}/{taskUpdateActionType}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_topology_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_topology_operations.py index 9f2d93a6b94c6..521d4f3e9bf70 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_topology_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_topology_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._topology_operations import build_get_request, build_list_by_home_region_request, build_list_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -57,34 +63,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('TopologyList', pipeline_response) + deserialized = self._deserialize("TopologyList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -102,11 +103,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/topologies'} # type: ignore + @distributed_trace def list_by_home_region( self, **kwargs: Any @@ -123,35 +126,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('TopologyList', pipeline_response) + deserialized = self._deserialize("TopologyList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -169,11 +168,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/topologies'} # type: ignore + @distributed_trace_async async def get( self, resource_group_name: str, @@ -197,28 +198,18 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'topologyResourceName': self._serialize.url("topology_resource_name", topology_resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + topology_resource_name=topology_resource_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -232,4 +223,6 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/topologies/{topologyResourceName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_workspace_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_workspace_settings_operations.py index e2f283eec3d40..a1458187f9e4b 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_workspace_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/aio/operations/_workspace_settings_operations.py @@ -5,17 +5,22 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import functools from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - +from ..._vendor import _convert_request +from ...operations._workspace_settings_operations import build_create_request, build_delete_request, build_get_request, build_list_request, build_update_request T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -41,6 +46,7 @@ def __init__(self, client, config, serializer, deserializer) -> None: self._deserialize = deserializer self._config = config + @distributed_trace def list( self, **kwargs: Any @@ -49,8 +55,10 @@ def list( means that no custom-workspace configuration was set. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either WorkspaceSettingList or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.WorkspaceSettingList] + :return: An iterator like instance of either WorkspaceSettingList or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.security.models.WorkspaceSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.WorkspaceSettingList"] @@ -58,34 +66,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('WorkspaceSettingList', pipeline_response) + deserialized = self._deserialize("WorkspaceSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -103,11 +106,13 @@ async def get_next(next_link=None): return pipeline_response + return AsyncItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings'} # type: ignore + @distributed_trace_async async def get( self, workspace_setting_name: str, @@ -128,26 +133,16 @@ async def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -161,8 +156,11 @@ async def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore + + @distributed_trace_async async def create( self, workspace_setting_name: str, @@ -185,31 +183,21 @@ async def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(workspace_setting, 'WorkspaceSetting') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(workspace_setting, 'WorkspaceSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -223,8 +211,11 @@ async def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore + + @distributed_trace_async async def update( self, workspace_setting_name: str, @@ -247,31 +238,21 @@ async def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(workspace_setting, 'WorkspaceSetting') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_update_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(workspace_setting, 'WorkspaceSetting') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -285,8 +266,11 @@ async def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore + + @distributed_trace_async async def delete( self, workspace_setting_name: str, @@ -307,26 +291,16 @@ async def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -338,3 +312,4 @@ async def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/__init__.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/models/__init__.py index f841b789d1e39..74d04c5de5e14 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/__init__.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/models/__init__.py @@ -6,478 +6,273 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -try: - from ._models_py3 import AadConnectivityState - from ._models_py3 import AadExternalSecuritySolution - from ._models_py3 import AadSolutionProperties - from ._models_py3 import ActiveConnectionsNotInAllowedRange - from ._models_py3 import AdaptiveApplicationControlGroup - from ._models_py3 import AdaptiveApplicationControlGroups - from ._models_py3 import AdaptiveApplicationControlIssueSummary - from ._models_py3 import AdaptiveNetworkHardening - from ._models_py3 import AdaptiveNetworkHardeningEnforceRequest - from ._models_py3 import AdaptiveNetworkHardeningsList - from ._models_py3 import AdditionalData - from ._models_py3 import AdditionalWorkspacesProperties - from ._models_py3 import AdvancedThreatProtectionSetting - from ._models_py3 import Alert - from ._models_py3 import AlertEntity - from ._models_py3 import AlertList - from ._models_py3 import AlertSimulatorBundlesRequestProperties - from ._models_py3 import AlertSimulatorRequestBody - from ._models_py3 import AlertSimulatorRequestProperties - from ._models_py3 import AlertSyncSettings - from ._models_py3 import AlertsSuppressionRule - from ._models_py3 import AlertsSuppressionRulesList - from ._models_py3 import AllowedConnectionsList - from ._models_py3 import AllowedConnectionsResource - from ._models_py3 import AllowlistCustomAlertRule - from ._models_py3 import AmqpC2DMessagesNotInAllowedRange - from ._models_py3 import AmqpC2DRejectedMessagesNotInAllowedRange - from ._models_py3 import AmqpD2CMessagesNotInAllowedRange - from ._models_py3 import AscLocation - from ._models_py3 import AscLocationList - from ._models_py3 import AssessmentLinks - from ._models_py3 import AssessmentStatus - from ._models_py3 import AtaExternalSecuritySolution - from ._models_py3 import AtaSolutionProperties - from ._models_py3 import AuthenticationDetailsProperties - from ._models_py3 import AutoProvisioningSetting - from ._models_py3 import AutoProvisioningSettingList - from ._models_py3 import Automation - from ._models_py3 import AutomationAction - from ._models_py3 import AutomationActionEventHub - from ._models_py3 import AutomationActionLogicApp - from ._models_py3 import AutomationActionWorkspace - from ._models_py3 import AutomationList - from ._models_py3 import AutomationRuleSet - from ._models_py3 import AutomationScope - from ._models_py3 import AutomationSource - from ._models_py3 import AutomationTriggeringRule - from ._models_py3 import AutomationValidationStatus - from ._models_py3 import AwAssumeRoleAuthenticationDetailsProperties - from ._models_py3 import AwsCredsAuthenticationDetailsProperties - from ._models_py3 import AzureResourceDetails - from ._models_py3 import AzureResourceIdentifier - from ._models_py3 import AzureResourceLink - from ._models_py3 import AzureTrackedResourceLocation - from ._models_py3 import Baseline - from ._models_py3 import BaselineAdjustedResult - from ._models_py3 import BenchmarkReference - from ._models_py3 import CVE - from ._models_py3 import CVSS - from ._models_py3 import CefExternalSecuritySolution - from ._models_py3 import CefSolutionProperties - from ._models_py3 import CloudErrorBody - from ._models_py3 import Compliance - from ._models_py3 import ComplianceList - from ._models_py3 import ComplianceResult - from ._models_py3 import ComplianceResultList - from ._models_py3 import ComplianceSegment - from ._models_py3 import ConnectableResource - from ._models_py3 import ConnectedResource - from ._models_py3 import ConnectedWorkspace - from ._models_py3 import ConnectionFromIpNotAllowed - from ._models_py3 import ConnectionStrings - from ._models_py3 import ConnectionToIpNotAllowed - from ._models_py3 import ConnectorSetting - from ._models_py3 import ConnectorSettingList - from ._models_py3 import ContainerRegistryVulnerabilityProperties - from ._models_py3 import CustomAlertRule - from ._models_py3 import DataExportSettings - from ._models_py3 import DenylistCustomAlertRule - from ._models_py3 import DeviceSecurityGroup - from ._models_py3 import DeviceSecurityGroupList - from ._models_py3 import DirectMethodInvokesNotInAllowedRange - from ._models_py3 import DiscoveredSecuritySolution - from ._models_py3 import DiscoveredSecuritySolutionList - from ._models_py3 import ETag - from ._models_py3 import EffectiveNetworkSecurityGroups - from ._models_py3 import ErrorAdditionalInfo - from ._models_py3 import ExternalSecuritySolution - from ._models_py3 import ExternalSecuritySolutionKind - from ._models_py3 import ExternalSecuritySolutionList - from ._models_py3 import ExternalSecuritySolutionProperties - from ._models_py3 import FailedLocalLoginsNotInAllowedRange - from ._models_py3 import FileUploadsNotInAllowedRange - from ._models_py3 import GcpCredentialsDetailsProperties - from ._models_py3 import HttpC2DMessagesNotInAllowedRange - from ._models_py3 import HttpC2DRejectedMessagesNotInAllowedRange - from ._models_py3 import HttpD2CMessagesNotInAllowedRange - from ._models_py3 import HybridComputeSettingsProperties - from ._models_py3 import InformationProtectionKeyword - from ._models_py3 import InformationProtectionPolicy - from ._models_py3 import InformationProtectionPolicyList - from ._models_py3 import InformationType - from ._models_py3 import IngestionConnectionString - from ._models_py3 import IngestionSetting - from ._models_py3 import IngestionSettingList - from ._models_py3 import IngestionSettingToken - from ._models_py3 import IoTSecurityAggregatedAlert - from ._models_py3 import IoTSecurityAggregatedAlertList - from ._models_py3 import IoTSecurityAggregatedAlertPropertiesTopDevicesListItem - from ._models_py3 import IoTSecurityAggregatedRecommendation - from ._models_py3 import IoTSecurityAggregatedRecommendationList - from ._models_py3 import IoTSecurityAlertedDevice - from ._models_py3 import IoTSecurityDeviceAlert - from ._models_py3 import IoTSecurityDeviceRecommendation - from ._models_py3 import IoTSecuritySolutionAnalyticsModel - from ._models_py3 import IoTSecuritySolutionAnalyticsModelList - from ._models_py3 import IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem - from ._models_py3 import IoTSecuritySolutionModel - from ._models_py3 import IoTSecuritySolutionsList - from ._models_py3 import IoTSeverityMetrics - from ._models_py3 import JitNetworkAccessPoliciesList - from ._models_py3 import JitNetworkAccessPolicy - from ._models_py3 import JitNetworkAccessPolicyInitiatePort - from ._models_py3 import JitNetworkAccessPolicyInitiateRequest - from ._models_py3 import JitNetworkAccessPolicyInitiateVirtualMachine - from ._models_py3 import JitNetworkAccessPolicyVirtualMachine - from ._models_py3 import JitNetworkAccessPortRule - from ._models_py3 import JitNetworkAccessRequest - from ._models_py3 import JitNetworkAccessRequestPort - from ._models_py3 import JitNetworkAccessRequestVirtualMachine - from ._models_py3 import Kind - from ._models_py3 import ListCustomAlertRule - from ._models_py3 import LocalUserNotAllowed - from ._models_py3 import Location - from ._models_py3 import LogAnalyticsIdentifier - from ._models_py3 import MqttC2DMessagesNotInAllowedRange - from ._models_py3 import MqttC2DRejectedMessagesNotInAllowedRange - from ._models_py3 import MqttD2CMessagesNotInAllowedRange - from ._models_py3 import OnPremiseResourceDetails - from ._models_py3 import OnPremiseSqlResourceDetails - from ._models_py3 import Operation - from ._models_py3 import OperationDisplay - from ._models_py3 import OperationList - from ._models_py3 import PathRecommendation - from ._models_py3 import Pricing - from ._models_py3 import PricingList - from ._models_py3 import ProcessNotAllowed - from ._models_py3 import ProtectionMode - from ._models_py3 import ProxyServerProperties - from ._models_py3 import PublisherInfo - from ._models_py3 import QueryCheck - from ._models_py3 import QueuePurgesNotInAllowedRange - from ._models_py3 import RecommendationConfigurationProperties - from ._models_py3 import RegulatoryComplianceAssessment - from ._models_py3 import RegulatoryComplianceAssessmentList - from ._models_py3 import RegulatoryComplianceControl - from ._models_py3 import RegulatoryComplianceControlList - from ._models_py3 import RegulatoryComplianceStandard - from ._models_py3 import RegulatoryComplianceStandardList - from ._models_py3 import Remediation - from ._models_py3 import Resource - from ._models_py3 import ResourceDetails - from ._models_py3 import ResourceIdentifier - from ._models_py3 import Rule - from ._models_py3 import RuleResults - from ._models_py3 import RuleResultsInput - from ._models_py3 import RuleResultsProperties - from ._models_py3 import RulesResults - from ._models_py3 import RulesResultsInput - from ._models_py3 import Scan - from ._models_py3 import ScanProperties - from ._models_py3 import ScanResult - from ._models_py3 import ScanResultProperties - from ._models_py3 import ScanResults - from ._models_py3 import Scans - from ._models_py3 import ScopeElement - from ._models_py3 import SecureScoreControlDefinitionItem - from ._models_py3 import SecureScoreControlDefinitionList - from ._models_py3 import SecureScoreControlDefinitionSource - from ._models_py3 import SecureScoreControlDetails - from ._models_py3 import SecureScoreControlList - from ._models_py3 import SecureScoreControlScore - from ._models_py3 import SecureScoreItem - from ._models_py3 import SecureScoresList - from ._models_py3 import SecurityAssessment - from ._models_py3 import SecurityAssessmentList - from ._models_py3 import SecurityAssessmentMetadata - from ._models_py3 import SecurityAssessmentMetadataList - from ._models_py3 import SecurityAssessmentMetadataPartnerData - from ._models_py3 import SecurityAssessmentMetadataProperties - from ._models_py3 import SecurityAssessmentPartnerData - from ._models_py3 import SecurityContact - from ._models_py3 import SecurityContactList - from ._models_py3 import SecuritySolution - from ._models_py3 import SecuritySolutionList - from ._models_py3 import SecuritySolutionsReferenceData - from ._models_py3 import SecuritySolutionsReferenceDataList - from ._models_py3 import SecuritySubAssessment - from ._models_py3 import SecuritySubAssessmentList - from ._models_py3 import SecurityTask - from ._models_py3 import SecurityTaskList - from ._models_py3 import SecurityTaskParameters - from ._models_py3 import SensitivityLabel - from ._models_py3 import ServerVulnerabilityAssessment - from ._models_py3 import ServerVulnerabilityAssessmentsList - from ._models_py3 import ServerVulnerabilityProperties - from ._models_py3 import ServicePrincipalProperties - from ._models_py3 import Setting - from ._models_py3 import SettingsList - from ._models_py3 import Software - from ._models_py3 import SoftwaresList - from ._models_py3 import SqlServerVulnerabilityProperties - from ._models_py3 import SubAssessmentStatus - from ._models_py3 import SuppressionAlertsScope - from ._models_py3 import SystemData - from ._models_py3 import Tags - from ._models_py3 import TagsResource - from ._models_py3 import ThresholdCustomAlertRule - from ._models_py3 import TimeWindowCustomAlertRule - from ._models_py3 import TopologyList - from ._models_py3 import TopologyResource - from ._models_py3 import TopologySingleResource - from ._models_py3 import TopologySingleResourceChild - from ._models_py3 import TopologySingleResourceParent - from ._models_py3 import TrackedResource - from ._models_py3 import TwinUpdatesNotInAllowedRange - from ._models_py3 import UnauthorizedOperationsNotInAllowedRange - from ._models_py3 import UpdateIotSecuritySolutionData - from ._models_py3 import UserDefinedResourcesProperties - from ._models_py3 import UserRecommendation - from ._models_py3 import VaRule - from ._models_py3 import VendorReference - from ._models_py3 import VmRecommendation - from ._models_py3 import WorkspaceSetting - from ._models_py3 import WorkspaceSettingList -except (SyntaxError, ImportError): - from ._models import AadConnectivityState # type: ignore - from ._models import AadExternalSecuritySolution # type: ignore - from ._models import AadSolutionProperties # type: ignore - from ._models import ActiveConnectionsNotInAllowedRange # type: ignore - from ._models import AdaptiveApplicationControlGroup # type: ignore - from ._models import AdaptiveApplicationControlGroups # type: ignore - from ._models import AdaptiveApplicationControlIssueSummary # type: ignore - from ._models import AdaptiveNetworkHardening # type: ignore - from ._models import AdaptiveNetworkHardeningEnforceRequest # type: ignore - from ._models import AdaptiveNetworkHardeningsList # type: ignore - from ._models import AdditionalData # type: ignore - from ._models import AdditionalWorkspacesProperties # type: ignore - from ._models import AdvancedThreatProtectionSetting # type: ignore - from ._models import Alert # type: ignore - from ._models import AlertEntity # type: ignore - from ._models import AlertList # type: ignore - from ._models import AlertSimulatorBundlesRequestProperties # type: ignore - from ._models import AlertSimulatorRequestBody # type: ignore - from ._models import AlertSimulatorRequestProperties # type: ignore - from ._models import AlertSyncSettings # type: ignore - from ._models import AlertsSuppressionRule # type: ignore - from ._models import AlertsSuppressionRulesList # type: ignore - from ._models import AllowedConnectionsList # type: ignore - from ._models import AllowedConnectionsResource # type: ignore - from ._models import AllowlistCustomAlertRule # type: ignore - from ._models import AmqpC2DMessagesNotInAllowedRange # type: ignore - from ._models import AmqpC2DRejectedMessagesNotInAllowedRange # type: ignore - from ._models import AmqpD2CMessagesNotInAllowedRange # type: ignore - from ._models import AscLocation # type: ignore - from ._models import AscLocationList # type: ignore - from ._models import AssessmentLinks # type: ignore - from ._models import AssessmentStatus # type: ignore - from ._models import AtaExternalSecuritySolution # type: ignore - from ._models import AtaSolutionProperties # type: ignore - from ._models import AuthenticationDetailsProperties # type: ignore - from ._models import AutoProvisioningSetting # type: ignore - from ._models import AutoProvisioningSettingList # type: ignore - from ._models import Automation # type: ignore - from ._models import AutomationAction # type: ignore - from ._models import AutomationActionEventHub # type: ignore - from ._models import AutomationActionLogicApp # type: ignore - from ._models import AutomationActionWorkspace # type: ignore - from ._models import AutomationList # type: ignore - from ._models import AutomationRuleSet # type: ignore - from ._models import AutomationScope # type: ignore - from ._models import AutomationSource # type: ignore - from ._models import AutomationTriggeringRule # type: ignore - from ._models import AutomationValidationStatus # type: ignore - from ._models import AwAssumeRoleAuthenticationDetailsProperties # type: ignore - from ._models import AwsCredsAuthenticationDetailsProperties # type: ignore - from ._models import AzureResourceDetails # type: ignore - from ._models import AzureResourceIdentifier # type: ignore - from ._models import AzureResourceLink # type: ignore - from ._models import AzureTrackedResourceLocation # type: ignore - from ._models import Baseline # type: ignore - from ._models import BaselineAdjustedResult # type: ignore - from ._models import BenchmarkReference # type: ignore - from ._models import CVE # type: ignore - from ._models import CVSS # type: ignore - from ._models import CefExternalSecuritySolution # type: ignore - from ._models import CefSolutionProperties # type: ignore - from ._models import CloudErrorBody # type: ignore - from ._models import Compliance # type: ignore - from ._models import ComplianceList # type: ignore - from ._models import ComplianceResult # type: ignore - from ._models import ComplianceResultList # type: ignore - from ._models import ComplianceSegment # type: ignore - from ._models import ConnectableResource # type: ignore - from ._models import ConnectedResource # type: ignore - from ._models import ConnectedWorkspace # type: ignore - from ._models import ConnectionFromIpNotAllowed # type: ignore - from ._models import ConnectionStrings # type: ignore - from ._models import ConnectionToIpNotAllowed # type: ignore - from ._models import ConnectorSetting # type: ignore - from ._models import ConnectorSettingList # type: ignore - from ._models import ContainerRegistryVulnerabilityProperties # type: ignore - from ._models import CustomAlertRule # type: ignore - from ._models import DataExportSettings # type: ignore - from ._models import DenylistCustomAlertRule # type: ignore - from ._models import DeviceSecurityGroup # type: ignore - from ._models import DeviceSecurityGroupList # type: ignore - from ._models import DirectMethodInvokesNotInAllowedRange # type: ignore - from ._models import DiscoveredSecuritySolution # type: ignore - from ._models import DiscoveredSecuritySolutionList # type: ignore - from ._models import ETag # type: ignore - from ._models import EffectiveNetworkSecurityGroups # type: ignore - from ._models import ErrorAdditionalInfo # type: ignore - from ._models import ExternalSecuritySolution # type: ignore - from ._models import ExternalSecuritySolutionKind # type: ignore - from ._models import ExternalSecuritySolutionList # type: ignore - from ._models import ExternalSecuritySolutionProperties # type: ignore - from ._models import FailedLocalLoginsNotInAllowedRange # type: ignore - from ._models import FileUploadsNotInAllowedRange # type: ignore - from ._models import GcpCredentialsDetailsProperties # type: ignore - from ._models import HttpC2DMessagesNotInAllowedRange # type: ignore - from ._models import HttpC2DRejectedMessagesNotInAllowedRange # type: ignore - from ._models import HttpD2CMessagesNotInAllowedRange # type: ignore - from ._models import HybridComputeSettingsProperties # type: ignore - from ._models import InformationProtectionKeyword # type: ignore - from ._models import InformationProtectionPolicy # type: ignore - from ._models import InformationProtectionPolicyList # type: ignore - from ._models import InformationType # type: ignore - from ._models import IngestionConnectionString # type: ignore - from ._models import IngestionSetting # type: ignore - from ._models import IngestionSettingList # type: ignore - from ._models import IngestionSettingToken # type: ignore - from ._models import IoTSecurityAggregatedAlert # type: ignore - from ._models import IoTSecurityAggregatedAlertList # type: ignore - from ._models import IoTSecurityAggregatedAlertPropertiesTopDevicesListItem # type: ignore - from ._models import IoTSecurityAggregatedRecommendation # type: ignore - from ._models import IoTSecurityAggregatedRecommendationList # type: ignore - from ._models import IoTSecurityAlertedDevice # type: ignore - from ._models import IoTSecurityDeviceAlert # type: ignore - from ._models import IoTSecurityDeviceRecommendation # type: ignore - from ._models import IoTSecuritySolutionAnalyticsModel # type: ignore - from ._models import IoTSecuritySolutionAnalyticsModelList # type: ignore - from ._models import IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem # type: ignore - from ._models import IoTSecuritySolutionModel # type: ignore - from ._models import IoTSecuritySolutionsList # type: ignore - from ._models import IoTSeverityMetrics # type: ignore - from ._models import JitNetworkAccessPoliciesList # type: ignore - from ._models import JitNetworkAccessPolicy # type: ignore - from ._models import JitNetworkAccessPolicyInitiatePort # type: ignore - from ._models import JitNetworkAccessPolicyInitiateRequest # type: ignore - from ._models import JitNetworkAccessPolicyInitiateVirtualMachine # type: ignore - from ._models import JitNetworkAccessPolicyVirtualMachine # type: ignore - from ._models import JitNetworkAccessPortRule # type: ignore - from ._models import JitNetworkAccessRequest # type: ignore - from ._models import JitNetworkAccessRequestPort # type: ignore - from ._models import JitNetworkAccessRequestVirtualMachine # type: ignore - from ._models import Kind # type: ignore - from ._models import ListCustomAlertRule # type: ignore - from ._models import LocalUserNotAllowed # type: ignore - from ._models import Location # type: ignore - from ._models import LogAnalyticsIdentifier # type: ignore - from ._models import MqttC2DMessagesNotInAllowedRange # type: ignore - from ._models import MqttC2DRejectedMessagesNotInAllowedRange # type: ignore - from ._models import MqttD2CMessagesNotInAllowedRange # type: ignore - from ._models import OnPremiseResourceDetails # type: ignore - from ._models import OnPremiseSqlResourceDetails # type: ignore - from ._models import Operation # type: ignore - from ._models import OperationDisplay # type: ignore - from ._models import OperationList # type: ignore - from ._models import PathRecommendation # type: ignore - from ._models import Pricing # type: ignore - from ._models import PricingList # type: ignore - from ._models import ProcessNotAllowed # type: ignore - from ._models import ProtectionMode # type: ignore - from ._models import ProxyServerProperties # type: ignore - from ._models import PublisherInfo # type: ignore - from ._models import QueryCheck # type: ignore - from ._models import QueuePurgesNotInAllowedRange # type: ignore - from ._models import RecommendationConfigurationProperties # type: ignore - from ._models import RegulatoryComplianceAssessment # type: ignore - from ._models import RegulatoryComplianceAssessmentList # type: ignore - from ._models import RegulatoryComplianceControl # type: ignore - from ._models import RegulatoryComplianceControlList # type: ignore - from ._models import RegulatoryComplianceStandard # type: ignore - from ._models import RegulatoryComplianceStandardList # type: ignore - from ._models import Remediation # type: ignore - from ._models import Resource # type: ignore - from ._models import ResourceDetails # type: ignore - from ._models import ResourceIdentifier # type: ignore - from ._models import Rule # type: ignore - from ._models import RuleResults # type: ignore - from ._models import RuleResultsInput # type: ignore - from ._models import RuleResultsProperties # type: ignore - from ._models import RulesResults # type: ignore - from ._models import RulesResultsInput # type: ignore - from ._models import Scan # type: ignore - from ._models import ScanProperties # type: ignore - from ._models import ScanResult # type: ignore - from ._models import ScanResultProperties # type: ignore - from ._models import ScanResults # type: ignore - from ._models import Scans # type: ignore - from ._models import ScopeElement # type: ignore - from ._models import SecureScoreControlDefinitionItem # type: ignore - from ._models import SecureScoreControlDefinitionList # type: ignore - from ._models import SecureScoreControlDefinitionSource # type: ignore - from ._models import SecureScoreControlDetails # type: ignore - from ._models import SecureScoreControlList # type: ignore - from ._models import SecureScoreControlScore # type: ignore - from ._models import SecureScoreItem # type: ignore - from ._models import SecureScoresList # type: ignore - from ._models import SecurityAssessment # type: ignore - from ._models import SecurityAssessmentList # type: ignore - from ._models import SecurityAssessmentMetadata # type: ignore - from ._models import SecurityAssessmentMetadataList # type: ignore - from ._models import SecurityAssessmentMetadataPartnerData # type: ignore - from ._models import SecurityAssessmentMetadataProperties # type: ignore - from ._models import SecurityAssessmentPartnerData # type: ignore - from ._models import SecurityContact # type: ignore - from ._models import SecurityContactList # type: ignore - from ._models import SecuritySolution # type: ignore - from ._models import SecuritySolutionList # type: ignore - from ._models import SecuritySolutionsReferenceData # type: ignore - from ._models import SecuritySolutionsReferenceDataList # type: ignore - from ._models import SecuritySubAssessment # type: ignore - from ._models import SecuritySubAssessmentList # type: ignore - from ._models import SecurityTask # type: ignore - from ._models import SecurityTaskList # type: ignore - from ._models import SecurityTaskParameters # type: ignore - from ._models import SensitivityLabel # type: ignore - from ._models import ServerVulnerabilityAssessment # type: ignore - from ._models import ServerVulnerabilityAssessmentsList # type: ignore - from ._models import ServerVulnerabilityProperties # type: ignore - from ._models import ServicePrincipalProperties # type: ignore - from ._models import Setting # type: ignore - from ._models import SettingsList # type: ignore - from ._models import Software # type: ignore - from ._models import SoftwaresList # type: ignore - from ._models import SqlServerVulnerabilityProperties # type: ignore - from ._models import SubAssessmentStatus # type: ignore - from ._models import SuppressionAlertsScope # type: ignore - from ._models import SystemData # type: ignore - from ._models import Tags # type: ignore - from ._models import TagsResource # type: ignore - from ._models import ThresholdCustomAlertRule # type: ignore - from ._models import TimeWindowCustomAlertRule # type: ignore - from ._models import TopologyList # type: ignore - from ._models import TopologyResource # type: ignore - from ._models import TopologySingleResource # type: ignore - from ._models import TopologySingleResourceChild # type: ignore - from ._models import TopologySingleResourceParent # type: ignore - from ._models import TrackedResource # type: ignore - from ._models import TwinUpdatesNotInAllowedRange # type: ignore - from ._models import UnauthorizedOperationsNotInAllowedRange # type: ignore - from ._models import UpdateIotSecuritySolutionData # type: ignore - from ._models import UserDefinedResourcesProperties # type: ignore - from ._models import UserRecommendation # type: ignore - from ._models import VaRule # type: ignore - from ._models import VendorReference # type: ignore - from ._models import VmRecommendation # type: ignore - from ._models import WorkspaceSetting # type: ignore - from ._models import WorkspaceSettingList # type: ignore +from ._models_py3 import AadConnectivityState +from ._models_py3 import AadExternalSecuritySolution +from ._models_py3 import AadSolutionProperties +from ._models_py3 import ActiveConnectionsNotInAllowedRange +from ._models_py3 import AdaptiveApplicationControlGroup +from ._models_py3 import AdaptiveApplicationControlGroups +from ._models_py3 import AdaptiveApplicationControlIssueSummary +from ._models_py3 import AdaptiveNetworkHardening +from ._models_py3 import AdaptiveNetworkHardeningEnforceRequest +from ._models_py3 import AdaptiveNetworkHardeningsList +from ._models_py3 import AdditionalData +from ._models_py3 import AdditionalWorkspacesProperties +from ._models_py3 import AdvancedThreatProtectionSetting +from ._models_py3 import Alert +from ._models_py3 import AlertEntity +from ._models_py3 import AlertList +from ._models_py3 import AlertSimulatorBundlesRequestProperties +from ._models_py3 import AlertSimulatorRequestBody +from ._models_py3 import AlertSimulatorRequestProperties +from ._models_py3 import AlertSyncSettings +from ._models_py3 import AlertsSuppressionRule +from ._models_py3 import AlertsSuppressionRulesList +from ._models_py3 import AllowedConnectionsList +from ._models_py3 import AllowedConnectionsResource +from ._models_py3 import AllowlistCustomAlertRule +from ._models_py3 import AmqpC2DMessagesNotInAllowedRange +from ._models_py3 import AmqpC2DRejectedMessagesNotInAllowedRange +from ._models_py3 import AmqpD2CMessagesNotInAllowedRange +from ._models_py3 import AscLocation +from ._models_py3 import AscLocationList +from ._models_py3 import AssessmentLinks +from ._models_py3 import AssessmentStatus +from ._models_py3 import AssessmentStatusResponse +from ._models_py3 import AtaExternalSecuritySolution +from ._models_py3 import AtaSolutionProperties +from ._models_py3 import AuthenticationDetailsProperties +from ._models_py3 import AutoProvisioningSetting +from ._models_py3 import AutoProvisioningSettingList +from ._models_py3 import Automation +from ._models_py3 import AutomationAction +from ._models_py3 import AutomationActionEventHub +from ._models_py3 import AutomationActionLogicApp +from ._models_py3 import AutomationActionWorkspace +from ._models_py3 import AutomationList +from ._models_py3 import AutomationRuleSet +from ._models_py3 import AutomationScope +from ._models_py3 import AutomationSource +from ._models_py3 import AutomationTriggeringRule +from ._models_py3 import AutomationValidationStatus +from ._models_py3 import AwAssumeRoleAuthenticationDetailsProperties +from ._models_py3 import AwsCredsAuthenticationDetailsProperties +from ._models_py3 import AzureResourceDetails +from ._models_py3 import AzureResourceIdentifier +from ._models_py3 import AzureResourceLink +from ._models_py3 import AzureTrackedResourceLocation +from ._models_py3 import Baseline +from ._models_py3 import BaselineAdjustedResult +from ._models_py3 import BenchmarkReference +from ._models_py3 import CVE +from ._models_py3 import CVSS +from ._models_py3 import CefExternalSecuritySolution +from ._models_py3 import CefSolutionProperties +from ._models_py3 import CloudErrorBody +from ._models_py3 import CloudOffering +from ._models_py3 import Compliance +from ._models_py3 import ComplianceList +from ._models_py3 import ComplianceResult +from ._models_py3 import ComplianceResultList +from ._models_py3 import ComplianceSegment +from ._models_py3 import ConnectableResource +from ._models_py3 import ConnectedResource +from ._models_py3 import ConnectedWorkspace +from ._models_py3 import ConnectionFromIpNotAllowed +from ._models_py3 import ConnectionStrings +from ._models_py3 import ConnectionToIpNotAllowed +from ._models_py3 import ConnectorSetting +from ._models_py3 import ConnectorSettingList +from ._models_py3 import ContainerRegistryVulnerabilityProperties +from ._models_py3 import CspmMonitorAwsOffering +from ._models_py3 import CspmMonitorAwsOfferingNativeCloudConnection +from ._models_py3 import CustomAlertRule +from ._models_py3 import CustomAssessmentAutomation +from ._models_py3 import CustomAssessmentAutomationRequest +from ._models_py3 import CustomAssessmentAutomationsListResult +from ._models_py3 import CustomEntityStoreAssignment +from ._models_py3 import CustomEntityStoreAssignmentRequest +from ._models_py3 import CustomEntityStoreAssignmentsListResult +from ._models_py3 import DataExportSettings +from ._models_py3 import DefenderForContainersAwsOffering +from ._models_py3 import DefenderForContainersAwsOfferingCloudWatchToKinesis +from ._models_py3 import DefenderForContainersAwsOfferingKinesisToS3 +from ._models_py3 import DefenderForContainersAwsOfferingKubernetesScubaReader +from ._models_py3 import DefenderForContainersAwsOfferingKubernetesService +from ._models_py3 import DefenderForServersAwsOffering +from ._models_py3 import DefenderForServersAwsOfferingArcAutoProvisioning +from ._models_py3 import DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata +from ._models_py3 import DefenderForServersAwsOfferingDefenderForServers +from ._models_py3 import DenylistCustomAlertRule +from ._models_py3 import DeviceSecurityGroup +from ._models_py3 import DeviceSecurityGroupList +from ._models_py3 import DirectMethodInvokesNotInAllowedRange +from ._models_py3 import DiscoveredSecuritySolution +from ._models_py3 import DiscoveredSecuritySolutionList +from ._models_py3 import ETag +from ._models_py3 import EffectiveNetworkSecurityGroups +from ._models_py3 import ErrorAdditionalInfo +from ._models_py3 import ExternalSecuritySolution +from ._models_py3 import ExternalSecuritySolutionKind +from ._models_py3 import ExternalSecuritySolutionList +from ._models_py3 import ExternalSecuritySolutionProperties +from ._models_py3 import FailedLocalLoginsNotInAllowedRange +from ._models_py3 import FileUploadsNotInAllowedRange +from ._models_py3 import GcpCredentialsDetailsProperties +from ._models_py3 import HttpC2DMessagesNotInAllowedRange +from ._models_py3 import HttpC2DRejectedMessagesNotInAllowedRange +from ._models_py3 import HttpD2CMessagesNotInAllowedRange +from ._models_py3 import HybridComputeSettingsProperties +from ._models_py3 import InformationProtectionKeyword +from ._models_py3 import InformationProtectionPolicy +from ._models_py3 import InformationProtectionPolicyList +from ._models_py3 import InformationType +from ._models_py3 import IngestionConnectionString +from ._models_py3 import IngestionSetting +from ._models_py3 import IngestionSettingList +from ._models_py3 import IngestionSettingToken +from ._models_py3 import IoTSecurityAggregatedAlert +from ._models_py3 import IoTSecurityAggregatedAlertList +from ._models_py3 import IoTSecurityAggregatedAlertPropertiesTopDevicesListItem +from ._models_py3 import IoTSecurityAggregatedRecommendation +from ._models_py3 import IoTSecurityAggregatedRecommendationList +from ._models_py3 import IoTSecurityAlertedDevice +from ._models_py3 import IoTSecurityDeviceAlert +from ._models_py3 import IoTSecurityDeviceRecommendation +from ._models_py3 import IoTSecuritySolutionAnalyticsModel +from ._models_py3 import IoTSecuritySolutionAnalyticsModelList +from ._models_py3 import IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem +from ._models_py3 import IoTSecuritySolutionModel +from ._models_py3 import IoTSecuritySolutionsList +from ._models_py3 import IoTSeverityMetrics +from ._models_py3 import JitNetworkAccessPoliciesList +from ._models_py3 import JitNetworkAccessPolicy +from ._models_py3 import JitNetworkAccessPolicyInitiatePort +from ._models_py3 import JitNetworkAccessPolicyInitiateRequest +from ._models_py3 import JitNetworkAccessPolicyInitiateVirtualMachine +from ._models_py3 import JitNetworkAccessPolicyVirtualMachine +from ._models_py3 import JitNetworkAccessPortRule +from ._models_py3 import JitNetworkAccessRequest +from ._models_py3 import JitNetworkAccessRequestPort +from ._models_py3 import JitNetworkAccessRequestVirtualMachine +from ._models_py3 import Kind +from ._models_py3 import ListCustomAlertRule +from ._models_py3 import LocalUserNotAllowed +from ._models_py3 import Location +from ._models_py3 import LogAnalyticsIdentifier +from ._models_py3 import MdeOnboardingData +from ._models_py3 import MdeOnboardingDataList +from ._models_py3 import MqttC2DMessagesNotInAllowedRange +from ._models_py3 import MqttC2DRejectedMessagesNotInAllowedRange +from ._models_py3 import MqttD2CMessagesNotInAllowedRange +from ._models_py3 import OnPremiseResourceDetails +from ._models_py3 import OnPremiseSqlResourceDetails +from ._models_py3 import Operation +from ._models_py3 import OperationDisplay +from ._models_py3 import OperationList +from ._models_py3 import PathRecommendation +from ._models_py3 import Pricing +from ._models_py3 import PricingList +from ._models_py3 import ProcessNotAllowed +from ._models_py3 import ProtectionMode +from ._models_py3 import ProxyServerProperties +from ._models_py3 import PublisherInfo +from ._models_py3 import QueryCheck +from ._models_py3 import QueuePurgesNotInAllowedRange +from ._models_py3 import RecommendationConfigurationProperties +from ._models_py3 import RegulatoryComplianceAssessment +from ._models_py3 import RegulatoryComplianceAssessmentList +from ._models_py3 import RegulatoryComplianceControl +from ._models_py3 import RegulatoryComplianceControlList +from ._models_py3 import RegulatoryComplianceStandard +from ._models_py3 import RegulatoryComplianceStandardList +from ._models_py3 import Remediation +from ._models_py3 import Resource +from ._models_py3 import ResourceDetails +from ._models_py3 import ResourceIdentifier +from ._models_py3 import Rule +from ._models_py3 import RuleResults +from ._models_py3 import RuleResultsInput +from ._models_py3 import RuleResultsProperties +from ._models_py3 import RulesResults +from ._models_py3 import RulesResultsInput +from ._models_py3 import Scan +from ._models_py3 import ScanProperties +from ._models_py3 import ScanResult +from ._models_py3 import ScanResultProperties +from ._models_py3 import ScanResults +from ._models_py3 import Scans +from ._models_py3 import ScopeElement +from ._models_py3 import SecureScoreControlDefinitionItem +from ._models_py3 import SecureScoreControlDefinitionList +from ._models_py3 import SecureScoreControlDefinitionSource +from ._models_py3 import SecureScoreControlDetails +from ._models_py3 import SecureScoreControlList +from ._models_py3 import SecureScoreControlScore +from ._models_py3 import SecureScoreItem +from ._models_py3 import SecureScoresList +from ._models_py3 import SecurityAssessment +from ._models_py3 import SecurityAssessmentList +from ._models_py3 import SecurityAssessmentMetadata +from ._models_py3 import SecurityAssessmentMetadataPartnerData +from ._models_py3 import SecurityAssessmentMetadataProperties +from ._models_py3 import SecurityAssessmentMetadataPropertiesResponse +from ._models_py3 import SecurityAssessmentMetadataPropertiesResponsePublishDates +from ._models_py3 import SecurityAssessmentMetadataResponse +from ._models_py3 import SecurityAssessmentMetadataResponseList +from ._models_py3 import SecurityAssessmentPartnerData +from ._models_py3 import SecurityAssessmentProperties +from ._models_py3 import SecurityAssessmentPropertiesBase +from ._models_py3 import SecurityAssessmentPropertiesResponse +from ._models_py3 import SecurityAssessmentResponse +from ._models_py3 import SecurityConnector +from ._models_py3 import SecurityConnectorPropertiesOrganizationalData +from ._models_py3 import SecurityConnectorsList +from ._models_py3 import SecurityContact +from ._models_py3 import SecurityContactList +from ._models_py3 import SecuritySolution +from ._models_py3 import SecuritySolutionList +from ._models_py3 import SecuritySolutionsReferenceData +from ._models_py3 import SecuritySolutionsReferenceDataList +from ._models_py3 import SecuritySubAssessment +from ._models_py3 import SecuritySubAssessmentList +from ._models_py3 import SecurityTask +from ._models_py3 import SecurityTaskList +from ._models_py3 import SecurityTaskParameters +from ._models_py3 import SensitivityLabel +from ._models_py3 import ServerVulnerabilityAssessment +from ._models_py3 import ServerVulnerabilityAssessmentsList +from ._models_py3 import ServerVulnerabilityProperties +from ._models_py3 import ServicePrincipalProperties +from ._models_py3 import Setting +from ._models_py3 import SettingsList +from ._models_py3 import Software +from ._models_py3 import SoftwaresList +from ._models_py3 import SqlServerVulnerabilityProperties +from ._models_py3 import SubAssessmentStatus +from ._models_py3 import SuppressionAlertsScope +from ._models_py3 import SystemData +from ._models_py3 import Tags +from ._models_py3 import TagsResource +from ._models_py3 import ThresholdCustomAlertRule +from ._models_py3 import TimeWindowCustomAlertRule +from ._models_py3 import TopologyList +from ._models_py3 import TopologyResource +from ._models_py3 import TopologySingleResource +from ._models_py3 import TopologySingleResourceChild +from ._models_py3 import TopologySingleResourceParent +from ._models_py3 import TrackedResource +from ._models_py3 import TwinUpdatesNotInAllowedRange +from ._models_py3 import UnauthorizedOperationsNotInAllowedRange +from ._models_py3 import UpdateIotSecuritySolutionData +from ._models_py3 import UserDefinedResourcesProperties +from ._models_py3 import UserRecommendation +from ._models_py3 import VaRule +from ._models_py3 import VendorReference +from ._models_py3 import VmRecommendation +from ._models_py3 import WorkspaceSetting +from ._models_py3 import WorkspaceSettingList + from ._security_center_enums import ( AadConnectivityStateEnum, @@ -497,6 +292,7 @@ AutoProvision, BundleType, Categories, + CloudName, ConfigurationStatus, ConnectionType, ControlType, @@ -506,9 +302,9 @@ EndOfSupportStatus, EnforcementMode, EnforcementSupport, - Enum13, Enum15, - Enum69, + Enum17, + Enum73, EventSource, ExpandControlsEnum, ExpandEnum, @@ -519,7 +315,9 @@ ImplementationEffort, Intent, KindEnum, + OfferingType, Operator, + OrganizationMembershipType, PermissionProperty, PricingTier, PropertyType, @@ -544,12 +342,16 @@ ServerVulnerabilityAssessmentPropertiesProvisioningState, SettingKind, Severity, + SeverityEnum, Source, SourceSystem, State, Status, StatusReason, SubAssessmentStatusCode, + SupportedCloudEnum, + Tactics, + Techniques, Threats, TransportProtocol, UnmaskedIpLoggingStatus, @@ -590,6 +392,7 @@ 'AscLocationList', 'AssessmentLinks', 'AssessmentStatus', + 'AssessmentStatusResponse', 'AtaExternalSecuritySolution', 'AtaSolutionProperties', 'AuthenticationDetailsProperties', @@ -620,6 +423,7 @@ 'CefExternalSecuritySolution', 'CefSolutionProperties', 'CloudErrorBody', + 'CloudOffering', 'Compliance', 'ComplianceList', 'ComplianceResult', @@ -634,8 +438,25 @@ 'ConnectorSetting', 'ConnectorSettingList', 'ContainerRegistryVulnerabilityProperties', + 'CspmMonitorAwsOffering', + 'CspmMonitorAwsOfferingNativeCloudConnection', 'CustomAlertRule', + 'CustomAssessmentAutomation', + 'CustomAssessmentAutomationRequest', + 'CustomAssessmentAutomationsListResult', + 'CustomEntityStoreAssignment', + 'CustomEntityStoreAssignmentRequest', + 'CustomEntityStoreAssignmentsListResult', 'DataExportSettings', + 'DefenderForContainersAwsOffering', + 'DefenderForContainersAwsOfferingCloudWatchToKinesis', + 'DefenderForContainersAwsOfferingKinesisToS3', + 'DefenderForContainersAwsOfferingKubernetesScubaReader', + 'DefenderForContainersAwsOfferingKubernetesService', + 'DefenderForServersAwsOffering', + 'DefenderForServersAwsOfferingArcAutoProvisioning', + 'DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata', + 'DefenderForServersAwsOfferingDefenderForServers', 'DenylistCustomAlertRule', 'DeviceSecurityGroup', 'DeviceSecurityGroupList', @@ -693,6 +514,8 @@ 'LocalUserNotAllowed', 'Location', 'LogAnalyticsIdentifier', + 'MdeOnboardingData', + 'MdeOnboardingDataList', 'MqttC2DMessagesNotInAllowedRange', 'MqttC2DRejectedMessagesNotInAllowedRange', 'MqttD2CMessagesNotInAllowedRange', @@ -745,10 +568,20 @@ 'SecurityAssessment', 'SecurityAssessmentList', 'SecurityAssessmentMetadata', - 'SecurityAssessmentMetadataList', 'SecurityAssessmentMetadataPartnerData', 'SecurityAssessmentMetadataProperties', + 'SecurityAssessmentMetadataPropertiesResponse', + 'SecurityAssessmentMetadataPropertiesResponsePublishDates', + 'SecurityAssessmentMetadataResponse', + 'SecurityAssessmentMetadataResponseList', 'SecurityAssessmentPartnerData', + 'SecurityAssessmentProperties', + 'SecurityAssessmentPropertiesBase', + 'SecurityAssessmentPropertiesResponse', + 'SecurityAssessmentResponse', + 'SecurityConnector', + 'SecurityConnectorPropertiesOrganizationalData', + 'SecurityConnectorsList', 'SecurityContact', 'SecurityContactList', 'SecuritySolution', @@ -810,6 +643,7 @@ 'AutoProvision', 'BundleType', 'Categories', + 'CloudName', 'ConfigurationStatus', 'ConnectionType', 'ControlType', @@ -819,9 +653,9 @@ 'EndOfSupportStatus', 'EnforcementMode', 'EnforcementSupport', - 'Enum13', 'Enum15', - 'Enum69', + 'Enum17', + 'Enum73', 'EventSource', 'ExpandControlsEnum', 'ExpandEnum', @@ -832,7 +666,9 @@ 'ImplementationEffort', 'Intent', 'KindEnum', + 'OfferingType', 'Operator', + 'OrganizationMembershipType', 'PermissionProperty', 'PricingTier', 'PropertyType', @@ -857,12 +693,16 @@ 'ServerVulnerabilityAssessmentPropertiesProvisioningState', 'SettingKind', 'Severity', + 'SeverityEnum', 'Source', 'SourceSystem', 'State', 'Status', 'StatusReason', 'SubAssessmentStatusCode', + 'SupportedCloudEnum', + 'Tactics', + 'Techniques', 'Threats', 'TransportProtocol', 'UnmaskedIpLoggingStatus', diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models.py deleted file mode 100644 index 13b2d159b9d31..0000000000000 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models.py +++ /dev/null @@ -1,9600 +0,0 @@ -# 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. -# -------------------------------------------------------------------------- - -import msrest.serialization - - -class AadConnectivityState(msrest.serialization.Model): - """Describes an Azure resource with kind. - - :param connectivity_state: The connectivity state of the external AAD solution. Possible values - include: "Discovered", "NotLicensed", "Connected". - :type connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum - """ - - _attribute_map = { - 'connectivity_state': {'key': 'connectivityState', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AadConnectivityState, self).__init__(**kwargs) - self.connectivity_state = kwargs.get('connectivity_state', None) - - -class ExternalSecuritySolutionKind(msrest.serialization.Model): - """Describes an Azure resource with kind. - - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum - """ - - _attribute_map = { - 'kind': {'key': 'kind', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ExternalSecuritySolutionKind, self).__init__(**kwargs) - self.kind = kwargs.get('kind', None) - - -class Location(msrest.serialization.Model): - """Describes an Azure resource with location. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar location: Location where the resource is stored. - :vartype location: str - """ - - _validation = { - 'location': {'readonly': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(Location, self).__init__(**kwargs) - self.location = None - - -class Resource(msrest.serialization.Model): - """Describes an Azure resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(Resource, self).__init__(**kwargs) - self.id = None - self.name = None - self.type = None - - -class ExternalSecuritySolution(Resource, Location, ExternalSecuritySolutionKind): - """Represents a security solution external to Azure Security Center which sends information to an OMS workspace and whose data is displayed by Azure Security Center. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'kind': {'key': 'kind', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ExternalSecuritySolution, self).__init__(**kwargs) - self.kind = kwargs.get('kind', None) - self.location = None - self.kind = kwargs.get('kind', None) - self.id = None - self.name = None - self.type = None - self.location = None - self.id = None - self.name = None - self.type = None - - -class AadExternalSecuritySolution(ExternalSecuritySolution): - """Represents an AAD identity protection solution which sends logs to an OMS workspace. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: The external security solution properties for AAD solutions. - :type properties: ~azure.mgmt.security.models.AadSolutionProperties - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'kind': {'key': 'kind', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'AadSolutionProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(AadExternalSecuritySolution, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class ExternalSecuritySolutionProperties(msrest.serialization.Model): - """The solution properties (correspond to the solution kind). - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace - """ - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'device_vendor': {'key': 'deviceVendor', 'type': 'str'}, - 'device_type': {'key': 'deviceType', 'type': 'str'}, - 'workspace': {'key': 'workspace', 'type': 'ConnectedWorkspace'}, - } - - def __init__( - self, - **kwargs - ): - super(ExternalSecuritySolutionProperties, self).__init__(**kwargs) - self.additional_properties = kwargs.get('additional_properties', None) - self.device_vendor = kwargs.get('device_vendor', None) - self.device_type = kwargs.get('device_type', None) - self.workspace = kwargs.get('workspace', None) - - -class AadSolutionProperties(ExternalSecuritySolutionProperties, AadConnectivityState): - """The external security solution properties for AAD solutions. - - :param connectivity_state: The connectivity state of the external AAD solution. Possible values - include: "Discovered", "NotLicensed", "Connected". - :type connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace - """ - - _attribute_map = { - 'connectivity_state': {'key': 'connectivityState', 'type': 'str'}, - 'additional_properties': {'key': '', 'type': '{object}'}, - 'device_vendor': {'key': 'deviceVendor', 'type': 'str'}, - 'device_type': {'key': 'deviceType', 'type': 'str'}, - 'workspace': {'key': 'workspace', 'type': 'ConnectedWorkspace'}, - } - - def __init__( - self, - **kwargs - ): - super(AadSolutionProperties, self).__init__(**kwargs) - self.connectivity_state = kwargs.get('connectivity_state', None) - self.additional_properties = kwargs.get('additional_properties', None) - self.device_vendor = kwargs.get('device_vendor', None) - self.device_type = kwargs.get('device_type', None) - self.workspace = kwargs.get('workspace', None) - - -class CustomAlertRule(msrest.serialization.Model): - """A custom alert rule. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: ListCustomAlertRule, ThresholdCustomAlertRule. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - } - - _subtype_map = { - 'rule_type': {'ListCustomAlertRule': 'ListCustomAlertRule', 'ThresholdCustomAlertRule': 'ThresholdCustomAlertRule'} - } - - def __init__( - self, - **kwargs - ): - super(CustomAlertRule, self).__init__(**kwargs) - self.display_name = None - self.description = None - self.is_enabled = kwargs['is_enabled'] - self.rule_type = None # type: Optional[str] - - -class ThresholdCustomAlertRule(CustomAlertRule): - """A custom alert rule that checks if a value (depends on the custom alert type) is within the given range. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: TimeWindowCustomAlertRule. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - } - - _subtype_map = { - 'rule_type': {'TimeWindowCustomAlertRule': 'TimeWindowCustomAlertRule'} - } - - def __init__( - self, - **kwargs - ): - super(ThresholdCustomAlertRule, self).__init__(**kwargs) - self.rule_type = 'ThresholdCustomAlertRule' # type: str - self.min_threshold = kwargs['min_threshold'] - self.max_threshold = kwargs['max_threshold'] - - -class TimeWindowCustomAlertRule(ThresholdCustomAlertRule): - """A custom alert rule that checks if the number of activities (depends on the custom alert type) in a time window is within the given range. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: ActiveConnectionsNotInAllowedRange, AmqpC2DMessagesNotInAllowedRange, AmqpC2DRejectedMessagesNotInAllowedRange, AmqpD2CMessagesNotInAllowedRange, DirectMethodInvokesNotInAllowedRange, FailedLocalLoginsNotInAllowedRange, FileUploadsNotInAllowedRange, HttpC2DMessagesNotInAllowedRange, HttpC2DRejectedMessagesNotInAllowedRange, HttpD2CMessagesNotInAllowedRange, MqttC2DMessagesNotInAllowedRange, MqttC2DRejectedMessagesNotInAllowedRange, MqttD2CMessagesNotInAllowedRange, QueuePurgesNotInAllowedRange, TwinUpdatesNotInAllowedRange, UnauthorizedOperationsNotInAllowedRange. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - _subtype_map = { - 'rule_type': {'ActiveConnectionsNotInAllowedRange': 'ActiveConnectionsNotInAllowedRange', 'AmqpC2DMessagesNotInAllowedRange': 'AmqpC2DMessagesNotInAllowedRange', 'AmqpC2DRejectedMessagesNotInAllowedRange': 'AmqpC2DRejectedMessagesNotInAllowedRange', 'AmqpD2CMessagesNotInAllowedRange': 'AmqpD2CMessagesNotInAllowedRange', 'DirectMethodInvokesNotInAllowedRange': 'DirectMethodInvokesNotInAllowedRange', 'FailedLocalLoginsNotInAllowedRange': 'FailedLocalLoginsNotInAllowedRange', 'FileUploadsNotInAllowedRange': 'FileUploadsNotInAllowedRange', 'HttpC2DMessagesNotInAllowedRange': 'HttpC2DMessagesNotInAllowedRange', 'HttpC2DRejectedMessagesNotInAllowedRange': 'HttpC2DRejectedMessagesNotInAllowedRange', 'HttpD2CMessagesNotInAllowedRange': 'HttpD2CMessagesNotInAllowedRange', 'MqttC2DMessagesNotInAllowedRange': 'MqttC2DMessagesNotInAllowedRange', 'MqttC2DRejectedMessagesNotInAllowedRange': 'MqttC2DRejectedMessagesNotInAllowedRange', 'MqttD2CMessagesNotInAllowedRange': 'MqttD2CMessagesNotInAllowedRange', 'QueuePurgesNotInAllowedRange': 'QueuePurgesNotInAllowedRange', 'TwinUpdatesNotInAllowedRange': 'TwinUpdatesNotInAllowedRange', 'UnauthorizedOperationsNotInAllowedRange': 'UnauthorizedOperationsNotInAllowedRange'} - } - - def __init__( - self, - **kwargs - ): - super(TimeWindowCustomAlertRule, self).__init__(**kwargs) - self.rule_type = 'TimeWindowCustomAlertRule' # type: str - self.time_window_size = kwargs['time_window_size'] - - -class ActiveConnectionsNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of active connections is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(ActiveConnectionsNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'ActiveConnectionsNotInAllowedRange' # type: str - - -class AdaptiveApplicationControlGroup(Resource, Location): - """AdaptiveApplicationControlGroup. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param enforcement_mode: The application control policy enforcement/protection mode of the - machine group. Possible values include: "Audit", "Enforce", "None". - :type enforcement_mode: str or ~azure.mgmt.security.models.EnforcementMode - :param protection_mode: The protection mode of the collection/file types. Exe/Msi/Script are - used for Windows, Executable is used for Linux. - :type protection_mode: ~azure.mgmt.security.models.ProtectionMode - :ivar configuration_status: The configuration status of the machines group or machine or rule. - Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", "NoStatus". - :vartype configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus - :ivar recommendation_status: The initial recommendation status of the machine group or machine. - Possible values include: "Recommended", "NotRecommended", "NotAvailable", "NoStatus". - :vartype recommendation_status: str or ~azure.mgmt.security.models.RecommendationStatus - :ivar issues: - :vartype issues: list[~azure.mgmt.security.models.AdaptiveApplicationControlIssueSummary] - :ivar source_system: The source type of the machine group. Possible values include: - "Azure_AppLocker", "Azure_AuditD", "NonAzure_AppLocker", "NonAzure_AuditD", "None". - :vartype source_system: str or ~azure.mgmt.security.models.SourceSystem - :param vm_recommendations: - :type vm_recommendations: list[~azure.mgmt.security.models.VmRecommendation] - :param path_recommendations: - :type path_recommendations: list[~azure.mgmt.security.models.PathRecommendation] - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'configuration_status': {'readonly': True}, - 'recommendation_status': {'readonly': True}, - 'issues': {'readonly': True}, - 'source_system': {'readonly': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'enforcement_mode': {'key': 'properties.enforcementMode', 'type': 'str'}, - 'protection_mode': {'key': 'properties.protectionMode', 'type': 'ProtectionMode'}, - 'configuration_status': {'key': 'properties.configurationStatus', 'type': 'str'}, - 'recommendation_status': {'key': 'properties.recommendationStatus', 'type': 'str'}, - 'issues': {'key': 'properties.issues', 'type': '[AdaptiveApplicationControlIssueSummary]'}, - 'source_system': {'key': 'properties.sourceSystem', 'type': 'str'}, - 'vm_recommendations': {'key': 'properties.vmRecommendations', 'type': '[VmRecommendation]'}, - 'path_recommendations': {'key': 'properties.pathRecommendations', 'type': '[PathRecommendation]'}, - } - - def __init__( - self, - **kwargs - ): - super(AdaptiveApplicationControlGroup, self).__init__(**kwargs) - self.location = None - self.enforcement_mode = kwargs.get('enforcement_mode', None) - self.protection_mode = kwargs.get('protection_mode', None) - self.configuration_status = None - self.recommendation_status = None - self.issues = None - self.source_system = None - self.vm_recommendations = kwargs.get('vm_recommendations', None) - self.path_recommendations = kwargs.get('path_recommendations', None) - self.id = None - self.name = None - self.type = None - self.enforcement_mode = kwargs.get('enforcement_mode', None) - self.protection_mode = kwargs.get('protection_mode', None) - self.configuration_status = None - self.recommendation_status = None - self.issues = None - self.source_system = None - self.vm_recommendations = kwargs.get('vm_recommendations', None) - self.path_recommendations = kwargs.get('path_recommendations', None) - - -class AdaptiveApplicationControlGroups(msrest.serialization.Model): - """Represents a list of machine groups and set of rules that are recommended by Azure Security Center to be allowed. - - :param value: - :type value: list[~azure.mgmt.security.models.AdaptiveApplicationControlGroup] - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[AdaptiveApplicationControlGroup]'}, - } - - def __init__( - self, - **kwargs - ): - super(AdaptiveApplicationControlGroups, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - - -class AdaptiveApplicationControlIssueSummary(msrest.serialization.Model): - """Represents a summary of the alerts of the machine group. - - :param issue: An alert that machines within a group can have. Possible values include: - "ViolationsAudited", "ViolationsBlocked", "MsiAndScriptViolationsAudited", - "MsiAndScriptViolationsBlocked", "ExecutableViolationsAudited", "RulesViolatedManually". - :type issue: str or ~azure.mgmt.security.models.AdaptiveApplicationControlIssue - :param number_of_vms: The number of machines in the group that have this alert. - :type number_of_vms: float - """ - - _attribute_map = { - 'issue': {'key': 'issue', 'type': 'str'}, - 'number_of_vms': {'key': 'numberOfVms', 'type': 'float'}, - } - - def __init__( - self, - **kwargs - ): - super(AdaptiveApplicationControlIssueSummary, self).__init__(**kwargs) - self.issue = kwargs.get('issue', None) - self.number_of_vms = kwargs.get('number_of_vms', None) - - -class AdaptiveNetworkHardening(Resource): - """The resource whose properties describes the Adaptive Network Hardening settings for some Azure resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param rules: The security rules which are recommended to be effective on the VM. - :type rules: list[~azure.mgmt.security.models.Rule] - :param rules_calculation_time: The UTC time on which the rules were calculated. - :type rules_calculation_time: ~datetime.datetime - :param effective_network_security_groups: The Network Security Groups effective on the network - interfaces of the protected resource. - :type effective_network_security_groups: - list[~azure.mgmt.security.models.EffectiveNetworkSecurityGroups] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'rules': {'key': 'properties.rules', 'type': '[Rule]'}, - 'rules_calculation_time': {'key': 'properties.rulesCalculationTime', 'type': 'iso-8601'}, - 'effective_network_security_groups': {'key': 'properties.effectiveNetworkSecurityGroups', 'type': '[EffectiveNetworkSecurityGroups]'}, - } - - def __init__( - self, - **kwargs - ): - super(AdaptiveNetworkHardening, self).__init__(**kwargs) - self.rules = kwargs.get('rules', None) - self.rules_calculation_time = kwargs.get('rules_calculation_time', None) - self.effective_network_security_groups = kwargs.get('effective_network_security_groups', None) - - -class AdaptiveNetworkHardeningEnforceRequest(msrest.serialization.Model): - """AdaptiveNetworkHardeningEnforceRequest. - - All required parameters must be populated in order to send to Azure. - - :param rules: Required. The rules to enforce. - :type rules: list[~azure.mgmt.security.models.Rule] - :param network_security_groups: Required. The Azure resource IDs of the effective network - security groups that will be updated with the created security rules from the Adaptive Network - Hardening rules. - :type network_security_groups: list[str] - """ - - _validation = { - 'rules': {'required': True}, - 'network_security_groups': {'required': True}, - } - - _attribute_map = { - 'rules': {'key': 'rules', 'type': '[Rule]'}, - 'network_security_groups': {'key': 'networkSecurityGroups', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(AdaptiveNetworkHardeningEnforceRequest, self).__init__(**kwargs) - self.rules = kwargs['rules'] - self.network_security_groups = kwargs['network_security_groups'] - - -class AdaptiveNetworkHardeningsList(msrest.serialization.Model): - """Response for ListAdaptiveNetworkHardenings API service call. - - :param value: A list of Adaptive Network Hardenings resources. - :type value: list[~azure.mgmt.security.models.AdaptiveNetworkHardening] - :param next_link: The URL to get the next set of results. - :type next_link: str - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[AdaptiveNetworkHardening]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AdaptiveNetworkHardeningsList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = kwargs.get('next_link', None) - - -class AdditionalData(msrest.serialization.Model): - """Details of the sub-assessment. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: ContainerRegistryVulnerabilityProperties, ServerVulnerabilityProperties, SqlServerVulnerabilityProperties. - - All required parameters must be populated in order to send to Azure. - - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", - "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType - """ - - _validation = { - 'assessed_resource_type': {'required': True}, - } - - _attribute_map = { - 'assessed_resource_type': {'key': 'assessedResourceType', 'type': 'str'}, - } - - _subtype_map = { - 'assessed_resource_type': {'ContainerRegistryVulnerability': 'ContainerRegistryVulnerabilityProperties', 'ServerVulnerabilityAssessment': 'ServerVulnerabilityProperties', 'SqlServerVulnerability': 'SqlServerVulnerabilityProperties'} - } - - def __init__( - self, - **kwargs - ): - super(AdditionalData, self).__init__(**kwargs) - self.assessed_resource_type = None # type: Optional[str] - - -class AdditionalWorkspacesProperties(msrest.serialization.Model): - """Properties of the additional workspaces. - - :param workspace: Workspace resource id. - :type workspace: str - :param type: Workspace type. Possible values include: "Sentinel". Default value: "Sentinel". - :type type: str or ~azure.mgmt.security.models.AdditionalWorkspaceType - :param data_types: List of data types sent to workspace. - :type data_types: list[str or ~azure.mgmt.security.models.AdditionalWorkspaceDataType] - """ - - _attribute_map = { - 'workspace': {'key': 'workspace', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'data_types': {'key': 'dataTypes', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(AdditionalWorkspacesProperties, self).__init__(**kwargs) - self.workspace = kwargs.get('workspace', None) - self.type = kwargs.get('type', "Sentinel") - self.data_types = kwargs.get('data_types', None) - - -class AdvancedThreatProtectionSetting(Resource): - """The Advanced Threat Protection resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param is_enabled: Indicates whether Advanced Threat Protection is enabled. - :type is_enabled: bool - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'is_enabled': {'key': 'properties.isEnabled', 'type': 'bool'}, - } - - def __init__( - self, - **kwargs - ): - super(AdvancedThreatProtectionSetting, self).__init__(**kwargs) - self.is_enabled = kwargs.get('is_enabled', None) - - -class Alert(Resource): - """Security alert. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar alert_type: Unique identifier for the detection logic (all alert instances from the same - detection logic will have the same alertType). - :vartype alert_type: str - :ivar system_alert_id: Unique identifier for the alert. - :vartype system_alert_id: str - :ivar product_component_name: The name of Azure Security Center pricing tier which powering - this alert. Learn more: - https://docs.microsoft.com/en-us/azure/security-center/security-center-pricing. - :vartype product_component_name: str - :ivar alert_display_name: The display name of the alert. - :vartype alert_display_name: str - :ivar description: Description of the suspicious activity that was detected. - :vartype description: str - :ivar severity: The risk level of the threat that was detected. Learn more: - https://docs.microsoft.com/en-us/azure/security-center/security-center-alerts-overview#how-are-alerts-classified. - Possible values include: "Informational", "Low", "Medium", "High". - :vartype severity: str or ~azure.mgmt.security.models.AlertSeverity - :ivar intent: The kill chain related intent behind the alert. For list of supported values, and - explanations of Azure Security Center's supported kill chain intents. Possible values include: - "Unknown", "PreAttack", "InitialAccess", "Persistence", "PrivilegeEscalation", - "DefenseEvasion", "CredentialAccess", "Discovery", "LateralMovement", "Execution", - "Collection", "Exfiltration", "CommandAndControl", "Impact", "Probing", "Exploitation". - :vartype intent: str or ~azure.mgmt.security.models.Intent - :ivar start_time_utc: The UTC time of the first event or activity included in the alert in - ISO8601 format. - :vartype start_time_utc: ~datetime.datetime - :ivar end_time_utc: The UTC time of the last event or activity included in the alert in ISO8601 - format. - :vartype end_time_utc: ~datetime.datetime - :ivar resource_identifiers: The resource identifiers that can be used to direct the alert to - the right product exposure group (tenant, workspace, subscription etc.). There can be multiple - identifiers of different type per alert. - :vartype resource_identifiers: list[~azure.mgmt.security.models.ResourceIdentifier] - :ivar remediation_steps: Manual action items to take to remediate the alert. - :vartype remediation_steps: list[str] - :ivar vendor_name: The name of the vendor that raises the alert. - :vartype vendor_name: str - :ivar status: The life cycle status of the alert. Possible values include: "Active", - "Resolved", "Dismissed". - :vartype status: str or ~azure.mgmt.security.models.AlertStatus - :ivar extended_links: Links related to the alert. - :vartype extended_links: list[dict[str, str]] - :ivar alert_uri: A direct link to the alert page in Azure Portal. - :vartype alert_uri: str - :ivar time_generated_utc: The UTC time the alert was generated in ISO8601 format. - :vartype time_generated_utc: ~datetime.datetime - :ivar product_name: The name of the product which published this alert (Azure Security Center, - Azure ATP, Microsoft Defender ATP, O365 ATP, MCAS, and so on). - :vartype product_name: str - :ivar processing_end_time_utc: The UTC processing end time of the alert in ISO8601 format. - :vartype processing_end_time_utc: ~datetime.datetime - :ivar entities: A list of entities related to the alert. - :vartype entities: list[~azure.mgmt.security.models.AlertEntity] - :ivar is_incident: This field determines whether the alert is an incident (a compound grouping - of several alerts) or a single alert. - :vartype is_incident: bool - :ivar correlation_key: Key for corelating related alerts. Alerts with the same correlation key - considered to be related. - :vartype correlation_key: str - :param extended_properties: Custom properties for the alert. - :type extended_properties: dict[str, str] - :ivar compromised_entity: The display name of the resource most related to this alert. - :vartype compromised_entity: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'alert_type': {'readonly': True}, - 'system_alert_id': {'readonly': True}, - 'product_component_name': {'readonly': True}, - 'alert_display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'severity': {'readonly': True}, - 'intent': {'readonly': True}, - 'start_time_utc': {'readonly': True}, - 'end_time_utc': {'readonly': True}, - 'resource_identifiers': {'readonly': True}, - 'remediation_steps': {'readonly': True}, - 'vendor_name': {'readonly': True}, - 'status': {'readonly': True}, - 'extended_links': {'readonly': True}, - 'alert_uri': {'readonly': True}, - 'time_generated_utc': {'readonly': True}, - 'product_name': {'readonly': True}, - 'processing_end_time_utc': {'readonly': True}, - 'entities': {'readonly': True}, - 'is_incident': {'readonly': True}, - 'correlation_key': {'readonly': True}, - 'compromised_entity': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'alert_type': {'key': 'properties.alertType', 'type': 'str'}, - 'system_alert_id': {'key': 'properties.systemAlertId', 'type': 'str'}, - 'product_component_name': {'key': 'properties.productComponentName', 'type': 'str'}, - 'alert_display_name': {'key': 'properties.alertDisplayName', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'severity': {'key': 'properties.severity', 'type': 'str'}, - 'intent': {'key': 'properties.intent', 'type': 'str'}, - 'start_time_utc': {'key': 'properties.startTimeUtc', 'type': 'iso-8601'}, - 'end_time_utc': {'key': 'properties.endTimeUtc', 'type': 'iso-8601'}, - 'resource_identifiers': {'key': 'properties.resourceIdentifiers', 'type': '[ResourceIdentifier]'}, - 'remediation_steps': {'key': 'properties.remediationSteps', 'type': '[str]'}, - 'vendor_name': {'key': 'properties.vendorName', 'type': 'str'}, - 'status': {'key': 'properties.status', 'type': 'str'}, - 'extended_links': {'key': 'properties.extendedLinks', 'type': '[{str}]'}, - 'alert_uri': {'key': 'properties.alertUri', 'type': 'str'}, - 'time_generated_utc': {'key': 'properties.timeGeneratedUtc', 'type': 'iso-8601'}, - 'product_name': {'key': 'properties.productName', 'type': 'str'}, - 'processing_end_time_utc': {'key': 'properties.processingEndTimeUtc', 'type': 'iso-8601'}, - 'entities': {'key': 'properties.entities', 'type': '[AlertEntity]'}, - 'is_incident': {'key': 'properties.isIncident', 'type': 'bool'}, - 'correlation_key': {'key': 'properties.correlationKey', 'type': 'str'}, - 'extended_properties': {'key': 'properties.extendedProperties', 'type': '{str}'}, - 'compromised_entity': {'key': 'properties.compromisedEntity', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(Alert, self).__init__(**kwargs) - self.alert_type = None - self.system_alert_id = None - self.product_component_name = None - self.alert_display_name = None - self.description = None - self.severity = None - self.intent = None - self.start_time_utc = None - self.end_time_utc = None - self.resource_identifiers = None - self.remediation_steps = None - self.vendor_name = None - self.status = None - self.extended_links = None - self.alert_uri = None - self.time_generated_utc = None - self.product_name = None - self.processing_end_time_utc = None - self.entities = None - self.is_incident = None - self.correlation_key = None - self.extended_properties = kwargs.get('extended_properties', None) - self.compromised_entity = None - - -class AlertEntity(msrest.serialization.Model): - """Changing set of properties depending on the entity type. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :ivar type: Type of entity. - :vartype type: str - """ - - _validation = { - 'type': {'readonly': True}, - } - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'type': {'key': 'type', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AlertEntity, self).__init__(**kwargs) - self.additional_properties = kwargs.get('additional_properties', None) - self.type = None - - -class AlertList(msrest.serialization.Model): - """List of security alerts. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: describes security alert properties. - :type value: list[~azure.mgmt.security.models.Alert] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Alert]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AlertList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class AlertSimulatorRequestProperties(msrest.serialization.Model): - """Describes properties of an alert simulation request. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: AlertSimulatorBundlesRequestProperties. - - All required parameters must be populated in order to send to Azure. - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param kind: Required. The kind of alert simulation.Constant filled by server. Possible values - include: "Bundles". - :type kind: str or ~azure.mgmt.security.models.KindEnum - """ - - _validation = { - 'kind': {'required': True}, - } - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'kind': {'key': 'kind', 'type': 'str'}, - } - - _subtype_map = { - 'kind': {'Bundles': 'AlertSimulatorBundlesRequestProperties'} - } - - def __init__( - self, - **kwargs - ): - super(AlertSimulatorRequestProperties, self).__init__(**kwargs) - self.additional_properties = kwargs.get('additional_properties', None) - self.kind = 'AlertSimulatorRequestProperties' # type: str - - -class AlertSimulatorBundlesRequestProperties(AlertSimulatorRequestProperties): - """Simulate alerts according to this bundles. - - All required parameters must be populated in order to send to Azure. - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param kind: Required. The kind of alert simulation.Constant filled by server. Possible values - include: "Bundles". - :type kind: str or ~azure.mgmt.security.models.KindEnum - :param bundles: Bundles list. - :type bundles: list[str or ~azure.mgmt.security.models.BundleType] - """ - - _validation = { - 'kind': {'required': True}, - } - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'kind': {'key': 'kind', 'type': 'str'}, - 'bundles': {'key': 'bundles', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(AlertSimulatorBundlesRequestProperties, self).__init__(**kwargs) - self.kind = 'Bundles' # type: str - self.bundles = kwargs.get('bundles', None) - - -class AlertSimulatorRequestBody(msrest.serialization.Model): - """Alert Simulator request body. - - :param properties: Alert Simulator request body data. - :type properties: ~azure.mgmt.security.models.AlertSimulatorRequestProperties - """ - - _attribute_map = { - 'properties': {'key': 'properties', 'type': 'AlertSimulatorRequestProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(AlertSimulatorRequestBody, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class AlertsSuppressionRule(Resource): - """Describes the suppression rule. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param alert_type: Type of the alert to automatically suppress. For all alert types, use '*'. - :type alert_type: str - :ivar last_modified_utc: The last time this rule was modified. - :vartype last_modified_utc: ~datetime.datetime - :param expiration_date_utc: Expiration date of the rule, if value is not provided or provided - as null this field will default to the maximum allowed expiration date. - :type expiration_date_utc: ~datetime.datetime - :param reason: The reason for dismissing the alert. - :type reason: str - :param state: Possible states of the rule. Possible values include: "Enabled", "Disabled", - "Expired". - :type state: str or ~azure.mgmt.security.models.RuleState - :param comment: Any comment regarding the rule. - :type comment: str - :param suppression_alerts_scope: The suppression conditions. - :type suppression_alerts_scope: ~azure.mgmt.security.models.SuppressionAlertsScope - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'last_modified_utc': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'alert_type': {'key': 'properties.alertType', 'type': 'str'}, - 'last_modified_utc': {'key': 'properties.lastModifiedUtc', 'type': 'iso-8601'}, - 'expiration_date_utc': {'key': 'properties.expirationDateUtc', 'type': 'iso-8601'}, - 'reason': {'key': 'properties.reason', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'comment': {'key': 'properties.comment', 'type': 'str'}, - 'suppression_alerts_scope': {'key': 'properties.suppressionAlertsScope', 'type': 'SuppressionAlertsScope'}, - } - - def __init__( - self, - **kwargs - ): - super(AlertsSuppressionRule, self).__init__(**kwargs) - self.alert_type = kwargs.get('alert_type', None) - self.last_modified_utc = None - self.expiration_date_utc = kwargs.get('expiration_date_utc', None) - self.reason = kwargs.get('reason', None) - self.state = kwargs.get('state', None) - self.comment = kwargs.get('comment', None) - self.suppression_alerts_scope = kwargs.get('suppression_alerts_scope', None) - - -class AlertsSuppressionRulesList(msrest.serialization.Model): - """Suppression rules list for subscription. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. - :type value: list[~azure.mgmt.security.models.AlertsSuppressionRule] - :ivar next_link: URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[AlertsSuppressionRule]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AlertsSuppressionRulesList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class Setting(Resource): - """The kind of the security setting. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: AlertSyncSettings, DataExportSettings. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param kind: Required. the kind of the settings string.Constant filled by server. Possible - values include: "DataExportSettings", "AlertSuppressionSetting", "AlertSyncSettings". - :type kind: str or ~azure.mgmt.security.models.SettingKind - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'kind': {'required': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, - } - - _subtype_map = { - 'kind': {'AlertSyncSettings': 'AlertSyncSettings', 'DataExportSettings': 'DataExportSettings'} - } - - def __init__( - self, - **kwargs - ): - super(Setting, self).__init__(**kwargs) - self.kind = 'Setting' # type: str - - -class AlertSyncSettings(Setting): - """Represents an alert sync setting. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param kind: Required. the kind of the settings string.Constant filled by server. Possible - values include: "DataExportSettings", "AlertSuppressionSetting", "AlertSyncSettings". - :type kind: str or ~azure.mgmt.security.models.SettingKind - :param enabled: Is the alert sync setting enabled. - :type enabled: bool - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'kind': {'required': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, - 'enabled': {'key': 'properties.enabled', 'type': 'bool'}, - } - - def __init__( - self, - **kwargs - ): - super(AlertSyncSettings, self).__init__(**kwargs) - self.kind = 'AlertSyncSettings' # type: str - self.enabled = kwargs.get('enabled', None) - - -class AllowedConnectionsList(msrest.serialization.Model): - """List of all possible traffic between Azure resources. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: - :vartype value: list[~azure.mgmt.security.models.AllowedConnectionsResource] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[AllowedConnectionsResource]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AllowedConnectionsList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class AllowedConnectionsResource(Resource, Location): - """The resource whose properties describes the allowed traffic between Azure resources. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar calculated_date_time: The UTC time on which the allowed connections resource was - calculated. - :vartype calculated_date_time: ~datetime.datetime - :ivar connectable_resources: List of connectable resources. - :vartype connectable_resources: list[~azure.mgmt.security.models.ConnectableResource] - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'calculated_date_time': {'readonly': True}, - 'connectable_resources': {'readonly': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'calculated_date_time': {'key': 'properties.calculatedDateTime', 'type': 'iso-8601'}, - 'connectable_resources': {'key': 'properties.connectableResources', 'type': '[ConnectableResource]'}, - } - - def __init__( - self, - **kwargs - ): - super(AllowedConnectionsResource, self).__init__(**kwargs) - self.location = None - self.calculated_date_time = None - self.connectable_resources = None - self.id = None - self.name = None - self.type = None - self.calculated_date_time = None - self.connectable_resources = None - - -class ListCustomAlertRule(CustomAlertRule): - """A List custom alert rule. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: AllowlistCustomAlertRule, DenylistCustomAlertRule. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - } - - _subtype_map = { - 'rule_type': {'AllowlistCustomAlertRule': 'AllowlistCustomAlertRule', 'DenylistCustomAlertRule': 'DenylistCustomAlertRule'} - } - - def __init__( - self, - **kwargs - ): - super(ListCustomAlertRule, self).__init__(**kwargs) - self.rule_type = 'ListCustomAlertRule' # type: str - self.value_type = None - - -class AllowlistCustomAlertRule(ListCustomAlertRule): - """A custom alert rule that checks if a value (depends on the custom alert type) is allowed. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: ConnectionFromIpNotAllowed, ConnectionToIpNotAllowed, LocalUserNotAllowed, ProcessNotAllowed. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the - rule type. - :type allowlist_values: list[str] - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'allowlist_values': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'allowlist_values': {'key': 'allowlistValues', 'type': '[str]'}, - } - - _subtype_map = { - 'rule_type': {'ConnectionFromIpNotAllowed': 'ConnectionFromIpNotAllowed', 'ConnectionToIpNotAllowed': 'ConnectionToIpNotAllowed', 'LocalUserNotAllowed': 'LocalUserNotAllowed', 'ProcessNotAllowed': 'ProcessNotAllowed'} - } - - def __init__( - self, - **kwargs - ): - super(AllowlistCustomAlertRule, self).__init__(**kwargs) - self.rule_type = 'AllowlistCustomAlertRule' # type: str - self.allowlist_values = kwargs['allowlist_values'] - - -class AmqpC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of cloud to device messages (AMQP protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(AmqpC2DMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'AmqpC2DMessagesNotInAllowedRange' # type: str - - -class AmqpC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of rejected cloud to device messages (AMQP protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(AmqpC2DRejectedMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'AmqpC2DRejectedMessagesNotInAllowedRange' # type: str - - -class AmqpD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of device to cloud messages (AMQP protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(AmqpD2CMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'AmqpD2CMessagesNotInAllowedRange' # type: str - - -class AscLocation(Resource): - """The ASC location of the subscription is in the "name" field. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: Any object. - :type properties: any - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'object'}, - } - - def __init__( - self, - **kwargs - ): - super(AscLocation, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class AscLocationList(msrest.serialization.Model): - """List of locations where ASC saves your data. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: - :vartype value: list[~azure.mgmt.security.models.AscLocation] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[AscLocation]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AscLocationList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class AssessmentLinks(msrest.serialization.Model): - """Links relevant to the assessment. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar azure_portal_uri: Link to assessment in Azure Portal. - :vartype azure_portal_uri: str - """ - - _validation = { - 'azure_portal_uri': {'readonly': True}, - } - - _attribute_map = { - 'azure_portal_uri': {'key': 'azurePortalUri', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AssessmentLinks, self).__init__(**kwargs) - self.azure_portal_uri = None - - -class AssessmentStatus(msrest.serialization.Model): - """The result of the assessment. - - All required parameters must be populated in order to send to Azure. - - :param code: Required. Programmatic code for the status of the assessment. Possible values - include: "Healthy", "Unhealthy", "NotApplicable". - :type code: str or ~azure.mgmt.security.models.AssessmentStatusCode - :param cause: Programmatic code for the cause of the assessment status. - :type cause: str - :param description: Human readable description of the assessment status. - :type description: str - """ - - _validation = { - 'code': {'required': True}, - } - - _attribute_map = { - 'code': {'key': 'code', 'type': 'str'}, - 'cause': {'key': 'cause', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AssessmentStatus, self).__init__(**kwargs) - self.code = kwargs['code'] - self.cause = kwargs.get('cause', None) - self.description = kwargs.get('description', None) - - -class AtaExternalSecuritySolution(ExternalSecuritySolution): - """Represents an ATA security solution which sends logs to an OMS workspace. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: The external security solution properties for ATA solutions. - :type properties: ~azure.mgmt.security.models.AtaSolutionProperties - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'kind': {'key': 'kind', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'AtaSolutionProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(AtaExternalSecuritySolution, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class AtaSolutionProperties(ExternalSecuritySolutionProperties): - """The external security solution properties for ATA solutions. - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace - :param last_event_received: - :type last_event_received: str - """ - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'device_vendor': {'key': 'deviceVendor', 'type': 'str'}, - 'device_type': {'key': 'deviceType', 'type': 'str'}, - 'workspace': {'key': 'workspace', 'type': 'ConnectedWorkspace'}, - 'last_event_received': {'key': 'lastEventReceived', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AtaSolutionProperties, self).__init__(**kwargs) - self.last_event_received = kwargs.get('last_event_received', None) - - -class AuthenticationDetailsProperties(msrest.serialization.Model): - """Settings for cloud authentication management. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: AwAssumeRoleAuthenticationDetailsProperties, AwsCredsAuthenticationDetailsProperties, GcpCredentialsDetailsProperties. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar authentication_provisioning_state: State of the multi-cloud connector. Possible values - include: "Valid", "Invalid", "Expired", "IncorrectPolicy". - :vartype authentication_provisioning_state: str or - ~azure.mgmt.security.models.AuthenticationProvisioningState - :ivar granted_permissions: The permissions detected in the cloud account. - :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account - credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType - """ - - _validation = { - 'authentication_provisioning_state': {'readonly': True}, - 'granted_permissions': {'readonly': True}, - 'authentication_type': {'required': True}, - } - - _attribute_map = { - 'authentication_provisioning_state': {'key': 'authenticationProvisioningState', 'type': 'str'}, - 'granted_permissions': {'key': 'grantedPermissions', 'type': '[str]'}, - 'authentication_type': {'key': 'authenticationType', 'type': 'str'}, - } - - _subtype_map = { - 'authentication_type': {'awsAssumeRole': 'AwAssumeRoleAuthenticationDetailsProperties', 'awsCreds': 'AwsCredsAuthenticationDetailsProperties', 'gcpCredentials': 'GcpCredentialsDetailsProperties'} - } - - def __init__( - self, - **kwargs - ): - super(AuthenticationDetailsProperties, self).__init__(**kwargs) - self.authentication_provisioning_state = None - self.granted_permissions = None - self.authentication_type = None # type: Optional[str] - - -class Tags(msrest.serialization.Model): - """A list of key value pairs that describe the resource. - - :param tags: A set of tags. A list of key value pairs that describe the resource. - :type tags: dict[str, str] - """ - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - } - - def __init__( - self, - **kwargs - ): - super(Tags, self).__init__(**kwargs) - self.tags = kwargs.get('tags', None) - - -class ETag(msrest.serialization.Model): - """Entity tag is used for comparing two or more entities from the same requested resource. - - :param etag: Entity tag is used for comparing two or more entities from the same requested - resource. - :type etag: str - """ - - _attribute_map = { - 'etag': {'key': 'etag', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ETag, self).__init__(**kwargs) - self.etag = kwargs.get('etag', None) - - -class Kind(msrest.serialization.Model): - """Describes an Azure resource with kind. - - :param kind: Kind of the resource. - :type kind: str - """ - - _attribute_map = { - 'kind': {'key': 'kind', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(Kind, self).__init__(**kwargs) - self.kind = kwargs.get('kind', None) - - -class AzureTrackedResourceLocation(msrest.serialization.Model): - """Describes an Azure resource with location. - - :param location: Location where the resource is stored. - :type location: str - """ - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AzureTrackedResourceLocation, self).__init__(**kwargs) - self.location = kwargs.get('location', None) - - -class TrackedResource(Resource, AzureTrackedResourceLocation, Kind, ETag, Tags): - """Describes an Azure tracked resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param tags: A set of tags. A list of key value pairs that describe the resource. - :type tags: dict[str, str] - :param etag: Entity tag is used for comparing two or more entities from the same requested - resource. - :type etag: str - :param kind: Kind of the resource. - :type kind: str - :param location: Location where the resource is stored. - :type location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'etag': {'key': 'etag', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(TrackedResource, self).__init__(**kwargs) - self.tags = kwargs.get('tags', None) - self.etag = kwargs.get('etag', None) - self.kind = kwargs.get('kind', None) - self.location = kwargs.get('location', None) - self.tags = kwargs.get('tags', None) - self.etag = kwargs.get('etag', None) - self.kind = kwargs.get('kind', None) - self.id = None - self.name = None - self.type = None - self.tags = kwargs.get('tags', None) - self.etag = kwargs.get('etag', None) - self.location = kwargs.get('location', None) - self.id = None - self.name = None - self.type = None - self.tags = kwargs.get('tags', None) - self.kind = kwargs.get('kind', None) - self.location = kwargs.get('location', None) - self.id = None - self.name = None - self.type = None - self.etag = kwargs.get('etag', None) - self.kind = kwargs.get('kind', None) - self.location = kwargs.get('location', None) - self.id = None - self.name = None - self.type = None - - -class Automation(TrackedResource): - """The security automation resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param tags: A set of tags. A list of key value pairs that describe the resource. - :type tags: dict[str, str] - :param etag: Entity tag is used for comparing two or more entities from the same requested - resource. - :type etag: str - :param kind: Kind of the resource. - :type kind: str - :param location: Location where the resource is stored. - :type location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param description: The security automation description. - :type description: str - :param is_enabled: Indicates whether the security automation is enabled. - :type is_enabled: bool - :param scopes: A collection of scopes on which the security automations logic is applied. - Supported scopes are the subscription itself or a resource group under that subscription. The - automation will only apply on defined scopes. - :type scopes: list[~azure.mgmt.security.models.AutomationScope] - :param sources: A collection of the source event types which evaluate the security automation - set of rules. - :type sources: list[~azure.mgmt.security.models.AutomationSource] - :param actions: A collection of the actions which are triggered if all the configured rules - evaluations, within at least one rule set, are true. - :type actions: list[~azure.mgmt.security.models.AutomationAction] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'etag': {'key': 'etag', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'is_enabled': {'key': 'properties.isEnabled', 'type': 'bool'}, - 'scopes': {'key': 'properties.scopes', 'type': '[AutomationScope]'}, - 'sources': {'key': 'properties.sources', 'type': '[AutomationSource]'}, - 'actions': {'key': 'properties.actions', 'type': '[AutomationAction]'}, - } - - def __init__( - self, - **kwargs - ): - super(Automation, self).__init__(**kwargs) - self.description = kwargs.get('description', None) - self.is_enabled = kwargs.get('is_enabled', None) - self.scopes = kwargs.get('scopes', None) - self.sources = kwargs.get('sources', None) - self.actions = kwargs.get('actions', None) - - -class AutomationAction(msrest.serialization.Model): - """The action that should be triggered. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: AutomationActionEventHub, AutomationActionLogicApp, AutomationActionWorkspace. - - All required parameters must be populated in order to send to Azure. - - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", - "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType - """ - - _validation = { - 'action_type': {'required': True}, - } - - _attribute_map = { - 'action_type': {'key': 'actionType', 'type': 'str'}, - } - - _subtype_map = { - 'action_type': {'EventHub': 'AutomationActionEventHub', 'LogicApp': 'AutomationActionLogicApp', 'Workspace': 'AutomationActionWorkspace'} - } - - def __init__( - self, - **kwargs - ): - super(AutomationAction, self).__init__(**kwargs) - self.action_type = None # type: Optional[str] - - -class AutomationActionEventHub(AutomationAction): - """The target Event Hub to which event data will be exported. To learn more about Security Center continuous export capabilities, visit https://aka.ms/ASCExportLearnMore. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", - "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType - :param event_hub_resource_id: The target Event Hub Azure Resource ID. - :type event_hub_resource_id: str - :ivar sas_policy_name: The target Event Hub SAS policy name. - :vartype sas_policy_name: str - :param connection_string: The target Event Hub connection string (it will not be included in - any response). - :type connection_string: str - """ - - _validation = { - 'action_type': {'required': True}, - 'sas_policy_name': {'readonly': True}, - } - - _attribute_map = { - 'action_type': {'key': 'actionType', 'type': 'str'}, - 'event_hub_resource_id': {'key': 'eventHubResourceId', 'type': 'str'}, - 'sas_policy_name': {'key': 'sasPolicyName', 'type': 'str'}, - 'connection_string': {'key': 'connectionString', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationActionEventHub, self).__init__(**kwargs) - self.action_type = 'EventHub' # type: str - self.event_hub_resource_id = kwargs.get('event_hub_resource_id', None) - self.sas_policy_name = None - self.connection_string = kwargs.get('connection_string', None) - - -class AutomationActionLogicApp(AutomationAction): - """The logic app action that should be triggered. To learn more about Security Center's Workflow Automation capabilities, visit https://aka.ms/ASCWorkflowAutomationLearnMore. - - All required parameters must be populated in order to send to Azure. - - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", - "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType - :param logic_app_resource_id: The triggered Logic App Azure Resource ID. This can also reside - on other subscriptions, given that you have permissions to trigger the Logic App. - :type logic_app_resource_id: str - :param uri: The Logic App trigger URI endpoint (it will not be included in any response). - :type uri: str - """ - - _validation = { - 'action_type': {'required': True}, - } - - _attribute_map = { - 'action_type': {'key': 'actionType', 'type': 'str'}, - 'logic_app_resource_id': {'key': 'logicAppResourceId', 'type': 'str'}, - 'uri': {'key': 'uri', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationActionLogicApp, self).__init__(**kwargs) - self.action_type = 'LogicApp' # type: str - self.logic_app_resource_id = kwargs.get('logic_app_resource_id', None) - self.uri = kwargs.get('uri', None) - - -class AutomationActionWorkspace(AutomationAction): - """The Log Analytics Workspace to which event data will be exported. Security alerts data will reside in the 'SecurityAlert' table and the assessments data will reside in the 'SecurityRecommendation' table (under the 'Security'/'SecurityCenterFree' solutions). Note that in order to view the data in the workspace, the Security Center Log Analytics free/standard solution needs to be enabled on that workspace. To learn more about Security Center continuous export capabilities, visit https://aka.ms/ASCExportLearnMore. - - All required parameters must be populated in order to send to Azure. - - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", - "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType - :param workspace_resource_id: The fully qualified Log Analytics Workspace Azure Resource ID. - :type workspace_resource_id: str - """ - - _validation = { - 'action_type': {'required': True}, - } - - _attribute_map = { - 'action_type': {'key': 'actionType', 'type': 'str'}, - 'workspace_resource_id': {'key': 'workspaceResourceId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationActionWorkspace, self).__init__(**kwargs) - self.action_type = 'Workspace' # type: str - self.workspace_resource_id = kwargs.get('workspace_resource_id', None) - - -class AutomationList(msrest.serialization.Model): - """List of security automations response. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. The list of security automations under the given scope. - :type value: list[~azure.mgmt.security.models.Automation] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Automation]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class AutomationRuleSet(msrest.serialization.Model): - """A rule set which evaluates all its rules upon an event interception. Only when all the included rules in the rule set will be evaluated as 'true', will the event trigger the defined actions. - - :param rules: - :type rules: list[~azure.mgmt.security.models.AutomationTriggeringRule] - """ - - _attribute_map = { - 'rules': {'key': 'rules', 'type': '[AutomationTriggeringRule]'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationRuleSet, self).__init__(**kwargs) - self.rules = kwargs.get('rules', None) - - -class AutomationScope(msrest.serialization.Model): - """A single automation scope. - - :param description: The resources scope description. - :type description: str - :param scope_path: The resources scope path. Can be the subscription on which the automation is - defined on or a resource group under that subscription (fully qualified Azure resource IDs). - :type scope_path: str - """ - - _attribute_map = { - 'description': {'key': 'description', 'type': 'str'}, - 'scope_path': {'key': 'scopePath', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationScope, self).__init__(**kwargs) - self.description = kwargs.get('description', None) - self.scope_path = kwargs.get('scope_path', None) - - -class AutomationSource(msrest.serialization.Model): - """The source event types which evaluate the security automation set of rules. For example - security alerts and security assessments. To learn more about the supported security events data models schemas - please visit https://aka.ms/ASCAutomationSchemas. - - :param event_source: A valid event source type. Possible values include: "Assessments", - "SubAssessments", "Alerts", "SecureScores", "SecureScoresSnapshot", "SecureScoreControls", - "SecureScoreControlsSnapshot", "RegulatoryComplianceAssessment", - "RegulatoryComplianceAssessmentSnapshot". - :type event_source: str or ~azure.mgmt.security.models.EventSource - :param rule_sets: A set of rules which evaluate upon event interception. A logical disjunction - is applied between defined rule sets (logical 'or'). - :type rule_sets: list[~azure.mgmt.security.models.AutomationRuleSet] - """ - - _attribute_map = { - 'event_source': {'key': 'eventSource', 'type': 'str'}, - 'rule_sets': {'key': 'ruleSets', 'type': '[AutomationRuleSet]'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationSource, self).__init__(**kwargs) - self.event_source = kwargs.get('event_source', None) - self.rule_sets = kwargs.get('rule_sets', None) - - -class AutomationTriggeringRule(msrest.serialization.Model): - """A rule which is evaluated upon event interception. The rule is configured by comparing a specific value from the event model to an expected value. This comparison is done by using one of the supported operators set. - - :param property_j_path: The JPath of the entity model property that should be checked. - :type property_j_path: str - :param property_type: The data type of the compared operands (string, integer, floating point - number or a boolean [true/false]]. Possible values include: "String", "Integer", "Number", - "Boolean". - :type property_type: str or ~azure.mgmt.security.models.PropertyType - :param expected_value: The expected value. - :type expected_value: str - :param operator: A valid comparer operator to use. A case-insensitive comparison will be - applied for String PropertyType. Possible values include: "Equals", "GreaterThan", - "GreaterThanOrEqualTo", "LesserThan", "LesserThanOrEqualTo", "NotEquals", "Contains", - "StartsWith", "EndsWith". - :type operator: str or ~azure.mgmt.security.models.Operator - """ - - _attribute_map = { - 'property_j_path': {'key': 'propertyJPath', 'type': 'str'}, - 'property_type': {'key': 'propertyType', 'type': 'str'}, - 'expected_value': {'key': 'expectedValue', 'type': 'str'}, - 'operator': {'key': 'operator', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationTriggeringRule, self).__init__(**kwargs) - self.property_j_path = kwargs.get('property_j_path', None) - self.property_type = kwargs.get('property_type', None) - self.expected_value = kwargs.get('expected_value', None) - self.operator = kwargs.get('operator', None) - - -class AutomationValidationStatus(msrest.serialization.Model): - """The security automation model state property bag. - - :param is_valid: Indicates whether the model is valid or not. - :type is_valid: bool - :param message: The validation message. - :type message: str - """ - - _attribute_map = { - 'is_valid': {'key': 'isValid', 'type': 'bool'}, - 'message': {'key': 'message', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutomationValidationStatus, self).__init__(**kwargs) - self.is_valid = kwargs.get('is_valid', None) - self.message = kwargs.get('message', None) - - -class AutoProvisioningSetting(Resource): - """Auto provisioning setting. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param auto_provision: Describes what kind of security agent provisioning action to take. - Possible values include: "On", "Off". - :type auto_provision: str or ~azure.mgmt.security.models.AutoProvision - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'auto_provision': {'key': 'properties.autoProvision', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutoProvisioningSetting, self).__init__(**kwargs) - self.auto_provision = kwargs.get('auto_provision', None) - - -class AutoProvisioningSettingList(msrest.serialization.Model): - """List of all the auto provisioning settings response. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: List of all the auto provisioning settings. - :type value: list[~azure.mgmt.security.models.AutoProvisioningSetting] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[AutoProvisioningSetting]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AutoProvisioningSettingList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class AwAssumeRoleAuthenticationDetailsProperties(AuthenticationDetailsProperties): - """AWS cloud account connector based assume role, the role enables delegating access to your AWS resources. The role is composed of role Amazon Resource Name (ARN) and external ID. For more details, refer to :code:`Creating a Role to Delegate Permissions to an IAM User (write only)`. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar authentication_provisioning_state: State of the multi-cloud connector. Possible values - include: "Valid", "Invalid", "Expired", "IncorrectPolicy". - :vartype authentication_provisioning_state: str or - ~azure.mgmt.security.models.AuthenticationProvisioningState - :ivar granted_permissions: The permissions detected in the cloud account. - :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account - credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType - :ivar account_id: The ID of the cloud account. - :vartype account_id: str - :param aws_assume_role_arn: Required. Assumed role ID is an identifier that you can use to - create temporary security credentials. - :type aws_assume_role_arn: str - :param aws_external_id: Required. A unique identifier that is required when you assume a role - in another account. - :type aws_external_id: str - """ - - _validation = { - 'authentication_provisioning_state': {'readonly': True}, - 'granted_permissions': {'readonly': True}, - 'authentication_type': {'required': True}, - 'account_id': {'readonly': True}, - 'aws_assume_role_arn': {'required': True}, - 'aws_external_id': {'required': True}, - } - - _attribute_map = { - 'authentication_provisioning_state': {'key': 'authenticationProvisioningState', 'type': 'str'}, - 'granted_permissions': {'key': 'grantedPermissions', 'type': '[str]'}, - 'authentication_type': {'key': 'authenticationType', 'type': 'str'}, - 'account_id': {'key': 'accountId', 'type': 'str'}, - 'aws_assume_role_arn': {'key': 'awsAssumeRoleArn', 'type': 'str'}, - 'aws_external_id': {'key': 'awsExternalId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AwAssumeRoleAuthenticationDetailsProperties, self).__init__(**kwargs) - self.authentication_type = 'awsAssumeRole' # type: str - self.account_id = None - self.aws_assume_role_arn = kwargs['aws_assume_role_arn'] - self.aws_external_id = kwargs['aws_external_id'] - - -class AwsCredsAuthenticationDetailsProperties(AuthenticationDetailsProperties): - """AWS cloud account connector based credentials, the credentials is composed of access key ID and secret key, for more details, refer to :code:`Creating an IAM User in Your AWS Account (write only)`. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar authentication_provisioning_state: State of the multi-cloud connector. Possible values - include: "Valid", "Invalid", "Expired", "IncorrectPolicy". - :vartype authentication_provisioning_state: str or - ~azure.mgmt.security.models.AuthenticationProvisioningState - :ivar granted_permissions: The permissions detected in the cloud account. - :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account - credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType - :ivar account_id: The ID of the cloud account. - :vartype account_id: str - :param aws_access_key_id: Required. Public key element of the AWS credential object (write - only). - :type aws_access_key_id: str - :param aws_secret_access_key: Required. Secret key element of the AWS credential object (write - only). - :type aws_secret_access_key: str - """ - - _validation = { - 'authentication_provisioning_state': {'readonly': True}, - 'granted_permissions': {'readonly': True}, - 'authentication_type': {'required': True}, - 'account_id': {'readonly': True}, - 'aws_access_key_id': {'required': True}, - 'aws_secret_access_key': {'required': True}, - } - - _attribute_map = { - 'authentication_provisioning_state': {'key': 'authenticationProvisioningState', 'type': 'str'}, - 'granted_permissions': {'key': 'grantedPermissions', 'type': '[str]'}, - 'authentication_type': {'key': 'authenticationType', 'type': 'str'}, - 'account_id': {'key': 'accountId', 'type': 'str'}, - 'aws_access_key_id': {'key': 'awsAccessKeyId', 'type': 'str'}, - 'aws_secret_access_key': {'key': 'awsSecretAccessKey', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AwsCredsAuthenticationDetailsProperties, self).__init__(**kwargs) - self.authentication_type = 'awsCreds' # type: str - self.account_id = None - self.aws_access_key_id = kwargs['aws_access_key_id'] - self.aws_secret_access_key = kwargs['aws_secret_access_key'] - - -class ResourceDetails(msrest.serialization.Model): - """Details of the resource that was assessed. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: AzureResourceDetails, OnPremiseResourceDetails. - - All required parameters must be populated in order to send to Azure. - - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source - """ - - _validation = { - 'source': {'required': True}, - } - - _attribute_map = { - 'source': {'key': 'source', 'type': 'str'}, - } - - _subtype_map = { - 'source': {'Azure': 'AzureResourceDetails', 'OnPremise': 'OnPremiseResourceDetails'} - } - - def __init__( - self, - **kwargs - ): - super(ResourceDetails, self).__init__(**kwargs) - self.source = None # type: Optional[str] - - -class AzureResourceDetails(ResourceDetails): - """Details of the Azure resource that was assessed. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source - :ivar id: Azure resource Id of the assessed resource. - :vartype id: str - """ - - _validation = { - 'source': {'required': True}, - 'id': {'readonly': True}, - } - - _attribute_map = { - 'source': {'key': 'source', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AzureResourceDetails, self).__init__(**kwargs) - self.source = 'Azure' # type: str - self.id = None - - -class ResourceIdentifier(msrest.serialization.Model): - """A resource identifier for an alert which can be used to direct the alert to the right product exposure group (tenant, workspace, subscription etc.). - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: AzureResourceIdentifier, LogAnalyticsIdentifier. - - All required parameters must be populated in order to send to Azure. - - :param type: Required. There can be multiple identifiers of different type per alert, this - field specify the identifier type.Constant filled by server. Possible values include: - "AzureResource", "LogAnalytics". - :type type: str or ~azure.mgmt.security.models.ResourceIdentifierType - """ - - _validation = { - 'type': {'required': True}, - } - - _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - } - - _subtype_map = { - 'type': {'AzureResource': 'AzureResourceIdentifier', 'LogAnalytics': 'LogAnalyticsIdentifier'} - } - - def __init__( - self, - **kwargs - ): - super(ResourceIdentifier, self).__init__(**kwargs) - self.type = None # type: Optional[str] - - -class AzureResourceIdentifier(ResourceIdentifier): - """Azure resource identifier. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param type: Required. There can be multiple identifiers of different type per alert, this - field specify the identifier type.Constant filled by server. Possible values include: - "AzureResource", "LogAnalytics". - :type type: str or ~azure.mgmt.security.models.ResourceIdentifierType - :ivar azure_resource_id: ARM resource identifier for the cloud resource being alerted on. - :vartype azure_resource_id: str - """ - - _validation = { - 'type': {'required': True}, - 'azure_resource_id': {'readonly': True}, - } - - _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'azure_resource_id': {'key': 'azureResourceId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AzureResourceIdentifier, self).__init__(**kwargs) - self.type = 'AzureResource' # type: str - self.azure_resource_id = None - - -class AzureResourceLink(msrest.serialization.Model): - """Describes an Azure resource with kind. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Azure resource Id. - :vartype id: str - """ - - _validation = { - 'id': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(AzureResourceLink, self).__init__(**kwargs) - self.id = None - - -class Baseline(msrest.serialization.Model): - """Baseline details. - - :param expected_results: Expected results. - :type expected_results: list[list[str]] - :param updated_time: Baseline update time (UTC). - :type updated_time: ~datetime.datetime - """ - - _attribute_map = { - 'expected_results': {'key': 'expectedResults', 'type': '[[str]]'}, - 'updated_time': {'key': 'updatedTime', 'type': 'iso-8601'}, - } - - def __init__( - self, - **kwargs - ): - super(Baseline, self).__init__(**kwargs) - self.expected_results = kwargs.get('expected_results', None) - self.updated_time = kwargs.get('updated_time', None) - - -class BaselineAdjustedResult(msrest.serialization.Model): - """The rule result adjusted with baseline. - - :param baseline: Baseline details. - :type baseline: ~azure.mgmt.security.models.Baseline - :param status: The rule result status. Possible values include: "NonFinding", "Finding", - "InternalError". - :type status: str or ~azure.mgmt.security.models.RuleStatus - :param results_not_in_baseline: Results the are not in baseline. - :type results_not_in_baseline: list[list[str]] - :param results_only_in_baseline: Results the are in baseline. - :type results_only_in_baseline: list[list[str]] - """ - - _attribute_map = { - 'baseline': {'key': 'baseline', 'type': 'Baseline'}, - 'status': {'key': 'status', 'type': 'str'}, - 'results_not_in_baseline': {'key': 'resultsNotInBaseline', 'type': '[[str]]'}, - 'results_only_in_baseline': {'key': 'resultsOnlyInBaseline', 'type': '[[str]]'}, - } - - def __init__( - self, - **kwargs - ): - super(BaselineAdjustedResult, self).__init__(**kwargs) - self.baseline = kwargs.get('baseline', None) - self.status = kwargs.get('status', None) - self.results_not_in_baseline = kwargs.get('results_not_in_baseline', None) - self.results_only_in_baseline = kwargs.get('results_only_in_baseline', None) - - -class BenchmarkReference(msrest.serialization.Model): - """The benchmark references. - - :param benchmark: The benchmark name. - :type benchmark: str - :param reference: The benchmark reference. - :type reference: str - """ - - _attribute_map = { - 'benchmark': {'key': 'benchmark', 'type': 'str'}, - 'reference': {'key': 'reference', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(BenchmarkReference, self).__init__(**kwargs) - self.benchmark = kwargs.get('benchmark', None) - self.reference = kwargs.get('reference', None) - - -class CefExternalSecuritySolution(ExternalSecuritySolution): - """Represents a security solution which sends CEF logs to an OMS workspace. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: The external security solution properties for CEF solutions. - :type properties: ~azure.mgmt.security.models.CefSolutionProperties - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'kind': {'key': 'kind', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'CefSolutionProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(CefExternalSecuritySolution, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class CefSolutionProperties(ExternalSecuritySolutionProperties): - """The external security solution properties for CEF solutions. - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace - :param hostname: - :type hostname: str - :param agent: - :type agent: str - :param last_event_received: - :type last_event_received: str - """ - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'device_vendor': {'key': 'deviceVendor', 'type': 'str'}, - 'device_type': {'key': 'deviceType', 'type': 'str'}, - 'workspace': {'key': 'workspace', 'type': 'ConnectedWorkspace'}, - 'hostname': {'key': 'hostname', 'type': 'str'}, - 'agent': {'key': 'agent', 'type': 'str'}, - 'last_event_received': {'key': 'lastEventReceived', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CefSolutionProperties, self).__init__(**kwargs) - self.hostname = kwargs.get('hostname', None) - self.agent = kwargs.get('agent', None) - self.last_event_received = kwargs.get('last_event_received', None) - - -class CloudErrorBody(msrest.serialization.Model): - """The error detail. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar code: The error code. - :vartype code: str - :ivar message: The error message. - :vartype message: str - :ivar target: The error target. - :vartype target: str - :ivar details: The error details. - :vartype details: list[~azure.mgmt.security.models.CloudErrorBody] - :ivar additional_info: The error additional info. - :vartype additional_info: list[~azure.mgmt.security.models.ErrorAdditionalInfo] - """ - - _validation = { - 'code': {'readonly': True}, - 'message': {'readonly': True}, - 'target': {'readonly': True}, - 'details': {'readonly': True}, - 'additional_info': {'readonly': True}, - } - - _attribute_map = { - 'code': {'key': 'code', 'type': 'str'}, - 'message': {'key': 'message', 'type': 'str'}, - 'target': {'key': 'target', 'type': 'str'}, - 'details': {'key': 'details', 'type': '[CloudErrorBody]'}, - 'additional_info': {'key': 'additionalInfo', 'type': '[ErrorAdditionalInfo]'}, - } - - def __init__( - self, - **kwargs - ): - super(CloudErrorBody, self).__init__(**kwargs) - self.code = None - self.message = None - self.target = None - self.details = None - self.additional_info = None - - -class Compliance(Resource): - """Compliance of a scope. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar assessment_timestamp_utc_date: The timestamp when the Compliance calculation was - conducted. - :vartype assessment_timestamp_utc_date: ~datetime.datetime - :ivar resource_count: The resource count of the given subscription for which the Compliance - calculation was conducted (needed for Management Group Compliance calculation). - :vartype resource_count: int - :ivar assessment_result: An array of segment, which is the actually the compliance assessment. - :vartype assessment_result: list[~azure.mgmt.security.models.ComplianceSegment] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'assessment_timestamp_utc_date': {'readonly': True}, - 'resource_count': {'readonly': True}, - 'assessment_result': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'assessment_timestamp_utc_date': {'key': 'properties.assessmentTimestampUtcDate', 'type': 'iso-8601'}, - 'resource_count': {'key': 'properties.resourceCount', 'type': 'int'}, - 'assessment_result': {'key': 'properties.assessmentResult', 'type': '[ComplianceSegment]'}, - } - - def __init__( - self, - **kwargs - ): - super(Compliance, self).__init__(**kwargs) - self.assessment_timestamp_utc_date = None - self.resource_count = None - self.assessment_result = None - - -class ComplianceList(msrest.serialization.Model): - """List of Compliance objects response. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: List of Compliance objects. - :type value: list[~azure.mgmt.security.models.Compliance] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Compliance]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ComplianceList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class ComplianceResult(Resource): - """a compliance result. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar resource_status: The status of the resource regarding a single assessment. Possible - values include: "Healthy", "NotApplicable", "OffByPolicy", "NotHealthy". - :vartype resource_status: str or ~azure.mgmt.security.models.ResourceStatus - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'resource_status': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'resource_status': {'key': 'properties.resourceStatus', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ComplianceResult, self).__init__(**kwargs) - self.resource_status = None - - -class ComplianceResultList(msrest.serialization.Model): - """List of compliance results response. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of compliance results. - :type value: list[~azure.mgmt.security.models.ComplianceResult] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[ComplianceResult]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ComplianceResultList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class ComplianceSegment(msrest.serialization.Model): - """A segment of a compliance assessment. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar segment_type: The segment type, e.g. compliant, non-compliance, insufficient coverage, - N/A, etc. - :vartype segment_type: str - :ivar percentage: The size (%) of the segment. - :vartype percentage: float - """ - - _validation = { - 'segment_type': {'readonly': True}, - 'percentage': {'readonly': True}, - } - - _attribute_map = { - 'segment_type': {'key': 'segmentType', 'type': 'str'}, - 'percentage': {'key': 'percentage', 'type': 'float'}, - } - - def __init__( - self, - **kwargs - ): - super(ComplianceSegment, self).__init__(**kwargs) - self.segment_type = None - self.percentage = None - - -class ConnectableResource(msrest.serialization.Model): - """Describes the allowed inbound and outbound traffic of an Azure resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The Azure resource id. - :vartype id: str - :ivar inbound_connected_resources: The list of Azure resources that the resource has inbound - allowed connection from. - :vartype inbound_connected_resources: list[~azure.mgmt.security.models.ConnectedResource] - :ivar outbound_connected_resources: The list of Azure resources that the resource has outbound - allowed connection to. - :vartype outbound_connected_resources: list[~azure.mgmt.security.models.ConnectedResource] - """ - - _validation = { - 'id': {'readonly': True}, - 'inbound_connected_resources': {'readonly': True}, - 'outbound_connected_resources': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'inbound_connected_resources': {'key': 'inboundConnectedResources', 'type': '[ConnectedResource]'}, - 'outbound_connected_resources': {'key': 'outboundConnectedResources', 'type': '[ConnectedResource]'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectableResource, self).__init__(**kwargs) - self.id = None - self.inbound_connected_resources = None - self.outbound_connected_resources = None - - -class ConnectedResource(msrest.serialization.Model): - """Describes properties of a connected resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar connected_resource_id: The Azure resource id of the connected resource. - :vartype connected_resource_id: str - :ivar tcp_ports: The allowed tcp ports. - :vartype tcp_ports: str - :ivar udp_ports: The allowed udp ports. - :vartype udp_ports: str - """ - - _validation = { - 'connected_resource_id': {'readonly': True}, - 'tcp_ports': {'readonly': True}, - 'udp_ports': {'readonly': True}, - } - - _attribute_map = { - 'connected_resource_id': {'key': 'connectedResourceId', 'type': 'str'}, - 'tcp_ports': {'key': 'tcpPorts', 'type': 'str'}, - 'udp_ports': {'key': 'udpPorts', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectedResource, self).__init__(**kwargs) - self.connected_resource_id = None - self.tcp_ports = None - self.udp_ports = None - - -class ConnectedWorkspace(msrest.serialization.Model): - """Represents an OMS workspace to which the solution is connected. - - :param id: Azure resource ID of the connected OMS workspace. - :type id: str - """ - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectedWorkspace, self).__init__(**kwargs) - self.id = kwargs.get('id', None) - - -class ConnectionFromIpNotAllowed(AllowlistCustomAlertRule): - """Inbound connection from an ip that isn't allowed. Allow list consists of ipv4 or ipv6 range in CIDR notation. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the - rule type. - :type allowlist_values: list[str] - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'allowlist_values': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'allowlist_values': {'key': 'allowlistValues', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectionFromIpNotAllowed, self).__init__(**kwargs) - self.rule_type = 'ConnectionFromIpNotAllowed' # type: str - - -class ConnectionStrings(msrest.serialization.Model): - """Connection string for ingesting security data and logs. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. Connection strings. - :type value: list[~azure.mgmt.security.models.IngestionConnectionString] - """ - - _validation = { - 'value': {'required': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[IngestionConnectionString]'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectionStrings, self).__init__(**kwargs) - self.value = kwargs['value'] - - -class ConnectionToIpNotAllowed(AllowlistCustomAlertRule): - """Outbound connection to an ip that isn't allowed. Allow list consists of ipv4 or ipv6 range in CIDR notation. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the - rule type. - :type allowlist_values: list[str] - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'allowlist_values': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'allowlist_values': {'key': 'allowlistValues', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectionToIpNotAllowed, self).__init__(**kwargs) - self.rule_type = 'ConnectionToIpNotAllowed' # type: str - - -class ConnectorSetting(Resource): - """The connector setting. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param hybrid_compute_settings: Settings for hybrid compute management. These settings are - relevant only for Arc autoProvision (Hybrid Compute). - :type hybrid_compute_settings: ~azure.mgmt.security.models.HybridComputeSettingsProperties - :param authentication_details: Settings for authentication management, these settings are - relevant only for the cloud connector. - :type authentication_details: ~azure.mgmt.security.models.AuthenticationDetailsProperties - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'hybrid_compute_settings': {'key': 'properties.hybridComputeSettings', 'type': 'HybridComputeSettingsProperties'}, - 'authentication_details': {'key': 'properties.authenticationDetails', 'type': 'AuthenticationDetailsProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectorSetting, self).__init__(**kwargs) - self.hybrid_compute_settings = kwargs.get('hybrid_compute_settings', None) - self.authentication_details = kwargs.get('authentication_details', None) - - -class ConnectorSettingList(msrest.serialization.Model): - """For a subscription, list of all cloud account connectors and their settings. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: List of all the cloud account connector settings. - :type value: list[~azure.mgmt.security.models.ConnectorSetting] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[ConnectorSetting]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ConnectorSettingList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class ContainerRegistryVulnerabilityProperties(AdditionalData): - """Additional context fields for container registry Vulnerability assessment. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", - "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType - :ivar type: Vulnerability Type. e.g: Vulnerability, Potential Vulnerability, Information - Gathered, Vulnerability. - :vartype type: str - :ivar cvss: Dictionary from cvss version to cvss details object. - :vartype cvss: dict[str, ~azure.mgmt.security.models.CVSS] - :ivar patchable: Indicates whether a patch is available or not. - :vartype patchable: bool - :ivar cve: List of CVEs. - :vartype cve: list[~azure.mgmt.security.models.CVE] - :ivar published_time: Published time. - :vartype published_time: ~datetime.datetime - :ivar vendor_references: - :vartype vendor_references: list[~azure.mgmt.security.models.VendorReference] - :ivar repository_name: Name of the repository which the vulnerable image belongs to. - :vartype repository_name: str - :ivar image_digest: Digest of the vulnerable image. - :vartype image_digest: str - """ - - _validation = { - 'assessed_resource_type': {'required': True}, - 'type': {'readonly': True}, - 'cvss': {'readonly': True}, - 'patchable': {'readonly': True}, - 'cve': {'readonly': True}, - 'published_time': {'readonly': True}, - 'vendor_references': {'readonly': True}, - 'repository_name': {'readonly': True}, - 'image_digest': {'readonly': True}, - } - - _attribute_map = { - 'assessed_resource_type': {'key': 'assessedResourceType', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'cvss': {'key': 'cvss', 'type': '{CVSS}'}, - 'patchable': {'key': 'patchable', 'type': 'bool'}, - 'cve': {'key': 'cve', 'type': '[CVE]'}, - 'published_time': {'key': 'publishedTime', 'type': 'iso-8601'}, - 'vendor_references': {'key': 'vendorReferences', 'type': '[VendorReference]'}, - 'repository_name': {'key': 'repositoryName', 'type': 'str'}, - 'image_digest': {'key': 'imageDigest', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ContainerRegistryVulnerabilityProperties, self).__init__(**kwargs) - self.assessed_resource_type = 'ContainerRegistryVulnerability' # type: str - self.type = None - self.cvss = None - self.patchable = None - self.cve = None - self.published_time = None - self.vendor_references = None - self.repository_name = None - self.image_digest = None - - -class CVE(msrest.serialization.Model): - """CVE details. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar title: CVE title. - :vartype title: str - :ivar link: Link url. - :vartype link: str - """ - - _validation = { - 'title': {'readonly': True}, - 'link': {'readonly': True}, - } - - _attribute_map = { - 'title': {'key': 'title', 'type': 'str'}, - 'link': {'key': 'link', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CVE, self).__init__(**kwargs) - self.title = None - self.link = None - - -class CVSS(msrest.serialization.Model): - """CVSS details. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar base: CVSS base. - :vartype base: float - """ - - _validation = { - 'base': {'readonly': True}, - } - - _attribute_map = { - 'base': {'key': 'base', 'type': 'float'}, - } - - def __init__( - self, - **kwargs - ): - super(CVSS, self).__init__(**kwargs) - self.base = None - - -class DataExportSettings(Setting): - """Represents a data export setting. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param kind: Required. the kind of the settings string.Constant filled by server. Possible - values include: "DataExportSettings", "AlertSuppressionSetting", "AlertSyncSettings". - :type kind: str or ~azure.mgmt.security.models.SettingKind - :param enabled: Is the data export setting enabled. - :type enabled: bool - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'kind': {'required': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, - 'enabled': {'key': 'properties.enabled', 'type': 'bool'}, - } - - def __init__( - self, - **kwargs - ): - super(DataExportSettings, self).__init__(**kwargs) - self.kind = 'DataExportSettings' # type: str - self.enabled = kwargs.get('enabled', None) - - -class DenylistCustomAlertRule(ListCustomAlertRule): - """A custom alert rule that checks if a value (depends on the custom alert type) is denied. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param denylist_values: Required. The values to deny. The format of the values depends on the - rule type. - :type denylist_values: list[str] - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'denylist_values': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'denylist_values': {'key': 'denylistValues', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(DenylistCustomAlertRule, self).__init__(**kwargs) - self.rule_type = 'DenylistCustomAlertRule' # type: str - self.denylist_values = kwargs['denylist_values'] - - -class DeviceSecurityGroup(Resource): - """The device security group resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param threshold_rules: The list of custom alert threshold rules. - :type threshold_rules: list[~azure.mgmt.security.models.ThresholdCustomAlertRule] - :param time_window_rules: The list of custom alert time-window rules. - :type time_window_rules: list[~azure.mgmt.security.models.TimeWindowCustomAlertRule] - :param allowlist_rules: The allow-list custom alert rules. - :type allowlist_rules: list[~azure.mgmt.security.models.AllowlistCustomAlertRule] - :param denylist_rules: The deny-list custom alert rules. - :type denylist_rules: list[~azure.mgmt.security.models.DenylistCustomAlertRule] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'threshold_rules': {'key': 'properties.thresholdRules', 'type': '[ThresholdCustomAlertRule]'}, - 'time_window_rules': {'key': 'properties.timeWindowRules', 'type': '[TimeWindowCustomAlertRule]'}, - 'allowlist_rules': {'key': 'properties.allowlistRules', 'type': '[AllowlistCustomAlertRule]'}, - 'denylist_rules': {'key': 'properties.denylistRules', 'type': '[DenylistCustomAlertRule]'}, - } - - def __init__( - self, - **kwargs - ): - super(DeviceSecurityGroup, self).__init__(**kwargs) - self.threshold_rules = kwargs.get('threshold_rules', None) - self.time_window_rules = kwargs.get('time_window_rules', None) - self.allowlist_rules = kwargs.get('allowlist_rules', None) - self.denylist_rules = kwargs.get('denylist_rules', None) - - -class DeviceSecurityGroupList(msrest.serialization.Model): - """List of device security groups. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: List of device security group objects. - :type value: list[~azure.mgmt.security.models.DeviceSecurityGroup] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[DeviceSecurityGroup]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DeviceSecurityGroupList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class DirectMethodInvokesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of direct method invokes is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(DirectMethodInvokesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'DirectMethodInvokesNotInAllowedRange' # type: str - - -class DiscoveredSecuritySolution(Resource, Location): - """DiscoveredSecuritySolution. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param security_family: Required. The security family of the discovered solution. Possible - values include: "Waf", "Ngfw", "SaasWaf", "Va". - :type security_family: str or ~azure.mgmt.security.models.SecurityFamily - :param offer: Required. The security solutions' image offer. - :type offer: str - :param publisher: Required. The security solutions' image publisher. - :type publisher: str - :param sku: Required. The security solutions' image sku. - :type sku: str - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'security_family': {'required': True}, - 'offer': {'required': True}, - 'publisher': {'required': True}, - 'sku': {'required': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'security_family': {'key': 'properties.securityFamily', 'type': 'str'}, - 'offer': {'key': 'properties.offer', 'type': 'str'}, - 'publisher': {'key': 'properties.publisher', 'type': 'str'}, - 'sku': {'key': 'properties.sku', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DiscoveredSecuritySolution, self).__init__(**kwargs) - self.location = None - self.security_family = kwargs['security_family'] - self.offer = kwargs['offer'] - self.publisher = kwargs['publisher'] - self.sku = kwargs['sku'] - self.id = None - self.name = None - self.type = None - self.security_family = kwargs['security_family'] - self.offer = kwargs['offer'] - self.publisher = kwargs['publisher'] - self.sku = kwargs['sku'] - - -class DiscoveredSecuritySolutionList(msrest.serialization.Model): - """DiscoveredSecuritySolutionList. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: - :type value: list[~azure.mgmt.security.models.DiscoveredSecuritySolution] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[DiscoveredSecuritySolution]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DiscoveredSecuritySolutionList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class EffectiveNetworkSecurityGroups(msrest.serialization.Model): - """Describes the Network Security Groups effective on a network interface. - - :param network_interface: The Azure resource ID of the network interface. - :type network_interface: str - :param network_security_groups: The Network Security Groups effective on the network interface. - :type network_security_groups: list[str] - """ - - _attribute_map = { - 'network_interface': {'key': 'networkInterface', 'type': 'str'}, - 'network_security_groups': {'key': 'networkSecurityGroups', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(EffectiveNetworkSecurityGroups, self).__init__(**kwargs) - self.network_interface = kwargs.get('network_interface', None) - self.network_security_groups = kwargs.get('network_security_groups', None) - - -class ErrorAdditionalInfo(msrest.serialization.Model): - """The resource management error additional info. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar type: The additional info type. - :vartype type: str - :ivar info: The additional info. - :vartype info: any - """ - - _validation = { - 'type': {'readonly': True}, - 'info': {'readonly': True}, - } - - _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'info': {'key': 'info', 'type': 'object'}, - } - - def __init__( - self, - **kwargs - ): - super(ErrorAdditionalInfo, self).__init__(**kwargs) - self.type = None - self.info = None - - -class ExternalSecuritySolutionList(msrest.serialization.Model): - """ExternalSecuritySolutionList. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: - :type value: list[~azure.mgmt.security.models.ExternalSecuritySolution] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[ExternalSecuritySolution]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ExternalSecuritySolutionList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class FailedLocalLoginsNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of failed local logins is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(FailedLocalLoginsNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'FailedLocalLoginsNotInAllowedRange' # type: str - - -class FileUploadsNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of file uploads is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(FileUploadsNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'FileUploadsNotInAllowedRange' # type: str - - -class GcpCredentialsDetailsProperties(AuthenticationDetailsProperties): - """GCP cloud account connector based service to service credentials, the credentials are composed of the organization ID and a JSON API key (write only). - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar authentication_provisioning_state: State of the multi-cloud connector. Possible values - include: "Valid", "Invalid", "Expired", "IncorrectPolicy". - :vartype authentication_provisioning_state: str or - ~azure.mgmt.security.models.AuthenticationProvisioningState - :ivar granted_permissions: The permissions detected in the cloud account. - :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account - credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType - :param organization_id: Required. The organization ID of the GCP cloud account. - :type organization_id: str - :param type: Required. Type field of the API key (write only). - :type type: str - :param project_id: Required. Project ID field of the API key (write only). - :type project_id: str - :param private_key_id: Required. Private key ID field of the API key (write only). - :type private_key_id: str - :param private_key: Required. Private key field of the API key (write only). - :type private_key: str - :param client_email: Required. Client email field of the API key (write only). - :type client_email: str - :param client_id: Required. Client ID field of the API key (write only). - :type client_id: str - :param auth_uri: Required. Auth URI field of the API key (write only). - :type auth_uri: str - :param token_uri: Required. Token URI field of the API key (write only). - :type token_uri: str - :param auth_provider_x509_cert_url: Required. Auth provider x509 certificate URL field of the - API key (write only). - :type auth_provider_x509_cert_url: str - :param client_x509_cert_url: Required. Client x509 certificate URL field of the API key (write - only). - :type client_x509_cert_url: str - """ - - _validation = { - 'authentication_provisioning_state': {'readonly': True}, - 'granted_permissions': {'readonly': True}, - 'authentication_type': {'required': True}, - 'organization_id': {'required': True}, - 'type': {'required': True}, - 'project_id': {'required': True}, - 'private_key_id': {'required': True}, - 'private_key': {'required': True}, - 'client_email': {'required': True}, - 'client_id': {'required': True}, - 'auth_uri': {'required': True}, - 'token_uri': {'required': True}, - 'auth_provider_x509_cert_url': {'required': True}, - 'client_x509_cert_url': {'required': True}, - } - - _attribute_map = { - 'authentication_provisioning_state': {'key': 'authenticationProvisioningState', 'type': 'str'}, - 'granted_permissions': {'key': 'grantedPermissions', 'type': '[str]'}, - 'authentication_type': {'key': 'authenticationType', 'type': 'str'}, - 'organization_id': {'key': 'organizationId', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'project_id': {'key': 'projectId', 'type': 'str'}, - 'private_key_id': {'key': 'privateKeyId', 'type': 'str'}, - 'private_key': {'key': 'privateKey', 'type': 'str'}, - 'client_email': {'key': 'clientEmail', 'type': 'str'}, - 'client_id': {'key': 'clientId', 'type': 'str'}, - 'auth_uri': {'key': 'authUri', 'type': 'str'}, - 'token_uri': {'key': 'tokenUri', 'type': 'str'}, - 'auth_provider_x509_cert_url': {'key': 'authProviderX509CertUrl', 'type': 'str'}, - 'client_x509_cert_url': {'key': 'clientX509CertUrl', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(GcpCredentialsDetailsProperties, self).__init__(**kwargs) - self.authentication_type = 'gcpCredentials' # type: str - self.organization_id = kwargs['organization_id'] - self.type = kwargs['type'] - self.project_id = kwargs['project_id'] - self.private_key_id = kwargs['private_key_id'] - self.private_key = kwargs['private_key'] - self.client_email = kwargs['client_email'] - self.client_id = kwargs['client_id'] - self.auth_uri = kwargs['auth_uri'] - self.token_uri = kwargs['token_uri'] - self.auth_provider_x509_cert_url = kwargs['auth_provider_x509_cert_url'] - self.client_x509_cert_url = kwargs['client_x509_cert_url'] - - -class HttpC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of cloud to device messages (HTTP protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(HttpC2DMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'HttpC2DMessagesNotInAllowedRange' # type: str - - -class HttpC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of rejected cloud to device messages (HTTP protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(HttpC2DRejectedMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'HttpC2DRejectedMessagesNotInAllowedRange' # type: str - - -class HttpD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of device to cloud messages (HTTP protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(HttpD2CMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'HttpD2CMessagesNotInAllowedRange' # type: str - - -class HybridComputeSettingsProperties(msrest.serialization.Model): - """Settings for hybrid compute management. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar hybrid_compute_provisioning_state: State of the service principal and its secret. - Possible values include: "Valid", "Invalid", "Expired". - :vartype hybrid_compute_provisioning_state: str or - ~azure.mgmt.security.models.HybridComputeProvisioningState - :param auto_provision: Required. Whether or not to automatically install Azure Arc (hybrid - compute) agents on machines. Possible values include: "On", "Off". - :type auto_provision: str or ~azure.mgmt.security.models.AutoProvision - :param resource_group_name: The name of the resource group where Arc (Hybrid Compute) - connectors are connected. - :type resource_group_name: str - :param region: The location where the metadata of machines will be stored. - :type region: str - :param proxy_server: For a non-Azure machine that is not connected directly to the internet, - specify a proxy server that the non-Azure machine can use. - :type proxy_server: ~azure.mgmt.security.models.ProxyServerProperties - :param service_principal: An object to access resources that are secured by an Azure AD tenant. - :type service_principal: ~azure.mgmt.security.models.ServicePrincipalProperties - """ - - _validation = { - 'hybrid_compute_provisioning_state': {'readonly': True}, - 'auto_provision': {'required': True}, - } - - _attribute_map = { - 'hybrid_compute_provisioning_state': {'key': 'hybridComputeProvisioningState', 'type': 'str'}, - 'auto_provision': {'key': 'autoProvision', 'type': 'str'}, - 'resource_group_name': {'key': 'resourceGroupName', 'type': 'str'}, - 'region': {'key': 'region', 'type': 'str'}, - 'proxy_server': {'key': 'proxyServer', 'type': 'ProxyServerProperties'}, - 'service_principal': {'key': 'servicePrincipal', 'type': 'ServicePrincipalProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(HybridComputeSettingsProperties, self).__init__(**kwargs) - self.hybrid_compute_provisioning_state = None - self.auto_provision = kwargs['auto_provision'] - self.resource_group_name = kwargs.get('resource_group_name', None) - self.region = kwargs.get('region', None) - self.proxy_server = kwargs.get('proxy_server', None) - self.service_principal = kwargs.get('service_principal', None) - - -class InformationProtectionKeyword(msrest.serialization.Model): - """The information type keyword. - - :param pattern: The keyword pattern. - :type pattern: str - :param custom: Indicates whether the keyword is custom or not. - :type custom: bool - :param can_be_numeric: Indicates whether the keyword can be applied on numeric types or not. - :type can_be_numeric: bool - :param excluded: Indicates whether the keyword is excluded or not. - :type excluded: bool - """ - - _attribute_map = { - 'pattern': {'key': 'pattern', 'type': 'str'}, - 'custom': {'key': 'custom', 'type': 'bool'}, - 'can_be_numeric': {'key': 'canBeNumeric', 'type': 'bool'}, - 'excluded': {'key': 'excluded', 'type': 'bool'}, - } - - def __init__( - self, - **kwargs - ): - super(InformationProtectionKeyword, self).__init__(**kwargs) - self.pattern = kwargs.get('pattern', None) - self.custom = kwargs.get('custom', None) - self.can_be_numeric = kwargs.get('can_be_numeric', None) - self.excluded = kwargs.get('excluded', None) - - -class InformationProtectionPolicy(Resource): - """Information protection policy. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar last_modified_utc: Describes the last UTC time the policy was modified. - :vartype last_modified_utc: ~datetime.datetime - :ivar version: Describes the version of the policy. - :vartype version: str - :param labels: Dictionary of sensitivity labels. - :type labels: dict[str, ~azure.mgmt.security.models.SensitivityLabel] - :param information_types: The sensitivity information types. - :type information_types: dict[str, ~azure.mgmt.security.models.InformationType] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'last_modified_utc': {'readonly': True}, - 'version': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'last_modified_utc': {'key': 'properties.lastModifiedUtc', 'type': 'iso-8601'}, - 'version': {'key': 'properties.version', 'type': 'str'}, - 'labels': {'key': 'properties.labels', 'type': '{SensitivityLabel}'}, - 'information_types': {'key': 'properties.informationTypes', 'type': '{InformationType}'}, - } - - def __init__( - self, - **kwargs - ): - super(InformationProtectionPolicy, self).__init__(**kwargs) - self.last_modified_utc = None - self.version = None - self.labels = kwargs.get('labels', None) - self.information_types = kwargs.get('information_types', None) - - -class InformationProtectionPolicyList(msrest.serialization.Model): - """Information protection policies response. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: List of information protection policies. - :type value: list[~azure.mgmt.security.models.InformationProtectionPolicy] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[InformationProtectionPolicy]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(InformationProtectionPolicyList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class InformationType(msrest.serialization.Model): - """The information type. - - :param display_name: The name of the information type. - :type display_name: str - :param description: The description of the information type. - :type description: str - :param order: The order of the information type. - :type order: int - :param recommended_label_id: The recommended label id to be associated with this information - type. - :type recommended_label_id: str - :param enabled: Indicates whether the information type is enabled or not. - :type enabled: bool - :param custom: Indicates whether the information type is custom or not. - :type custom: bool - :param keywords: The information type keywords. - :type keywords: list[~azure.mgmt.security.models.InformationProtectionKeyword] - """ - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'order': {'key': 'order', 'type': 'int'}, - 'recommended_label_id': {'key': 'recommendedLabelId', 'type': 'str'}, - 'enabled': {'key': 'enabled', 'type': 'bool'}, - 'custom': {'key': 'custom', 'type': 'bool'}, - 'keywords': {'key': 'keywords', 'type': '[InformationProtectionKeyword]'}, - } - - def __init__( - self, - **kwargs - ): - super(InformationType, self).__init__(**kwargs) - self.display_name = kwargs.get('display_name', None) - self.description = kwargs.get('description', None) - self.order = kwargs.get('order', None) - self.recommended_label_id = kwargs.get('recommended_label_id', None) - self.enabled = kwargs.get('enabled', None) - self.custom = kwargs.get('custom', None) - self.keywords = kwargs.get('keywords', None) - - -class IngestionConnectionString(msrest.serialization.Model): - """Connection string for ingesting security data and logs. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar location: The region where ingested logs and data resides. - :vartype location: str - :ivar value: Connection string value. - :vartype value: str - """ - - _validation = { - 'location': {'readonly': True}, - 'value': {'readonly': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'value': {'key': 'value', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IngestionConnectionString, self).__init__(**kwargs) - self.location = None - self.value = None - - -class IngestionSetting(Resource): - """Configures how to correlate scan data and logs with resources associated with the subscription. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: Ingestion setting data. - :type properties: any - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'object'}, - } - - def __init__( - self, - **kwargs - ): - super(IngestionSetting, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class IngestionSettingList(msrest.serialization.Model): - """List of ingestion settings. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: List of ingestion settings. - :vartype value: list[~azure.mgmt.security.models.IngestionSetting] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[IngestionSetting]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IngestionSettingList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class IngestionSettingToken(msrest.serialization.Model): - """Configures how to correlate scan data and logs with resources associated with the subscription. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar token: The token is used for correlating security data and logs with the resources in the - subscription. - :vartype token: str - """ - - _validation = { - 'token': {'readonly': True}, - } - - _attribute_map = { - 'token': {'key': 'token', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IngestionSettingToken, self).__init__(**kwargs) - self.token = None - - -class TagsResource(msrest.serialization.Model): - """A container holding only the Tags for a resource, allowing the user to update the tags. - - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - """ - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - } - - def __init__( - self, - **kwargs - ): - super(TagsResource, self).__init__(**kwargs) - self.tags = kwargs.get('tags', None) - - -class IoTSecurityAggregatedAlert(Resource, TagsResource): - """Security Solution Aggregated Alert information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar alert_type: Name of the alert type. - :vartype alert_type: str - :ivar alert_display_name: Display name of the alert type. - :vartype alert_display_name: str - :ivar aggregated_date_utc: Date of detection. - :vartype aggregated_date_utc: ~datetime.date - :ivar vendor_name: Name of the organization that raised the alert. - :vartype vendor_name: str - :ivar reported_severity: Assessed alert severity. Possible values include: "Informational", - "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar remediation_steps: Recommended steps for remediation. - :vartype remediation_steps: str - :ivar description: Description of the suspected vulnerability and meaning. - :vartype description: str - :ivar count: Number of alerts occurrences within the aggregated time window. - :vartype count: long - :ivar effected_resource_type: Azure resource ID of the resource that received the alerts. - :vartype effected_resource_type: str - :ivar system_source: The type of the alerted resource (Azure, Non-Azure). - :vartype system_source: str - :ivar action_taken: IoT Security solution alert response. - :vartype action_taken: str - :ivar log_analytics_query: Log analytics query for getting the list of affected devices/alerts. - :vartype log_analytics_query: str - :ivar top_devices_list: 10 devices with the highest number of occurrences of this alert type, - on this day. - :vartype top_devices_list: - list[~azure.mgmt.security.models.IoTSecurityAggregatedAlertPropertiesTopDevicesListItem] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'alert_type': {'readonly': True}, - 'alert_display_name': {'readonly': True}, - 'aggregated_date_utc': {'readonly': True}, - 'vendor_name': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'remediation_steps': {'readonly': True}, - 'description': {'readonly': True}, - 'count': {'readonly': True}, - 'effected_resource_type': {'readonly': True}, - 'system_source': {'readonly': True}, - 'action_taken': {'readonly': True}, - 'log_analytics_query': {'readonly': True}, - 'top_devices_list': {'readonly': True}, - } - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'alert_type': {'key': 'properties.alertType', 'type': 'str'}, - 'alert_display_name': {'key': 'properties.alertDisplayName', 'type': 'str'}, - 'aggregated_date_utc': {'key': 'properties.aggregatedDateUtc', 'type': 'date'}, - 'vendor_name': {'key': 'properties.vendorName', 'type': 'str'}, - 'reported_severity': {'key': 'properties.reportedSeverity', 'type': 'str'}, - 'remediation_steps': {'key': 'properties.remediationSteps', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'count': {'key': 'properties.count', 'type': 'long'}, - 'effected_resource_type': {'key': 'properties.effectedResourceType', 'type': 'str'}, - 'system_source': {'key': 'properties.systemSource', 'type': 'str'}, - 'action_taken': {'key': 'properties.actionTaken', 'type': 'str'}, - 'log_analytics_query': {'key': 'properties.logAnalyticsQuery', 'type': 'str'}, - 'top_devices_list': {'key': 'properties.topDevicesList', 'type': '[IoTSecurityAggregatedAlertPropertiesTopDevicesListItem]'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityAggregatedAlert, self).__init__(**kwargs) - self.tags = kwargs.get('tags', None) - self.alert_type = None - self.alert_display_name = None - self.aggregated_date_utc = None - self.vendor_name = None - self.reported_severity = None - self.remediation_steps = None - self.description = None - self.count = None - self.effected_resource_type = None - self.system_source = None - self.action_taken = None - self.log_analytics_query = None - self.top_devices_list = None - self.id = None - self.name = None - self.type = None - self.alert_type = None - self.alert_display_name = None - self.aggregated_date_utc = None - self.vendor_name = None - self.reported_severity = None - self.remediation_steps = None - self.description = None - self.count = None - self.effected_resource_type = None - self.system_source = None - self.action_taken = None - self.log_analytics_query = None - self.top_devices_list = None - - -class IoTSecurityAggregatedAlertList(msrest.serialization.Model): - """List of IoT Security solution aggregated alert data. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of aggregated alerts data. - :type value: list[~azure.mgmt.security.models.IoTSecurityAggregatedAlert] - :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next - page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecurityAggregatedAlert]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityAggregatedAlertList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class IoTSecurityAggregatedAlertPropertiesTopDevicesListItem(msrest.serialization.Model): - """IoTSecurityAggregatedAlertPropertiesTopDevicesListItem. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar device_id: Name of the device. - :vartype device_id: str - :ivar alerts_count: Number of alerts raised for this device. - :vartype alerts_count: long - :ivar last_occurrence: Most recent time this alert was raised for this device, on this day. - :vartype last_occurrence: str - """ - - _validation = { - 'device_id': {'readonly': True}, - 'alerts_count': {'readonly': True}, - 'last_occurrence': {'readonly': True}, - } - - _attribute_map = { - 'device_id': {'key': 'deviceId', 'type': 'str'}, - 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, - 'last_occurrence': {'key': 'lastOccurrence', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityAggregatedAlertPropertiesTopDevicesListItem, self).__init__(**kwargs) - self.device_id = None - self.alerts_count = None - self.last_occurrence = None - - -class IoTSecurityAggregatedRecommendation(Resource, TagsResource): - """IoT Security solution recommendation information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param recommendation_name: Name of the recommendation. - :type recommendation_name: str - :ivar recommendation_display_name: Display name of the recommendation type. - :vartype recommendation_display_name: str - :ivar description: Description of the suspected vulnerability and meaning. - :vartype description: str - :ivar recommendation_type_id: Recommendation-type GUID. - :vartype recommendation_type_id: str - :ivar detected_by: Name of the organization that made the recommendation. - :vartype detected_by: str - :ivar remediation_steps: Recommended steps for remediation. - :vartype remediation_steps: str - :ivar reported_severity: Assessed recommendation severity. Possible values include: - "Informational", "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar healthy_devices: Number of healthy devices within the IoT Security solution. - :vartype healthy_devices: long - :ivar unhealthy_device_count: Number of unhealthy devices within the IoT Security solution. - :vartype unhealthy_device_count: long - :ivar log_analytics_query: Log analytics query for getting the list of affected devices/alerts. - :vartype log_analytics_query: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'recommendation_display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'recommendation_type_id': {'readonly': True}, - 'detected_by': {'readonly': True}, - 'remediation_steps': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'healthy_devices': {'readonly': True}, - 'unhealthy_device_count': {'readonly': True}, - 'log_analytics_query': {'readonly': True}, - } - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'recommendation_name': {'key': 'properties.recommendationName', 'type': 'str'}, - 'recommendation_display_name': {'key': 'properties.recommendationDisplayName', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'recommendation_type_id': {'key': 'properties.recommendationTypeId', 'type': 'str'}, - 'detected_by': {'key': 'properties.detectedBy', 'type': 'str'}, - 'remediation_steps': {'key': 'properties.remediationSteps', 'type': 'str'}, - 'reported_severity': {'key': 'properties.reportedSeverity', 'type': 'str'}, - 'healthy_devices': {'key': 'properties.healthyDevices', 'type': 'long'}, - 'unhealthy_device_count': {'key': 'properties.unhealthyDeviceCount', 'type': 'long'}, - 'log_analytics_query': {'key': 'properties.logAnalyticsQuery', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityAggregatedRecommendation, self).__init__(**kwargs) - self.tags = kwargs.get('tags', None) - self.recommendation_name = kwargs.get('recommendation_name', None) - self.recommendation_display_name = None - self.description = None - self.recommendation_type_id = None - self.detected_by = None - self.remediation_steps = None - self.reported_severity = None - self.healthy_devices = None - self.unhealthy_device_count = None - self.log_analytics_query = None - self.id = None - self.name = None - self.type = None - self.recommendation_name = kwargs.get('recommendation_name', None) - self.recommendation_display_name = None - self.description = None - self.recommendation_type_id = None - self.detected_by = None - self.remediation_steps = None - self.reported_severity = None - self.healthy_devices = None - self.unhealthy_device_count = None - self.log_analytics_query = None - - -class IoTSecurityAggregatedRecommendationList(msrest.serialization.Model): - """List of IoT Security solution aggregated recommendations. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of aggregated recommendations data. - :type value: list[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendation] - :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next - page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecurityAggregatedRecommendation]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityAggregatedRecommendationList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class IoTSecurityAlertedDevice(msrest.serialization.Model): - """Statistical information about the number of alerts per device during last set number of days. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar device_id: Device identifier. - :vartype device_id: str - :ivar alerts_count: Number of alerts raised for this device. - :vartype alerts_count: long - """ - - _validation = { - 'device_id': {'readonly': True}, - 'alerts_count': {'readonly': True}, - } - - _attribute_map = { - 'device_id': {'key': 'deviceId', 'type': 'str'}, - 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityAlertedDevice, self).__init__(**kwargs) - self.device_id = None - self.alerts_count = None - - -class IoTSecurityDeviceAlert(msrest.serialization.Model): - """Statistical information about the number of alerts per alert type during last set number of days. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar alert_display_name: Display name of the alert. - :vartype alert_display_name: str - :ivar reported_severity: Assessed Alert severity. Possible values include: "Informational", - "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar alerts_count: Number of alerts raised for this alert type. - :vartype alerts_count: long - """ - - _validation = { - 'alert_display_name': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'alerts_count': {'readonly': True}, - } - - _attribute_map = { - 'alert_display_name': {'key': 'alertDisplayName', 'type': 'str'}, - 'reported_severity': {'key': 'reportedSeverity', 'type': 'str'}, - 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityDeviceAlert, self).__init__(**kwargs) - self.alert_display_name = None - self.reported_severity = None - self.alerts_count = None - - -class IoTSecurityDeviceRecommendation(msrest.serialization.Model): - """Statistical information about the number of recommendations per device, per recommendation type. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar recommendation_display_name: Display name of the recommendation. - :vartype recommendation_display_name: str - :ivar reported_severity: Assessed recommendation severity. Possible values include: - "Informational", "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar devices_count: Number of devices with this recommendation. - :vartype devices_count: long - """ - - _validation = { - 'recommendation_display_name': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'devices_count': {'readonly': True}, - } - - _attribute_map = { - 'recommendation_display_name': {'key': 'recommendationDisplayName', 'type': 'str'}, - 'reported_severity': {'key': 'reportedSeverity', 'type': 'str'}, - 'devices_count': {'key': 'devicesCount', 'type': 'long'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecurityDeviceRecommendation, self).__init__(**kwargs) - self.recommendation_display_name = None - self.reported_severity = None - self.devices_count = None - - -class IoTSecuritySolutionAnalyticsModel(Resource): - """Security analytics of your IoT Security solution. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar metrics: Security analytics of your IoT Security solution. - :vartype metrics: ~azure.mgmt.security.models.IoTSeverityMetrics - :ivar unhealthy_device_count: Number of unhealthy devices within your IoT Security solution. - :vartype unhealthy_device_count: long - :ivar devices_metrics: List of device metrics by the aggregation date. - :vartype devices_metrics: - list[~azure.mgmt.security.models.IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem] - :param top_alerted_devices: List of the 3 devices with the most alerts. - :type top_alerted_devices: list[~azure.mgmt.security.models.IoTSecurityAlertedDevice] - :param most_prevalent_device_alerts: List of the 3 most prevalent device alerts. - :type most_prevalent_device_alerts: list[~azure.mgmt.security.models.IoTSecurityDeviceAlert] - :param most_prevalent_device_recommendations: List of the 3 most prevalent device - recommendations. - :type most_prevalent_device_recommendations: - list[~azure.mgmt.security.models.IoTSecurityDeviceRecommendation] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'metrics': {'readonly': True}, - 'unhealthy_device_count': {'readonly': True}, - 'devices_metrics': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'metrics': {'key': 'properties.metrics', 'type': 'IoTSeverityMetrics'}, - 'unhealthy_device_count': {'key': 'properties.unhealthyDeviceCount', 'type': 'long'}, - 'devices_metrics': {'key': 'properties.devicesMetrics', 'type': '[IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem]'}, - 'top_alerted_devices': {'key': 'properties.topAlertedDevices', 'type': '[IoTSecurityAlertedDevice]'}, - 'most_prevalent_device_alerts': {'key': 'properties.mostPrevalentDeviceAlerts', 'type': '[IoTSecurityDeviceAlert]'}, - 'most_prevalent_device_recommendations': {'key': 'properties.mostPrevalentDeviceRecommendations', 'type': '[IoTSecurityDeviceRecommendation]'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecuritySolutionAnalyticsModel, self).__init__(**kwargs) - self.metrics = None - self.unhealthy_device_count = None - self.devices_metrics = None - self.top_alerted_devices = kwargs.get('top_alerted_devices', None) - self.most_prevalent_device_alerts = kwargs.get('most_prevalent_device_alerts', None) - self.most_prevalent_device_recommendations = kwargs.get('most_prevalent_device_recommendations', None) - - -class IoTSecuritySolutionAnalyticsModelList(msrest.serialization.Model): - """List of Security analytics of your IoT Security solution. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of Security analytics of your IoT Security solution. - :type value: list[~azure.mgmt.security.models.IoTSecuritySolutionAnalyticsModel] - :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next - page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecuritySolutionAnalyticsModel]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecuritySolutionAnalyticsModelList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem(msrest.serialization.Model): - """IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem. - - :param date: Aggregation of IoT Security solution device alert metrics by date. - :type date: ~datetime.datetime - :param devices_metrics: Device alert count by severity. - :type devices_metrics: ~azure.mgmt.security.models.IoTSeverityMetrics - """ - - _attribute_map = { - 'date': {'key': 'date', 'type': 'iso-8601'}, - 'devices_metrics': {'key': 'devicesMetrics', 'type': 'IoTSeverityMetrics'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem, self).__init__(**kwargs) - self.date = kwargs.get('date', None) - self.devices_metrics = kwargs.get('devices_metrics', None) - - -class IoTSecuritySolutionModel(Resource, TagsResource): - """IoT Security solution configuration and resource information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param location: The resource location. - :type location: str - :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy - information. - :vartype system_data: ~azure.mgmt.security.models.SystemData - :param workspace: Workspace resource ID. - :type workspace: str - :param display_name: Resource display name. - :type display_name: str - :param status: Status of the IoT Security solution. Possible values include: "Enabled", - "Disabled". Default value: "Enabled". - :type status: str or ~azure.mgmt.security.models.SecuritySolutionStatus - :param export: List of additional options for exporting to workspace data. - :type export: list[str or ~azure.mgmt.security.models.ExportData] - :param disabled_data_sources: Disabled data sources. Disabling these data sources compromises - the system. - :type disabled_data_sources: list[str or ~azure.mgmt.security.models.DataSource] - :param iot_hubs: IoT Hub resource IDs. - :type iot_hubs: list[str] - :param user_defined_resources: Properties of the IoT Security solution's user defined - resources. - :type user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties - :ivar auto_discovered_resources: List of resources that were automatically discovered as - relevant to the security solution. - :vartype auto_discovered_resources: list[str] - :param recommendations_configuration: List of the configuration status for each recommendation - type. - :type recommendations_configuration: - list[~azure.mgmt.security.models.RecommendationConfigurationProperties] - :param unmasked_ip_logging_status: Unmasked IP address logging status. Possible values include: - "Disabled", "Enabled". Default value: "Disabled". - :type unmasked_ip_logging_status: str or ~azure.mgmt.security.models.UnmaskedIpLoggingStatus - :param additional_workspaces: List of additional workspaces. - :type additional_workspaces: list[~azure.mgmt.security.models.AdditionalWorkspacesProperties] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'system_data': {'readonly': True}, - 'auto_discovered_resources': {'readonly': True}, - } - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'system_data': {'key': 'systemData', 'type': 'SystemData'}, - 'workspace': {'key': 'properties.workspace', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'status': {'key': 'properties.status', 'type': 'str'}, - 'export': {'key': 'properties.export', 'type': '[str]'}, - 'disabled_data_sources': {'key': 'properties.disabledDataSources', 'type': '[str]'}, - 'iot_hubs': {'key': 'properties.iotHubs', 'type': '[str]'}, - 'user_defined_resources': {'key': 'properties.userDefinedResources', 'type': 'UserDefinedResourcesProperties'}, - 'auto_discovered_resources': {'key': 'properties.autoDiscoveredResources', 'type': '[str]'}, - 'recommendations_configuration': {'key': 'properties.recommendationsConfiguration', 'type': '[RecommendationConfigurationProperties]'}, - 'unmasked_ip_logging_status': {'key': 'properties.unmaskedIpLoggingStatus', 'type': 'str'}, - 'additional_workspaces': {'key': 'properties.additionalWorkspaces', 'type': '[AdditionalWorkspacesProperties]'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecuritySolutionModel, self).__init__(**kwargs) - self.tags = kwargs.get('tags', None) - self.location = kwargs.get('location', None) - self.system_data = None - self.workspace = kwargs.get('workspace', None) - self.display_name = kwargs.get('display_name', None) - self.status = kwargs.get('status', "Enabled") - self.export = kwargs.get('export', None) - self.disabled_data_sources = kwargs.get('disabled_data_sources', None) - self.iot_hubs = kwargs.get('iot_hubs', None) - self.user_defined_resources = kwargs.get('user_defined_resources', None) - self.auto_discovered_resources = None - self.recommendations_configuration = kwargs.get('recommendations_configuration', None) - self.unmasked_ip_logging_status = kwargs.get('unmasked_ip_logging_status', "Disabled") - self.additional_workspaces = kwargs.get('additional_workspaces', None) - self.id = None - self.name = None - self.type = None - self.location = kwargs.get('location', None) - self.system_data = None - self.workspace = kwargs.get('workspace', None) - self.display_name = kwargs.get('display_name', None) - self.status = kwargs.get('status', "Enabled") - self.export = kwargs.get('export', None) - self.disabled_data_sources = kwargs.get('disabled_data_sources', None) - self.iot_hubs = kwargs.get('iot_hubs', None) - self.user_defined_resources = kwargs.get('user_defined_resources', None) - self.auto_discovered_resources = None - self.recommendations_configuration = kwargs.get('recommendations_configuration', None) - self.unmasked_ip_logging_status = kwargs.get('unmasked_ip_logging_status', "Disabled") - self.additional_workspaces = kwargs.get('additional_workspaces', None) - - -class IoTSecuritySolutionsList(msrest.serialization.Model): - """List of IoT Security solutions. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of IoT Security solutions. - :type value: list[~azure.mgmt.security.models.IoTSecuritySolutionModel] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecuritySolutionModel]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSecuritySolutionsList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class IoTSeverityMetrics(msrest.serialization.Model): - """IoT Security solution analytics severity metrics. - - :param high: Count of high severity alerts/recommendations. - :type high: long - :param medium: Count of medium severity alerts/recommendations. - :type medium: long - :param low: Count of low severity alerts/recommendations. - :type low: long - """ - - _attribute_map = { - 'high': {'key': 'high', 'type': 'long'}, - 'medium': {'key': 'medium', 'type': 'long'}, - 'low': {'key': 'low', 'type': 'long'}, - } - - def __init__( - self, - **kwargs - ): - super(IoTSeverityMetrics, self).__init__(**kwargs) - self.high = kwargs.get('high', None) - self.medium = kwargs.get('medium', None) - self.low = kwargs.get('low', None) - - -class JitNetworkAccessPoliciesList(msrest.serialization.Model): - """JitNetworkAccessPoliciesList. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: - :type value: list[~azure.mgmt.security.models.JitNetworkAccessPolicy] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[JitNetworkAccessPolicy]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessPoliciesList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class JitNetworkAccessPolicy(Resource, Kind, Location): - """JitNetworkAccessPolicy. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar location: Location where the resource is stored. - :vartype location: str - :param kind: Kind of the resource. - :type kind: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param virtual_machines: Required. Configurations for Microsoft.Compute/virtualMachines - resource type. - :type virtual_machines: list[~azure.mgmt.security.models.JitNetworkAccessPolicyVirtualMachine] - :param requests: - :type requests: list[~azure.mgmt.security.models.JitNetworkAccessRequest] - :ivar provisioning_state: Gets the provisioning state of the Just-in-Time policy. - :vartype provisioning_state: str - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'virtual_machines': {'required': True}, - 'provisioning_state': {'readonly': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'virtual_machines': {'key': 'properties.virtualMachines', 'type': '[JitNetworkAccessPolicyVirtualMachine]'}, - 'requests': {'key': 'properties.requests', 'type': '[JitNetworkAccessRequest]'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessPolicy, self).__init__(**kwargs) - self.location = None - self.kind = kwargs.get('kind', None) - self.virtual_machines = kwargs['virtual_machines'] - self.requests = kwargs.get('requests', None) - self.provisioning_state = None - self.location = None - self.id = None - self.name = None - self.type = None - self.virtual_machines = kwargs['virtual_machines'] - self.requests = kwargs.get('requests', None) - self.provisioning_state = None - self.kind = kwargs.get('kind', None) - self.id = None - self.name = None - self.type = None - self.virtual_machines = kwargs['virtual_machines'] - self.requests = kwargs.get('requests', None) - self.provisioning_state = None - - -class JitNetworkAccessPolicyInitiatePort(msrest.serialization.Model): - """JitNetworkAccessPolicyInitiatePort. - - All required parameters must be populated in order to send to Azure. - - :param number: Required. - :type number: int - :param allowed_source_address_prefix: Source of the allowed traffic. If omitted, the request - will be for the source IP address of the initiate request. - :type allowed_source_address_prefix: str - :param end_time_utc: Required. The time to close the request in UTC. - :type end_time_utc: ~datetime.datetime - """ - - _validation = { - 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, - 'end_time_utc': {'required': True}, - } - - _attribute_map = { - 'number': {'key': 'number', 'type': 'int'}, - 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, - 'end_time_utc': {'key': 'endTimeUtc', 'type': 'iso-8601'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessPolicyInitiatePort, self).__init__(**kwargs) - self.number = kwargs['number'] - self.allowed_source_address_prefix = kwargs.get('allowed_source_address_prefix', None) - self.end_time_utc = kwargs['end_time_utc'] - - -class JitNetworkAccessPolicyInitiateRequest(msrest.serialization.Model): - """JitNetworkAccessPolicyInitiateRequest. - - All required parameters must be populated in order to send to Azure. - - :param virtual_machines: Required. A list of virtual machines & ports to open access for. - :type virtual_machines: - list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiateVirtualMachine] - :param justification: The justification for making the initiate request. - :type justification: str - """ - - _validation = { - 'virtual_machines': {'required': True}, - } - - _attribute_map = { - 'virtual_machines': {'key': 'virtualMachines', 'type': '[JitNetworkAccessPolicyInitiateVirtualMachine]'}, - 'justification': {'key': 'justification', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessPolicyInitiateRequest, self).__init__(**kwargs) - self.virtual_machines = kwargs['virtual_machines'] - self.justification = kwargs.get('justification', None) - - -class JitNetworkAccessPolicyInitiateVirtualMachine(msrest.serialization.Model): - """JitNetworkAccessPolicyInitiateVirtualMachine. - - All required parameters must be populated in order to send to Azure. - - :param id: Required. Resource ID of the virtual machine that is linked to this policy. - :type id: str - :param ports: Required. The ports to open for the resource with the ``id``. - :type ports: list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiatePort] - """ - - _validation = { - 'id': {'required': True}, - 'ports': {'required': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'ports': {'key': 'ports', 'type': '[JitNetworkAccessPolicyInitiatePort]'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessPolicyInitiateVirtualMachine, self).__init__(**kwargs) - self.id = kwargs['id'] - self.ports = kwargs['ports'] - - -class JitNetworkAccessPolicyVirtualMachine(msrest.serialization.Model): - """JitNetworkAccessPolicyVirtualMachine. - - All required parameters must be populated in order to send to Azure. - - :param id: Required. Resource ID of the virtual machine that is linked to this policy. - :type id: str - :param ports: Required. Port configurations for the virtual machine. - :type ports: list[~azure.mgmt.security.models.JitNetworkAccessPortRule] - :param public_ip_address: Public IP address of the Azure Firewall that is linked to this - policy, if applicable. - :type public_ip_address: str - """ - - _validation = { - 'id': {'required': True}, - 'ports': {'required': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'ports': {'key': 'ports', 'type': '[JitNetworkAccessPortRule]'}, - 'public_ip_address': {'key': 'publicIpAddress', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessPolicyVirtualMachine, self).__init__(**kwargs) - self.id = kwargs['id'] - self.ports = kwargs['ports'] - self.public_ip_address = kwargs.get('public_ip_address', None) - - -class JitNetworkAccessPortRule(msrest.serialization.Model): - """JitNetworkAccessPortRule. - - All required parameters must be populated in order to send to Azure. - - :param number: Required. - :type number: int - :param protocol: Required. Possible values include: "TCP", "UDP", "*". - :type protocol: str or ~azure.mgmt.security.models.ProtocolEnum - :param allowed_source_address_prefix: Mutually exclusive with the - "allowedSourceAddressPrefixes" parameter. Should be an IP address or CIDR, for example - "192.168.0.3" or "192.168.0.0/16". - :type allowed_source_address_prefix: str - :param allowed_source_address_prefixes: Mutually exclusive with the - "allowedSourceAddressPrefix" parameter. - :type allowed_source_address_prefixes: list[str] - :param max_request_access_duration: Required. Maximum duration requests can be made for. In ISO - 8601 duration format. Minimum 5 minutes, maximum 1 day. - :type max_request_access_duration: str - """ - - _validation = { - 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, - 'protocol': {'required': True}, - 'max_request_access_duration': {'required': True}, - } - - _attribute_map = { - 'number': {'key': 'number', 'type': 'int'}, - 'protocol': {'key': 'protocol', 'type': 'str'}, - 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, - 'allowed_source_address_prefixes': {'key': 'allowedSourceAddressPrefixes', 'type': '[str]'}, - 'max_request_access_duration': {'key': 'maxRequestAccessDuration', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessPortRule, self).__init__(**kwargs) - self.number = kwargs['number'] - self.protocol = kwargs['protocol'] - self.allowed_source_address_prefix = kwargs.get('allowed_source_address_prefix', None) - self.allowed_source_address_prefixes = kwargs.get('allowed_source_address_prefixes', None) - self.max_request_access_duration = kwargs['max_request_access_duration'] - - -class JitNetworkAccessRequest(msrest.serialization.Model): - """JitNetworkAccessRequest. - - All required parameters must be populated in order to send to Azure. - - :param virtual_machines: Required. - :type virtual_machines: list[~azure.mgmt.security.models.JitNetworkAccessRequestVirtualMachine] - :param start_time_utc: Required. The start time of the request in UTC. - :type start_time_utc: ~datetime.datetime - :param requestor: Required. The identity of the person who made the request. - :type requestor: str - :param justification: The justification for making the initiate request. - :type justification: str - """ - - _validation = { - 'virtual_machines': {'required': True}, - 'start_time_utc': {'required': True}, - 'requestor': {'required': True}, - } - - _attribute_map = { - 'virtual_machines': {'key': 'virtualMachines', 'type': '[JitNetworkAccessRequestVirtualMachine]'}, - 'start_time_utc': {'key': 'startTimeUtc', 'type': 'iso-8601'}, - 'requestor': {'key': 'requestor', 'type': 'str'}, - 'justification': {'key': 'justification', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessRequest, self).__init__(**kwargs) - self.virtual_machines = kwargs['virtual_machines'] - self.start_time_utc = kwargs['start_time_utc'] - self.requestor = kwargs['requestor'] - self.justification = kwargs.get('justification', None) - - -class JitNetworkAccessRequestPort(msrest.serialization.Model): - """JitNetworkAccessRequestPort. - - All required parameters must be populated in order to send to Azure. - - :param number: Required. - :type number: int - :param allowed_source_address_prefix: Mutually exclusive with the - "allowedSourceAddressPrefixes" parameter. Should be an IP address or CIDR, for example - "192.168.0.3" or "192.168.0.0/16". - :type allowed_source_address_prefix: str - :param allowed_source_address_prefixes: Mutually exclusive with the - "allowedSourceAddressPrefix" parameter. - :type allowed_source_address_prefixes: list[str] - :param end_time_utc: Required. The date & time at which the request ends in UTC. - :type end_time_utc: ~datetime.datetime - :param status: Required. The status of the port. Possible values include: "Revoked", - "Initiated". - :type status: str or ~azure.mgmt.security.models.Status - :param status_reason: Required. A description of why the ``status`` has its value. Possible - values include: "Expired", "UserRequested", "NewerRequestInitiated". - :type status_reason: str or ~azure.mgmt.security.models.StatusReason - :param mapped_port: The port which is mapped to this port's ``number`` in the Azure Firewall, - if applicable. - :type mapped_port: int - """ - - _validation = { - 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, - 'end_time_utc': {'required': True}, - 'status': {'required': True}, - 'status_reason': {'required': True}, - } - - _attribute_map = { - 'number': {'key': 'number', 'type': 'int'}, - 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, - 'allowed_source_address_prefixes': {'key': 'allowedSourceAddressPrefixes', 'type': '[str]'}, - 'end_time_utc': {'key': 'endTimeUtc', 'type': 'iso-8601'}, - 'status': {'key': 'status', 'type': 'str'}, - 'status_reason': {'key': 'statusReason', 'type': 'str'}, - 'mapped_port': {'key': 'mappedPort', 'type': 'int'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessRequestPort, self).__init__(**kwargs) - self.number = kwargs['number'] - self.allowed_source_address_prefix = kwargs.get('allowed_source_address_prefix', None) - self.allowed_source_address_prefixes = kwargs.get('allowed_source_address_prefixes', None) - self.end_time_utc = kwargs['end_time_utc'] - self.status = kwargs['status'] - self.status_reason = kwargs['status_reason'] - self.mapped_port = kwargs.get('mapped_port', None) - - -class JitNetworkAccessRequestVirtualMachine(msrest.serialization.Model): - """JitNetworkAccessRequestVirtualMachine. - - All required parameters must be populated in order to send to Azure. - - :param id: Required. Resource ID of the virtual machine that is linked to this policy. - :type id: str - :param ports: Required. The ports that were opened for the virtual machine. - :type ports: list[~azure.mgmt.security.models.JitNetworkAccessRequestPort] - """ - - _validation = { - 'id': {'required': True}, - 'ports': {'required': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'ports': {'key': 'ports', 'type': '[JitNetworkAccessRequestPort]'}, - } - - def __init__( - self, - **kwargs - ): - super(JitNetworkAccessRequestVirtualMachine, self).__init__(**kwargs) - self.id = kwargs['id'] - self.ports = kwargs['ports'] - - -class LocalUserNotAllowed(AllowlistCustomAlertRule): - """Login by a local user that isn't allowed. Allow list consists of login names to allow. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the - rule type. - :type allowlist_values: list[str] - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'allowlist_values': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'allowlist_values': {'key': 'allowlistValues', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(LocalUserNotAllowed, self).__init__(**kwargs) - self.rule_type = 'LocalUserNotAllowed' # type: str - - -class LogAnalyticsIdentifier(ResourceIdentifier): - """Represents a Log Analytics workspace scope identifier. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param type: Required. There can be multiple identifiers of different type per alert, this - field specify the identifier type.Constant filled by server. Possible values include: - "AzureResource", "LogAnalytics". - :type type: str or ~azure.mgmt.security.models.ResourceIdentifierType - :ivar workspace_id: The LogAnalytics workspace id that stores this alert. - :vartype workspace_id: str - :ivar workspace_subscription_id: The azure subscription id for the LogAnalytics workspace - storing this alert. - :vartype workspace_subscription_id: str - :ivar workspace_resource_group: The azure resource group for the LogAnalytics workspace storing - this alert. - :vartype workspace_resource_group: str - :ivar agent_id: (optional) The LogAnalytics agent id reporting the event that this alert is - based on. - :vartype agent_id: str - """ - - _validation = { - 'type': {'required': True}, - 'workspace_id': {'readonly': True}, - 'workspace_subscription_id': {'readonly': True, 'pattern': r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'}, - 'workspace_resource_group': {'readonly': True}, - 'agent_id': {'readonly': True}, - } - - _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, - 'workspace_subscription_id': {'key': 'workspaceSubscriptionId', 'type': 'str'}, - 'workspace_resource_group': {'key': 'workspaceResourceGroup', 'type': 'str'}, - 'agent_id': {'key': 'agentId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(LogAnalyticsIdentifier, self).__init__(**kwargs) - self.type = 'LogAnalytics' # type: str - self.workspace_id = None - self.workspace_subscription_id = None - self.workspace_resource_group = None - self.agent_id = None - - -class MqttC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of cloud to device messages (MQTT protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(MqttC2DMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'MqttC2DMessagesNotInAllowedRange' # type: str - - -class MqttC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of rejected cloud to device messages (MQTT protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(MqttC2DRejectedMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'MqttC2DRejectedMessagesNotInAllowedRange' # type: str - - -class MqttD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of device to cloud messages (MQTT protocol) is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(MqttD2CMessagesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'MqttD2CMessagesNotInAllowedRange' # type: str - - -class OnPremiseResourceDetails(ResourceDetails): - """Details of the On Premise resource that was assessed. - - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: OnPremiseSqlResourceDetails. - - All required parameters must be populated in order to send to Azure. - - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source - :param workspace_id: Required. Azure resource Id of the workspace the machine is attached to. - :type workspace_id: str - :param vmuuid: Required. The unique Id of the machine. - :type vmuuid: str - :param source_computer_id: Required. The oms agent Id installed on the machine. - :type source_computer_id: str - :param machine_name: Required. The name of the machine. - :type machine_name: str - """ - - _validation = { - 'source': {'required': True}, - 'workspace_id': {'required': True}, - 'vmuuid': {'required': True}, - 'source_computer_id': {'required': True}, - 'machine_name': {'required': True}, - } - - _attribute_map = { - 'source': {'key': 'source', 'type': 'str'}, - 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, - 'vmuuid': {'key': 'vmuuid', 'type': 'str'}, - 'source_computer_id': {'key': 'sourceComputerId', 'type': 'str'}, - 'machine_name': {'key': 'machineName', 'type': 'str'}, - } - - _subtype_map = { - 'source': {'OnPremiseSql': 'OnPremiseSqlResourceDetails'} - } - - def __init__( - self, - **kwargs - ): - super(OnPremiseResourceDetails, self).__init__(**kwargs) - self.source = 'OnPremise' # type: str - self.workspace_id = kwargs['workspace_id'] - self.vmuuid = kwargs['vmuuid'] - self.source_computer_id = kwargs['source_computer_id'] - self.machine_name = kwargs['machine_name'] - - -class OnPremiseSqlResourceDetails(OnPremiseResourceDetails): - """Details of the On Premise Sql resource that was assessed. - - All required parameters must be populated in order to send to Azure. - - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source - :param workspace_id: Required. Azure resource Id of the workspace the machine is attached to. - :type workspace_id: str - :param vmuuid: Required. The unique Id of the machine. - :type vmuuid: str - :param source_computer_id: Required. The oms agent Id installed on the machine. - :type source_computer_id: str - :param machine_name: Required. The name of the machine. - :type machine_name: str - :param server_name: Required. The Sql server name installed on the machine. - :type server_name: str - :param database_name: Required. The Sql database name installed on the machine. - :type database_name: str - """ - - _validation = { - 'source': {'required': True}, - 'workspace_id': {'required': True}, - 'vmuuid': {'required': True}, - 'source_computer_id': {'required': True}, - 'machine_name': {'required': True}, - 'server_name': {'required': True}, - 'database_name': {'required': True}, - } - - _attribute_map = { - 'source': {'key': 'source', 'type': 'str'}, - 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, - 'vmuuid': {'key': 'vmuuid', 'type': 'str'}, - 'source_computer_id': {'key': 'sourceComputerId', 'type': 'str'}, - 'machine_name': {'key': 'machineName', 'type': 'str'}, - 'server_name': {'key': 'serverName', 'type': 'str'}, - 'database_name': {'key': 'databaseName', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(OnPremiseSqlResourceDetails, self).__init__(**kwargs) - self.source = 'OnPremiseSql' # type: str - self.server_name = kwargs['server_name'] - self.database_name = kwargs['database_name'] - - -class Operation(msrest.serialization.Model): - """Possible operation in the REST API of Microsoft.Security. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: Name of the operation. - :vartype name: str - :ivar origin: Where the operation is originated. - :vartype origin: str - :param display: Security operation display. - :type display: ~azure.mgmt.security.models.OperationDisplay - """ - - _validation = { - 'name': {'readonly': True}, - 'origin': {'readonly': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'origin': {'key': 'origin', 'type': 'str'}, - 'display': {'key': 'display', 'type': 'OperationDisplay'}, - } - - def __init__( - self, - **kwargs - ): - super(Operation, self).__init__(**kwargs) - self.name = None - self.origin = None - self.display = kwargs.get('display', None) - - -class OperationDisplay(msrest.serialization.Model): - """Security operation display. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar provider: The resource provider for the operation. - :vartype provider: str - :ivar resource: The display name of the resource the operation applies to. - :vartype resource: str - :ivar operation: The display name of the security operation. - :vartype operation: str - :ivar description: The description of the operation. - :vartype description: str - """ - - _validation = { - 'provider': {'readonly': True}, - 'resource': {'readonly': True}, - 'operation': {'readonly': True}, - 'description': {'readonly': True}, - } - - _attribute_map = { - 'provider': {'key': 'provider', 'type': 'str'}, - 'resource': {'key': 'resource', 'type': 'str'}, - 'operation': {'key': 'operation', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(OperationDisplay, self).__init__(**kwargs) - self.provider = None - self.resource = None - self.operation = None - self.description = None - - -class OperationList(msrest.serialization.Model): - """List of possible operations for Microsoft.Security resource provider. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: List of Security operations. - :type value: list[~azure.mgmt.security.models.Operation] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Operation]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(OperationList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class PathRecommendation(msrest.serialization.Model): - """Represents a path that is recommended to be allowed and its properties. - - :param path: The full path of the file, or an identifier of the application. - :type path: str - :param action: The recommendation action of the machine or rule. Possible values include: - "Recommended", "Add", "Remove". - :type action: str or ~azure.mgmt.security.models.RecommendationAction - :param type: The type of IoT Security recommendation. Possible values include: - "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", - "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", - "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", - "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", - "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", - "IoT_VulnerableTLSCipherSuite". - :type type: str or ~azure.mgmt.security.models.RecommendationType - :param publisher_info: Represents the publisher information of a process/rule. - :type publisher_info: ~azure.mgmt.security.models.PublisherInfo - :param common: Whether the application is commonly run on the machine. - :type common: bool - :param user_sids: - :type user_sids: list[str] - :param usernames: - :type usernames: list[~azure.mgmt.security.models.UserRecommendation] - :param file_type: The type of the file (for Linux files - Executable is used). Possible values - include: "Exe", "Dll", "Msi", "Script", "Executable", "Unknown". - :type file_type: str or ~azure.mgmt.security.models.FileType - :param configuration_status: The configuration status of the machines group or machine or rule. - Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", "NoStatus". - :type configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus - """ - - _attribute_map = { - 'path': {'key': 'path', 'type': 'str'}, - 'action': {'key': 'action', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'publisher_info': {'key': 'publisherInfo', 'type': 'PublisherInfo'}, - 'common': {'key': 'common', 'type': 'bool'}, - 'user_sids': {'key': 'userSids', 'type': '[str]'}, - 'usernames': {'key': 'usernames', 'type': '[UserRecommendation]'}, - 'file_type': {'key': 'fileType', 'type': 'str'}, - 'configuration_status': {'key': 'configurationStatus', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(PathRecommendation, self).__init__(**kwargs) - self.path = kwargs.get('path', None) - self.action = kwargs.get('action', None) - self.type = kwargs.get('type', None) - self.publisher_info = kwargs.get('publisher_info', None) - self.common = kwargs.get('common', None) - self.user_sids = kwargs.get('user_sids', None) - self.usernames = kwargs.get('usernames', None) - self.file_type = kwargs.get('file_type', None) - self.configuration_status = kwargs.get('configuration_status', None) - - -class Pricing(Resource): - """Azure Security Center is provided in two pricing tiers: free and standard, with the standard tier available with a trial period. The standard tier offers advanced security capabilities, while the free tier offers basic security features. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param pricing_tier: The pricing tier value. Azure Security Center is provided in two pricing - tiers: free and standard, with the standard tier available with a trial period. The standard - tier offers advanced security capabilities, while the free tier offers basic security features. - Possible values include: "Free", "Standard". - :type pricing_tier: str or ~azure.mgmt.security.models.PricingTier - :ivar free_trial_remaining_time: The duration left for the subscriptions free trial period - in - ISO 8601 format (e.g. P3Y6M4DT12H30M5S). - :vartype free_trial_remaining_time: ~datetime.timedelta - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'free_trial_remaining_time': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'pricing_tier': {'key': 'properties.pricingTier', 'type': 'str'}, - 'free_trial_remaining_time': {'key': 'properties.freeTrialRemainingTime', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(Pricing, self).__init__(**kwargs) - self.pricing_tier = kwargs.get('pricing_tier', None) - self.free_trial_remaining_time = None - - -class PricingList(msrest.serialization.Model): - """List of pricing configurations response. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of pricing configurations. - :type value: list[~azure.mgmt.security.models.Pricing] - """ - - _validation = { - 'value': {'required': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Pricing]'}, - } - - def __init__( - self, - **kwargs - ): - super(PricingList, self).__init__(**kwargs) - self.value = kwargs['value'] - - -class ProcessNotAllowed(AllowlistCustomAlertRule): - """Execution of a process that isn't allowed. Allow list consists of process names to allow. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the - rule type. - :type allowlist_values: list[str] - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'allowlist_values': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'allowlist_values': {'key': 'allowlistValues', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(ProcessNotAllowed, self).__init__(**kwargs) - self.rule_type = 'ProcessNotAllowed' # type: str - - -class ProtectionMode(msrest.serialization.Model): - """The protection mode of the collection/file types. Exe/Msi/Script are used for Windows, Executable is used for Linux. - - :param exe: The application control policy enforcement/protection mode of the machine group. - Possible values include: "Audit", "Enforce", "None". - :type exe: str or ~azure.mgmt.security.models.EnforcementMode - :param msi: The application control policy enforcement/protection mode of the machine group. - Possible values include: "Audit", "Enforce", "None". - :type msi: str or ~azure.mgmt.security.models.EnforcementMode - :param script: The application control policy enforcement/protection mode of the machine group. - Possible values include: "Audit", "Enforce", "None". - :type script: str or ~azure.mgmt.security.models.EnforcementMode - :param executable: The application control policy enforcement/protection mode of the machine - group. Possible values include: "Audit", "Enforce", "None". - :type executable: str or ~azure.mgmt.security.models.EnforcementMode - """ - - _attribute_map = { - 'exe': {'key': 'exe', 'type': 'str'}, - 'msi': {'key': 'msi', 'type': 'str'}, - 'script': {'key': 'script', 'type': 'str'}, - 'executable': {'key': 'executable', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ProtectionMode, self).__init__(**kwargs) - self.exe = kwargs.get('exe', None) - self.msi = kwargs.get('msi', None) - self.script = kwargs.get('script', None) - self.executable = kwargs.get('executable', None) - - -class ProxyServerProperties(msrest.serialization.Model): - """For a non-Azure machine that is not connected directly to the internet, specify a proxy server that the non-Azure machine can use. - - :param ip: Proxy server IP. - :type ip: str - :param port: Proxy server port. - :type port: str - """ - - _attribute_map = { - 'ip': {'key': 'ip', 'type': 'str'}, - 'port': {'key': 'port', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ProxyServerProperties, self).__init__(**kwargs) - self.ip = kwargs.get('ip', None) - self.port = kwargs.get('port', None) - - -class PublisherInfo(msrest.serialization.Model): - """Represents the publisher information of a process/rule. - - :param publisher_name: The Subject field of the x.509 certificate used to sign the code, using - the following fields - O = Organization, L = Locality, S = State or Province, and C = Country. - :type publisher_name: str - :param product_name: The product name taken from the file's version resource. - :type product_name: str - :param binary_name: The "OriginalName" field taken from the file's version resource. - :type binary_name: str - :param version: The binary file version taken from the file's version resource. - :type version: str - """ - - _attribute_map = { - 'publisher_name': {'key': 'publisherName', 'type': 'str'}, - 'product_name': {'key': 'productName', 'type': 'str'}, - 'binary_name': {'key': 'binaryName', 'type': 'str'}, - 'version': {'key': 'version', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(PublisherInfo, self).__init__(**kwargs) - self.publisher_name = kwargs.get('publisher_name', None) - self.product_name = kwargs.get('product_name', None) - self.binary_name = kwargs.get('binary_name', None) - self.version = kwargs.get('version', None) - - -class QueryCheck(msrest.serialization.Model): - """The rule query details. - - :param query: The rule query. - :type query: str - :param expected_result: Expected result. - :type expected_result: list[list[str]] - :param column_names: Column names of expected result. - :type column_names: list[str] - """ - - _attribute_map = { - 'query': {'key': 'query', 'type': 'str'}, - 'expected_result': {'key': 'expectedResult', 'type': '[[str]]'}, - 'column_names': {'key': 'columnNames', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(QueryCheck, self).__init__(**kwargs) - self.query = kwargs.get('query', None) - self.expected_result = kwargs.get('expected_result', None) - self.column_names = kwargs.get('column_names', None) - - -class QueuePurgesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of device queue purges is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(QueuePurgesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'QueuePurgesNotInAllowedRange' # type: str - - -class RecommendationConfigurationProperties(msrest.serialization.Model): - """The type of IoT Security recommendation. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param recommendation_type: Required. The type of IoT Security recommendation. Possible values - include: "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", - "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", - "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", - "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", - "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", - "IoT_VulnerableTLSCipherSuite". - :type recommendation_type: str or ~azure.mgmt.security.models.RecommendationType - :ivar name: - :vartype name: str - :param status: Required. Recommendation status. When the recommendation status is disabled - recommendations are not generated. Possible values include: "Disabled", "Enabled". Default - value: "Enabled". - :type status: str or ~azure.mgmt.security.models.RecommendationConfigStatus - """ - - _validation = { - 'recommendation_type': {'required': True}, - 'name': {'readonly': True}, - 'status': {'required': True}, - } - - _attribute_map = { - 'recommendation_type': {'key': 'recommendationType', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'status': {'key': 'status', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(RecommendationConfigurationProperties, self).__init__(**kwargs) - self.recommendation_type = kwargs['recommendation_type'] - self.name = None - self.status = kwargs.get('status', "Enabled") - - -class RegulatoryComplianceAssessment(Resource): - """Regulatory compliance assessment details and state. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar description: The description of the regulatory compliance assessment. - :vartype description: str - :ivar assessment_type: The expected type of assessment contained in the AssessmentDetailsLink. - :vartype assessment_type: str - :ivar assessment_details_link: Link to more detailed assessment results data. The response type - will be according to the assessmentType field. - :vartype assessment_details_link: str - :param state: Aggregative state based on the assessment's scanned resources states. Possible - values include: "Passed", "Failed", "Skipped", "Unsupported". - :type state: str or ~azure.mgmt.security.models.State - :ivar passed_resources: The given assessment's related resources count with passed state. - :vartype passed_resources: int - :ivar failed_resources: The given assessment's related resources count with failed state. - :vartype failed_resources: int - :ivar skipped_resources: The given assessment's related resources count with skipped state. - :vartype skipped_resources: int - :ivar unsupported_resources: The given assessment's related resources count with unsupported - state. - :vartype unsupported_resources: int - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'description': {'readonly': True}, - 'assessment_type': {'readonly': True}, - 'assessment_details_link': {'readonly': True}, - 'passed_resources': {'readonly': True}, - 'failed_resources': {'readonly': True}, - 'skipped_resources': {'readonly': True}, - 'unsupported_resources': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'assessment_type': {'key': 'properties.assessmentType', 'type': 'str'}, - 'assessment_details_link': {'key': 'properties.assessmentDetailsLink', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'passed_resources': {'key': 'properties.passedResources', 'type': 'int'}, - 'failed_resources': {'key': 'properties.failedResources', 'type': 'int'}, - 'skipped_resources': {'key': 'properties.skippedResources', 'type': 'int'}, - 'unsupported_resources': {'key': 'properties.unsupportedResources', 'type': 'int'}, - } - - def __init__( - self, - **kwargs - ): - super(RegulatoryComplianceAssessment, self).__init__(**kwargs) - self.description = None - self.assessment_type = None - self.assessment_details_link = None - self.state = kwargs.get('state', None) - self.passed_resources = None - self.failed_resources = None - self.skipped_resources = None - self.unsupported_resources = None - - -class RegulatoryComplianceAssessmentList(msrest.serialization.Model): - """List of regulatory compliance assessment response. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. - :type value: list[~azure.mgmt.security.models.RegulatoryComplianceAssessment] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[RegulatoryComplianceAssessment]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(RegulatoryComplianceAssessmentList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class RegulatoryComplianceControl(Resource): - """Regulatory compliance control details and state. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar description: The description of the regulatory compliance control. - :vartype description: str - :param state: Aggregative state based on the control's supported assessments states. Possible - values include: "Passed", "Failed", "Skipped", "Unsupported". - :type state: str or ~azure.mgmt.security.models.State - :ivar passed_assessments: The number of supported regulatory compliance assessments of the - given control with a passed state. - :vartype passed_assessments: int - :ivar failed_assessments: The number of supported regulatory compliance assessments of the - given control with a failed state. - :vartype failed_assessments: int - :ivar skipped_assessments: The number of supported regulatory compliance assessments of the - given control with a skipped state. - :vartype skipped_assessments: int - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'description': {'readonly': True}, - 'passed_assessments': {'readonly': True}, - 'failed_assessments': {'readonly': True}, - 'skipped_assessments': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'passed_assessments': {'key': 'properties.passedAssessments', 'type': 'int'}, - 'failed_assessments': {'key': 'properties.failedAssessments', 'type': 'int'}, - 'skipped_assessments': {'key': 'properties.skippedAssessments', 'type': 'int'}, - } - - def __init__( - self, - **kwargs - ): - super(RegulatoryComplianceControl, self).__init__(**kwargs) - self.description = None - self.state = kwargs.get('state', None) - self.passed_assessments = None - self.failed_assessments = None - self.skipped_assessments = None - - -class RegulatoryComplianceControlList(msrest.serialization.Model): - """List of regulatory compliance controls response. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of regulatory compliance controls. - :type value: list[~azure.mgmt.security.models.RegulatoryComplianceControl] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[RegulatoryComplianceControl]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(RegulatoryComplianceControlList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class RegulatoryComplianceStandard(Resource): - """Regulatory compliance standard details and state. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param state: Aggregative state based on the standard's supported controls states. Possible - values include: "Passed", "Failed", "Skipped", "Unsupported". - :type state: str or ~azure.mgmt.security.models.State - :ivar passed_controls: The number of supported regulatory compliance controls of the given - standard with a passed state. - :vartype passed_controls: int - :ivar failed_controls: The number of supported regulatory compliance controls of the given - standard with a failed state. - :vartype failed_controls: int - :ivar skipped_controls: The number of supported regulatory compliance controls of the given - standard with a skipped state. - :vartype skipped_controls: int - :ivar unsupported_controls: The number of regulatory compliance controls of the given standard - which are unsupported by automated assessments. - :vartype unsupported_controls: int - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'passed_controls': {'readonly': True}, - 'failed_controls': {'readonly': True}, - 'skipped_controls': {'readonly': True}, - 'unsupported_controls': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'passed_controls': {'key': 'properties.passedControls', 'type': 'int'}, - 'failed_controls': {'key': 'properties.failedControls', 'type': 'int'}, - 'skipped_controls': {'key': 'properties.skippedControls', 'type': 'int'}, - 'unsupported_controls': {'key': 'properties.unsupportedControls', 'type': 'int'}, - } - - def __init__( - self, - **kwargs - ): - super(RegulatoryComplianceStandard, self).__init__(**kwargs) - self.state = kwargs.get('state', None) - self.passed_controls = None - self.failed_controls = None - self.skipped_controls = None - self.unsupported_controls = None - - -class RegulatoryComplianceStandardList(msrest.serialization.Model): - """List of regulatory compliance standards response. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. - :type value: list[~azure.mgmt.security.models.RegulatoryComplianceStandard] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[RegulatoryComplianceStandard]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(RegulatoryComplianceStandardList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None - - -class Remediation(msrest.serialization.Model): - """Remediation details. - - :param description: Remediation description. - :type description: str - :param scripts: Remediation script. - :type scripts: list[str] - :param automated: Is remediation automated. - :type automated: bool - :param portal_link: Optional link to remediate in Azure Portal. - :type portal_link: str - """ - - _attribute_map = { - 'description': {'key': 'description', 'type': 'str'}, - 'scripts': {'key': 'scripts', 'type': '[str]'}, - 'automated': {'key': 'automated', 'type': 'bool'}, - 'portal_link': {'key': 'portalLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(Remediation, self).__init__(**kwargs) - self.description = kwargs.get('description', None) - self.scripts = kwargs.get('scripts', None) - self.automated = kwargs.get('automated', None) - self.portal_link = kwargs.get('portal_link', None) - - -class Rule(msrest.serialization.Model): - """Describes remote addresses that is recommended to communicate with the Azure resource on some (Protocol, Port, Direction). All other remote addresses are recommended to be blocked. - - :param name: The name of the rule. - :type name: str - :param direction: The rule's direction. Possible values include: "Inbound", "Outbound". - :type direction: str or ~azure.mgmt.security.models.Direction - :param destination_port: The rule's destination port. - :type destination_port: int - :param protocols: The rule's transport protocols. - :type protocols: list[str or ~azure.mgmt.security.models.TransportProtocol] - :param ip_addresses: The remote IP addresses that should be able to communicate with the Azure - resource on the rule's destination port and protocol. - :type ip_addresses: list[str] - """ - - _validation = { - 'destination_port': {'maximum': 65535, 'minimum': 0}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'direction': {'key': 'direction', 'type': 'str'}, - 'destination_port': {'key': 'destinationPort', 'type': 'int'}, - 'protocols': {'key': 'protocols', 'type': '[str]'}, - 'ip_addresses': {'key': 'ipAddresses', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(Rule, self).__init__(**kwargs) - self.name = kwargs.get('name', None) - self.direction = kwargs.get('direction', None) - self.destination_port = kwargs.get('destination_port', None) - self.protocols = kwargs.get('protocols', None) - self.ip_addresses = kwargs.get('ip_addresses', None) - - -class RuleResults(Resource): - """Rule results. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: Rule results properties. - :type properties: ~azure.mgmt.security.models.RuleResultsProperties - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'RuleResultsProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(RuleResults, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class RuleResultsInput(msrest.serialization.Model): - """Rule results input. - - :param latest_scan: Take results from latest scan. - :type latest_scan: bool - :param results: Expected results to be inserted into the baseline. - Leave this field empty it LatestScan == true. - :type results: list[list[str]] - """ - - _attribute_map = { - 'latest_scan': {'key': 'latestScan', 'type': 'bool'}, - 'results': {'key': 'results', 'type': '[[str]]'}, - } - - def __init__( - self, - **kwargs - ): - super(RuleResultsInput, self).__init__(**kwargs) - self.latest_scan = kwargs.get('latest_scan', None) - self.results = kwargs.get('results', None) - - -class RuleResultsProperties(msrest.serialization.Model): - """Rule results properties. - - :param results: Expected results in the baseline. - :type results: list[list[str]] - """ - - _attribute_map = { - 'results': {'key': 'results', 'type': '[[str]]'}, - } - - def __init__( - self, - **kwargs - ): - super(RuleResultsProperties, self).__init__(**kwargs) - self.results = kwargs.get('results', None) - - -class RulesResults(msrest.serialization.Model): - """A list of rules results. - - :param value: List of rule results. - :type value: list[~azure.mgmt.security.models.RuleResults] - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[RuleResults]'}, - } - - def __init__( - self, - **kwargs - ): - super(RulesResults, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - - -class RulesResultsInput(msrest.serialization.Model): - """Rules results input. - - :param latest_scan: Take results from latest scan. - :type latest_scan: bool - :param results: Expected results to be inserted into the baseline. - Leave this field empty it LatestScan == true. - :type results: dict[str, list[list[str]]] - """ - - _attribute_map = { - 'latest_scan': {'key': 'latestScan', 'type': 'bool'}, - 'results': {'key': 'results', 'type': '{[[str]]}'}, - } - - def __init__( - self, - **kwargs - ): - super(RulesResultsInput, self).__init__(**kwargs) - self.latest_scan = kwargs.get('latest_scan', None) - self.results = kwargs.get('results', None) - - -class Scan(Resource): - """A vulnerability assessment scan record. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: A vulnerability assessment scan record properties. - :type properties: ~azure.mgmt.security.models.ScanProperties - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'ScanProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(Scan, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class ScanProperties(msrest.serialization.Model): - """A vulnerability assessment scan record properties. - - :param trigger_type: The scan trigger type. Possible values include: "OnDemand", "Recurring". - :type trigger_type: str or ~azure.mgmt.security.models.ScanTriggerType - :param state: The scan status. Possible values include: "Failed", "FailedToRun", "InProgress", - "Passed". - :type state: str or ~azure.mgmt.security.models.ScanState - :param server: The server name. - :type server: str - :param database: The database name. - :type database: str - :param sql_version: The SQL version. - :type sql_version: str - :param start_time: The scan start time (UTC). - :type start_time: ~datetime.datetime - :param end_time: Scan results are valid until end time (UTC). - :type end_time: ~datetime.datetime - :param high_severity_failed_rules_count: The number of failed rules with high severity. - :type high_severity_failed_rules_count: int - :param medium_severity_failed_rules_count: The number of failed rules with medium severity. - :type medium_severity_failed_rules_count: int - :param low_severity_failed_rules_count: The number of failed rules with low severity. - :type low_severity_failed_rules_count: int - :param total_passed_rules_count: The number of total passed rules. - :type total_passed_rules_count: int - :param total_failed_rules_count: The number of total failed rules. - :type total_failed_rules_count: int - :param total_rules_count: The number of total rules assessed. - :type total_rules_count: int - :param is_baseline_applied: Baseline created for this database, and has one or more rules. - :type is_baseline_applied: bool - """ - - _attribute_map = { - 'trigger_type': {'key': 'triggerType', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'server': {'key': 'server', 'type': 'str'}, - 'database': {'key': 'database', 'type': 'str'}, - 'sql_version': {'key': 'sqlVersion', 'type': 'str'}, - 'start_time': {'key': 'startTime', 'type': 'iso-8601'}, - 'end_time': {'key': 'endTime', 'type': 'iso-8601'}, - 'high_severity_failed_rules_count': {'key': 'highSeverityFailedRulesCount', 'type': 'int'}, - 'medium_severity_failed_rules_count': {'key': 'mediumSeverityFailedRulesCount', 'type': 'int'}, - 'low_severity_failed_rules_count': {'key': 'lowSeverityFailedRulesCount', 'type': 'int'}, - 'total_passed_rules_count': {'key': 'totalPassedRulesCount', 'type': 'int'}, - 'total_failed_rules_count': {'key': 'totalFailedRulesCount', 'type': 'int'}, - 'total_rules_count': {'key': 'totalRulesCount', 'type': 'int'}, - 'is_baseline_applied': {'key': 'isBaselineApplied', 'type': 'bool'}, - } - - def __init__( - self, - **kwargs - ): - super(ScanProperties, self).__init__(**kwargs) - self.trigger_type = kwargs.get('trigger_type', None) - self.state = kwargs.get('state', None) - self.server = kwargs.get('server', None) - self.database = kwargs.get('database', None) - self.sql_version = kwargs.get('sql_version', None) - self.start_time = kwargs.get('start_time', None) - self.end_time = kwargs.get('end_time', None) - self.high_severity_failed_rules_count = kwargs.get('high_severity_failed_rules_count', None) - self.medium_severity_failed_rules_count = kwargs.get('medium_severity_failed_rules_count', None) - self.low_severity_failed_rules_count = kwargs.get('low_severity_failed_rules_count', None) - self.total_passed_rules_count = kwargs.get('total_passed_rules_count', None) - self.total_failed_rules_count = kwargs.get('total_failed_rules_count', None) - self.total_rules_count = kwargs.get('total_rules_count', None) - self.is_baseline_applied = kwargs.get('is_baseline_applied', None) - - -class ScanResult(Resource): - """A vulnerability assessment scan result for a single rule. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: A vulnerability assessment scan result properties for a single rule. - :type properties: ~azure.mgmt.security.models.ScanResultProperties - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'ScanResultProperties'}, - } - - def __init__( - self, - **kwargs - ): - super(ScanResult, self).__init__(**kwargs) - self.properties = kwargs.get('properties', None) - - -class ScanResultProperties(msrest.serialization.Model): - """A vulnerability assessment scan result properties for a single rule. - - :param rule_id: The rule Id. - :type rule_id: str - :param status: The rule result status. Possible values include: "NonFinding", "Finding", - "InternalError". - :type status: str or ~azure.mgmt.security.models.RuleStatus - :param is_trimmed: Indicated whether the results specified here are trimmed. - :type is_trimmed: bool - :param query_results: The results of the query that was run. - :type query_results: list[list[str]] - :param remediation: Remediation details. - :type remediation: ~azure.mgmt.security.models.Remediation - :param baseline_adjusted_result: The rule result adjusted with baseline. - :type baseline_adjusted_result: ~azure.mgmt.security.models.BaselineAdjustedResult - :param rule_metadata: vulnerability assessment rule metadata details. - :type rule_metadata: ~azure.mgmt.security.models.VaRule - """ - - _attribute_map = { - 'rule_id': {'key': 'ruleId', 'type': 'str'}, - 'status': {'key': 'status', 'type': 'str'}, - 'is_trimmed': {'key': 'isTrimmed', 'type': 'bool'}, - 'query_results': {'key': 'queryResults', 'type': '[[str]]'}, - 'remediation': {'key': 'remediation', 'type': 'Remediation'}, - 'baseline_adjusted_result': {'key': 'baselineAdjustedResult', 'type': 'BaselineAdjustedResult'}, - 'rule_metadata': {'key': 'ruleMetadata', 'type': 'VaRule'}, - } - - def __init__( - self, - **kwargs - ): - super(ScanResultProperties, self).__init__(**kwargs) - self.rule_id = kwargs.get('rule_id', None) - self.status = kwargs.get('status', None) - self.is_trimmed = kwargs.get('is_trimmed', None) - self.query_results = kwargs.get('query_results', None) - self.remediation = kwargs.get('remediation', None) - self.baseline_adjusted_result = kwargs.get('baseline_adjusted_result', None) - self.rule_metadata = kwargs.get('rule_metadata', None) - - -class ScanResults(msrest.serialization.Model): - """A list of vulnerability assessment scan results. - - :param value: List of vulnerability assessment scan results. - :type value: list[~azure.mgmt.security.models.ScanResult] - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[ScanResult]'}, - } - - def __init__( - self, - **kwargs - ): - super(ScanResults, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - - -class Scans(msrest.serialization.Model): - """A list of vulnerability assessment scan records. - - :param value: List of vulnerability assessment scan records. - :type value: list[~azure.mgmt.security.models.Scan] - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Scan]'}, - } - - def __init__( - self, - **kwargs - ): - super(Scans, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - - -class ScopeElement(msrest.serialization.Model): - """A more specific scope used to identify the alerts to suppress. - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param field: The alert entity type to suppress by. - :type field: str - """ - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'field': {'key': 'field', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ScopeElement, self).__init__(**kwargs) - self.additional_properties = kwargs.get('additional_properties', None) - self.field = kwargs.get('field', None) - - -class SecureScoreControlDefinitionItem(Resource): - """Information about the security control. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar display_name: User friendly display name of the control. - :vartype display_name: str - :ivar description: User friendly description of the control. - :vartype description: str - :ivar max_score: Maximum control score (0..10). - :vartype max_score: int - :ivar source: Source object from which the control was created. - :vartype source: ~azure.mgmt.security.models.SecureScoreControlDefinitionSource - :ivar assessment_definitions: Array of assessments metadata IDs that are included in this - security control. - :vartype assessment_definitions: list[~azure.mgmt.security.models.AzureResourceLink] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'display_name': {'readonly': True}, - 'description': {'readonly': True, 'max_length': 256, 'min_length': 0}, - 'max_score': {'readonly': True, 'maximum': 10, 'minimum': 0}, - 'source': {'readonly': True}, - 'assessment_definitions': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'max_score': {'key': 'properties.maxScore', 'type': 'int'}, - 'source': {'key': 'properties.source', 'type': 'SecureScoreControlDefinitionSource'}, - 'assessment_definitions': {'key': 'properties.assessmentDefinitions', 'type': '[AzureResourceLink]'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoreControlDefinitionItem, self).__init__(**kwargs) - self.display_name = None - self.description = None - self.max_score = None - self.source = None - self.assessment_definitions = None - - -class SecureScoreControlDefinitionList(msrest.serialization.Model): - """List of security controls definition. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Collection of security controls definition in this page. - :vartype value: list[~azure.mgmt.security.models.SecureScoreControlDefinitionItem] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecureScoreControlDefinitionItem]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoreControlDefinitionList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecureScoreControlDefinitionSource(msrest.serialization.Model): - """The type of the security control (For example, BuiltIn). - - :param source_type: The type of security control (for example, BuiltIn). Possible values - include: "BuiltIn", "Custom". - :type source_type: str or ~azure.mgmt.security.models.ControlType - """ - - _attribute_map = { - 'source_type': {'key': 'sourceType', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoreControlDefinitionSource, self).__init__(**kwargs) - self.source_type = kwargs.get('source_type', None) - - -class SecureScoreControlDetails(Resource): - """Details of the security control, its score, and the health status of the relevant resources. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar display_name: User friendly display name of the control. - :vartype display_name: str - :ivar healthy_resource_count: Number of healthy resources in the control. - :vartype healthy_resource_count: int - :ivar unhealthy_resource_count: Number of unhealthy resources in the control. - :vartype unhealthy_resource_count: int - :ivar not_applicable_resource_count: Number of not applicable resources in the control. - :vartype not_applicable_resource_count: int - :ivar weight: The relative weight for this specific control in each of your subscriptions. Used - when calculating an aggregated score for this control across all of your subscriptions. - :vartype weight: long - :param definition: Information about the security control. - :type definition: ~azure.mgmt.security.models.SecureScoreControlDefinitionItem - :ivar max: Maximum score available. - :vartype max: int - :ivar current: Current score. - :vartype current: float - :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after - the decimal point. - :vartype percentage: float - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'display_name': {'readonly': True}, - 'healthy_resource_count': {'readonly': True}, - 'unhealthy_resource_count': {'readonly': True}, - 'not_applicable_resource_count': {'readonly': True}, - 'weight': {'readonly': True, 'minimum': 0}, - 'max': {'readonly': True, 'minimum': 0}, - 'current': {'readonly': True, 'minimum': 0}, - 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'healthy_resource_count': {'key': 'properties.healthyResourceCount', 'type': 'int'}, - 'unhealthy_resource_count': {'key': 'properties.unhealthyResourceCount', 'type': 'int'}, - 'not_applicable_resource_count': {'key': 'properties.notApplicableResourceCount', 'type': 'int'}, - 'weight': {'key': 'properties.weight', 'type': 'long'}, - 'definition': {'key': 'properties.definition', 'type': 'SecureScoreControlDefinitionItem'}, - 'max': {'key': 'properties.score.max', 'type': 'int'}, - 'current': {'key': 'properties.score.current', 'type': 'float'}, - 'percentage': {'key': 'properties.score.percentage', 'type': 'float'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoreControlDetails, self).__init__(**kwargs) - self.display_name = None - self.healthy_resource_count = None - self.unhealthy_resource_count = None - self.not_applicable_resource_count = None - self.weight = None - self.definition = kwargs.get('definition', None) - self.max = None - self.current = None - self.percentage = None - - -class SecureScoreControlList(msrest.serialization.Model): - """List of security controls. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Collection of security controls in this page. - :vartype value: list[~azure.mgmt.security.models.SecureScoreControlDetails] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecureScoreControlDetails]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoreControlList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecureScoreControlScore(msrest.serialization.Model): - """Calculation result data. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar max: Maximum control score (0..10). - :vartype max: int - :ivar current: Actual score for the control = (achieved points / total points) * max score. if - total points is zeroed, the return number is 0.00. - :vartype current: float - :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after - the decimal point. - :vartype percentage: float - """ - - _validation = { - 'max': {'readonly': True, 'maximum': 10, 'minimum': 0}, - 'current': {'readonly': True, 'maximum': 10, 'minimum': 0}, - 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, - } - - _attribute_map = { - 'max': {'key': 'max', 'type': 'int'}, - 'current': {'key': 'current', 'type': 'float'}, - 'percentage': {'key': 'percentage', 'type': 'float'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoreControlScore, self).__init__(**kwargs) - self.max = None - self.current = None - self.percentage = None - - -class SecureScoreItem(Resource): - """Secure score item data model. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar display_name: The initiative’s name. - :vartype display_name: str - :ivar weight: The relative weight for each subscription. Used when calculating an aggregated - secure score for multiple subscriptions. - :vartype weight: long - :ivar max: Maximum score available. - :vartype max: int - :ivar current: Current score. - :vartype current: float - :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after - the decimal point. - :vartype percentage: float - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'display_name': {'readonly': True}, - 'weight': {'readonly': True, 'minimum': 0}, - 'max': {'readonly': True, 'minimum': 0}, - 'current': {'readonly': True, 'minimum': 0}, - 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'weight': {'key': 'properties.weight', 'type': 'long'}, - 'max': {'key': 'properties.score.max', 'type': 'int'}, - 'current': {'key': 'properties.score.current', 'type': 'float'}, - 'percentage': {'key': 'properties.score.percentage', 'type': 'float'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoreItem, self).__init__(**kwargs) - self.display_name = None - self.weight = None - self.max = None - self.current = None - self.percentage = None - - -class SecureScoresList(msrest.serialization.Model): - """List of secure scores. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Collection of secure scores in this page. - :vartype value: list[~azure.mgmt.security.models.SecureScoreItem] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecureScoreItem]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecureScoresList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecurityAssessment(Resource): - """Security assessment on a resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param resource_details: Details of the resource that was assessed. - :type resource_details: ~azure.mgmt.security.models.ResourceDetails - :ivar display_name: User friendly display name of the assessment. - :vartype display_name: str - :param status: The result of the assessment. - :type status: ~azure.mgmt.security.models.AssessmentStatus - :param additional_data: Additional data regarding the assessment. - :type additional_data: dict[str, str] - :ivar links: Links relevant to the assessment. - :vartype links: ~azure.mgmt.security.models.AssessmentLinks - :param metadata: Describes properties of an assessment metadata. - :type metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties - :param partners_data: Data regarding 3rd party partner integration. - :type partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'display_name': {'readonly': True}, - 'links': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'resource_details': {'key': 'properties.resourceDetails', 'type': 'ResourceDetails'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'status': {'key': 'properties.status', 'type': 'AssessmentStatus'}, - 'additional_data': {'key': 'properties.additionalData', 'type': '{str}'}, - 'links': {'key': 'properties.links', 'type': 'AssessmentLinks'}, - 'metadata': {'key': 'properties.metadata', 'type': 'SecurityAssessmentMetadataProperties'}, - 'partners_data': {'key': 'properties.partnersData', 'type': 'SecurityAssessmentPartnerData'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessment, self).__init__(**kwargs) - self.resource_details = kwargs.get('resource_details', None) - self.display_name = None - self.status = kwargs.get('status', None) - self.additional_data = kwargs.get('additional_data', None) - self.links = None - self.metadata = kwargs.get('metadata', None) - self.partners_data = kwargs.get('partners_data', None) - - -class SecurityAssessmentList(msrest.serialization.Model): - """Page of a security assessments list. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Collection of security assessments in this page. - :vartype value: list[~azure.mgmt.security.models.SecurityAssessment] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecurityAssessment]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecurityAssessmentMetadata(Resource): - """Security assessment metadata. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param display_name: User friendly display name of the assessment. - :type display_name: str - :ivar policy_definition_id: Azure resource ID of the policy definition that turns this - assessment calculation on. - :vartype policy_definition_id: str - :param description: Human readable description of the assessment. - :type description: str - :param remediation_description: Human readable description of what you should do to mitigate - this security issue. - :type remediation_description: str - :param categories: - :type categories: list[str or ~azure.mgmt.security.models.Categories] - :param severity: The severity level of the assessment. Possible values include: "Low", - "Medium", "High". - :type severity: str or ~azure.mgmt.security.models.Severity - :param user_impact: The user impact of the assessment. Possible values include: "Low", - "Moderate", "High". - :type user_impact: str or ~azure.mgmt.security.models.UserImpact - :param implementation_effort: The implementation effort required to remediate this assessment. - Possible values include: "Low", "Moderate", "High". - :type implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort - :param threats: - :type threats: list[str or ~azure.mgmt.security.models.Threats] - :param preview: True if this assessment is in preview release status. - :type preview: bool - :param assessment_type: BuiltIn if the assessment based on built-in Azure Policy definition, - Custom if the assessment based on custom Azure Policy definition. Possible values include: - "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". - :type assessment_type: str or ~azure.mgmt.security.models.AssessmentType - :param partner_data: Describes the partner that created the assessment. - :type partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'policy_definition_id': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'policy_definition_id': {'key': 'properties.policyDefinitionId', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'remediation_description': {'key': 'properties.remediationDescription', 'type': 'str'}, - 'categories': {'key': 'properties.categories', 'type': '[str]'}, - 'severity': {'key': 'properties.severity', 'type': 'str'}, - 'user_impact': {'key': 'properties.userImpact', 'type': 'str'}, - 'implementation_effort': {'key': 'properties.implementationEffort', 'type': 'str'}, - 'threats': {'key': 'properties.threats', 'type': '[str]'}, - 'preview': {'key': 'properties.preview', 'type': 'bool'}, - 'assessment_type': {'key': 'properties.assessmentType', 'type': 'str'}, - 'partner_data': {'key': 'properties.partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentMetadata, self).__init__(**kwargs) - self.display_name = kwargs.get('display_name', None) - self.policy_definition_id = None - self.description = kwargs.get('description', None) - self.remediation_description = kwargs.get('remediation_description', None) - self.categories = kwargs.get('categories', None) - self.severity = kwargs.get('severity', None) - self.user_impact = kwargs.get('user_impact', None) - self.implementation_effort = kwargs.get('implementation_effort', None) - self.threats = kwargs.get('threats', None) - self.preview = kwargs.get('preview', None) - self.assessment_type = kwargs.get('assessment_type', None) - self.partner_data = kwargs.get('partner_data', None) - - -class SecurityAssessmentMetadataList(msrest.serialization.Model): - """List of security assessment metadata. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: - :vartype value: list[~azure.mgmt.security.models.SecurityAssessmentMetadata] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecurityAssessmentMetadata]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentMetadataList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecurityAssessmentMetadataPartnerData(msrest.serialization.Model): - """Describes the partner that created the assessment. - - All required parameters must be populated in order to send to Azure. - - :param partner_name: Required. Name of the company of the partner. - :type partner_name: str - :param product_name: Name of the product of the partner that created the assessment. - :type product_name: str - :param secret: Required. Secret to authenticate the partner and verify it created the - assessment - write only. - :type secret: str - """ - - _validation = { - 'partner_name': {'required': True}, - 'secret': {'required': True}, - } - - _attribute_map = { - 'partner_name': {'key': 'partnerName', 'type': 'str'}, - 'product_name': {'key': 'productName', 'type': 'str'}, - 'secret': {'key': 'secret', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentMetadataPartnerData, self).__init__(**kwargs) - self.partner_name = kwargs['partner_name'] - self.product_name = kwargs.get('product_name', None) - self.secret = kwargs['secret'] - - -class SecurityAssessmentMetadataProperties(msrest.serialization.Model): - """Describes properties of an assessment metadata. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param display_name: Required. User friendly display name of the assessment. - :type display_name: str - :ivar policy_definition_id: Azure resource ID of the policy definition that turns this - assessment calculation on. - :vartype policy_definition_id: str - :param description: Human readable description of the assessment. - :type description: str - :param remediation_description: Human readable description of what you should do to mitigate - this security issue. - :type remediation_description: str - :param categories: - :type categories: list[str or ~azure.mgmt.security.models.Categories] - :param severity: Required. The severity level of the assessment. Possible values include: - "Low", "Medium", "High". - :type severity: str or ~azure.mgmt.security.models.Severity - :param user_impact: The user impact of the assessment. Possible values include: "Low", - "Moderate", "High". - :type user_impact: str or ~azure.mgmt.security.models.UserImpact - :param implementation_effort: The implementation effort required to remediate this assessment. - Possible values include: "Low", "Moderate", "High". - :type implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort - :param threats: - :type threats: list[str or ~azure.mgmt.security.models.Threats] - :param preview: True if this assessment is in preview release status. - :type preview: bool - :param assessment_type: Required. BuiltIn if the assessment based on built-in Azure Policy - definition, Custom if the assessment based on custom Azure Policy definition. Possible values - include: "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". - :type assessment_type: str or ~azure.mgmt.security.models.AssessmentType - :param partner_data: Describes the partner that created the assessment. - :type partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData - """ - - _validation = { - 'display_name': {'required': True}, - 'policy_definition_id': {'readonly': True}, - 'severity': {'required': True}, - 'assessment_type': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'policy_definition_id': {'key': 'policyDefinitionId', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'remediation_description': {'key': 'remediationDescription', 'type': 'str'}, - 'categories': {'key': 'categories', 'type': '[str]'}, - 'severity': {'key': 'severity', 'type': 'str'}, - 'user_impact': {'key': 'userImpact', 'type': 'str'}, - 'implementation_effort': {'key': 'implementationEffort', 'type': 'str'}, - 'threats': {'key': 'threats', 'type': '[str]'}, - 'preview': {'key': 'preview', 'type': 'bool'}, - 'assessment_type': {'key': 'assessmentType', 'type': 'str'}, - 'partner_data': {'key': 'partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentMetadataProperties, self).__init__(**kwargs) - self.display_name = kwargs['display_name'] - self.policy_definition_id = None - self.description = kwargs.get('description', None) - self.remediation_description = kwargs.get('remediation_description', None) - self.categories = kwargs.get('categories', None) - self.severity = kwargs['severity'] - self.user_impact = kwargs.get('user_impact', None) - self.implementation_effort = kwargs.get('implementation_effort', None) - self.threats = kwargs.get('threats', None) - self.preview = kwargs.get('preview', None) - self.assessment_type = kwargs['assessment_type'] - self.partner_data = kwargs.get('partner_data', None) - - -class SecurityAssessmentPartnerData(msrest.serialization.Model): - """Data regarding 3rd party partner integration. - - All required parameters must be populated in order to send to Azure. - - :param partner_name: Required. Name of the company of the partner. - :type partner_name: str - :param secret: Required. secret to authenticate the partner - write only. - :type secret: str - """ - - _validation = { - 'partner_name': {'required': True}, - 'secret': {'required': True}, - } - - _attribute_map = { - 'partner_name': {'key': 'partnerName', 'type': 'str'}, - 'secret': {'key': 'secret', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentPartnerData, self).__init__(**kwargs) - self.partner_name = kwargs['partner_name'] - self.secret = kwargs['secret'] - - -class SecurityContact(Resource): - """Contact details for security issues. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param email: The email of this security contact. - :type email: str - :param phone: The phone number of this security contact. - :type phone: str - :param alert_notifications: Whether to send security alerts notifications to the security - contact. Possible values include: "On", "Off". - :type alert_notifications: str or ~azure.mgmt.security.models.AlertNotifications - :param alerts_to_admins: Whether to send security alerts notifications to subscription admins. - Possible values include: "On", "Off". - :type alerts_to_admins: str or ~azure.mgmt.security.models.AlertsToAdmins - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'email': {'key': 'properties.email', 'type': 'str'}, - 'phone': {'key': 'properties.phone', 'type': 'str'}, - 'alert_notifications': {'key': 'properties.alertNotifications', 'type': 'str'}, - 'alerts_to_admins': {'key': 'properties.alertsToAdmins', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityContact, self).__init__(**kwargs) - self.email = kwargs.get('email', None) - self.phone = kwargs.get('phone', None) - self.alert_notifications = kwargs.get('alert_notifications', None) - self.alerts_to_admins = kwargs.get('alerts_to_admins', None) - - -class SecurityContactList(msrest.serialization.Model): - """List of security contacts response. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: List of security contacts. - :vartype value: list[~azure.mgmt.security.models.SecurityContact] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecurityContact]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityContactList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecuritySolution(Resource, Location): - """SecuritySolution. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param security_family: The security family of the security solution. Possible values include: - "Waf", "Ngfw", "SaasWaf", "Va". - :type security_family: str or ~azure.mgmt.security.models.SecurityFamily - :param provisioning_state: The security family provisioning State. Possible values include: - "Succeeded", "Failed", "Updating". - :type provisioning_state: str or ~azure.mgmt.security.models.ProvisioningState - :param template: The security solutions' template. - :type template: str - :param protection_status: The security solutions' status. - :type protection_status: str - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'security_family': {'key': 'properties.securityFamily', 'type': 'str'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, - 'template': {'key': 'properties.template', 'type': 'str'}, - 'protection_status': {'key': 'properties.protectionStatus', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecuritySolution, self).__init__(**kwargs) - self.location = None - self.security_family = kwargs.get('security_family', None) - self.provisioning_state = kwargs.get('provisioning_state', None) - self.template = kwargs.get('template', None) - self.protection_status = kwargs.get('protection_status', None) - self.id = None - self.name = None - self.type = None - self.security_family = kwargs.get('security_family', None) - self.provisioning_state = kwargs.get('provisioning_state', None) - self.template = kwargs.get('template', None) - self.protection_status = kwargs.get('protection_status', None) - - -class SecuritySolutionList(msrest.serialization.Model): - """SecuritySolutionList. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: - :type value: list[~azure.mgmt.security.models.SecuritySolution] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecuritySolution]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecuritySolutionList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class SecuritySolutionsReferenceData(Resource, Location): - """SecuritySolutionsReferenceData. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param security_family: Required. The security family of the security solution. Possible values - include: "Waf", "Ngfw", "SaasWaf", "Va". - :type security_family: str or ~azure.mgmt.security.models.SecurityFamily - :param alert_vendor_name: Required. The security solutions' vendor name. - :type alert_vendor_name: str - :param package_info_url: Required. The security solutions' package info url. - :type package_info_url: str - :param product_name: Required. The security solutions' product name. - :type product_name: str - :param publisher: Required. The security solutions' publisher. - :type publisher: str - :param publisher_display_name: Required. The security solutions' publisher display name. - :type publisher_display_name: str - :param template: Required. The security solutions' template. - :type template: str - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'security_family': {'required': True}, - 'alert_vendor_name': {'required': True}, - 'package_info_url': {'required': True}, - 'product_name': {'required': True}, - 'publisher': {'required': True}, - 'publisher_display_name': {'required': True}, - 'template': {'required': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'security_family': {'key': 'properties.securityFamily', 'type': 'str'}, - 'alert_vendor_name': {'key': 'properties.alertVendorName', 'type': 'str'}, - 'package_info_url': {'key': 'properties.packageInfoUrl', 'type': 'str'}, - 'product_name': {'key': 'properties.productName', 'type': 'str'}, - 'publisher': {'key': 'properties.publisher', 'type': 'str'}, - 'publisher_display_name': {'key': 'properties.publisherDisplayName', 'type': 'str'}, - 'template': {'key': 'properties.template', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecuritySolutionsReferenceData, self).__init__(**kwargs) - self.location = None - self.security_family = kwargs['security_family'] - self.alert_vendor_name = kwargs['alert_vendor_name'] - self.package_info_url = kwargs['package_info_url'] - self.product_name = kwargs['product_name'] - self.publisher = kwargs['publisher'] - self.publisher_display_name = kwargs['publisher_display_name'] - self.template = kwargs['template'] - self.id = None - self.name = None - self.type = None - self.security_family = kwargs['security_family'] - self.alert_vendor_name = kwargs['alert_vendor_name'] - self.package_info_url = kwargs['package_info_url'] - self.product_name = kwargs['product_name'] - self.publisher = kwargs['publisher'] - self.publisher_display_name = kwargs['publisher_display_name'] - self.template = kwargs['template'] - - -class SecuritySolutionsReferenceDataList(msrest.serialization.Model): - """SecuritySolutionsReferenceDataList. - - :param value: - :type value: list[~azure.mgmt.security.models.SecuritySolutionsReferenceData] - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecuritySolutionsReferenceData]'}, - } - - def __init__( - self, - **kwargs - ): - super(SecuritySolutionsReferenceDataList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - - -class SecuritySubAssessment(Resource): - """Security sub-assessment on a resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar id_properties_id: Vulnerability ID. - :vartype id_properties_id: str - :ivar display_name: User friendly display name of the sub-assessment. - :vartype display_name: str - :param status: Status of the sub-assessment. - :type status: ~azure.mgmt.security.models.SubAssessmentStatus - :ivar remediation: Information on how to remediate this sub-assessment. - :vartype remediation: str - :ivar impact: Description of the impact of this sub-assessment. - :vartype impact: str - :ivar category: Category of the sub-assessment. - :vartype category: str - :ivar description: Human readable description of the assessment status. - :vartype description: str - :ivar time_generated: The date and time the sub-assessment was generated. - :vartype time_generated: ~datetime.datetime - :param resource_details: Details of the resource that was assessed. - :type resource_details: ~azure.mgmt.security.models.ResourceDetails - :param additional_data: Details of the sub-assessment. - :type additional_data: ~azure.mgmt.security.models.AdditionalData - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'id_properties_id': {'readonly': True}, - 'display_name': {'readonly': True}, - 'remediation': {'readonly': True}, - 'impact': {'readonly': True}, - 'category': {'readonly': True}, - 'description': {'readonly': True}, - 'time_generated': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'id_properties_id': {'key': 'properties.id', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'status': {'key': 'properties.status', 'type': 'SubAssessmentStatus'}, - 'remediation': {'key': 'properties.remediation', 'type': 'str'}, - 'impact': {'key': 'properties.impact', 'type': 'str'}, - 'category': {'key': 'properties.category', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'time_generated': {'key': 'properties.timeGenerated', 'type': 'iso-8601'}, - 'resource_details': {'key': 'properties.resourceDetails', 'type': 'ResourceDetails'}, - 'additional_data': {'key': 'properties.additionalData', 'type': 'AdditionalData'}, - } - - def __init__( - self, - **kwargs - ): - super(SecuritySubAssessment, self).__init__(**kwargs) - self.id_properties_id = None - self.display_name = None - self.status = kwargs.get('status', None) - self.remediation = None - self.impact = None - self.category = None - self.description = None - self.time_generated = None - self.resource_details = kwargs.get('resource_details', None) - self.additional_data = kwargs.get('additional_data', None) - - -class SecuritySubAssessmentList(msrest.serialization.Model): - """List of security sub-assessments. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: - :vartype value: list[~azure.mgmt.security.models.SecuritySubAssessment] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecuritySubAssessment]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecuritySubAssessmentList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecurityTask(Resource): - """Security task that we recommend to do in order to strengthen security. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar state: State of the task (Active, Resolved etc.). - :vartype state: str - :ivar creation_time_utc: The time this task was discovered in UTC. - :vartype creation_time_utc: ~datetime.datetime - :param security_task_parameters: Changing set of properties, depending on the task type that is - derived from the name field. - :type security_task_parameters: ~azure.mgmt.security.models.SecurityTaskParameters - :ivar last_state_change_time_utc: The time this task's details were last changed in UTC. - :vartype last_state_change_time_utc: ~datetime.datetime - :ivar sub_state: Additional data on the state of the task. - :vartype sub_state: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time_utc': {'readonly': True}, - 'last_state_change_time_utc': {'readonly': True}, - 'sub_state': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'creation_time_utc': {'key': 'properties.creationTimeUtc', 'type': 'iso-8601'}, - 'security_task_parameters': {'key': 'properties.securityTaskParameters', 'type': 'SecurityTaskParameters'}, - 'last_state_change_time_utc': {'key': 'properties.lastStateChangeTimeUtc', 'type': 'iso-8601'}, - 'sub_state': {'key': 'properties.subState', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityTask, self).__init__(**kwargs) - self.state = None - self.creation_time_utc = None - self.security_task_parameters = kwargs.get('security_task_parameters', None) - self.last_state_change_time_utc = None - self.sub_state = None - - -class SecurityTaskList(msrest.serialization.Model): - """List of security task recommendations. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: - :vartype value: list[~azure.mgmt.security.models.SecurityTask] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecurityTask]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityTaskList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecurityTaskParameters(msrest.serialization.Model): - """Changing set of properties, depending on the task type that is derived from the name field. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :ivar name: Name of the task type. - :vartype name: str - """ - - _validation = { - 'name': {'readonly': True}, - } - - _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'name': {'key': 'name', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityTaskParameters, self).__init__(**kwargs) - self.additional_properties = kwargs.get('additional_properties', None) - self.name = None - - -class SensitivityLabel(msrest.serialization.Model): - """The sensitivity label. - - :param display_name: The name of the sensitivity label. - :type display_name: str - :param description: The description of the sensitivity label. - :type description: str - :param rank: The rank of the sensitivity label. Possible values include: "None", "Low", - "Medium", "High", "Critical". - :type rank: str or ~azure.mgmt.security.models.Rank - :param order: The order of the sensitivity label. - :type order: int - :param enabled: Indicates whether the label is enabled or not. - :type enabled: bool - """ - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'rank': {'key': 'rank', 'type': 'str'}, - 'order': {'key': 'order', 'type': 'int'}, - 'enabled': {'key': 'enabled', 'type': 'bool'}, - } - - def __init__( - self, - **kwargs - ): - super(SensitivityLabel, self).__init__(**kwargs) - self.display_name = kwargs.get('display_name', None) - self.description = kwargs.get('description', None) - self.rank = kwargs.get('rank', None) - self.order = kwargs.get('order', None) - self.enabled = kwargs.get('enabled', None) - - -class ServerVulnerabilityAssessment(Resource): - """Describes the server vulnerability assessment details on a resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar provisioning_state: The provisioningState of the vulnerability assessment capability on - the VM. Possible values include: "Succeeded", "Failed", "Canceled", "Provisioning", - "Deprovisioning". - :vartype provisioning_state: str or - ~azure.mgmt.security.models.ServerVulnerabilityAssessmentPropertiesProvisioningState - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ServerVulnerabilityAssessment, self).__init__(**kwargs) - self.provisioning_state = None - - -class ServerVulnerabilityAssessmentsList(msrest.serialization.Model): - """List of server vulnerability assessments. - - :param value: - :type value: list[~azure.mgmt.security.models.ServerVulnerabilityAssessment] - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[ServerVulnerabilityAssessment]'}, - } - - def __init__( - self, - **kwargs - ): - super(ServerVulnerabilityAssessmentsList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - - -class ServerVulnerabilityProperties(AdditionalData): - """Additional context fields for server vulnerability assessment. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", - "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType - :ivar type: Vulnerability Type. e.g: Vulnerability, Potential Vulnerability, Information - Gathered. - :vartype type: str - :ivar cvss: Dictionary from cvss version to cvss details object. - :vartype cvss: dict[str, ~azure.mgmt.security.models.CVSS] - :ivar patchable: Indicates whether a patch is available or not. - :vartype patchable: bool - :ivar cve: List of CVEs. - :vartype cve: list[~azure.mgmt.security.models.CVE] - :ivar threat: Threat name. - :vartype threat: str - :ivar published_time: Published time. - :vartype published_time: ~datetime.datetime - :ivar vendor_references: - :vartype vendor_references: list[~azure.mgmt.security.models.VendorReference] - """ - - _validation = { - 'assessed_resource_type': {'required': True}, - 'type': {'readonly': True}, - 'cvss': {'readonly': True}, - 'patchable': {'readonly': True}, - 'cve': {'readonly': True}, - 'threat': {'readonly': True}, - 'published_time': {'readonly': True}, - 'vendor_references': {'readonly': True}, - } - - _attribute_map = { - 'assessed_resource_type': {'key': 'assessedResourceType', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'cvss': {'key': 'cvss', 'type': '{CVSS}'}, - 'patchable': {'key': 'patchable', 'type': 'bool'}, - 'cve': {'key': 'cve', 'type': '[CVE]'}, - 'threat': {'key': 'threat', 'type': 'str'}, - 'published_time': {'key': 'publishedTime', 'type': 'iso-8601'}, - 'vendor_references': {'key': 'vendorReferences', 'type': '[VendorReference]'}, - } - - def __init__( - self, - **kwargs - ): - super(ServerVulnerabilityProperties, self).__init__(**kwargs) - self.assessed_resource_type = 'ServerVulnerabilityAssessment' # type: str - self.type = None - self.cvss = None - self.patchable = None - self.cve = None - self.threat = None - self.published_time = None - self.vendor_references = None - - -class ServicePrincipalProperties(msrest.serialization.Model): - """Details of the service principal. - - :param application_id: Application ID of service principal. - :type application_id: str - :param secret: A secret string that the application uses to prove its identity, also can be - referred to as application password (write only). - :type secret: str - """ - - _attribute_map = { - 'application_id': {'key': 'applicationId', 'type': 'str'}, - 'secret': {'key': 'secret', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(ServicePrincipalProperties, self).__init__(**kwargs) - self.application_id = kwargs.get('application_id', None) - self.secret = kwargs.get('secret', None) - - -class SettingsList(msrest.serialization.Model): - """Subscription settings list. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: The settings list. - :type value: list[~azure.mgmt.security.models.Setting] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Setting]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SettingsList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class Software(Resource): - """Represents a software data. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param device_id: Unique identifier for the virtual machine in the service. - :type device_id: str - :param os_platform: Platform of the operating system running on the device. - :type os_platform: str - :param vendor: Name of the software vendor. - :type vendor: str - :param software_name: Name of the software product. - :type software_name: str - :param version: Version number of the software product. - :type version: str - :param end_of_support_status: End of support status. Possible values include: "None", - "noLongerSupported", "versionNoLongerSupported", "upcomingNoLongerSupported", - "upcomingVersionNoLongerSupported". - :type end_of_support_status: str or ~azure.mgmt.security.models.EndOfSupportStatus - :param end_of_support_date: The end of support date in case the product is upcoming end of - support. - :type end_of_support_date: str - :param number_of_known_vulnerabilities: Number of weaknesses. - :type number_of_known_vulnerabilities: int - :param first_seen_at: First time that the software was seen in the device. - :type first_seen_at: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'device_id': {'key': 'properties.deviceId', 'type': 'str'}, - 'os_platform': {'key': 'properties.osPlatform', 'type': 'str'}, - 'vendor': {'key': 'properties.vendor', 'type': 'str'}, - 'software_name': {'key': 'properties.softwareName', 'type': 'str'}, - 'version': {'key': 'properties.version', 'type': 'str'}, - 'end_of_support_status': {'key': 'properties.endOfSupportStatus', 'type': 'str'}, - 'end_of_support_date': {'key': 'properties.endOfSupportDate', 'type': 'str'}, - 'number_of_known_vulnerabilities': {'key': 'properties.numberOfKnownVulnerabilities', 'type': 'int'}, - 'first_seen_at': {'key': 'properties.firstSeenAt', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(Software, self).__init__(**kwargs) - self.device_id = kwargs.get('device_id', None) - self.os_platform = kwargs.get('os_platform', None) - self.vendor = kwargs.get('vendor', None) - self.software_name = kwargs.get('software_name', None) - self.version = kwargs.get('version', None) - self.end_of_support_status = kwargs.get('end_of_support_status', None) - self.end_of_support_date = kwargs.get('end_of_support_date', None) - self.number_of_known_vulnerabilities = kwargs.get('number_of_known_vulnerabilities', None) - self.first_seen_at = kwargs.get('first_seen_at', None) - - -class SoftwaresList(msrest.serialization.Model): - """Represents the software inventory of the virtual machine. - - Variables are only populated by the server, and will be ignored when sending a request. - - :param value: - :type value: list[~azure.mgmt.security.models.Software] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Software]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SoftwaresList, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - self.next_link = None - - -class SqlServerVulnerabilityProperties(AdditionalData): - """Details of the resource that was assessed. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", - "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType - :ivar type: The resource type the sub assessment refers to in its resource details. - :vartype type: str - :ivar query: The T-SQL query that runs on your SQL database to perform the particular check. - :vartype query: str - """ - - _validation = { - 'assessed_resource_type': {'required': True}, - 'type': {'readonly': True}, - 'query': {'readonly': True}, - } - - _attribute_map = { - 'assessed_resource_type': {'key': 'assessedResourceType', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'query': {'key': 'query', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SqlServerVulnerabilityProperties, self).__init__(**kwargs) - self.assessed_resource_type = 'SqlServerVulnerability' # type: str - self.type = None - self.query = None - - -class SubAssessmentStatus(msrest.serialization.Model): - """Status of the sub-assessment. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar code: Programmatic code for the status of the assessment. Possible values include: - "Healthy", "Unhealthy", "NotApplicable". - :vartype code: str or ~azure.mgmt.security.models.SubAssessmentStatusCode - :ivar cause: Programmatic code for the cause of the assessment status. - :vartype cause: str - :ivar description: Human readable description of the assessment status. - :vartype description: str - :ivar severity: The sub-assessment severity level. Possible values include: "Low", "Medium", - "High". - :vartype severity: str or ~azure.mgmt.security.models.Severity - """ - - _validation = { - 'code': {'readonly': True}, - 'cause': {'readonly': True}, - 'description': {'readonly': True}, - 'severity': {'readonly': True}, - } - - _attribute_map = { - 'code': {'key': 'code', 'type': 'str'}, - 'cause': {'key': 'cause', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'severity': {'key': 'severity', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SubAssessmentStatus, self).__init__(**kwargs) - self.code = None - self.cause = None - self.description = None - self.severity = None - - -class SuppressionAlertsScope(msrest.serialization.Model): - """SuppressionAlertsScope. - - All required parameters must be populated in order to send to Azure. - - :param all_of: Required. All the conditions inside need to be true in order to suppress the - alert. - :type all_of: list[~azure.mgmt.security.models.ScopeElement] - """ - - _validation = { - 'all_of': {'required': True}, - } - - _attribute_map = { - 'all_of': {'key': 'allOf', 'type': '[ScopeElement]'}, - } - - def __init__( - self, - **kwargs - ): - super(SuppressionAlertsScope, self).__init__(**kwargs) - self.all_of = kwargs['all_of'] - - -class SystemData(msrest.serialization.Model): - """Metadata pertaining to creation and last modification of the resource. - - :param created_by: The identity that created the resource. - :type created_by: str - :param created_by_type: The type of identity that created the resource. Possible values - include: "User", "Application", "ManagedIdentity", "Key". - :type created_by_type: str or ~azure.mgmt.security.models.CreatedByType - :param created_at: The timestamp of resource creation (UTC). - :type created_at: ~datetime.datetime - :param last_modified_by: The identity that last modified the resource. - :type last_modified_by: str - :param last_modified_by_type: The type of identity that last modified the resource. Possible - values include: "User", "Application", "ManagedIdentity", "Key". - :type last_modified_by_type: str or ~azure.mgmt.security.models.CreatedByType - :param last_modified_at: The timestamp of resource last modification (UTC). - :type last_modified_at: ~datetime.datetime - """ - - _attribute_map = { - 'created_by': {'key': 'createdBy', 'type': 'str'}, - 'created_by_type': {'key': 'createdByType', 'type': 'str'}, - 'created_at': {'key': 'createdAt', 'type': 'iso-8601'}, - 'last_modified_by': {'key': 'lastModifiedBy', 'type': 'str'}, - 'last_modified_by_type': {'key': 'lastModifiedByType', 'type': 'str'}, - 'last_modified_at': {'key': 'lastModifiedAt', 'type': 'iso-8601'}, - } - - def __init__( - self, - **kwargs - ): - super(SystemData, self).__init__(**kwargs) - self.created_by = kwargs.get('created_by', None) - self.created_by_type = kwargs.get('created_by_type', None) - self.created_at = kwargs.get('created_at', None) - self.last_modified_by = kwargs.get('last_modified_by', None) - self.last_modified_by_type = kwargs.get('last_modified_by_type', None) - self.last_modified_at = kwargs.get('last_modified_at', None) - - -class TopologyList(msrest.serialization.Model): - """TopologyList. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: - :vartype value: list[~azure.mgmt.security.models.TopologyResource] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[TopologyResource]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(TopologyList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class TopologyResource(Resource, Location): - """TopologyResource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar location: Location where the resource is stored. - :vartype location: str - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar calculated_date_time: The UTC time on which the topology was calculated. - :vartype calculated_date_time: ~datetime.datetime - :ivar topology_resources: Azure resources which are part of this topology resource. - :vartype topology_resources: list[~azure.mgmt.security.models.TopologySingleResource] - """ - - _validation = { - 'location': {'readonly': True}, - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'calculated_date_time': {'readonly': True}, - 'topology_resources': {'readonly': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'calculated_date_time': {'key': 'properties.calculatedDateTime', 'type': 'iso-8601'}, - 'topology_resources': {'key': 'properties.topologyResources', 'type': '[TopologySingleResource]'}, - } - - def __init__( - self, - **kwargs - ): - super(TopologyResource, self).__init__(**kwargs) - self.location = None - self.calculated_date_time = None - self.topology_resources = None - self.id = None - self.name = None - self.type = None - self.calculated_date_time = None - self.topology_resources = None - - -class TopologySingleResource(msrest.serialization.Model): - """TopologySingleResource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar resource_id: Azure resource id. - :vartype resource_id: str - :ivar severity: The security severity of the resource. - :vartype severity: str - :ivar recommendations_exist: Indicates if the resource has security recommendations. - :vartype recommendations_exist: bool - :ivar network_zones: Indicates the resource connectivity level to the Internet (InternetFacing, - Internal ,etc.). - :vartype network_zones: str - :ivar topology_score: Score of the resource based on its security severity. - :vartype topology_score: int - :ivar location: The location of this resource. - :vartype location: str - :ivar parents: Azure resources connected to this resource which are in higher level in the - topology view. - :vartype parents: list[~azure.mgmt.security.models.TopologySingleResourceParent] - :ivar children: Azure resources connected to this resource which are in lower level in the - topology view. - :vartype children: list[~azure.mgmt.security.models.TopologySingleResourceChild] - """ - - _validation = { - 'resource_id': {'readonly': True}, - 'severity': {'readonly': True}, - 'recommendations_exist': {'readonly': True}, - 'network_zones': {'readonly': True}, - 'topology_score': {'readonly': True}, - 'location': {'readonly': True}, - 'parents': {'readonly': True}, - 'children': {'readonly': True}, - } - - _attribute_map = { - 'resource_id': {'key': 'resourceId', 'type': 'str'}, - 'severity': {'key': 'severity', 'type': 'str'}, - 'recommendations_exist': {'key': 'recommendationsExist', 'type': 'bool'}, - 'network_zones': {'key': 'networkZones', 'type': 'str'}, - 'topology_score': {'key': 'topologyScore', 'type': 'int'}, - 'location': {'key': 'location', 'type': 'str'}, - 'parents': {'key': 'parents', 'type': '[TopologySingleResourceParent]'}, - 'children': {'key': 'children', 'type': '[TopologySingleResourceChild]'}, - } - - def __init__( - self, - **kwargs - ): - super(TopologySingleResource, self).__init__(**kwargs) - self.resource_id = None - self.severity = None - self.recommendations_exist = None - self.network_zones = None - self.topology_score = None - self.location = None - self.parents = None - self.children = None - - -class TopologySingleResourceChild(msrest.serialization.Model): - """TopologySingleResourceChild. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar resource_id: Azure resource id which serves as child resource in topology view. - :vartype resource_id: str - """ - - _validation = { - 'resource_id': {'readonly': True}, - } - - _attribute_map = { - 'resource_id': {'key': 'resourceId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(TopologySingleResourceChild, self).__init__(**kwargs) - self.resource_id = None - - -class TopologySingleResourceParent(msrest.serialization.Model): - """TopologySingleResourceParent. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar resource_id: Azure resource id which serves as parent resource in topology view. - :vartype resource_id: str - """ - - _validation = { - 'resource_id': {'readonly': True}, - } - - _attribute_map = { - 'resource_id': {'key': 'resourceId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(TopologySingleResourceParent, self).__init__(**kwargs) - self.resource_id = None - - -class TwinUpdatesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of twin updates is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(TwinUpdatesNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'TwinUpdatesNotInAllowedRange' # type: str - - -class UnauthorizedOperationsNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of unauthorized operations is not in allowed range. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, - } - - def __init__( - self, - **kwargs - ): - super(UnauthorizedOperationsNotInAllowedRange, self).__init__(**kwargs) - self.rule_type = 'UnauthorizedOperationsNotInAllowedRange' # type: str - - -class UpdateIotSecuritySolutionData(TagsResource): - """UpdateIotSecuritySolutionData. - - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :param user_defined_resources: Properties of the IoT Security solution's user defined - resources. - :type user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties - :param recommendations_configuration: List of the configuration status for each recommendation - type. - :type recommendations_configuration: - list[~azure.mgmt.security.models.RecommendationConfigurationProperties] - """ - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'user_defined_resources': {'key': 'properties.userDefinedResources', 'type': 'UserDefinedResourcesProperties'}, - 'recommendations_configuration': {'key': 'properties.recommendationsConfiguration', 'type': '[RecommendationConfigurationProperties]'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateIotSecuritySolutionData, self).__init__(**kwargs) - self.user_defined_resources = kwargs.get('user_defined_resources', None) - self.recommendations_configuration = kwargs.get('recommendations_configuration', None) - - -class UserDefinedResourcesProperties(msrest.serialization.Model): - """Properties of the IoT Security solution's user defined resources. - - All required parameters must be populated in order to send to Azure. - - :param query: Required. Azure Resource Graph query which represents the security solution's - user defined resources. Required to start with "where type != "Microsoft.Devices/IotHubs"". - :type query: str - :param query_subscriptions: Required. List of Azure subscription ids on which the user defined - resources query should be executed. - :type query_subscriptions: list[str] - """ - - _validation = { - 'query': {'required': True}, - 'query_subscriptions': {'required': True}, - } - - _attribute_map = { - 'query': {'key': 'query', 'type': 'str'}, - 'query_subscriptions': {'key': 'querySubscriptions', 'type': '[str]'}, - } - - def __init__( - self, - **kwargs - ): - super(UserDefinedResourcesProperties, self).__init__(**kwargs) - self.query = kwargs['query'] - self.query_subscriptions = kwargs['query_subscriptions'] - - -class UserRecommendation(msrest.serialization.Model): - """Represents a user that is recommended to be allowed for a certain rule. - - :param username: Represents a user that is recommended to be allowed for a certain rule. - :type username: str - :param recommendation_action: The recommendation action of the machine or rule. Possible values - include: "Recommended", "Add", "Remove". - :type recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction - """ - - _attribute_map = { - 'username': {'key': 'username', 'type': 'str'}, - 'recommendation_action': {'key': 'recommendationAction', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UserRecommendation, self).__init__(**kwargs) - self.username = kwargs.get('username', None) - self.recommendation_action = kwargs.get('recommendation_action', None) - - -class VaRule(msrest.serialization.Model): - """vulnerability assessment rule metadata details. - - :param rule_id: The rule Id. - :type rule_id: str - :param severity: The rule severity. Possible values include: "High", "Medium", "Low", - "Informational", "Obsolete". - :type severity: str or ~azure.mgmt.security.models.RuleSeverity - :param category: The rule category. - :type category: str - :param rule_type: The rule type. Possible values include: "Binary", "BaselineExpected", - "PositiveList", "NegativeList". - :type rule_type: str or ~azure.mgmt.security.models.RuleType - :param title: The rule title. - :type title: str - :param description: The rule description. - :type description: str - :param rationale: The rule rationale. - :type rationale: str - :param query_check: The rule query details. - :type query_check: ~azure.mgmt.security.models.QueryCheck - :param benchmark_references: The benchmark references. - :type benchmark_references: list[~azure.mgmt.security.models.BenchmarkReference] - """ - - _attribute_map = { - 'rule_id': {'key': 'ruleId', 'type': 'str'}, - 'severity': {'key': 'severity', 'type': 'str'}, - 'category': {'key': 'category', 'type': 'str'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'title': {'key': 'title', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'rationale': {'key': 'rationale', 'type': 'str'}, - 'query_check': {'key': 'queryCheck', 'type': 'QueryCheck'}, - 'benchmark_references': {'key': 'benchmarkReferences', 'type': '[BenchmarkReference]'}, - } - - def __init__( - self, - **kwargs - ): - super(VaRule, self).__init__(**kwargs) - self.rule_id = kwargs.get('rule_id', None) - self.severity = kwargs.get('severity', None) - self.category = kwargs.get('category', None) - self.rule_type = kwargs.get('rule_type', None) - self.title = kwargs.get('title', None) - self.description = kwargs.get('description', None) - self.rationale = kwargs.get('rationale', None) - self.query_check = kwargs.get('query_check', None) - self.benchmark_references = kwargs.get('benchmark_references', None) - - -class VendorReference(msrest.serialization.Model): - """Vendor reference. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar title: Link title. - :vartype title: str - :ivar link: Link url. - :vartype link: str - """ - - _validation = { - 'title': {'readonly': True}, - 'link': {'readonly': True}, - } - - _attribute_map = { - 'title': {'key': 'title', 'type': 'str'}, - 'link': {'key': 'link', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(VendorReference, self).__init__(**kwargs) - self.title = None - self.link = None - - -class VmRecommendation(msrest.serialization.Model): - """Represents a machine that is part of a machine group. - - :param configuration_status: The configuration status of the machines group or machine or rule. - Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", "NoStatus". - :type configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus - :param recommendation_action: The recommendation action of the machine or rule. Possible values - include: "Recommended", "Add", "Remove". - :type recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction - :param resource_id: The full resource id of the machine. - :type resource_id: str - :param enforcement_support: The machine supportability of Enforce feature. Possible values - include: "Supported", "NotSupported", "Unknown". - :type enforcement_support: str or ~azure.mgmt.security.models.EnforcementSupport - """ - - _attribute_map = { - 'configuration_status': {'key': 'configurationStatus', 'type': 'str'}, - 'recommendation_action': {'key': 'recommendationAction', 'type': 'str'}, - 'resource_id': {'key': 'resourceId', 'type': 'str'}, - 'enforcement_support': {'key': 'enforcementSupport', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(VmRecommendation, self).__init__(**kwargs) - self.configuration_status = kwargs.get('configuration_status', None) - self.recommendation_action = kwargs.get('recommendation_action', None) - self.resource_id = kwargs.get('resource_id', None) - self.enforcement_support = kwargs.get('enforcement_support', None) - - -class WorkspaceSetting(Resource): - """Configures where to store the OMS agent data for workspaces under a scope. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param workspace_id: The full Azure ID of the workspace to save the data in. - :type workspace_id: str - :param scope: All the VMs in this scope will send their security data to the mentioned - workspace unless overridden by a setting with more specific scope. - :type scope: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'workspace_id': {'key': 'properties.workspaceId', 'type': 'str'}, - 'scope': {'key': 'properties.scope', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(WorkspaceSetting, self).__init__(**kwargs) - self.workspace_id = kwargs.get('workspace_id', None) - self.scope = kwargs.get('scope', None) - - -class WorkspaceSettingList(msrest.serialization.Model): - """List of workspace settings response. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of workspace settings. - :type value: list[~azure.mgmt.security.models.WorkspaceSetting] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[WorkspaceSetting]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(WorkspaceSettingList, self).__init__(**kwargs) - self.value = kwargs['value'] - self.next_link = None diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models_py3.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models_py3.py index cc336c5b2ea0b..8a26aa22525ac 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models_py3.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_models_py3.py @@ -17,9 +17,9 @@ class AadConnectivityState(msrest.serialization.Model): """Describes an Azure resource with kind. - :param connectivity_state: The connectivity state of the external AAD solution. Possible values + :ivar connectivity_state: The connectivity state of the external AAD solution. Possible values include: "Discovered", "NotLicensed", "Connected". - :type connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum + :vartype connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum """ _attribute_map = { @@ -32,6 +32,11 @@ def __init__( connectivity_state: Optional[Union[str, "AadConnectivityStateEnum"]] = None, **kwargs ): + """ + :keyword connectivity_state: The connectivity state of the external AAD solution. Possible + values include: "Discovered", "NotLicensed", "Connected". + :paramtype connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum + """ super(AadConnectivityState, self).__init__(**kwargs) self.connectivity_state = connectivity_state @@ -39,8 +44,8 @@ def __init__( class ExternalSecuritySolutionKind(msrest.serialization.Model): """Describes an Azure resource with kind. - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :ivar kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :vartype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum """ _attribute_map = { @@ -53,6 +58,10 @@ def __init__( kind: Optional[Union[str, "ExternalSecuritySolutionKindEnum"]] = None, **kwargs ): + """ + :keyword kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :paramtype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + """ super(ExternalSecuritySolutionKind, self).__init__(**kwargs) self.kind = kind @@ -78,6 +87,8 @@ def __init__( self, **kwargs ): + """ + """ super(Location, self).__init__(**kwargs) self.location = None @@ -111,6 +122,8 @@ def __init__( self, **kwargs ): + """ + """ super(Resource, self).__init__(**kwargs) self.id = None self.name = None @@ -122,8 +135,8 @@ class ExternalSecuritySolution(Resource, Location, ExternalSecuritySolutionKind) Variables are only populated by the server, and will be ignored when sending a request. - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :ivar kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :vartype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum :ivar location: Location where the resource is stored. :vartype location: str :ivar id: Resource Id. @@ -155,6 +168,10 @@ def __init__( kind: Optional[Union[str, "ExternalSecuritySolutionKindEnum"]] = None, **kwargs ): + """ + :keyword kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :paramtype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + """ super(ExternalSecuritySolution, self).__init__(kind=kind, **kwargs) self.kind = kind self.location = None @@ -173,8 +190,8 @@ class AadExternalSecuritySolution(ExternalSecuritySolution): Variables are only populated by the server, and will be ignored when sending a request. - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :ivar kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :vartype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum :ivar location: Location where the resource is stored. :vartype location: str :ivar id: Resource Id. @@ -183,8 +200,8 @@ class AadExternalSecuritySolution(ExternalSecuritySolution): :vartype name: str :ivar type: Resource type. :vartype type: str - :param properties: The external security solution properties for AAD solutions. - :type properties: ~azure.mgmt.security.models.AadSolutionProperties + :ivar properties: The external security solution properties for AAD solutions. + :vartype properties: ~azure.mgmt.security.models.AadSolutionProperties """ _validation = { @@ -210,6 +227,12 @@ def __init__( properties: Optional["AadSolutionProperties"] = None, **kwargs ): + """ + :keyword kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :paramtype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :keyword properties: The external security solution properties for AAD solutions. + :paramtype properties: ~azure.mgmt.security.models.AadSolutionProperties + """ super(AadExternalSecuritySolution, self).__init__(kind=kind, **kwargs) self.properties = properties @@ -217,15 +240,15 @@ def __init__( class ExternalSecuritySolutionProperties(msrest.serialization.Model): """The solution properties (correspond to the solution kind). - :param additional_properties: Unmatched properties from the message are deserialized to this + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace + :vartype additional_properties: dict[str, any] + :ivar device_vendor: + :vartype device_vendor: str + :ivar device_type: + :vartype device_type: str + :ivar workspace: Represents an OMS workspace to which the solution is connected. + :vartype workspace: ~azure.mgmt.security.models.ConnectedWorkspace """ _attribute_map = { @@ -244,6 +267,17 @@ def __init__( workspace: Optional["ConnectedWorkspace"] = None, **kwargs ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + :keyword device_vendor: + :paramtype device_vendor: str + :keyword device_type: + :paramtype device_type: str + :keyword workspace: Represents an OMS workspace to which the solution is connected. + :paramtype workspace: ~azure.mgmt.security.models.ConnectedWorkspace + """ super(ExternalSecuritySolutionProperties, self).__init__(**kwargs) self.additional_properties = additional_properties self.device_vendor = device_vendor @@ -254,18 +288,18 @@ def __init__( class AadSolutionProperties(ExternalSecuritySolutionProperties, AadConnectivityState): """The external security solution properties for AAD solutions. - :param connectivity_state: The connectivity state of the external AAD solution. Possible values + :ivar connectivity_state: The connectivity state of the external AAD solution. Possible values include: "Discovered", "NotLicensed", "Connected". - :type connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum - :param additional_properties: Unmatched properties from the message are deserialized to this + :vartype connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace + :vartype additional_properties: dict[str, any] + :ivar device_vendor: + :vartype device_vendor: str + :ivar device_type: + :vartype device_type: str + :ivar workspace: Represents an OMS workspace to which the solution is connected. + :vartype workspace: ~azure.mgmt.security.models.ConnectedWorkspace """ _attribute_map = { @@ -286,6 +320,20 @@ def __init__( workspace: Optional["ConnectedWorkspace"] = None, **kwargs ): + """ + :keyword connectivity_state: The connectivity state of the external AAD solution. Possible + values include: "Discovered", "NotLicensed", "Connected". + :paramtype connectivity_state: str or ~azure.mgmt.security.models.AadConnectivityStateEnum + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + :keyword device_vendor: + :paramtype device_vendor: str + :keyword device_type: + :paramtype device_type: str + :keyword workspace: Represents an OMS workspace to which the solution is connected. + :paramtype workspace: ~azure.mgmt.security.models.ConnectedWorkspace + """ super(AadSolutionProperties, self).__init__(additional_properties=additional_properties, device_vendor=device_vendor, device_type=device_type, workspace=workspace, connectivity_state=connectivity_state, **kwargs) self.connectivity_state = connectivity_state self.additional_properties = additional_properties @@ -308,10 +356,10 @@ class CustomAlertRule(msrest.serialization.Model): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str """ _validation = { @@ -338,6 +386,10 @@ def __init__( is_enabled: bool, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + """ super(CustomAlertRule, self).__init__(**kwargs) self.display_name = None self.description = None @@ -359,14 +411,14 @@ class ThresholdCustomAlertRule(CustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int """ _validation = { @@ -399,6 +451,14 @@ def __init__( max_threshold: int, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + """ super(ThresholdCustomAlertRule, self).__init__(is_enabled=is_enabled, **kwargs) self.rule_type = 'ThresholdCustomAlertRule' # type: str self.min_threshold = min_threshold @@ -419,16 +479,16 @@ class TimeWindowCustomAlertRule(ThresholdCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -464,6 +524,16 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ super(TimeWindowCustomAlertRule, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, **kwargs) self.rule_type = 'TimeWindowCustomAlertRule' # type: str self.time_window_size = time_window_size @@ -480,16 +550,16 @@ class ActiveConnectionsNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -521,6 +591,16 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ super(ActiveConnectionsNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) self.rule_type = 'ActiveConnectionsNotInAllowedRange' # type: str @@ -538,12 +618,12 @@ class AdaptiveApplicationControlGroup(Resource, Location): :vartype name: str :ivar type: Resource type. :vartype type: str - :param enforcement_mode: The application control policy enforcement/protection mode of the + :ivar enforcement_mode: The application control policy enforcement/protection mode of the machine group. Possible values include: "Audit", "Enforce", "None". - :type enforcement_mode: str or ~azure.mgmt.security.models.EnforcementMode - :param protection_mode: The protection mode of the collection/file types. Exe/Msi/Script are + :vartype enforcement_mode: str or ~azure.mgmt.security.models.EnforcementMode + :ivar protection_mode: The protection mode of the collection/file types. Exe/Msi/Script are used for Windows, Executable is used for Linux. - :type protection_mode: ~azure.mgmt.security.models.ProtectionMode + :vartype protection_mode: ~azure.mgmt.security.models.ProtectionMode :ivar configuration_status: The configuration status of the machines group or machine or rule. Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", "NoStatus". :vartype configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus @@ -555,10 +635,10 @@ class AdaptiveApplicationControlGroup(Resource, Location): :ivar source_system: The source type of the machine group. Possible values include: "Azure_AppLocker", "Azure_AuditD", "NonAzure_AppLocker", "NonAzure_AuditD", "None". :vartype source_system: str or ~azure.mgmt.security.models.SourceSystem - :param vm_recommendations: - :type vm_recommendations: list[~azure.mgmt.security.models.VmRecommendation] - :param path_recommendations: - :type path_recommendations: list[~azure.mgmt.security.models.PathRecommendation] + :ivar vm_recommendations: + :vartype vm_recommendations: list[~azure.mgmt.security.models.VmRecommendation] + :ivar path_recommendations: + :vartype path_recommendations: list[~azure.mgmt.security.models.PathRecommendation] """ _validation = { @@ -596,6 +676,18 @@ def __init__( path_recommendations: Optional[List["PathRecommendation"]] = None, **kwargs ): + """ + :keyword enforcement_mode: The application control policy enforcement/protection mode of the + machine group. Possible values include: "Audit", "Enforce", "None". + :paramtype enforcement_mode: str or ~azure.mgmt.security.models.EnforcementMode + :keyword protection_mode: The protection mode of the collection/file types. Exe/Msi/Script are + used for Windows, Executable is used for Linux. + :paramtype protection_mode: ~azure.mgmt.security.models.ProtectionMode + :keyword vm_recommendations: + :paramtype vm_recommendations: list[~azure.mgmt.security.models.VmRecommendation] + :keyword path_recommendations: + :paramtype path_recommendations: list[~azure.mgmt.security.models.PathRecommendation] + """ super(AdaptiveApplicationControlGroup, self).__init__(**kwargs) self.location = None self.enforcement_mode = enforcement_mode @@ -622,8 +714,8 @@ def __init__( class AdaptiveApplicationControlGroups(msrest.serialization.Model): """Represents a list of machine groups and set of rules that are recommended by Azure Security Center to be allowed. - :param value: - :type value: list[~azure.mgmt.security.models.AdaptiveApplicationControlGroup] + :ivar value: + :vartype value: list[~azure.mgmt.security.models.AdaptiveApplicationControlGroup] """ _attribute_map = { @@ -636,6 +728,10 @@ def __init__( value: Optional[List["AdaptiveApplicationControlGroup"]] = None, **kwargs ): + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.AdaptiveApplicationControlGroup] + """ super(AdaptiveApplicationControlGroups, self).__init__(**kwargs) self.value = value @@ -643,12 +739,12 @@ def __init__( class AdaptiveApplicationControlIssueSummary(msrest.serialization.Model): """Represents a summary of the alerts of the machine group. - :param issue: An alert that machines within a group can have. Possible values include: + :ivar issue: An alert that machines within a group can have. Possible values include: "ViolationsAudited", "ViolationsBlocked", "MsiAndScriptViolationsAudited", "MsiAndScriptViolationsBlocked", "ExecutableViolationsAudited", "RulesViolatedManually". - :type issue: str or ~azure.mgmt.security.models.AdaptiveApplicationControlIssue - :param number_of_vms: The number of machines in the group that have this alert. - :type number_of_vms: float + :vartype issue: str or ~azure.mgmt.security.models.AdaptiveApplicationControlIssue + :ivar number_of_vms: The number of machines in the group that have this alert. + :vartype number_of_vms: float """ _attribute_map = { @@ -663,6 +759,14 @@ def __init__( number_of_vms: Optional[float] = None, **kwargs ): + """ + :keyword issue: An alert that machines within a group can have. Possible values include: + "ViolationsAudited", "ViolationsBlocked", "MsiAndScriptViolationsAudited", + "MsiAndScriptViolationsBlocked", "ExecutableViolationsAudited", "RulesViolatedManually". + :paramtype issue: str or ~azure.mgmt.security.models.AdaptiveApplicationControlIssue + :keyword number_of_vms: The number of machines in the group that have this alert. + :paramtype number_of_vms: float + """ super(AdaptiveApplicationControlIssueSummary, self).__init__(**kwargs) self.issue = issue self.number_of_vms = number_of_vms @@ -679,13 +783,13 @@ class AdaptiveNetworkHardening(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param rules: The security rules which are recommended to be effective on the VM. - :type rules: list[~azure.mgmt.security.models.Rule] - :param rules_calculation_time: The UTC time on which the rules were calculated. - :type rules_calculation_time: ~datetime.datetime - :param effective_network_security_groups: The Network Security Groups effective on the network + :ivar rules: The security rules which are recommended to be effective on the VM. + :vartype rules: list[~azure.mgmt.security.models.Rule] + :ivar rules_calculation_time: The UTC time on which the rules were calculated. + :vartype rules_calculation_time: ~datetime.datetime + :ivar effective_network_security_groups: The Network Security Groups effective on the network interfaces of the protected resource. - :type effective_network_security_groups: + :vartype effective_network_security_groups: list[~azure.mgmt.security.models.EffectiveNetworkSecurityGroups] """ @@ -712,6 +816,16 @@ def __init__( effective_network_security_groups: Optional[List["EffectiveNetworkSecurityGroups"]] = None, **kwargs ): + """ + :keyword rules: The security rules which are recommended to be effective on the VM. + :paramtype rules: list[~azure.mgmt.security.models.Rule] + :keyword rules_calculation_time: The UTC time on which the rules were calculated. + :paramtype rules_calculation_time: ~datetime.datetime + :keyword effective_network_security_groups: The Network Security Groups effective on the + network interfaces of the protected resource. + :paramtype effective_network_security_groups: + list[~azure.mgmt.security.models.EffectiveNetworkSecurityGroups] + """ super(AdaptiveNetworkHardening, self).__init__(**kwargs) self.rules = rules self.rules_calculation_time = rules_calculation_time @@ -723,12 +837,12 @@ class AdaptiveNetworkHardeningEnforceRequest(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param rules: Required. The rules to enforce. - :type rules: list[~azure.mgmt.security.models.Rule] - :param network_security_groups: Required. The Azure resource IDs of the effective network + :ivar rules: Required. The rules to enforce. + :vartype rules: list[~azure.mgmt.security.models.Rule] + :ivar network_security_groups: Required. The Azure resource IDs of the effective network security groups that will be updated with the created security rules from the Adaptive Network Hardening rules. - :type network_security_groups: list[str] + :vartype network_security_groups: list[str] """ _validation = { @@ -748,6 +862,14 @@ def __init__( network_security_groups: List[str], **kwargs ): + """ + :keyword rules: Required. The rules to enforce. + :paramtype rules: list[~azure.mgmt.security.models.Rule] + :keyword network_security_groups: Required. The Azure resource IDs of the effective network + security groups that will be updated with the created security rules from the Adaptive Network + Hardening rules. + :paramtype network_security_groups: list[str] + """ super(AdaptiveNetworkHardeningEnforceRequest, self).__init__(**kwargs) self.rules = rules self.network_security_groups = network_security_groups @@ -756,10 +878,10 @@ def __init__( class AdaptiveNetworkHardeningsList(msrest.serialization.Model): """Response for ListAdaptiveNetworkHardenings API service call. - :param value: A list of Adaptive Network Hardenings resources. - :type value: list[~azure.mgmt.security.models.AdaptiveNetworkHardening] - :param next_link: The URL to get the next set of results. - :type next_link: str + :ivar value: A list of Adaptive Network Hardenings resources. + :vartype value: list[~azure.mgmt.security.models.AdaptiveNetworkHardening] + :ivar next_link: The URL to get the next set of results. + :vartype next_link: str """ _attribute_map = { @@ -774,6 +896,12 @@ def __init__( next_link: Optional[str] = None, **kwargs ): + """ + :keyword value: A list of Adaptive Network Hardenings resources. + :paramtype value: list[~azure.mgmt.security.models.AdaptiveNetworkHardening] + :keyword next_link: The URL to get the next set of results. + :paramtype next_link: str + """ super(AdaptiveNetworkHardeningsList, self).__init__(**kwargs) self.value = value self.next_link = next_link @@ -787,10 +915,10 @@ class AdditionalData(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", + :ivar assessed_resource_type: Required. Sub-assessment resource type.Constant filled by server. + Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType + :vartype assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType """ _validation = { @@ -809,6 +937,8 @@ def __init__( self, **kwargs ): + """ + """ super(AdditionalData, self).__init__(**kwargs) self.assessed_resource_type = None # type: Optional[str] @@ -816,12 +946,12 @@ def __init__( class AdditionalWorkspacesProperties(msrest.serialization.Model): """Properties of the additional workspaces. - :param workspace: Workspace resource id. - :type workspace: str - :param type: Workspace type. Possible values include: "Sentinel". Default value: "Sentinel". - :type type: str or ~azure.mgmt.security.models.AdditionalWorkspaceType - :param data_types: List of data types sent to workspace. - :type data_types: list[str or ~azure.mgmt.security.models.AdditionalWorkspaceDataType] + :ivar workspace: Workspace resource id. + :vartype workspace: str + :ivar type: Workspace type. Possible values include: "Sentinel". Default value: "Sentinel". + :vartype type: str or ~azure.mgmt.security.models.AdditionalWorkspaceType + :ivar data_types: List of data types sent to workspace. + :vartype data_types: list[str or ~azure.mgmt.security.models.AdditionalWorkspaceDataType] """ _attribute_map = { @@ -838,6 +968,14 @@ def __init__( data_types: Optional[List[Union[str, "AdditionalWorkspaceDataType"]]] = None, **kwargs ): + """ + :keyword workspace: Workspace resource id. + :paramtype workspace: str + :keyword type: Workspace type. Possible values include: "Sentinel". Default value: "Sentinel". + :paramtype type: str or ~azure.mgmt.security.models.AdditionalWorkspaceType + :keyword data_types: List of data types sent to workspace. + :paramtype data_types: list[str or ~azure.mgmt.security.models.AdditionalWorkspaceDataType] + """ super(AdditionalWorkspacesProperties, self).__init__(**kwargs) self.workspace = workspace self.type = type @@ -855,8 +993,8 @@ class AdvancedThreatProtectionSetting(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param is_enabled: Indicates whether Advanced Threat Protection is enabled. - :type is_enabled: bool + :ivar is_enabled: Indicates whether Advanced Threat Protection is enabled. + :vartype is_enabled: bool """ _validation = { @@ -878,6 +1016,10 @@ def __init__( is_enabled: Optional[bool] = None, **kwargs ): + """ + :keyword is_enabled: Indicates whether Advanced Threat Protection is enabled. + :paramtype is_enabled: bool + """ super(AdvancedThreatProtectionSetting, self).__init__(**kwargs) self.is_enabled = is_enabled @@ -952,8 +1094,8 @@ class Alert(Resource): :ivar correlation_key: Key for corelating related alerts. Alerts with the same correlation key considered to be related. :vartype correlation_key: str - :param extended_properties: Custom properties for the alert. - :type extended_properties: dict[str, str] + :ivar extended_properties: Custom properties for the alert. + :vartype extended_properties: dict[str, str] :ivar compromised_entity: The display name of the resource most related to this alert. :vartype compromised_entity: str """ @@ -1021,6 +1163,10 @@ def __init__( extended_properties: Optional[Dict[str, str]] = None, **kwargs ): + """ + :keyword extended_properties: Custom properties for the alert. + :paramtype extended_properties: dict[str, str] + """ super(Alert, self).__init__(**kwargs) self.alert_type = None self.system_alert_id = None @@ -1052,9 +1198,9 @@ class AlertEntity(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param additional_properties: Unmatched properties from the message are deserialized to this + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] + :vartype additional_properties: dict[str, any] :ivar type: Type of entity. :vartype type: str """ @@ -1074,6 +1220,11 @@ def __init__( additional_properties: Optional[Dict[str, Any]] = None, **kwargs ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + """ super(AlertEntity, self).__init__(**kwargs) self.additional_properties = additional_properties self.type = None @@ -1084,8 +1235,8 @@ class AlertList(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param value: describes security alert properties. - :type value: list[~azure.mgmt.security.models.Alert] + :ivar value: describes security alert properties. + :vartype value: list[~azure.mgmt.security.models.Alert] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -1105,6 +1256,10 @@ def __init__( value: Optional[List["Alert"]] = None, **kwargs ): + """ + :keyword value: describes security alert properties. + :paramtype value: list[~azure.mgmt.security.models.Alert] + """ super(AlertList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -1118,12 +1273,12 @@ class AlertSimulatorRequestProperties(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param additional_properties: Unmatched properties from the message are deserialized to this + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] - :param kind: Required. The kind of alert simulation.Constant filled by server. Possible values + :vartype additional_properties: dict[str, any] + :ivar kind: Required. The kind of alert simulation.Constant filled by server. Possible values include: "Bundles". - :type kind: str or ~azure.mgmt.security.models.KindEnum + :vartype kind: str or ~azure.mgmt.security.models.KindEnum """ _validation = { @@ -1145,6 +1300,11 @@ def __init__( additional_properties: Optional[Dict[str, Any]] = None, **kwargs ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + """ super(AlertSimulatorRequestProperties, self).__init__(**kwargs) self.additional_properties = additional_properties self.kind = 'AlertSimulatorRequestProperties' # type: str @@ -1155,14 +1315,14 @@ class AlertSimulatorBundlesRequestProperties(AlertSimulatorRequestProperties): All required parameters must be populated in order to send to Azure. - :param additional_properties: Unmatched properties from the message are deserialized to this + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] - :param kind: Required. The kind of alert simulation.Constant filled by server. Possible values + :vartype additional_properties: dict[str, any] + :ivar kind: Required. The kind of alert simulation.Constant filled by server. Possible values include: "Bundles". - :type kind: str or ~azure.mgmt.security.models.KindEnum - :param bundles: Bundles list. - :type bundles: list[str or ~azure.mgmt.security.models.BundleType] + :vartype kind: str or ~azure.mgmt.security.models.KindEnum + :ivar bundles: Bundles list. + :vartype bundles: list[str or ~azure.mgmt.security.models.BundleType] """ _validation = { @@ -1182,6 +1342,13 @@ def __init__( bundles: Optional[List[Union[str, "BundleType"]]] = None, **kwargs ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + :keyword bundles: Bundles list. + :paramtype bundles: list[str or ~azure.mgmt.security.models.BundleType] + """ super(AlertSimulatorBundlesRequestProperties, self).__init__(additional_properties=additional_properties, **kwargs) self.kind = 'Bundles' # type: str self.bundles = bundles @@ -1190,8 +1357,8 @@ def __init__( class AlertSimulatorRequestBody(msrest.serialization.Model): """Alert Simulator request body. - :param properties: Alert Simulator request body data. - :type properties: ~azure.mgmt.security.models.AlertSimulatorRequestProperties + :ivar properties: Alert Simulator request body data. + :vartype properties: ~azure.mgmt.security.models.AlertSimulatorRequestProperties """ _attribute_map = { @@ -1204,6 +1371,10 @@ def __init__( properties: Optional["AlertSimulatorRequestProperties"] = None, **kwargs ): + """ + :keyword properties: Alert Simulator request body data. + :paramtype properties: ~azure.mgmt.security.models.AlertSimulatorRequestProperties + """ super(AlertSimulatorRequestBody, self).__init__(**kwargs) self.properties = properties @@ -1219,22 +1390,22 @@ class AlertsSuppressionRule(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param alert_type: Type of the alert to automatically suppress. For all alert types, use '*'. - :type alert_type: str + :ivar alert_type: Type of the alert to automatically suppress. For all alert types, use '*'. + :vartype alert_type: str :ivar last_modified_utc: The last time this rule was modified. :vartype last_modified_utc: ~datetime.datetime - :param expiration_date_utc: Expiration date of the rule, if value is not provided or provided - as null this field will default to the maximum allowed expiration date. - :type expiration_date_utc: ~datetime.datetime - :param reason: The reason for dismissing the alert. - :type reason: str - :param state: Possible states of the rule. Possible values include: "Enabled", "Disabled", + :ivar expiration_date_utc: Expiration date of the rule, if value is not provided or provided as + null this field will default to the maximum allowed expiration date. + :vartype expiration_date_utc: ~datetime.datetime + :ivar reason: The reason for dismissing the alert. + :vartype reason: str + :ivar state: Possible states of the rule. Possible values include: "Enabled", "Disabled", "Expired". - :type state: str or ~azure.mgmt.security.models.RuleState - :param comment: Any comment regarding the rule. - :type comment: str - :param suppression_alerts_scope: The suppression conditions. - :type suppression_alerts_scope: ~azure.mgmt.security.models.SuppressionAlertsScope + :vartype state: str or ~azure.mgmt.security.models.RuleState + :ivar comment: Any comment regarding the rule. + :vartype comment: str + :ivar suppression_alerts_scope: The suppression conditions. + :vartype suppression_alerts_scope: ~azure.mgmt.security.models.SuppressionAlertsScope """ _validation = { @@ -1268,6 +1439,22 @@ def __init__( suppression_alerts_scope: Optional["SuppressionAlertsScope"] = None, **kwargs ): + """ + :keyword alert_type: Type of the alert to automatically suppress. For all alert types, use '*'. + :paramtype alert_type: str + :keyword expiration_date_utc: Expiration date of the rule, if value is not provided or provided + as null this field will default to the maximum allowed expiration date. + :paramtype expiration_date_utc: ~datetime.datetime + :keyword reason: The reason for dismissing the alert. + :paramtype reason: str + :keyword state: Possible states of the rule. Possible values include: "Enabled", "Disabled", + "Expired". + :paramtype state: str or ~azure.mgmt.security.models.RuleState + :keyword comment: Any comment regarding the rule. + :paramtype comment: str + :keyword suppression_alerts_scope: The suppression conditions. + :paramtype suppression_alerts_scope: ~azure.mgmt.security.models.SuppressionAlertsScope + """ super(AlertsSuppressionRule, self).__init__(**kwargs) self.alert_type = alert_type self.last_modified_utc = None @@ -1285,8 +1472,8 @@ class AlertsSuppressionRulesList(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param value: Required. - :type value: list[~azure.mgmt.security.models.AlertsSuppressionRule] + :ivar value: Required. + :vartype value: list[~azure.mgmt.security.models.AlertsSuppressionRule] :ivar next_link: URI to fetch the next page. :vartype next_link: str """ @@ -1307,6 +1494,10 @@ def __init__( value: List["AlertsSuppressionRule"], **kwargs ): + """ + :keyword value: Required. + :paramtype value: list[~azure.mgmt.security.models.AlertsSuppressionRule] + """ super(AlertsSuppressionRulesList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -1328,9 +1519,9 @@ class Setting(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param kind: Required. the kind of the settings string.Constant filled by server. Possible + :ivar kind: Required. the kind of the settings string.Constant filled by server. Possible values include: "DataExportSettings", "AlertSuppressionSetting", "AlertSyncSettings". - :type kind: str or ~azure.mgmt.security.models.SettingKind + :vartype kind: str or ~azure.mgmt.security.models.SettingKind """ _validation = { @@ -1355,6 +1546,8 @@ def __init__( self, **kwargs ): + """ + """ super(Setting, self).__init__(**kwargs) self.kind = 'Setting' # type: str @@ -1372,11 +1565,11 @@ class AlertSyncSettings(Setting): :vartype name: str :ivar type: Resource type. :vartype type: str - :param kind: Required. the kind of the settings string.Constant filled by server. Possible + :ivar kind: Required. the kind of the settings string.Constant filled by server. Possible values include: "DataExportSettings", "AlertSuppressionSetting", "AlertSyncSettings". - :type kind: str or ~azure.mgmt.security.models.SettingKind - :param enabled: Is the alert sync setting enabled. - :type enabled: bool + :vartype kind: str or ~azure.mgmt.security.models.SettingKind + :ivar enabled: Is the alert sync setting enabled. + :vartype enabled: bool """ _validation = { @@ -1400,6 +1593,10 @@ def __init__( enabled: Optional[bool] = None, **kwargs ): + """ + :keyword enabled: Is the alert sync setting enabled. + :paramtype enabled: bool + """ super(AlertSyncSettings, self).__init__(**kwargs) self.kind = 'AlertSyncSettings' # type: str self.enabled = enabled @@ -1430,6 +1627,8 @@ def __init__( self, **kwargs ): + """ + """ super(AllowedConnectionsList, self).__init__(**kwargs) self.value = None self.next_link = None @@ -1477,6 +1676,8 @@ def __init__( self, **kwargs ): + """ + """ super(AllowedConnectionsResource, self).__init__(**kwargs) self.location = None self.calculated_date_time = None @@ -1502,10 +1703,10 @@ class ListCustomAlertRule(CustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", "String". :vartype value_type: str or ~azure.mgmt.security.models.ValueType @@ -1537,6 +1738,10 @@ def __init__( is_enabled: bool, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + """ super(ListCustomAlertRule, self).__init__(is_enabled=is_enabled, **kwargs) self.rule_type = 'ListCustomAlertRule' # type: str self.value_type = None @@ -1556,16 +1761,16 @@ class AllowlistCustomAlertRule(ListCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", "String". :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the + :ivar allowlist_values: Required. The values to allow. The format of the values depends on the rule type. - :type allowlist_values: list[str] + :vartype allowlist_values: list[str] """ _validation = { @@ -1597,6 +1802,13 @@ def __init__( allowlist_values: List[str], **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword allowlist_values: Required. The values to allow. The format of the values depends on + the rule type. + :paramtype allowlist_values: list[str] + """ super(AllowlistCustomAlertRule, self).__init__(is_enabled=is_enabled, **kwargs) self.rule_type = 'AllowlistCustomAlertRule' # type: str self.allowlist_values = allowlist_values @@ -1613,16 +1825,16 @@ class AmqpC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -1654,6 +1866,16 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ super(AmqpC2DMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) self.rule_type = 'AmqpC2DMessagesNotInAllowedRange' # type: str @@ -1669,16 +1891,16 @@ class AmqpC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -1710,6 +1932,16 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ super(AmqpC2DRejectedMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) self.rule_type = 'AmqpC2DRejectedMessagesNotInAllowedRange' # type: str @@ -1725,16 +1957,16 @@ class AmqpD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -1766,6 +1998,16 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ super(AmqpD2CMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) self.rule_type = 'AmqpD2CMessagesNotInAllowedRange' # type: str @@ -1781,8 +2023,8 @@ class AscLocation(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param properties: Any object. - :type properties: any + :ivar properties: Any object. + :vartype properties: any """ _validation = { @@ -1804,6 +2046,10 @@ def __init__( properties: Optional[Any] = None, **kwargs ): + """ + :keyword properties: Any object. + :paramtype properties: any + """ super(AscLocation, self).__init__(**kwargs) self.properties = properties @@ -1833,6 +2079,8 @@ def __init__( self, **kwargs ): + """ + """ super(AscLocationList, self).__init__(**kwargs) self.value = None self.next_link = None @@ -1859,6 +2107,8 @@ def __init__( self, **kwargs ): + """ + """ super(AssessmentLinks, self).__init__(**kwargs) self.azure_portal_uri = None @@ -1868,13 +2118,13 @@ class AssessmentStatus(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param code: Required. Programmatic code for the status of the assessment. Possible values + :ivar code: Required. Programmatic code for the status of the assessment. Possible values include: "Healthy", "Unhealthy", "NotApplicable". - :type code: str or ~azure.mgmt.security.models.AssessmentStatusCode - :param cause: Programmatic code for the cause of the assessment status. - :type cause: str - :param description: Human readable description of the assessment status. - :type description: str + :vartype code: str or ~azure.mgmt.security.models.AssessmentStatusCode + :ivar cause: Programmatic code for the cause of the assessment status. + :vartype cause: str + :ivar description: Human readable description of the assessment status. + :vartype description: str """ _validation = { @@ -1895,19 +2145,86 @@ def __init__( description: Optional[str] = None, **kwargs ): + """ + :keyword code: Required. Programmatic code for the status of the assessment. Possible values + include: "Healthy", "Unhealthy", "NotApplicable". + :paramtype code: str or ~azure.mgmt.security.models.AssessmentStatusCode + :keyword cause: Programmatic code for the cause of the assessment status. + :paramtype cause: str + :keyword description: Human readable description of the assessment status. + :paramtype description: str + """ super(AssessmentStatus, self).__init__(**kwargs) self.code = code self.cause = cause self.description = description +class AssessmentStatusResponse(AssessmentStatus): + """The result of the assessment. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar code: Required. Programmatic code for the status of the assessment. Possible values + include: "Healthy", "Unhealthy", "NotApplicable". + :vartype code: str or ~azure.mgmt.security.models.AssessmentStatusCode + :ivar cause: Programmatic code for the cause of the assessment status. + :vartype cause: str + :ivar description: Human readable description of the assessment status. + :vartype description: str + :ivar first_evaluation_date: The time that the assessment was created and first evaluated. + Returned as UTC time in ISO 8601 format. + :vartype first_evaluation_date: ~datetime.datetime + :ivar status_change_date: The time that the status of the assessment last changed. Returned as + UTC time in ISO 8601 format. + :vartype status_change_date: ~datetime.datetime + """ + + _validation = { + 'code': {'required': True}, + 'first_evaluation_date': {'readonly': True}, + 'status_change_date': {'readonly': True}, + } + + _attribute_map = { + 'code': {'key': 'code', 'type': 'str'}, + 'cause': {'key': 'cause', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'first_evaluation_date': {'key': 'firstEvaluationDate', 'type': 'iso-8601'}, + 'status_change_date': {'key': 'statusChangeDate', 'type': 'iso-8601'}, + } + + def __init__( + self, + *, + code: Union[str, "AssessmentStatusCode"], + cause: Optional[str] = None, + description: Optional[str] = None, + **kwargs + ): + """ + :keyword code: Required. Programmatic code for the status of the assessment. Possible values + include: "Healthy", "Unhealthy", "NotApplicable". + :paramtype code: str or ~azure.mgmt.security.models.AssessmentStatusCode + :keyword cause: Programmatic code for the cause of the assessment status. + :paramtype cause: str + :keyword description: Human readable description of the assessment status. + :paramtype description: str + """ + super(AssessmentStatusResponse, self).__init__(code=code, cause=cause, description=description, **kwargs) + self.first_evaluation_date = None + self.status_change_date = None + + class AtaExternalSecuritySolution(ExternalSecuritySolution): """Represents an ATA security solution which sends logs to an OMS workspace. Variables are only populated by the server, and will be ignored when sending a request. - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :ivar kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :vartype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum :ivar location: Location where the resource is stored. :vartype location: str :ivar id: Resource Id. @@ -1916,8 +2233,8 @@ class AtaExternalSecuritySolution(ExternalSecuritySolution): :vartype name: str :ivar type: Resource type. :vartype type: str - :param properties: The external security solution properties for ATA solutions. - :type properties: ~azure.mgmt.security.models.AtaSolutionProperties + :ivar properties: The external security solution properties for ATA solutions. + :vartype properties: ~azure.mgmt.security.models.AtaSolutionProperties """ _validation = { @@ -1943,6 +2260,12 @@ def __init__( properties: Optional["AtaSolutionProperties"] = None, **kwargs ): + """ + :keyword kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :paramtype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :keyword properties: The external security solution properties for ATA solutions. + :paramtype properties: ~azure.mgmt.security.models.AtaSolutionProperties + """ super(AtaExternalSecuritySolution, self).__init__(kind=kind, **kwargs) self.properties = properties @@ -1950,17 +2273,17 @@ def __init__( class AtaSolutionProperties(ExternalSecuritySolutionProperties): """The external security solution properties for ATA solutions. - :param additional_properties: Unmatched properties from the message are deserialized to this + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace - :param last_event_received: - :type last_event_received: str + :vartype additional_properties: dict[str, any] + :ivar device_vendor: + :vartype device_vendor: str + :ivar device_type: + :vartype device_type: str + :ivar workspace: Represents an OMS workspace to which the solution is connected. + :vartype workspace: ~azure.mgmt.security.models.ConnectedWorkspace + :ivar last_event_received: + :vartype last_event_received: str """ _attribute_map = { @@ -1981,6 +2304,19 @@ def __init__( last_event_received: Optional[str] = None, **kwargs ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + :keyword device_vendor: + :paramtype device_vendor: str + :keyword device_type: + :paramtype device_type: str + :keyword workspace: Represents an OMS workspace to which the solution is connected. + :paramtype workspace: ~azure.mgmt.security.models.ConnectedWorkspace + :keyword last_event_received: + :paramtype last_event_received: str + """ super(AtaSolutionProperties, self).__init__(additional_properties=additional_properties, device_vendor=device_vendor, device_type=device_type, workspace=workspace, **kwargs) self.last_event_received = last_event_received @@ -2001,10 +2337,10 @@ class AuthenticationDetailsProperties(msrest.serialization.Model): ~azure.mgmt.security.models.AuthenticationProvisioningState :ivar granted_permissions: The permissions detected in the cloud account. :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account + :ivar authentication_type: Required. Connect to your cloud account, for AWS use either account credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType + filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". + :vartype authentication_type: str or ~azure.mgmt.security.models.AuthenticationType """ _validation = { @@ -2027,6 +2363,8 @@ def __init__( self, **kwargs ): + """ + """ super(AuthenticationDetailsProperties, self).__init__(**kwargs) self.authentication_provisioning_state = None self.granted_permissions = None @@ -2036,8 +2374,8 @@ def __init__( class Tags(msrest.serialization.Model): """A list of key value pairs that describe the resource. - :param tags: A set of tags. A list of key value pairs that describe the resource. - :type tags: dict[str, str] + :ivar tags: A set of tags. A list of key value pairs that describe the resource. + :vartype tags: dict[str, str] """ _attribute_map = { @@ -2050,6 +2388,10 @@ def __init__( tags: Optional[Dict[str, str]] = None, **kwargs ): + """ + :keyword tags: A set of tags. A list of key value pairs that describe the resource. + :paramtype tags: dict[str, str] + """ super(Tags, self).__init__(**kwargs) self.tags = tags @@ -2057,9 +2399,9 @@ def __init__( class ETag(msrest.serialization.Model): """Entity tag is used for comparing two or more entities from the same requested resource. - :param etag: Entity tag is used for comparing two or more entities from the same requested + :ivar etag: Entity tag is used for comparing two or more entities from the same requested resource. - :type etag: str + :vartype etag: str """ _attribute_map = { @@ -2072,6 +2414,11 @@ def __init__( etag: Optional[str] = None, **kwargs ): + """ + :keyword etag: Entity tag is used for comparing two or more entities from the same requested + resource. + :paramtype etag: str + """ super(ETag, self).__init__(**kwargs) self.etag = etag @@ -2079,8 +2426,8 @@ def __init__( class Kind(msrest.serialization.Model): """Describes an Azure resource with kind. - :param kind: Kind of the resource. - :type kind: str + :ivar kind: Kind of the resource. + :vartype kind: str """ _attribute_map = { @@ -2093,6 +2440,10 @@ def __init__( kind: Optional[str] = None, **kwargs ): + """ + :keyword kind: Kind of the resource. + :paramtype kind: str + """ super(Kind, self).__init__(**kwargs) self.kind = kind @@ -2100,8 +2451,8 @@ def __init__( class AzureTrackedResourceLocation(msrest.serialization.Model): """Describes an Azure resource with location. - :param location: Location where the resource is stored. - :type location: str + :ivar location: Location where the resource is stored. + :vartype location: str """ _attribute_map = { @@ -2114,6 +2465,10 @@ def __init__( location: Optional[str] = None, **kwargs ): + """ + :keyword location: Location where the resource is stored. + :paramtype location: str + """ super(AzureTrackedResourceLocation, self).__init__(**kwargs) self.location = location @@ -2123,15 +2478,15 @@ class TrackedResource(Resource, AzureTrackedResourceLocation, Kind, ETag, Tags): Variables are only populated by the server, and will be ignored when sending a request. - :param tags: A set of tags. A list of key value pairs that describe the resource. - :type tags: dict[str, str] - :param etag: Entity tag is used for comparing two or more entities from the same requested + :ivar tags: A set of tags. A list of key value pairs that describe the resource. + :vartype tags: dict[str, str] + :ivar etag: Entity tag is used for comparing two or more entities from the same requested resource. - :type etag: str - :param kind: Kind of the resource. - :type kind: str - :param location: Location where the resource is stored. - :type location: str + :vartype etag: str + :ivar kind: Kind of the resource. + :vartype kind: str + :ivar location: Location where the resource is stored. + :vartype location: str :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. @@ -2165,6 +2520,17 @@ def __init__( location: Optional[str] = None, **kwargs ): + """ + :keyword tags: A set of tags. A list of key value pairs that describe the resource. + :paramtype tags: dict[str, str] + :keyword etag: Entity tag is used for comparing two or more entities from the same requested + resource. + :paramtype etag: str + :keyword kind: Kind of the resource. + :paramtype kind: str + :keyword location: Location where the resource is stored. + :paramtype location: str + """ super(TrackedResource, self).__init__(location=location, kind=kind, etag=etag, tags=tags, **kwargs) self.tags = tags self.etag = etag @@ -2201,35 +2567,35 @@ class Automation(TrackedResource): Variables are only populated by the server, and will be ignored when sending a request. - :param tags: A set of tags. A list of key value pairs that describe the resource. - :type tags: dict[str, str] - :param etag: Entity tag is used for comparing two or more entities from the same requested + :ivar tags: A set of tags. A list of key value pairs that describe the resource. + :vartype tags: dict[str, str] + :ivar etag: Entity tag is used for comparing two or more entities from the same requested resource. - :type etag: str - :param kind: Kind of the resource. - :type kind: str - :param location: Location where the resource is stored. - :type location: str + :vartype etag: str + :ivar kind: Kind of the resource. + :vartype kind: str + :ivar location: Location where the resource is stored. + :vartype location: str :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. :vartype name: str :ivar type: Resource type. :vartype type: str - :param description: The security automation description. - :type description: str - :param is_enabled: Indicates whether the security automation is enabled. - :type is_enabled: bool - :param scopes: A collection of scopes on which the security automations logic is applied. + :ivar description: The security automation description. + :vartype description: str + :ivar is_enabled: Indicates whether the security automation is enabled. + :vartype is_enabled: bool + :ivar scopes: A collection of scopes on which the security automations logic is applied. Supported scopes are the subscription itself or a resource group under that subscription. The automation will only apply on defined scopes. - :type scopes: list[~azure.mgmt.security.models.AutomationScope] - :param sources: A collection of the source event types which evaluate the security automation + :vartype scopes: list[~azure.mgmt.security.models.AutomationScope] + :ivar sources: A collection of the source event types which evaluate the security automation set of rules. - :type sources: list[~azure.mgmt.security.models.AutomationSource] - :param actions: A collection of the actions which are triggered if all the configured rules + :vartype sources: list[~azure.mgmt.security.models.AutomationSource] + :ivar actions: A collection of the actions which are triggered if all the configured rules evaluations, within at least one rule set, are true. - :type actions: list[~azure.mgmt.security.models.AutomationAction] + :vartype actions: list[~azure.mgmt.security.models.AutomationAction] """ _validation = { @@ -2267,6 +2633,31 @@ def __init__( actions: Optional[List["AutomationAction"]] = None, **kwargs ): + """ + :keyword tags: A set of tags. A list of key value pairs that describe the resource. + :paramtype tags: dict[str, str] + :keyword etag: Entity tag is used for comparing two or more entities from the same requested + resource. + :paramtype etag: str + :keyword kind: Kind of the resource. + :paramtype kind: str + :keyword location: Location where the resource is stored. + :paramtype location: str + :keyword description: The security automation description. + :paramtype description: str + :keyword is_enabled: Indicates whether the security automation is enabled. + :paramtype is_enabled: bool + :keyword scopes: A collection of scopes on which the security automations logic is applied. + Supported scopes are the subscription itself or a resource group under that subscription. The + automation will only apply on defined scopes. + :paramtype scopes: list[~azure.mgmt.security.models.AutomationScope] + :keyword sources: A collection of the source event types which evaluate the security automation + set of rules. + :paramtype sources: list[~azure.mgmt.security.models.AutomationSource] + :keyword actions: A collection of the actions which are triggered if all the configured rules + evaluations, within at least one rule set, are true. + :paramtype actions: list[~azure.mgmt.security.models.AutomationAction] + """ super(Automation, self).__init__(tags=tags, etag=etag, kind=kind, location=location, **kwargs) self.description = description self.is_enabled = is_enabled @@ -2283,10 +2674,10 @@ class AutomationAction(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", + :ivar action_type: Required. The type of the action that will be triggered by the + Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType + :vartype action_type: str or ~azure.mgmt.security.models.ActionType """ _validation = { @@ -2305,6 +2696,8 @@ def __init__( self, **kwargs ): + """ + """ super(AutomationAction, self).__init__(**kwargs) self.action_type = None # type: Optional[str] @@ -2316,17 +2709,17 @@ class AutomationActionEventHub(AutomationAction): All required parameters must be populated in order to send to Azure. - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", + :ivar action_type: Required. The type of the action that will be triggered by the + Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType - :param event_hub_resource_id: The target Event Hub Azure Resource ID. - :type event_hub_resource_id: str + :vartype action_type: str or ~azure.mgmt.security.models.ActionType + :ivar event_hub_resource_id: The target Event Hub Azure Resource ID. + :vartype event_hub_resource_id: str :ivar sas_policy_name: The target Event Hub SAS policy name. :vartype sas_policy_name: str - :param connection_string: The target Event Hub connection string (it will not be included in - any response). - :type connection_string: str + :ivar connection_string: The target Event Hub connection string (it will not be included in any + response). + :vartype connection_string: str """ _validation = { @@ -2348,6 +2741,13 @@ def __init__( connection_string: Optional[str] = None, **kwargs ): + """ + :keyword event_hub_resource_id: The target Event Hub Azure Resource ID. + :paramtype event_hub_resource_id: str + :keyword connection_string: The target Event Hub connection string (it will not be included in + any response). + :paramtype connection_string: str + """ super(AutomationActionEventHub, self).__init__(**kwargs) self.action_type = 'EventHub' # type: str self.event_hub_resource_id = event_hub_resource_id @@ -2360,15 +2760,15 @@ class AutomationActionLogicApp(AutomationAction): All required parameters must be populated in order to send to Azure. - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", + :ivar action_type: Required. The type of the action that will be triggered by the + Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType - :param logic_app_resource_id: The triggered Logic App Azure Resource ID. This can also reside - on other subscriptions, given that you have permissions to trigger the Logic App. - :type logic_app_resource_id: str - :param uri: The Logic App trigger URI endpoint (it will not be included in any response). - :type uri: str + :vartype action_type: str or ~azure.mgmt.security.models.ActionType + :ivar logic_app_resource_id: The triggered Logic App Azure Resource ID. This can also reside on + other subscriptions, given that you have permissions to trigger the Logic App. + :vartype logic_app_resource_id: str + :ivar uri: The Logic App trigger URI endpoint (it will not be included in any response). + :vartype uri: str """ _validation = { @@ -2388,6 +2788,13 @@ def __init__( uri: Optional[str] = None, **kwargs ): + """ + :keyword logic_app_resource_id: The triggered Logic App Azure Resource ID. This can also reside + on other subscriptions, given that you have permissions to trigger the Logic App. + :paramtype logic_app_resource_id: str + :keyword uri: The Logic App trigger URI endpoint (it will not be included in any response). + :paramtype uri: str + """ super(AutomationActionLogicApp, self).__init__(**kwargs) self.action_type = 'LogicApp' # type: str self.logic_app_resource_id = logic_app_resource_id @@ -2399,12 +2806,12 @@ class AutomationActionWorkspace(AutomationAction): All required parameters must be populated in order to send to Azure. - :param action_type: Required. The type of the action that will be triggered by the - Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", + :ivar action_type: Required. The type of the action that will be triggered by the + Automation.Constant filled by server. Possible values include: "LogicApp", "EventHub", "Workspace". - :type action_type: str or ~azure.mgmt.security.models.ActionType - :param workspace_resource_id: The fully qualified Log Analytics Workspace Azure Resource ID. - :type workspace_resource_id: str + :vartype action_type: str or ~azure.mgmt.security.models.ActionType + :ivar workspace_resource_id: The fully qualified Log Analytics Workspace Azure Resource ID. + :vartype workspace_resource_id: str """ _validation = { @@ -2422,6 +2829,10 @@ def __init__( workspace_resource_id: Optional[str] = None, **kwargs ): + """ + :keyword workspace_resource_id: The fully qualified Log Analytics Workspace Azure Resource ID. + :paramtype workspace_resource_id: str + """ super(AutomationActionWorkspace, self).__init__(**kwargs) self.action_type = 'Workspace' # type: str self.workspace_resource_id = workspace_resource_id @@ -2434,8 +2845,8 @@ class AutomationList(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param value: Required. The list of security automations under the given scope. - :type value: list[~azure.mgmt.security.models.Automation] + :ivar value: Required. The list of security automations under the given scope. + :vartype value: list[~azure.mgmt.security.models.Automation] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -2456,6 +2867,10 @@ def __init__( value: List["Automation"], **kwargs ): + """ + :keyword value: Required. The list of security automations under the given scope. + :paramtype value: list[~azure.mgmt.security.models.Automation] + """ super(AutomationList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -2464,8 +2879,8 @@ def __init__( class AutomationRuleSet(msrest.serialization.Model): """A rule set which evaluates all its rules upon an event interception. Only when all the included rules in the rule set will be evaluated as 'true', will the event trigger the defined actions. - :param rules: - :type rules: list[~azure.mgmt.security.models.AutomationTriggeringRule] + :ivar rules: + :vartype rules: list[~azure.mgmt.security.models.AutomationTriggeringRule] """ _attribute_map = { @@ -2478,6 +2893,10 @@ def __init__( rules: Optional[List["AutomationTriggeringRule"]] = None, **kwargs ): + """ + :keyword rules: + :paramtype rules: list[~azure.mgmt.security.models.AutomationTriggeringRule] + """ super(AutomationRuleSet, self).__init__(**kwargs) self.rules = rules @@ -2485,11 +2904,11 @@ def __init__( class AutomationScope(msrest.serialization.Model): """A single automation scope. - :param description: The resources scope description. - :type description: str - :param scope_path: The resources scope path. Can be the subscription on which the automation is + :ivar description: The resources scope description. + :vartype description: str + :ivar scope_path: The resources scope path. Can be the subscription on which the automation is defined on or a resource group under that subscription (fully qualified Azure resource IDs). - :type scope_path: str + :vartype scope_path: str """ _attribute_map = { @@ -2504,6 +2923,13 @@ def __init__( scope_path: Optional[str] = None, **kwargs ): + """ + :keyword description: The resources scope description. + :paramtype description: str + :keyword scope_path: The resources scope path. Can be the subscription on which the automation + is defined on or a resource group under that subscription (fully qualified Azure resource IDs). + :paramtype scope_path: str + """ super(AutomationScope, self).__init__(**kwargs) self.description = description self.scope_path = scope_path @@ -2512,14 +2938,14 @@ def __init__( class AutomationSource(msrest.serialization.Model): """The source event types which evaluate the security automation set of rules. For example - security alerts and security assessments. To learn more about the supported security events data models schemas - please visit https://aka.ms/ASCAutomationSchemas. - :param event_source: A valid event source type. Possible values include: "Assessments", - "SubAssessments", "Alerts", "SecureScores", "SecureScoresSnapshot", "SecureScoreControls", - "SecureScoreControlsSnapshot", "RegulatoryComplianceAssessment", - "RegulatoryComplianceAssessmentSnapshot". - :type event_source: str or ~azure.mgmt.security.models.EventSource - :param rule_sets: A set of rules which evaluate upon event interception. A logical disjunction + :ivar event_source: A valid event source type. Possible values include: "Assessments", + "AssessmentsSnapshot", "SubAssessments", "SubAssessmentsSnapshot", "Alerts", "SecureScores", + "SecureScoresSnapshot", "SecureScoreControls", "SecureScoreControlsSnapshot", + "RegulatoryComplianceAssessment", "RegulatoryComplianceAssessmentSnapshot". + :vartype event_source: str or ~azure.mgmt.security.models.EventSource + :ivar rule_sets: A set of rules which evaluate upon event interception. A logical disjunction is applied between defined rule sets (logical 'or'). - :type rule_sets: list[~azure.mgmt.security.models.AutomationRuleSet] + :vartype rule_sets: list[~azure.mgmt.security.models.AutomationRuleSet] """ _attribute_map = { @@ -2534,6 +2960,16 @@ def __init__( rule_sets: Optional[List["AutomationRuleSet"]] = None, **kwargs ): + """ + :keyword event_source: A valid event source type. Possible values include: "Assessments", + "AssessmentsSnapshot", "SubAssessments", "SubAssessmentsSnapshot", "Alerts", "SecureScores", + "SecureScoresSnapshot", "SecureScoreControls", "SecureScoreControlsSnapshot", + "RegulatoryComplianceAssessment", "RegulatoryComplianceAssessmentSnapshot". + :paramtype event_source: str or ~azure.mgmt.security.models.EventSource + :keyword rule_sets: A set of rules which evaluate upon event interception. A logical + disjunction is applied between defined rule sets (logical 'or'). + :paramtype rule_sets: list[~azure.mgmt.security.models.AutomationRuleSet] + """ super(AutomationSource, self).__init__(**kwargs) self.event_source = event_source self.rule_sets = rule_sets @@ -2542,19 +2978,19 @@ def __init__( class AutomationTriggeringRule(msrest.serialization.Model): """A rule which is evaluated upon event interception. The rule is configured by comparing a specific value from the event model to an expected value. This comparison is done by using one of the supported operators set. - :param property_j_path: The JPath of the entity model property that should be checked. - :type property_j_path: str - :param property_type: The data type of the compared operands (string, integer, floating point + :ivar property_j_path: The JPath of the entity model property that should be checked. + :vartype property_j_path: str + :ivar property_type: The data type of the compared operands (string, integer, floating point number or a boolean [true/false]]. Possible values include: "String", "Integer", "Number", "Boolean". - :type property_type: str or ~azure.mgmt.security.models.PropertyType - :param expected_value: The expected value. - :type expected_value: str - :param operator: A valid comparer operator to use. A case-insensitive comparison will be - applied for String PropertyType. Possible values include: "Equals", "GreaterThan", + :vartype property_type: str or ~azure.mgmt.security.models.PropertyType + :ivar expected_value: The expected value. + :vartype expected_value: str + :ivar operator: A valid comparer operator to use. A case-insensitive comparison will be applied + for String PropertyType. Possible values include: "Equals", "GreaterThan", "GreaterThanOrEqualTo", "LesserThan", "LesserThanOrEqualTo", "NotEquals", "Contains", "StartsWith", "EndsWith". - :type operator: str or ~azure.mgmt.security.models.Operator + :vartype operator: str or ~azure.mgmt.security.models.Operator """ _attribute_map = { @@ -2573,6 +3009,21 @@ def __init__( operator: Optional[Union[str, "Operator"]] = None, **kwargs ): + """ + :keyword property_j_path: The JPath of the entity model property that should be checked. + :paramtype property_j_path: str + :keyword property_type: The data type of the compared operands (string, integer, floating point + number or a boolean [true/false]]. Possible values include: "String", "Integer", "Number", + "Boolean". + :paramtype property_type: str or ~azure.mgmt.security.models.PropertyType + :keyword expected_value: The expected value. + :paramtype expected_value: str + :keyword operator: A valid comparer operator to use. A case-insensitive comparison will be + applied for String PropertyType. Possible values include: "Equals", "GreaterThan", + "GreaterThanOrEqualTo", "LesserThan", "LesserThanOrEqualTo", "NotEquals", "Contains", + "StartsWith", "EndsWith". + :paramtype operator: str or ~azure.mgmt.security.models.Operator + """ super(AutomationTriggeringRule, self).__init__(**kwargs) self.property_j_path = property_j_path self.property_type = property_type @@ -2583,10 +3034,10 @@ def __init__( class AutomationValidationStatus(msrest.serialization.Model): """The security automation model state property bag. - :param is_valid: Indicates whether the model is valid or not. - :type is_valid: bool - :param message: The validation message. - :type message: str + :ivar is_valid: Indicates whether the model is valid or not. + :vartype is_valid: bool + :ivar message: The validation message. + :vartype message: str """ _attribute_map = { @@ -2601,6 +3052,12 @@ def __init__( message: Optional[str] = None, **kwargs ): + """ + :keyword is_valid: Indicates whether the model is valid or not. + :paramtype is_valid: bool + :keyword message: The validation message. + :paramtype message: str + """ super(AutomationValidationStatus, self).__init__(**kwargs) self.is_valid = is_valid self.message = message @@ -2617,9 +3074,9 @@ class AutoProvisioningSetting(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param auto_provision: Describes what kind of security agent provisioning action to take. + :ivar auto_provision: Describes what kind of security agent provisioning action to take. Possible values include: "On", "Off". - :type auto_provision: str or ~azure.mgmt.security.models.AutoProvision + :vartype auto_provision: str or ~azure.mgmt.security.models.AutoProvision """ _validation = { @@ -2641,6 +3098,11 @@ def __init__( auto_provision: Optional[Union[str, "AutoProvision"]] = None, **kwargs ): + """ + :keyword auto_provision: Describes what kind of security agent provisioning action to take. + Possible values include: "On", "Off". + :paramtype auto_provision: str or ~azure.mgmt.security.models.AutoProvision + """ super(AutoProvisioningSetting, self).__init__(**kwargs) self.auto_provision = auto_provision @@ -2650,8 +3112,8 @@ class AutoProvisioningSettingList(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param value: List of all the auto provisioning settings. - :type value: list[~azure.mgmt.security.models.AutoProvisioningSetting] + :ivar value: List of all the auto provisioning settings. + :vartype value: list[~azure.mgmt.security.models.AutoProvisioningSetting] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -2671,6 +3133,10 @@ def __init__( value: Optional[List["AutoProvisioningSetting"]] = None, **kwargs ): + """ + :keyword value: List of all the auto provisioning settings. + :paramtype value: list[~azure.mgmt.security.models.AutoProvisioningSetting] + """ super(AutoProvisioningSettingList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -2689,18 +3155,18 @@ class AwAssumeRoleAuthenticationDetailsProperties(AuthenticationDetailsPropertie ~azure.mgmt.security.models.AuthenticationProvisioningState :ivar granted_permissions: The permissions detected in the cloud account. :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account + :ivar authentication_type: Required. Connect to your cloud account, for AWS use either account credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType + filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". + :vartype authentication_type: str or ~azure.mgmt.security.models.AuthenticationType :ivar account_id: The ID of the cloud account. :vartype account_id: str - :param aws_assume_role_arn: Required. Assumed role ID is an identifier that you can use to + :ivar aws_assume_role_arn: Required. Assumed role ID is an identifier that you can use to create temporary security credentials. - :type aws_assume_role_arn: str - :param aws_external_id: Required. A unique identifier that is required when you assume a role - in another account. - :type aws_external_id: str + :vartype aws_assume_role_arn: str + :ivar aws_external_id: Required. A unique identifier that is required when you assume a role in + another account. + :vartype aws_external_id: str """ _validation = { @@ -2728,6 +3194,14 @@ def __init__( aws_external_id: str, **kwargs ): + """ + :keyword aws_assume_role_arn: Required. Assumed role ID is an identifier that you can use to + create temporary security credentials. + :paramtype aws_assume_role_arn: str + :keyword aws_external_id: Required. A unique identifier that is required when you assume a role + in another account. + :paramtype aws_external_id: str + """ super(AwAssumeRoleAuthenticationDetailsProperties, self).__init__(**kwargs) self.authentication_type = 'awsAssumeRole' # type: str self.account_id = None @@ -2748,18 +3222,18 @@ class AwsCredsAuthenticationDetailsProperties(AuthenticationDetailsProperties): ~azure.mgmt.security.models.AuthenticationProvisioningState :ivar granted_permissions: The permissions detected in the cloud account. :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account + :ivar authentication_type: Required. Connect to your cloud account, for AWS use either account credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType + filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". + :vartype authentication_type: str or ~azure.mgmt.security.models.AuthenticationType :ivar account_id: The ID of the cloud account. :vartype account_id: str - :param aws_access_key_id: Required. Public key element of the AWS credential object (write + :ivar aws_access_key_id: Required. Public key element of the AWS credential object (write only). - :type aws_access_key_id: str - :param aws_secret_access_key: Required. Secret key element of the AWS credential object (write + :vartype aws_access_key_id: str + :ivar aws_secret_access_key: Required. Secret key element of the AWS credential object (write only). - :type aws_secret_access_key: str + :vartype aws_secret_access_key: str """ _validation = { @@ -2787,6 +3261,14 @@ def __init__( aws_secret_access_key: str, **kwargs ): + """ + :keyword aws_access_key_id: Required. Public key element of the AWS credential object (write + only). + :paramtype aws_access_key_id: str + :keyword aws_secret_access_key: Required. Secret key element of the AWS credential object + (write only). + :paramtype aws_secret_access_key: str + """ super(AwsCredsAuthenticationDetailsProperties, self).__init__(**kwargs) self.authentication_type = 'awsCreds' # type: str self.account_id = None @@ -2802,9 +3284,9 @@ class ResourceDetails(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source + :ivar source: Required. The platform where the assessed resource resides.Constant filled by + server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". + :vartype source: str or ~azure.mgmt.security.models.Source """ _validation = { @@ -2823,6 +3305,8 @@ def __init__( self, **kwargs ): + """ + """ super(ResourceDetails, self).__init__(**kwargs) self.source = None # type: Optional[str] @@ -2834,9 +3318,9 @@ class AzureResourceDetails(ResourceDetails): All required parameters must be populated in order to send to Azure. - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source + :ivar source: Required. The platform where the assessed resource resides.Constant filled by + server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". + :vartype source: str or ~azure.mgmt.security.models.Source :ivar id: Azure resource Id of the assessed resource. :vartype id: str """ @@ -2855,6 +3339,8 @@ def __init__( self, **kwargs ): + """ + """ super(AzureResourceDetails, self).__init__(**kwargs) self.source = 'Azure' # type: str self.id = None @@ -2868,10 +3354,10 @@ class ResourceIdentifier(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param type: Required. There can be multiple identifiers of different type per alert, this - field specify the identifier type.Constant filled by server. Possible values include: + :ivar type: Required. There can be multiple identifiers of different type per alert, this field + specify the identifier type.Constant filled by server. Possible values include: "AzureResource", "LogAnalytics". - :type type: str or ~azure.mgmt.security.models.ResourceIdentifierType + :vartype type: str or ~azure.mgmt.security.models.ResourceIdentifierType """ _validation = { @@ -2890,6 +3376,8 @@ def __init__( self, **kwargs ): + """ + """ super(ResourceIdentifier, self).__init__(**kwargs) self.type = None # type: Optional[str] @@ -2901,10 +3389,10 @@ class AzureResourceIdentifier(ResourceIdentifier): All required parameters must be populated in order to send to Azure. - :param type: Required. There can be multiple identifiers of different type per alert, this - field specify the identifier type.Constant filled by server. Possible values include: + :ivar type: Required. There can be multiple identifiers of different type per alert, this field + specify the identifier type.Constant filled by server. Possible values include: "AzureResource", "LogAnalytics". - :type type: str or ~azure.mgmt.security.models.ResourceIdentifierType + :vartype type: str or ~azure.mgmt.security.models.ResourceIdentifierType :ivar azure_resource_id: ARM resource identifier for the cloud resource being alerted on. :vartype azure_resource_id: str """ @@ -2923,6 +3411,8 @@ def __init__( self, **kwargs ): + """ + """ super(AzureResourceIdentifier, self).__init__(**kwargs) self.type = 'AzureResource' # type: str self.azure_resource_id = None @@ -2949,6 +3439,8 @@ def __init__( self, **kwargs ): + """ + """ super(AzureResourceLink, self).__init__(**kwargs) self.id = None @@ -2956,10 +3448,10 @@ def __init__( class Baseline(msrest.serialization.Model): """Baseline details. - :param expected_results: Expected results. - :type expected_results: list[list[str]] - :param updated_time: Baseline update time (UTC). - :type updated_time: ~datetime.datetime + :ivar expected_results: Expected results. + :vartype expected_results: list[list[str]] + :ivar updated_time: Baseline update time (UTC). + :vartype updated_time: ~datetime.datetime """ _attribute_map = { @@ -2974,6 +3466,12 @@ def __init__( updated_time: Optional[datetime.datetime] = None, **kwargs ): + """ + :keyword expected_results: Expected results. + :paramtype expected_results: list[list[str]] + :keyword updated_time: Baseline update time (UTC). + :paramtype updated_time: ~datetime.datetime + """ super(Baseline, self).__init__(**kwargs) self.expected_results = expected_results self.updated_time = updated_time @@ -2982,15 +3480,15 @@ def __init__( class BaselineAdjustedResult(msrest.serialization.Model): """The rule result adjusted with baseline. - :param baseline: Baseline details. - :type baseline: ~azure.mgmt.security.models.Baseline - :param status: The rule result status. Possible values include: "NonFinding", "Finding", + :ivar baseline: Baseline details. + :vartype baseline: ~azure.mgmt.security.models.Baseline + :ivar status: The rule result status. Possible values include: "NonFinding", "Finding", "InternalError". - :type status: str or ~azure.mgmt.security.models.RuleStatus - :param results_not_in_baseline: Results the are not in baseline. - :type results_not_in_baseline: list[list[str]] - :param results_only_in_baseline: Results the are in baseline. - :type results_only_in_baseline: list[list[str]] + :vartype status: str or ~azure.mgmt.security.models.RuleStatus + :ivar results_not_in_baseline: Results the are not in baseline. + :vartype results_not_in_baseline: list[list[str]] + :ivar results_only_in_baseline: Results the are in baseline. + :vartype results_only_in_baseline: list[list[str]] """ _attribute_map = { @@ -3009,6 +3507,17 @@ def __init__( results_only_in_baseline: Optional[List[List[str]]] = None, **kwargs ): + """ + :keyword baseline: Baseline details. + :paramtype baseline: ~azure.mgmt.security.models.Baseline + :keyword status: The rule result status. Possible values include: "NonFinding", "Finding", + "InternalError". + :paramtype status: str or ~azure.mgmt.security.models.RuleStatus + :keyword results_not_in_baseline: Results the are not in baseline. + :paramtype results_not_in_baseline: list[list[str]] + :keyword results_only_in_baseline: Results the are in baseline. + :paramtype results_only_in_baseline: list[list[str]] + """ super(BaselineAdjustedResult, self).__init__(**kwargs) self.baseline = baseline self.status = status @@ -3019,10 +3528,10 @@ def __init__( class BenchmarkReference(msrest.serialization.Model): """The benchmark references. - :param benchmark: The benchmark name. - :type benchmark: str - :param reference: The benchmark reference. - :type reference: str + :ivar benchmark: The benchmark name. + :vartype benchmark: str + :ivar reference: The benchmark reference. + :vartype reference: str """ _attribute_map = { @@ -3037,6 +3546,12 @@ def __init__( reference: Optional[str] = None, **kwargs ): + """ + :keyword benchmark: The benchmark name. + :paramtype benchmark: str + :keyword reference: The benchmark reference. + :paramtype reference: str + """ super(BenchmarkReference, self).__init__(**kwargs) self.benchmark = benchmark self.reference = reference @@ -3047,8 +3562,8 @@ class CefExternalSecuritySolution(ExternalSecuritySolution): Variables are only populated by the server, and will be ignored when sending a request. - :param kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". - :type kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :ivar kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :vartype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum :ivar location: Location where the resource is stored. :vartype location: str :ivar id: Resource Id. @@ -3057,8 +3572,8 @@ class CefExternalSecuritySolution(ExternalSecuritySolution): :vartype name: str :ivar type: Resource type. :vartype type: str - :param properties: The external security solution properties for CEF solutions. - :type properties: ~azure.mgmt.security.models.CefSolutionProperties + :ivar properties: The external security solution properties for CEF solutions. + :vartype properties: ~azure.mgmt.security.models.CefSolutionProperties """ _validation = { @@ -3084,6 +3599,12 @@ def __init__( properties: Optional["CefSolutionProperties"] = None, **kwargs ): + """ + :keyword kind: The kind of the external solution. Possible values include: "CEF", "ATA", "AAD". + :paramtype kind: str or ~azure.mgmt.security.models.ExternalSecuritySolutionKindEnum + :keyword properties: The external security solution properties for CEF solutions. + :paramtype properties: ~azure.mgmt.security.models.CefSolutionProperties + """ super(CefExternalSecuritySolution, self).__init__(kind=kind, **kwargs) self.properties = properties @@ -3091,21 +3612,21 @@ def __init__( class CefSolutionProperties(ExternalSecuritySolutionProperties): """The external security solution properties for CEF solutions. - :param additional_properties: Unmatched properties from the message are deserialized to this + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] - :param device_vendor: - :type device_vendor: str - :param device_type: - :type device_type: str - :param workspace: Represents an OMS workspace to which the solution is connected. - :type workspace: ~azure.mgmt.security.models.ConnectedWorkspace - :param hostname: - :type hostname: str - :param agent: - :type agent: str - :param last_event_received: - :type last_event_received: str + :vartype additional_properties: dict[str, any] + :ivar device_vendor: + :vartype device_vendor: str + :ivar device_type: + :vartype device_type: str + :ivar workspace: Represents an OMS workspace to which the solution is connected. + :vartype workspace: ~azure.mgmt.security.models.ConnectedWorkspace + :ivar hostname: + :vartype hostname: str + :ivar agent: + :vartype agent: str + :ivar last_event_received: + :vartype last_event_received: str """ _attribute_map = { @@ -3130,6 +3651,23 @@ def __init__( last_event_received: Optional[str] = None, **kwargs ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + :keyword device_vendor: + :paramtype device_vendor: str + :keyword device_type: + :paramtype device_type: str + :keyword workspace: Represents an OMS workspace to which the solution is connected. + :paramtype workspace: ~azure.mgmt.security.models.ConnectedWorkspace + :keyword hostname: + :paramtype hostname: str + :keyword agent: + :paramtype agent: str + :keyword last_event_received: + :paramtype last_event_received: str + """ super(CefSolutionProperties, self).__init__(additional_properties=additional_properties, device_vendor=device_vendor, device_type=device_type, workspace=workspace, **kwargs) self.hostname = hostname self.agent = agent @@ -3173,6 +3711,8 @@ def __init__( self, **kwargs ): + """ + """ super(CloudErrorBody, self).__init__(**kwargs) self.code = None self.message = None @@ -3181,6 +3721,48 @@ def __init__( self.additional_info = None +class CloudOffering(msrest.serialization.Model): + """The security offering details. + + You probably want to use the sub-classes and not this class directly. Known + sub-classes are: CspmMonitorAwsOffering, DefenderForContainersAwsOffering, DefenderForServersAwsOffering. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar offering_type: Required. The type of the security offering.Constant filled by server. + Possible values include: "CspmMonitorAws", "DefenderForContainersAws", "DefenderForServersAws". + :vartype offering_type: str or ~azure.mgmt.security.models.OfferingType + :ivar description: The offering description. + :vartype description: str + """ + + _validation = { + 'offering_type': {'required': True}, + 'description': {'readonly': True}, + } + + _attribute_map = { + 'offering_type': {'key': 'offeringType', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + } + + _subtype_map = { + 'offering_type': {'CspmMonitorAws': 'CspmMonitorAwsOffering', 'DefenderForContainersAws': 'DefenderForContainersAwsOffering', 'DefenderForServersAWS': 'DefenderForServersAwsOffering'} + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(CloudOffering, self).__init__(**kwargs) + self.offering_type = None # type: Optional[str] + self.description = None + + class Compliance(Resource): """Compliance of a scope. @@ -3224,6 +3806,8 @@ def __init__( self, **kwargs ): + """ + """ super(Compliance, self).__init__(**kwargs) self.assessment_timestamp_utc_date = None self.resource_count = None @@ -3235,8 +3819,8 @@ class ComplianceList(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param value: List of Compliance objects. - :type value: list[~azure.mgmt.security.models.Compliance] + :ivar value: List of Compliance objects. + :vartype value: list[~azure.mgmt.security.models.Compliance] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -3256,6 +3840,10 @@ def __init__( value: Optional[List["Compliance"]] = None, **kwargs ): + """ + :keyword value: List of Compliance objects. + :paramtype value: list[~azure.mgmt.security.models.Compliance] + """ super(ComplianceList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -3295,6 +3883,8 @@ def __init__( self, **kwargs ): + """ + """ super(ComplianceResult, self).__init__(**kwargs) self.resource_status = None @@ -3306,8 +3896,8 @@ class ComplianceResultList(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param value: Required. List of compliance results. - :type value: list[~azure.mgmt.security.models.ComplianceResult] + :ivar value: Required. List of compliance results. + :vartype value: list[~azure.mgmt.security.models.ComplianceResult] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -3328,6 +3918,10 @@ def __init__( value: List["ComplianceResult"], **kwargs ): + """ + :keyword value: Required. List of compliance results. + :paramtype value: list[~azure.mgmt.security.models.ComplianceResult] + """ super(ComplianceResultList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -3359,6 +3953,8 @@ def __init__( self, **kwargs ): + """ + """ super(ComplianceSegment, self).__init__(**kwargs) self.segment_type = None self.percentage = None @@ -3395,6 +3991,8 @@ def __init__( self, **kwargs ): + """ + """ super(ConnectableResource, self).__init__(**kwargs) self.id = None self.inbound_connected_resources = None @@ -3430,6 +4028,8 @@ def __init__( self, **kwargs ): + """ + """ super(ConnectedResource, self).__init__(**kwargs) self.connected_resource_id = None self.tcp_ports = None @@ -3439,8 +4039,8 @@ def __init__( class ConnectedWorkspace(msrest.serialization.Model): """Represents an OMS workspace to which the solution is connected. - :param id: Azure resource ID of the connected OMS workspace. - :type id: str + :ivar id: Azure resource ID of the connected OMS workspace. + :vartype id: str """ _attribute_map = { @@ -3453,6 +4053,10 @@ def __init__( id: Optional[str] = None, **kwargs ): + """ + :keyword id: Azure resource ID of the connected OMS workspace. + :paramtype id: str + """ super(ConnectedWorkspace, self).__init__(**kwargs) self.id = id @@ -3468,16 +4072,16 @@ class ConnectionFromIpNotAllowed(AllowlistCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", "String". :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the + :ivar allowlist_values: Required. The values to allow. The format of the values depends on the rule type. - :type allowlist_values: list[str] + :vartype allowlist_values: list[str] """ _validation = { @@ -3505,6 +4109,13 @@ def __init__( allowlist_values: List[str], **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword allowlist_values: Required. The values to allow. The format of the values depends on + the rule type. + :paramtype allowlist_values: list[str] + """ super(ConnectionFromIpNotAllowed, self).__init__(is_enabled=is_enabled, allowlist_values=allowlist_values, **kwargs) self.rule_type = 'ConnectionFromIpNotAllowed' # type: str @@ -3514,8 +4125,8 @@ class ConnectionStrings(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param value: Required. Connection strings. - :type value: list[~azure.mgmt.security.models.IngestionConnectionString] + :ivar value: Required. Connection strings. + :vartype value: list[~azure.mgmt.security.models.IngestionConnectionString] """ _validation = { @@ -3532,6 +4143,10 @@ def __init__( value: List["IngestionConnectionString"], **kwargs ): + """ + :keyword value: Required. Connection strings. + :paramtype value: list[~azure.mgmt.security.models.IngestionConnectionString] + """ super(ConnectionStrings, self).__init__(**kwargs) self.value = value @@ -3547,16 +4162,16 @@ class ConnectionToIpNotAllowed(AllowlistCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", "String". :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the + :ivar allowlist_values: Required. The values to allow. The format of the values depends on the rule type. - :type allowlist_values: list[str] + :vartype allowlist_values: list[str] """ _validation = { @@ -3584,6 +4199,13 @@ def __init__( allowlist_values: List[str], **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword allowlist_values: Required. The values to allow. The format of the values depends on + the rule type. + :paramtype allowlist_values: list[str] + """ super(ConnectionToIpNotAllowed, self).__init__(is_enabled=is_enabled, allowlist_values=allowlist_values, **kwargs) self.rule_type = 'ConnectionToIpNotAllowed' # type: str @@ -3599,12 +4221,12 @@ class ConnectorSetting(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param hybrid_compute_settings: Settings for hybrid compute management. These settings are + :ivar hybrid_compute_settings: Settings for hybrid compute management. These settings are relevant only for Arc autoProvision (Hybrid Compute). - :type hybrid_compute_settings: ~azure.mgmt.security.models.HybridComputeSettingsProperties - :param authentication_details: Settings for authentication management, these settings are + :vartype hybrid_compute_settings: ~azure.mgmt.security.models.HybridComputeSettingsProperties + :ivar authentication_details: Settings for authentication management, these settings are relevant only for the cloud connector. - :type authentication_details: ~azure.mgmt.security.models.AuthenticationDetailsProperties + :vartype authentication_details: ~azure.mgmt.security.models.AuthenticationDetailsProperties """ _validation = { @@ -3628,6 +4250,14 @@ def __init__( authentication_details: Optional["AuthenticationDetailsProperties"] = None, **kwargs ): + """ + :keyword hybrid_compute_settings: Settings for hybrid compute management. These settings are + relevant only for Arc autoProvision (Hybrid Compute). + :paramtype hybrid_compute_settings: ~azure.mgmt.security.models.HybridComputeSettingsProperties + :keyword authentication_details: Settings for authentication management, these settings are + relevant only for the cloud connector. + :paramtype authentication_details: ~azure.mgmt.security.models.AuthenticationDetailsProperties + """ super(ConnectorSetting, self).__init__(**kwargs) self.hybrid_compute_settings = hybrid_compute_settings self.authentication_details = authentication_details @@ -3638,8 +4268,8 @@ class ConnectorSettingList(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param value: List of all the cloud account connector settings. - :type value: list[~azure.mgmt.security.models.ConnectorSetting] + :ivar value: List of all the cloud account connector settings. + :vartype value: list[~azure.mgmt.security.models.ConnectorSetting] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -3659,6 +4289,10 @@ def __init__( value: Optional[List["ConnectorSetting"]] = None, **kwargs ): + """ + :keyword value: List of all the cloud account connector settings. + :paramtype value: list[~azure.mgmt.security.models.ConnectorSetting] + """ super(ConnectorSettingList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -3671,10 +4305,10 @@ class ContainerRegistryVulnerabilityProperties(AdditionalData): All required parameters must be populated in order to send to Azure. - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", + :ivar assessed_resource_type: Required. Sub-assessment resource type.Constant filled by server. + Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType + :vartype assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType :ivar type: Vulnerability Type. e.g: Vulnerability, Potential Vulnerability, Information Gathered, Vulnerability. :vartype type: str @@ -3722,6 +4356,8 @@ def __init__( self, **kwargs ): + """ + """ super(ContainerRegistryVulnerabilityProperties, self).__init__(**kwargs) self.assessed_resource_type = 'ContainerRegistryVulnerability' # type: str self.type = None @@ -3734,162 +4370,305 @@ def __init__( self.image_digest = None -class CVE(msrest.serialization.Model): - """CVE details. +class CspmMonitorAwsOffering(CloudOffering): + """The CSPM monitoring for AWS offering configurations. Variables are only populated by the server, and will be ignored when sending a request. - :ivar title: CVE title. - :vartype title: str - :ivar link: Link url. - :vartype link: str + All required parameters must be populated in order to send to Azure. + + :ivar offering_type: Required. The type of the security offering.Constant filled by server. + Possible values include: "CspmMonitorAws", "DefenderForContainersAws", "DefenderForServersAws". + :vartype offering_type: str or ~azure.mgmt.security.models.OfferingType + :ivar description: The offering description. + :vartype description: str + :ivar native_cloud_connection: The native cloud connection configuration. + :vartype native_cloud_connection: + ~azure.mgmt.security.models.CspmMonitorAwsOfferingNativeCloudConnection """ _validation = { - 'title': {'readonly': True}, - 'link': {'readonly': True}, + 'offering_type': {'required': True}, + 'description': {'readonly': True}, } _attribute_map = { - 'title': {'key': 'title', 'type': 'str'}, - 'link': {'key': 'link', 'type': 'str'}, + 'offering_type': {'key': 'offeringType', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'native_cloud_connection': {'key': 'nativeCloudConnection', 'type': 'CspmMonitorAwsOfferingNativeCloudConnection'}, } def __init__( self, + *, + native_cloud_connection: Optional["CspmMonitorAwsOfferingNativeCloudConnection"] = None, **kwargs ): - super(CVE, self).__init__(**kwargs) - self.title = None - self.link = None + """ + :keyword native_cloud_connection: The native cloud connection configuration. + :paramtype native_cloud_connection: + ~azure.mgmt.security.models.CspmMonitorAwsOfferingNativeCloudConnection + """ + super(CspmMonitorAwsOffering, self).__init__(**kwargs) + self.offering_type = 'CspmMonitorAws' # type: str + self.native_cloud_connection = native_cloud_connection -class CVSS(msrest.serialization.Model): - """CVSS details. +class CspmMonitorAwsOfferingNativeCloudConnection(msrest.serialization.Model): + """The native cloud connection configuration. + + :ivar cloud_role_arn: The cloud role ARN in AWS for this feature. + :vartype cloud_role_arn: str + """ + + _attribute_map = { + 'cloud_role_arn': {'key': 'cloudRoleArn', 'type': 'str'}, + } + + def __init__( + self, + *, + cloud_role_arn: Optional[str] = None, + **kwargs + ): + """ + :keyword cloud_role_arn: The cloud role ARN in AWS for this feature. + :paramtype cloud_role_arn: str + """ + super(CspmMonitorAwsOfferingNativeCloudConnection, self).__init__(**kwargs) + self.cloud_role_arn = cloud_role_arn + + +class CustomAssessmentAutomation(Resource): + """Custom Assessment Automation. Variables are only populated by the server, and will be ignored when sending a request. - :ivar base: CVSS base. - :vartype base: float + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.security.models.SystemData + :ivar compressed_query: GZip encoded KQL query representing the assessment automation results + required. + :vartype compressed_query: str + :ivar supported_cloud: Relevant cloud for the custom assessment automation. Possible values + include: "AWS". + :vartype supported_cloud: str or ~azure.mgmt.security.models.SupportedCloudEnum + :ivar severity: The severity to relate to the assessments generated by this assessment + automation. Possible values include: "High", "Medium", "Low". + :vartype severity: str or ~azure.mgmt.security.models.SeverityEnum + :ivar display_name: The display name of the assessments generated by this assessment + automation. + :vartype display_name: str + :ivar description: The description to relate to the assessments generated by this assessment + automation. + :vartype description: str + :ivar remediation_description: The remediation description to relate to the assessments + generated by this assessment automation. + :vartype remediation_description: str + :ivar assessment_key: The assessment metadata key used when an assessment is generated for this + assessment automation. + :vartype assessment_key: str """ _validation = { - 'base': {'readonly': True}, + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'system_data': {'readonly': True}, } _attribute_map = { - 'base': {'key': 'base', 'type': 'float'}, + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'system_data': {'key': 'systemData', 'type': 'SystemData'}, + 'compressed_query': {'key': 'properties.compressedQuery', 'type': 'str'}, + 'supported_cloud': {'key': 'properties.supportedCloud', 'type': 'str'}, + 'severity': {'key': 'properties.severity', 'type': 'str'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'remediation_description': {'key': 'properties.remediationDescription', 'type': 'str'}, + 'assessment_key': {'key': 'properties.assessmentKey', 'type': 'str'}, } def __init__( self, - **kwargs - ): - super(CVSS, self).__init__(**kwargs) - self.base = None + *, + compressed_query: Optional[str] = None, + supported_cloud: Optional[Union[str, "SupportedCloudEnum"]] = "AWS", + severity: Optional[Union[str, "SeverityEnum"]] = "Low", + display_name: Optional[str] = None, + description: Optional[str] = None, + remediation_description: Optional[str] = None, + assessment_key: Optional[str] = None, + **kwargs + ): + """ + :keyword compressed_query: GZip encoded KQL query representing the assessment automation + results required. + :paramtype compressed_query: str + :keyword supported_cloud: Relevant cloud for the custom assessment automation. Possible values + include: "AWS". + :paramtype supported_cloud: str or ~azure.mgmt.security.models.SupportedCloudEnum + :keyword severity: The severity to relate to the assessments generated by this assessment + automation. Possible values include: "High", "Medium", "Low". + :paramtype severity: str or ~azure.mgmt.security.models.SeverityEnum + :keyword display_name: The display name of the assessments generated by this assessment + automation. + :paramtype display_name: str + :keyword description: The description to relate to the assessments generated by this assessment + automation. + :paramtype description: str + :keyword remediation_description: The remediation description to relate to the assessments + generated by this assessment automation. + :paramtype remediation_description: str + :keyword assessment_key: The assessment metadata key used when an assessment is generated for + this assessment automation. + :paramtype assessment_key: str + """ + super(CustomAssessmentAutomation, self).__init__(**kwargs) + self.system_data = None + self.compressed_query = compressed_query + self.supported_cloud = supported_cloud + self.severity = severity + self.display_name = display_name + self.description = description + self.remediation_description = remediation_description + self.assessment_key = assessment_key -class DataExportSettings(Setting): - """Represents a data export setting. +class CustomAssessmentAutomationRequest(Resource): + """Custom Assessment Automation request. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. :vartype name: str :ivar type: Resource type. :vartype type: str - :param kind: Required. the kind of the settings string.Constant filled by server. Possible - values include: "DataExportSettings", "AlertSuppressionSetting", "AlertSyncSettings". - :type kind: str or ~azure.mgmt.security.models.SettingKind - :param enabled: Is the data export setting enabled. - :type enabled: bool + :ivar compressed_query: Base 64 encoded KQL query representing the assessment automation + results required. + :vartype compressed_query: str + :ivar supported_cloud: Relevant cloud for the custom assessment automation. Possible values + include: "AWS". + :vartype supported_cloud: str or ~azure.mgmt.security.models.SupportedCloudEnum + :ivar severity: The severity to relate to the assessments generated by this assessment + automation. Possible values include: "High", "Medium", "Low". + :vartype severity: str or ~azure.mgmt.security.models.SeverityEnum + :ivar display_name: The display name of the assessments generated by this assessment + automation. + :vartype display_name: str + :ivar description: The description to relate to the assessments generated by this assessment + automation. + :vartype description: str + :ivar remediation_description: The remediation description to relate to the assessments + generated by this assessment automation. + :vartype remediation_description: str """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'kind': {'required': True}, } _attribute_map = { 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, - 'enabled': {'key': 'properties.enabled', 'type': 'bool'}, + 'compressed_query': {'key': 'properties.compressedQuery', 'type': 'str'}, + 'supported_cloud': {'key': 'properties.supportedCloud', 'type': 'str'}, + 'severity': {'key': 'properties.severity', 'type': 'str'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'remediation_description': {'key': 'properties.remediationDescription', 'type': 'str'}, } def __init__( self, *, - enabled: Optional[bool] = None, + compressed_query: Optional[str] = None, + supported_cloud: Optional[Union[str, "SupportedCloudEnum"]] = "AWS", + severity: Optional[Union[str, "SeverityEnum"]] = "Low", + display_name: Optional[str] = None, + description: Optional[str] = None, + remediation_description: Optional[str] = None, **kwargs ): - super(DataExportSettings, self).__init__(**kwargs) - self.kind = 'DataExportSettings' # type: str - self.enabled = enabled + """ + :keyword compressed_query: Base 64 encoded KQL query representing the assessment automation + results required. + :paramtype compressed_query: str + :keyword supported_cloud: Relevant cloud for the custom assessment automation. Possible values + include: "AWS". + :paramtype supported_cloud: str or ~azure.mgmt.security.models.SupportedCloudEnum + :keyword severity: The severity to relate to the assessments generated by this assessment + automation. Possible values include: "High", "Medium", "Low". + :paramtype severity: str or ~azure.mgmt.security.models.SeverityEnum + :keyword display_name: The display name of the assessments generated by this assessment + automation. + :paramtype display_name: str + :keyword description: The description to relate to the assessments generated by this assessment + automation. + :paramtype description: str + :keyword remediation_description: The remediation description to relate to the assessments + generated by this assessment automation. + :paramtype remediation_description: str + """ + super(CustomAssessmentAutomationRequest, self).__init__(**kwargs) + self.compressed_query = compressed_query + self.supported_cloud = supported_cloud + self.severity = severity + self.display_name = display_name + self.description = description + self.remediation_description = remediation_description -class DenylistCustomAlertRule(ListCustomAlertRule): - """A custom alert rule that checks if a value (depends on the custom alert type) is denied. +class CustomAssessmentAutomationsListResult(msrest.serialization.Model): + """A list of Custom Assessment Automations. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param denylist_values: Required. The values to deny. The format of the values depends on the - rule type. - :type denylist_values: list[str] + :ivar value: Collection of Custom Assessment Automations. + :vartype value: list[~azure.mgmt.security.models.CustomAssessmentAutomation] + :ivar next_link: The link used to get the next page of operations. + :vartype next_link: str """ _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'denylist_values': {'required': True}, + 'value': {'readonly': True}, } _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'denylist_values': {'key': 'denylistValues', 'type': '[str]'}, + 'value': {'key': 'value', 'type': '[CustomAssessmentAutomation]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - is_enabled: bool, - denylist_values: List[str], + next_link: Optional[str] = None, **kwargs ): - super(DenylistCustomAlertRule, self).__init__(is_enabled=is_enabled, **kwargs) - self.rule_type = 'DenylistCustomAlertRule' # type: str - self.denylist_values = denylist_values + """ + :keyword next_link: The link used to get the next page of operations. + :paramtype next_link: str + """ + super(CustomAssessmentAutomationsListResult, self).__init__(**kwargs) + self.value = None + self.next_link = next_link -class DeviceSecurityGroup(Resource): - """The device security group resource. +class CustomEntityStoreAssignment(Resource): + """Custom entity store assignment. Variables are only populated by the server, and will be ignored when sending a request. @@ -3899,383 +4678,548 @@ class DeviceSecurityGroup(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param threshold_rules: The list of custom alert threshold rules. - :type threshold_rules: list[~azure.mgmt.security.models.ThresholdCustomAlertRule] - :param time_window_rules: The list of custom alert time-window rules. - :type time_window_rules: list[~azure.mgmt.security.models.TimeWindowCustomAlertRule] - :param allowlist_rules: The allow-list custom alert rules. - :type allowlist_rules: list[~azure.mgmt.security.models.AllowlistCustomAlertRule] - :param denylist_rules: The deny-list custom alert rules. - :type denylist_rules: list[~azure.mgmt.security.models.DenylistCustomAlertRule] + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.security.models.SystemData + :ivar principal: The principal assigned with entity store. Format of principal is: [AAD + type]=[PrincipalObjectId];[TenantId]. + :vartype principal: str + :ivar entity_store_database_link: The link to entity store database. + :vartype entity_store_database_link: str """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, + 'system_data': {'readonly': True}, } _attribute_map = { 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'threshold_rules': {'key': 'properties.thresholdRules', 'type': '[ThresholdCustomAlertRule]'}, - 'time_window_rules': {'key': 'properties.timeWindowRules', 'type': '[TimeWindowCustomAlertRule]'}, - 'allowlist_rules': {'key': 'properties.allowlistRules', 'type': '[AllowlistCustomAlertRule]'}, - 'denylist_rules': {'key': 'properties.denylistRules', 'type': '[DenylistCustomAlertRule]'}, + 'system_data': {'key': 'systemData', 'type': 'SystemData'}, + 'principal': {'key': 'properties.principal', 'type': 'str'}, + 'entity_store_database_link': {'key': 'properties.entityStoreDatabaseLink', 'type': 'str'}, } def __init__( self, *, - threshold_rules: Optional[List["ThresholdCustomAlertRule"]] = None, - time_window_rules: Optional[List["TimeWindowCustomAlertRule"]] = None, - allowlist_rules: Optional[List["AllowlistCustomAlertRule"]] = None, - denylist_rules: Optional[List["DenylistCustomAlertRule"]] = None, + principal: Optional[str] = None, + entity_store_database_link: Optional[str] = None, **kwargs ): - super(DeviceSecurityGroup, self).__init__(**kwargs) - self.threshold_rules = threshold_rules - self.time_window_rules = time_window_rules - self.allowlist_rules = allowlist_rules - self.denylist_rules = denylist_rules + """ + :keyword principal: The principal assigned with entity store. Format of principal is: [AAD + type]=[PrincipalObjectId];[TenantId]. + :paramtype principal: str + :keyword entity_store_database_link: The link to entity store database. + :paramtype entity_store_database_link: str + """ + super(CustomEntityStoreAssignment, self).__init__(**kwargs) + self.system_data = None + self.principal = principal + self.entity_store_database_link = entity_store_database_link -class DeviceSecurityGroupList(msrest.serialization.Model): - """List of device security groups. +class CustomEntityStoreAssignmentRequest(msrest.serialization.Model): + """describes the custom entity store assignment request. + + :ivar principal: The principal assigned with entity store. If not provided, will use caller + principal. Format of principal is: [AAD type]=[PrincipalObjectId];[TenantId]. + :vartype principal: str + """ + + _attribute_map = { + 'principal': {'key': 'properties.principal', 'type': 'str'}, + } + + def __init__( + self, + *, + principal: Optional[str] = None, + **kwargs + ): + """ + :keyword principal: The principal assigned with entity store. If not provided, will use caller + principal. Format of principal is: [AAD type]=[PrincipalObjectId];[TenantId]. + :paramtype principal: str + """ + super(CustomEntityStoreAssignmentRequest, self).__init__(**kwargs) + self.principal = principal + + +class CustomEntityStoreAssignmentsListResult(msrest.serialization.Model): + """A list of custom entity store assignments. Variables are only populated by the server, and will be ignored when sending a request. - :param value: List of device security group objects. - :type value: list[~azure.mgmt.security.models.DeviceSecurityGroup] - :ivar next_link: The URI to fetch the next page. + :ivar value: Collection of custom entity store assignments. + :vartype value: list[~azure.mgmt.security.models.CustomEntityStoreAssignment] + :ivar next_link: The link used to get the next page of operations. :vartype next_link: str """ _validation = { - 'next_link': {'readonly': True}, + 'value': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[DeviceSecurityGroup]'}, + 'value': {'key': 'value', 'type': '[CustomEntityStoreAssignment]'}, 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - value: Optional[List["DeviceSecurityGroup"]] = None, + next_link: Optional[str] = None, **kwargs ): - super(DeviceSecurityGroupList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + :keyword next_link: The link used to get the next page of operations. + :paramtype next_link: str + """ + super(CustomEntityStoreAssignmentsListResult, self).__init__(**kwargs) + self.value = None + self.next_link = next_link -class DirectMethodInvokesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of direct method invokes is not in allowed range. +class CVE(msrest.serialization.Model): + """CVE details. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + :ivar title: CVE title. + :vartype title: str + :ivar link: Link url. + :vartype link: str + """ - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + _validation = { + 'title': {'readonly': True}, + 'link': {'readonly': True}, + } + + _attribute_map = { + 'title': {'key': 'title', 'type': 'str'}, + 'link': {'key': 'link', 'type': 'str'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(CVE, self).__init__(**kwargs) + self.title = None + self.link = None + + +class CVSS(msrest.serialization.Model): + """CVSS details. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar base: CVSS base. + :vartype base: float """ _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, + 'base': {'readonly': True}, } _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'base': {'key': 'base', 'type': 'float'}, } def __init__( self, - *, - is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, **kwargs ): - super(DirectMethodInvokesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'DirectMethodInvokesNotInAllowedRange' # type: str + """ + """ + super(CVSS, self).__init__(**kwargs) + self.base = None -class DiscoveredSecuritySolution(Resource, Location): - """DiscoveredSecuritySolution. +class DataExportSettings(Setting): + """Represents a data export setting. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :ivar location: Location where the resource is stored. - :vartype location: str :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. :vartype name: str :ivar type: Resource type. :vartype type: str - :param security_family: Required. The security family of the discovered solution. Possible - values include: "Waf", "Ngfw", "SaasWaf", "Va". - :type security_family: str or ~azure.mgmt.security.models.SecurityFamily - :param offer: Required. The security solutions' image offer. - :type offer: str - :param publisher: Required. The security solutions' image publisher. - :type publisher: str - :param sku: Required. The security solutions' image sku. - :type sku: str + :ivar kind: Required. the kind of the settings string.Constant filled by server. Possible + values include: "DataExportSettings", "AlertSuppressionSetting", "AlertSyncSettings". + :vartype kind: str or ~azure.mgmt.security.models.SettingKind + :ivar enabled: Is the data export setting enabled. + :vartype enabled: bool """ _validation = { - 'location': {'readonly': True}, 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'security_family': {'required': True}, - 'offer': {'required': True}, - 'publisher': {'required': True}, - 'sku': {'required': True}, + 'kind': {'required': True}, } _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'security_family': {'key': 'properties.securityFamily', 'type': 'str'}, - 'offer': {'key': 'properties.offer', 'type': 'str'}, - 'publisher': {'key': 'properties.publisher', 'type': 'str'}, - 'sku': {'key': 'properties.sku', 'type': 'str'}, + 'kind': {'key': 'kind', 'type': 'str'}, + 'enabled': {'key': 'properties.enabled', 'type': 'bool'}, } def __init__( self, *, - security_family: Union[str, "SecurityFamily"], - offer: str, - publisher: str, - sku: str, + enabled: Optional[bool] = None, **kwargs ): - super(DiscoveredSecuritySolution, self).__init__(**kwargs) - self.location = None - self.security_family = security_family - self.offer = offer - self.publisher = publisher - self.sku = sku - self.id = None - self.name = None - self.type = None - self.security_family = security_family - self.offer = offer - self.publisher = publisher - self.sku = sku + """ + :keyword enabled: Is the data export setting enabled. + :paramtype enabled: bool + """ + super(DataExportSettings, self).__init__(**kwargs) + self.kind = 'DataExportSettings' # type: str + self.enabled = enabled -class DiscoveredSecuritySolutionList(msrest.serialization.Model): - """DiscoveredSecuritySolutionList. +class DefenderForContainersAwsOffering(CloudOffering): + """The Defender for Containers AWS offering configurations. Variables are only populated by the server, and will be ignored when sending a request. - :param value: - :type value: list[~azure.mgmt.security.models.DiscoveredSecuritySolution] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + All required parameters must be populated in order to send to Azure. + + :ivar offering_type: Required. The type of the security offering.Constant filled by server. + Possible values include: "CspmMonitorAws", "DefenderForContainersAws", "DefenderForServersAws". + :vartype offering_type: str or ~azure.mgmt.security.models.OfferingType + :ivar description: The offering description. + :vartype description: str + :ivar kubernetes_service: The kubernetes service connection configuration. + :vartype kubernetes_service: + ~azure.mgmt.security.models.DefenderForContainersAwsOfferingKubernetesService + :ivar kubernetes_scuba_reader: The kubernetes to scuba connection configuration. + :vartype kubernetes_scuba_reader: + ~azure.mgmt.security.models.DefenderForContainersAwsOfferingKubernetesScubaReader + :ivar cloud_watch_to_kinesis: The cloudwatch to kinesis connection configuration. + :vartype cloud_watch_to_kinesis: + ~azure.mgmt.security.models.DefenderForContainersAwsOfferingCloudWatchToKinesis + :ivar kinesis_to_s3: The kinesis to s3 connection configuration. + :vartype kinesis_to_s3: ~azure.mgmt.security.models.DefenderForContainersAwsOfferingKinesisToS3 """ _validation = { - 'next_link': {'readonly': True}, + 'offering_type': {'required': True}, + 'description': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[DiscoveredSecuritySolution]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'offering_type': {'key': 'offeringType', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'kubernetes_service': {'key': 'kubernetesService', 'type': 'DefenderForContainersAwsOfferingKubernetesService'}, + 'kubernetes_scuba_reader': {'key': 'kubernetesScubaReader', 'type': 'DefenderForContainersAwsOfferingKubernetesScubaReader'}, + 'cloud_watch_to_kinesis': {'key': 'cloudWatchToKinesis', 'type': 'DefenderForContainersAwsOfferingCloudWatchToKinesis'}, + 'kinesis_to_s3': {'key': 'kinesisToS3', 'type': 'DefenderForContainersAwsOfferingKinesisToS3'}, } def __init__( self, *, - value: Optional[List["DiscoveredSecuritySolution"]] = None, + kubernetes_service: Optional["DefenderForContainersAwsOfferingKubernetesService"] = None, + kubernetes_scuba_reader: Optional["DefenderForContainersAwsOfferingKubernetesScubaReader"] = None, + cloud_watch_to_kinesis: Optional["DefenderForContainersAwsOfferingCloudWatchToKinesis"] = None, + kinesis_to_s3: Optional["DefenderForContainersAwsOfferingKinesisToS3"] = None, **kwargs ): - super(DiscoveredSecuritySolutionList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + :keyword kubernetes_service: The kubernetes service connection configuration. + :paramtype kubernetes_service: + ~azure.mgmt.security.models.DefenderForContainersAwsOfferingKubernetesService + :keyword kubernetes_scuba_reader: The kubernetes to scuba connection configuration. + :paramtype kubernetes_scuba_reader: + ~azure.mgmt.security.models.DefenderForContainersAwsOfferingKubernetesScubaReader + :keyword cloud_watch_to_kinesis: The cloudwatch to kinesis connection configuration. + :paramtype cloud_watch_to_kinesis: + ~azure.mgmt.security.models.DefenderForContainersAwsOfferingCloudWatchToKinesis + :keyword kinesis_to_s3: The kinesis to s3 connection configuration. + :paramtype kinesis_to_s3: + ~azure.mgmt.security.models.DefenderForContainersAwsOfferingKinesisToS3 + """ + super(DefenderForContainersAwsOffering, self).__init__(**kwargs) + self.offering_type = 'DefenderForContainersAws' # type: str + self.kubernetes_service = kubernetes_service + self.kubernetes_scuba_reader = kubernetes_scuba_reader + self.cloud_watch_to_kinesis = cloud_watch_to_kinesis + self.kinesis_to_s3 = kinesis_to_s3 -class EffectiveNetworkSecurityGroups(msrest.serialization.Model): - """Describes the Network Security Groups effective on a network interface. +class DefenderForContainersAwsOfferingCloudWatchToKinesis(msrest.serialization.Model): + """The cloudwatch to kinesis connection configuration. - :param network_interface: The Azure resource ID of the network interface. - :type network_interface: str - :param network_security_groups: The Network Security Groups effective on the network interface. - :type network_security_groups: list[str] + :ivar cloud_role_arn: The cloud role ARN in AWS for this feature. + :vartype cloud_role_arn: str """ _attribute_map = { - 'network_interface': {'key': 'networkInterface', 'type': 'str'}, - 'network_security_groups': {'key': 'networkSecurityGroups', 'type': '[str]'}, + 'cloud_role_arn': {'key': 'cloudRoleArn', 'type': 'str'}, } def __init__( self, *, - network_interface: Optional[str] = None, - network_security_groups: Optional[List[str]] = None, + cloud_role_arn: Optional[str] = None, **kwargs ): - super(EffectiveNetworkSecurityGroups, self).__init__(**kwargs) - self.network_interface = network_interface - self.network_security_groups = network_security_groups - + """ + :keyword cloud_role_arn: The cloud role ARN in AWS for this feature. + :paramtype cloud_role_arn: str + """ + super(DefenderForContainersAwsOfferingCloudWatchToKinesis, self).__init__(**kwargs) + self.cloud_role_arn = cloud_role_arn -class ErrorAdditionalInfo(msrest.serialization.Model): - """The resource management error additional info. - Variables are only populated by the server, and will be ignored when sending a request. +class DefenderForContainersAwsOfferingKinesisToS3(msrest.serialization.Model): + """The kinesis to s3 connection configuration. - :ivar type: The additional info type. - :vartype type: str - :ivar info: The additional info. - :vartype info: any + :ivar cloud_role_arn: The cloud role ARN in AWS for this feature. + :vartype cloud_role_arn: str """ - _validation = { - 'type': {'readonly': True}, - 'info': {'readonly': True}, - } - _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'info': {'key': 'info', 'type': 'object'}, + 'cloud_role_arn': {'key': 'cloudRoleArn', 'type': 'str'}, } def __init__( self, + *, + cloud_role_arn: Optional[str] = None, **kwargs ): - super(ErrorAdditionalInfo, self).__init__(**kwargs) - self.type = None - self.info = None + """ + :keyword cloud_role_arn: The cloud role ARN in AWS for this feature. + :paramtype cloud_role_arn: str + """ + super(DefenderForContainersAwsOfferingKinesisToS3, self).__init__(**kwargs) + self.cloud_role_arn = cloud_role_arn -class ExternalSecuritySolutionList(msrest.serialization.Model): - """ExternalSecuritySolutionList. - - Variables are only populated by the server, and will be ignored when sending a request. +class DefenderForContainersAwsOfferingKubernetesScubaReader(msrest.serialization.Model): + """The kubernetes to scuba connection configuration. - :param value: - :type value: list[~azure.mgmt.security.models.ExternalSecuritySolution] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + :ivar cloud_role_arn: The cloud role ARN in AWS for this feature. + :vartype cloud_role_arn: str """ - _validation = { - 'next_link': {'readonly': True}, + _attribute_map = { + 'cloud_role_arn': {'key': 'cloudRoleArn', 'type': 'str'}, } + def __init__( + self, + *, + cloud_role_arn: Optional[str] = None, + **kwargs + ): + """ + :keyword cloud_role_arn: The cloud role ARN in AWS for this feature. + :paramtype cloud_role_arn: str + """ + super(DefenderForContainersAwsOfferingKubernetesScubaReader, self).__init__(**kwargs) + self.cloud_role_arn = cloud_role_arn + + +class DefenderForContainersAwsOfferingKubernetesService(msrest.serialization.Model): + """The kubernetes service connection configuration. + + :ivar cloud_role_arn: The cloud role ARN in AWS for this feature. + :vartype cloud_role_arn: str + """ + _attribute_map = { - 'value': {'key': 'value', 'type': '[ExternalSecuritySolution]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'cloud_role_arn': {'key': 'cloudRoleArn', 'type': 'str'}, } def __init__( self, *, - value: Optional[List["ExternalSecuritySolution"]] = None, + cloud_role_arn: Optional[str] = None, **kwargs ): - super(ExternalSecuritySolutionList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + :keyword cloud_role_arn: The cloud role ARN in AWS for this feature. + :paramtype cloud_role_arn: str + """ + super(DefenderForContainersAwsOfferingKubernetesService, self).__init__(**kwargs) + self.cloud_role_arn = cloud_role_arn -class FailedLocalLoginsNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of failed local logins is not in allowed range. +class DefenderForServersAwsOffering(CloudOffering): + """The Defender for Servers AWS offering configurations. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. + :ivar offering_type: Required. The type of the security offering.Constant filled by server. + Possible values include: "CspmMonitorAws", "DefenderForContainersAws", "DefenderForServersAws". + :vartype offering_type: str or ~azure.mgmt.security.models.OfferingType + :ivar description: The offering description. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar defender_for_servers: The Defender for servers connection configuration. + :vartype defender_for_servers: + ~azure.mgmt.security.models.DefenderForServersAwsOfferingDefenderForServers + :ivar arc_auto_provisioning: The ARC autoprovisioning configuration. + :vartype arc_auto_provisioning: + ~azure.mgmt.security.models.DefenderForServersAwsOfferingArcAutoProvisioning """ _validation = { - 'display_name': {'readonly': True}, + 'offering_type': {'required': True}, 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, } _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, + 'offering_type': {'key': 'offeringType', 'type': 'str'}, 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'defender_for_servers': {'key': 'defenderForServers', 'type': 'DefenderForServersAwsOfferingDefenderForServers'}, + 'arc_auto_provisioning': {'key': 'arcAutoProvisioning', 'type': 'DefenderForServersAwsOfferingArcAutoProvisioning'}, } def __init__( self, *, - is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, + defender_for_servers: Optional["DefenderForServersAwsOfferingDefenderForServers"] = None, + arc_auto_provisioning: Optional["DefenderForServersAwsOfferingArcAutoProvisioning"] = None, **kwargs ): - super(FailedLocalLoginsNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'FailedLocalLoginsNotInAllowedRange' # type: str + """ + :keyword defender_for_servers: The Defender for servers connection configuration. + :paramtype defender_for_servers: + ~azure.mgmt.security.models.DefenderForServersAwsOfferingDefenderForServers + :keyword arc_auto_provisioning: The ARC autoprovisioning configuration. + :paramtype arc_auto_provisioning: + ~azure.mgmt.security.models.DefenderForServersAwsOfferingArcAutoProvisioning + """ + super(DefenderForServersAwsOffering, self).__init__(**kwargs) + self.offering_type = 'DefenderForServersAWS' # type: str + self.defender_for_servers = defender_for_servers + self.arc_auto_provisioning = arc_auto_provisioning -class FileUploadsNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of file uploads is not in allowed range. +class DefenderForServersAwsOfferingArcAutoProvisioning(msrest.serialization.Model): + """The ARC autoprovisioning configuration. + + :ivar enabled: Is arc auto provisioning enabled. + :vartype enabled: bool + :ivar service_principal_secret_metadata: Metadata of Service Principal secret for + autoprovisioning. + :vartype service_principal_secret_metadata: + ~azure.mgmt.security.models.DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata + """ + + _attribute_map = { + 'enabled': {'key': 'enabled', 'type': 'bool'}, + 'service_principal_secret_metadata': {'key': 'servicePrincipalSecretMetadata', 'type': 'DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata'}, + } + + def __init__( + self, + *, + enabled: Optional[bool] = None, + service_principal_secret_metadata: Optional["DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata"] = None, + **kwargs + ): + """ + :keyword enabled: Is arc auto provisioning enabled. + :paramtype enabled: bool + :keyword service_principal_secret_metadata: Metadata of Service Principal secret for + autoprovisioning. + :paramtype service_principal_secret_metadata: + ~azure.mgmt.security.models.DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata + """ + super(DefenderForServersAwsOfferingArcAutoProvisioning, self).__init__(**kwargs) + self.enabled = enabled + self.service_principal_secret_metadata = service_principal_secret_metadata + + +class DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata(msrest.serialization.Model): + """Metadata of Service Principal secret for autoprovisioning. + + :ivar expiry_date: expiration date of service principal secret. + :vartype expiry_date: str + :ivar parameter_store_region: region of parameter store where secret is kept. + :vartype parameter_store_region: str + :ivar parameter_name_in_store: name of secret resource in parameter store. + :vartype parameter_name_in_store: str + """ + + _attribute_map = { + 'expiry_date': {'key': 'expiryDate', 'type': 'str'}, + 'parameter_store_region': {'key': 'parameterStoreRegion', 'type': 'str'}, + 'parameter_name_in_store': {'key': 'parameterNameInStore', 'type': 'str'}, + } + + def __init__( + self, + *, + expiry_date: Optional[str] = None, + parameter_store_region: Optional[str] = None, + parameter_name_in_store: Optional[str] = None, + **kwargs + ): + """ + :keyword expiry_date: expiration date of service principal secret. + :paramtype expiry_date: str + :keyword parameter_store_region: region of parameter store where secret is kept. + :paramtype parameter_store_region: str + :keyword parameter_name_in_store: name of secret resource in parameter store. + :paramtype parameter_name_in_store: str + """ + super(DefenderForServersAwsOfferingArcAutoProvisioningServicePrincipalSecretMetadata, self).__init__(**kwargs) + self.expiry_date = expiry_date + self.parameter_store_region = parameter_store_region + self.parameter_name_in_store = parameter_name_in_store + + +class DefenderForServersAwsOfferingDefenderForServers(msrest.serialization.Model): + """The Defender for servers connection configuration. + + :ivar cloud_role_arn: The cloud role ARN in AWS for this feature. + :vartype cloud_role_arn: str + """ + + _attribute_map = { + 'cloud_role_arn': {'key': 'cloudRoleArn', 'type': 'str'}, + } + + def __init__( + self, + *, + cloud_role_arn: Optional[str] = None, + **kwargs + ): + """ + :keyword cloud_role_arn: The cloud role ARN in AWS for this feature. + :paramtype cloud_role_arn: str + """ + super(DefenderForServersAwsOfferingDefenderForServers, self).__init__(**kwargs) + self.cloud_role_arn = cloud_role_arn + + +class DenylistCustomAlertRule(ListCustomAlertRule): + """A custom alert rule that checks if a value (depends on the custom alert type) is denied. Variables are only populated by the server, and will be ignored when sending a request. @@ -4285,16 +5229,16 @@ class FileUploadsNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", + "String". + :vartype value_type: str or ~azure.mgmt.security.models.ValueType + :ivar denylist_values: Required. The values to deny. The format of the values depends on the + rule type. + :vartype denylist_values: list[str] """ _validation = { @@ -4302,9 +5246,8 @@ class FileUploadsNotInAllowedRange(TimeWindowCustomAlertRule): 'description': {'readonly': True}, 'is_enabled': {'required': True}, 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, + 'value_type': {'readonly': True}, + 'denylist_values': {'required': True}, } _attribute_map = { @@ -4312,190 +5255,129 @@ class FileUploadsNotInAllowedRange(TimeWindowCustomAlertRule): 'description': {'key': 'description', 'type': 'str'}, 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'value_type': {'key': 'valueType', 'type': 'str'}, + 'denylist_values': {'key': 'denylistValues', 'type': '[str]'}, } def __init__( self, *, is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, + denylist_values: List[str], **kwargs ): - super(FileUploadsNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'FileUploadsNotInAllowedRange' # type: str + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword denylist_values: Required. The values to deny. The format of the values depends on the + rule type. + :paramtype denylist_values: list[str] + """ + super(DenylistCustomAlertRule, self).__init__(is_enabled=is_enabled, **kwargs) + self.rule_type = 'DenylistCustomAlertRule' # type: str + self.denylist_values = denylist_values -class GcpCredentialsDetailsProperties(AuthenticationDetailsProperties): - """GCP cloud account connector based service to service credentials, the credentials are composed of the organization ID and a JSON API key (write only). +class DeviceSecurityGroup(Resource): + """The device security group resource. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :ivar authentication_provisioning_state: State of the multi-cloud connector. Possible values - include: "Valid", "Invalid", "Expired", "IncorrectPolicy". - :vartype authentication_provisioning_state: str or - ~azure.mgmt.security.models.AuthenticationProvisioningState - :ivar granted_permissions: The permissions detected in the cloud account. - :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] - :param authentication_type: Required. Connect to your cloud account, for AWS use either account - credentials or role-based authentication. For GCP use account organization credentials.Constant - filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". - :type authentication_type: str or ~azure.mgmt.security.models.AuthenticationType - :param organization_id: Required. The organization ID of the GCP cloud account. - :type organization_id: str - :param type: Required. Type field of the API key (write only). - :type type: str - :param project_id: Required. Project ID field of the API key (write only). - :type project_id: str - :param private_key_id: Required. Private key ID field of the API key (write only). - :type private_key_id: str - :param private_key: Required. Private key field of the API key (write only). - :type private_key: str - :param client_email: Required. Client email field of the API key (write only). - :type client_email: str - :param client_id: Required. Client ID field of the API key (write only). - :type client_id: str - :param auth_uri: Required. Auth URI field of the API key (write only). - :type auth_uri: str - :param token_uri: Required. Token URI field of the API key (write only). - :type token_uri: str - :param auth_provider_x509_cert_url: Required. Auth provider x509 certificate URL field of the - API key (write only). - :type auth_provider_x509_cert_url: str - :param client_x509_cert_url: Required. Client x509 certificate URL field of the API key (write - only). - :type client_x509_cert_url: str + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar threshold_rules: The list of custom alert threshold rules. + :vartype threshold_rules: list[~azure.mgmt.security.models.ThresholdCustomAlertRule] + :ivar time_window_rules: The list of custom alert time-window rules. + :vartype time_window_rules: list[~azure.mgmt.security.models.TimeWindowCustomAlertRule] + :ivar allowlist_rules: The allow-list custom alert rules. + :vartype allowlist_rules: list[~azure.mgmt.security.models.AllowlistCustomAlertRule] + :ivar denylist_rules: The deny-list custom alert rules. + :vartype denylist_rules: list[~azure.mgmt.security.models.DenylistCustomAlertRule] """ _validation = { - 'authentication_provisioning_state': {'readonly': True}, - 'granted_permissions': {'readonly': True}, - 'authentication_type': {'required': True}, - 'organization_id': {'required': True}, - 'type': {'required': True}, - 'project_id': {'required': True}, - 'private_key_id': {'required': True}, - 'private_key': {'required': True}, - 'client_email': {'required': True}, - 'client_id': {'required': True}, - 'auth_uri': {'required': True}, - 'token_uri': {'required': True}, - 'auth_provider_x509_cert_url': {'required': True}, - 'client_x509_cert_url': {'required': True}, + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, } _attribute_map = { - 'authentication_provisioning_state': {'key': 'authenticationProvisioningState', 'type': 'str'}, - 'granted_permissions': {'key': 'grantedPermissions', 'type': '[str]'}, - 'authentication_type': {'key': 'authenticationType', 'type': 'str'}, - 'organization_id': {'key': 'organizationId', 'type': 'str'}, + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'project_id': {'key': 'projectId', 'type': 'str'}, - 'private_key_id': {'key': 'privateKeyId', 'type': 'str'}, - 'private_key': {'key': 'privateKey', 'type': 'str'}, - 'client_email': {'key': 'clientEmail', 'type': 'str'}, - 'client_id': {'key': 'clientId', 'type': 'str'}, - 'auth_uri': {'key': 'authUri', 'type': 'str'}, - 'token_uri': {'key': 'tokenUri', 'type': 'str'}, - 'auth_provider_x509_cert_url': {'key': 'authProviderX509CertUrl', 'type': 'str'}, - 'client_x509_cert_url': {'key': 'clientX509CertUrl', 'type': 'str'}, + 'threshold_rules': {'key': 'properties.thresholdRules', 'type': '[ThresholdCustomAlertRule]'}, + 'time_window_rules': {'key': 'properties.timeWindowRules', 'type': '[TimeWindowCustomAlertRule]'}, + 'allowlist_rules': {'key': 'properties.allowlistRules', 'type': '[AllowlistCustomAlertRule]'}, + 'denylist_rules': {'key': 'properties.denylistRules', 'type': '[DenylistCustomAlertRule]'}, } def __init__( self, *, - organization_id: str, - type: str, - project_id: str, - private_key_id: str, - private_key: str, - client_email: str, - client_id: str, - auth_uri: str, - token_uri: str, - auth_provider_x509_cert_url: str, - client_x509_cert_url: str, + threshold_rules: Optional[List["ThresholdCustomAlertRule"]] = None, + time_window_rules: Optional[List["TimeWindowCustomAlertRule"]] = None, + allowlist_rules: Optional[List["AllowlistCustomAlertRule"]] = None, + denylist_rules: Optional[List["DenylistCustomAlertRule"]] = None, **kwargs ): - super(GcpCredentialsDetailsProperties, self).__init__(**kwargs) - self.authentication_type = 'gcpCredentials' # type: str - self.organization_id = organization_id - self.type = type - self.project_id = project_id - self.private_key_id = private_key_id - self.private_key = private_key - self.client_email = client_email - self.client_id = client_id - self.auth_uri = auth_uri - self.token_uri = token_uri - self.auth_provider_x509_cert_url = auth_provider_x509_cert_url - self.client_x509_cert_url = client_x509_cert_url + """ + :keyword threshold_rules: The list of custom alert threshold rules. + :paramtype threshold_rules: list[~azure.mgmt.security.models.ThresholdCustomAlertRule] + :keyword time_window_rules: The list of custom alert time-window rules. + :paramtype time_window_rules: list[~azure.mgmt.security.models.TimeWindowCustomAlertRule] + :keyword allowlist_rules: The allow-list custom alert rules. + :paramtype allowlist_rules: list[~azure.mgmt.security.models.AllowlistCustomAlertRule] + :keyword denylist_rules: The deny-list custom alert rules. + :paramtype denylist_rules: list[~azure.mgmt.security.models.DenylistCustomAlertRule] + """ + super(DeviceSecurityGroup, self).__init__(**kwargs) + self.threshold_rules = threshold_rules + self.time_window_rules = time_window_rules + self.allowlist_rules = allowlist_rules + self.denylist_rules = denylist_rules -class HttpC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of cloud to device messages (HTTP protocol) is not in allowed range. +class DeviceSecurityGroupList(msrest.serialization.Model): + """List of device security groups. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar value: List of device security group objects. + :vartype value: list[~azure.mgmt.security.models.DeviceSecurityGroup] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str """ _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'value': {'key': 'value', 'type': '[DeviceSecurityGroup]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, + value: Optional[List["DeviceSecurityGroup"]] = None, **kwargs ): - super(HttpC2DMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'HttpC2DMessagesNotInAllowedRange' # type: str + """ + :keyword value: List of device security group objects. + :paramtype value: list[~azure.mgmt.security.models.DeviceSecurityGroup] + """ + super(DeviceSecurityGroupList, self).__init__(**kwargs) + self.value = value + self.next_link = None -class HttpC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of rejected cloud to device messages (HTTP protocol) is not in allowed range. +class DirectMethodInvokesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of direct method invokes is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. @@ -4505,16 +5387,16 @@ class HttpC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -4546,221 +5428,210 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): - super(HttpC2DRejectedMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'HttpC2DRejectedMessagesNotInAllowedRange' # type: str + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(DirectMethodInvokesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'DirectMethodInvokesNotInAllowedRange' # type: str -class HttpD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of device to cloud messages (HTTP protocol) is not in allowed range. +class DiscoveredSecuritySolution(Resource, Location): + """DiscoveredSecuritySolution. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar location: Location where the resource is stored. + :vartype location: str + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar security_family: Required. The security family of the discovered solution. Possible + values include: "Waf", "Ngfw", "SaasWaf", "Va". + :vartype security_family: str or ~azure.mgmt.security.models.SecurityFamily + :ivar offer: Required. The security solutions' image offer. + :vartype offer: str + :ivar publisher: Required. The security solutions' image publisher. + :vartype publisher: str + :ivar sku: Required. The security solutions' image sku. + :vartype sku: str """ _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, + 'location': {'readonly': True}, + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'security_family': {'required': True}, + 'offer': {'required': True}, + 'publisher': {'required': True}, + 'sku': {'required': True}, } _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'location': {'key': 'location', 'type': 'str'}, + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'security_family': {'key': 'properties.securityFamily', 'type': 'str'}, + 'offer': {'key': 'properties.offer', 'type': 'str'}, + 'publisher': {'key': 'properties.publisher', 'type': 'str'}, + 'sku': {'key': 'properties.sku', 'type': 'str'}, } def __init__( self, *, - is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, + security_family: Union[str, "SecurityFamily"], + offer: str, + publisher: str, + sku: str, **kwargs ): - super(HttpD2CMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'HttpD2CMessagesNotInAllowedRange' # type: str + """ + :keyword security_family: Required. The security family of the discovered solution. Possible + values include: "Waf", "Ngfw", "SaasWaf", "Va". + :paramtype security_family: str or ~azure.mgmt.security.models.SecurityFamily + :keyword offer: Required. The security solutions' image offer. + :paramtype offer: str + :keyword publisher: Required. The security solutions' image publisher. + :paramtype publisher: str + :keyword sku: Required. The security solutions' image sku. + :paramtype sku: str + """ + super(DiscoveredSecuritySolution, self).__init__(**kwargs) + self.location = None + self.security_family = security_family + self.offer = offer + self.publisher = publisher + self.sku = sku + self.id = None + self.name = None + self.type = None + self.security_family = security_family + self.offer = offer + self.publisher = publisher + self.sku = sku -class HybridComputeSettingsProperties(msrest.serialization.Model): - """Settings for hybrid compute management. +class DiscoveredSecuritySolutionList(msrest.serialization.Model): + """DiscoveredSecuritySolutionList. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :ivar hybrid_compute_provisioning_state: State of the service principal and its secret. - Possible values include: "Valid", "Invalid", "Expired". - :vartype hybrid_compute_provisioning_state: str or - ~azure.mgmt.security.models.HybridComputeProvisioningState - :param auto_provision: Required. Whether or not to automatically install Azure Arc (hybrid - compute) agents on machines. Possible values include: "On", "Off". - :type auto_provision: str or ~azure.mgmt.security.models.AutoProvision - :param resource_group_name: The name of the resource group where Arc (Hybrid Compute) - connectors are connected. - :type resource_group_name: str - :param region: The location where the metadata of machines will be stored. - :type region: str - :param proxy_server: For a non-Azure machine that is not connected directly to the internet, - specify a proxy server that the non-Azure machine can use. - :type proxy_server: ~azure.mgmt.security.models.ProxyServerProperties - :param service_principal: An object to access resources that are secured by an Azure AD tenant. - :type service_principal: ~azure.mgmt.security.models.ServicePrincipalProperties + :ivar value: + :vartype value: list[~azure.mgmt.security.models.DiscoveredSecuritySolution] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str """ _validation = { - 'hybrid_compute_provisioning_state': {'readonly': True}, - 'auto_provision': {'required': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'hybrid_compute_provisioning_state': {'key': 'hybridComputeProvisioningState', 'type': 'str'}, - 'auto_provision': {'key': 'autoProvision', 'type': 'str'}, - 'resource_group_name': {'key': 'resourceGroupName', 'type': 'str'}, - 'region': {'key': 'region', 'type': 'str'}, - 'proxy_server': {'key': 'proxyServer', 'type': 'ProxyServerProperties'}, - 'service_principal': {'key': 'servicePrincipal', 'type': 'ServicePrincipalProperties'}, + 'value': {'key': 'value', 'type': '[DiscoveredSecuritySolution]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - auto_provision: Union[str, "AutoProvision"], - resource_group_name: Optional[str] = None, - region: Optional[str] = None, - proxy_server: Optional["ProxyServerProperties"] = None, - service_principal: Optional["ServicePrincipalProperties"] = None, + value: Optional[List["DiscoveredSecuritySolution"]] = None, **kwargs ): - super(HybridComputeSettingsProperties, self).__init__(**kwargs) - self.hybrid_compute_provisioning_state = None - self.auto_provision = auto_provision - self.resource_group_name = resource_group_name - self.region = region - self.proxy_server = proxy_server - self.service_principal = service_principal + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.DiscoveredSecuritySolution] + """ + super(DiscoveredSecuritySolutionList, self).__init__(**kwargs) + self.value = value + self.next_link = None -class InformationProtectionKeyword(msrest.serialization.Model): - """The information type keyword. +class EffectiveNetworkSecurityGroups(msrest.serialization.Model): + """Describes the Network Security Groups effective on a network interface. - :param pattern: The keyword pattern. - :type pattern: str - :param custom: Indicates whether the keyword is custom or not. - :type custom: bool - :param can_be_numeric: Indicates whether the keyword can be applied on numeric types or not. - :type can_be_numeric: bool - :param excluded: Indicates whether the keyword is excluded or not. - :type excluded: bool + :ivar network_interface: The Azure resource ID of the network interface. + :vartype network_interface: str + :ivar network_security_groups: The Network Security Groups effective on the network interface. + :vartype network_security_groups: list[str] """ _attribute_map = { - 'pattern': {'key': 'pattern', 'type': 'str'}, - 'custom': {'key': 'custom', 'type': 'bool'}, - 'can_be_numeric': {'key': 'canBeNumeric', 'type': 'bool'}, - 'excluded': {'key': 'excluded', 'type': 'bool'}, + 'network_interface': {'key': 'networkInterface', 'type': 'str'}, + 'network_security_groups': {'key': 'networkSecurityGroups', 'type': '[str]'}, } def __init__( self, *, - pattern: Optional[str] = None, - custom: Optional[bool] = None, - can_be_numeric: Optional[bool] = None, - excluded: Optional[bool] = None, + network_interface: Optional[str] = None, + network_security_groups: Optional[List[str]] = None, **kwargs ): - super(InformationProtectionKeyword, self).__init__(**kwargs) - self.pattern = pattern - self.custom = custom - self.can_be_numeric = can_be_numeric - self.excluded = excluded + """ + :keyword network_interface: The Azure resource ID of the network interface. + :paramtype network_interface: str + :keyword network_security_groups: The Network Security Groups effective on the network + interface. + :paramtype network_security_groups: list[str] + """ + super(EffectiveNetworkSecurityGroups, self).__init__(**kwargs) + self.network_interface = network_interface + self.network_security_groups = network_security_groups -class InformationProtectionPolicy(Resource): - """Information protection policy. +class ErrorAdditionalInfo(msrest.serialization.Model): + """The resource management error additional info. Variables are only populated by the server, and will be ignored when sending a request. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. + :ivar type: The additional info type. :vartype type: str - :ivar last_modified_utc: Describes the last UTC time the policy was modified. - :vartype last_modified_utc: ~datetime.datetime - :ivar version: Describes the version of the policy. - :vartype version: str - :param labels: Dictionary of sensitivity labels. - :type labels: dict[str, ~azure.mgmt.security.models.SensitivityLabel] - :param information_types: The sensitivity information types. - :type information_types: dict[str, ~azure.mgmt.security.models.InformationType] + :ivar info: The additional info. + :vartype info: any """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, 'type': {'readonly': True}, - 'last_modified_utc': {'readonly': True}, - 'version': {'readonly': True}, + 'info': {'readonly': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'last_modified_utc': {'key': 'properties.lastModifiedUtc', 'type': 'iso-8601'}, - 'version': {'key': 'properties.version', 'type': 'str'}, - 'labels': {'key': 'properties.labels', 'type': '{SensitivityLabel}'}, - 'information_types': {'key': 'properties.informationTypes', 'type': '{InformationType}'}, + 'info': {'key': 'info', 'type': 'object'}, } def __init__( self, - *, - labels: Optional[Dict[str, "SensitivityLabel"]] = None, - information_types: Optional[Dict[str, "InformationType"]] = None, **kwargs ): - super(InformationProtectionPolicy, self).__init__(**kwargs) - self.last_modified_utc = None - self.version = None - self.labels = labels - self.information_types = information_types + """ + """ + super(ErrorAdditionalInfo, self).__init__(**kwargs) + self.type = None + self.info = None -class InformationProtectionPolicyList(msrest.serialization.Model): - """Information protection policies response. +class ExternalSecuritySolutionList(msrest.serialization.Model): + """ExternalSecuritySolutionList. Variables are only populated by the server, and will be ignored when sending a request. - :param value: List of information protection policies. - :type value: list[~azure.mgmt.security.models.InformationProtectionPolicy] + :ivar value: + :vartype value: list[~azure.mgmt.security.models.ExternalSecuritySolution] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -4770,831 +5641,993 @@ class InformationProtectionPolicyList(msrest.serialization.Model): } _attribute_map = { - 'value': {'key': 'value', 'type': '[InformationProtectionPolicy]'}, + 'value': {'key': 'value', 'type': '[ExternalSecuritySolution]'}, 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - value: Optional[List["InformationProtectionPolicy"]] = None, + value: Optional[List["ExternalSecuritySolution"]] = None, **kwargs ): - super(InformationProtectionPolicyList, self).__init__(**kwargs) + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.ExternalSecuritySolution] + """ + super(ExternalSecuritySolutionList, self).__init__(**kwargs) self.value = value self.next_link = None -class InformationType(msrest.serialization.Model): - """The information type. +class FailedLocalLoginsNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of failed local logins is not in allowed range. - :param display_name: The name of the information type. - :type display_name: str - :param description: The description of the information type. - :type description: str - :param order: The order of the information type. - :type order: int - :param recommended_label_id: The recommended label id to be associated with this information - type. - :type recommended_label_id: str - :param enabled: Indicates whether the information type is enabled or not. - :type enabled: bool - :param custom: Indicates whether the information type is custom or not. - :type custom: bool - :param keywords: The information type keywords. - :type keywords: list[~azure.mgmt.security.models.InformationProtectionKeyword] + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. + :vartype description: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ + _validation = { + 'display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, + } + _attribute_map = { 'display_name': {'key': 'displayName', 'type': 'str'}, 'description': {'key': 'description', 'type': 'str'}, - 'order': {'key': 'order', 'type': 'int'}, - 'recommended_label_id': {'key': 'recommendedLabelId', 'type': 'str'}, - 'enabled': {'key': 'enabled', 'type': 'bool'}, - 'custom': {'key': 'custom', 'type': 'bool'}, - 'keywords': {'key': 'keywords', 'type': '[InformationProtectionKeyword]'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, *, - display_name: Optional[str] = None, - description: Optional[str] = None, - order: Optional[int] = None, - recommended_label_id: Optional[str] = None, - enabled: Optional[bool] = None, - custom: Optional[bool] = None, - keywords: Optional[List["InformationProtectionKeyword"]] = None, + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(InformationType, self).__init__(**kwargs) - self.display_name = display_name - self.description = description - self.order = order - self.recommended_label_id = recommended_label_id - self.enabled = enabled - self.custom = custom - self.keywords = keywords + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(FailedLocalLoginsNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'FailedLocalLoginsNotInAllowedRange' # type: str -class IngestionConnectionString(msrest.serialization.Model): - """Connection string for ingesting security data and logs. +class FileUploadsNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of file uploads is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. - :ivar location: The region where ingested logs and data resides. - :vartype location: str - :ivar value: Connection string value. - :vartype value: str + All required parameters must be populated in order to send to Azure. + + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. + :vartype description: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { - 'location': {'readonly': True}, - 'value': {'readonly': True}, + 'display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, } _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'value': {'key': 'value', 'type': 'str'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, + *, + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(IngestionConnectionString, self).__init__(**kwargs) - self.location = None - self.value = None + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(FileUploadsNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'FileUploadsNotInAllowedRange' # type: str -class IngestionSetting(Resource): - """Configures how to correlate scan data and logs with resources associated with the subscription. +class GcpCredentialsDetailsProperties(AuthenticationDetailsProperties): + """GCP cloud account connector based service to service credentials, the credentials are composed of the organization ID and a JSON API key (write only). Variables are only populated by the server, and will be ignored when sending a request. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. + All required parameters must be populated in order to send to Azure. + + :ivar authentication_provisioning_state: State of the multi-cloud connector. Possible values + include: "Valid", "Invalid", "Expired", "IncorrectPolicy". + :vartype authentication_provisioning_state: str or + ~azure.mgmt.security.models.AuthenticationProvisioningState + :ivar granted_permissions: The permissions detected in the cloud account. + :vartype granted_permissions: list[str or ~azure.mgmt.security.models.PermissionProperty] + :ivar authentication_type: Required. Connect to your cloud account, for AWS use either account + credentials or role-based authentication. For GCP use account organization credentials.Constant + filled by server. Possible values include: "awsCreds", "awsAssumeRole", "gcpCredentials". + :vartype authentication_type: str or ~azure.mgmt.security.models.AuthenticationType + :ivar organization_id: Required. The organization ID of the GCP cloud account. + :vartype organization_id: str + :ivar type: Required. Type field of the API key (write only). :vartype type: str - :param properties: Ingestion setting data. - :type properties: any + :ivar project_id: Required. Project ID field of the API key (write only). + :vartype project_id: str + :ivar private_key_id: Required. Private key ID field of the API key (write only). + :vartype private_key_id: str + :ivar private_key: Required. Private key field of the API key (write only). + :vartype private_key: str + :ivar client_email: Required. Client email field of the API key (write only). + :vartype client_email: str + :ivar client_id: Required. Client ID field of the API key (write only). + :vartype client_id: str + :ivar auth_uri: Required. Auth URI field of the API key (write only). + :vartype auth_uri: str + :ivar token_uri: Required. Token URI field of the API key (write only). + :vartype token_uri: str + :ivar auth_provider_x509_cert_url: Required. Auth provider x509 certificate URL field of the + API key (write only). + :vartype auth_provider_x509_cert_url: str + :ivar client_x509_cert_url: Required. Client x509 certificate URL field of the API key (write + only). + :vartype client_x509_cert_url: str """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, + 'authentication_provisioning_state': {'readonly': True}, + 'granted_permissions': {'readonly': True}, + 'authentication_type': {'required': True}, + 'organization_id': {'required': True}, + 'type': {'required': True}, + 'project_id': {'required': True}, + 'private_key_id': {'required': True}, + 'private_key': {'required': True}, + 'client_email': {'required': True}, + 'client_id': {'required': True}, + 'auth_uri': {'required': True}, + 'token_uri': {'required': True}, + 'auth_provider_x509_cert_url': {'required': True}, + 'client_x509_cert_url': {'required': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, + 'authentication_provisioning_state': {'key': 'authenticationProvisioningState', 'type': 'str'}, + 'granted_permissions': {'key': 'grantedPermissions', 'type': '[str]'}, + 'authentication_type': {'key': 'authenticationType', 'type': 'str'}, + 'organization_id': {'key': 'organizationId', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'object'}, + 'project_id': {'key': 'projectId', 'type': 'str'}, + 'private_key_id': {'key': 'privateKeyId', 'type': 'str'}, + 'private_key': {'key': 'privateKey', 'type': 'str'}, + 'client_email': {'key': 'clientEmail', 'type': 'str'}, + 'client_id': {'key': 'clientId', 'type': 'str'}, + 'auth_uri': {'key': 'authUri', 'type': 'str'}, + 'token_uri': {'key': 'tokenUri', 'type': 'str'}, + 'auth_provider_x509_cert_url': {'key': 'authProviderX509CertUrl', 'type': 'str'}, + 'client_x509_cert_url': {'key': 'clientX509CertUrl', 'type': 'str'}, } def __init__( self, *, - properties: Optional[Any] = None, + organization_id: str, + type: str, + project_id: str, + private_key_id: str, + private_key: str, + client_email: str, + client_id: str, + auth_uri: str, + token_uri: str, + auth_provider_x509_cert_url: str, + client_x509_cert_url: str, **kwargs ): - super(IngestionSetting, self).__init__(**kwargs) - self.properties = properties + """ + :keyword organization_id: Required. The organization ID of the GCP cloud account. + :paramtype organization_id: str + :keyword type: Required. Type field of the API key (write only). + :paramtype type: str + :keyword project_id: Required. Project ID field of the API key (write only). + :paramtype project_id: str + :keyword private_key_id: Required. Private key ID field of the API key (write only). + :paramtype private_key_id: str + :keyword private_key: Required. Private key field of the API key (write only). + :paramtype private_key: str + :keyword client_email: Required. Client email field of the API key (write only). + :paramtype client_email: str + :keyword client_id: Required. Client ID field of the API key (write only). + :paramtype client_id: str + :keyword auth_uri: Required. Auth URI field of the API key (write only). + :paramtype auth_uri: str + :keyword token_uri: Required. Token URI field of the API key (write only). + :paramtype token_uri: str + :keyword auth_provider_x509_cert_url: Required. Auth provider x509 certificate URL field of the + API key (write only). + :paramtype auth_provider_x509_cert_url: str + :keyword client_x509_cert_url: Required. Client x509 certificate URL field of the API key + (write only). + :paramtype client_x509_cert_url: str + """ + super(GcpCredentialsDetailsProperties, self).__init__(**kwargs) + self.authentication_type = 'gcpCredentials' # type: str + self.organization_id = organization_id + self.type = type + self.project_id = project_id + self.private_key_id = private_key_id + self.private_key = private_key + self.client_email = client_email + self.client_id = client_id + self.auth_uri = auth_uri + self.token_uri = token_uri + self.auth_provider_x509_cert_url = auth_provider_x509_cert_url + self.client_x509_cert_url = client_x509_cert_url -class IngestionSettingList(msrest.serialization.Model): - """List of ingestion settings. +class HttpC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of cloud to device messages (HTTP protocol) is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. - :ivar value: List of ingestion settings. - :vartype value: list[~azure.mgmt.security.models.IngestionSetting] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + All required parameters must be populated in order to send to Azure. + + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. + :vartype description: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, + 'display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[IngestionSetting]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, + *, + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(IngestionSettingList, self).__init__(**kwargs) - self.value = None - self.next_link = None + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(HttpC2DMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'HttpC2DMessagesNotInAllowedRange' # type: str -class IngestionSettingToken(msrest.serialization.Model): - """Configures how to correlate scan data and logs with resources associated with the subscription. +class HttpC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of rejected cloud to device messages (HTTP protocol) is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. - :ivar token: The token is used for correlating security data and logs with the resources in the - subscription. - :vartype token: str + All required parameters must be populated in order to send to Azure. + + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. + :vartype description: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { - 'token': {'readonly': True}, - } - - _attribute_map = { - 'token': {'key': 'token', 'type': 'str'}, + 'display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, } - def __init__( - self, - **kwargs - ): - super(IngestionSettingToken, self).__init__(**kwargs) - self.token = None - - -class TagsResource(msrest.serialization.Model): - """A container holding only the Tags for a resource, allowing the user to update the tags. - - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - """ - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, *, - tags: Optional[Dict[str, str]] = None, + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(TagsResource, self).__init__(**kwargs) - self.tags = tags + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(HttpC2DRejectedMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'HttpC2DRejectedMessagesNotInAllowedRange' # type: str -class IoTSecurityAggregatedAlert(Resource, TagsResource): - """Security Solution Aggregated Alert information. +class HttpD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of device to cloud messages (HTTP protocol) is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar alert_type: Name of the alert type. - :vartype alert_type: str - :ivar alert_display_name: Display name of the alert type. - :vartype alert_display_name: str - :ivar aggregated_date_utc: Date of detection. - :vartype aggregated_date_utc: ~datetime.date - :ivar vendor_name: Name of the organization that raised the alert. - :vartype vendor_name: str - :ivar reported_severity: Assessed alert severity. Possible values include: "Informational", - "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar remediation_steps: Recommended steps for remediation. - :vartype remediation_steps: str - :ivar description: Description of the suspected vulnerability and meaning. + All required parameters must be populated in order to send to Azure. + + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. :vartype description: str - :ivar count: Number of alerts occurrences within the aggregated time window. - :vartype count: long - :ivar effected_resource_type: Azure resource ID of the resource that received the alerts. - :vartype effected_resource_type: str - :ivar system_source: The type of the alerted resource (Azure, Non-Azure). - :vartype system_source: str - :ivar action_taken: IoT Security solution alert response. - :vartype action_taken: str - :ivar log_analytics_query: Log analytics query for getting the list of affected devices/alerts. - :vartype log_analytics_query: str - :ivar top_devices_list: 10 devices with the highest number of occurrences of this alert type, - on this day. - :vartype top_devices_list: - list[~azure.mgmt.security.models.IoTSecurityAggregatedAlertPropertiesTopDevicesListItem] + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'alert_type': {'readonly': True}, - 'alert_display_name': {'readonly': True}, - 'aggregated_date_utc': {'readonly': True}, - 'vendor_name': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'remediation_steps': {'readonly': True}, + 'display_name': {'readonly': True}, 'description': {'readonly': True}, - 'count': {'readonly': True}, - 'effected_resource_type': {'readonly': True}, - 'system_source': {'readonly': True}, - 'action_taken': {'readonly': True}, - 'log_analytics_query': {'readonly': True}, - 'top_devices_list': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, } _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'alert_type': {'key': 'properties.alertType', 'type': 'str'}, - 'alert_display_name': {'key': 'properties.alertDisplayName', 'type': 'str'}, - 'aggregated_date_utc': {'key': 'properties.aggregatedDateUtc', 'type': 'date'}, - 'vendor_name': {'key': 'properties.vendorName', 'type': 'str'}, - 'reported_severity': {'key': 'properties.reportedSeverity', 'type': 'str'}, - 'remediation_steps': {'key': 'properties.remediationSteps', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'count': {'key': 'properties.count', 'type': 'long'}, - 'effected_resource_type': {'key': 'properties.effectedResourceType', 'type': 'str'}, - 'system_source': {'key': 'properties.systemSource', 'type': 'str'}, - 'action_taken': {'key': 'properties.actionTaken', 'type': 'str'}, - 'log_analytics_query': {'key': 'properties.logAnalyticsQuery', 'type': 'str'}, - 'top_devices_list': {'key': 'properties.topDevicesList', 'type': '[IoTSecurityAggregatedAlertPropertiesTopDevicesListItem]'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, *, - tags: Optional[Dict[str, str]] = None, + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(IoTSecurityAggregatedAlert, self).__init__(tags=tags, **kwargs) - self.tags = tags - self.alert_type = None - self.alert_display_name = None - self.aggregated_date_utc = None - self.vendor_name = None - self.reported_severity = None - self.remediation_steps = None - self.description = None - self.count = None - self.effected_resource_type = None - self.system_source = None - self.action_taken = None - self.log_analytics_query = None - self.top_devices_list = None - self.id = None - self.name = None - self.type = None - self.alert_type = None - self.alert_display_name = None - self.aggregated_date_utc = None - self.vendor_name = None - self.reported_severity = None - self.remediation_steps = None - self.description = None - self.count = None - self.effected_resource_type = None - self.system_source = None - self.action_taken = None - self.log_analytics_query = None - self.top_devices_list = None + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(HttpD2CMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'HttpD2CMessagesNotInAllowedRange' # type: str -class IoTSecurityAggregatedAlertList(msrest.serialization.Model): - """List of IoT Security solution aggregated alert data. +class HybridComputeSettingsProperties(msrest.serialization.Model): + """Settings for hybrid compute management. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :param value: Required. List of aggregated alerts data. - :type value: list[~azure.mgmt.security.models.IoTSecurityAggregatedAlert] - :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next - page. - :vartype next_link: str + :ivar hybrid_compute_provisioning_state: State of the service principal and its secret. + Possible values include: "Valid", "Invalid", "Expired". + :vartype hybrid_compute_provisioning_state: str or + ~azure.mgmt.security.models.HybridComputeProvisioningState + :ivar auto_provision: Required. Whether or not to automatically install Azure Arc (hybrid + compute) agents on machines. Possible values include: "On", "Off". + :vartype auto_provision: str or ~azure.mgmt.security.models.AutoProvision + :ivar resource_group_name: The name of the resource group where Arc (Hybrid Compute) connectors + are connected. + :vartype resource_group_name: str + :ivar region: The location where the metadata of machines will be stored. + :vartype region: str + :ivar proxy_server: For a non-Azure machine that is not connected directly to the internet, + specify a proxy server that the non-Azure machine can use. + :vartype proxy_server: ~azure.mgmt.security.models.ProxyServerProperties + :ivar service_principal: An object to access resources that are secured by an Azure AD tenant. + :vartype service_principal: ~azure.mgmt.security.models.ServicePrincipalProperties """ _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, + 'hybrid_compute_provisioning_state': {'readonly': True}, + 'auto_provision': {'required': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecurityAggregatedAlert]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'hybrid_compute_provisioning_state': {'key': 'hybridComputeProvisioningState', 'type': 'str'}, + 'auto_provision': {'key': 'autoProvision', 'type': 'str'}, + 'resource_group_name': {'key': 'resourceGroupName', 'type': 'str'}, + 'region': {'key': 'region', 'type': 'str'}, + 'proxy_server': {'key': 'proxyServer', 'type': 'ProxyServerProperties'}, + 'service_principal': {'key': 'servicePrincipal', 'type': 'ServicePrincipalProperties'}, } def __init__( self, *, - value: List["IoTSecurityAggregatedAlert"], + auto_provision: Union[str, "AutoProvision"], + resource_group_name: Optional[str] = None, + region: Optional[str] = None, + proxy_server: Optional["ProxyServerProperties"] = None, + service_principal: Optional["ServicePrincipalProperties"] = None, **kwargs ): - super(IoTSecurityAggregatedAlertList, self).__init__(**kwargs) - self.value = value - self.next_link = None - + """ + :keyword auto_provision: Required. Whether or not to automatically install Azure Arc (hybrid + compute) agents on machines. Possible values include: "On", "Off". + :paramtype auto_provision: str or ~azure.mgmt.security.models.AutoProvision + :keyword resource_group_name: The name of the resource group where Arc (Hybrid Compute) + connectors are connected. + :paramtype resource_group_name: str + :keyword region: The location where the metadata of machines will be stored. + :paramtype region: str + :keyword proxy_server: For a non-Azure machine that is not connected directly to the internet, + specify a proxy server that the non-Azure machine can use. + :paramtype proxy_server: ~azure.mgmt.security.models.ProxyServerProperties + :keyword service_principal: An object to access resources that are secured by an Azure AD + tenant. + :paramtype service_principal: ~azure.mgmt.security.models.ServicePrincipalProperties + """ + super(HybridComputeSettingsProperties, self).__init__(**kwargs) + self.hybrid_compute_provisioning_state = None + self.auto_provision = auto_provision + self.resource_group_name = resource_group_name + self.region = region + self.proxy_server = proxy_server + self.service_principal = service_principal -class IoTSecurityAggregatedAlertPropertiesTopDevicesListItem(msrest.serialization.Model): - """IoTSecurityAggregatedAlertPropertiesTopDevicesListItem. - Variables are only populated by the server, and will be ignored when sending a request. +class InformationProtectionKeyword(msrest.serialization.Model): + """The information type keyword. - :ivar device_id: Name of the device. - :vartype device_id: str - :ivar alerts_count: Number of alerts raised for this device. - :vartype alerts_count: long - :ivar last_occurrence: Most recent time this alert was raised for this device, on this day. - :vartype last_occurrence: str + :ivar pattern: The keyword pattern. + :vartype pattern: str + :ivar custom: Indicates whether the keyword is custom or not. + :vartype custom: bool + :ivar can_be_numeric: Indicates whether the keyword can be applied on numeric types or not. + :vartype can_be_numeric: bool + :ivar excluded: Indicates whether the keyword is excluded or not. + :vartype excluded: bool """ - _validation = { - 'device_id': {'readonly': True}, - 'alerts_count': {'readonly': True}, - 'last_occurrence': {'readonly': True}, - } - _attribute_map = { - 'device_id': {'key': 'deviceId', 'type': 'str'}, - 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, - 'last_occurrence': {'key': 'lastOccurrence', 'type': 'str'}, + 'pattern': {'key': 'pattern', 'type': 'str'}, + 'custom': {'key': 'custom', 'type': 'bool'}, + 'can_be_numeric': {'key': 'canBeNumeric', 'type': 'bool'}, + 'excluded': {'key': 'excluded', 'type': 'bool'}, } def __init__( self, + *, + pattern: Optional[str] = None, + custom: Optional[bool] = None, + can_be_numeric: Optional[bool] = None, + excluded: Optional[bool] = None, **kwargs ): - super(IoTSecurityAggregatedAlertPropertiesTopDevicesListItem, self).__init__(**kwargs) - self.device_id = None - self.alerts_count = None - self.last_occurrence = None + """ + :keyword pattern: The keyword pattern. + :paramtype pattern: str + :keyword custom: Indicates whether the keyword is custom or not. + :paramtype custom: bool + :keyword can_be_numeric: Indicates whether the keyword can be applied on numeric types or not. + :paramtype can_be_numeric: bool + :keyword excluded: Indicates whether the keyword is excluded or not. + :paramtype excluded: bool + """ + super(InformationProtectionKeyword, self).__init__(**kwargs) + self.pattern = pattern + self.custom = custom + self.can_be_numeric = can_be_numeric + self.excluded = excluded -class IoTSecurityAggregatedRecommendation(Resource, TagsResource): - """IoT Security solution recommendation information. +class InformationProtectionPolicy(Resource): + """Information protection policy. Variables are only populated by the server, and will be ignored when sending a request. - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. :vartype name: str :ivar type: Resource type. :vartype type: str - :param recommendation_name: Name of the recommendation. - :type recommendation_name: str - :ivar recommendation_display_name: Display name of the recommendation type. - :vartype recommendation_display_name: str - :ivar description: Description of the suspected vulnerability and meaning. - :vartype description: str - :ivar recommendation_type_id: Recommendation-type GUID. - :vartype recommendation_type_id: str - :ivar detected_by: Name of the organization that made the recommendation. - :vartype detected_by: str - :ivar remediation_steps: Recommended steps for remediation. - :vartype remediation_steps: str - :ivar reported_severity: Assessed recommendation severity. Possible values include: - "Informational", "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar healthy_devices: Number of healthy devices within the IoT Security solution. - :vartype healthy_devices: long - :ivar unhealthy_device_count: Number of unhealthy devices within the IoT Security solution. - :vartype unhealthy_device_count: long - :ivar log_analytics_query: Log analytics query for getting the list of affected devices/alerts. - :vartype log_analytics_query: str + :ivar last_modified_utc: Describes the last UTC time the policy was modified. + :vartype last_modified_utc: ~datetime.datetime + :ivar version: Describes the version of the policy. + :vartype version: str + :ivar labels: Dictionary of sensitivity labels. + :vartype labels: dict[str, ~azure.mgmt.security.models.SensitivityLabel] + :ivar information_types: The sensitivity information types. + :vartype information_types: dict[str, ~azure.mgmt.security.models.InformationType] """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'recommendation_display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'recommendation_type_id': {'readonly': True}, - 'detected_by': {'readonly': True}, - 'remediation_steps': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'healthy_devices': {'readonly': True}, - 'unhealthy_device_count': {'readonly': True}, - 'log_analytics_query': {'readonly': True}, + 'last_modified_utc': {'readonly': True}, + 'version': {'readonly': True}, } _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'recommendation_name': {'key': 'properties.recommendationName', 'type': 'str'}, - 'recommendation_display_name': {'key': 'properties.recommendationDisplayName', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'recommendation_type_id': {'key': 'properties.recommendationTypeId', 'type': 'str'}, - 'detected_by': {'key': 'properties.detectedBy', 'type': 'str'}, - 'remediation_steps': {'key': 'properties.remediationSteps', 'type': 'str'}, - 'reported_severity': {'key': 'properties.reportedSeverity', 'type': 'str'}, - 'healthy_devices': {'key': 'properties.healthyDevices', 'type': 'long'}, - 'unhealthy_device_count': {'key': 'properties.unhealthyDeviceCount', 'type': 'long'}, - 'log_analytics_query': {'key': 'properties.logAnalyticsQuery', 'type': 'str'}, + 'last_modified_utc': {'key': 'properties.lastModifiedUtc', 'type': 'iso-8601'}, + 'version': {'key': 'properties.version', 'type': 'str'}, + 'labels': {'key': 'properties.labels', 'type': '{SensitivityLabel}'}, + 'information_types': {'key': 'properties.informationTypes', 'type': '{InformationType}'}, } def __init__( self, *, - tags: Optional[Dict[str, str]] = None, - recommendation_name: Optional[str] = None, + labels: Optional[Dict[str, "SensitivityLabel"]] = None, + information_types: Optional[Dict[str, "InformationType"]] = None, **kwargs ): - super(IoTSecurityAggregatedRecommendation, self).__init__(tags=tags, **kwargs) - self.tags = tags - self.recommendation_name = recommendation_name - self.recommendation_display_name = None - self.description = None - self.recommendation_type_id = None - self.detected_by = None - self.remediation_steps = None - self.reported_severity = None - self.healthy_devices = None - self.unhealthy_device_count = None - self.log_analytics_query = None - self.id = None - self.name = None - self.type = None - self.recommendation_name = recommendation_name - self.recommendation_display_name = None - self.description = None - self.recommendation_type_id = None - self.detected_by = None - self.remediation_steps = None - self.reported_severity = None - self.healthy_devices = None - self.unhealthy_device_count = None - self.log_analytics_query = None + """ + :keyword labels: Dictionary of sensitivity labels. + :paramtype labels: dict[str, ~azure.mgmt.security.models.SensitivityLabel] + :keyword information_types: The sensitivity information types. + :paramtype information_types: dict[str, ~azure.mgmt.security.models.InformationType] + """ + super(InformationProtectionPolicy, self).__init__(**kwargs) + self.last_modified_utc = None + self.version = None + self.labels = labels + self.information_types = information_types -class IoTSecurityAggregatedRecommendationList(msrest.serialization.Model): - """List of IoT Security solution aggregated recommendations. +class InformationProtectionPolicyList(msrest.serialization.Model): + """Information protection policies response. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of aggregated recommendations data. - :type value: list[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendation] - :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next - page. + :ivar value: List of information protection policies. + :vartype value: list[~azure.mgmt.security.models.InformationProtectionPolicy] + :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ _validation = { - 'value': {'required': True}, 'next_link': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecurityAggregatedRecommendation]'}, + 'value': {'key': 'value', 'type': '[InformationProtectionPolicy]'}, 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - value: List["IoTSecurityAggregatedRecommendation"], + value: Optional[List["InformationProtectionPolicy"]] = None, **kwargs ): - super(IoTSecurityAggregatedRecommendationList, self).__init__(**kwargs) + """ + :keyword value: List of information protection policies. + :paramtype value: list[~azure.mgmt.security.models.InformationProtectionPolicy] + """ + super(InformationProtectionPolicyList, self).__init__(**kwargs) self.value = value self.next_link = None -class IoTSecurityAlertedDevice(msrest.serialization.Model): - """Statistical information about the number of alerts per device during last set number of days. - - Variables are only populated by the server, and will be ignored when sending a request. +class InformationType(msrest.serialization.Model): + """The information type. - :ivar device_id: Device identifier. - :vartype device_id: str - :ivar alerts_count: Number of alerts raised for this device. - :vartype alerts_count: long + :ivar display_name: The name of the information type. + :vartype display_name: str + :ivar description: The description of the information type. + :vartype description: str + :ivar order: The order of the information type. + :vartype order: int + :ivar recommended_label_id: The recommended label id to be associated with this information + type. + :vartype recommended_label_id: str + :ivar enabled: Indicates whether the information type is enabled or not. + :vartype enabled: bool + :ivar custom: Indicates whether the information type is custom or not. + :vartype custom: bool + :ivar keywords: The information type keywords. + :vartype keywords: list[~azure.mgmt.security.models.InformationProtectionKeyword] """ - _validation = { - 'device_id': {'readonly': True}, - 'alerts_count': {'readonly': True}, - } - _attribute_map = { - 'device_id': {'key': 'deviceId', 'type': 'str'}, - 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'order': {'key': 'order', 'type': 'int'}, + 'recommended_label_id': {'key': 'recommendedLabelId', 'type': 'str'}, + 'enabled': {'key': 'enabled', 'type': 'bool'}, + 'custom': {'key': 'custom', 'type': 'bool'}, + 'keywords': {'key': 'keywords', 'type': '[InformationProtectionKeyword]'}, } def __init__( self, + *, + display_name: Optional[str] = None, + description: Optional[str] = None, + order: Optional[int] = None, + recommended_label_id: Optional[str] = None, + enabled: Optional[bool] = None, + custom: Optional[bool] = None, + keywords: Optional[List["InformationProtectionKeyword"]] = None, **kwargs ): - super(IoTSecurityAlertedDevice, self).__init__(**kwargs) - self.device_id = None - self.alerts_count = None + """ + :keyword display_name: The name of the information type. + :paramtype display_name: str + :keyword description: The description of the information type. + :paramtype description: str + :keyword order: The order of the information type. + :paramtype order: int + :keyword recommended_label_id: The recommended label id to be associated with this information + type. + :paramtype recommended_label_id: str + :keyword enabled: Indicates whether the information type is enabled or not. + :paramtype enabled: bool + :keyword custom: Indicates whether the information type is custom or not. + :paramtype custom: bool + :keyword keywords: The information type keywords. + :paramtype keywords: list[~azure.mgmt.security.models.InformationProtectionKeyword] + """ + super(InformationType, self).__init__(**kwargs) + self.display_name = display_name + self.description = description + self.order = order + self.recommended_label_id = recommended_label_id + self.enabled = enabled + self.custom = custom + self.keywords = keywords -class IoTSecurityDeviceAlert(msrest.serialization.Model): - """Statistical information about the number of alerts per alert type during last set number of days. +class IngestionConnectionString(msrest.serialization.Model): + """Connection string for ingesting security data and logs. Variables are only populated by the server, and will be ignored when sending a request. - :ivar alert_display_name: Display name of the alert. - :vartype alert_display_name: str - :ivar reported_severity: Assessed Alert severity. Possible values include: "Informational", - "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar alerts_count: Number of alerts raised for this alert type. - :vartype alerts_count: long + :ivar location: The region where ingested logs and data resides. + :vartype location: str + :ivar value: Connection string value. + :vartype value: str """ _validation = { - 'alert_display_name': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'alerts_count': {'readonly': True}, + 'location': {'readonly': True}, + 'value': {'readonly': True}, } _attribute_map = { - 'alert_display_name': {'key': 'alertDisplayName', 'type': 'str'}, - 'reported_severity': {'key': 'reportedSeverity', 'type': 'str'}, - 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, + 'location': {'key': 'location', 'type': 'str'}, + 'value': {'key': 'value', 'type': 'str'}, } def __init__( self, **kwargs ): - super(IoTSecurityDeviceAlert, self).__init__(**kwargs) - self.alert_display_name = None - self.reported_severity = None - self.alerts_count = None + """ + """ + super(IngestionConnectionString, self).__init__(**kwargs) + self.location = None + self.value = None -class IoTSecurityDeviceRecommendation(msrest.serialization.Model): - """Statistical information about the number of recommendations per device, per recommendation type. +class IngestionSetting(Resource): + """Configures how to correlate scan data and logs with resources associated with the subscription. Variables are only populated by the server, and will be ignored when sending a request. - :ivar recommendation_display_name: Display name of the recommendation. - :vartype recommendation_display_name: str - :ivar reported_severity: Assessed recommendation severity. Possible values include: - "Informational", "Low", "Medium", "High". - :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity - :ivar devices_count: Number of devices with this recommendation. - :vartype devices_count: long + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar properties: Ingestion setting data. + :vartype properties: any """ _validation = { - 'recommendation_display_name': {'readonly': True}, - 'reported_severity': {'readonly': True}, - 'devices_count': {'readonly': True}, + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, } _attribute_map = { - 'recommendation_display_name': {'key': 'recommendationDisplayName', 'type': 'str'}, - 'reported_severity': {'key': 'reportedSeverity', 'type': 'str'}, - 'devices_count': {'key': 'devicesCount', 'type': 'long'}, - } - + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'properties': {'key': 'properties', 'type': 'object'}, + } + def __init__( self, + *, + properties: Optional[Any] = None, **kwargs ): - super(IoTSecurityDeviceRecommendation, self).__init__(**kwargs) - self.recommendation_display_name = None - self.reported_severity = None - self.devices_count = None + """ + :keyword properties: Ingestion setting data. + :paramtype properties: any + """ + super(IngestionSetting, self).__init__(**kwargs) + self.properties = properties -class IoTSecuritySolutionAnalyticsModel(Resource): - """Security analytics of your IoT Security solution. +class IngestionSettingList(msrest.serialization.Model): + """List of ingestion settings. Variables are only populated by the server, and will be ignored when sending a request. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar metrics: Security analytics of your IoT Security solution. - :vartype metrics: ~azure.mgmt.security.models.IoTSeverityMetrics - :ivar unhealthy_device_count: Number of unhealthy devices within your IoT Security solution. - :vartype unhealthy_device_count: long - :ivar devices_metrics: List of device metrics by the aggregation date. - :vartype devices_metrics: - list[~azure.mgmt.security.models.IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem] - :param top_alerted_devices: List of the 3 devices with the most alerts. - :type top_alerted_devices: list[~azure.mgmt.security.models.IoTSecurityAlertedDevice] - :param most_prevalent_device_alerts: List of the 3 most prevalent device alerts. - :type most_prevalent_device_alerts: list[~azure.mgmt.security.models.IoTSecurityDeviceAlert] - :param most_prevalent_device_recommendations: List of the 3 most prevalent device - recommendations. - :type most_prevalent_device_recommendations: - list[~azure.mgmt.security.models.IoTSecurityDeviceRecommendation] + :ivar value: List of ingestion settings. + :vartype value: list[~azure.mgmt.security.models.IngestionSetting] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'metrics': {'readonly': True}, - 'unhealthy_device_count': {'readonly': True}, - 'devices_metrics': {'readonly': True}, + 'value': {'readonly': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'metrics': {'key': 'properties.metrics', 'type': 'IoTSeverityMetrics'}, - 'unhealthy_device_count': {'key': 'properties.unhealthyDeviceCount', 'type': 'long'}, - 'devices_metrics': {'key': 'properties.devicesMetrics', 'type': '[IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem]'}, - 'top_alerted_devices': {'key': 'properties.topAlertedDevices', 'type': '[IoTSecurityAlertedDevice]'}, - 'most_prevalent_device_alerts': {'key': 'properties.mostPrevalentDeviceAlerts', 'type': '[IoTSecurityDeviceAlert]'}, - 'most_prevalent_device_recommendations': {'key': 'properties.mostPrevalentDeviceRecommendations', 'type': '[IoTSecurityDeviceRecommendation]'}, + 'value': {'key': 'value', 'type': '[IngestionSetting]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, - *, - top_alerted_devices: Optional[List["IoTSecurityAlertedDevice"]] = None, - most_prevalent_device_alerts: Optional[List["IoTSecurityDeviceAlert"]] = None, - most_prevalent_device_recommendations: Optional[List["IoTSecurityDeviceRecommendation"]] = None, **kwargs ): - super(IoTSecuritySolutionAnalyticsModel, self).__init__(**kwargs) - self.metrics = None - self.unhealthy_device_count = None - self.devices_metrics = None - self.top_alerted_devices = top_alerted_devices - self.most_prevalent_device_alerts = most_prevalent_device_alerts - self.most_prevalent_device_recommendations = most_prevalent_device_recommendations + """ + """ + super(IngestionSettingList, self).__init__(**kwargs) + self.value = None + self.next_link = None -class IoTSecuritySolutionAnalyticsModelList(msrest.serialization.Model): - """List of Security analytics of your IoT Security solution. +class IngestionSettingToken(msrest.serialization.Model): + """Configures how to correlate scan data and logs with resources associated with the subscription. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of Security analytics of your IoT Security solution. - :type value: list[~azure.mgmt.security.models.IoTSecuritySolutionAnalyticsModel] - :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next - page. - :vartype next_link: str + :ivar token: The token is used for correlating security data and logs with the resources in the + subscription. + :vartype token: str """ _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, + 'token': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecuritySolutionAnalyticsModel]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'token': {'key': 'token', 'type': 'str'}, } def __init__( self, - *, - value: List["IoTSecuritySolutionAnalyticsModel"], **kwargs ): - super(IoTSecuritySolutionAnalyticsModelList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + """ + super(IngestionSettingToken, self).__init__(**kwargs) + self.token = None -class IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem(msrest.serialization.Model): - """IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem. +class TagsResource(msrest.serialization.Model): + """A container holding only the Tags for a resource, allowing the user to update the tags. - :param date: Aggregation of IoT Security solution device alert metrics by date. - :type date: ~datetime.datetime - :param devices_metrics: Device alert count by severity. - :type devices_metrics: ~azure.mgmt.security.models.IoTSeverityMetrics + :ivar tags: A set of tags. Resource tags. + :vartype tags: dict[str, str] """ _attribute_map = { - 'date': {'key': 'date', 'type': 'iso-8601'}, - 'devices_metrics': {'key': 'devicesMetrics', 'type': 'IoTSeverityMetrics'}, + 'tags': {'key': 'tags', 'type': '{str}'}, } def __init__( self, *, - date: Optional[datetime.datetime] = None, - devices_metrics: Optional["IoTSeverityMetrics"] = None, + tags: Optional[Dict[str, str]] = None, **kwargs ): - super(IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem, self).__init__(**kwargs) - self.date = date - self.devices_metrics = devices_metrics + """ + :keyword tags: A set of tags. Resource tags. + :paramtype tags: dict[str, str] + """ + super(TagsResource, self).__init__(**kwargs) + self.tags = tags -class IoTSecuritySolutionModel(Resource, TagsResource): - """IoT Security solution configuration and resource information. +class IoTSecurityAggregatedAlert(Resource, TagsResource): + """Security Solution Aggregated Alert information. Variables are only populated by the server, and will be ignored when sending a request. - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] + :ivar tags: A set of tags. Resource tags. + :vartype tags: dict[str, str] :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. :vartype name: str :ivar type: Resource type. :vartype type: str - :param location: The resource location. - :type location: str - :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy - information. - :vartype system_data: ~azure.mgmt.security.models.SystemData - :param workspace: Workspace resource ID. - :type workspace: str - :param display_name: Resource display name. - :type display_name: str - :param status: Status of the IoT Security solution. Possible values include: "Enabled", - "Disabled". Default value: "Enabled". - :type status: str or ~azure.mgmt.security.models.SecuritySolutionStatus - :param export: List of additional options for exporting to workspace data. - :type export: list[str or ~azure.mgmt.security.models.ExportData] - :param disabled_data_sources: Disabled data sources. Disabling these data sources compromises - the system. - :type disabled_data_sources: list[str or ~azure.mgmt.security.models.DataSource] - :param iot_hubs: IoT Hub resource IDs. - :type iot_hubs: list[str] - :param user_defined_resources: Properties of the IoT Security solution's user defined - resources. - :type user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties - :ivar auto_discovered_resources: List of resources that were automatically discovered as - relevant to the security solution. - :vartype auto_discovered_resources: list[str] - :param recommendations_configuration: List of the configuration status for each recommendation - type. - :type recommendations_configuration: - list[~azure.mgmt.security.models.RecommendationConfigurationProperties] - :param unmasked_ip_logging_status: Unmasked IP address logging status. Possible values include: - "Disabled", "Enabled". Default value: "Disabled". - :type unmasked_ip_logging_status: str or ~azure.mgmt.security.models.UnmaskedIpLoggingStatus - :param additional_workspaces: List of additional workspaces. - :type additional_workspaces: list[~azure.mgmt.security.models.AdditionalWorkspacesProperties] + :ivar alert_type: Name of the alert type. + :vartype alert_type: str + :ivar alert_display_name: Display name of the alert type. + :vartype alert_display_name: str + :ivar aggregated_date_utc: Date of detection. + :vartype aggregated_date_utc: ~datetime.date + :ivar vendor_name: Name of the organization that raised the alert. + :vartype vendor_name: str + :ivar reported_severity: Assessed alert severity. Possible values include: "Informational", + "Low", "Medium", "High". + :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity + :ivar remediation_steps: Recommended steps for remediation. + :vartype remediation_steps: str + :ivar description: Description of the suspected vulnerability and meaning. + :vartype description: str + :ivar count: Number of alerts occurrences within the aggregated time window. + :vartype count: long + :ivar effected_resource_type: Azure resource ID of the resource that received the alerts. + :vartype effected_resource_type: str + :ivar system_source: The type of the alerted resource (Azure, Non-Azure). + :vartype system_source: str + :ivar action_taken: IoT Security solution alert response. + :vartype action_taken: str + :ivar log_analytics_query: Log analytics query for getting the list of affected devices/alerts. + :vartype log_analytics_query: str + :ivar top_devices_list: 10 devices with the highest number of occurrences of this alert type, + on this day. + :vartype top_devices_list: + list[~azure.mgmt.security.models.IoTSecurityAggregatedAlertPropertiesTopDevicesListItem] """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'system_data': {'readonly': True}, - 'auto_discovered_resources': {'readonly': True}, + 'alert_type': {'readonly': True}, + 'alert_display_name': {'readonly': True}, + 'aggregated_date_utc': {'readonly': True}, + 'vendor_name': {'readonly': True}, + 'reported_severity': {'readonly': True}, + 'remediation_steps': {'readonly': True}, + 'description': {'readonly': True}, + 'count': {'readonly': True}, + 'effected_resource_type': {'readonly': True}, + 'system_source': {'readonly': True}, + 'action_taken': {'readonly': True}, + 'log_analytics_query': {'readonly': True}, + 'top_devices_list': {'readonly': True}, } _attribute_map = { @@ -5602,81 +6635,75 @@ class IoTSecuritySolutionModel(Resource, TagsResource): 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'system_data': {'key': 'systemData', 'type': 'SystemData'}, - 'workspace': {'key': 'properties.workspace', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'status': {'key': 'properties.status', 'type': 'str'}, - 'export': {'key': 'properties.export', 'type': '[str]'}, - 'disabled_data_sources': {'key': 'properties.disabledDataSources', 'type': '[str]'}, - 'iot_hubs': {'key': 'properties.iotHubs', 'type': '[str]'}, - 'user_defined_resources': {'key': 'properties.userDefinedResources', 'type': 'UserDefinedResourcesProperties'}, - 'auto_discovered_resources': {'key': 'properties.autoDiscoveredResources', 'type': '[str]'}, - 'recommendations_configuration': {'key': 'properties.recommendationsConfiguration', 'type': '[RecommendationConfigurationProperties]'}, - 'unmasked_ip_logging_status': {'key': 'properties.unmaskedIpLoggingStatus', 'type': 'str'}, - 'additional_workspaces': {'key': 'properties.additionalWorkspaces', 'type': '[AdditionalWorkspacesProperties]'}, + 'alert_type': {'key': 'properties.alertType', 'type': 'str'}, + 'alert_display_name': {'key': 'properties.alertDisplayName', 'type': 'str'}, + 'aggregated_date_utc': {'key': 'properties.aggregatedDateUtc', 'type': 'date'}, + 'vendor_name': {'key': 'properties.vendorName', 'type': 'str'}, + 'reported_severity': {'key': 'properties.reportedSeverity', 'type': 'str'}, + 'remediation_steps': {'key': 'properties.remediationSteps', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'count': {'key': 'properties.count', 'type': 'long'}, + 'effected_resource_type': {'key': 'properties.effectedResourceType', 'type': 'str'}, + 'system_source': {'key': 'properties.systemSource', 'type': 'str'}, + 'action_taken': {'key': 'properties.actionTaken', 'type': 'str'}, + 'log_analytics_query': {'key': 'properties.logAnalyticsQuery', 'type': 'str'}, + 'top_devices_list': {'key': 'properties.topDevicesList', 'type': '[IoTSecurityAggregatedAlertPropertiesTopDevicesListItem]'}, } def __init__( self, *, tags: Optional[Dict[str, str]] = None, - location: Optional[str] = None, - workspace: Optional[str] = None, - display_name: Optional[str] = None, - status: Optional[Union[str, "SecuritySolutionStatus"]] = "Enabled", - export: Optional[List[Union[str, "ExportData"]]] = None, - disabled_data_sources: Optional[List[Union[str, "DataSource"]]] = None, - iot_hubs: Optional[List[str]] = None, - user_defined_resources: Optional["UserDefinedResourcesProperties"] = None, - recommendations_configuration: Optional[List["RecommendationConfigurationProperties"]] = None, - unmasked_ip_logging_status: Optional[Union[str, "UnmaskedIpLoggingStatus"]] = "Disabled", - additional_workspaces: Optional[List["AdditionalWorkspacesProperties"]] = None, **kwargs ): - super(IoTSecuritySolutionModel, self).__init__(tags=tags, **kwargs) + """ + :keyword tags: A set of tags. Resource tags. + :paramtype tags: dict[str, str] + """ + super(IoTSecurityAggregatedAlert, self).__init__(tags=tags, **kwargs) self.tags = tags - self.location = location - self.system_data = None - self.workspace = workspace - self.display_name = display_name - self.status = status - self.export = export - self.disabled_data_sources = disabled_data_sources - self.iot_hubs = iot_hubs - self.user_defined_resources = user_defined_resources - self.auto_discovered_resources = None - self.recommendations_configuration = recommendations_configuration - self.unmasked_ip_logging_status = unmasked_ip_logging_status - self.additional_workspaces = additional_workspaces + self.alert_type = None + self.alert_display_name = None + self.aggregated_date_utc = None + self.vendor_name = None + self.reported_severity = None + self.remediation_steps = None + self.description = None + self.count = None + self.effected_resource_type = None + self.system_source = None + self.action_taken = None + self.log_analytics_query = None + self.top_devices_list = None self.id = None self.name = None self.type = None - self.location = location - self.system_data = None - self.workspace = workspace - self.display_name = display_name - self.status = status - self.export = export - self.disabled_data_sources = disabled_data_sources - self.iot_hubs = iot_hubs - self.user_defined_resources = user_defined_resources - self.auto_discovered_resources = None - self.recommendations_configuration = recommendations_configuration - self.unmasked_ip_logging_status = unmasked_ip_logging_status - self.additional_workspaces = additional_workspaces - - -class IoTSecuritySolutionsList(msrest.serialization.Model): - """List of IoT Security solutions. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of IoT Security solutions. - :type value: list[~azure.mgmt.security.models.IoTSecuritySolutionModel] - :ivar next_link: The URI to fetch the next page. + self.alert_type = None + self.alert_display_name = None + self.aggregated_date_utc = None + self.vendor_name = None + self.reported_severity = None + self.remediation_steps = None + self.description = None + self.count = None + self.effected_resource_type = None + self.system_source = None + self.action_taken = None + self.log_analytics_query = None + self.top_devices_list = None + + +class IoTSecurityAggregatedAlertList(msrest.serialization.Model): + """List of IoT Security solution aggregated alert data. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar value: Required. List of aggregated alerts data. + :vartype value: list[~azure.mgmt.security.models.IoTSecurityAggregatedAlert] + :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next + page. :vartype next_link: str """ @@ -5686,1141 +6713,1252 @@ class IoTSecuritySolutionsList(msrest.serialization.Model): } _attribute_map = { - 'value': {'key': 'value', 'type': '[IoTSecuritySolutionModel]'}, + 'value': {'key': 'value', 'type': '[IoTSecurityAggregatedAlert]'}, 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - value: List["IoTSecuritySolutionModel"], + value: List["IoTSecurityAggregatedAlert"], **kwargs ): - super(IoTSecuritySolutionsList, self).__init__(**kwargs) + """ + :keyword value: Required. List of aggregated alerts data. + :paramtype value: list[~azure.mgmt.security.models.IoTSecurityAggregatedAlert] + """ + super(IoTSecurityAggregatedAlertList, self).__init__(**kwargs) self.value = value self.next_link = None -class IoTSeverityMetrics(msrest.serialization.Model): - """IoT Security solution analytics severity metrics. - - :param high: Count of high severity alerts/recommendations. - :type high: long - :param medium: Count of medium severity alerts/recommendations. - :type medium: long - :param low: Count of low severity alerts/recommendations. - :type low: long - """ - - _attribute_map = { - 'high': {'key': 'high', 'type': 'long'}, - 'medium': {'key': 'medium', 'type': 'long'}, - 'low': {'key': 'low', 'type': 'long'}, - } - - def __init__( - self, - *, - high: Optional[int] = None, - medium: Optional[int] = None, - low: Optional[int] = None, - **kwargs - ): - super(IoTSeverityMetrics, self).__init__(**kwargs) - self.high = high - self.medium = medium - self.low = low - - -class JitNetworkAccessPoliciesList(msrest.serialization.Model): - """JitNetworkAccessPoliciesList. +class IoTSecurityAggregatedAlertPropertiesTopDevicesListItem(msrest.serialization.Model): + """IoTSecurityAggregatedAlertPropertiesTopDevicesListItem. Variables are only populated by the server, and will be ignored when sending a request. - :param value: - :type value: list[~azure.mgmt.security.models.JitNetworkAccessPolicy] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + :ivar device_id: Name of the device. + :vartype device_id: str + :ivar alerts_count: Number of alerts raised for this device. + :vartype alerts_count: long + :ivar last_occurrence: Most recent time this alert was raised for this device, on this day. + :vartype last_occurrence: str """ _validation = { - 'next_link': {'readonly': True}, + 'device_id': {'readonly': True}, + 'alerts_count': {'readonly': True}, + 'last_occurrence': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[JitNetworkAccessPolicy]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'device_id': {'key': 'deviceId', 'type': 'str'}, + 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, + 'last_occurrence': {'key': 'lastOccurrence', 'type': 'str'}, } def __init__( self, - *, - value: Optional[List["JitNetworkAccessPolicy"]] = None, **kwargs ): - super(JitNetworkAccessPoliciesList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + """ + super(IoTSecurityAggregatedAlertPropertiesTopDevicesListItem, self).__init__(**kwargs) + self.device_id = None + self.alerts_count = None + self.last_occurrence = None -class JitNetworkAccessPolicy(Resource, Kind, Location): - """JitNetworkAccessPolicy. +class IoTSecurityAggregatedRecommendation(Resource, TagsResource): + """IoT Security solution recommendation information. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :ivar location: Location where the resource is stored. - :vartype location: str - :param kind: Kind of the resource. - :type kind: str + :ivar tags: A set of tags. Resource tags. + :vartype tags: dict[str, str] :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. :vartype name: str :ivar type: Resource type. :vartype type: str - :param virtual_machines: Required. Configurations for Microsoft.Compute/virtualMachines - resource type. - :type virtual_machines: list[~azure.mgmt.security.models.JitNetworkAccessPolicyVirtualMachine] - :param requests: - :type requests: list[~azure.mgmt.security.models.JitNetworkAccessRequest] - :ivar provisioning_state: Gets the provisioning state of the Just-in-Time policy. - :vartype provisioning_state: str + :ivar recommendation_name: Name of the recommendation. + :vartype recommendation_name: str + :ivar recommendation_display_name: Display name of the recommendation type. + :vartype recommendation_display_name: str + :ivar description: Description of the suspected vulnerability and meaning. + :vartype description: str + :ivar recommendation_type_id: Recommendation-type GUID. + :vartype recommendation_type_id: str + :ivar detected_by: Name of the organization that made the recommendation. + :vartype detected_by: str + :ivar remediation_steps: Recommended steps for remediation. + :vartype remediation_steps: str + :ivar reported_severity: Assessed recommendation severity. Possible values include: + "Informational", "Low", "Medium", "High". + :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity + :ivar healthy_devices: Number of healthy devices within the IoT Security solution. + :vartype healthy_devices: long + :ivar unhealthy_device_count: Number of unhealthy devices within the IoT Security solution. + :vartype unhealthy_device_count: long + :ivar log_analytics_query: Log analytics query for getting the list of affected devices/alerts. + :vartype log_analytics_query: str """ _validation = { - 'location': {'readonly': True}, 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'virtual_machines': {'required': True}, - 'provisioning_state': {'readonly': True}, + 'recommendation_display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'recommendation_type_id': {'readonly': True}, + 'detected_by': {'readonly': True}, + 'remediation_steps': {'readonly': True}, + 'reported_severity': {'readonly': True}, + 'healthy_devices': {'readonly': True}, + 'unhealthy_device_count': {'readonly': True}, + 'log_analytics_query': {'readonly': True}, } _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'kind': {'key': 'kind', 'type': 'str'}, + 'tags': {'key': 'tags', 'type': '{str}'}, 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'virtual_machines': {'key': 'properties.virtualMachines', 'type': '[JitNetworkAccessPolicyVirtualMachine]'}, - 'requests': {'key': 'properties.requests', 'type': '[JitNetworkAccessRequest]'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, + 'recommendation_name': {'key': 'properties.recommendationName', 'type': 'str'}, + 'recommendation_display_name': {'key': 'properties.recommendationDisplayName', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'recommendation_type_id': {'key': 'properties.recommendationTypeId', 'type': 'str'}, + 'detected_by': {'key': 'properties.detectedBy', 'type': 'str'}, + 'remediation_steps': {'key': 'properties.remediationSteps', 'type': 'str'}, + 'reported_severity': {'key': 'properties.reportedSeverity', 'type': 'str'}, + 'healthy_devices': {'key': 'properties.healthyDevices', 'type': 'long'}, + 'unhealthy_device_count': {'key': 'properties.unhealthyDeviceCount', 'type': 'long'}, + 'log_analytics_query': {'key': 'properties.logAnalyticsQuery', 'type': 'str'}, } def __init__( self, *, - virtual_machines: List["JitNetworkAccessPolicyVirtualMachine"], - kind: Optional[str] = None, - requests: Optional[List["JitNetworkAccessRequest"]] = None, + tags: Optional[Dict[str, str]] = None, + recommendation_name: Optional[str] = None, **kwargs ): - super(JitNetworkAccessPolicy, self).__init__(kind=kind, **kwargs) - self.location = None - self.kind = kind - self.virtual_machines = virtual_machines - self.requests = requests - self.provisioning_state = None - self.location = None - self.id = None - self.name = None - self.type = None - self.virtual_machines = virtual_machines - self.requests = requests - self.provisioning_state = None - self.kind = kind + """ + :keyword tags: A set of tags. Resource tags. + :paramtype tags: dict[str, str] + :keyword recommendation_name: Name of the recommendation. + :paramtype recommendation_name: str + """ + super(IoTSecurityAggregatedRecommendation, self).__init__(tags=tags, **kwargs) + self.tags = tags + self.recommendation_name = recommendation_name + self.recommendation_display_name = None + self.description = None + self.recommendation_type_id = None + self.detected_by = None + self.remediation_steps = None + self.reported_severity = None + self.healthy_devices = None + self.unhealthy_device_count = None + self.log_analytics_query = None self.id = None self.name = None self.type = None - self.virtual_machines = virtual_machines - self.requests = requests - self.provisioning_state = None + self.recommendation_name = recommendation_name + self.recommendation_display_name = None + self.description = None + self.recommendation_type_id = None + self.detected_by = None + self.remediation_steps = None + self.reported_severity = None + self.healthy_devices = None + self.unhealthy_device_count = None + self.log_analytics_query = None -class JitNetworkAccessPolicyInitiatePort(msrest.serialization.Model): - """JitNetworkAccessPolicyInitiatePort. +class IoTSecurityAggregatedRecommendationList(msrest.serialization.Model): + """List of IoT Security solution aggregated recommendations. + + Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :param number: Required. - :type number: int - :param allowed_source_address_prefix: Source of the allowed traffic. If omitted, the request - will be for the source IP address of the initiate request. - :type allowed_source_address_prefix: str - :param end_time_utc: Required. The time to close the request in UTC. - :type end_time_utc: ~datetime.datetime + :ivar value: Required. List of aggregated recommendations data. + :vartype value: list[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendation] + :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next + page. + :vartype next_link: str """ _validation = { - 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, - 'end_time_utc': {'required': True}, + 'value': {'required': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'number': {'key': 'number', 'type': 'int'}, - 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, - 'end_time_utc': {'key': 'endTimeUtc', 'type': 'iso-8601'}, + 'value': {'key': 'value', 'type': '[IoTSecurityAggregatedRecommendation]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - number: int, - end_time_utc: datetime.datetime, - allowed_source_address_prefix: Optional[str] = None, + value: List["IoTSecurityAggregatedRecommendation"], **kwargs ): - super(JitNetworkAccessPolicyInitiatePort, self).__init__(**kwargs) - self.number = number - self.allowed_source_address_prefix = allowed_source_address_prefix - self.end_time_utc = end_time_utc + """ + :keyword value: Required. List of aggregated recommendations data. + :paramtype value: list[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendation] + """ + super(IoTSecurityAggregatedRecommendationList, self).__init__(**kwargs) + self.value = value + self.next_link = None -class JitNetworkAccessPolicyInitiateRequest(msrest.serialization.Model): - """JitNetworkAccessPolicyInitiateRequest. +class IoTSecurityAlertedDevice(msrest.serialization.Model): + """Statistical information about the number of alerts per device during last set number of days. - All required parameters must be populated in order to send to Azure. + Variables are only populated by the server, and will be ignored when sending a request. - :param virtual_machines: Required. A list of virtual machines & ports to open access for. - :type virtual_machines: - list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiateVirtualMachine] - :param justification: The justification for making the initiate request. - :type justification: str + :ivar device_id: Device identifier. + :vartype device_id: str + :ivar alerts_count: Number of alerts raised for this device. + :vartype alerts_count: long """ _validation = { - 'virtual_machines': {'required': True}, + 'device_id': {'readonly': True}, + 'alerts_count': {'readonly': True}, } _attribute_map = { - 'virtual_machines': {'key': 'virtualMachines', 'type': '[JitNetworkAccessPolicyInitiateVirtualMachine]'}, - 'justification': {'key': 'justification', 'type': 'str'}, + 'device_id': {'key': 'deviceId', 'type': 'str'}, + 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, } def __init__( self, - *, - virtual_machines: List["JitNetworkAccessPolicyInitiateVirtualMachine"], - justification: Optional[str] = None, **kwargs ): - super(JitNetworkAccessPolicyInitiateRequest, self).__init__(**kwargs) - self.virtual_machines = virtual_machines - self.justification = justification - + """ + """ + super(IoTSecurityAlertedDevice, self).__init__(**kwargs) + self.device_id = None + self.alerts_count = None -class JitNetworkAccessPolicyInitiateVirtualMachine(msrest.serialization.Model): - """JitNetworkAccessPolicyInitiateVirtualMachine. - All required parameters must be populated in order to send to Azure. +class IoTSecurityDeviceAlert(msrest.serialization.Model): + """Statistical information about the number of alerts per alert type during last set number of days. - :param id: Required. Resource ID of the virtual machine that is linked to this policy. - :type id: str - :param ports: Required. The ports to open for the resource with the ``id``. - :type ports: list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiatePort] - """ + Variables are only populated by the server, and will be ignored when sending a request. - _validation = { - 'id': {'required': True}, - 'ports': {'required': True}, - } + :ivar alert_display_name: Display name of the alert. + :vartype alert_display_name: str + :ivar reported_severity: Assessed Alert severity. Possible values include: "Informational", + "Low", "Medium", "High". + :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity + :ivar alerts_count: Number of alerts raised for this alert type. + :vartype alerts_count: long + """ + + _validation = { + 'alert_display_name': {'readonly': True}, + 'reported_severity': {'readonly': True}, + 'alerts_count': {'readonly': True}, + } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'ports': {'key': 'ports', 'type': '[JitNetworkAccessPolicyInitiatePort]'}, + 'alert_display_name': {'key': 'alertDisplayName', 'type': 'str'}, + 'reported_severity': {'key': 'reportedSeverity', 'type': 'str'}, + 'alerts_count': {'key': 'alertsCount', 'type': 'long'}, } def __init__( self, - *, - id: str, - ports: List["JitNetworkAccessPolicyInitiatePort"], **kwargs ): - super(JitNetworkAccessPolicyInitiateVirtualMachine, self).__init__(**kwargs) - self.id = id - self.ports = ports + """ + """ + super(IoTSecurityDeviceAlert, self).__init__(**kwargs) + self.alert_display_name = None + self.reported_severity = None + self.alerts_count = None -class JitNetworkAccessPolicyVirtualMachine(msrest.serialization.Model): - """JitNetworkAccessPolicyVirtualMachine. +class IoTSecurityDeviceRecommendation(msrest.serialization.Model): + """Statistical information about the number of recommendations per device, per recommendation type. - All required parameters must be populated in order to send to Azure. + Variables are only populated by the server, and will be ignored when sending a request. - :param id: Required. Resource ID of the virtual machine that is linked to this policy. - :type id: str - :param ports: Required. Port configurations for the virtual machine. - :type ports: list[~azure.mgmt.security.models.JitNetworkAccessPortRule] - :param public_ip_address: Public IP address of the Azure Firewall that is linked to this - policy, if applicable. - :type public_ip_address: str + :ivar recommendation_display_name: Display name of the recommendation. + :vartype recommendation_display_name: str + :ivar reported_severity: Assessed recommendation severity. Possible values include: + "Informational", "Low", "Medium", "High". + :vartype reported_severity: str or ~azure.mgmt.security.models.ReportedSeverity + :ivar devices_count: Number of devices with this recommendation. + :vartype devices_count: long """ _validation = { - 'id': {'required': True}, - 'ports': {'required': True}, + 'recommendation_display_name': {'readonly': True}, + 'reported_severity': {'readonly': True}, + 'devices_count': {'readonly': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'ports': {'key': 'ports', 'type': '[JitNetworkAccessPortRule]'}, - 'public_ip_address': {'key': 'publicIpAddress', 'type': 'str'}, + 'recommendation_display_name': {'key': 'recommendationDisplayName', 'type': 'str'}, + 'reported_severity': {'key': 'reportedSeverity', 'type': 'str'}, + 'devices_count': {'key': 'devicesCount', 'type': 'long'}, } def __init__( self, - *, - id: str, - ports: List["JitNetworkAccessPortRule"], - public_ip_address: Optional[str] = None, **kwargs ): - super(JitNetworkAccessPolicyVirtualMachine, self).__init__(**kwargs) - self.id = id - self.ports = ports - self.public_ip_address = public_ip_address + """ + """ + super(IoTSecurityDeviceRecommendation, self).__init__(**kwargs) + self.recommendation_display_name = None + self.reported_severity = None + self.devices_count = None -class JitNetworkAccessPortRule(msrest.serialization.Model): - """JitNetworkAccessPortRule. +class IoTSecuritySolutionAnalyticsModel(Resource): + """Security analytics of your IoT Security solution. - All required parameters must be populated in order to send to Azure. + Variables are only populated by the server, and will be ignored when sending a request. - :param number: Required. - :type number: int - :param protocol: Required. Possible values include: "TCP", "UDP", "*". - :type protocol: str or ~azure.mgmt.security.models.ProtocolEnum - :param allowed_source_address_prefix: Mutually exclusive with the - "allowedSourceAddressPrefixes" parameter. Should be an IP address or CIDR, for example - "192.168.0.3" or "192.168.0.0/16". - :type allowed_source_address_prefix: str - :param allowed_source_address_prefixes: Mutually exclusive with the - "allowedSourceAddressPrefix" parameter. - :type allowed_source_address_prefixes: list[str] - :param max_request_access_duration: Required. Maximum duration requests can be made for. In ISO - 8601 duration format. Minimum 5 minutes, maximum 1 day. - :type max_request_access_duration: str + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar metrics: Security analytics of your IoT Security solution. + :vartype metrics: ~azure.mgmt.security.models.IoTSeverityMetrics + :ivar unhealthy_device_count: Number of unhealthy devices within your IoT Security solution. + :vartype unhealthy_device_count: long + :ivar devices_metrics: List of device metrics by the aggregation date. + :vartype devices_metrics: + list[~azure.mgmt.security.models.IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem] + :ivar top_alerted_devices: List of the 3 devices with the most alerts. + :vartype top_alerted_devices: list[~azure.mgmt.security.models.IoTSecurityAlertedDevice] + :ivar most_prevalent_device_alerts: List of the 3 most prevalent device alerts. + :vartype most_prevalent_device_alerts: list[~azure.mgmt.security.models.IoTSecurityDeviceAlert] + :ivar most_prevalent_device_recommendations: List of the 3 most prevalent device + recommendations. + :vartype most_prevalent_device_recommendations: + list[~azure.mgmt.security.models.IoTSecurityDeviceRecommendation] """ _validation = { - 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, - 'protocol': {'required': True}, - 'max_request_access_duration': {'required': True}, + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'metrics': {'readonly': True}, + 'unhealthy_device_count': {'readonly': True}, + 'devices_metrics': {'readonly': True}, } _attribute_map = { - 'number': {'key': 'number', 'type': 'int'}, - 'protocol': {'key': 'protocol', 'type': 'str'}, - 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, - 'allowed_source_address_prefixes': {'key': 'allowedSourceAddressPrefixes', 'type': '[str]'}, - 'max_request_access_duration': {'key': 'maxRequestAccessDuration', 'type': 'str'}, + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'metrics': {'key': 'properties.metrics', 'type': 'IoTSeverityMetrics'}, + 'unhealthy_device_count': {'key': 'properties.unhealthyDeviceCount', 'type': 'long'}, + 'devices_metrics': {'key': 'properties.devicesMetrics', 'type': '[IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem]'}, + 'top_alerted_devices': {'key': 'properties.topAlertedDevices', 'type': '[IoTSecurityAlertedDevice]'}, + 'most_prevalent_device_alerts': {'key': 'properties.mostPrevalentDeviceAlerts', 'type': '[IoTSecurityDeviceAlert]'}, + 'most_prevalent_device_recommendations': {'key': 'properties.mostPrevalentDeviceRecommendations', 'type': '[IoTSecurityDeviceRecommendation]'}, } def __init__( self, *, - number: int, - protocol: Union[str, "ProtocolEnum"], - max_request_access_duration: str, - allowed_source_address_prefix: Optional[str] = None, - allowed_source_address_prefixes: Optional[List[str]] = None, + top_alerted_devices: Optional[List["IoTSecurityAlertedDevice"]] = None, + most_prevalent_device_alerts: Optional[List["IoTSecurityDeviceAlert"]] = None, + most_prevalent_device_recommendations: Optional[List["IoTSecurityDeviceRecommendation"]] = None, **kwargs ): - super(JitNetworkAccessPortRule, self).__init__(**kwargs) - self.number = number - self.protocol = protocol - self.allowed_source_address_prefix = allowed_source_address_prefix - self.allowed_source_address_prefixes = allowed_source_address_prefixes - self.max_request_access_duration = max_request_access_duration + """ + :keyword top_alerted_devices: List of the 3 devices with the most alerts. + :paramtype top_alerted_devices: list[~azure.mgmt.security.models.IoTSecurityAlertedDevice] + :keyword most_prevalent_device_alerts: List of the 3 most prevalent device alerts. + :paramtype most_prevalent_device_alerts: + list[~azure.mgmt.security.models.IoTSecurityDeviceAlert] + :keyword most_prevalent_device_recommendations: List of the 3 most prevalent device + recommendations. + :paramtype most_prevalent_device_recommendations: + list[~azure.mgmt.security.models.IoTSecurityDeviceRecommendation] + """ + super(IoTSecuritySolutionAnalyticsModel, self).__init__(**kwargs) + self.metrics = None + self.unhealthy_device_count = None + self.devices_metrics = None + self.top_alerted_devices = top_alerted_devices + self.most_prevalent_device_alerts = most_prevalent_device_alerts + self.most_prevalent_device_recommendations = most_prevalent_device_recommendations -class JitNetworkAccessRequest(msrest.serialization.Model): - """JitNetworkAccessRequest. +class IoTSecuritySolutionAnalyticsModelList(msrest.serialization.Model): + """List of Security analytics of your IoT Security solution. + + Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :param virtual_machines: Required. - :type virtual_machines: list[~azure.mgmt.security.models.JitNetworkAccessRequestVirtualMachine] - :param start_time_utc: Required. The start time of the request in UTC. - :type start_time_utc: ~datetime.datetime - :param requestor: Required. The identity of the person who made the request. - :type requestor: str - :param justification: The justification for making the initiate request. - :type justification: str + :ivar value: Required. List of Security analytics of your IoT Security solution. + :vartype value: list[~azure.mgmt.security.models.IoTSecuritySolutionAnalyticsModel] + :ivar next_link: When there is too much alert data for one page, use this URI to fetch the next + page. + :vartype next_link: str """ _validation = { - 'virtual_machines': {'required': True}, - 'start_time_utc': {'required': True}, - 'requestor': {'required': True}, + 'value': {'required': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'virtual_machines': {'key': 'virtualMachines', 'type': '[JitNetworkAccessRequestVirtualMachine]'}, - 'start_time_utc': {'key': 'startTimeUtc', 'type': 'iso-8601'}, - 'requestor': {'key': 'requestor', 'type': 'str'}, - 'justification': {'key': 'justification', 'type': 'str'}, + 'value': {'key': 'value', 'type': '[IoTSecuritySolutionAnalyticsModel]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - virtual_machines: List["JitNetworkAccessRequestVirtualMachine"], - start_time_utc: datetime.datetime, - requestor: str, - justification: Optional[str] = None, + value: List["IoTSecuritySolutionAnalyticsModel"], **kwargs ): - super(JitNetworkAccessRequest, self).__init__(**kwargs) - self.virtual_machines = virtual_machines - self.start_time_utc = start_time_utc - self.requestor = requestor - self.justification = justification - + """ + :keyword value: Required. List of Security analytics of your IoT Security solution. + :paramtype value: list[~azure.mgmt.security.models.IoTSecuritySolutionAnalyticsModel] + """ + super(IoTSecuritySolutionAnalyticsModelList, self).__init__(**kwargs) + self.value = value + self.next_link = None -class JitNetworkAccessRequestPort(msrest.serialization.Model): - """JitNetworkAccessRequestPort. - All required parameters must be populated in order to send to Azure. +class IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem(msrest.serialization.Model): + """IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem. - :param number: Required. - :type number: int - :param allowed_source_address_prefix: Mutually exclusive with the - "allowedSourceAddressPrefixes" parameter. Should be an IP address or CIDR, for example - "192.168.0.3" or "192.168.0.0/16". - :type allowed_source_address_prefix: str - :param allowed_source_address_prefixes: Mutually exclusive with the - "allowedSourceAddressPrefix" parameter. - :type allowed_source_address_prefixes: list[str] - :param end_time_utc: Required. The date & time at which the request ends in UTC. - :type end_time_utc: ~datetime.datetime - :param status: Required. The status of the port. Possible values include: "Revoked", - "Initiated". - :type status: str or ~azure.mgmt.security.models.Status - :param status_reason: Required. A description of why the ``status`` has its value. Possible - values include: "Expired", "UserRequested", "NewerRequestInitiated". - :type status_reason: str or ~azure.mgmt.security.models.StatusReason - :param mapped_port: The port which is mapped to this port's ``number`` in the Azure Firewall, - if applicable. - :type mapped_port: int + :ivar date: Aggregation of IoT Security solution device alert metrics by date. + :vartype date: ~datetime.datetime + :ivar devices_metrics: Device alert count by severity. + :vartype devices_metrics: ~azure.mgmt.security.models.IoTSeverityMetrics """ - _validation = { - 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, - 'end_time_utc': {'required': True}, - 'status': {'required': True}, - 'status_reason': {'required': True}, - } - _attribute_map = { - 'number': {'key': 'number', 'type': 'int'}, - 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, - 'allowed_source_address_prefixes': {'key': 'allowedSourceAddressPrefixes', 'type': '[str]'}, - 'end_time_utc': {'key': 'endTimeUtc', 'type': 'iso-8601'}, - 'status': {'key': 'status', 'type': 'str'}, - 'status_reason': {'key': 'statusReason', 'type': 'str'}, - 'mapped_port': {'key': 'mappedPort', 'type': 'int'}, + 'date': {'key': 'date', 'type': 'iso-8601'}, + 'devices_metrics': {'key': 'devicesMetrics', 'type': 'IoTSeverityMetrics'}, } def __init__( self, *, - number: int, - end_time_utc: datetime.datetime, - status: Union[str, "Status"], - status_reason: Union[str, "StatusReason"], - allowed_source_address_prefix: Optional[str] = None, - allowed_source_address_prefixes: Optional[List[str]] = None, - mapped_port: Optional[int] = None, + date: Optional[datetime.datetime] = None, + devices_metrics: Optional["IoTSeverityMetrics"] = None, **kwargs ): - super(JitNetworkAccessRequestPort, self).__init__(**kwargs) - self.number = number - self.allowed_source_address_prefix = allowed_source_address_prefix - self.allowed_source_address_prefixes = allowed_source_address_prefixes - self.end_time_utc = end_time_utc - self.status = status - self.status_reason = status_reason - self.mapped_port = mapped_port + """ + :keyword date: Aggregation of IoT Security solution device alert metrics by date. + :paramtype date: ~datetime.datetime + :keyword devices_metrics: Device alert count by severity. + :paramtype devices_metrics: ~azure.mgmt.security.models.IoTSeverityMetrics + """ + super(IoTSecuritySolutionAnalyticsModelPropertiesDevicesMetricsItem, self).__init__(**kwargs) + self.date = date + self.devices_metrics = devices_metrics -class JitNetworkAccessRequestVirtualMachine(msrest.serialization.Model): - """JitNetworkAccessRequestVirtualMachine. +class IoTSecuritySolutionModel(Resource, TagsResource): + """IoT Security solution configuration and resource information. - All required parameters must be populated in order to send to Azure. + Variables are only populated by the server, and will be ignored when sending a request. - :param id: Required. Resource ID of the virtual machine that is linked to this policy. - :type id: str - :param ports: Required. The ports that were opened for the virtual machine. - :type ports: list[~azure.mgmt.security.models.JitNetworkAccessRequestPort] + :ivar tags: A set of tags. Resource tags. + :vartype tags: dict[str, str] + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar location: The resource location. + :vartype location: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.security.models.SystemData + :ivar workspace: Workspace resource ID. + :vartype workspace: str + :ivar display_name: Resource display name. + :vartype display_name: str + :ivar status: Status of the IoT Security solution. Possible values include: "Enabled", + "Disabled". Default value: "Enabled". + :vartype status: str or ~azure.mgmt.security.models.SecuritySolutionStatus + :ivar export: List of additional options for exporting to workspace data. + :vartype export: list[str or ~azure.mgmt.security.models.ExportData] + :ivar disabled_data_sources: Disabled data sources. Disabling these data sources compromises + the system. + :vartype disabled_data_sources: list[str or ~azure.mgmt.security.models.DataSource] + :ivar iot_hubs: IoT Hub resource IDs. + :vartype iot_hubs: list[str] + :ivar user_defined_resources: Properties of the IoT Security solution's user defined resources. + :vartype user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties + :ivar auto_discovered_resources: List of resources that were automatically discovered as + relevant to the security solution. + :vartype auto_discovered_resources: list[str] + :ivar recommendations_configuration: List of the configuration status for each recommendation + type. + :vartype recommendations_configuration: + list[~azure.mgmt.security.models.RecommendationConfigurationProperties] + :ivar unmasked_ip_logging_status: Unmasked IP address logging status. Possible values include: + "Disabled", "Enabled". Default value: "Disabled". + :vartype unmasked_ip_logging_status: str or ~azure.mgmt.security.models.UnmaskedIpLoggingStatus + :ivar additional_workspaces: List of additional workspaces. + :vartype additional_workspaces: + list[~azure.mgmt.security.models.AdditionalWorkspacesProperties] """ _validation = { - 'id': {'required': True}, - 'ports': {'required': True}, + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'system_data': {'readonly': True}, + 'auto_discovered_resources': {'readonly': True}, } _attribute_map = { + 'tags': {'key': 'tags', 'type': '{str}'}, 'id': {'key': 'id', 'type': 'str'}, - 'ports': {'key': 'ports', 'type': '[JitNetworkAccessRequestPort]'}, - } - - def __init__( - self, - *, - id: str, - ports: List["JitNetworkAccessRequestPort"], - **kwargs - ): - super(JitNetworkAccessRequestVirtualMachine, self).__init__(**kwargs) - self.id = id - self.ports = ports - - -class LocalUserNotAllowed(AllowlistCustomAlertRule): - """Login by a local user that isn't allowed. Allow list consists of login names to allow. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", - "String". - :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the - rule type. - :type allowlist_values: list[str] - """ - - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'value_type': {'readonly': True}, - 'allowlist_values': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'value_type': {'key': 'valueType', 'type': 'str'}, - 'allowlist_values': {'key': 'allowlistValues', 'type': '[str]'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'location': {'key': 'location', 'type': 'str'}, + 'system_data': {'key': 'systemData', 'type': 'SystemData'}, + 'workspace': {'key': 'properties.workspace', 'type': 'str'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'status': {'key': 'properties.status', 'type': 'str'}, + 'export': {'key': 'properties.export', 'type': '[str]'}, + 'disabled_data_sources': {'key': 'properties.disabledDataSources', 'type': '[str]'}, + 'iot_hubs': {'key': 'properties.iotHubs', 'type': '[str]'}, + 'user_defined_resources': {'key': 'properties.userDefinedResources', 'type': 'UserDefinedResourcesProperties'}, + 'auto_discovered_resources': {'key': 'properties.autoDiscoveredResources', 'type': '[str]'}, + 'recommendations_configuration': {'key': 'properties.recommendationsConfiguration', 'type': '[RecommendationConfigurationProperties]'}, + 'unmasked_ip_logging_status': {'key': 'properties.unmaskedIpLoggingStatus', 'type': 'str'}, + 'additional_workspaces': {'key': 'properties.additionalWorkspaces', 'type': '[AdditionalWorkspacesProperties]'}, } def __init__( self, *, - is_enabled: bool, - allowlist_values: List[str], + tags: Optional[Dict[str, str]] = None, + location: Optional[str] = None, + workspace: Optional[str] = None, + display_name: Optional[str] = None, + status: Optional[Union[str, "SecuritySolutionStatus"]] = "Enabled", + export: Optional[List[Union[str, "ExportData"]]] = None, + disabled_data_sources: Optional[List[Union[str, "DataSource"]]] = None, + iot_hubs: Optional[List[str]] = None, + user_defined_resources: Optional["UserDefinedResourcesProperties"] = None, + recommendations_configuration: Optional[List["RecommendationConfigurationProperties"]] = None, + unmasked_ip_logging_status: Optional[Union[str, "UnmaskedIpLoggingStatus"]] = "Disabled", + additional_workspaces: Optional[List["AdditionalWorkspacesProperties"]] = None, **kwargs ): - super(LocalUserNotAllowed, self).__init__(is_enabled=is_enabled, allowlist_values=allowlist_values, **kwargs) - self.rule_type = 'LocalUserNotAllowed' # type: str + """ + :keyword tags: A set of tags. Resource tags. + :paramtype tags: dict[str, str] + :keyword location: The resource location. + :paramtype location: str + :keyword workspace: Workspace resource ID. + :paramtype workspace: str + :keyword display_name: Resource display name. + :paramtype display_name: str + :keyword status: Status of the IoT Security solution. Possible values include: "Enabled", + "Disabled". Default value: "Enabled". + :paramtype status: str or ~azure.mgmt.security.models.SecuritySolutionStatus + :keyword export: List of additional options for exporting to workspace data. + :paramtype export: list[str or ~azure.mgmt.security.models.ExportData] + :keyword disabled_data_sources: Disabled data sources. Disabling these data sources compromises + the system. + :paramtype disabled_data_sources: list[str or ~azure.mgmt.security.models.DataSource] + :keyword iot_hubs: IoT Hub resource IDs. + :paramtype iot_hubs: list[str] + :keyword user_defined_resources: Properties of the IoT Security solution's user defined + resources. + :paramtype user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties + :keyword recommendations_configuration: List of the configuration status for each + recommendation type. + :paramtype recommendations_configuration: + list[~azure.mgmt.security.models.RecommendationConfigurationProperties] + :keyword unmasked_ip_logging_status: Unmasked IP address logging status. Possible values + include: "Disabled", "Enabled". Default value: "Disabled". + :paramtype unmasked_ip_logging_status: str or + ~azure.mgmt.security.models.UnmaskedIpLoggingStatus + :keyword additional_workspaces: List of additional workspaces. + :paramtype additional_workspaces: + list[~azure.mgmt.security.models.AdditionalWorkspacesProperties] + """ + super(IoTSecuritySolutionModel, self).__init__(tags=tags, **kwargs) + self.tags = tags + self.location = location + self.system_data = None + self.workspace = workspace + self.display_name = display_name + self.status = status + self.export = export + self.disabled_data_sources = disabled_data_sources + self.iot_hubs = iot_hubs + self.user_defined_resources = user_defined_resources + self.auto_discovered_resources = None + self.recommendations_configuration = recommendations_configuration + self.unmasked_ip_logging_status = unmasked_ip_logging_status + self.additional_workspaces = additional_workspaces + self.id = None + self.name = None + self.type = None + self.location = location + self.system_data = None + self.workspace = workspace + self.display_name = display_name + self.status = status + self.export = export + self.disabled_data_sources = disabled_data_sources + self.iot_hubs = iot_hubs + self.user_defined_resources = user_defined_resources + self.auto_discovered_resources = None + self.recommendations_configuration = recommendations_configuration + self.unmasked_ip_logging_status = unmasked_ip_logging_status + self.additional_workspaces = additional_workspaces -class LogAnalyticsIdentifier(ResourceIdentifier): - """Represents a Log Analytics workspace scope identifier. +class IoTSecuritySolutionsList(msrest.serialization.Model): + """List of IoT Security solutions. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :param type: Required. There can be multiple identifiers of different type per alert, this - field specify the identifier type.Constant filled by server. Possible values include: - "AzureResource", "LogAnalytics". - :type type: str or ~azure.mgmt.security.models.ResourceIdentifierType - :ivar workspace_id: The LogAnalytics workspace id that stores this alert. - :vartype workspace_id: str - :ivar workspace_subscription_id: The azure subscription id for the LogAnalytics workspace - storing this alert. - :vartype workspace_subscription_id: str - :ivar workspace_resource_group: The azure resource group for the LogAnalytics workspace storing - this alert. - :vartype workspace_resource_group: str - :ivar agent_id: (optional) The LogAnalytics agent id reporting the event that this alert is - based on. - :vartype agent_id: str + :ivar value: Required. List of IoT Security solutions. + :vartype value: list[~azure.mgmt.security.models.IoTSecuritySolutionModel] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str """ _validation = { - 'type': {'required': True}, - 'workspace_id': {'readonly': True}, - 'workspace_subscription_id': {'readonly': True, 'pattern': r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'}, - 'workspace_resource_group': {'readonly': True}, - 'agent_id': {'readonly': True}, + 'value': {'required': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, - 'workspace_subscription_id': {'key': 'workspaceSubscriptionId', 'type': 'str'}, - 'workspace_resource_group': {'key': 'workspaceResourceGroup', 'type': 'str'}, - 'agent_id': {'key': 'agentId', 'type': 'str'}, + 'value': {'key': 'value', 'type': '[IoTSecuritySolutionModel]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, + *, + value: List["IoTSecuritySolutionModel"], **kwargs ): - super(LogAnalyticsIdentifier, self).__init__(**kwargs) - self.type = 'LogAnalytics' # type: str - self.workspace_id = None - self.workspace_subscription_id = None - self.workspace_resource_group = None - self.agent_id = None - - -class MqttC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of cloud to device messages (MQTT protocol) is not in allowed range. + """ + :keyword value: Required. List of IoT Security solutions. + :paramtype value: list[~azure.mgmt.security.models.IoTSecuritySolutionModel] + """ + super(IoTSecuritySolutionsList, self).__init__(**kwargs) + self.value = value + self.next_link = None - Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. +class IoTSeverityMetrics(msrest.serialization.Model): + """IoT Security solution analytics severity metrics. - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar high: Count of high severity alerts/recommendations. + :vartype high: long + :ivar medium: Count of medium severity alerts/recommendations. + :vartype medium: long + :ivar low: Count of low severity alerts/recommendations. + :vartype low: long """ - _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, - } - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'high': {'key': 'high', 'type': 'long'}, + 'medium': {'key': 'medium', 'type': 'long'}, + 'low': {'key': 'low', 'type': 'long'}, } def __init__( self, *, - is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, + high: Optional[int] = None, + medium: Optional[int] = None, + low: Optional[int] = None, **kwargs ): - super(MqttC2DMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'MqttC2DMessagesNotInAllowedRange' # type: str + """ + :keyword high: Count of high severity alerts/recommendations. + :paramtype high: long + :keyword medium: Count of medium severity alerts/recommendations. + :paramtype medium: long + :keyword low: Count of low severity alerts/recommendations. + :paramtype low: long + """ + super(IoTSeverityMetrics, self).__init__(**kwargs) + self.high = high + self.medium = medium + self.low = low -class MqttC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of rejected cloud to device messages (MQTT protocol) is not in allowed range. +class JitNetworkAccessPoliciesList(msrest.serialization.Model): + """JitNetworkAccessPoliciesList. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar value: + :vartype value: list[~azure.mgmt.security.models.JitNetworkAccessPolicy] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str """ _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'value': {'key': 'value', 'type': '[JitNetworkAccessPolicy]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, + value: Optional[List["JitNetworkAccessPolicy"]] = None, **kwargs ): - super(MqttC2DRejectedMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'MqttC2DRejectedMessagesNotInAllowedRange' # type: str + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.JitNetworkAccessPolicy] + """ + super(JitNetworkAccessPoliciesList, self).__init__(**kwargs) + self.value = value + self.next_link = None -class MqttD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of device to cloud messages (MQTT protocol) is not in allowed range. +class JitNetworkAccessPolicy(Resource, Kind, Location): + """JitNetworkAccessPolicy. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :ivar display_name: The display name of the custom alert. - :vartype display_name: str - :ivar description: The description of the custom alert. - :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar location: Location where the resource is stored. + :vartype location: str + :ivar kind: Kind of the resource. + :vartype kind: str + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar virtual_machines: Required. Configurations for Microsoft.Compute/virtualMachines resource + type. + :vartype virtual_machines: + list[~azure.mgmt.security.models.JitNetworkAccessPolicyVirtualMachine] + :ivar requests: + :vartype requests: list[~azure.mgmt.security.models.JitNetworkAccessRequest] + :ivar provisioning_state: Gets the provisioning state of the Just-in-Time policy. + :vartype provisioning_state: str """ _validation = { - 'display_name': {'readonly': True}, - 'description': {'readonly': True}, - 'is_enabled': {'required': True}, - 'rule_type': {'required': True}, - 'min_threshold': {'required': True}, - 'max_threshold': {'required': True}, - 'time_window_size': {'required': True}, + 'location': {'readonly': True}, + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'virtual_machines': {'required': True}, + 'provisioning_state': {'readonly': True}, } _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, - 'rule_type': {'key': 'ruleType', 'type': 'str'}, - 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, - 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, - 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, + 'location': {'key': 'location', 'type': 'str'}, + 'kind': {'key': 'kind', 'type': 'str'}, + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'virtual_machines': {'key': 'properties.virtualMachines', 'type': '[JitNetworkAccessPolicyVirtualMachine]'}, + 'requests': {'key': 'properties.requests', 'type': '[JitNetworkAccessRequest]'}, + 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, } def __init__( self, *, - is_enabled: bool, - min_threshold: int, - max_threshold: int, - time_window_size: datetime.timedelta, + virtual_machines: List["JitNetworkAccessPolicyVirtualMachine"], + kind: Optional[str] = None, + requests: Optional[List["JitNetworkAccessRequest"]] = None, **kwargs ): - super(MqttD2CMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'MqttD2CMessagesNotInAllowedRange' # type: str - + """ + :keyword kind: Kind of the resource. + :paramtype kind: str + :keyword virtual_machines: Required. Configurations for Microsoft.Compute/virtualMachines + resource type. + :paramtype virtual_machines: + list[~azure.mgmt.security.models.JitNetworkAccessPolicyVirtualMachine] + :keyword requests: + :paramtype requests: list[~azure.mgmt.security.models.JitNetworkAccessRequest] + """ + super(JitNetworkAccessPolicy, self).__init__(kind=kind, **kwargs) + self.location = None + self.kind = kind + self.virtual_machines = virtual_machines + self.requests = requests + self.provisioning_state = None + self.location = None + self.id = None + self.name = None + self.type = None + self.virtual_machines = virtual_machines + self.requests = requests + self.provisioning_state = None + self.kind = kind + self.id = None + self.name = None + self.type = None + self.virtual_machines = virtual_machines + self.requests = requests + self.provisioning_state = None -class OnPremiseResourceDetails(ResourceDetails): - """Details of the On Premise resource that was assessed. - You probably want to use the sub-classes and not this class directly. Known - sub-classes are: OnPremiseSqlResourceDetails. +class JitNetworkAccessPolicyInitiatePort(msrest.serialization.Model): + """JitNetworkAccessPolicyInitiatePort. All required parameters must be populated in order to send to Azure. - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source - :param workspace_id: Required. Azure resource Id of the workspace the machine is attached to. - :type workspace_id: str - :param vmuuid: Required. The unique Id of the machine. - :type vmuuid: str - :param source_computer_id: Required. The oms agent Id installed on the machine. - :type source_computer_id: str - :param machine_name: Required. The name of the machine. - :type machine_name: str + :ivar number: Required. + :vartype number: int + :ivar allowed_source_address_prefix: Source of the allowed traffic. If omitted, the request + will be for the source IP address of the initiate request. + :vartype allowed_source_address_prefix: str + :ivar end_time_utc: Required. The time to close the request in UTC. + :vartype end_time_utc: ~datetime.datetime """ _validation = { - 'source': {'required': True}, - 'workspace_id': {'required': True}, - 'vmuuid': {'required': True}, - 'source_computer_id': {'required': True}, - 'machine_name': {'required': True}, + 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, + 'end_time_utc': {'required': True}, } _attribute_map = { - 'source': {'key': 'source', 'type': 'str'}, - 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, - 'vmuuid': {'key': 'vmuuid', 'type': 'str'}, - 'source_computer_id': {'key': 'sourceComputerId', 'type': 'str'}, - 'machine_name': {'key': 'machineName', 'type': 'str'}, - } - - _subtype_map = { - 'source': {'OnPremiseSql': 'OnPremiseSqlResourceDetails'} + 'number': {'key': 'number', 'type': 'int'}, + 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, + 'end_time_utc': {'key': 'endTimeUtc', 'type': 'iso-8601'}, } def __init__( self, *, - workspace_id: str, - vmuuid: str, - source_computer_id: str, - machine_name: str, + number: int, + end_time_utc: datetime.datetime, + allowed_source_address_prefix: Optional[str] = None, **kwargs ): - super(OnPremiseResourceDetails, self).__init__(**kwargs) - self.source = 'OnPremise' # type: str - self.workspace_id = workspace_id - self.vmuuid = vmuuid - self.source_computer_id = source_computer_id - self.machine_name = machine_name + """ + :keyword number: Required. + :paramtype number: int + :keyword allowed_source_address_prefix: Source of the allowed traffic. If omitted, the request + will be for the source IP address of the initiate request. + :paramtype allowed_source_address_prefix: str + :keyword end_time_utc: Required. The time to close the request in UTC. + :paramtype end_time_utc: ~datetime.datetime + """ + super(JitNetworkAccessPolicyInitiatePort, self).__init__(**kwargs) + self.number = number + self.allowed_source_address_prefix = allowed_source_address_prefix + self.end_time_utc = end_time_utc -class OnPremiseSqlResourceDetails(OnPremiseResourceDetails): - """Details of the On Premise Sql resource that was assessed. +class JitNetworkAccessPolicyInitiateRequest(msrest.serialization.Model): + """JitNetworkAccessPolicyInitiateRequest. All required parameters must be populated in order to send to Azure. - :param source: Required. The platform where the assessed resource resides.Constant filled by - server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". - :type source: str or ~azure.mgmt.security.models.Source - :param workspace_id: Required. Azure resource Id of the workspace the machine is attached to. - :type workspace_id: str - :param vmuuid: Required. The unique Id of the machine. - :type vmuuid: str - :param source_computer_id: Required. The oms agent Id installed on the machine. - :type source_computer_id: str - :param machine_name: Required. The name of the machine. - :type machine_name: str - :param server_name: Required. The Sql server name installed on the machine. - :type server_name: str - :param database_name: Required. The Sql database name installed on the machine. - :type database_name: str + :ivar virtual_machines: Required. A list of virtual machines & ports to open access for. + :vartype virtual_machines: + list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiateVirtualMachine] + :ivar justification: The justification for making the initiate request. + :vartype justification: str """ _validation = { - 'source': {'required': True}, - 'workspace_id': {'required': True}, - 'vmuuid': {'required': True}, - 'source_computer_id': {'required': True}, - 'machine_name': {'required': True}, - 'server_name': {'required': True}, - 'database_name': {'required': True}, + 'virtual_machines': {'required': True}, } _attribute_map = { - 'source': {'key': 'source', 'type': 'str'}, - 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, - 'vmuuid': {'key': 'vmuuid', 'type': 'str'}, - 'source_computer_id': {'key': 'sourceComputerId', 'type': 'str'}, - 'machine_name': {'key': 'machineName', 'type': 'str'}, - 'server_name': {'key': 'serverName', 'type': 'str'}, - 'database_name': {'key': 'databaseName', 'type': 'str'}, + 'virtual_machines': {'key': 'virtualMachines', 'type': '[JitNetworkAccessPolicyInitiateVirtualMachine]'}, + 'justification': {'key': 'justification', 'type': 'str'}, } def __init__( self, *, - workspace_id: str, - vmuuid: str, - source_computer_id: str, - machine_name: str, - server_name: str, - database_name: str, + virtual_machines: List["JitNetworkAccessPolicyInitiateVirtualMachine"], + justification: Optional[str] = None, **kwargs ): - super(OnPremiseSqlResourceDetails, self).__init__(workspace_id=workspace_id, vmuuid=vmuuid, source_computer_id=source_computer_id, machine_name=machine_name, **kwargs) - self.source = 'OnPremiseSql' # type: str - self.server_name = server_name - self.database_name = database_name + """ + :keyword virtual_machines: Required. A list of virtual machines & ports to open access for. + :paramtype virtual_machines: + list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiateVirtualMachine] + :keyword justification: The justification for making the initiate request. + :paramtype justification: str + """ + super(JitNetworkAccessPolicyInitiateRequest, self).__init__(**kwargs) + self.virtual_machines = virtual_machines + self.justification = justification -class Operation(msrest.serialization.Model): - """Possible operation in the REST API of Microsoft.Security. +class JitNetworkAccessPolicyInitiateVirtualMachine(msrest.serialization.Model): + """JitNetworkAccessPolicyInitiateVirtualMachine. - Variables are only populated by the server, and will be ignored when sending a request. + All required parameters must be populated in order to send to Azure. - :ivar name: Name of the operation. - :vartype name: str - :ivar origin: Where the operation is originated. - :vartype origin: str - :param display: Security operation display. - :type display: ~azure.mgmt.security.models.OperationDisplay + :ivar id: Required. Resource ID of the virtual machine that is linked to this policy. + :vartype id: str + :ivar ports: Required. The ports to open for the resource with the ``id``. + :vartype ports: list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiatePort] """ _validation = { - 'name': {'readonly': True}, - 'origin': {'readonly': True}, + 'id': {'required': True}, + 'ports': {'required': True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'origin': {'key': 'origin', 'type': 'str'}, - 'display': {'key': 'display', 'type': 'OperationDisplay'}, + 'id': {'key': 'id', 'type': 'str'}, + 'ports': {'key': 'ports', 'type': '[JitNetworkAccessPolicyInitiatePort]'}, } def __init__( self, *, - display: Optional["OperationDisplay"] = None, + id: str, + ports: List["JitNetworkAccessPolicyInitiatePort"], **kwargs ): - super(Operation, self).__init__(**kwargs) - self.name = None - self.origin = None - self.display = display + """ + :keyword id: Required. Resource ID of the virtual machine that is linked to this policy. + :paramtype id: str + :keyword ports: Required. The ports to open for the resource with the ``id``. + :paramtype ports: list[~azure.mgmt.security.models.JitNetworkAccessPolicyInitiatePort] + """ + super(JitNetworkAccessPolicyInitiateVirtualMachine, self).__init__(**kwargs) + self.id = id + self.ports = ports -class OperationDisplay(msrest.serialization.Model): - """Security operation display. +class JitNetworkAccessPolicyVirtualMachine(msrest.serialization.Model): + """JitNetworkAccessPolicyVirtualMachine. - Variables are only populated by the server, and will be ignored when sending a request. + All required parameters must be populated in order to send to Azure. - :ivar provider: The resource provider for the operation. - :vartype provider: str - :ivar resource: The display name of the resource the operation applies to. - :vartype resource: str - :ivar operation: The display name of the security operation. - :vartype operation: str - :ivar description: The description of the operation. - :vartype description: str + :ivar id: Required. Resource ID of the virtual machine that is linked to this policy. + :vartype id: str + :ivar ports: Required. Port configurations for the virtual machine. + :vartype ports: list[~azure.mgmt.security.models.JitNetworkAccessPortRule] + :ivar public_ip_address: Public IP address of the Azure Firewall that is linked to this policy, + if applicable. + :vartype public_ip_address: str """ _validation = { - 'provider': {'readonly': True}, - 'resource': {'readonly': True}, - 'operation': {'readonly': True}, - 'description': {'readonly': True}, + 'id': {'required': True}, + 'ports': {'required': True}, } _attribute_map = { - 'provider': {'key': 'provider', 'type': 'str'}, - 'resource': {'key': 'resource', 'type': 'str'}, - 'operation': {'key': 'operation', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, + 'id': {'key': 'id', 'type': 'str'}, + 'ports': {'key': 'ports', 'type': '[JitNetworkAccessPortRule]'}, + 'public_ip_address': {'key': 'publicIpAddress', 'type': 'str'}, } def __init__( self, + *, + id: str, + ports: List["JitNetworkAccessPortRule"], + public_ip_address: Optional[str] = None, **kwargs ): - super(OperationDisplay, self).__init__(**kwargs) - self.provider = None - self.resource = None - self.operation = None - self.description = None + """ + :keyword id: Required. Resource ID of the virtual machine that is linked to this policy. + :paramtype id: str + :keyword ports: Required. Port configurations for the virtual machine. + :paramtype ports: list[~azure.mgmt.security.models.JitNetworkAccessPortRule] + :keyword public_ip_address: Public IP address of the Azure Firewall that is linked to this + policy, if applicable. + :paramtype public_ip_address: str + """ + super(JitNetworkAccessPolicyVirtualMachine, self).__init__(**kwargs) + self.id = id + self.ports = ports + self.public_ip_address = public_ip_address -class OperationList(msrest.serialization.Model): - """List of possible operations for Microsoft.Security resource provider. +class JitNetworkAccessPortRule(msrest.serialization.Model): + """JitNetworkAccessPortRule. - Variables are only populated by the server, and will be ignored when sending a request. + All required parameters must be populated in order to send to Azure. - :param value: List of Security operations. - :type value: list[~azure.mgmt.security.models.Operation] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + :ivar number: Required. + :vartype number: int + :ivar protocol: Required. Possible values include: "TCP", "UDP", "*". + :vartype protocol: str or ~azure.mgmt.security.models.ProtocolEnum + :ivar allowed_source_address_prefix: Mutually exclusive with the "allowedSourceAddressPrefixes" + parameter. Should be an IP address or CIDR, for example "192.168.0.3" or "192.168.0.0/16". + :vartype allowed_source_address_prefix: str + :ivar allowed_source_address_prefixes: Mutually exclusive with the "allowedSourceAddressPrefix" + parameter. + :vartype allowed_source_address_prefixes: list[str] + :ivar max_request_access_duration: Required. Maximum duration requests can be made for. In ISO + 8601 duration format. Minimum 5 minutes, maximum 1 day. + :vartype max_request_access_duration: str """ _validation = { - 'next_link': {'readonly': True}, + 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, + 'protocol': {'required': True}, + 'max_request_access_duration': {'required': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[Operation]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'number': {'key': 'number', 'type': 'int'}, + 'protocol': {'key': 'protocol', 'type': 'str'}, + 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, + 'allowed_source_address_prefixes': {'key': 'allowedSourceAddressPrefixes', 'type': '[str]'}, + 'max_request_access_duration': {'key': 'maxRequestAccessDuration', 'type': 'str'}, } def __init__( self, *, - value: Optional[List["Operation"]] = None, + number: int, + protocol: Union[str, "ProtocolEnum"], + max_request_access_duration: str, + allowed_source_address_prefix: Optional[str] = None, + allowed_source_address_prefixes: Optional[List[str]] = None, **kwargs ): - super(OperationList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + :keyword number: Required. + :paramtype number: int + :keyword protocol: Required. Possible values include: "TCP", "UDP", "*". + :paramtype protocol: str or ~azure.mgmt.security.models.ProtocolEnum + :keyword allowed_source_address_prefix: Mutually exclusive with the + "allowedSourceAddressPrefixes" parameter. Should be an IP address or CIDR, for example + "192.168.0.3" or "192.168.0.0/16". + :paramtype allowed_source_address_prefix: str + :keyword allowed_source_address_prefixes: Mutually exclusive with the + "allowedSourceAddressPrefix" parameter. + :paramtype allowed_source_address_prefixes: list[str] + :keyword max_request_access_duration: Required. Maximum duration requests can be made for. In + ISO 8601 duration format. Minimum 5 minutes, maximum 1 day. + :paramtype max_request_access_duration: str + """ + super(JitNetworkAccessPortRule, self).__init__(**kwargs) + self.number = number + self.protocol = protocol + self.allowed_source_address_prefix = allowed_source_address_prefix + self.allowed_source_address_prefixes = allowed_source_address_prefixes + self.max_request_access_duration = max_request_access_duration -class PathRecommendation(msrest.serialization.Model): - """Represents a path that is recommended to be allowed and its properties. +class JitNetworkAccessRequest(msrest.serialization.Model): + """JitNetworkAccessRequest. - :param path: The full path of the file, or an identifier of the application. - :type path: str - :param action: The recommendation action of the machine or rule. Possible values include: - "Recommended", "Add", "Remove". - :type action: str or ~azure.mgmt.security.models.RecommendationAction - :param type: The type of IoT Security recommendation. Possible values include: - "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", - "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", - "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", - "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", - "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", - "IoT_VulnerableTLSCipherSuite". - :type type: str or ~azure.mgmt.security.models.RecommendationType - :param publisher_info: Represents the publisher information of a process/rule. - :type publisher_info: ~azure.mgmt.security.models.PublisherInfo - :param common: Whether the application is commonly run on the machine. - :type common: bool - :param user_sids: - :type user_sids: list[str] - :param usernames: - :type usernames: list[~azure.mgmt.security.models.UserRecommendation] - :param file_type: The type of the file (for Linux files - Executable is used). Possible values - include: "Exe", "Dll", "Msi", "Script", "Executable", "Unknown". - :type file_type: str or ~azure.mgmt.security.models.FileType - :param configuration_status: The configuration status of the machines group or machine or rule. - Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", "NoStatus". - :type configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus + All required parameters must be populated in order to send to Azure. + + :ivar virtual_machines: Required. + :vartype virtual_machines: + list[~azure.mgmt.security.models.JitNetworkAccessRequestVirtualMachine] + :ivar start_time_utc: Required. The start time of the request in UTC. + :vartype start_time_utc: ~datetime.datetime + :ivar requestor: Required. The identity of the person who made the request. + :vartype requestor: str + :ivar justification: The justification for making the initiate request. + :vartype justification: str """ + _validation = { + 'virtual_machines': {'required': True}, + 'start_time_utc': {'required': True}, + 'requestor': {'required': True}, + } + _attribute_map = { - 'path': {'key': 'path', 'type': 'str'}, - 'action': {'key': 'action', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'publisher_info': {'key': 'publisherInfo', 'type': 'PublisherInfo'}, - 'common': {'key': 'common', 'type': 'bool'}, - 'user_sids': {'key': 'userSids', 'type': '[str]'}, - 'usernames': {'key': 'usernames', 'type': '[UserRecommendation]'}, - 'file_type': {'key': 'fileType', 'type': 'str'}, - 'configuration_status': {'key': 'configurationStatus', 'type': 'str'}, + 'virtual_machines': {'key': 'virtualMachines', 'type': '[JitNetworkAccessRequestVirtualMachine]'}, + 'start_time_utc': {'key': 'startTimeUtc', 'type': 'iso-8601'}, + 'requestor': {'key': 'requestor', 'type': 'str'}, + 'justification': {'key': 'justification', 'type': 'str'}, } def __init__( self, *, - path: Optional[str] = None, - action: Optional[Union[str, "RecommendationAction"]] = None, - type: Optional[Union[str, "RecommendationType"]] = None, - publisher_info: Optional["PublisherInfo"] = None, - common: Optional[bool] = None, - user_sids: Optional[List[str]] = None, - usernames: Optional[List["UserRecommendation"]] = None, - file_type: Optional[Union[str, "FileType"]] = None, - configuration_status: Optional[Union[str, "ConfigurationStatus"]] = None, + virtual_machines: List["JitNetworkAccessRequestVirtualMachine"], + start_time_utc: datetime.datetime, + requestor: str, + justification: Optional[str] = None, **kwargs ): - super(PathRecommendation, self).__init__(**kwargs) - self.path = path - self.action = action - self.type = type - self.publisher_info = publisher_info - self.common = common - self.user_sids = user_sids - self.usernames = usernames - self.file_type = file_type - self.configuration_status = configuration_status + """ + :keyword virtual_machines: Required. + :paramtype virtual_machines: + list[~azure.mgmt.security.models.JitNetworkAccessRequestVirtualMachine] + :keyword start_time_utc: Required. The start time of the request in UTC. + :paramtype start_time_utc: ~datetime.datetime + :keyword requestor: Required. The identity of the person who made the request. + :paramtype requestor: str + :keyword justification: The justification for making the initiate request. + :paramtype justification: str + """ + super(JitNetworkAccessRequest, self).__init__(**kwargs) + self.virtual_machines = virtual_machines + self.start_time_utc = start_time_utc + self.requestor = requestor + self.justification = justification -class Pricing(Resource): - """Azure Security Center is provided in two pricing tiers: free and standard, with the standard tier available with a trial period. The standard tier offers advanced security capabilities, while the free tier offers basic security features. +class JitNetworkAccessRequestPort(msrest.serialization.Model): + """JitNetworkAccessRequestPort. - Variables are only populated by the server, and will be ignored when sending a request. + All required parameters must be populated in order to send to Azure. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param pricing_tier: The pricing tier value. Azure Security Center is provided in two pricing - tiers: free and standard, with the standard tier available with a trial period. The standard - tier offers advanced security capabilities, while the free tier offers basic security features. - Possible values include: "Free", "Standard". - :type pricing_tier: str or ~azure.mgmt.security.models.PricingTier - :ivar free_trial_remaining_time: The duration left for the subscriptions free trial period - in - ISO 8601 format (e.g. P3Y6M4DT12H30M5S). - :vartype free_trial_remaining_time: ~datetime.timedelta + :ivar number: Required. + :vartype number: int + :ivar allowed_source_address_prefix: Mutually exclusive with the "allowedSourceAddressPrefixes" + parameter. Should be an IP address or CIDR, for example "192.168.0.3" or "192.168.0.0/16". + :vartype allowed_source_address_prefix: str + :ivar allowed_source_address_prefixes: Mutually exclusive with the "allowedSourceAddressPrefix" + parameter. + :vartype allowed_source_address_prefixes: list[str] + :ivar end_time_utc: Required. The date & time at which the request ends in UTC. + :vartype end_time_utc: ~datetime.datetime + :ivar status: Required. The status of the port. Possible values include: "Revoked", + "Initiated". + :vartype status: str or ~azure.mgmt.security.models.Status + :ivar status_reason: Required. A description of why the ``status`` has its value. Possible + values include: "Expired", "UserRequested", "NewerRequestInitiated". + :vartype status_reason: str or ~azure.mgmt.security.models.StatusReason + :ivar mapped_port: The port which is mapped to this port's ``number`` in the Azure Firewall, if + applicable. + :vartype mapped_port: int """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'free_trial_remaining_time': {'readonly': True}, + 'number': {'required': True, 'maximum': 65535, 'minimum': 0}, + 'end_time_utc': {'required': True}, + 'status': {'required': True}, + 'status_reason': {'required': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'pricing_tier': {'key': 'properties.pricingTier', 'type': 'str'}, - 'free_trial_remaining_time': {'key': 'properties.freeTrialRemainingTime', 'type': 'duration'}, + 'number': {'key': 'number', 'type': 'int'}, + 'allowed_source_address_prefix': {'key': 'allowedSourceAddressPrefix', 'type': 'str'}, + 'allowed_source_address_prefixes': {'key': 'allowedSourceAddressPrefixes', 'type': '[str]'}, + 'end_time_utc': {'key': 'endTimeUtc', 'type': 'iso-8601'}, + 'status': {'key': 'status', 'type': 'str'}, + 'status_reason': {'key': 'statusReason', 'type': 'str'}, + 'mapped_port': {'key': 'mappedPort', 'type': 'int'}, } def __init__( self, *, - pricing_tier: Optional[Union[str, "PricingTier"]] = None, + number: int, + end_time_utc: datetime.datetime, + status: Union[str, "Status"], + status_reason: Union[str, "StatusReason"], + allowed_source_address_prefix: Optional[str] = None, + allowed_source_address_prefixes: Optional[List[str]] = None, + mapped_port: Optional[int] = None, **kwargs ): - super(Pricing, self).__init__(**kwargs) - self.pricing_tier = pricing_tier - self.free_trial_remaining_time = None + """ + :keyword number: Required. + :paramtype number: int + :keyword allowed_source_address_prefix: Mutually exclusive with the + "allowedSourceAddressPrefixes" parameter. Should be an IP address or CIDR, for example + "192.168.0.3" or "192.168.0.0/16". + :paramtype allowed_source_address_prefix: str + :keyword allowed_source_address_prefixes: Mutually exclusive with the + "allowedSourceAddressPrefix" parameter. + :paramtype allowed_source_address_prefixes: list[str] + :keyword end_time_utc: Required. The date & time at which the request ends in UTC. + :paramtype end_time_utc: ~datetime.datetime + :keyword status: Required. The status of the port. Possible values include: "Revoked", + "Initiated". + :paramtype status: str or ~azure.mgmt.security.models.Status + :keyword status_reason: Required. A description of why the ``status`` has its value. Possible + values include: "Expired", "UserRequested", "NewerRequestInitiated". + :paramtype status_reason: str or ~azure.mgmt.security.models.StatusReason + :keyword mapped_port: The port which is mapped to this port's ``number`` in the Azure Firewall, + if applicable. + :paramtype mapped_port: int + """ + super(JitNetworkAccessRequestPort, self).__init__(**kwargs) + self.number = number + self.allowed_source_address_prefix = allowed_source_address_prefix + self.allowed_source_address_prefixes = allowed_source_address_prefixes + self.end_time_utc = end_time_utc + self.status = status + self.status_reason = status_reason + self.mapped_port = mapped_port -class PricingList(msrest.serialization.Model): - """List of pricing configurations response. +class JitNetworkAccessRequestVirtualMachine(msrest.serialization.Model): + """JitNetworkAccessRequestVirtualMachine. All required parameters must be populated in order to send to Azure. - :param value: Required. List of pricing configurations. - :type value: list[~azure.mgmt.security.models.Pricing] + :ivar id: Required. Resource ID of the virtual machine that is linked to this policy. + :vartype id: str + :ivar ports: Required. The ports that were opened for the virtual machine. + :vartype ports: list[~azure.mgmt.security.models.JitNetworkAccessRequestPort] """ _validation = { - 'value': {'required': True}, + 'id': {'required': True}, + 'ports': {'required': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[Pricing]'}, + 'id': {'key': 'id', 'type': 'str'}, + 'ports': {'key': 'ports', 'type': '[JitNetworkAccessRequestPort]'}, } def __init__( self, *, - value: List["Pricing"], + id: str, + ports: List["JitNetworkAccessRequestPort"], **kwargs ): - super(PricingList, self).__init__(**kwargs) - self.value = value + """ + :keyword id: Required. Resource ID of the virtual machine that is linked to this policy. + :paramtype id: str + :keyword ports: Required. The ports that were opened for the virtual machine. + :paramtype ports: list[~azure.mgmt.security.models.JitNetworkAccessRequestPort] + """ + super(JitNetworkAccessRequestVirtualMachine, self).__init__(**kwargs) + self.id = id + self.ports = ports -class ProcessNotAllowed(AllowlistCustomAlertRule): - """Execution of a process that isn't allowed. Allow list consists of process names to allow. +class LocalUserNotAllowed(AllowlistCustomAlertRule): + """Login by a local user that isn't allowed. Allow list consists of login names to allow. Variables are only populated by the server, and will be ignored when sending a request. @@ -6830,16 +7968,16 @@ class ProcessNotAllowed(AllowlistCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", "String". :vartype value_type: str or ~azure.mgmt.security.models.ValueType - :param allowlist_values: Required. The values to allow. The format of the values depends on the + :ivar allowlist_values: Required. The values to allow. The format of the values depends on the rule type. - :type allowlist_values: list[str] + :vartype allowlist_values: list[str] """ _validation = { @@ -6867,146 +8005,153 @@ def __init__( allowlist_values: List[str], **kwargs ): - super(ProcessNotAllowed, self).__init__(is_enabled=is_enabled, allowlist_values=allowlist_values, **kwargs) - self.rule_type = 'ProcessNotAllowed' # type: str - - -class ProtectionMode(msrest.serialization.Model): - """The protection mode of the collection/file types. Exe/Msi/Script are used for Windows, Executable is used for Linux. - - :param exe: The application control policy enforcement/protection mode of the machine group. - Possible values include: "Audit", "Enforce", "None". - :type exe: str or ~azure.mgmt.security.models.EnforcementMode - :param msi: The application control policy enforcement/protection mode of the machine group. - Possible values include: "Audit", "Enforce", "None". - :type msi: str or ~azure.mgmt.security.models.EnforcementMode - :param script: The application control policy enforcement/protection mode of the machine group. - Possible values include: "Audit", "Enforce", "None". - :type script: str or ~azure.mgmt.security.models.EnforcementMode - :param executable: The application control policy enforcement/protection mode of the machine - group. Possible values include: "Audit", "Enforce", "None". - :type executable: str or ~azure.mgmt.security.models.EnforcementMode - """ + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword allowlist_values: Required. The values to allow. The format of the values depends on + the rule type. + :paramtype allowlist_values: list[str] + """ + super(LocalUserNotAllowed, self).__init__(is_enabled=is_enabled, allowlist_values=allowlist_values, **kwargs) + self.rule_type = 'LocalUserNotAllowed' # type: str - _attribute_map = { - 'exe': {'key': 'exe', 'type': 'str'}, - 'msi': {'key': 'msi', 'type': 'str'}, - 'script': {'key': 'script', 'type': 'str'}, - 'executable': {'key': 'executable', 'type': 'str'}, - } - def __init__( - self, - *, - exe: Optional[Union[str, "EnforcementMode"]] = None, - msi: Optional[Union[str, "EnforcementMode"]] = None, - script: Optional[Union[str, "EnforcementMode"]] = None, - executable: Optional[Union[str, "EnforcementMode"]] = None, - **kwargs - ): - super(ProtectionMode, self).__init__(**kwargs) - self.exe = exe - self.msi = msi - self.script = script - self.executable = executable +class LogAnalyticsIdentifier(ResourceIdentifier): + """Represents a Log Analytics workspace scope identifier. + Variables are only populated by the server, and will be ignored when sending a request. -class ProxyServerProperties(msrest.serialization.Model): - """For a non-Azure machine that is not connected directly to the internet, specify a proxy server that the non-Azure machine can use. + All required parameters must be populated in order to send to Azure. - :param ip: Proxy server IP. - :type ip: str - :param port: Proxy server port. - :type port: str + :ivar type: Required. There can be multiple identifiers of different type per alert, this field + specify the identifier type.Constant filled by server. Possible values include: + "AzureResource", "LogAnalytics". + :vartype type: str or ~azure.mgmt.security.models.ResourceIdentifierType + :ivar workspace_id: The LogAnalytics workspace id that stores this alert. + :vartype workspace_id: str + :ivar workspace_subscription_id: The azure subscription id for the LogAnalytics workspace + storing this alert. + :vartype workspace_subscription_id: str + :ivar workspace_resource_group: The azure resource group for the LogAnalytics workspace storing + this alert. + :vartype workspace_resource_group: str + :ivar agent_id: (optional) The LogAnalytics agent id reporting the event that this alert is + based on. + :vartype agent_id: str """ + _validation = { + 'type': {'required': True}, + 'workspace_id': {'readonly': True}, + 'workspace_subscription_id': {'readonly': True, 'pattern': r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'}, + 'workspace_resource_group': {'readonly': True}, + 'agent_id': {'readonly': True}, + } + _attribute_map = { - 'ip': {'key': 'ip', 'type': 'str'}, - 'port': {'key': 'port', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, + 'workspace_subscription_id': {'key': 'workspaceSubscriptionId', 'type': 'str'}, + 'workspace_resource_group': {'key': 'workspaceResourceGroup', 'type': 'str'}, + 'agent_id': {'key': 'agentId', 'type': 'str'}, } def __init__( self, - *, - ip: Optional[str] = None, - port: Optional[str] = None, **kwargs ): - super(ProxyServerProperties, self).__init__(**kwargs) - self.ip = ip - self.port = port + """ + """ + super(LogAnalyticsIdentifier, self).__init__(**kwargs) + self.type = 'LogAnalytics' # type: str + self.workspace_id = None + self.workspace_subscription_id = None + self.workspace_resource_group = None + self.agent_id = None -class PublisherInfo(msrest.serialization.Model): - """Represents the publisher information of a process/rule. +class MdeOnboardingData(Resource): + """The resource of the configuration or data needed to onboard the machine to MDE. - :param publisher_name: The Subject field of the x.509 certificate used to sign the code, using - the following fields - O = Organization, L = Locality, S = State or Province, and C = Country. - :type publisher_name: str - :param product_name: The product name taken from the file's version resource. - :type product_name: str - :param binary_name: The "OriginalName" field taken from the file's version resource. - :type binary_name: str - :param version: The binary file version taken from the file's version resource. - :type version: str + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar onboarding_package_windows: The onboarding package used to onboard Windows machines to + MDE, coded in base64. This can also be used for onboarding using the dedicated VM Extension. + :vartype onboarding_package_windows: bytearray + :ivar onboarding_package_linux: The onboarding package used to onboard Linux machines to MDE, + coded in base64. This can also be used for onboarding using the dedicated VM Extension. + :vartype onboarding_package_linux: bytearray """ + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + } + _attribute_map = { - 'publisher_name': {'key': 'publisherName', 'type': 'str'}, - 'product_name': {'key': 'productName', 'type': 'str'}, - 'binary_name': {'key': 'binaryName', 'type': 'str'}, - 'version': {'key': 'version', 'type': 'str'}, + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'onboarding_package_windows': {'key': 'properties.onboardingPackageWindows', 'type': 'bytearray'}, + 'onboarding_package_linux': {'key': 'properties.onboardingPackageLinux', 'type': 'bytearray'}, } def __init__( self, *, - publisher_name: Optional[str] = None, - product_name: Optional[str] = None, - binary_name: Optional[str] = None, - version: Optional[str] = None, + onboarding_package_windows: Optional[bytearray] = None, + onboarding_package_linux: Optional[bytearray] = None, **kwargs ): - super(PublisherInfo, self).__init__(**kwargs) - self.publisher_name = publisher_name - self.product_name = product_name - self.binary_name = binary_name - self.version = version + """ + :keyword onboarding_package_windows: The onboarding package used to onboard Windows machines to + MDE, coded in base64. This can also be used for onboarding using the dedicated VM Extension. + :paramtype onboarding_package_windows: bytearray + :keyword onboarding_package_linux: The onboarding package used to onboard Linux machines to + MDE, coded in base64. This can also be used for onboarding using the dedicated VM Extension. + :paramtype onboarding_package_linux: bytearray + """ + super(MdeOnboardingData, self).__init__(**kwargs) + self.onboarding_package_windows = onboarding_package_windows + self.onboarding_package_linux = onboarding_package_linux -class QueryCheck(msrest.serialization.Model): - """The rule query details. +class MdeOnboardingDataList(msrest.serialization.Model): + """List of all MDE onboarding data resources. - :param query: The rule query. - :type query: str - :param expected_result: Expected result. - :type expected_result: list[list[str]] - :param column_names: Column names of expected result. - :type column_names: list[str] + :ivar value: List of the resources of the configuration or data needed to onboard the machine + to MDE. + :vartype value: list[~azure.mgmt.security.models.MdeOnboardingData] """ _attribute_map = { - 'query': {'key': 'query', 'type': 'str'}, - 'expected_result': {'key': 'expectedResult', 'type': '[[str]]'}, - 'column_names': {'key': 'columnNames', 'type': '[str]'}, + 'value': {'key': 'value', 'type': '[MdeOnboardingData]'}, } def __init__( self, *, - query: Optional[str] = None, - expected_result: Optional[List[List[str]]] = None, - column_names: Optional[List[str]] = None, + value: Optional[List["MdeOnboardingData"]] = None, **kwargs ): - super(QueryCheck, self).__init__(**kwargs) - self.query = query - self.expected_result = expected_result - self.column_names = column_names + """ + :keyword value: List of the resources of the configuration or data needed to onboard the + machine to MDE. + :paramtype value: list[~azure.mgmt.security.models.MdeOnboardingData] + """ + super(MdeOnboardingDataList, self).__init__(**kwargs) + self.value = value -class QueuePurgesNotInAllowedRange(TimeWindowCustomAlertRule): - """Number of device queue purges is not in allowed range. +class MqttC2DMessagesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of cloud to device messages (MQTT protocol) is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. @@ -7016,16 +8161,16 @@ class QueuePurgesNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -7057,445 +8202,512 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): - super(QueuePurgesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) - self.rule_type = 'QueuePurgesNotInAllowedRange' # type: str + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(MqttC2DMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'MqttC2DMessagesNotInAllowedRange' # type: str -class RecommendationConfigurationProperties(msrest.serialization.Model): - """The type of IoT Security recommendation. +class MqttC2DRejectedMessagesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of rejected cloud to device messages (MQTT protocol) is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :param recommendation_type: Required. The type of IoT Security recommendation. Possible values - include: "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", - "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", - "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", - "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", - "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", - "IoT_VulnerableTLSCipherSuite". - :type recommendation_type: str or ~azure.mgmt.security.models.RecommendationType - :ivar name: - :vartype name: str - :param status: Required. Recommendation status. When the recommendation status is disabled - recommendations are not generated. Possible values include: "Disabled", "Enabled". Default - value: "Enabled". - :type status: str or ~azure.mgmt.security.models.RecommendationConfigStatus + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. + :vartype description: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { - 'recommendation_type': {'required': True}, - 'name': {'readonly': True}, - 'status': {'required': True}, + 'display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, } _attribute_map = { - 'recommendation_type': {'key': 'recommendationType', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'status': {'key': 'status', 'type': 'str'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, *, - recommendation_type: Union[str, "RecommendationType"], - status: Union[str, "RecommendationConfigStatus"] = "Enabled", + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(RecommendationConfigurationProperties, self).__init__(**kwargs) - self.recommendation_type = recommendation_type - self.name = None - self.status = status + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(MqttC2DRejectedMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'MqttC2DRejectedMessagesNotInAllowedRange' # type: str -class RegulatoryComplianceAssessment(Resource): - """Regulatory compliance assessment details and state. +class MqttD2CMessagesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of device to cloud messages (MQTT protocol) is not in allowed range. Variables are only populated by the server, and will be ignored when sending a request. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar description: The description of the regulatory compliance assessment. + All required parameters must be populated in order to send to Azure. + + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. :vartype description: str - :ivar assessment_type: The expected type of assessment contained in the AssessmentDetailsLink. - :vartype assessment_type: str - :ivar assessment_details_link: Link to more detailed assessment results data. The response type - will be according to the assessmentType field. - :vartype assessment_details_link: str - :param state: Aggregative state based on the assessment's scanned resources states. Possible - values include: "Passed", "Failed", "Skipped", "Unsupported". - :type state: str or ~azure.mgmt.security.models.State - :ivar passed_resources: The given assessment's related resources count with passed state. - :vartype passed_resources: int - :ivar failed_resources: The given assessment's related resources count with failed state. - :vartype failed_resources: int - :ivar skipped_resources: The given assessment's related resources count with skipped state. - :vartype skipped_resources: int - :ivar unsupported_resources: The given assessment's related resources count with unsupported - state. - :vartype unsupported_resources: int + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, + 'display_name': {'readonly': True}, 'description': {'readonly': True}, - 'assessment_type': {'readonly': True}, - 'assessment_details_link': {'readonly': True}, - 'passed_resources': {'readonly': True}, - 'failed_resources': {'readonly': True}, - 'skipped_resources': {'readonly': True}, - 'unsupported_resources': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'assessment_type': {'key': 'properties.assessmentType', 'type': 'str'}, - 'assessment_details_link': {'key': 'properties.assessmentDetailsLink', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'passed_resources': {'key': 'properties.passedResources', 'type': 'int'}, - 'failed_resources': {'key': 'properties.failedResources', 'type': 'int'}, - 'skipped_resources': {'key': 'properties.skippedResources', 'type': 'int'}, - 'unsupported_resources': {'key': 'properties.unsupportedResources', 'type': 'int'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, *, - state: Optional[Union[str, "State"]] = None, + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(RegulatoryComplianceAssessment, self).__init__(**kwargs) - self.description = None - self.assessment_type = None - self.assessment_details_link = None - self.state = state - self.passed_resources = None - self.failed_resources = None - self.skipped_resources = None - self.unsupported_resources = None + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(MqttD2CMessagesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'MqttD2CMessagesNotInAllowedRange' # type: str -class RegulatoryComplianceAssessmentList(msrest.serialization.Model): - """List of regulatory compliance assessment response. +class OnPremiseResourceDetails(ResourceDetails): + """Details of the On Premise resource that was assessed. - Variables are only populated by the server, and will be ignored when sending a request. + You probably want to use the sub-classes and not this class directly. Known + sub-classes are: OnPremiseSqlResourceDetails. All required parameters must be populated in order to send to Azure. - :param value: Required. - :type value: list[~azure.mgmt.security.models.RegulatoryComplianceAssessment] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + :ivar source: Required. The platform where the assessed resource resides.Constant filled by + server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". + :vartype source: str or ~azure.mgmt.security.models.Source + :ivar workspace_id: Required. Azure resource Id of the workspace the machine is attached to. + :vartype workspace_id: str + :ivar vmuuid: Required. The unique Id of the machine. + :vartype vmuuid: str + :ivar source_computer_id: Required. The oms agent Id installed on the machine. + :vartype source_computer_id: str + :ivar machine_name: Required. The name of the machine. + :vartype machine_name: str """ _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, + 'source': {'required': True}, + 'workspace_id': {'required': True}, + 'vmuuid': {'required': True}, + 'source_computer_id': {'required': True}, + 'machine_name': {'required': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[RegulatoryComplianceAssessment]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'source': {'key': 'source', 'type': 'str'}, + 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, + 'vmuuid': {'key': 'vmuuid', 'type': 'str'}, + 'source_computer_id': {'key': 'sourceComputerId', 'type': 'str'}, + 'machine_name': {'key': 'machineName', 'type': 'str'}, + } + + _subtype_map = { + 'source': {'OnPremiseSql': 'OnPremiseSqlResourceDetails'} } def __init__( self, *, - value: List["RegulatoryComplianceAssessment"], + workspace_id: str, + vmuuid: str, + source_computer_id: str, + machine_name: str, **kwargs ): - super(RegulatoryComplianceAssessmentList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + :keyword workspace_id: Required. Azure resource Id of the workspace the machine is attached to. + :paramtype workspace_id: str + :keyword vmuuid: Required. The unique Id of the machine. + :paramtype vmuuid: str + :keyword source_computer_id: Required. The oms agent Id installed on the machine. + :paramtype source_computer_id: str + :keyword machine_name: Required. The name of the machine. + :paramtype machine_name: str + """ + super(OnPremiseResourceDetails, self).__init__(**kwargs) + self.source = 'OnPremise' # type: str + self.workspace_id = workspace_id + self.vmuuid = vmuuid + self.source_computer_id = source_computer_id + self.machine_name = machine_name -class RegulatoryComplianceControl(Resource): - """Regulatory compliance control details and state. +class OnPremiseSqlResourceDetails(OnPremiseResourceDetails): + """Details of the On Premise Sql resource that was assessed. - Variables are only populated by the server, and will be ignored when sending a request. + All required parameters must be populated in order to send to Azure. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar description: The description of the regulatory compliance control. - :vartype description: str - :param state: Aggregative state based on the control's supported assessments states. Possible - values include: "Passed", "Failed", "Skipped", "Unsupported". - :type state: str or ~azure.mgmt.security.models.State - :ivar passed_assessments: The number of supported regulatory compliance assessments of the - given control with a passed state. - :vartype passed_assessments: int - :ivar failed_assessments: The number of supported regulatory compliance assessments of the - given control with a failed state. - :vartype failed_assessments: int - :ivar skipped_assessments: The number of supported regulatory compliance assessments of the - given control with a skipped state. - :vartype skipped_assessments: int + :ivar source: Required. The platform where the assessed resource resides.Constant filled by + server. Possible values include: "Azure", "OnPremise", "OnPremiseSql". + :vartype source: str or ~azure.mgmt.security.models.Source + :ivar workspace_id: Required. Azure resource Id of the workspace the machine is attached to. + :vartype workspace_id: str + :ivar vmuuid: Required. The unique Id of the machine. + :vartype vmuuid: str + :ivar source_computer_id: Required. The oms agent Id installed on the machine. + :vartype source_computer_id: str + :ivar machine_name: Required. The name of the machine. + :vartype machine_name: str + :ivar server_name: Required. The Sql server name installed on the machine. + :vartype server_name: str + :ivar database_name: Required. The Sql database name installed on the machine. + :vartype database_name: str """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'description': {'readonly': True}, - 'passed_assessments': {'readonly': True}, - 'failed_assessments': {'readonly': True}, - 'skipped_assessments': {'readonly': True}, + 'source': {'required': True}, + 'workspace_id': {'required': True}, + 'vmuuid': {'required': True}, + 'source_computer_id': {'required': True}, + 'machine_name': {'required': True}, + 'server_name': {'required': True}, + 'database_name': {'required': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'passed_assessments': {'key': 'properties.passedAssessments', 'type': 'int'}, - 'failed_assessments': {'key': 'properties.failedAssessments', 'type': 'int'}, - 'skipped_assessments': {'key': 'properties.skippedAssessments', 'type': 'int'}, + 'source': {'key': 'source', 'type': 'str'}, + 'workspace_id': {'key': 'workspaceId', 'type': 'str'}, + 'vmuuid': {'key': 'vmuuid', 'type': 'str'}, + 'source_computer_id': {'key': 'sourceComputerId', 'type': 'str'}, + 'machine_name': {'key': 'machineName', 'type': 'str'}, + 'server_name': {'key': 'serverName', 'type': 'str'}, + 'database_name': {'key': 'databaseName', 'type': 'str'}, } def __init__( self, *, - state: Optional[Union[str, "State"]] = None, + workspace_id: str, + vmuuid: str, + source_computer_id: str, + machine_name: str, + server_name: str, + database_name: str, **kwargs ): - super(RegulatoryComplianceControl, self).__init__(**kwargs) - self.description = None - self.state = state - self.passed_assessments = None - self.failed_assessments = None - self.skipped_assessments = None + """ + :keyword workspace_id: Required. Azure resource Id of the workspace the machine is attached to. + :paramtype workspace_id: str + :keyword vmuuid: Required. The unique Id of the machine. + :paramtype vmuuid: str + :keyword source_computer_id: Required. The oms agent Id installed on the machine. + :paramtype source_computer_id: str + :keyword machine_name: Required. The name of the machine. + :paramtype machine_name: str + :keyword server_name: Required. The Sql server name installed on the machine. + :paramtype server_name: str + :keyword database_name: Required. The Sql database name installed on the machine. + :paramtype database_name: str + """ + super(OnPremiseSqlResourceDetails, self).__init__(workspace_id=workspace_id, vmuuid=vmuuid, source_computer_id=source_computer_id, machine_name=machine_name, **kwargs) + self.source = 'OnPremiseSql' # type: str + self.server_name = server_name + self.database_name = database_name -class RegulatoryComplianceControlList(msrest.serialization.Model): - """List of regulatory compliance controls response. +class Operation(msrest.serialization.Model): + """Possible operation in the REST API of Microsoft.Security. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :param value: Required. List of regulatory compliance controls. - :type value: list[~azure.mgmt.security.models.RegulatoryComplianceControl] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + :ivar name: Name of the operation. + :vartype name: str + :ivar origin: Where the operation is originated. + :vartype origin: str + :ivar display: Security operation display. + :vartype display: ~azure.mgmt.security.models.OperationDisplay """ _validation = { - 'value': {'required': True}, - 'next_link': {'readonly': True}, + 'name': {'readonly': True}, + 'origin': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[RegulatoryComplianceControl]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'origin': {'key': 'origin', 'type': 'str'}, + 'display': {'key': 'display', 'type': 'OperationDisplay'}, } def __init__( self, *, - value: List["RegulatoryComplianceControl"], + display: Optional["OperationDisplay"] = None, **kwargs ): - super(RegulatoryComplianceControlList, self).__init__(**kwargs) - self.value = value - self.next_link = None + """ + :keyword display: Security operation display. + :paramtype display: ~azure.mgmt.security.models.OperationDisplay + """ + super(Operation, self).__init__(**kwargs) + self.name = None + self.origin = None + self.display = display -class RegulatoryComplianceStandard(Resource): - """Regulatory compliance standard details and state. +class OperationDisplay(msrest.serialization.Model): + """Security operation display. Variables are only populated by the server, and will be ignored when sending a request. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param state: Aggregative state based on the standard's supported controls states. Possible - values include: "Passed", "Failed", "Skipped", "Unsupported". - :type state: str or ~azure.mgmt.security.models.State - :ivar passed_controls: The number of supported regulatory compliance controls of the given - standard with a passed state. - :vartype passed_controls: int - :ivar failed_controls: The number of supported regulatory compliance controls of the given - standard with a failed state. - :vartype failed_controls: int - :ivar skipped_controls: The number of supported regulatory compliance controls of the given - standard with a skipped state. - :vartype skipped_controls: int - :ivar unsupported_controls: The number of regulatory compliance controls of the given standard - which are unsupported by automated assessments. - :vartype unsupported_controls: int + :ivar provider: The resource provider for the operation. + :vartype provider: str + :ivar resource: The display name of the resource the operation applies to. + :vartype resource: str + :ivar operation: The display name of the security operation. + :vartype operation: str + :ivar description: The description of the operation. + :vartype description: str """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'passed_controls': {'readonly': True}, - 'failed_controls': {'readonly': True}, - 'skipped_controls': {'readonly': True}, - 'unsupported_controls': {'readonly': True}, + 'provider': {'readonly': True}, + 'resource': {'readonly': True}, + 'operation': {'readonly': True}, + 'description': {'readonly': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'passed_controls': {'key': 'properties.passedControls', 'type': 'int'}, - 'failed_controls': {'key': 'properties.failedControls', 'type': 'int'}, - 'skipped_controls': {'key': 'properties.skippedControls', 'type': 'int'}, - 'unsupported_controls': {'key': 'properties.unsupportedControls', 'type': 'int'}, + 'provider': {'key': 'provider', 'type': 'str'}, + 'resource': {'key': 'resource', 'type': 'str'}, + 'operation': {'key': 'operation', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, } def __init__( self, - *, - state: Optional[Union[str, "State"]] = None, **kwargs ): - super(RegulatoryComplianceStandard, self).__init__(**kwargs) - self.state = state - self.passed_controls = None - self.failed_controls = None - self.skipped_controls = None - self.unsupported_controls = None + """ + """ + super(OperationDisplay, self).__init__(**kwargs) + self.provider = None + self.resource = None + self.operation = None + self.description = None -class RegulatoryComplianceStandardList(msrest.serialization.Model): - """List of regulatory compliance standards response. +class OperationList(msrest.serialization.Model): + """List of possible operations for Microsoft.Security resource provider. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. - - :param value: Required. - :type value: list[~azure.mgmt.security.models.RegulatoryComplianceStandard] + :ivar value: List of Security operations. + :vartype value: list[~azure.mgmt.security.models.Operation] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ _validation = { - 'value': {'required': True}, 'next_link': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[RegulatoryComplianceStandard]'}, + 'value': {'key': 'value', 'type': '[Operation]'}, 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - value: List["RegulatoryComplianceStandard"], + value: Optional[List["Operation"]] = None, **kwargs ): - super(RegulatoryComplianceStandardList, self).__init__(**kwargs) + """ + :keyword value: List of Security operations. + :paramtype value: list[~azure.mgmt.security.models.Operation] + """ + super(OperationList, self).__init__(**kwargs) self.value = value self.next_link = None -class Remediation(msrest.serialization.Model): - """Remediation details. - - :param description: Remediation description. - :type description: str - :param scripts: Remediation script. - :type scripts: list[str] - :param automated: Is remediation automated. - :type automated: bool - :param portal_link: Optional link to remediate in Azure Portal. - :type portal_link: str - """ - - _attribute_map = { - 'description': {'key': 'description', 'type': 'str'}, - 'scripts': {'key': 'scripts', 'type': '[str]'}, - 'automated': {'key': 'automated', 'type': 'bool'}, - 'portal_link': {'key': 'portalLink', 'type': 'str'}, - } - - def __init__( - self, - *, - description: Optional[str] = None, - scripts: Optional[List[str]] = None, - automated: Optional[bool] = None, - portal_link: Optional[str] = None, - **kwargs - ): - super(Remediation, self).__init__(**kwargs) - self.description = description - self.scripts = scripts - self.automated = automated - self.portal_link = portal_link - - -class Rule(msrest.serialization.Model): - """Describes remote addresses that is recommended to communicate with the Azure resource on some (Protocol, Port, Direction). All other remote addresses are recommended to be blocked. +class PathRecommendation(msrest.serialization.Model): + """Represents a path that is recommended to be allowed and its properties. - :param name: The name of the rule. - :type name: str - :param direction: The rule's direction. Possible values include: "Inbound", "Outbound". - :type direction: str or ~azure.mgmt.security.models.Direction - :param destination_port: The rule's destination port. - :type destination_port: int - :param protocols: The rule's transport protocols. - :type protocols: list[str or ~azure.mgmt.security.models.TransportProtocol] - :param ip_addresses: The remote IP addresses that should be able to communicate with the Azure - resource on the rule's destination port and protocol. - :type ip_addresses: list[str] + :ivar path: The full path of the file, or an identifier of the application. + :vartype path: str + :ivar action: The recommendation action of the machine or rule. Possible values include: + "Recommended", "Add", "Remove". + :vartype action: str or ~azure.mgmt.security.models.RecommendationAction + :ivar type: The type of IoT Security recommendation. Possible values include: + "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", + "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", + "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", + "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", + "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", + "IoT_VulnerableTLSCipherSuite". + :vartype type: str or ~azure.mgmt.security.models.RecommendationType + :ivar publisher_info: Represents the publisher information of a process/rule. + :vartype publisher_info: ~azure.mgmt.security.models.PublisherInfo + :ivar common: Whether the application is commonly run on the machine. + :vartype common: bool + :ivar user_sids: + :vartype user_sids: list[str] + :ivar usernames: + :vartype usernames: list[~azure.mgmt.security.models.UserRecommendation] + :ivar file_type: The type of the file (for Linux files - Executable is used). Possible values + include: "Exe", "Dll", "Msi", "Script", "Executable", "Unknown". + :vartype file_type: str or ~azure.mgmt.security.models.FileType + :ivar configuration_status: The configuration status of the machines group or machine or rule. + Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", "NoStatus". + :vartype configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus """ - _validation = { - 'destination_port': {'maximum': 65535, 'minimum': 0}, - } - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'direction': {'key': 'direction', 'type': 'str'}, - 'destination_port': {'key': 'destinationPort', 'type': 'int'}, - 'protocols': {'key': 'protocols', 'type': '[str]'}, - 'ip_addresses': {'key': 'ipAddresses', 'type': '[str]'}, + 'path': {'key': 'path', 'type': 'str'}, + 'action': {'key': 'action', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'publisher_info': {'key': 'publisherInfo', 'type': 'PublisherInfo'}, + 'common': {'key': 'common', 'type': 'bool'}, + 'user_sids': {'key': 'userSids', 'type': '[str]'}, + 'usernames': {'key': 'usernames', 'type': '[UserRecommendation]'}, + 'file_type': {'key': 'fileType', 'type': 'str'}, + 'configuration_status': {'key': 'configurationStatus', 'type': 'str'}, } def __init__( self, *, - name: Optional[str] = None, - direction: Optional[Union[str, "Direction"]] = None, - destination_port: Optional[int] = None, - protocols: Optional[List[Union[str, "TransportProtocol"]]] = None, - ip_addresses: Optional[List[str]] = None, + path: Optional[str] = None, + action: Optional[Union[str, "RecommendationAction"]] = None, + type: Optional[Union[str, "RecommendationType"]] = None, + publisher_info: Optional["PublisherInfo"] = None, + common: Optional[bool] = None, + user_sids: Optional[List[str]] = None, + usernames: Optional[List["UserRecommendation"]] = None, + file_type: Optional[Union[str, "FileType"]] = None, + configuration_status: Optional[Union[str, "ConfigurationStatus"]] = None, **kwargs ): - super(Rule, self).__init__(**kwargs) - self.name = name - self.direction = direction - self.destination_port = destination_port - self.protocols = protocols - self.ip_addresses = ip_addresses + """ + :keyword path: The full path of the file, or an identifier of the application. + :paramtype path: str + :keyword action: The recommendation action of the machine or rule. Possible values include: + "Recommended", "Add", "Remove". + :paramtype action: str or ~azure.mgmt.security.models.RecommendationAction + :keyword type: The type of IoT Security recommendation. Possible values include: + "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", + "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", + "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", + "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", + "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", + "IoT_VulnerableTLSCipherSuite". + :paramtype type: str or ~azure.mgmt.security.models.RecommendationType + :keyword publisher_info: Represents the publisher information of a process/rule. + :paramtype publisher_info: ~azure.mgmt.security.models.PublisherInfo + :keyword common: Whether the application is commonly run on the machine. + :paramtype common: bool + :keyword user_sids: + :paramtype user_sids: list[str] + :keyword usernames: + :paramtype usernames: list[~azure.mgmt.security.models.UserRecommendation] + :keyword file_type: The type of the file (for Linux files - Executable is used). Possible + values include: "Exe", "Dll", "Msi", "Script", "Executable", "Unknown". + :paramtype file_type: str or ~azure.mgmt.security.models.FileType + :keyword configuration_status: The configuration status of the machines group or machine or + rule. Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", + "NoStatus". + :paramtype configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus + """ + super(PathRecommendation, self).__init__(**kwargs) + self.path = path + self.action = action + self.type = type + self.publisher_info = publisher_info + self.common = common + self.user_sids = user_sids + self.usernames = usernames + self.file_type = file_type + self.configuration_status = configuration_status -class RuleResults(Resource): - """Rule results. +class Pricing(Resource): + """Azure Security Center is provided in two pricing tiers: free and standard, with the standard tier available with a trial period. The standard tier offers advanced security capabilities, while the free tier offers basic security features. Variables are only populated by the server, and will be ignored when sending a request. @@ -7505,415 +8717,562 @@ class RuleResults(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param properties: Rule results properties. - :type properties: ~azure.mgmt.security.models.RuleResultsProperties + :ivar pricing_tier: The pricing tier value. Azure Security Center is provided in two pricing + tiers: free and standard, with the standard tier available with a trial period. The standard + tier offers advanced security capabilities, while the free tier offers basic security features. + Possible values include: "Free", "Standard". + :vartype pricing_tier: str or ~azure.mgmt.security.models.PricingTier + :ivar free_trial_remaining_time: The duration left for the subscriptions free trial period - in + ISO 8601 format (e.g. P3Y6M4DT12H30M5S). + :vartype free_trial_remaining_time: ~datetime.timedelta """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, + 'free_trial_remaining_time': {'readonly': True}, } _attribute_map = { 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'RuleResultsProperties'}, + 'pricing_tier': {'key': 'properties.pricingTier', 'type': 'str'}, + 'free_trial_remaining_time': {'key': 'properties.freeTrialRemainingTime', 'type': 'duration'}, } def __init__( self, *, - properties: Optional["RuleResultsProperties"] = None, + pricing_tier: Optional[Union[str, "PricingTier"]] = None, **kwargs ): - super(RuleResults, self).__init__(**kwargs) - self.properties = properties + """ + :keyword pricing_tier: The pricing tier value. Azure Security Center is provided in two pricing + tiers: free and standard, with the standard tier available with a trial period. The standard + tier offers advanced security capabilities, while the free tier offers basic security features. + Possible values include: "Free", "Standard". + :paramtype pricing_tier: str or ~azure.mgmt.security.models.PricingTier + """ + super(Pricing, self).__init__(**kwargs) + self.pricing_tier = pricing_tier + self.free_trial_remaining_time = None -class RuleResultsInput(msrest.serialization.Model): - """Rule results input. +class PricingList(msrest.serialization.Model): + """List of pricing configurations response. - :param latest_scan: Take results from latest scan. - :type latest_scan: bool - :param results: Expected results to be inserted into the baseline. - Leave this field empty it LatestScan == true. - :type results: list[list[str]] + All required parameters must be populated in order to send to Azure. + + :ivar value: Required. List of pricing configurations. + :vartype value: list[~azure.mgmt.security.models.Pricing] """ - _attribute_map = { - 'latest_scan': {'key': 'latestScan', 'type': 'bool'}, - 'results': {'key': 'results', 'type': '[[str]]'}, + _validation = { + 'value': {'required': True}, } - def __init__( - self, - *, - latest_scan: Optional[bool] = None, - results: Optional[List[List[str]]] = None, - **kwargs - ): - super(RuleResultsInput, self).__init__(**kwargs) - self.latest_scan = latest_scan - self.results = results - - -class RuleResultsProperties(msrest.serialization.Model): - """Rule results properties. - - :param results: Expected results in the baseline. - :type results: list[list[str]] - """ - _attribute_map = { - 'results': {'key': 'results', 'type': '[[str]]'}, + 'value': {'key': 'value', 'type': '[Pricing]'}, } def __init__( self, *, - results: Optional[List[List[str]]] = None, + value: List["Pricing"], **kwargs ): - super(RuleResultsProperties, self).__init__(**kwargs) - self.results = results + """ + :keyword value: Required. List of pricing configurations. + :paramtype value: list[~azure.mgmt.security.models.Pricing] + """ + super(PricingList, self).__init__(**kwargs) + self.value = value -class RulesResults(msrest.serialization.Model): - """A list of rules results. +class ProcessNotAllowed(AllowlistCustomAlertRule): + """Execution of a process that isn't allowed. Allow list consists of process names to allow. - :param value: List of rule results. - :type value: list[~azure.mgmt.security.models.RuleResults] - """ + Variables are only populated by the server, and will be ignored when sending a request. - _attribute_map = { - 'value': {'key': 'value', 'type': '[RuleResults]'}, - } - - def __init__( - self, - *, - value: Optional[List["RuleResults"]] = None, - **kwargs - ): - super(RulesResults, self).__init__(**kwargs) - self.value = value - - -class RulesResultsInput(msrest.serialization.Model): - """Rules results input. + All required parameters must be populated in order to send to Azure. - :param latest_scan: Take results from latest scan. - :type latest_scan: bool - :param results: Expected results to be inserted into the baseline. - Leave this field empty it LatestScan == true. - :type results: dict[str, list[list[str]]] + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. + :vartype description: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar value_type: The value type of the items in the list. Possible values include: "IpCidr", + "String". + :vartype value_type: str or ~azure.mgmt.security.models.ValueType + :ivar allowlist_values: Required. The values to allow. The format of the values depends on the + rule type. + :vartype allowlist_values: list[str] """ + _validation = { + 'display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'value_type': {'readonly': True}, + 'allowlist_values': {'required': True}, + } + _attribute_map = { - 'latest_scan': {'key': 'latestScan', 'type': 'bool'}, - 'results': {'key': 'results', 'type': '{[[str]]}'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'value_type': {'key': 'valueType', 'type': 'str'}, + 'allowlist_values': {'key': 'allowlistValues', 'type': '[str]'}, } def __init__( self, *, - latest_scan: Optional[bool] = None, - results: Optional[Dict[str, List[List[str]]]] = None, + is_enabled: bool, + allowlist_values: List[str], **kwargs ): - super(RulesResultsInput, self).__init__(**kwargs) - self.latest_scan = latest_scan - self.results = results - + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword allowlist_values: Required. The values to allow. The format of the values depends on + the rule type. + :paramtype allowlist_values: list[str] + """ + super(ProcessNotAllowed, self).__init__(is_enabled=is_enabled, allowlist_values=allowlist_values, **kwargs) + self.rule_type = 'ProcessNotAllowed' # type: str -class Scan(Resource): - """A vulnerability assessment scan record. - Variables are only populated by the server, and will be ignored when sending a request. +class ProtectionMode(msrest.serialization.Model): + """The protection mode of the collection/file types. Exe/Msi/Script are used for Windows, Executable is used for Linux. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: A vulnerability assessment scan record properties. - :type properties: ~azure.mgmt.security.models.ScanProperties + :ivar exe: The application control policy enforcement/protection mode of the machine group. + Possible values include: "Audit", "Enforce", "None". + :vartype exe: str or ~azure.mgmt.security.models.EnforcementMode + :ivar msi: The application control policy enforcement/protection mode of the machine group. + Possible values include: "Audit", "Enforce", "None". + :vartype msi: str or ~azure.mgmt.security.models.EnforcementMode + :ivar script: The application control policy enforcement/protection mode of the machine group. + Possible values include: "Audit", "Enforce", "None". + :vartype script: str or ~azure.mgmt.security.models.EnforcementMode + :ivar executable: The application control policy enforcement/protection mode of the machine + group. Possible values include: "Audit", "Enforce", "None". + :vartype executable: str or ~azure.mgmt.security.models.EnforcementMode """ - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'ScanProperties'}, + 'exe': {'key': 'exe', 'type': 'str'}, + 'msi': {'key': 'msi', 'type': 'str'}, + 'script': {'key': 'script', 'type': 'str'}, + 'executable': {'key': 'executable', 'type': 'str'}, } def __init__( self, *, - properties: Optional["ScanProperties"] = None, + exe: Optional[Union[str, "EnforcementMode"]] = None, + msi: Optional[Union[str, "EnforcementMode"]] = None, + script: Optional[Union[str, "EnforcementMode"]] = None, + executable: Optional[Union[str, "EnforcementMode"]] = None, **kwargs ): - super(Scan, self).__init__(**kwargs) - self.properties = properties + """ + :keyword exe: The application control policy enforcement/protection mode of the machine group. + Possible values include: "Audit", "Enforce", "None". + :paramtype exe: str or ~azure.mgmt.security.models.EnforcementMode + :keyword msi: The application control policy enforcement/protection mode of the machine group. + Possible values include: "Audit", "Enforce", "None". + :paramtype msi: str or ~azure.mgmt.security.models.EnforcementMode + :keyword script: The application control policy enforcement/protection mode of the machine + group. Possible values include: "Audit", "Enforce", "None". + :paramtype script: str or ~azure.mgmt.security.models.EnforcementMode + :keyword executable: The application control policy enforcement/protection mode of the machine + group. Possible values include: "Audit", "Enforce", "None". + :paramtype executable: str or ~azure.mgmt.security.models.EnforcementMode + """ + super(ProtectionMode, self).__init__(**kwargs) + self.exe = exe + self.msi = msi + self.script = script + self.executable = executable -class ScanProperties(msrest.serialization.Model): - """A vulnerability assessment scan record properties. +class ProxyServerProperties(msrest.serialization.Model): + """For a non-Azure machine that is not connected directly to the internet, specify a proxy server that the non-Azure machine can use. - :param trigger_type: The scan trigger type. Possible values include: "OnDemand", "Recurring". - :type trigger_type: str or ~azure.mgmt.security.models.ScanTriggerType - :param state: The scan status. Possible values include: "Failed", "FailedToRun", "InProgress", - "Passed". - :type state: str or ~azure.mgmt.security.models.ScanState - :param server: The server name. - :type server: str - :param database: The database name. - :type database: str - :param sql_version: The SQL version. - :type sql_version: str - :param start_time: The scan start time (UTC). - :type start_time: ~datetime.datetime - :param end_time: Scan results are valid until end time (UTC). - :type end_time: ~datetime.datetime - :param high_severity_failed_rules_count: The number of failed rules with high severity. - :type high_severity_failed_rules_count: int - :param medium_severity_failed_rules_count: The number of failed rules with medium severity. - :type medium_severity_failed_rules_count: int - :param low_severity_failed_rules_count: The number of failed rules with low severity. - :type low_severity_failed_rules_count: int - :param total_passed_rules_count: The number of total passed rules. - :type total_passed_rules_count: int - :param total_failed_rules_count: The number of total failed rules. - :type total_failed_rules_count: int - :param total_rules_count: The number of total rules assessed. - :type total_rules_count: int - :param is_baseline_applied: Baseline created for this database, and has one or more rules. - :type is_baseline_applied: bool + :ivar ip: Proxy server IP. + :vartype ip: str + :ivar port: Proxy server port. + :vartype port: str """ _attribute_map = { - 'trigger_type': {'key': 'triggerType', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'server': {'key': 'server', 'type': 'str'}, - 'database': {'key': 'database', 'type': 'str'}, - 'sql_version': {'key': 'sqlVersion', 'type': 'str'}, - 'start_time': {'key': 'startTime', 'type': 'iso-8601'}, - 'end_time': {'key': 'endTime', 'type': 'iso-8601'}, - 'high_severity_failed_rules_count': {'key': 'highSeverityFailedRulesCount', 'type': 'int'}, - 'medium_severity_failed_rules_count': {'key': 'mediumSeverityFailedRulesCount', 'type': 'int'}, - 'low_severity_failed_rules_count': {'key': 'lowSeverityFailedRulesCount', 'type': 'int'}, - 'total_passed_rules_count': {'key': 'totalPassedRulesCount', 'type': 'int'}, - 'total_failed_rules_count': {'key': 'totalFailedRulesCount', 'type': 'int'}, - 'total_rules_count': {'key': 'totalRulesCount', 'type': 'int'}, - 'is_baseline_applied': {'key': 'isBaselineApplied', 'type': 'bool'}, + 'ip': {'key': 'ip', 'type': 'str'}, + 'port': {'key': 'port', 'type': 'str'}, } def __init__( self, *, - trigger_type: Optional[Union[str, "ScanTriggerType"]] = None, - state: Optional[Union[str, "ScanState"]] = None, - server: Optional[str] = None, - database: Optional[str] = None, - sql_version: Optional[str] = None, - start_time: Optional[datetime.datetime] = None, - end_time: Optional[datetime.datetime] = None, - high_severity_failed_rules_count: Optional[int] = None, - medium_severity_failed_rules_count: Optional[int] = None, - low_severity_failed_rules_count: Optional[int] = None, - total_passed_rules_count: Optional[int] = None, - total_failed_rules_count: Optional[int] = None, - total_rules_count: Optional[int] = None, - is_baseline_applied: Optional[bool] = None, + ip: Optional[str] = None, + port: Optional[str] = None, **kwargs ): - super(ScanProperties, self).__init__(**kwargs) - self.trigger_type = trigger_type - self.state = state - self.server = server - self.database = database - self.sql_version = sql_version - self.start_time = start_time - self.end_time = end_time - self.high_severity_failed_rules_count = high_severity_failed_rules_count - self.medium_severity_failed_rules_count = medium_severity_failed_rules_count - self.low_severity_failed_rules_count = low_severity_failed_rules_count - self.total_passed_rules_count = total_passed_rules_count - self.total_failed_rules_count = total_failed_rules_count - self.total_rules_count = total_rules_count - self.is_baseline_applied = is_baseline_applied - + """ + :keyword ip: Proxy server IP. + :paramtype ip: str + :keyword port: Proxy server port. + :paramtype port: str + """ + super(ProxyServerProperties, self).__init__(**kwargs) + self.ip = ip + self.port = port -class ScanResult(Resource): - """A vulnerability assessment scan result for a single rule. - Variables are only populated by the server, and will be ignored when sending a request. +class PublisherInfo(msrest.serialization.Model): + """Represents the publisher information of a process/rule. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :param properties: A vulnerability assessment scan result properties for a single rule. - :type properties: ~azure.mgmt.security.models.ScanResultProperties + :ivar publisher_name: The Subject field of the x.509 certificate used to sign the code, using + the following fields - O = Organization, L = Locality, S = State or Province, and C = Country. + :vartype publisher_name: str + :ivar product_name: The product name taken from the file's version resource. + :vartype product_name: str + :ivar binary_name: The "OriginalName" field taken from the file's version resource. + :vartype binary_name: str + :ivar version: The binary file version taken from the file's version resource. + :vartype version: str """ - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'properties': {'key': 'properties', 'type': 'ScanResultProperties'}, + 'publisher_name': {'key': 'publisherName', 'type': 'str'}, + 'product_name': {'key': 'productName', 'type': 'str'}, + 'binary_name': {'key': 'binaryName', 'type': 'str'}, + 'version': {'key': 'version', 'type': 'str'}, } def __init__( self, *, - properties: Optional["ScanResultProperties"] = None, + publisher_name: Optional[str] = None, + product_name: Optional[str] = None, + binary_name: Optional[str] = None, + version: Optional[str] = None, **kwargs ): - super(ScanResult, self).__init__(**kwargs) - self.properties = properties + """ + :keyword publisher_name: The Subject field of the x.509 certificate used to sign the code, + using the following fields - O = Organization, L = Locality, S = State or Province, and C = + Country. + :paramtype publisher_name: str + :keyword product_name: The product name taken from the file's version resource. + :paramtype product_name: str + :keyword binary_name: The "OriginalName" field taken from the file's version resource. + :paramtype binary_name: str + :keyword version: The binary file version taken from the file's version resource. + :paramtype version: str + """ + super(PublisherInfo, self).__init__(**kwargs) + self.publisher_name = publisher_name + self.product_name = product_name + self.binary_name = binary_name + self.version = version -class ScanResultProperties(msrest.serialization.Model): - """A vulnerability assessment scan result properties for a single rule. +class QueryCheck(msrest.serialization.Model): + """The rule query details. - :param rule_id: The rule Id. - :type rule_id: str - :param status: The rule result status. Possible values include: "NonFinding", "Finding", - "InternalError". - :type status: str or ~azure.mgmt.security.models.RuleStatus - :param is_trimmed: Indicated whether the results specified here are trimmed. - :type is_trimmed: bool - :param query_results: The results of the query that was run. - :type query_results: list[list[str]] - :param remediation: Remediation details. - :type remediation: ~azure.mgmt.security.models.Remediation - :param baseline_adjusted_result: The rule result adjusted with baseline. - :type baseline_adjusted_result: ~azure.mgmt.security.models.BaselineAdjustedResult - :param rule_metadata: vulnerability assessment rule metadata details. - :type rule_metadata: ~azure.mgmt.security.models.VaRule + :ivar query: The rule query. + :vartype query: str + :ivar expected_result: Expected result. + :vartype expected_result: list[list[str]] + :ivar column_names: Column names of expected result. + :vartype column_names: list[str] """ _attribute_map = { - 'rule_id': {'key': 'ruleId', 'type': 'str'}, - 'status': {'key': 'status', 'type': 'str'}, - 'is_trimmed': {'key': 'isTrimmed', 'type': 'bool'}, - 'query_results': {'key': 'queryResults', 'type': '[[str]]'}, - 'remediation': {'key': 'remediation', 'type': 'Remediation'}, - 'baseline_adjusted_result': {'key': 'baselineAdjustedResult', 'type': 'BaselineAdjustedResult'}, - 'rule_metadata': {'key': 'ruleMetadata', 'type': 'VaRule'}, + 'query': {'key': 'query', 'type': 'str'}, + 'expected_result': {'key': 'expectedResult', 'type': '[[str]]'}, + 'column_names': {'key': 'columnNames', 'type': '[str]'}, } def __init__( self, *, - rule_id: Optional[str] = None, - status: Optional[Union[str, "RuleStatus"]] = None, - is_trimmed: Optional[bool] = None, - query_results: Optional[List[List[str]]] = None, - remediation: Optional["Remediation"] = None, - baseline_adjusted_result: Optional["BaselineAdjustedResult"] = None, - rule_metadata: Optional["VaRule"] = None, + query: Optional[str] = None, + expected_result: Optional[List[List[str]]] = None, + column_names: Optional[List[str]] = None, **kwargs ): - super(ScanResultProperties, self).__init__(**kwargs) - self.rule_id = rule_id - self.status = status - self.is_trimmed = is_trimmed - self.query_results = query_results - self.remediation = remediation - self.baseline_adjusted_result = baseline_adjusted_result - self.rule_metadata = rule_metadata + """ + :keyword query: The rule query. + :paramtype query: str + :keyword expected_result: Expected result. + :paramtype expected_result: list[list[str]] + :keyword column_names: Column names of expected result. + :paramtype column_names: list[str] + """ + super(QueryCheck, self).__init__(**kwargs) + self.query = query + self.expected_result = expected_result + self.column_names = column_names -class ScanResults(msrest.serialization.Model): - """A list of vulnerability assessment scan results. +class QueuePurgesNotInAllowedRange(TimeWindowCustomAlertRule): + """Number of device queue purges is not in allowed range. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. - :param value: List of vulnerability assessment scan results. - :type value: list[~azure.mgmt.security.models.ScanResult] + :ivar display_name: The display name of the custom alert. + :vartype display_name: str + :ivar description: The description of the custom alert. + :vartype description: str + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ + _validation = { + 'display_name': {'readonly': True}, + 'description': {'readonly': True}, + 'is_enabled': {'required': True}, + 'rule_type': {'required': True}, + 'min_threshold': {'required': True}, + 'max_threshold': {'required': True}, + 'time_window_size': {'required': True}, + } + _attribute_map = { - 'value': {'key': 'value', 'type': '[ScanResult]'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'is_enabled': {'key': 'isEnabled', 'type': 'bool'}, + 'rule_type': {'key': 'ruleType', 'type': 'str'}, + 'min_threshold': {'key': 'minThreshold', 'type': 'int'}, + 'max_threshold': {'key': 'maxThreshold', 'type': 'int'}, + 'time_window_size': {'key': 'timeWindowSize', 'type': 'duration'}, } def __init__( self, *, - value: Optional[List["ScanResult"]] = None, + is_enabled: bool, + min_threshold: int, + max_threshold: int, + time_window_size: datetime.timedelta, **kwargs ): - super(ScanResults, self).__init__(**kwargs) - self.value = value + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ + super(QueuePurgesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) + self.rule_type = 'QueuePurgesNotInAllowedRange' # type: str -class Scans(msrest.serialization.Model): - """A list of vulnerability assessment scan records. +class RecommendationConfigurationProperties(msrest.serialization.Model): + """The type of IoT Security recommendation. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. - :param value: List of vulnerability assessment scan records. - :type value: list[~azure.mgmt.security.models.Scan] + :ivar recommendation_type: Required. The type of IoT Security recommendation. Possible values + include: "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", + "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", + "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", + "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", + "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", + "IoT_VulnerableTLSCipherSuite". + :vartype recommendation_type: str or ~azure.mgmt.security.models.RecommendationType + :ivar name: + :vartype name: str + :ivar status: Required. Recommendation status. When the recommendation status is disabled + recommendations are not generated. Possible values include: "Disabled", "Enabled". Default + value: "Enabled". + :vartype status: str or ~azure.mgmt.security.models.RecommendationConfigStatus """ + _validation = { + 'recommendation_type': {'required': True}, + 'name': {'readonly': True}, + 'status': {'required': True}, + } + _attribute_map = { - 'value': {'key': 'value', 'type': '[Scan]'}, + 'recommendation_type': {'key': 'recommendationType', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'status': {'key': 'status', 'type': 'str'}, } def __init__( self, *, - value: Optional[List["Scan"]] = None, + recommendation_type: Union[str, "RecommendationType"], + status: Union[str, "RecommendationConfigStatus"] = "Enabled", **kwargs ): - super(Scans, self).__init__(**kwargs) - self.value = value + """ + :keyword recommendation_type: Required. The type of IoT Security recommendation. Possible + values include: "IoT_ACRAuthentication", "IoT_AgentSendsUnutilizedMessages", "IoT_Baseline", + "IoT_EdgeHubMemOptimize", "IoT_EdgeLoggingOptions", "IoT_InconsistentModuleSettings", + "IoT_InstallAgent", "IoT_IPFilter_DenyAll", "IoT_IPFilter_PermissiveRule", "IoT_OpenPorts", + "IoT_PermissiveFirewallPolicy", "IoT_PermissiveInputFirewallRules", + "IoT_PermissiveOutputFirewallRules", "IoT_PrivilegedDockerOptions", "IoT_SharedCredentials", + "IoT_VulnerableTLSCipherSuite". + :paramtype recommendation_type: str or ~azure.mgmt.security.models.RecommendationType + :keyword status: Required. Recommendation status. When the recommendation status is disabled + recommendations are not generated. Possible values include: "Disabled", "Enabled". Default + value: "Enabled". + :paramtype status: str or ~azure.mgmt.security.models.RecommendationConfigStatus + """ + super(RecommendationConfigurationProperties, self).__init__(**kwargs) + self.recommendation_type = recommendation_type + self.name = None + self.status = status -class ScopeElement(msrest.serialization.Model): - """A more specific scope used to identify the alerts to suppress. +class RegulatoryComplianceAssessment(Resource): + """Regulatory compliance assessment details and state. - :param additional_properties: Unmatched properties from the message are deserialized to this - collection. - :type additional_properties: dict[str, any] - :param field: The alert entity type to suppress by. - :type field: str + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar description: The description of the regulatory compliance assessment. + :vartype description: str + :ivar assessment_type: The expected type of assessment contained in the AssessmentDetailsLink. + :vartype assessment_type: str + :ivar assessment_details_link: Link to more detailed assessment results data. The response type + will be according to the assessmentType field. + :vartype assessment_details_link: str + :ivar state: Aggregative state based on the assessment's scanned resources states. Possible + values include: "Passed", "Failed", "Skipped", "Unsupported". + :vartype state: str or ~azure.mgmt.security.models.State + :ivar passed_resources: The given assessment's related resources count with passed state. + :vartype passed_resources: int + :ivar failed_resources: The given assessment's related resources count with failed state. + :vartype failed_resources: int + :ivar skipped_resources: The given assessment's related resources count with skipped state. + :vartype skipped_resources: int + :ivar unsupported_resources: The given assessment's related resources count with unsupported + state. + :vartype unsupported_resources: int """ + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'description': {'readonly': True}, + 'assessment_type': {'readonly': True}, + 'assessment_details_link': {'readonly': True}, + 'passed_resources': {'readonly': True}, + 'failed_resources': {'readonly': True}, + 'skipped_resources': {'readonly': True}, + 'unsupported_resources': {'readonly': True}, + } + _attribute_map = { - 'additional_properties': {'key': '', 'type': '{object}'}, - 'field': {'key': 'field', 'type': 'str'}, + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'assessment_type': {'key': 'properties.assessmentType', 'type': 'str'}, + 'assessment_details_link': {'key': 'properties.assessmentDetailsLink', 'type': 'str'}, + 'state': {'key': 'properties.state', 'type': 'str'}, + 'passed_resources': {'key': 'properties.passedResources', 'type': 'int'}, + 'failed_resources': {'key': 'properties.failedResources', 'type': 'int'}, + 'skipped_resources': {'key': 'properties.skippedResources', 'type': 'int'}, + 'unsupported_resources': {'key': 'properties.unsupportedResources', 'type': 'int'}, } def __init__( self, *, - additional_properties: Optional[Dict[str, Any]] = None, - field: Optional[str] = None, + state: Optional[Union[str, "State"]] = None, **kwargs ): - super(ScopeElement, self).__init__(**kwargs) - self.additional_properties = additional_properties - self.field = field + """ + :keyword state: Aggregative state based on the assessment's scanned resources states. Possible + values include: "Passed", "Failed", "Skipped", "Unsupported". + :paramtype state: str or ~azure.mgmt.security.models.State + """ + super(RegulatoryComplianceAssessment, self).__init__(**kwargs) + self.description = None + self.assessment_type = None + self.assessment_details_link = None + self.state = state + self.passed_resources = None + self.failed_resources = None + self.skipped_resources = None + self.unsupported_resources = None -class SecureScoreControlDefinitionItem(Resource): - """Information about the security control. +class RegulatoryComplianceAssessmentList(msrest.serialization.Model): + """List of regulatory compliance assessment response. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar value: Required. + :vartype value: list[~azure.mgmt.security.models.RegulatoryComplianceAssessment] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str + """ + + _validation = { + 'value': {'required': True}, + 'next_link': {'readonly': True}, + } + + _attribute_map = { + 'value': {'key': 'value', 'type': '[RegulatoryComplianceAssessment]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, + } + + def __init__( + self, + *, + value: List["RegulatoryComplianceAssessment"], + **kwargs + ): + """ + :keyword value: Required. + :paramtype value: list[~azure.mgmt.security.models.RegulatoryComplianceAssessment] + """ + super(RegulatoryComplianceAssessmentList, self).__init__(**kwargs) + self.value = value + self.next_link = None + + +class RegulatoryComplianceControl(Resource): + """Regulatory compliance control details and state. Variables are only populated by the server, and will be ignored when sending a request. @@ -7923,107 +9282,1789 @@ class SecureScoreControlDefinitionItem(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :ivar display_name: User friendly display name of the control. - :vartype display_name: str - :ivar description: User friendly description of the control. + :ivar description: The description of the regulatory compliance control. :vartype description: str - :ivar max_score: Maximum control score (0..10). - :vartype max_score: int - :ivar source: Source object from which the control was created. - :vartype source: ~azure.mgmt.security.models.SecureScoreControlDefinitionSource - :ivar assessment_definitions: Array of assessments metadata IDs that are included in this - security control. - :vartype assessment_definitions: list[~azure.mgmt.security.models.AzureResourceLink] + :ivar state: Aggregative state based on the control's supported assessments states. Possible + values include: "Passed", "Failed", "Skipped", "Unsupported". + :vartype state: str or ~azure.mgmt.security.models.State + :ivar passed_assessments: The number of supported regulatory compliance assessments of the + given control with a passed state. + :vartype passed_assessments: int + :ivar failed_assessments: The number of supported regulatory compliance assessments of the + given control with a failed state. + :vartype failed_assessments: int + :ivar skipped_assessments: The number of supported regulatory compliance assessments of the + given control with a skipped state. + :vartype skipped_assessments: int """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'display_name': {'readonly': True}, - 'description': {'readonly': True, 'max_length': 256, 'min_length': 0}, - 'max_score': {'readonly': True, 'maximum': 10, 'minimum': 0}, - 'source': {'readonly': True}, - 'assessment_definitions': {'readonly': True}, + 'description': {'readonly': True}, + 'passed_assessments': {'readonly': True}, + 'failed_assessments': {'readonly': True}, + 'skipped_assessments': {'readonly': True}, } _attribute_map = { 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, 'description': {'key': 'properties.description', 'type': 'str'}, - 'max_score': {'key': 'properties.maxScore', 'type': 'int'}, - 'source': {'key': 'properties.source', 'type': 'SecureScoreControlDefinitionSource'}, - 'assessment_definitions': {'key': 'properties.assessmentDefinitions', 'type': '[AzureResourceLink]'}, + 'state': {'key': 'properties.state', 'type': 'str'}, + 'passed_assessments': {'key': 'properties.passedAssessments', 'type': 'int'}, + 'failed_assessments': {'key': 'properties.failedAssessments', 'type': 'int'}, + 'skipped_assessments': {'key': 'properties.skippedAssessments', 'type': 'int'}, } def __init__( self, + *, + state: Optional[Union[str, "State"]] = None, **kwargs ): - super(SecureScoreControlDefinitionItem, self).__init__(**kwargs) - self.display_name = None + """ + :keyword state: Aggregative state based on the control's supported assessments states. Possible + values include: "Passed", "Failed", "Skipped", "Unsupported". + :paramtype state: str or ~azure.mgmt.security.models.State + """ + super(RegulatoryComplianceControl, self).__init__(**kwargs) self.description = None - self.max_score = None - self.source = None - self.assessment_definitions = None + self.state = state + self.passed_assessments = None + self.failed_assessments = None + self.skipped_assessments = None -class SecureScoreControlDefinitionList(msrest.serialization.Model): - """List of security controls definition. +class RegulatoryComplianceControlList(msrest.serialization.Model): + """List of regulatory compliance controls response. Variables are only populated by the server, and will be ignored when sending a request. - :ivar value: Collection of security controls definition in this page. - :vartype value: list[~azure.mgmt.security.models.SecureScoreControlDefinitionItem] + All required parameters must be populated in order to send to Azure. + + :ivar value: Required. List of regulatory compliance controls. + :vartype value: list[~azure.mgmt.security.models.RegulatoryComplianceControl] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ _validation = { - 'value': {'readonly': True}, + 'value': {'required': True}, 'next_link': {'readonly': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[SecureScoreControlDefinitionItem]'}, + 'value': {'key': 'value', 'type': '[RegulatoryComplianceControl]'}, 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, + *, + value: List["RegulatoryComplianceControl"], **kwargs ): - super(SecureScoreControlDefinitionList, self).__init__(**kwargs) - self.value = None + """ + :keyword value: Required. List of regulatory compliance controls. + :paramtype value: list[~azure.mgmt.security.models.RegulatoryComplianceControl] + """ + super(RegulatoryComplianceControlList, self).__init__(**kwargs) + self.value = value self.next_link = None -class SecureScoreControlDefinitionSource(msrest.serialization.Model): - """The type of the security control (For example, BuiltIn). +class RegulatoryComplianceStandard(Resource): + """Regulatory compliance standard details and state. - :param source_type: The type of security control (for example, BuiltIn). Possible values - include: "BuiltIn", "Custom". - :type source_type: str or ~azure.mgmt.security.models.ControlType + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar state: Aggregative state based on the standard's supported controls states. Possible + values include: "Passed", "Failed", "Skipped", "Unsupported". + :vartype state: str or ~azure.mgmt.security.models.State + :ivar passed_controls: The number of supported regulatory compliance controls of the given + standard with a passed state. + :vartype passed_controls: int + :ivar failed_controls: The number of supported regulatory compliance controls of the given + standard with a failed state. + :vartype failed_controls: int + :ivar skipped_controls: The number of supported regulatory compliance controls of the given + standard with a skipped state. + :vartype skipped_controls: int + :ivar unsupported_controls: The number of regulatory compliance controls of the given standard + which are unsupported by automated assessments. + :vartype unsupported_controls: int + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'passed_controls': {'readonly': True}, + 'failed_controls': {'readonly': True}, + 'skipped_controls': {'readonly': True}, + 'unsupported_controls': {'readonly': True}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'state': {'key': 'properties.state', 'type': 'str'}, + 'passed_controls': {'key': 'properties.passedControls', 'type': 'int'}, + 'failed_controls': {'key': 'properties.failedControls', 'type': 'int'}, + 'skipped_controls': {'key': 'properties.skippedControls', 'type': 'int'}, + 'unsupported_controls': {'key': 'properties.unsupportedControls', 'type': 'int'}, + } + + def __init__( + self, + *, + state: Optional[Union[str, "State"]] = None, + **kwargs + ): + """ + :keyword state: Aggregative state based on the standard's supported controls states. Possible + values include: "Passed", "Failed", "Skipped", "Unsupported". + :paramtype state: str or ~azure.mgmt.security.models.State + """ + super(RegulatoryComplianceStandard, self).__init__(**kwargs) + self.state = state + self.passed_controls = None + self.failed_controls = None + self.skipped_controls = None + self.unsupported_controls = None + + +class RegulatoryComplianceStandardList(msrest.serialization.Model): + """List of regulatory compliance standards response. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar value: Required. + :vartype value: list[~azure.mgmt.security.models.RegulatoryComplianceStandard] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str + """ + + _validation = { + 'value': {'required': True}, + 'next_link': {'readonly': True}, + } + + _attribute_map = { + 'value': {'key': 'value', 'type': '[RegulatoryComplianceStandard]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, + } + + def __init__( + self, + *, + value: List["RegulatoryComplianceStandard"], + **kwargs + ): + """ + :keyword value: Required. + :paramtype value: list[~azure.mgmt.security.models.RegulatoryComplianceStandard] + """ + super(RegulatoryComplianceStandardList, self).__init__(**kwargs) + self.value = value + self.next_link = None + + +class Remediation(msrest.serialization.Model): + """Remediation details. + + :ivar description: Remediation description. + :vartype description: str + :ivar scripts: Remediation script. + :vartype scripts: list[str] + :ivar automated: Is remediation automated. + :vartype automated: bool + :ivar portal_link: Optional link to remediate in Azure Portal. + :vartype portal_link: str + """ + + _attribute_map = { + 'description': {'key': 'description', 'type': 'str'}, + 'scripts': {'key': 'scripts', 'type': '[str]'}, + 'automated': {'key': 'automated', 'type': 'bool'}, + 'portal_link': {'key': 'portalLink', 'type': 'str'}, + } + + def __init__( + self, + *, + description: Optional[str] = None, + scripts: Optional[List[str]] = None, + automated: Optional[bool] = None, + portal_link: Optional[str] = None, + **kwargs + ): + """ + :keyword description: Remediation description. + :paramtype description: str + :keyword scripts: Remediation script. + :paramtype scripts: list[str] + :keyword automated: Is remediation automated. + :paramtype automated: bool + :keyword portal_link: Optional link to remediate in Azure Portal. + :paramtype portal_link: str + """ + super(Remediation, self).__init__(**kwargs) + self.description = description + self.scripts = scripts + self.automated = automated + self.portal_link = portal_link + + +class Rule(msrest.serialization.Model): + """Describes remote addresses that is recommended to communicate with the Azure resource on some (Protocol, Port, Direction). All other remote addresses are recommended to be blocked. + + :ivar name: The name of the rule. + :vartype name: str + :ivar direction: The rule's direction. Possible values include: "Inbound", "Outbound". + :vartype direction: str or ~azure.mgmt.security.models.Direction + :ivar destination_port: The rule's destination port. + :vartype destination_port: int + :ivar protocols: The rule's transport protocols. + :vartype protocols: list[str or ~azure.mgmt.security.models.TransportProtocol] + :ivar ip_addresses: The remote IP addresses that should be able to communicate with the Azure + resource on the rule's destination port and protocol. + :vartype ip_addresses: list[str] + """ + + _validation = { + 'destination_port': {'maximum': 65535, 'minimum': 0}, + } + + _attribute_map = { + 'name': {'key': 'name', 'type': 'str'}, + 'direction': {'key': 'direction', 'type': 'str'}, + 'destination_port': {'key': 'destinationPort', 'type': 'int'}, + 'protocols': {'key': 'protocols', 'type': '[str]'}, + 'ip_addresses': {'key': 'ipAddresses', 'type': '[str]'}, + } + + def __init__( + self, + *, + name: Optional[str] = None, + direction: Optional[Union[str, "Direction"]] = None, + destination_port: Optional[int] = None, + protocols: Optional[List[Union[str, "TransportProtocol"]]] = None, + ip_addresses: Optional[List[str]] = None, + **kwargs + ): + """ + :keyword name: The name of the rule. + :paramtype name: str + :keyword direction: The rule's direction. Possible values include: "Inbound", "Outbound". + :paramtype direction: str or ~azure.mgmt.security.models.Direction + :keyword destination_port: The rule's destination port. + :paramtype destination_port: int + :keyword protocols: The rule's transport protocols. + :paramtype protocols: list[str or ~azure.mgmt.security.models.TransportProtocol] + :keyword ip_addresses: The remote IP addresses that should be able to communicate with the + Azure resource on the rule's destination port and protocol. + :paramtype ip_addresses: list[str] + """ + super(Rule, self).__init__(**kwargs) + self.name = name + self.direction = direction + self.destination_port = destination_port + self.protocols = protocols + self.ip_addresses = ip_addresses + + +class RuleResults(Resource): + """Rule results. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar properties: Rule results properties. + :vartype properties: ~azure.mgmt.security.models.RuleResultsProperties + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'properties': {'key': 'properties', 'type': 'RuleResultsProperties'}, + } + + def __init__( + self, + *, + properties: Optional["RuleResultsProperties"] = None, + **kwargs + ): + """ + :keyword properties: Rule results properties. + :paramtype properties: ~azure.mgmt.security.models.RuleResultsProperties + """ + super(RuleResults, self).__init__(**kwargs) + self.properties = properties + + +class RuleResultsInput(msrest.serialization.Model): + """Rule results input. + + :ivar latest_scan: Take results from latest scan. + :vartype latest_scan: bool + :ivar results: Expected results to be inserted into the baseline. + Leave this field empty it LatestScan == true. + :vartype results: list[list[str]] + """ + + _attribute_map = { + 'latest_scan': {'key': 'latestScan', 'type': 'bool'}, + 'results': {'key': 'results', 'type': '[[str]]'}, + } + + def __init__( + self, + *, + latest_scan: Optional[bool] = None, + results: Optional[List[List[str]]] = None, + **kwargs + ): + """ + :keyword latest_scan: Take results from latest scan. + :paramtype latest_scan: bool + :keyword results: Expected results to be inserted into the baseline. + Leave this field empty it LatestScan == true. + :paramtype results: list[list[str]] + """ + super(RuleResultsInput, self).__init__(**kwargs) + self.latest_scan = latest_scan + self.results = results + + +class RuleResultsProperties(msrest.serialization.Model): + """Rule results properties. + + :ivar results: Expected results in the baseline. + :vartype results: list[list[str]] + """ + + _attribute_map = { + 'results': {'key': 'results', 'type': '[[str]]'}, + } + + def __init__( + self, + *, + results: Optional[List[List[str]]] = None, + **kwargs + ): + """ + :keyword results: Expected results in the baseline. + :paramtype results: list[list[str]] + """ + super(RuleResultsProperties, self).__init__(**kwargs) + self.results = results + + +class RulesResults(msrest.serialization.Model): + """A list of rules results. + + :ivar value: List of rule results. + :vartype value: list[~azure.mgmt.security.models.RuleResults] + """ + + _attribute_map = { + 'value': {'key': 'value', 'type': '[RuleResults]'}, + } + + def __init__( + self, + *, + value: Optional[List["RuleResults"]] = None, + **kwargs + ): + """ + :keyword value: List of rule results. + :paramtype value: list[~azure.mgmt.security.models.RuleResults] + """ + super(RulesResults, self).__init__(**kwargs) + self.value = value + + +class RulesResultsInput(msrest.serialization.Model): + """Rules results input. + + :ivar latest_scan: Take results from latest scan. + :vartype latest_scan: bool + :ivar results: Expected results to be inserted into the baseline. + Leave this field empty it LatestScan == true. + :vartype results: dict[str, list[list[str]]] + """ + + _attribute_map = { + 'latest_scan': {'key': 'latestScan', 'type': 'bool'}, + 'results': {'key': 'results', 'type': '{[[str]]}'}, + } + + def __init__( + self, + *, + latest_scan: Optional[bool] = None, + results: Optional[Dict[str, List[List[str]]]] = None, + **kwargs + ): + """ + :keyword latest_scan: Take results from latest scan. + :paramtype latest_scan: bool + :keyword results: Expected results to be inserted into the baseline. + Leave this field empty it LatestScan == true. + :paramtype results: dict[str, list[list[str]]] + """ + super(RulesResultsInput, self).__init__(**kwargs) + self.latest_scan = latest_scan + self.results = results + + +class Scan(Resource): + """A vulnerability assessment scan record. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar properties: A vulnerability assessment scan record properties. + :vartype properties: ~azure.mgmt.security.models.ScanProperties + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'properties': {'key': 'properties', 'type': 'ScanProperties'}, + } + + def __init__( + self, + *, + properties: Optional["ScanProperties"] = None, + **kwargs + ): + """ + :keyword properties: A vulnerability assessment scan record properties. + :paramtype properties: ~azure.mgmt.security.models.ScanProperties + """ + super(Scan, self).__init__(**kwargs) + self.properties = properties + + +class ScanProperties(msrest.serialization.Model): + """A vulnerability assessment scan record properties. + + :ivar trigger_type: The scan trigger type. Possible values include: "OnDemand", "Recurring". + :vartype trigger_type: str or ~azure.mgmt.security.models.ScanTriggerType + :ivar state: The scan status. Possible values include: "Failed", "FailedToRun", "InProgress", + "Passed". + :vartype state: str or ~azure.mgmt.security.models.ScanState + :ivar server: The server name. + :vartype server: str + :ivar database: The database name. + :vartype database: str + :ivar sql_version: The SQL version. + :vartype sql_version: str + :ivar start_time: The scan start time (UTC). + :vartype start_time: ~datetime.datetime + :ivar end_time: Scan results are valid until end time (UTC). + :vartype end_time: ~datetime.datetime + :ivar high_severity_failed_rules_count: The number of failed rules with high severity. + :vartype high_severity_failed_rules_count: int + :ivar medium_severity_failed_rules_count: The number of failed rules with medium severity. + :vartype medium_severity_failed_rules_count: int + :ivar low_severity_failed_rules_count: The number of failed rules with low severity. + :vartype low_severity_failed_rules_count: int + :ivar total_passed_rules_count: The number of total passed rules. + :vartype total_passed_rules_count: int + :ivar total_failed_rules_count: The number of total failed rules. + :vartype total_failed_rules_count: int + :ivar total_rules_count: The number of total rules assessed. + :vartype total_rules_count: int + :ivar is_baseline_applied: Baseline created for this database, and has one or more rules. + :vartype is_baseline_applied: bool + """ + + _attribute_map = { + 'trigger_type': {'key': 'triggerType', 'type': 'str'}, + 'state': {'key': 'state', 'type': 'str'}, + 'server': {'key': 'server', 'type': 'str'}, + 'database': {'key': 'database', 'type': 'str'}, + 'sql_version': {'key': 'sqlVersion', 'type': 'str'}, + 'start_time': {'key': 'startTime', 'type': 'iso-8601'}, + 'end_time': {'key': 'endTime', 'type': 'iso-8601'}, + 'high_severity_failed_rules_count': {'key': 'highSeverityFailedRulesCount', 'type': 'int'}, + 'medium_severity_failed_rules_count': {'key': 'mediumSeverityFailedRulesCount', 'type': 'int'}, + 'low_severity_failed_rules_count': {'key': 'lowSeverityFailedRulesCount', 'type': 'int'}, + 'total_passed_rules_count': {'key': 'totalPassedRulesCount', 'type': 'int'}, + 'total_failed_rules_count': {'key': 'totalFailedRulesCount', 'type': 'int'}, + 'total_rules_count': {'key': 'totalRulesCount', 'type': 'int'}, + 'is_baseline_applied': {'key': 'isBaselineApplied', 'type': 'bool'}, + } + + def __init__( + self, + *, + trigger_type: Optional[Union[str, "ScanTriggerType"]] = None, + state: Optional[Union[str, "ScanState"]] = None, + server: Optional[str] = None, + database: Optional[str] = None, + sql_version: Optional[str] = None, + start_time: Optional[datetime.datetime] = None, + end_time: Optional[datetime.datetime] = None, + high_severity_failed_rules_count: Optional[int] = None, + medium_severity_failed_rules_count: Optional[int] = None, + low_severity_failed_rules_count: Optional[int] = None, + total_passed_rules_count: Optional[int] = None, + total_failed_rules_count: Optional[int] = None, + total_rules_count: Optional[int] = None, + is_baseline_applied: Optional[bool] = None, + **kwargs + ): + """ + :keyword trigger_type: The scan trigger type. Possible values include: "OnDemand", "Recurring". + :paramtype trigger_type: str or ~azure.mgmt.security.models.ScanTriggerType + :keyword state: The scan status. Possible values include: "Failed", "FailedToRun", + "InProgress", "Passed". + :paramtype state: str or ~azure.mgmt.security.models.ScanState + :keyword server: The server name. + :paramtype server: str + :keyword database: The database name. + :paramtype database: str + :keyword sql_version: The SQL version. + :paramtype sql_version: str + :keyword start_time: The scan start time (UTC). + :paramtype start_time: ~datetime.datetime + :keyword end_time: Scan results are valid until end time (UTC). + :paramtype end_time: ~datetime.datetime + :keyword high_severity_failed_rules_count: The number of failed rules with high severity. + :paramtype high_severity_failed_rules_count: int + :keyword medium_severity_failed_rules_count: The number of failed rules with medium severity. + :paramtype medium_severity_failed_rules_count: int + :keyword low_severity_failed_rules_count: The number of failed rules with low severity. + :paramtype low_severity_failed_rules_count: int + :keyword total_passed_rules_count: The number of total passed rules. + :paramtype total_passed_rules_count: int + :keyword total_failed_rules_count: The number of total failed rules. + :paramtype total_failed_rules_count: int + :keyword total_rules_count: The number of total rules assessed. + :paramtype total_rules_count: int + :keyword is_baseline_applied: Baseline created for this database, and has one or more rules. + :paramtype is_baseline_applied: bool + """ + super(ScanProperties, self).__init__(**kwargs) + self.trigger_type = trigger_type + self.state = state + self.server = server + self.database = database + self.sql_version = sql_version + self.start_time = start_time + self.end_time = end_time + self.high_severity_failed_rules_count = high_severity_failed_rules_count + self.medium_severity_failed_rules_count = medium_severity_failed_rules_count + self.low_severity_failed_rules_count = low_severity_failed_rules_count + self.total_passed_rules_count = total_passed_rules_count + self.total_failed_rules_count = total_failed_rules_count + self.total_rules_count = total_rules_count + self.is_baseline_applied = is_baseline_applied + + +class ScanResult(Resource): + """A vulnerability assessment scan result for a single rule. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar properties: A vulnerability assessment scan result properties for a single rule. + :vartype properties: ~azure.mgmt.security.models.ScanResultProperties + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'properties': {'key': 'properties', 'type': 'ScanResultProperties'}, + } + + def __init__( + self, + *, + properties: Optional["ScanResultProperties"] = None, + **kwargs + ): + """ + :keyword properties: A vulnerability assessment scan result properties for a single rule. + :paramtype properties: ~azure.mgmt.security.models.ScanResultProperties + """ + super(ScanResult, self).__init__(**kwargs) + self.properties = properties + + +class ScanResultProperties(msrest.serialization.Model): + """A vulnerability assessment scan result properties for a single rule. + + :ivar rule_id: The rule Id. + :vartype rule_id: str + :ivar status: The rule result status. Possible values include: "NonFinding", "Finding", + "InternalError". + :vartype status: str or ~azure.mgmt.security.models.RuleStatus + :ivar is_trimmed: Indicated whether the results specified here are trimmed. + :vartype is_trimmed: bool + :ivar query_results: The results of the query that was run. + :vartype query_results: list[list[str]] + :ivar remediation: Remediation details. + :vartype remediation: ~azure.mgmt.security.models.Remediation + :ivar baseline_adjusted_result: The rule result adjusted with baseline. + :vartype baseline_adjusted_result: ~azure.mgmt.security.models.BaselineAdjustedResult + :ivar rule_metadata: vulnerability assessment rule metadata details. + :vartype rule_metadata: ~azure.mgmt.security.models.VaRule + """ + + _attribute_map = { + 'rule_id': {'key': 'ruleId', 'type': 'str'}, + 'status': {'key': 'status', 'type': 'str'}, + 'is_trimmed': {'key': 'isTrimmed', 'type': 'bool'}, + 'query_results': {'key': 'queryResults', 'type': '[[str]]'}, + 'remediation': {'key': 'remediation', 'type': 'Remediation'}, + 'baseline_adjusted_result': {'key': 'baselineAdjustedResult', 'type': 'BaselineAdjustedResult'}, + 'rule_metadata': {'key': 'ruleMetadata', 'type': 'VaRule'}, + } + + def __init__( + self, + *, + rule_id: Optional[str] = None, + status: Optional[Union[str, "RuleStatus"]] = None, + is_trimmed: Optional[bool] = None, + query_results: Optional[List[List[str]]] = None, + remediation: Optional["Remediation"] = None, + baseline_adjusted_result: Optional["BaselineAdjustedResult"] = None, + rule_metadata: Optional["VaRule"] = None, + **kwargs + ): + """ + :keyword rule_id: The rule Id. + :paramtype rule_id: str + :keyword status: The rule result status. Possible values include: "NonFinding", "Finding", + "InternalError". + :paramtype status: str or ~azure.mgmt.security.models.RuleStatus + :keyword is_trimmed: Indicated whether the results specified here are trimmed. + :paramtype is_trimmed: bool + :keyword query_results: The results of the query that was run. + :paramtype query_results: list[list[str]] + :keyword remediation: Remediation details. + :paramtype remediation: ~azure.mgmt.security.models.Remediation + :keyword baseline_adjusted_result: The rule result adjusted with baseline. + :paramtype baseline_adjusted_result: ~azure.mgmt.security.models.BaselineAdjustedResult + :keyword rule_metadata: vulnerability assessment rule metadata details. + :paramtype rule_metadata: ~azure.mgmt.security.models.VaRule + """ + super(ScanResultProperties, self).__init__(**kwargs) + self.rule_id = rule_id + self.status = status + self.is_trimmed = is_trimmed + self.query_results = query_results + self.remediation = remediation + self.baseline_adjusted_result = baseline_adjusted_result + self.rule_metadata = rule_metadata + + +class ScanResults(msrest.serialization.Model): + """A list of vulnerability assessment scan results. + + :ivar value: List of vulnerability assessment scan results. + :vartype value: list[~azure.mgmt.security.models.ScanResult] + """ + + _attribute_map = { + 'value': {'key': 'value', 'type': '[ScanResult]'}, + } + + def __init__( + self, + *, + value: Optional[List["ScanResult"]] = None, + **kwargs + ): + """ + :keyword value: List of vulnerability assessment scan results. + :paramtype value: list[~azure.mgmt.security.models.ScanResult] + """ + super(ScanResults, self).__init__(**kwargs) + self.value = value + + +class Scans(msrest.serialization.Model): + """A list of vulnerability assessment scan records. + + :ivar value: List of vulnerability assessment scan records. + :vartype value: list[~azure.mgmt.security.models.Scan] + """ + + _attribute_map = { + 'value': {'key': 'value', 'type': '[Scan]'}, + } + + def __init__( + self, + *, + value: Optional[List["Scan"]] = None, + **kwargs + ): + """ + :keyword value: List of vulnerability assessment scan records. + :paramtype value: list[~azure.mgmt.security.models.Scan] + """ + super(Scans, self).__init__(**kwargs) + self.value = value + + +class ScopeElement(msrest.serialization.Model): + """A more specific scope used to identify the alerts to suppress. + + :ivar additional_properties: Unmatched properties from the message are deserialized to this + collection. + :vartype additional_properties: dict[str, any] + :ivar field: The alert entity type to suppress by. + :vartype field: str + """ + + _attribute_map = { + 'additional_properties': {'key': '', 'type': '{object}'}, + 'field': {'key': 'field', 'type': 'str'}, + } + + def __init__( + self, + *, + additional_properties: Optional[Dict[str, Any]] = None, + field: Optional[str] = None, + **kwargs + ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + :keyword field: The alert entity type to suppress by. + :paramtype field: str + """ + super(ScopeElement, self).__init__(**kwargs) + self.additional_properties = additional_properties + self.field = field + + +class SecureScoreControlDefinitionItem(Resource): + """Information about the security control. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar display_name: User friendly display name of the control. + :vartype display_name: str + :ivar description: User friendly description of the control. + :vartype description: str + :ivar max_score: Maximum control score (0..10). + :vartype max_score: int + :ivar source: Source object from which the control was created. + :vartype source: ~azure.mgmt.security.models.SecureScoreControlDefinitionSource + :ivar assessment_definitions: Array of assessments metadata IDs that are included in this + security control. + :vartype assessment_definitions: list[~azure.mgmt.security.models.AzureResourceLink] + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'display_name': {'readonly': True}, + 'description': {'readonly': True, 'max_length': 256, 'min_length': 0}, + 'max_score': {'readonly': True, 'maximum': 10, 'minimum': 0}, + 'source': {'readonly': True}, + 'assessment_definitions': {'readonly': True}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'max_score': {'key': 'properties.maxScore', 'type': 'int'}, + 'source': {'key': 'properties.source', 'type': 'SecureScoreControlDefinitionSource'}, + 'assessment_definitions': {'key': 'properties.assessmentDefinitions', 'type': '[AzureResourceLink]'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(SecureScoreControlDefinitionItem, self).__init__(**kwargs) + self.display_name = None + self.description = None + self.max_score = None + self.source = None + self.assessment_definitions = None + + +class SecureScoreControlDefinitionList(msrest.serialization.Model): + """List of security controls definition. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: Collection of security controls definition in this page. + :vartype value: list[~azure.mgmt.security.models.SecureScoreControlDefinitionItem] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str + """ + + _validation = { + 'value': {'readonly': True}, + 'next_link': {'readonly': True}, + } + + _attribute_map = { + 'value': {'key': 'value', 'type': '[SecureScoreControlDefinitionItem]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(SecureScoreControlDefinitionList, self).__init__(**kwargs) + self.value = None + self.next_link = None + + +class SecureScoreControlDefinitionSource(msrest.serialization.Model): + """The type of the security control (For example, BuiltIn). + + :ivar source_type: The type of security control (for example, BuiltIn). Possible values + include: "BuiltIn", "Custom". + :vartype source_type: str or ~azure.mgmt.security.models.ControlType + """ + + _attribute_map = { + 'source_type': {'key': 'sourceType', 'type': 'str'}, + } + + def __init__( + self, + *, + source_type: Optional[Union[str, "ControlType"]] = None, + **kwargs + ): + """ + :keyword source_type: The type of security control (for example, BuiltIn). Possible values + include: "BuiltIn", "Custom". + :paramtype source_type: str or ~azure.mgmt.security.models.ControlType + """ + super(SecureScoreControlDefinitionSource, self).__init__(**kwargs) + self.source_type = source_type + + +class SecureScoreControlDetails(Resource): + """Details of the security control, its score, and the health status of the relevant resources. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar display_name: User friendly display name of the control. + :vartype display_name: str + :ivar healthy_resource_count: Number of healthy resources in the control. + :vartype healthy_resource_count: int + :ivar unhealthy_resource_count: Number of unhealthy resources in the control. + :vartype unhealthy_resource_count: int + :ivar not_applicable_resource_count: Number of not applicable resources in the control. + :vartype not_applicable_resource_count: int + :ivar weight: The relative weight for this specific control in each of your subscriptions. Used + when calculating an aggregated score for this control across all of your subscriptions. + :vartype weight: long + :ivar definition: Information about the security control. + :vartype definition: ~azure.mgmt.security.models.SecureScoreControlDefinitionItem + :ivar max: Maximum score available. + :vartype max: int + :ivar current: Current score. + :vartype current: float + :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after + the decimal point. + :vartype percentage: float + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'display_name': {'readonly': True}, + 'healthy_resource_count': {'readonly': True}, + 'unhealthy_resource_count': {'readonly': True}, + 'not_applicable_resource_count': {'readonly': True}, + 'weight': {'readonly': True, 'minimum': 0}, + 'max': {'readonly': True, 'minimum': 0}, + 'current': {'readonly': True, 'minimum': 0}, + 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'healthy_resource_count': {'key': 'properties.healthyResourceCount', 'type': 'int'}, + 'unhealthy_resource_count': {'key': 'properties.unhealthyResourceCount', 'type': 'int'}, + 'not_applicable_resource_count': {'key': 'properties.notApplicableResourceCount', 'type': 'int'}, + 'weight': {'key': 'properties.weight', 'type': 'long'}, + 'definition': {'key': 'properties.definition', 'type': 'SecureScoreControlDefinitionItem'}, + 'max': {'key': 'properties.score.max', 'type': 'int'}, + 'current': {'key': 'properties.score.current', 'type': 'float'}, + 'percentage': {'key': 'properties.score.percentage', 'type': 'float'}, + } + + def __init__( + self, + *, + definition: Optional["SecureScoreControlDefinitionItem"] = None, + **kwargs + ): + """ + :keyword definition: Information about the security control. + :paramtype definition: ~azure.mgmt.security.models.SecureScoreControlDefinitionItem + """ + super(SecureScoreControlDetails, self).__init__(**kwargs) + self.display_name = None + self.healthy_resource_count = None + self.unhealthy_resource_count = None + self.not_applicable_resource_count = None + self.weight = None + self.definition = definition + self.max = None + self.current = None + self.percentage = None + + +class SecureScoreControlList(msrest.serialization.Model): + """List of security controls. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: Collection of security controls in this page. + :vartype value: list[~azure.mgmt.security.models.SecureScoreControlDetails] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str + """ + + _validation = { + 'value': {'readonly': True}, + 'next_link': {'readonly': True}, + } + + _attribute_map = { + 'value': {'key': 'value', 'type': '[SecureScoreControlDetails]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(SecureScoreControlList, self).__init__(**kwargs) + self.value = None + self.next_link = None + + +class SecureScoreControlScore(msrest.serialization.Model): + """Calculation result data. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar max: Maximum control score (0..10). + :vartype max: int + :ivar current: Actual score for the control = (achieved points / total points) * max score. if + total points is zeroed, the return number is 0.00. + :vartype current: float + :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after + the decimal point. + :vartype percentage: float + """ + + _validation = { + 'max': {'readonly': True, 'maximum': 10, 'minimum': 0}, + 'current': {'readonly': True, 'maximum': 10, 'minimum': 0}, + 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, + } + + _attribute_map = { + 'max': {'key': 'max', 'type': 'int'}, + 'current': {'key': 'current', 'type': 'float'}, + 'percentage': {'key': 'percentage', 'type': 'float'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(SecureScoreControlScore, self).__init__(**kwargs) + self.max = None + self.current = None + self.percentage = None + + +class SecureScoreItem(Resource): + """Secure score item data model. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar display_name: The initiative’s name. + :vartype display_name: str + :ivar weight: The relative weight for each subscription. Used when calculating an aggregated + secure score for multiple subscriptions. + :vartype weight: long + :ivar max: Maximum score available. + :vartype max: int + :ivar current: Current score. + :vartype current: float + :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after + the decimal point. + :vartype percentage: float + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'display_name': {'readonly': True}, + 'weight': {'readonly': True, 'minimum': 0}, + 'max': {'readonly': True, 'minimum': 0}, + 'current': {'readonly': True, 'minimum': 0}, + 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'weight': {'key': 'properties.weight', 'type': 'long'}, + 'max': {'key': 'properties.score.max', 'type': 'int'}, + 'current': {'key': 'properties.score.current', 'type': 'float'}, + 'percentage': {'key': 'properties.score.percentage', 'type': 'float'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(SecureScoreItem, self).__init__(**kwargs) + self.display_name = None + self.weight = None + self.max = None + self.current = None + self.percentage = None + + +class SecureScoresList(msrest.serialization.Model): + """List of secure scores. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: Collection of secure scores in this page. + :vartype value: list[~azure.mgmt.security.models.SecureScoreItem] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str + """ + + _validation = { + 'value': {'readonly': True}, + 'next_link': {'readonly': True}, + } + + _attribute_map = { + 'value': {'key': 'value', 'type': '[SecureScoreItem]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(SecureScoresList, self).__init__(**kwargs) + self.value = None + self.next_link = None + + +class SecurityAssessment(Resource): + """Security assessment on a resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar resource_details: Details of the resource that was assessed. + :vartype resource_details: ~azure.mgmt.security.models.ResourceDetails + :ivar display_name: User friendly display name of the assessment. + :vartype display_name: str + :ivar additional_data: Additional data regarding the assessment. + :vartype additional_data: dict[str, str] + :ivar links: Links relevant to the assessment. + :vartype links: ~azure.mgmt.security.models.AssessmentLinks + :ivar metadata: Describes properties of an assessment metadata. + :vartype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :ivar partners_data: Data regarding 3rd party partner integration. + :vartype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :ivar status: The result of the assessment. + :vartype status: ~azure.mgmt.security.models.AssessmentStatus + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'display_name': {'readonly': True}, + 'links': {'readonly': True}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'resource_details': {'key': 'properties.resourceDetails', 'type': 'ResourceDetails'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'additional_data': {'key': 'properties.additionalData', 'type': '{str}'}, + 'links': {'key': 'properties.links', 'type': 'AssessmentLinks'}, + 'metadata': {'key': 'properties.metadata', 'type': 'SecurityAssessmentMetadataProperties'}, + 'partners_data': {'key': 'properties.partnersData', 'type': 'SecurityAssessmentPartnerData'}, + 'status': {'key': 'properties.status', 'type': 'AssessmentStatus'}, + } + + def __init__( + self, + *, + resource_details: Optional["ResourceDetails"] = None, + additional_data: Optional[Dict[str, str]] = None, + metadata: Optional["SecurityAssessmentMetadataProperties"] = None, + partners_data: Optional["SecurityAssessmentPartnerData"] = None, + status: Optional["AssessmentStatus"] = None, + **kwargs + ): + """ + :keyword resource_details: Details of the resource that was assessed. + :paramtype resource_details: ~azure.mgmt.security.models.ResourceDetails + :keyword additional_data: Additional data regarding the assessment. + :paramtype additional_data: dict[str, str] + :keyword metadata: Describes properties of an assessment metadata. + :paramtype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :keyword partners_data: Data regarding 3rd party partner integration. + :paramtype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :keyword status: The result of the assessment. + :paramtype status: ~azure.mgmt.security.models.AssessmentStatus + """ + super(SecurityAssessment, self).__init__(**kwargs) + self.resource_details = resource_details + self.display_name = None + self.additional_data = additional_data + self.links = None + self.metadata = metadata + self.partners_data = partners_data + self.status = status + + +class SecurityAssessmentList(msrest.serialization.Model): + """Page of a security assessments list. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: Collection of security assessments in this page. + :vartype value: list[~azure.mgmt.security.models.SecurityAssessmentResponse] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str + """ + + _validation = { + 'value': {'readonly': True}, + 'next_link': {'readonly': True}, + } + + _attribute_map = { + 'value': {'key': 'value', 'type': '[SecurityAssessmentResponse]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, + } + + def __init__( + self, + **kwargs + ): + """ + """ + super(SecurityAssessmentList, self).__init__(**kwargs) + self.value = None + self.next_link = None + + +class SecurityAssessmentMetadata(Resource): + """Security assessment metadata. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Resource Id. + :vartype id: str + :ivar name: Resource name. + :vartype name: str + :ivar type: Resource type. + :vartype type: str + :ivar display_name: User friendly display name of the assessment. + :vartype display_name: str + :ivar policy_definition_id: Azure resource ID of the policy definition that turns this + assessment calculation on. + :vartype policy_definition_id: str + :ivar description: Human readable description of the assessment. + :vartype description: str + :ivar remediation_description: Human readable description of what you should do to mitigate + this security issue. + :vartype remediation_description: str + :ivar categories: + :vartype categories: list[str or ~azure.mgmt.security.models.Categories] + :ivar severity: The severity level of the assessment. Possible values include: "Low", "Medium", + "High". + :vartype severity: str or ~azure.mgmt.security.models.Severity + :ivar user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :vartype user_impact: str or ~azure.mgmt.security.models.UserImpact + :ivar implementation_effort: The implementation effort required to remediate this assessment. + Possible values include: "Low", "Moderate", "High". + :vartype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :ivar threats: + :vartype threats: list[str or ~azure.mgmt.security.models.Threats] + :ivar preview: True if this assessment is in preview release status. + :vartype preview: bool + :ivar assessment_type: BuiltIn if the assessment based on built-in Azure Policy definition, + Custom if the assessment based on custom Azure Policy definition. Possible values include: + "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :vartype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :ivar partner_data: Describes the partner that created the assessment. + :vartype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + """ + + _validation = { + 'id': {'readonly': True}, + 'name': {'readonly': True}, + 'type': {'readonly': True}, + 'policy_definition_id': {'readonly': True}, + } + + _attribute_map = { + 'id': {'key': 'id', 'type': 'str'}, + 'name': {'key': 'name', 'type': 'str'}, + 'type': {'key': 'type', 'type': 'str'}, + 'display_name': {'key': 'properties.displayName', 'type': 'str'}, + 'policy_definition_id': {'key': 'properties.policyDefinitionId', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'remediation_description': {'key': 'properties.remediationDescription', 'type': 'str'}, + 'categories': {'key': 'properties.categories', 'type': '[str]'}, + 'severity': {'key': 'properties.severity', 'type': 'str'}, + 'user_impact': {'key': 'properties.userImpact', 'type': 'str'}, + 'implementation_effort': {'key': 'properties.implementationEffort', 'type': 'str'}, + 'threats': {'key': 'properties.threats', 'type': '[str]'}, + 'preview': {'key': 'properties.preview', 'type': 'bool'}, + 'assessment_type': {'key': 'properties.assessmentType', 'type': 'str'}, + 'partner_data': {'key': 'properties.partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, + } + + def __init__( + self, + *, + display_name: Optional[str] = None, + description: Optional[str] = None, + remediation_description: Optional[str] = None, + categories: Optional[List[Union[str, "Categories"]]] = None, + severity: Optional[Union[str, "Severity"]] = None, + user_impact: Optional[Union[str, "UserImpact"]] = None, + implementation_effort: Optional[Union[str, "ImplementationEffort"]] = None, + threats: Optional[List[Union[str, "Threats"]]] = None, + preview: Optional[bool] = None, + assessment_type: Optional[Union[str, "AssessmentType"]] = None, + partner_data: Optional["SecurityAssessmentMetadataPartnerData"] = None, + **kwargs + ): + """ + :keyword display_name: User friendly display name of the assessment. + :paramtype display_name: str + :keyword description: Human readable description of the assessment. + :paramtype description: str + :keyword remediation_description: Human readable description of what you should do to mitigate + this security issue. + :paramtype remediation_description: str + :keyword categories: + :paramtype categories: list[str or ~azure.mgmt.security.models.Categories] + :keyword severity: The severity level of the assessment. Possible values include: "Low", + "Medium", "High". + :paramtype severity: str or ~azure.mgmt.security.models.Severity + :keyword user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :paramtype user_impact: str or ~azure.mgmt.security.models.UserImpact + :keyword implementation_effort: The implementation effort required to remediate this + assessment. Possible values include: "Low", "Moderate", "High". + :paramtype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :keyword threats: + :paramtype threats: list[str or ~azure.mgmt.security.models.Threats] + :keyword preview: True if this assessment is in preview release status. + :paramtype preview: bool + :keyword assessment_type: BuiltIn if the assessment based on built-in Azure Policy definition, + Custom if the assessment based on custom Azure Policy definition. Possible values include: + "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :paramtype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :keyword partner_data: Describes the partner that created the assessment. + :paramtype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + """ + super(SecurityAssessmentMetadata, self).__init__(**kwargs) + self.display_name = display_name + self.policy_definition_id = None + self.description = description + self.remediation_description = remediation_description + self.categories = categories + self.severity = severity + self.user_impact = user_impact + self.implementation_effort = implementation_effort + self.threats = threats + self.preview = preview + self.assessment_type = assessment_type + self.partner_data = partner_data + + +class SecurityAssessmentMetadataPartnerData(msrest.serialization.Model): + """Describes the partner that created the assessment. + + All required parameters must be populated in order to send to Azure. + + :ivar partner_name: Required. Name of the company of the partner. + :vartype partner_name: str + :ivar product_name: Name of the product of the partner that created the assessment. + :vartype product_name: str + :ivar secret: Required. Secret to authenticate the partner and verify it created the assessment + - write only. + :vartype secret: str + """ + + _validation = { + 'partner_name': {'required': True}, + 'secret': {'required': True}, + } + + _attribute_map = { + 'partner_name': {'key': 'partnerName', 'type': 'str'}, + 'product_name': {'key': 'productName', 'type': 'str'}, + 'secret': {'key': 'secret', 'type': 'str'}, + } + + def __init__( + self, + *, + partner_name: str, + secret: str, + product_name: Optional[str] = None, + **kwargs + ): + """ + :keyword partner_name: Required. Name of the company of the partner. + :paramtype partner_name: str + :keyword product_name: Name of the product of the partner that created the assessment. + :paramtype product_name: str + :keyword secret: Required. Secret to authenticate the partner and verify it created the + assessment - write only. + :paramtype secret: str + """ + super(SecurityAssessmentMetadataPartnerData, self).__init__(**kwargs) + self.partner_name = partner_name + self.product_name = product_name + self.secret = secret + + +class SecurityAssessmentMetadataProperties(msrest.serialization.Model): + """Describes properties of an assessment metadata. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar display_name: Required. User friendly display name of the assessment. + :vartype display_name: str + :ivar policy_definition_id: Azure resource ID of the policy definition that turns this + assessment calculation on. + :vartype policy_definition_id: str + :ivar description: Human readable description of the assessment. + :vartype description: str + :ivar remediation_description: Human readable description of what you should do to mitigate + this security issue. + :vartype remediation_description: str + :ivar categories: + :vartype categories: list[str or ~azure.mgmt.security.models.Categories] + :ivar severity: Required. The severity level of the assessment. Possible values include: "Low", + "Medium", "High". + :vartype severity: str or ~azure.mgmt.security.models.Severity + :ivar user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :vartype user_impact: str or ~azure.mgmt.security.models.UserImpact + :ivar implementation_effort: The implementation effort required to remediate this assessment. + Possible values include: "Low", "Moderate", "High". + :vartype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :ivar threats: + :vartype threats: list[str or ~azure.mgmt.security.models.Threats] + :ivar preview: True if this assessment is in preview release status. + :vartype preview: bool + :ivar assessment_type: Required. BuiltIn if the assessment based on built-in Azure Policy + definition, Custom if the assessment based on custom Azure Policy definition. Possible values + include: "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :vartype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :ivar partner_data: Describes the partner that created the assessment. + :vartype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + """ + + _validation = { + 'display_name': {'required': True}, + 'policy_definition_id': {'readonly': True}, + 'severity': {'required': True}, + 'assessment_type': {'required': True}, + } + + _attribute_map = { + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'policy_definition_id': {'key': 'policyDefinitionId', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'remediation_description': {'key': 'remediationDescription', 'type': 'str'}, + 'categories': {'key': 'categories', 'type': '[str]'}, + 'severity': {'key': 'severity', 'type': 'str'}, + 'user_impact': {'key': 'userImpact', 'type': 'str'}, + 'implementation_effort': {'key': 'implementationEffort', 'type': 'str'}, + 'threats': {'key': 'threats', 'type': '[str]'}, + 'preview': {'key': 'preview', 'type': 'bool'}, + 'assessment_type': {'key': 'assessmentType', 'type': 'str'}, + 'partner_data': {'key': 'partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, + } + + def __init__( + self, + *, + display_name: str, + severity: Union[str, "Severity"], + assessment_type: Union[str, "AssessmentType"], + description: Optional[str] = None, + remediation_description: Optional[str] = None, + categories: Optional[List[Union[str, "Categories"]]] = None, + user_impact: Optional[Union[str, "UserImpact"]] = None, + implementation_effort: Optional[Union[str, "ImplementationEffort"]] = None, + threats: Optional[List[Union[str, "Threats"]]] = None, + preview: Optional[bool] = None, + partner_data: Optional["SecurityAssessmentMetadataPartnerData"] = None, + **kwargs + ): + """ + :keyword display_name: Required. User friendly display name of the assessment. + :paramtype display_name: str + :keyword description: Human readable description of the assessment. + :paramtype description: str + :keyword remediation_description: Human readable description of what you should do to mitigate + this security issue. + :paramtype remediation_description: str + :keyword categories: + :paramtype categories: list[str or ~azure.mgmt.security.models.Categories] + :keyword severity: Required. The severity level of the assessment. Possible values include: + "Low", "Medium", "High". + :paramtype severity: str or ~azure.mgmt.security.models.Severity + :keyword user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :paramtype user_impact: str or ~azure.mgmt.security.models.UserImpact + :keyword implementation_effort: The implementation effort required to remediate this + assessment. Possible values include: "Low", "Moderate", "High". + :paramtype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :keyword threats: + :paramtype threats: list[str or ~azure.mgmt.security.models.Threats] + :keyword preview: True if this assessment is in preview release status. + :paramtype preview: bool + :keyword assessment_type: Required. BuiltIn if the assessment based on built-in Azure Policy + definition, Custom if the assessment based on custom Azure Policy definition. Possible values + include: "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :paramtype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :keyword partner_data: Describes the partner that created the assessment. + :paramtype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + """ + super(SecurityAssessmentMetadataProperties, self).__init__(**kwargs) + self.display_name = display_name + self.policy_definition_id = None + self.description = description + self.remediation_description = remediation_description + self.categories = categories + self.severity = severity + self.user_impact = user_impact + self.implementation_effort = implementation_effort + self.threats = threats + self.preview = preview + self.assessment_type = assessment_type + self.partner_data = partner_data + + +class SecurityAssessmentMetadataPropertiesResponse(SecurityAssessmentMetadataProperties): + """Describes properties of an assessment metadata response. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar display_name: Required. User friendly display name of the assessment. + :vartype display_name: str + :ivar policy_definition_id: Azure resource ID of the policy definition that turns this + assessment calculation on. + :vartype policy_definition_id: str + :ivar description: Human readable description of the assessment. + :vartype description: str + :ivar remediation_description: Human readable description of what you should do to mitigate + this security issue. + :vartype remediation_description: str + :ivar categories: + :vartype categories: list[str or ~azure.mgmt.security.models.Categories] + :ivar severity: Required. The severity level of the assessment. Possible values include: "Low", + "Medium", "High". + :vartype severity: str or ~azure.mgmt.security.models.Severity + :ivar user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :vartype user_impact: str or ~azure.mgmt.security.models.UserImpact + :ivar implementation_effort: The implementation effort required to remediate this assessment. + Possible values include: "Low", "Moderate", "High". + :vartype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :ivar threats: + :vartype threats: list[str or ~azure.mgmt.security.models.Threats] + :ivar preview: True if this assessment is in preview release status. + :vartype preview: bool + :ivar assessment_type: Required. BuiltIn if the assessment based on built-in Azure Policy + definition, Custom if the assessment based on custom Azure Policy definition. Possible values + include: "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :vartype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :ivar partner_data: Describes the partner that created the assessment. + :vartype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + :ivar publish_dates: + :vartype publish_dates: + ~azure.mgmt.security.models.SecurityAssessmentMetadataPropertiesResponsePublishDates + :ivar planned_deprecation_date: + :vartype planned_deprecation_date: str + :ivar tactics: + :vartype tactics: list[str or ~azure.mgmt.security.models.Tactics] + :ivar techniques: + :vartype techniques: list[str or ~azure.mgmt.security.models.Techniques] + """ + + _validation = { + 'display_name': {'required': True}, + 'policy_definition_id': {'readonly': True}, + 'severity': {'required': True}, + 'assessment_type': {'required': True}, + 'planned_deprecation_date': {'pattern': r'^[0-9]{2}/[0-9]{4}$'}, + } + + _attribute_map = { + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'policy_definition_id': {'key': 'policyDefinitionId', 'type': 'str'}, + 'description': {'key': 'description', 'type': 'str'}, + 'remediation_description': {'key': 'remediationDescription', 'type': 'str'}, + 'categories': {'key': 'categories', 'type': '[str]'}, + 'severity': {'key': 'severity', 'type': 'str'}, + 'user_impact': {'key': 'userImpact', 'type': 'str'}, + 'implementation_effort': {'key': 'implementationEffort', 'type': 'str'}, + 'threats': {'key': 'threats', 'type': '[str]'}, + 'preview': {'key': 'preview', 'type': 'bool'}, + 'assessment_type': {'key': 'assessmentType', 'type': 'str'}, + 'partner_data': {'key': 'partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, + 'publish_dates': {'key': 'publishDates', 'type': 'SecurityAssessmentMetadataPropertiesResponsePublishDates'}, + 'planned_deprecation_date': {'key': 'plannedDeprecationDate', 'type': 'str'}, + 'tactics': {'key': 'tactics', 'type': '[str]'}, + 'techniques': {'key': 'techniques', 'type': '[str]'}, + } + + def __init__( + self, + *, + display_name: str, + severity: Union[str, "Severity"], + assessment_type: Union[str, "AssessmentType"], + description: Optional[str] = None, + remediation_description: Optional[str] = None, + categories: Optional[List[Union[str, "Categories"]]] = None, + user_impact: Optional[Union[str, "UserImpact"]] = None, + implementation_effort: Optional[Union[str, "ImplementationEffort"]] = None, + threats: Optional[List[Union[str, "Threats"]]] = None, + preview: Optional[bool] = None, + partner_data: Optional["SecurityAssessmentMetadataPartnerData"] = None, + publish_dates: Optional["SecurityAssessmentMetadataPropertiesResponsePublishDates"] = None, + planned_deprecation_date: Optional[str] = None, + tactics: Optional[List[Union[str, "Tactics"]]] = None, + techniques: Optional[List[Union[str, "Techniques"]]] = None, + **kwargs + ): + """ + :keyword display_name: Required. User friendly display name of the assessment. + :paramtype display_name: str + :keyword description: Human readable description of the assessment. + :paramtype description: str + :keyword remediation_description: Human readable description of what you should do to mitigate + this security issue. + :paramtype remediation_description: str + :keyword categories: + :paramtype categories: list[str or ~azure.mgmt.security.models.Categories] + :keyword severity: Required. The severity level of the assessment. Possible values include: + "Low", "Medium", "High". + :paramtype severity: str or ~azure.mgmt.security.models.Severity + :keyword user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :paramtype user_impact: str or ~azure.mgmt.security.models.UserImpact + :keyword implementation_effort: The implementation effort required to remediate this + assessment. Possible values include: "Low", "Moderate", "High". + :paramtype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :keyword threats: + :paramtype threats: list[str or ~azure.mgmt.security.models.Threats] + :keyword preview: True if this assessment is in preview release status. + :paramtype preview: bool + :keyword assessment_type: Required. BuiltIn if the assessment based on built-in Azure Policy + definition, Custom if the assessment based on custom Azure Policy definition. Possible values + include: "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :paramtype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :keyword partner_data: Describes the partner that created the assessment. + :paramtype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + :keyword publish_dates: + :paramtype publish_dates: + ~azure.mgmt.security.models.SecurityAssessmentMetadataPropertiesResponsePublishDates + :keyword planned_deprecation_date: + :paramtype planned_deprecation_date: str + :keyword tactics: + :paramtype tactics: list[str or ~azure.mgmt.security.models.Tactics] + :keyword techniques: + :paramtype techniques: list[str or ~azure.mgmt.security.models.Techniques] + """ + super(SecurityAssessmentMetadataPropertiesResponse, self).__init__(display_name=display_name, description=description, remediation_description=remediation_description, categories=categories, severity=severity, user_impact=user_impact, implementation_effort=implementation_effort, threats=threats, preview=preview, assessment_type=assessment_type, partner_data=partner_data, **kwargs) + self.publish_dates = publish_dates + self.planned_deprecation_date = planned_deprecation_date + self.tactics = tactics + self.techniques = techniques + + +class SecurityAssessmentMetadataPropertiesResponsePublishDates(msrest.serialization.Model): + """SecurityAssessmentMetadataPropertiesResponsePublishDates. + + All required parameters must be populated in order to send to Azure. + + :ivar ga: + :vartype ga: str + :ivar public: Required. + :vartype public: str """ + _validation = { + 'ga': {'pattern': r'^([0-9]{2}/){2}[0-9]{4}$'}, + 'public': {'required': True, 'pattern': r'^([0-9]{2}/){2}[0-9]{4}$'}, + } + _attribute_map = { - 'source_type': {'key': 'sourceType', 'type': 'str'}, + 'ga': {'key': 'GA', 'type': 'str'}, + 'public': {'key': 'public', 'type': 'str'}, } def __init__( self, *, - source_type: Optional[Union[str, "ControlType"]] = None, + public: str, + ga: Optional[str] = None, **kwargs ): - super(SecureScoreControlDefinitionSource, self).__init__(**kwargs) - self.source_type = source_type + """ + :keyword ga: + :paramtype ga: str + :keyword public: Required. + :paramtype public: str + """ + super(SecurityAssessmentMetadataPropertiesResponsePublishDates, self).__init__(**kwargs) + self.ga = ga + self.public = public -class SecureScoreControlDetails(Resource): - """Details of the security control, its score, and the health status of the relevant resources. +class SecurityAssessmentMetadataResponse(Resource): + """Security assessment metadata response. Variables are only populated by the server, and will be ignored when sending a request. @@ -8033,40 +11074,54 @@ class SecureScoreControlDetails(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :ivar display_name: User friendly display name of the control. + :ivar display_name: User friendly display name of the assessment. :vartype display_name: str - :ivar healthy_resource_count: Number of healthy resources in the control. - :vartype healthy_resource_count: int - :ivar unhealthy_resource_count: Number of unhealthy resources in the control. - :vartype unhealthy_resource_count: int - :ivar not_applicable_resource_count: Number of not applicable resources in the control. - :vartype not_applicable_resource_count: int - :ivar weight: The relative weight for this specific control in each of your subscriptions. Used - when calculating an aggregated score for this control across all of your subscriptions. - :vartype weight: long - :param definition: Information about the security control. - :type definition: ~azure.mgmt.security.models.SecureScoreControlDefinitionItem - :ivar max: Maximum score available. - :vartype max: int - :ivar current: Current score. - :vartype current: float - :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after - the decimal point. - :vartype percentage: float + :ivar policy_definition_id: Azure resource ID of the policy definition that turns this + assessment calculation on. + :vartype policy_definition_id: str + :ivar description: Human readable description of the assessment. + :vartype description: str + :ivar remediation_description: Human readable description of what you should do to mitigate + this security issue. + :vartype remediation_description: str + :ivar categories: + :vartype categories: list[str or ~azure.mgmt.security.models.Categories] + :ivar severity: The severity level of the assessment. Possible values include: "Low", "Medium", + "High". + :vartype severity: str or ~azure.mgmt.security.models.Severity + :ivar user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :vartype user_impact: str or ~azure.mgmt.security.models.UserImpact + :ivar implementation_effort: The implementation effort required to remediate this assessment. + Possible values include: "Low", "Moderate", "High". + :vartype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :ivar threats: + :vartype threats: list[str or ~azure.mgmt.security.models.Threats] + :ivar preview: True if this assessment is in preview release status. + :vartype preview: bool + :ivar assessment_type: BuiltIn if the assessment based on built-in Azure Policy definition, + Custom if the assessment based on custom Azure Policy definition. Possible values include: + "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :vartype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :ivar partner_data: Describes the partner that created the assessment. + :vartype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + :ivar publish_dates: + :vartype publish_dates: + ~azure.mgmt.security.models.SecurityAssessmentMetadataPropertiesResponsePublishDates + :ivar planned_deprecation_date: + :vartype planned_deprecation_date: str + :ivar tactics: + :vartype tactics: list[str or ~azure.mgmt.security.models.Tactics] + :ivar techniques: + :vartype techniques: list[str or ~azure.mgmt.security.models.Techniques] """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'display_name': {'readonly': True}, - 'healthy_resource_count': {'readonly': True}, - 'unhealthy_resource_count': {'readonly': True}, - 'not_applicable_resource_count': {'readonly': True}, - 'weight': {'readonly': True, 'minimum': 0}, - 'max': {'readonly': True, 'minimum': 0}, - 'current': {'readonly': True, 'minimum': 0}, - 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, + 'policy_definition_id': {'readonly': True}, + 'planned_deprecation_date': {'pattern': r'^[0-9]{2}/[0-9]{4}$'}, } _attribute_map = { @@ -8074,41 +11129,108 @@ class SecureScoreControlDetails(Resource): 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'healthy_resource_count': {'key': 'properties.healthyResourceCount', 'type': 'int'}, - 'unhealthy_resource_count': {'key': 'properties.unhealthyResourceCount', 'type': 'int'}, - 'not_applicable_resource_count': {'key': 'properties.notApplicableResourceCount', 'type': 'int'}, - 'weight': {'key': 'properties.weight', 'type': 'long'}, - 'definition': {'key': 'properties.definition', 'type': 'SecureScoreControlDefinitionItem'}, - 'max': {'key': 'properties.score.max', 'type': 'int'}, - 'current': {'key': 'properties.score.current', 'type': 'float'}, - 'percentage': {'key': 'properties.score.percentage', 'type': 'float'}, + 'policy_definition_id': {'key': 'properties.policyDefinitionId', 'type': 'str'}, + 'description': {'key': 'properties.description', 'type': 'str'}, + 'remediation_description': {'key': 'properties.remediationDescription', 'type': 'str'}, + 'categories': {'key': 'properties.categories', 'type': '[str]'}, + 'severity': {'key': 'properties.severity', 'type': 'str'}, + 'user_impact': {'key': 'properties.userImpact', 'type': 'str'}, + 'implementation_effort': {'key': 'properties.implementationEffort', 'type': 'str'}, + 'threats': {'key': 'properties.threats', 'type': '[str]'}, + 'preview': {'key': 'properties.preview', 'type': 'bool'}, + 'assessment_type': {'key': 'properties.assessmentType', 'type': 'str'}, + 'partner_data': {'key': 'properties.partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, + 'publish_dates': {'key': 'properties.publishDates', 'type': 'SecurityAssessmentMetadataPropertiesResponsePublishDates'}, + 'planned_deprecation_date': {'key': 'properties.plannedDeprecationDate', 'type': 'str'}, + 'tactics': {'key': 'properties.tactics', 'type': '[str]'}, + 'techniques': {'key': 'properties.techniques', 'type': '[str]'}, } def __init__( self, *, - definition: Optional["SecureScoreControlDefinitionItem"] = None, - **kwargs - ): - super(SecureScoreControlDetails, self).__init__(**kwargs) - self.display_name = None - self.healthy_resource_count = None - self.unhealthy_resource_count = None - self.not_applicable_resource_count = None - self.weight = None - self.definition = definition - self.max = None - self.current = None - self.percentage = None + display_name: Optional[str] = None, + description: Optional[str] = None, + remediation_description: Optional[str] = None, + categories: Optional[List[Union[str, "Categories"]]] = None, + severity: Optional[Union[str, "Severity"]] = None, + user_impact: Optional[Union[str, "UserImpact"]] = None, + implementation_effort: Optional[Union[str, "ImplementationEffort"]] = None, + threats: Optional[List[Union[str, "Threats"]]] = None, + preview: Optional[bool] = None, + assessment_type: Optional[Union[str, "AssessmentType"]] = None, + partner_data: Optional["SecurityAssessmentMetadataPartnerData"] = None, + publish_dates: Optional["SecurityAssessmentMetadataPropertiesResponsePublishDates"] = None, + planned_deprecation_date: Optional[str] = None, + tactics: Optional[List[Union[str, "Tactics"]]] = None, + techniques: Optional[List[Union[str, "Techniques"]]] = None, + **kwargs + ): + """ + :keyword display_name: User friendly display name of the assessment. + :paramtype display_name: str + :keyword description: Human readable description of the assessment. + :paramtype description: str + :keyword remediation_description: Human readable description of what you should do to mitigate + this security issue. + :paramtype remediation_description: str + :keyword categories: + :paramtype categories: list[str or ~azure.mgmt.security.models.Categories] + :keyword severity: The severity level of the assessment. Possible values include: "Low", + "Medium", "High". + :paramtype severity: str or ~azure.mgmt.security.models.Severity + :keyword user_impact: The user impact of the assessment. Possible values include: "Low", + "Moderate", "High". + :paramtype user_impact: str or ~azure.mgmt.security.models.UserImpact + :keyword implementation_effort: The implementation effort required to remediate this + assessment. Possible values include: "Low", "Moderate", "High". + :paramtype implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort + :keyword threats: + :paramtype threats: list[str or ~azure.mgmt.security.models.Threats] + :keyword preview: True if this assessment is in preview release status. + :paramtype preview: bool + :keyword assessment_type: BuiltIn if the assessment based on built-in Azure Policy definition, + Custom if the assessment based on custom Azure Policy definition. Possible values include: + "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". + :paramtype assessment_type: str or ~azure.mgmt.security.models.AssessmentType + :keyword partner_data: Describes the partner that created the assessment. + :paramtype partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + :keyword publish_dates: + :paramtype publish_dates: + ~azure.mgmt.security.models.SecurityAssessmentMetadataPropertiesResponsePublishDates + :keyword planned_deprecation_date: + :paramtype planned_deprecation_date: str + :keyword tactics: + :paramtype tactics: list[str or ~azure.mgmt.security.models.Tactics] + :keyword techniques: + :paramtype techniques: list[str or ~azure.mgmt.security.models.Techniques] + """ + super(SecurityAssessmentMetadataResponse, self).__init__(**kwargs) + self.display_name = display_name + self.policy_definition_id = None + self.description = description + self.remediation_description = remediation_description + self.categories = categories + self.severity = severity + self.user_impact = user_impact + self.implementation_effort = implementation_effort + self.threats = threats + self.preview = preview + self.assessment_type = assessment_type + self.partner_data = partner_data + self.publish_dates = publish_dates + self.planned_deprecation_date = planned_deprecation_date + self.tactics = tactics + self.techniques = techniques -class SecureScoreControlList(msrest.serialization.Model): - """List of security controls. +class SecurityAssessmentMetadataResponseList(msrest.serialization.Model): + """List of security assessment metadata. Variables are only populated by the server, and will be ignored when sending a request. - :ivar value: Collection of security controls in this page. - :vartype value: list[~azure.mgmt.security.models.SecureScoreControlDetails] + :ivar value: + :vartype value: list[~azure.mgmt.security.models.SecurityAssessmentMetadataResponse] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -8119,7 +11241,7 @@ class SecureScoreControlList(msrest.serialization.Model): } _attribute_map = { - 'value': {'key': 'value', 'type': '[SecureScoreControlDetails]'}, + 'value': {'key': 'value', 'type': '[SecurityAssessmentMetadataResponse]'}, 'next_link': {'key': 'nextLink', 'type': 'str'}, } @@ -8127,139 +11249,250 @@ def __init__( self, **kwargs ): - super(SecureScoreControlList, self).__init__(**kwargs) + """ + """ + super(SecurityAssessmentMetadataResponseList, self).__init__(**kwargs) self.value = None self.next_link = None -class SecureScoreControlScore(msrest.serialization.Model): - """Calculation result data. +class SecurityAssessmentPartnerData(msrest.serialization.Model): + """Data regarding 3rd party partner integration. - Variables are only populated by the server, and will be ignored when sending a request. + All required parameters must be populated in order to send to Azure. - :ivar max: Maximum control score (0..10). - :vartype max: int - :ivar current: Actual score for the control = (achieved points / total points) * max score. if - total points is zeroed, the return number is 0.00. - :vartype current: float - :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after - the decimal point. - :vartype percentage: float + :ivar partner_name: Required. Name of the company of the partner. + :vartype partner_name: str + :ivar secret: Required. secret to authenticate the partner - write only. + :vartype secret: str """ _validation = { - 'max': {'readonly': True, 'maximum': 10, 'minimum': 0}, - 'current': {'readonly': True, 'maximum': 10, 'minimum': 0}, - 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, + 'partner_name': {'required': True}, + 'secret': {'required': True}, } _attribute_map = { - 'max': {'key': 'max', 'type': 'int'}, - 'current': {'key': 'current', 'type': 'float'}, - 'percentage': {'key': 'percentage', 'type': 'float'}, + 'partner_name': {'key': 'partnerName', 'type': 'str'}, + 'secret': {'key': 'secret', 'type': 'str'}, } def __init__( self, + *, + partner_name: str, + secret: str, **kwargs ): - super(SecureScoreControlScore, self).__init__(**kwargs) - self.max = None - self.current = None - self.percentage = None + """ + :keyword partner_name: Required. Name of the company of the partner. + :paramtype partner_name: str + :keyword secret: Required. secret to authenticate the partner - write only. + :paramtype secret: str + """ + super(SecurityAssessmentPartnerData, self).__init__(**kwargs) + self.partner_name = partner_name + self.secret = secret -class SecureScoreItem(Resource): - """Secure score item data model. +class SecurityAssessmentPropertiesBase(msrest.serialization.Model): + """Describes properties of an assessment. Variables are only populated by the server, and will be ignored when sending a request. - :ivar id: Resource Id. - :vartype id: str - :ivar name: Resource name. - :vartype name: str - :ivar type: Resource type. - :vartype type: str - :ivar display_name: The initiative’s name. + All required parameters must be populated in order to send to Azure. + + :ivar resource_details: Required. Details of the resource that was assessed. + :vartype resource_details: ~azure.mgmt.security.models.ResourceDetails + :ivar display_name: User friendly display name of the assessment. :vartype display_name: str - :ivar weight: The relative weight for each subscription. Used when calculating an aggregated - secure score for multiple subscriptions. - :vartype weight: long - :ivar max: Maximum score available. - :vartype max: int - :ivar current: Current score. - :vartype current: float - :ivar percentage: Ratio of the current score divided by the maximum. Rounded to 4 digits after - the decimal point. - :vartype percentage: float + :ivar additional_data: Additional data regarding the assessment. + :vartype additional_data: dict[str, str] + :ivar links: Links relevant to the assessment. + :vartype links: ~azure.mgmt.security.models.AssessmentLinks + :ivar metadata: Describes properties of an assessment metadata. + :vartype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :ivar partners_data: Data regarding 3rd party partner integration. + :vartype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, + 'resource_details': {'required': True}, 'display_name': {'readonly': True}, - 'weight': {'readonly': True, 'minimum': 0}, - 'max': {'readonly': True, 'minimum': 0}, - 'current': {'readonly': True, 'minimum': 0}, - 'percentage': {'readonly': True, 'maximum': 1, 'minimum': 0}, + 'links': {'readonly': True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'weight': {'key': 'properties.weight', 'type': 'long'}, - 'max': {'key': 'properties.score.max', 'type': 'int'}, - 'current': {'key': 'properties.score.current', 'type': 'float'}, - 'percentage': {'key': 'properties.score.percentage', 'type': 'float'}, + 'resource_details': {'key': 'resourceDetails', 'type': 'ResourceDetails'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'additional_data': {'key': 'additionalData', 'type': '{str}'}, + 'links': {'key': 'links', 'type': 'AssessmentLinks'}, + 'metadata': {'key': 'metadata', 'type': 'SecurityAssessmentMetadataProperties'}, + 'partners_data': {'key': 'partnersData', 'type': 'SecurityAssessmentPartnerData'}, } def __init__( self, + *, + resource_details: "ResourceDetails", + additional_data: Optional[Dict[str, str]] = None, + metadata: Optional["SecurityAssessmentMetadataProperties"] = None, + partners_data: Optional["SecurityAssessmentPartnerData"] = None, **kwargs ): - super(SecureScoreItem, self).__init__(**kwargs) + """ + :keyword resource_details: Required. Details of the resource that was assessed. + :paramtype resource_details: ~azure.mgmt.security.models.ResourceDetails + :keyword additional_data: Additional data regarding the assessment. + :paramtype additional_data: dict[str, str] + :keyword metadata: Describes properties of an assessment metadata. + :paramtype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :keyword partners_data: Data regarding 3rd party partner integration. + :paramtype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + """ + super(SecurityAssessmentPropertiesBase, self).__init__(**kwargs) + self.resource_details = resource_details self.display_name = None - self.weight = None - self.max = None - self.current = None - self.percentage = None + self.additional_data = additional_data + self.links = None + self.metadata = metadata + self.partners_data = partners_data -class SecureScoresList(msrest.serialization.Model): - """List of secure scores. +class SecurityAssessmentProperties(SecurityAssessmentPropertiesBase): + """Describes properties of an assessment. Variables are only populated by the server, and will be ignored when sending a request. - :ivar value: Collection of secure scores in this page. - :vartype value: list[~azure.mgmt.security.models.SecureScoreItem] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str + All required parameters must be populated in order to send to Azure. + + :ivar resource_details: Required. Details of the resource that was assessed. + :vartype resource_details: ~azure.mgmt.security.models.ResourceDetails + :ivar display_name: User friendly display name of the assessment. + :vartype display_name: str + :ivar additional_data: Additional data regarding the assessment. + :vartype additional_data: dict[str, str] + :ivar links: Links relevant to the assessment. + :vartype links: ~azure.mgmt.security.models.AssessmentLinks + :ivar metadata: Describes properties of an assessment metadata. + :vartype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :ivar partners_data: Data regarding 3rd party partner integration. + :vartype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :ivar status: Required. The result of the assessment. + :vartype status: ~azure.mgmt.security.models.AssessmentStatus """ _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, + 'resource_details': {'required': True}, + 'display_name': {'readonly': True}, + 'links': {'readonly': True}, + 'status': {'required': True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[SecureScoreItem]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + 'resource_details': {'key': 'resourceDetails', 'type': 'ResourceDetails'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'additional_data': {'key': 'additionalData', 'type': '{str}'}, + 'links': {'key': 'links', 'type': 'AssessmentLinks'}, + 'metadata': {'key': 'metadata', 'type': 'SecurityAssessmentMetadataProperties'}, + 'partners_data': {'key': 'partnersData', 'type': 'SecurityAssessmentPartnerData'}, + 'status': {'key': 'status', 'type': 'AssessmentStatus'}, } def __init__( self, + *, + resource_details: "ResourceDetails", + status: "AssessmentStatus", + additional_data: Optional[Dict[str, str]] = None, + metadata: Optional["SecurityAssessmentMetadataProperties"] = None, + partners_data: Optional["SecurityAssessmentPartnerData"] = None, **kwargs ): - super(SecureScoresList, self).__init__(**kwargs) - self.value = None - self.next_link = None + """ + :keyword resource_details: Required. Details of the resource that was assessed. + :paramtype resource_details: ~azure.mgmt.security.models.ResourceDetails + :keyword additional_data: Additional data regarding the assessment. + :paramtype additional_data: dict[str, str] + :keyword metadata: Describes properties of an assessment metadata. + :paramtype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :keyword partners_data: Data regarding 3rd party partner integration. + :paramtype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :keyword status: Required. The result of the assessment. + :paramtype status: ~azure.mgmt.security.models.AssessmentStatus + """ + super(SecurityAssessmentProperties, self).__init__(resource_details=resource_details, additional_data=additional_data, metadata=metadata, partners_data=partners_data, **kwargs) + self.status = status + + +class SecurityAssessmentPropertiesResponse(SecurityAssessmentPropertiesBase): + """Describes properties of an assessment. + + Variables are only populated by the server, and will be ignored when sending a request. + + All required parameters must be populated in order to send to Azure. + + :ivar resource_details: Required. Details of the resource that was assessed. + :vartype resource_details: ~azure.mgmt.security.models.ResourceDetails + :ivar display_name: User friendly display name of the assessment. + :vartype display_name: str + :ivar additional_data: Additional data regarding the assessment. + :vartype additional_data: dict[str, str] + :ivar links: Links relevant to the assessment. + :vartype links: ~azure.mgmt.security.models.AssessmentLinks + :ivar metadata: Describes properties of an assessment metadata. + :vartype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :ivar partners_data: Data regarding 3rd party partner integration. + :vartype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :ivar status: Required. The result of the assessment. + :vartype status: ~azure.mgmt.security.models.AssessmentStatusResponse + """ + + _validation = { + 'resource_details': {'required': True}, + 'display_name': {'readonly': True}, + 'links': {'readonly': True}, + 'status': {'required': True}, + } + + _attribute_map = { + 'resource_details': {'key': 'resourceDetails', 'type': 'ResourceDetails'}, + 'display_name': {'key': 'displayName', 'type': 'str'}, + 'additional_data': {'key': 'additionalData', 'type': '{str}'}, + 'links': {'key': 'links', 'type': 'AssessmentLinks'}, + 'metadata': {'key': 'metadata', 'type': 'SecurityAssessmentMetadataProperties'}, + 'partners_data': {'key': 'partnersData', 'type': 'SecurityAssessmentPartnerData'}, + 'status': {'key': 'status', 'type': 'AssessmentStatusResponse'}, + } + + def __init__( + self, + *, + resource_details: "ResourceDetails", + status: "AssessmentStatusResponse", + additional_data: Optional[Dict[str, str]] = None, + metadata: Optional["SecurityAssessmentMetadataProperties"] = None, + partners_data: Optional["SecurityAssessmentPartnerData"] = None, + **kwargs + ): + """ + :keyword resource_details: Required. Details of the resource that was assessed. + :paramtype resource_details: ~azure.mgmt.security.models.ResourceDetails + :keyword additional_data: Additional data regarding the assessment. + :paramtype additional_data: dict[str, str] + :keyword metadata: Describes properties of an assessment metadata. + :paramtype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :keyword partners_data: Data regarding 3rd party partner integration. + :paramtype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :keyword status: Required. The result of the assessment. + :paramtype status: ~azure.mgmt.security.models.AssessmentStatusResponse + """ + super(SecurityAssessmentPropertiesResponse, self).__init__(resource_details=resource_details, additional_data=additional_data, metadata=metadata, partners_data=partners_data, **kwargs) + self.status = status -class SecurityAssessment(Resource): - """Security assessment on a resource. +class SecurityAssessmentResponse(Resource): + """Security assessment on a resource - response format. Variables are only populated by the server, and will be ignored when sending a request. @@ -8269,20 +11502,20 @@ class SecurityAssessment(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param resource_details: Details of the resource that was assessed. - :type resource_details: ~azure.mgmt.security.models.ResourceDetails + :ivar resource_details: Details of the resource that was assessed. + :vartype resource_details: ~azure.mgmt.security.models.ResourceDetails :ivar display_name: User friendly display name of the assessment. :vartype display_name: str - :param status: The result of the assessment. - :type status: ~azure.mgmt.security.models.AssessmentStatus - :param additional_data: Additional data regarding the assessment. - :type additional_data: dict[str, str] + :ivar additional_data: Additional data regarding the assessment. + :vartype additional_data: dict[str, str] :ivar links: Links relevant to the assessment. :vartype links: ~azure.mgmt.security.models.AssessmentLinks - :param metadata: Describes properties of an assessment metadata. - :type metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties - :param partners_data: Data regarding 3rd party partner integration. - :type partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :ivar metadata: Describes properties of an assessment metadata. + :vartype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :ivar partners_data: Data regarding 3rd party partner integration. + :vartype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :ivar status: The result of the assessment. + :vartype status: ~azure.mgmt.security.models.AssessmentStatusResponse """ _validation = { @@ -8299,356 +11532,238 @@ class SecurityAssessment(Resource): 'type': {'key': 'type', 'type': 'str'}, 'resource_details': {'key': 'properties.resourceDetails', 'type': 'ResourceDetails'}, 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'status': {'key': 'properties.status', 'type': 'AssessmentStatus'}, 'additional_data': {'key': 'properties.additionalData', 'type': '{str}'}, 'links': {'key': 'properties.links', 'type': 'AssessmentLinks'}, 'metadata': {'key': 'properties.metadata', 'type': 'SecurityAssessmentMetadataProperties'}, 'partners_data': {'key': 'properties.partnersData', 'type': 'SecurityAssessmentPartnerData'}, + 'status': {'key': 'properties.status', 'type': 'AssessmentStatusResponse'}, } def __init__( self, *, resource_details: Optional["ResourceDetails"] = None, - status: Optional["AssessmentStatus"] = None, additional_data: Optional[Dict[str, str]] = None, metadata: Optional["SecurityAssessmentMetadataProperties"] = None, partners_data: Optional["SecurityAssessmentPartnerData"] = None, - **kwargs - ): - super(SecurityAssessment, self).__init__(**kwargs) + status: Optional["AssessmentStatusResponse"] = None, + **kwargs + ): + """ + :keyword resource_details: Details of the resource that was assessed. + :paramtype resource_details: ~azure.mgmt.security.models.ResourceDetails + :keyword additional_data: Additional data regarding the assessment. + :paramtype additional_data: dict[str, str] + :keyword metadata: Describes properties of an assessment metadata. + :paramtype metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataProperties + :keyword partners_data: Data regarding 3rd party partner integration. + :paramtype partners_data: ~azure.mgmt.security.models.SecurityAssessmentPartnerData + :keyword status: The result of the assessment. + :paramtype status: ~azure.mgmt.security.models.AssessmentStatusResponse + """ + super(SecurityAssessmentResponse, self).__init__(**kwargs) self.resource_details = resource_details self.display_name = None - self.status = status self.additional_data = additional_data self.links = None self.metadata = metadata self.partners_data = partners_data + self.status = status -class SecurityAssessmentList(msrest.serialization.Model): - """Page of a security assessments list. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Collection of security assessments in this page. - :vartype value: list[~azure.mgmt.security.models.SecurityAssessment] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecurityAssessment]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentList, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class SecurityAssessmentMetadata(Resource): - """Security assessment metadata. +class SecurityConnector(TrackedResource): + """The security connector resource. Variables are only populated by the server, and will be ignored when sending a request. + :ivar tags: A set of tags. A list of key value pairs that describe the resource. + :vartype tags: dict[str, str] + :ivar etag: Entity tag is used for comparing two or more entities from the same requested + resource. + :vartype etag: str + :ivar kind: Kind of the resource. + :vartype kind: str + :ivar location: Location where the resource is stored. + :vartype location: str :ivar id: Resource Id. :vartype id: str :ivar name: Resource name. :vartype name: str :ivar type: Resource type. :vartype type: str - :param display_name: User friendly display name of the assessment. - :type display_name: str - :ivar policy_definition_id: Azure resource ID of the policy definition that turns this - assessment calculation on. - :vartype policy_definition_id: str - :param description: Human readable description of the assessment. - :type description: str - :param remediation_description: Human readable description of what you should do to mitigate - this security issue. - :type remediation_description: str - :param categories: - :type categories: list[str or ~azure.mgmt.security.models.Categories] - :param severity: The severity level of the assessment. Possible values include: "Low", - "Medium", "High". - :type severity: str or ~azure.mgmt.security.models.Severity - :param user_impact: The user impact of the assessment. Possible values include: "Low", - "Moderate", "High". - :type user_impact: str or ~azure.mgmt.security.models.UserImpact - :param implementation_effort: The implementation effort required to remediate this assessment. - Possible values include: "Low", "Moderate", "High". - :type implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort - :param threats: - :type threats: list[str or ~azure.mgmt.security.models.Threats] - :param preview: True if this assessment is in preview release status. - :type preview: bool - :param assessment_type: BuiltIn if the assessment based on built-in Azure Policy definition, - Custom if the assessment based on custom Azure Policy definition. Possible values include: - "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". - :type assessment_type: str or ~azure.mgmt.security.models.AssessmentType - :param partner_data: Describes the partner that created the assessment. - :type partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.security.models.SystemData + :ivar hierarchy_identifier: The multi cloud resource identifier (account id in case of AWS + connector). + :vartype hierarchy_identifier: str + :ivar cloud_name: The multi cloud resource's cloud name. Possible values include: "Azure", + "AWS", "GCP". + :vartype cloud_name: str or ~azure.mgmt.security.models.CloudName + :ivar offerings: A collection of offerings for the security connector. + :vartype offerings: list[~azure.mgmt.security.models.CloudOffering] + :ivar organizational_data: The multi cloud account's organizational data. + :vartype organizational_data: + ~azure.mgmt.security.models.SecurityConnectorPropertiesOrganizationalData """ _validation = { 'id': {'readonly': True}, 'name': {'readonly': True}, 'type': {'readonly': True}, - 'policy_definition_id': {'readonly': True}, + 'system_data': {'readonly': True}, } _attribute_map = { + 'tags': {'key': 'tags', 'type': '{str}'}, + 'etag': {'key': 'etag', 'type': 'str'}, + 'kind': {'key': 'kind', 'type': 'str'}, + 'location': {'key': 'location', 'type': 'str'}, 'id': {'key': 'id', 'type': 'str'}, 'name': {'key': 'name', 'type': 'str'}, 'type': {'key': 'type', 'type': 'str'}, - 'display_name': {'key': 'properties.displayName', 'type': 'str'}, - 'policy_definition_id': {'key': 'properties.policyDefinitionId', 'type': 'str'}, - 'description': {'key': 'properties.description', 'type': 'str'}, - 'remediation_description': {'key': 'properties.remediationDescription', 'type': 'str'}, - 'categories': {'key': 'properties.categories', 'type': '[str]'}, - 'severity': {'key': 'properties.severity', 'type': 'str'}, - 'user_impact': {'key': 'properties.userImpact', 'type': 'str'}, - 'implementation_effort': {'key': 'properties.implementationEffort', 'type': 'str'}, - 'threats': {'key': 'properties.threats', 'type': '[str]'}, - 'preview': {'key': 'properties.preview', 'type': 'bool'}, - 'assessment_type': {'key': 'properties.assessmentType', 'type': 'str'}, - 'partner_data': {'key': 'properties.partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, + 'system_data': {'key': 'systemData', 'type': 'SystemData'}, + 'hierarchy_identifier': {'key': 'properties.hierarchyIdentifier', 'type': 'str'}, + 'cloud_name': {'key': 'properties.cloudName', 'type': 'str'}, + 'offerings': {'key': 'properties.offerings', 'type': '[CloudOffering]'}, + 'organizational_data': {'key': 'properties.organizationalData', 'type': 'SecurityConnectorPropertiesOrganizationalData'}, } def __init__( self, *, - display_name: Optional[str] = None, - description: Optional[str] = None, - remediation_description: Optional[str] = None, - categories: Optional[List[Union[str, "Categories"]]] = None, - severity: Optional[Union[str, "Severity"]] = None, - user_impact: Optional[Union[str, "UserImpact"]] = None, - implementation_effort: Optional[Union[str, "ImplementationEffort"]] = None, - threats: Optional[List[Union[str, "Threats"]]] = None, - preview: Optional[bool] = None, - assessment_type: Optional[Union[str, "AssessmentType"]] = None, - partner_data: Optional["SecurityAssessmentMetadataPartnerData"] = None, - **kwargs - ): - super(SecurityAssessmentMetadata, self).__init__(**kwargs) - self.display_name = display_name - self.policy_definition_id = None - self.description = description - self.remediation_description = remediation_description - self.categories = categories - self.severity = severity - self.user_impact = user_impact - self.implementation_effort = implementation_effort - self.threats = threats - self.preview = preview - self.assessment_type = assessment_type - self.partner_data = partner_data - - -class SecurityAssessmentMetadataList(msrest.serialization.Model): - """List of security assessment metadata. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: - :vartype value: list[~azure.mgmt.security.models.SecurityAssessmentMetadata] - :ivar next_link: The URI to fetch the next page. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[SecurityAssessmentMetadata]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SecurityAssessmentMetadataList, self).__init__(**kwargs) - self.value = None - self.next_link = None - + tags: Optional[Dict[str, str]] = None, + etag: Optional[str] = None, + kind: Optional[str] = None, + location: Optional[str] = None, + hierarchy_identifier: Optional[str] = None, + cloud_name: Optional[Union[str, "CloudName"]] = None, + offerings: Optional[List["CloudOffering"]] = None, + organizational_data: Optional["SecurityConnectorPropertiesOrganizationalData"] = None, + **kwargs + ): + """ + :keyword tags: A set of tags. A list of key value pairs that describe the resource. + :paramtype tags: dict[str, str] + :keyword etag: Entity tag is used for comparing two or more entities from the same requested + resource. + :paramtype etag: str + :keyword kind: Kind of the resource. + :paramtype kind: str + :keyword location: Location where the resource is stored. + :paramtype location: str + :keyword hierarchy_identifier: The multi cloud resource identifier (account id in case of AWS + connector). + :paramtype hierarchy_identifier: str + :keyword cloud_name: The multi cloud resource's cloud name. Possible values include: "Azure", + "AWS", "GCP". + :paramtype cloud_name: str or ~azure.mgmt.security.models.CloudName + :keyword offerings: A collection of offerings for the security connector. + :paramtype offerings: list[~azure.mgmt.security.models.CloudOffering] + :keyword organizational_data: The multi cloud account's organizational data. + :paramtype organizational_data: + ~azure.mgmt.security.models.SecurityConnectorPropertiesOrganizationalData + """ + super(SecurityConnector, self).__init__(tags=tags, etag=etag, kind=kind, location=location, **kwargs) + self.system_data = None + self.hierarchy_identifier = hierarchy_identifier + self.cloud_name = cloud_name + self.offerings = offerings + self.organizational_data = organizational_data -class SecurityAssessmentMetadataPartnerData(msrest.serialization.Model): - """Describes the partner that created the assessment. - All required parameters must be populated in order to send to Azure. +class SecurityConnectorPropertiesOrganizationalData(msrest.serialization.Model): + """The multi cloud account's organizational data. - :param partner_name: Required. Name of the company of the partner. - :type partner_name: str - :param product_name: Name of the product of the partner that created the assessment. - :type product_name: str - :param secret: Required. Secret to authenticate the partner and verify it created the - assessment - write only. - :type secret: str + :ivar organization_membership_type: The multi cloud account's membership type in the + organization. Possible values include: "Member", "Organization". + :vartype organization_membership_type: str or + ~azure.mgmt.security.models.OrganizationMembershipType + :ivar parent_hierarchy_id: If the multi cloud account is not of membership type organization, + this will be the ID of the account's parent. + :vartype parent_hierarchy_id: str + :ivar stackset_name: If the multi cloud account is of membership type organization, this will + be the name of the onboarding stackset. + :vartype stackset_name: str + :ivar excluded_account_ids: If the multi cloud account is of membership type organization, list + of accounts excluded from offering. + :vartype excluded_account_ids: list[str] """ - _validation = { - 'partner_name': {'required': True}, - 'secret': {'required': True}, - } - _attribute_map = { - 'partner_name': {'key': 'partnerName', 'type': 'str'}, - 'product_name': {'key': 'productName', 'type': 'str'}, - 'secret': {'key': 'secret', 'type': 'str'}, + 'organization_membership_type': {'key': 'organizationMembershipType', 'type': 'str'}, + 'parent_hierarchy_id': {'key': 'parentHierarchyId', 'type': 'str'}, + 'stackset_name': {'key': 'stacksetName', 'type': 'str'}, + 'excluded_account_ids': {'key': 'excludedAccountIds', 'type': '[str]'}, } def __init__( self, *, - partner_name: str, - secret: str, - product_name: Optional[str] = None, + organization_membership_type: Optional[Union[str, "OrganizationMembershipType"]] = None, + parent_hierarchy_id: Optional[str] = None, + stackset_name: Optional[str] = None, + excluded_account_ids: Optional[List[str]] = None, **kwargs ): - super(SecurityAssessmentMetadataPartnerData, self).__init__(**kwargs) - self.partner_name = partner_name - self.product_name = product_name - self.secret = secret + """ + :keyword organization_membership_type: The multi cloud account's membership type in the + organization. Possible values include: "Member", "Organization". + :paramtype organization_membership_type: str or + ~azure.mgmt.security.models.OrganizationMembershipType + :keyword parent_hierarchy_id: If the multi cloud account is not of membership type + organization, this will be the ID of the account's parent. + :paramtype parent_hierarchy_id: str + :keyword stackset_name: If the multi cloud account is of membership type organization, this + will be the name of the onboarding stackset. + :paramtype stackset_name: str + :keyword excluded_account_ids: If the multi cloud account is of membership type organization, + list of accounts excluded from offering. + :paramtype excluded_account_ids: list[str] + """ + super(SecurityConnectorPropertiesOrganizationalData, self).__init__(**kwargs) + self.organization_membership_type = organization_membership_type + self.parent_hierarchy_id = parent_hierarchy_id + self.stackset_name = stackset_name + self.excluded_account_ids = excluded_account_ids -class SecurityAssessmentMetadataProperties(msrest.serialization.Model): - """Describes properties of an assessment metadata. +class SecurityConnectorsList(msrest.serialization.Model): + """List of security connectors response. Variables are only populated by the server, and will be ignored when sending a request. All required parameters must be populated in order to send to Azure. - :param display_name: Required. User friendly display name of the assessment. - :type display_name: str - :ivar policy_definition_id: Azure resource ID of the policy definition that turns this - assessment calculation on. - :vartype policy_definition_id: str - :param description: Human readable description of the assessment. - :type description: str - :param remediation_description: Human readable description of what you should do to mitigate - this security issue. - :type remediation_description: str - :param categories: - :type categories: list[str or ~azure.mgmt.security.models.Categories] - :param severity: Required. The severity level of the assessment. Possible values include: - "Low", "Medium", "High". - :type severity: str or ~azure.mgmt.security.models.Severity - :param user_impact: The user impact of the assessment. Possible values include: "Low", - "Moderate", "High". - :type user_impact: str or ~azure.mgmt.security.models.UserImpact - :param implementation_effort: The implementation effort required to remediate this assessment. - Possible values include: "Low", "Moderate", "High". - :type implementation_effort: str or ~azure.mgmt.security.models.ImplementationEffort - :param threats: - :type threats: list[str or ~azure.mgmt.security.models.Threats] - :param preview: True if this assessment is in preview release status. - :type preview: bool - :param assessment_type: Required. BuiltIn if the assessment based on built-in Azure Policy - definition, Custom if the assessment based on custom Azure Policy definition. Possible values - include: "BuiltIn", "CustomPolicy", "CustomerManaged", "VerifiedPartner". - :type assessment_type: str or ~azure.mgmt.security.models.AssessmentType - :param partner_data: Describes the partner that created the assessment. - :type partner_data: ~azure.mgmt.security.models.SecurityAssessmentMetadataPartnerData - """ - - _validation = { - 'display_name': {'required': True}, - 'policy_definition_id': {'readonly': True}, - 'severity': {'required': True}, - 'assessment_type': {'required': True}, - } - - _attribute_map = { - 'display_name': {'key': 'displayName', 'type': 'str'}, - 'policy_definition_id': {'key': 'policyDefinitionId', 'type': 'str'}, - 'description': {'key': 'description', 'type': 'str'}, - 'remediation_description': {'key': 'remediationDescription', 'type': 'str'}, - 'categories': {'key': 'categories', 'type': '[str]'}, - 'severity': {'key': 'severity', 'type': 'str'}, - 'user_impact': {'key': 'userImpact', 'type': 'str'}, - 'implementation_effort': {'key': 'implementationEffort', 'type': 'str'}, - 'threats': {'key': 'threats', 'type': '[str]'}, - 'preview': {'key': 'preview', 'type': 'bool'}, - 'assessment_type': {'key': 'assessmentType', 'type': 'str'}, - 'partner_data': {'key': 'partnerData', 'type': 'SecurityAssessmentMetadataPartnerData'}, - } - - def __init__( - self, - *, - display_name: str, - severity: Union[str, "Severity"], - assessment_type: Union[str, "AssessmentType"], - description: Optional[str] = None, - remediation_description: Optional[str] = None, - categories: Optional[List[Union[str, "Categories"]]] = None, - user_impact: Optional[Union[str, "UserImpact"]] = None, - implementation_effort: Optional[Union[str, "ImplementationEffort"]] = None, - threats: Optional[List[Union[str, "Threats"]]] = None, - preview: Optional[bool] = None, - partner_data: Optional["SecurityAssessmentMetadataPartnerData"] = None, - **kwargs - ): - super(SecurityAssessmentMetadataProperties, self).__init__(**kwargs) - self.display_name = display_name - self.policy_definition_id = None - self.description = description - self.remediation_description = remediation_description - self.categories = categories - self.severity = severity - self.user_impact = user_impact - self.implementation_effort = implementation_effort - self.threats = threats - self.preview = preview - self.assessment_type = assessment_type - self.partner_data = partner_data - - -class SecurityAssessmentPartnerData(msrest.serialization.Model): - """Data regarding 3rd party partner integration. - - All required parameters must be populated in order to send to Azure. - - :param partner_name: Required. Name of the company of the partner. - :type partner_name: str - :param secret: Required. secret to authenticate the partner - write only. - :type secret: str + :ivar value: Required. The list of security connectors under the given scope. + :vartype value: list[~azure.mgmt.security.models.SecurityConnector] + :ivar next_link: The URI to fetch the next page. + :vartype next_link: str """ _validation = { - 'partner_name': {'required': True}, - 'secret': {'required': True}, + 'value': {'required': True}, + 'next_link': {'readonly': True}, } _attribute_map = { - 'partner_name': {'key': 'partnerName', 'type': 'str'}, - 'secret': {'key': 'secret', 'type': 'str'}, + 'value': {'key': 'value', 'type': '[SecurityConnector]'}, + 'next_link': {'key': 'nextLink', 'type': 'str'}, } def __init__( self, *, - partner_name: str, - secret: str, + value: List["SecurityConnector"], **kwargs ): - super(SecurityAssessmentPartnerData, self).__init__(**kwargs) - self.partner_name = partner_name - self.secret = secret + """ + :keyword value: Required. The list of security connectors under the given scope. + :paramtype value: list[~azure.mgmt.security.models.SecurityConnector] + """ + super(SecurityConnectorsList, self).__init__(**kwargs) + self.value = value + self.next_link = None class SecurityContact(Resource): @@ -8662,16 +11777,16 @@ class SecurityContact(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param email: The email of this security contact. - :type email: str - :param phone: The phone number of this security contact. - :type phone: str - :param alert_notifications: Whether to send security alerts notifications to the security + :ivar email: The email of this security contact. + :vartype email: str + :ivar phone: The phone number of this security contact. + :vartype phone: str + :ivar alert_notifications: Whether to send security alerts notifications to the security contact. Possible values include: "On", "Off". - :type alert_notifications: str or ~azure.mgmt.security.models.AlertNotifications - :param alerts_to_admins: Whether to send security alerts notifications to subscription admins. + :vartype alert_notifications: str or ~azure.mgmt.security.models.AlertNotifications + :ivar alerts_to_admins: Whether to send security alerts notifications to subscription admins. Possible values include: "On", "Off". - :type alerts_to_admins: str or ~azure.mgmt.security.models.AlertsToAdmins + :vartype alerts_to_admins: str or ~azure.mgmt.security.models.AlertsToAdmins """ _validation = { @@ -8699,6 +11814,18 @@ def __init__( alerts_to_admins: Optional[Union[str, "AlertsToAdmins"]] = None, **kwargs ): + """ + :keyword email: The email of this security contact. + :paramtype email: str + :keyword phone: The phone number of this security contact. + :paramtype phone: str + :keyword alert_notifications: Whether to send security alerts notifications to the security + contact. Possible values include: "On", "Off". + :paramtype alert_notifications: str or ~azure.mgmt.security.models.AlertNotifications + :keyword alerts_to_admins: Whether to send security alerts notifications to subscription + admins. Possible values include: "On", "Off". + :paramtype alerts_to_admins: str or ~azure.mgmt.security.models.AlertsToAdmins + """ super(SecurityContact, self).__init__(**kwargs) self.email = email self.phone = phone @@ -8731,6 +11858,8 @@ def __init__( self, **kwargs ): + """ + """ super(SecurityContactList, self).__init__(**kwargs) self.value = None self.next_link = None @@ -8749,16 +11878,16 @@ class SecuritySolution(Resource, Location): :vartype name: str :ivar type: Resource type. :vartype type: str - :param security_family: The security family of the security solution. Possible values include: + :ivar security_family: The security family of the security solution. Possible values include: "Waf", "Ngfw", "SaasWaf", "Va". - :type security_family: str or ~azure.mgmt.security.models.SecurityFamily - :param provisioning_state: The security family provisioning State. Possible values include: + :vartype security_family: str or ~azure.mgmt.security.models.SecurityFamily + :ivar provisioning_state: The security family provisioning State. Possible values include: "Succeeded", "Failed", "Updating". - :type provisioning_state: str or ~azure.mgmt.security.models.ProvisioningState - :param template: The security solutions' template. - :type template: str - :param protection_status: The security solutions' status. - :type protection_status: str + :vartype provisioning_state: str or ~azure.mgmt.security.models.ProvisioningState + :ivar template: The security solutions' template. + :vartype template: str + :ivar protection_status: The security solutions' status. + :vartype protection_status: str """ _validation = { @@ -8788,6 +11917,18 @@ def __init__( protection_status: Optional[str] = None, **kwargs ): + """ + :keyword security_family: The security family of the security solution. Possible values + include: "Waf", "Ngfw", "SaasWaf", "Va". + :paramtype security_family: str or ~azure.mgmt.security.models.SecurityFamily + :keyword provisioning_state: The security family provisioning State. Possible values include: + "Succeeded", "Failed", "Updating". + :paramtype provisioning_state: str or ~azure.mgmt.security.models.ProvisioningState + :keyword template: The security solutions' template. + :paramtype template: str + :keyword protection_status: The security solutions' status. + :paramtype protection_status: str + """ super(SecuritySolution, self).__init__(**kwargs) self.location = None self.security_family = security_family @@ -8808,8 +11949,8 @@ class SecuritySolutionList(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param value: - :type value: list[~azure.mgmt.security.models.SecuritySolution] + :ivar value: + :vartype value: list[~azure.mgmt.security.models.SecuritySolution] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -8829,6 +11970,10 @@ def __init__( value: Optional[List["SecuritySolution"]] = None, **kwargs ): + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.SecuritySolution] + """ super(SecuritySolutionList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -8849,21 +11994,21 @@ class SecuritySolutionsReferenceData(Resource, Location): :vartype name: str :ivar type: Resource type. :vartype type: str - :param security_family: Required. The security family of the security solution. Possible values + :ivar security_family: Required. The security family of the security solution. Possible values include: "Waf", "Ngfw", "SaasWaf", "Va". - :type security_family: str or ~azure.mgmt.security.models.SecurityFamily - :param alert_vendor_name: Required. The security solutions' vendor name. - :type alert_vendor_name: str - :param package_info_url: Required. The security solutions' package info url. - :type package_info_url: str - :param product_name: Required. The security solutions' product name. - :type product_name: str - :param publisher: Required. The security solutions' publisher. - :type publisher: str - :param publisher_display_name: Required. The security solutions' publisher display name. - :type publisher_display_name: str - :param template: Required. The security solutions' template. - :type template: str + :vartype security_family: str or ~azure.mgmt.security.models.SecurityFamily + :ivar alert_vendor_name: Required. The security solutions' vendor name. + :vartype alert_vendor_name: str + :ivar package_info_url: Required. The security solutions' package info url. + :vartype package_info_url: str + :ivar product_name: Required. The security solutions' product name. + :vartype product_name: str + :ivar publisher: Required. The security solutions' publisher. + :vartype publisher: str + :ivar publisher_display_name: Required. The security solutions' publisher display name. + :vartype publisher_display_name: str + :ivar template: Required. The security solutions' template. + :vartype template: str """ _validation = { @@ -8906,6 +12051,23 @@ def __init__( template: str, **kwargs ): + """ + :keyword security_family: Required. The security family of the security solution. Possible + values include: "Waf", "Ngfw", "SaasWaf", "Va". + :paramtype security_family: str or ~azure.mgmt.security.models.SecurityFamily + :keyword alert_vendor_name: Required. The security solutions' vendor name. + :paramtype alert_vendor_name: str + :keyword package_info_url: Required. The security solutions' package info url. + :paramtype package_info_url: str + :keyword product_name: Required. The security solutions' product name. + :paramtype product_name: str + :keyword publisher: Required. The security solutions' publisher. + :paramtype publisher: str + :keyword publisher_display_name: Required. The security solutions' publisher display name. + :paramtype publisher_display_name: str + :keyword template: Required. The security solutions' template. + :paramtype template: str + """ super(SecuritySolutionsReferenceData, self).__init__(**kwargs) self.location = None self.security_family = security_family @@ -8930,8 +12092,8 @@ def __init__( class SecuritySolutionsReferenceDataList(msrest.serialization.Model): """SecuritySolutionsReferenceDataList. - :param value: - :type value: list[~azure.mgmt.security.models.SecuritySolutionsReferenceData] + :ivar value: + :vartype value: list[~azure.mgmt.security.models.SecuritySolutionsReferenceData] """ _attribute_map = { @@ -8944,6 +12106,10 @@ def __init__( value: Optional[List["SecuritySolutionsReferenceData"]] = None, **kwargs ): + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.SecuritySolutionsReferenceData] + """ super(SecuritySolutionsReferenceDataList, self).__init__(**kwargs) self.value = value @@ -8963,8 +12129,8 @@ class SecuritySubAssessment(Resource): :vartype id_properties_id: str :ivar display_name: User friendly display name of the sub-assessment. :vartype display_name: str - :param status: Status of the sub-assessment. - :type status: ~azure.mgmt.security.models.SubAssessmentStatus + :ivar status: Status of the sub-assessment. + :vartype status: ~azure.mgmt.security.models.SubAssessmentStatus :ivar remediation: Information on how to remediate this sub-assessment. :vartype remediation: str :ivar impact: Description of the impact of this sub-assessment. @@ -8975,10 +12141,10 @@ class SecuritySubAssessment(Resource): :vartype description: str :ivar time_generated: The date and time the sub-assessment was generated. :vartype time_generated: ~datetime.datetime - :param resource_details: Details of the resource that was assessed. - :type resource_details: ~azure.mgmt.security.models.ResourceDetails - :param additional_data: Details of the sub-assessment. - :type additional_data: ~azure.mgmt.security.models.AdditionalData + :ivar resource_details: Details of the resource that was assessed. + :vartype resource_details: ~azure.mgmt.security.models.ResourceDetails + :ivar additional_data: Details of the sub-assessment. + :vartype additional_data: ~azure.mgmt.security.models.AdditionalData """ _validation = { @@ -9018,6 +12184,14 @@ def __init__( additional_data: Optional["AdditionalData"] = None, **kwargs ): + """ + :keyword status: Status of the sub-assessment. + :paramtype status: ~azure.mgmt.security.models.SubAssessmentStatus + :keyword resource_details: Details of the resource that was assessed. + :paramtype resource_details: ~azure.mgmt.security.models.ResourceDetails + :keyword additional_data: Details of the sub-assessment. + :paramtype additional_data: ~azure.mgmt.security.models.AdditionalData + """ super(SecuritySubAssessment, self).__init__(**kwargs) self.id_properties_id = None self.display_name = None @@ -9056,6 +12230,8 @@ def __init__( self, **kwargs ): + """ + """ super(SecuritySubAssessmentList, self).__init__(**kwargs) self.value = None self.next_link = None @@ -9076,9 +12252,9 @@ class SecurityTask(Resource): :vartype state: str :ivar creation_time_utc: The time this task was discovered in UTC. :vartype creation_time_utc: ~datetime.datetime - :param security_task_parameters: Changing set of properties, depending on the task type that is + :ivar security_task_parameters: Changing set of properties, depending on the task type that is derived from the name field. - :type security_task_parameters: ~azure.mgmt.security.models.SecurityTaskParameters + :vartype security_task_parameters: ~azure.mgmt.security.models.SecurityTaskParameters :ivar last_state_change_time_utc: The time this task's details were last changed in UTC. :vartype last_state_change_time_utc: ~datetime.datetime :ivar sub_state: Additional data on the state of the task. @@ -9112,6 +12288,11 @@ def __init__( security_task_parameters: Optional["SecurityTaskParameters"] = None, **kwargs ): + """ + :keyword security_task_parameters: Changing set of properties, depending on the task type that + is derived from the name field. + :paramtype security_task_parameters: ~azure.mgmt.security.models.SecurityTaskParameters + """ super(SecurityTask, self).__init__(**kwargs) self.state = None self.creation_time_utc = None @@ -9145,6 +12326,8 @@ def __init__( self, **kwargs ): + """ + """ super(SecurityTaskList, self).__init__(**kwargs) self.value = None self.next_link = None @@ -9155,9 +12338,9 @@ class SecurityTaskParameters(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param additional_properties: Unmatched properties from the message are deserialized to this + :ivar additional_properties: Unmatched properties from the message are deserialized to this collection. - :type additional_properties: dict[str, any] + :vartype additional_properties: dict[str, any] :ivar name: Name of the task type. :vartype name: str """ @@ -9177,6 +12360,11 @@ def __init__( additional_properties: Optional[Dict[str, Any]] = None, **kwargs ): + """ + :keyword additional_properties: Unmatched properties from the message are deserialized to this + collection. + :paramtype additional_properties: dict[str, any] + """ super(SecurityTaskParameters, self).__init__(**kwargs) self.additional_properties = additional_properties self.name = None @@ -9185,17 +12373,17 @@ def __init__( class SensitivityLabel(msrest.serialization.Model): """The sensitivity label. - :param display_name: The name of the sensitivity label. - :type display_name: str - :param description: The description of the sensitivity label. - :type description: str - :param rank: The rank of the sensitivity label. Possible values include: "None", "Low", + :ivar display_name: The name of the sensitivity label. + :vartype display_name: str + :ivar description: The description of the sensitivity label. + :vartype description: str + :ivar rank: The rank of the sensitivity label. Possible values include: "None", "Low", "Medium", "High", "Critical". - :type rank: str or ~azure.mgmt.security.models.Rank - :param order: The order of the sensitivity label. - :type order: int - :param enabled: Indicates whether the label is enabled or not. - :type enabled: bool + :vartype rank: str or ~azure.mgmt.security.models.Rank + :ivar order: The order of the sensitivity label. + :vartype order: int + :ivar enabled: Indicates whether the label is enabled or not. + :vartype enabled: bool """ _attribute_map = { @@ -9216,6 +12404,19 @@ def __init__( enabled: Optional[bool] = None, **kwargs ): + """ + :keyword display_name: The name of the sensitivity label. + :paramtype display_name: str + :keyword description: The description of the sensitivity label. + :paramtype description: str + :keyword rank: The rank of the sensitivity label. Possible values include: "None", "Low", + "Medium", "High", "Critical". + :paramtype rank: str or ~azure.mgmt.security.models.Rank + :keyword order: The order of the sensitivity label. + :paramtype order: int + :keyword enabled: Indicates whether the label is enabled or not. + :paramtype enabled: bool + """ super(SensitivityLabel, self).__init__(**kwargs) self.display_name = display_name self.description = description @@ -9260,6 +12461,8 @@ def __init__( self, **kwargs ): + """ + """ super(ServerVulnerabilityAssessment, self).__init__(**kwargs) self.provisioning_state = None @@ -9267,8 +12470,8 @@ def __init__( class ServerVulnerabilityAssessmentsList(msrest.serialization.Model): """List of server vulnerability assessments. - :param value: - :type value: list[~azure.mgmt.security.models.ServerVulnerabilityAssessment] + :ivar value: + :vartype value: list[~azure.mgmt.security.models.ServerVulnerabilityAssessment] """ _attribute_map = { @@ -9281,6 +12484,10 @@ def __init__( value: Optional[List["ServerVulnerabilityAssessment"]] = None, **kwargs ): + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.ServerVulnerabilityAssessment] + """ super(ServerVulnerabilityAssessmentsList, self).__init__(**kwargs) self.value = value @@ -9292,10 +12499,10 @@ class ServerVulnerabilityProperties(AdditionalData): All required parameters must be populated in order to send to Azure. - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", + :ivar assessed_resource_type: Required. Sub-assessment resource type.Constant filled by server. + Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType + :vartype assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType :ivar type: Vulnerability Type. e.g: Vulnerability, Potential Vulnerability, Information Gathered. :vartype type: str @@ -9339,6 +12546,8 @@ def __init__( self, **kwargs ): + """ + """ super(ServerVulnerabilityProperties, self).__init__(**kwargs) self.assessed_resource_type = 'ServerVulnerabilityAssessment' # type: str self.type = None @@ -9353,11 +12562,11 @@ def __init__( class ServicePrincipalProperties(msrest.serialization.Model): """Details of the service principal. - :param application_id: Application ID of service principal. - :type application_id: str - :param secret: A secret string that the application uses to prove its identity, also can be + :ivar application_id: Application ID of service principal. + :vartype application_id: str + :ivar secret: A secret string that the application uses to prove its identity, also can be referred to as application password (write only). - :type secret: str + :vartype secret: str """ _attribute_map = { @@ -9372,6 +12581,13 @@ def __init__( secret: Optional[str] = None, **kwargs ): + """ + :keyword application_id: Application ID of service principal. + :paramtype application_id: str + :keyword secret: A secret string that the application uses to prove its identity, also can be + referred to as application password (write only). + :paramtype secret: str + """ super(ServicePrincipalProperties, self).__init__(**kwargs) self.application_id = application_id self.secret = secret @@ -9382,8 +12598,8 @@ class SettingsList(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param value: The settings list. - :type value: list[~azure.mgmt.security.models.Setting] + :ivar value: The settings list. + :vartype value: list[~azure.mgmt.security.models.Setting] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -9403,6 +12619,10 @@ def __init__( value: Optional[List["Setting"]] = None, **kwargs ): + """ + :keyword value: The settings list. + :paramtype value: list[~azure.mgmt.security.models.Setting] + """ super(SettingsList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -9419,27 +12639,27 @@ class Software(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param device_id: Unique identifier for the virtual machine in the service. - :type device_id: str - :param os_platform: Platform of the operating system running on the device. - :type os_platform: str - :param vendor: Name of the software vendor. - :type vendor: str - :param software_name: Name of the software product. - :type software_name: str - :param version: Version number of the software product. - :type version: str - :param end_of_support_status: End of support status. Possible values include: "None", + :ivar device_id: Unique identifier for the virtual machine in the service. + :vartype device_id: str + :ivar os_platform: Platform of the operating system running on the device. + :vartype os_platform: str + :ivar vendor: Name of the software vendor. + :vartype vendor: str + :ivar software_name: Name of the software product. + :vartype software_name: str + :ivar version: Version number of the software product. + :vartype version: str + :ivar end_of_support_status: End of support status. Possible values include: "None", "noLongerSupported", "versionNoLongerSupported", "upcomingNoLongerSupported", "upcomingVersionNoLongerSupported". - :type end_of_support_status: str or ~azure.mgmt.security.models.EndOfSupportStatus - :param end_of_support_date: The end of support date in case the product is upcoming end of + :vartype end_of_support_status: str or ~azure.mgmt.security.models.EndOfSupportStatus + :ivar end_of_support_date: The end of support date in case the product is upcoming end of support. - :type end_of_support_date: str - :param number_of_known_vulnerabilities: Number of weaknesses. - :type number_of_known_vulnerabilities: int - :param first_seen_at: First time that the software was seen in the device. - :type first_seen_at: str + :vartype end_of_support_date: str + :ivar number_of_known_vulnerabilities: Number of weaknesses. + :vartype number_of_known_vulnerabilities: int + :ivar first_seen_at: First time that the software was seen in the device. + :vartype first_seen_at: str """ _validation = { @@ -9477,6 +12697,29 @@ def __init__( first_seen_at: Optional[str] = None, **kwargs ): + """ + :keyword device_id: Unique identifier for the virtual machine in the service. + :paramtype device_id: str + :keyword os_platform: Platform of the operating system running on the device. + :paramtype os_platform: str + :keyword vendor: Name of the software vendor. + :paramtype vendor: str + :keyword software_name: Name of the software product. + :paramtype software_name: str + :keyword version: Version number of the software product. + :paramtype version: str + :keyword end_of_support_status: End of support status. Possible values include: "None", + "noLongerSupported", "versionNoLongerSupported", "upcomingNoLongerSupported", + "upcomingVersionNoLongerSupported". + :paramtype end_of_support_status: str or ~azure.mgmt.security.models.EndOfSupportStatus + :keyword end_of_support_date: The end of support date in case the product is upcoming end of + support. + :paramtype end_of_support_date: str + :keyword number_of_known_vulnerabilities: Number of weaknesses. + :paramtype number_of_known_vulnerabilities: int + :keyword first_seen_at: First time that the software was seen in the device. + :paramtype first_seen_at: str + """ super(Software, self).__init__(**kwargs) self.device_id = device_id self.os_platform = os_platform @@ -9494,8 +12737,8 @@ class SoftwaresList(msrest.serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - :param value: - :type value: list[~azure.mgmt.security.models.Software] + :ivar value: + :vartype value: list[~azure.mgmt.security.models.Software] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -9515,6 +12758,10 @@ def __init__( value: Optional[List["Software"]] = None, **kwargs ): + """ + :keyword value: + :paramtype value: list[~azure.mgmt.security.models.Software] + """ super(SoftwaresList, self).__init__(**kwargs) self.value = value self.next_link = None @@ -9527,10 +12774,10 @@ class SqlServerVulnerabilityProperties(AdditionalData): All required parameters must be populated in order to send to Azure. - :param assessed_resource_type: Required. Sub-assessment resource type.Constant filled by - server. Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", + :ivar assessed_resource_type: Required. Sub-assessment resource type.Constant filled by server. + Possible values include: "SqlServerVulnerability", "ContainerRegistryVulnerability", "ServerVulnerability". - :type assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType + :vartype assessed_resource_type: str or ~azure.mgmt.security.models.AssessedResourceType :ivar type: The resource type the sub assessment refers to in its resource details. :vartype type: str :ivar query: The T-SQL query that runs on your SQL database to perform the particular check. @@ -9553,6 +12800,8 @@ def __init__( self, **kwargs ): + """ + """ super(SqlServerVulnerabilityProperties, self).__init__(**kwargs) self.assessed_resource_type = 'SqlServerVulnerability' # type: str self.type = None @@ -9594,6 +12843,8 @@ def __init__( self, **kwargs ): + """ + """ super(SubAssessmentStatus, self).__init__(**kwargs) self.code = None self.cause = None @@ -9606,9 +12857,9 @@ class SuppressionAlertsScope(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param all_of: Required. All the conditions inside need to be true in order to suppress the + :ivar all_of: Required. All the conditions inside need to be true in order to suppress the alert. - :type all_of: list[~azure.mgmt.security.models.ScopeElement] + :vartype all_of: list[~azure.mgmt.security.models.ScopeElement] """ _validation = { @@ -9625,6 +12876,11 @@ def __init__( all_of: List["ScopeElement"], **kwargs ): + """ + :keyword all_of: Required. All the conditions inside need to be true in order to suppress the + alert. + :paramtype all_of: list[~azure.mgmt.security.models.ScopeElement] + """ super(SuppressionAlertsScope, self).__init__(**kwargs) self.all_of = all_of @@ -9632,20 +12888,20 @@ def __init__( class SystemData(msrest.serialization.Model): """Metadata pertaining to creation and last modification of the resource. - :param created_by: The identity that created the resource. - :type created_by: str - :param created_by_type: The type of identity that created the resource. Possible values - include: "User", "Application", "ManagedIdentity", "Key". - :type created_by_type: str or ~azure.mgmt.security.models.CreatedByType - :param created_at: The timestamp of resource creation (UTC). - :type created_at: ~datetime.datetime - :param last_modified_by: The identity that last modified the resource. - :type last_modified_by: str - :param last_modified_by_type: The type of identity that last modified the resource. Possible + :ivar created_by: The identity that created the resource. + :vartype created_by: str + :ivar created_by_type: The type of identity that created the resource. Possible values include: + "User", "Application", "ManagedIdentity", "Key". + :vartype created_by_type: str or ~azure.mgmt.security.models.CreatedByType + :ivar created_at: The timestamp of resource creation (UTC). + :vartype created_at: ~datetime.datetime + :ivar last_modified_by: The identity that last modified the resource. + :vartype last_modified_by: str + :ivar last_modified_by_type: The type of identity that last modified the resource. Possible values include: "User", "Application", "ManagedIdentity", "Key". - :type last_modified_by_type: str or ~azure.mgmt.security.models.CreatedByType - :param last_modified_at: The timestamp of resource last modification (UTC). - :type last_modified_at: ~datetime.datetime + :vartype last_modified_by_type: str or ~azure.mgmt.security.models.CreatedByType + :ivar last_modified_at: The timestamp of resource last modification (UTC). + :vartype last_modified_at: ~datetime.datetime """ _attribute_map = { @@ -9668,6 +12924,22 @@ def __init__( last_modified_at: Optional[datetime.datetime] = None, **kwargs ): + """ + :keyword created_by: The identity that created the resource. + :paramtype created_by: str + :keyword created_by_type: The type of identity that created the resource. Possible values + include: "User", "Application", "ManagedIdentity", "Key". + :paramtype created_by_type: str or ~azure.mgmt.security.models.CreatedByType + :keyword created_at: The timestamp of resource creation (UTC). + :paramtype created_at: ~datetime.datetime + :keyword last_modified_by: The identity that last modified the resource. + :paramtype last_modified_by: str + :keyword last_modified_by_type: The type of identity that last modified the resource. Possible + values include: "User", "Application", "ManagedIdentity", "Key". + :paramtype last_modified_by_type: str or ~azure.mgmt.security.models.CreatedByType + :keyword last_modified_at: The timestamp of resource last modification (UTC). + :paramtype last_modified_at: ~datetime.datetime + """ super(SystemData, self).__init__(**kwargs) self.created_by = created_by self.created_by_type = created_by_type @@ -9702,6 +12974,8 @@ def __init__( self, **kwargs ): + """ + """ super(TopologyList, self).__init__(**kwargs) self.value = None self.next_link = None @@ -9748,6 +13022,8 @@ def __init__( self, **kwargs ): + """ + """ super(TopologyResource, self).__init__(**kwargs) self.location = None self.calculated_date_time = None @@ -9811,6 +13087,8 @@ def __init__( self, **kwargs ): + """ + """ super(TopologySingleResource, self).__init__(**kwargs) self.resource_id = None self.severity = None @@ -9843,6 +13121,8 @@ def __init__( self, **kwargs ): + """ + """ super(TopologySingleResourceChild, self).__init__(**kwargs) self.resource_id = None @@ -9868,6 +13148,8 @@ def __init__( self, **kwargs ): + """ + """ super(TopologySingleResourceParent, self).__init__(**kwargs) self.resource_id = None @@ -9883,16 +13165,16 @@ class TwinUpdatesNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -9924,6 +13206,16 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ super(TwinUpdatesNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) self.rule_type = 'TwinUpdatesNotInAllowedRange' # type: str @@ -9939,16 +13231,16 @@ class UnauthorizedOperationsNotInAllowedRange(TimeWindowCustomAlertRule): :vartype display_name: str :ivar description: The description of the custom alert. :vartype description: str - :param is_enabled: Required. Status of the custom alert. - :type is_enabled: bool - :param rule_type: Required. The type of the custom alert rule.Constant filled by server. - :type rule_type: str - :param min_threshold: Required. The minimum threshold. - :type min_threshold: int - :param max_threshold: Required. The maximum threshold. - :type max_threshold: int - :param time_window_size: Required. The time window size in iso8601 format. - :type time_window_size: ~datetime.timedelta + :ivar is_enabled: Required. Status of the custom alert. + :vartype is_enabled: bool + :ivar rule_type: Required. The type of the custom alert rule.Constant filled by server. + :vartype rule_type: str + :ivar min_threshold: Required. The minimum threshold. + :vartype min_threshold: int + :ivar max_threshold: Required. The maximum threshold. + :vartype max_threshold: int + :ivar time_window_size: Required. The time window size in iso8601 format. + :vartype time_window_size: ~datetime.timedelta """ _validation = { @@ -9980,6 +13272,16 @@ def __init__( time_window_size: datetime.timedelta, **kwargs ): + """ + :keyword is_enabled: Required. Status of the custom alert. + :paramtype is_enabled: bool + :keyword min_threshold: Required. The minimum threshold. + :paramtype min_threshold: int + :keyword max_threshold: Required. The maximum threshold. + :paramtype max_threshold: int + :keyword time_window_size: Required. The time window size in iso8601 format. + :paramtype time_window_size: ~datetime.timedelta + """ super(UnauthorizedOperationsNotInAllowedRange, self).__init__(is_enabled=is_enabled, min_threshold=min_threshold, max_threshold=max_threshold, time_window_size=time_window_size, **kwargs) self.rule_type = 'UnauthorizedOperationsNotInAllowedRange' # type: str @@ -9987,14 +13289,13 @@ def __init__( class UpdateIotSecuritySolutionData(TagsResource): """UpdateIotSecuritySolutionData. - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :param user_defined_resources: Properties of the IoT Security solution's user defined - resources. - :type user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties - :param recommendations_configuration: List of the configuration status for each recommendation + :ivar tags: A set of tags. Resource tags. + :vartype tags: dict[str, str] + :ivar user_defined_resources: Properties of the IoT Security solution's user defined resources. + :vartype user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties + :ivar recommendations_configuration: List of the configuration status for each recommendation type. - :type recommendations_configuration: + :vartype recommendations_configuration: list[~azure.mgmt.security.models.RecommendationConfigurationProperties] """ @@ -10012,6 +13313,17 @@ def __init__( recommendations_configuration: Optional[List["RecommendationConfigurationProperties"]] = None, **kwargs ): + """ + :keyword tags: A set of tags. Resource tags. + :paramtype tags: dict[str, str] + :keyword user_defined_resources: Properties of the IoT Security solution's user defined + resources. + :paramtype user_defined_resources: ~azure.mgmt.security.models.UserDefinedResourcesProperties + :keyword recommendations_configuration: List of the configuration status for each + recommendation type. + :paramtype recommendations_configuration: + list[~azure.mgmt.security.models.RecommendationConfigurationProperties] + """ super(UpdateIotSecuritySolutionData, self).__init__(tags=tags, **kwargs) self.user_defined_resources = user_defined_resources self.recommendations_configuration = recommendations_configuration @@ -10022,12 +13334,12 @@ class UserDefinedResourcesProperties(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param query: Required. Azure Resource Graph query which represents the security solution's - user defined resources. Required to start with "where type != "Microsoft.Devices/IotHubs"". - :type query: str - :param query_subscriptions: Required. List of Azure subscription ids on which the user defined + :ivar query: Required. Azure Resource Graph query which represents the security solution's user + defined resources. Required to start with "where type != "Microsoft.Devices/IotHubs"". + :vartype query: str + :ivar query_subscriptions: Required. List of Azure subscription ids on which the user defined resources query should be executed. - :type query_subscriptions: list[str] + :vartype query_subscriptions: list[str] """ _validation = { @@ -10047,6 +13359,14 @@ def __init__( query_subscriptions: List[str], **kwargs ): + """ + :keyword query: Required. Azure Resource Graph query which represents the security solution's + user defined resources. Required to start with "where type != "Microsoft.Devices/IotHubs"". + :paramtype query: str + :keyword query_subscriptions: Required. List of Azure subscription ids on which the user + defined resources query should be executed. + :paramtype query_subscriptions: list[str] + """ super(UserDefinedResourcesProperties, self).__init__(**kwargs) self.query = query self.query_subscriptions = query_subscriptions @@ -10055,11 +13375,11 @@ def __init__( class UserRecommendation(msrest.serialization.Model): """Represents a user that is recommended to be allowed for a certain rule. - :param username: Represents a user that is recommended to be allowed for a certain rule. - :type username: str - :param recommendation_action: The recommendation action of the machine or rule. Possible values + :ivar username: Represents a user that is recommended to be allowed for a certain rule. + :vartype username: str + :ivar recommendation_action: The recommendation action of the machine or rule. Possible values include: "Recommended", "Add", "Remove". - :type recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction + :vartype recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction """ _attribute_map = { @@ -10074,6 +13394,13 @@ def __init__( recommendation_action: Optional[Union[str, "RecommendationAction"]] = None, **kwargs ): + """ + :keyword username: Represents a user that is recommended to be allowed for a certain rule. + :paramtype username: str + :keyword recommendation_action: The recommendation action of the machine or rule. Possible + values include: "Recommended", "Add", "Remove". + :paramtype recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction + """ super(UserRecommendation, self).__init__(**kwargs) self.username = username self.recommendation_action = recommendation_action @@ -10082,26 +13409,26 @@ def __init__( class VaRule(msrest.serialization.Model): """vulnerability assessment rule metadata details. - :param rule_id: The rule Id. - :type rule_id: str - :param severity: The rule severity. Possible values include: "High", "Medium", "Low", + :ivar rule_id: The rule Id. + :vartype rule_id: str + :ivar severity: The rule severity. Possible values include: "High", "Medium", "Low", "Informational", "Obsolete". - :type severity: str or ~azure.mgmt.security.models.RuleSeverity - :param category: The rule category. - :type category: str - :param rule_type: The rule type. Possible values include: "Binary", "BaselineExpected", + :vartype severity: str or ~azure.mgmt.security.models.RuleSeverity + :ivar category: The rule category. + :vartype category: str + :ivar rule_type: The rule type. Possible values include: "Binary", "BaselineExpected", "PositiveList", "NegativeList". - :type rule_type: str or ~azure.mgmt.security.models.RuleType - :param title: The rule title. - :type title: str - :param description: The rule description. - :type description: str - :param rationale: The rule rationale. - :type rationale: str - :param query_check: The rule query details. - :type query_check: ~azure.mgmt.security.models.QueryCheck - :param benchmark_references: The benchmark references. - :type benchmark_references: list[~azure.mgmt.security.models.BenchmarkReference] + :vartype rule_type: str or ~azure.mgmt.security.models.RuleType + :ivar title: The rule title. + :vartype title: str + :ivar description: The rule description. + :vartype description: str + :ivar rationale: The rule rationale. + :vartype rationale: str + :ivar query_check: The rule query details. + :vartype query_check: ~azure.mgmt.security.models.QueryCheck + :ivar benchmark_references: The benchmark references. + :vartype benchmark_references: list[~azure.mgmt.security.models.BenchmarkReference] """ _attribute_map = { @@ -10130,6 +13457,28 @@ def __init__( benchmark_references: Optional[List["BenchmarkReference"]] = None, **kwargs ): + """ + :keyword rule_id: The rule Id. + :paramtype rule_id: str + :keyword severity: The rule severity. Possible values include: "High", "Medium", "Low", + "Informational", "Obsolete". + :paramtype severity: str or ~azure.mgmt.security.models.RuleSeverity + :keyword category: The rule category. + :paramtype category: str + :keyword rule_type: The rule type. Possible values include: "Binary", "BaselineExpected", + "PositiveList", "NegativeList". + :paramtype rule_type: str or ~azure.mgmt.security.models.RuleType + :keyword title: The rule title. + :paramtype title: str + :keyword description: The rule description. + :paramtype description: str + :keyword rationale: The rule rationale. + :paramtype rationale: str + :keyword query_check: The rule query details. + :paramtype query_check: ~azure.mgmt.security.models.QueryCheck + :keyword benchmark_references: The benchmark references. + :paramtype benchmark_references: list[~azure.mgmt.security.models.BenchmarkReference] + """ super(VaRule, self).__init__(**kwargs) self.rule_id = rule_id self.severity = severity @@ -10167,6 +13516,8 @@ def __init__( self, **kwargs ): + """ + """ super(VendorReference, self).__init__(**kwargs) self.title = None self.link = None @@ -10175,17 +13526,17 @@ def __init__( class VmRecommendation(msrest.serialization.Model): """Represents a machine that is part of a machine group. - :param configuration_status: The configuration status of the machines group or machine or rule. + :ivar configuration_status: The configuration status of the machines group or machine or rule. Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", "NoStatus". - :type configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus - :param recommendation_action: The recommendation action of the machine or rule. Possible values + :vartype configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus + :ivar recommendation_action: The recommendation action of the machine or rule. Possible values include: "Recommended", "Add", "Remove". - :type recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction - :param resource_id: The full resource id of the machine. - :type resource_id: str - :param enforcement_support: The machine supportability of Enforce feature. Possible values + :vartype recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction + :ivar resource_id: The full resource id of the machine. + :vartype resource_id: str + :ivar enforcement_support: The machine supportability of Enforce feature. Possible values include: "Supported", "NotSupported", "Unknown". - :type enforcement_support: str or ~azure.mgmt.security.models.EnforcementSupport + :vartype enforcement_support: str or ~azure.mgmt.security.models.EnforcementSupport """ _attribute_map = { @@ -10204,6 +13555,20 @@ def __init__( enforcement_support: Optional[Union[str, "EnforcementSupport"]] = None, **kwargs ): + """ + :keyword configuration_status: The configuration status of the machines group or machine or + rule. Possible values include: "Configured", "NotConfigured", "InProgress", "Failed", + "NoStatus". + :paramtype configuration_status: str or ~azure.mgmt.security.models.ConfigurationStatus + :keyword recommendation_action: The recommendation action of the machine or rule. Possible + values include: "Recommended", "Add", "Remove". + :paramtype recommendation_action: str or ~azure.mgmt.security.models.RecommendationAction + :keyword resource_id: The full resource id of the machine. + :paramtype resource_id: str + :keyword enforcement_support: The machine supportability of Enforce feature. Possible values + include: "Supported", "NotSupported", "Unknown". + :paramtype enforcement_support: str or ~azure.mgmt.security.models.EnforcementSupport + """ super(VmRecommendation, self).__init__(**kwargs) self.configuration_status = configuration_status self.recommendation_action = recommendation_action @@ -10222,11 +13587,11 @@ class WorkspaceSetting(Resource): :vartype name: str :ivar type: Resource type. :vartype type: str - :param workspace_id: The full Azure ID of the workspace to save the data in. - :type workspace_id: str - :param scope: All the VMs in this scope will send their security data to the mentioned - workspace unless overridden by a setting with more specific scope. - :type scope: str + :ivar workspace_id: The full Azure ID of the workspace to save the data in. + :vartype workspace_id: str + :ivar scope: All the VMs in this scope will send their security data to the mentioned workspace + unless overridden by a setting with more specific scope. + :vartype scope: str """ _validation = { @@ -10250,6 +13615,13 @@ def __init__( scope: Optional[str] = None, **kwargs ): + """ + :keyword workspace_id: The full Azure ID of the workspace to save the data in. + :paramtype workspace_id: str + :keyword scope: All the VMs in this scope will send their security data to the mentioned + workspace unless overridden by a setting with more specific scope. + :paramtype scope: str + """ super(WorkspaceSetting, self).__init__(**kwargs) self.workspace_id = workspace_id self.scope = scope @@ -10262,8 +13634,8 @@ class WorkspaceSettingList(msrest.serialization.Model): All required parameters must be populated in order to send to Azure. - :param value: Required. List of workspace settings. - :type value: list[~azure.mgmt.security.models.WorkspaceSetting] + :ivar value: Required. List of workspace settings. + :vartype value: list[~azure.mgmt.security.models.WorkspaceSetting] :ivar next_link: The URI to fetch the next page. :vartype next_link: str """ @@ -10284,6 +13656,10 @@ def __init__( value: List["WorkspaceSetting"], **kwargs ): + """ + :keyword value: Required. List of workspace settings. + :paramtype value: list[~azure.mgmt.security.models.WorkspaceSetting] + """ super(WorkspaceSettingList, self).__init__(**kwargs) self.value = value self.next_link = None diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_security_center_enums.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_security_center_enums.py index 5a8510226b69d..4dd2b5744d260 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_security_center_enums.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/models/_security_center_enums.py @@ -6,27 +6,12 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from enum import Enum, EnumMeta +from enum import Enum from six import with_metaclass +from azure.core import CaseInsensitiveEnumMeta -class _CaseInsensitiveEnumMeta(EnumMeta): - def __getitem__(self, name): - return super().__getitem__(name.upper()) - def __getattr__(cls, name): - """Return the enum member matching `name` - We use __getattr__ instead of descriptors or inserting into the enum - class' __dict__ in order to support `name` and `value` being both - properties for enum members (which live in the class' __dict__) and - enum members themselves. - """ - try: - return cls._member_map_[name.upper()] - except KeyError: - raise AttributeError(name) - - -class AadConnectivityStateEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AadConnectivityStateEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The connectivity state of the external AAD solution """ @@ -34,7 +19,7 @@ class AadConnectivityStateEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enu NOT_LICENSED = "NotLicensed" CONNECTED = "Connected" -class ActionType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ActionType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The type of the action that will be triggered by the Automation """ @@ -42,7 +27,7 @@ class ActionType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): EVENT_HUB = "EventHub" WORKSPACE = "Workspace" -class AdaptiveApplicationControlIssue(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AdaptiveApplicationControlIssue(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """An alert that machines within a group can have """ @@ -53,20 +38,20 @@ class AdaptiveApplicationControlIssue(with_metaclass(_CaseInsensitiveEnumMeta, s EXECUTABLE_VIOLATIONS_AUDITED = "ExecutableViolationsAudited" RULES_VIOLATED_MANUALLY = "RulesViolatedManually" -class AdditionalWorkspaceDataType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AdditionalWorkspaceDataType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Data types sent to workspace. """ ALERTS = "Alerts" RAW_EVENTS = "RawEvents" -class AdditionalWorkspaceType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AdditionalWorkspaceType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Workspace type. """ SENTINEL = "Sentinel" -class AlertNotifications(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AlertNotifications(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Whether to send security alerts notifications to the security contact """ @@ -75,7 +60,7 @@ class AlertNotifications(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Don't get notifications on new alerts. OFF = "Off" -class AlertSeverity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AlertSeverity(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The risk level of the threat that was detected. Learn more: https://docs.microsoft.com/en-us/azure/security-center/security-center-alerts-overview#how-are-alerts-classified. """ @@ -89,7 +74,7 @@ class AlertSeverity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: High. HIGH = "High" -class AlertStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AlertStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The life cycle status of the alert. """ @@ -100,7 +85,7 @@ class AlertStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Alert dismissed as false positive. DISMISSED = "Dismissed" -class AlertsToAdmins(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AlertsToAdmins(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Whether to send security alerts notifications to subscription admins """ @@ -109,7 +94,7 @@ class AlertsToAdmins(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Don't send notification on new alerts to the subscription's admins. OFF = "Off" -class AssessedResourceType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AssessedResourceType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Sub-assessment resource type """ @@ -117,7 +102,7 @@ class AssessedResourceType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): CONTAINER_REGISTRY_VULNERABILITY = "ContainerRegistryVulnerability" SERVER_VULNERABILITY = "ServerVulnerability" -class AssessmentStatusCode(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AssessmentStatusCode(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Programmatic code for the status of the assessment """ @@ -128,7 +113,7 @@ class AssessmentStatusCode(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Assessment for this resource did not happen. NOT_APPLICABLE = "NotApplicable" -class AssessmentType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AssessmentType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """BuiltIn if the assessment based on built-in Azure Policy definition, Custom if the assessment based on custom Azure Policy definition """ @@ -143,7 +128,7 @@ class AssessmentType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: An assessment that was created by a verified 3rd party if the user connected it to ASC. VERIFIED_PARTNER = "VerifiedPartner" -class AuthenticationProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AuthenticationProvisioningState(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """State of the multi-cloud connector """ @@ -156,7 +141,7 @@ class AuthenticationProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, s #: Incorrect policy of the connector. INCORRECT_POLICY = "IncorrectPolicy" -class AuthenticationType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AuthenticationType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Connect to your cloud account, for AWS use either account credentials or role-based authentication. For GCP use account organization credentials. """ @@ -168,7 +153,7 @@ class AuthenticationType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: GCP account connector service to service authentication. GCP_CREDENTIALS = "gcpCredentials" -class AutoProvision(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class AutoProvision(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Describes what kind of security agent provisioning action to take """ @@ -177,7 +162,7 @@ class AutoProvision(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Do not install security agent on the VMs automatically. OFF = "Off" -class BundleType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class BundleType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Alert Simulator supported bundles. """ @@ -190,7 +175,7 @@ class BundleType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): STORAGE_ACCOUNTS = "StorageAccounts" VIRTUAL_MACHINES = "VirtualMachines" -class Categories(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Categories(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The categories of resource that is at risk when the assessment is unhealthy """ @@ -200,7 +185,15 @@ class Categories(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): IDENTITY_AND_ACCESS = "IdentityAndAccess" IO_T = "IoT" -class ConfigurationStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class CloudName(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): + """The multi cloud resource's cloud name. + """ + + AZURE = "Azure" + AWS = "AWS" + GCP = "GCP" + +class ConfigurationStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The configuration status of the machines group or machine or rule """ @@ -210,12 +203,12 @@ class ConfigurationStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): FAILED = "Failed" NO_STATUS = "NoStatus" -class ConnectionType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ConnectionType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): INTERNAL = "Internal" EXTERNAL = "External" -class ControlType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ControlType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The type of security control (for example, BuiltIn) """ @@ -224,7 +217,7 @@ class ControlType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Non Azure Security Center managed assessments. CUSTOM = "Custom" -class CreatedByType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class CreatedByType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The type of identity that created the resource. """ @@ -233,19 +226,19 @@ class CreatedByType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): MANAGED_IDENTITY = "ManagedIdentity" KEY = "Key" -class DataSource(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class DataSource(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): #: Devices twin data. TWIN_DATA = "TwinData" -class Direction(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Direction(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The rule's direction """ INBOUND = "Inbound" OUTBOUND = "Outbound" -class EndOfSupportStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class EndOfSupportStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """End of support status. """ @@ -255,7 +248,7 @@ class EndOfSupportStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): UPCOMING_NO_LONGER_SUPPORTED = "upcomingNoLongerSupported" UPCOMING_VERSION_NO_LONGER_SUPPORTED = "upcomingVersionNoLongerSupported" -class EnforcementMode(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class EnforcementMode(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The application control policy enforcement/protection mode of the machine group """ @@ -263,7 +256,7 @@ class EnforcementMode(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): ENFORCE = "Enforce" NONE = "None" -class EnforcementSupport(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class EnforcementSupport(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The machine supportability of Enforce feature """ @@ -271,7 +264,7 @@ class EnforcementSupport(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): NOT_SUPPORTED = "NotSupported" UNKNOWN = "Unknown" -class Enum13(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Enum15(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): ACTIVATE = "Activate" DISMISS = "Dismiss" @@ -279,24 +272,26 @@ class Enum13(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): RESOLVE = "Resolve" CLOSE = "Close" -class Enum15(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Enum17(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): EFFECTIVE = "effective" CUSTOM = "custom" -class Enum69(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Enum73(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): MCAS = "MCAS" WDATP = "WDATP" WDATP_EXCLUDE_LINUX_PUBLIC_PREVIEW = "WDATP_EXCLUDE_LINUX_PUBLIC_PREVIEW" SENTINEL = "Sentinel" -class EventSource(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class EventSource(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """A valid event source type. """ ASSESSMENTS = "Assessments" + ASSESSMENTS_SNAPSHOT = "AssessmentsSnapshot" SUB_ASSESSMENTS = "SubAssessments" + SUB_ASSESSMENTS_SNAPSHOT = "SubAssessmentsSnapshot" ALERTS = "Alerts" SECURE_SCORES = "SecureScores" SECURE_SCORES_SNAPSHOT = "SecureScoresSnapshot" @@ -305,24 +300,24 @@ class EventSource(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): REGULATORY_COMPLIANCE_ASSESSMENT = "RegulatoryComplianceAssessment" REGULATORY_COMPLIANCE_ASSESSMENT_SNAPSHOT = "RegulatoryComplianceAssessmentSnapshot" -class ExpandControlsEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ExpandControlsEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): #: Add definition object for each control. DEFINITION = "definition" -class ExpandEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ExpandEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): #: All links associated with an assessment. LINKS = "links" #: Assessment metadata. METADATA = "metadata" -class ExportData(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ExportData(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): #: Agent raw events. RAW_EVENTS = "RawEvents" -class ExternalSecuritySolutionKindEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ExternalSecuritySolutionKindEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The kind of the external solution """ @@ -330,7 +325,7 @@ class ExternalSecuritySolutionKindEnum(with_metaclass(_CaseInsensitiveEnumMeta, ATA = "ATA" AAD = "AAD" -class FileType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class FileType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The type of the file (for Linux files - Executable is used) """ @@ -341,7 +336,7 @@ class FileType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): EXECUTABLE = "Executable" UNKNOWN = "Unknown" -class HybridComputeProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class HybridComputeProvisioningState(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """State of the service principal and its secret """ @@ -352,7 +347,7 @@ class HybridComputeProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, st #: the service principal details are expired. EXPIRED = "Expired" -class ImplementationEffort(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ImplementationEffort(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The implementation effort required to remediate this assessment """ @@ -360,7 +355,7 @@ class ImplementationEffort(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): MODERATE = "Moderate" HIGH = "High" -class Intent(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Intent(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The kill chain related intent behind the alert. For list of supported values, and explanations of Azure Security Center's supported kill chain intents. """ @@ -418,14 +413,22 @@ class Intent(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: This stage is relevant for compute hosts and resources such as user accounts, certificates etc. EXPLOITATION = "Exploitation" -class KindEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class KindEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The kind of alert simulation. """ #: Simulate alerts according to bundles. BUNDLES = "Bundles" -class Operator(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class OfferingType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): + """The type of the security offering. + """ + + CSPM_MONITOR_AWS = "CspmMonitorAws" + DEFENDER_FOR_CONTAINERS_AWS = "DefenderForContainersAws" + DEFENDER_FOR_SERVERS_AWS = "DefenderForServersAws" + +class Operator(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """A valid comparer operator to use. A case-insensitive comparison will be applied for String PropertyType. """ @@ -449,7 +452,14 @@ class Operator(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Applies only for non-decimal operands. ENDS_WITH = "EndsWith" -class PermissionProperty(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class OrganizationMembershipType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): + """The multi cloud account's membership type in the organization + """ + + MEMBER = "Member" + ORGANIZATION = "Organization" + +class PermissionProperty(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """A permission detected in the cloud account. """ @@ -463,7 +473,7 @@ class PermissionProperty(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: This permission provides read only access to GCP Security Command Center. GCP_SECURITY_CENTER_ADMIN_VIEWER = "GCP::Security Center Admin Viewer" -class PricingTier(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class PricingTier(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The pricing tier value. Azure Security Center is provided in two pricing tiers: free and standard, with the standard tier available with a trial period. The standard tier offers advanced security capabilities, while the free tier offers basic security features. @@ -474,7 +484,7 @@ class PricingTier(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Get the standard Azure security center experience with advanced security features. STANDARD = "Standard" -class PropertyType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class PropertyType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The data type of the compared operands (string, integer, floating point number or a boolean [true/false]] """ @@ -484,13 +494,13 @@ class PropertyType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): NUMBER = "Number" BOOLEAN = "Boolean" -class ProtocolEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ProtocolEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): TCP = "TCP" UDP = "UDP" ALL = "*" -class ProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ProvisioningState(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The security family provisioning State """ @@ -498,7 +508,7 @@ class ProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): FAILED = "Failed" UPDATING = "Updating" -class Rank(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Rank(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The rank of the sensitivity label. """ @@ -508,7 +518,7 @@ class Rank(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): HIGH = "High" CRITICAL = "Critical" -class RecommendationAction(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RecommendationAction(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The recommendation action of the machine or rule """ @@ -516,7 +526,7 @@ class RecommendationAction(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): ADD = "Add" REMOVE = "Remove" -class RecommendationConfigStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RecommendationConfigStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Recommendation status. When the recommendation status is disabled recommendations are not generated. """ @@ -524,7 +534,7 @@ class RecommendationConfigStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, E DISABLED = "Disabled" ENABLED = "Enabled" -class RecommendationStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RecommendationStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The initial recommendation status of the machine group or machine """ @@ -533,7 +543,7 @@ class RecommendationStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): NOT_AVAILABLE = "NotAvailable" NO_STATUS = "NoStatus" -class RecommendationType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RecommendationType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The type of IoT Security recommendation. """ @@ -582,7 +592,7 @@ class RecommendationType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Insecure TLS configurations detected. Immediate upgrade recommended. IO_T_VULNERABLE_TLS_CIPHER_SUITE = "IoT_VulnerableTLSCipherSuite" -class ReportedSeverity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ReportedSeverity(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Assessed alert severity. """ @@ -591,7 +601,7 @@ class ReportedSeverity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): MEDIUM = "Medium" HIGH = "High" -class ResourceIdentifierType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ResourceIdentifierType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """There can be multiple identifiers of different type per alert, this field specify the identifier type. """ @@ -599,7 +609,7 @@ class ResourceIdentifierType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum) AZURE_RESOURCE = "AzureResource" LOG_ANALYTICS = "LogAnalytics" -class ResourceStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ResourceStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The status of the resource regarding a single assessment """ @@ -612,7 +622,7 @@ class ResourceStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: This assessment on the resource is not healthy. NOT_HEALTHY = "NotHealthy" -class RuleSeverity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RuleSeverity(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The rule severity. """ @@ -627,7 +637,7 @@ class RuleSeverity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Obsolete. OBSOLETE = "Obsolete" -class RuleState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RuleState(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Possible states of the rule """ @@ -635,7 +645,7 @@ class RuleState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): DISABLED = "Disabled" EXPIRED = "Expired" -class RuleStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RuleStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The rule result status. """ @@ -646,7 +656,7 @@ class RuleStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: InternalError. INTERNAL_ERROR = "InternalError" -class RuleType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class RuleType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The rule type. """ @@ -659,7 +669,7 @@ class RuleType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: NegativeList. NEGATIVE_LIST = "NegativeList" -class ScanState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ScanState(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The scan status. """ @@ -672,7 +682,7 @@ class ScanState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Passed. PASSED = "Passed" -class ScanTriggerType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ScanTriggerType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The scan trigger type. """ @@ -681,7 +691,7 @@ class ScanTriggerType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: Recurring. RECURRING = "Recurring" -class SecurityFamily(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class SecurityFamily(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The security family of the discovered solution """ @@ -690,14 +700,14 @@ class SecurityFamily(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): SAAS_WAF = "SaasWaf" VA = "Va" -class SecuritySolutionStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class SecuritySolutionStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Status of the IoT Security solution. """ ENABLED = "Enabled" DISABLED = "Disabled" -class ServerVulnerabilityAssessmentPropertiesProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ServerVulnerabilityAssessmentPropertiesProvisioningState(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The provisioningState of the vulnerability assessment capability on the VM """ @@ -707,7 +717,7 @@ class ServerVulnerabilityAssessmentPropertiesProvisioningState(with_metaclass(_C PROVISIONING = "Provisioning" DEPROVISIONING = "Deprovisioning" -class SettingKind(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class SettingKind(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """the kind of the settings string """ @@ -715,7 +725,7 @@ class SettingKind(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): ALERT_SUPPRESSION_SETTING = "AlertSuppressionSetting" ALERT_SYNC_SETTINGS = "AlertSyncSettings" -class Severity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Severity(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The sub-assessment severity level """ @@ -723,7 +733,15 @@ class Severity(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): MEDIUM = "Medium" HIGH = "High" -class Source(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class SeverityEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): + """The severity to relate to the assessments generated by this assessment automation. + """ + + HIGH = "High" + MEDIUM = "Medium" + LOW = "Low" + +class Source(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The platform where the assessed resource resides """ @@ -734,7 +752,7 @@ class Source(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: SQL Resource in an on premise machine connected to Azure cloud. ON_PREMISE_SQL = "OnPremiseSql" -class SourceSystem(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class SourceSystem(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The source type of the machine group """ @@ -744,7 +762,7 @@ class SourceSystem(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): NON_AZURE_AUDIT_D = "NonAzure_AuditD" NONE = "None" -class State(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class State(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Aggregative state based on the standard's supported controls states """ @@ -758,14 +776,14 @@ class State(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): #: No supported regulatory compliance data for the given standard. UNSUPPORTED = "Unsupported" -class Status(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class Status(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The status of the port """ REVOKED = "Revoked" INITIATED = "Initiated" -class StatusReason(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class StatusReason(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """A description of why the ``status`` has its value """ @@ -773,7 +791,7 @@ class StatusReason(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): USER_REQUESTED = "UserRequested" NEWER_REQUEST_INITIATED = "NewerRequestInitiated" -class SubAssessmentStatusCode(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class SubAssessmentStatusCode(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Programmatic code for the status of the assessment """ @@ -784,7 +802,141 @@ class SubAssessmentStatusCode(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum #: Assessment for this resource did not happen. NOT_APPLICABLE = "NotApplicable" -class Threats(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class SupportedCloudEnum(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): + """Relevant cloud for the custom assessment automation. + """ + + AWS = "AWS" + +class Tactics(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): + """Tactic of the assessment + """ + + RECONNAISSANCE = "Reconnaissance" + RESOURCE_DEVELOPMENT = "Resource Development" + INITIAL_ACCESS = "Initial Access" + EXECUTION = "Execution" + PERSISTENCE = "Persistence" + PRIVILEGE_ESCALATION = "Privilege Escalation" + DEFENSE_EVASION = "Defense Evasion" + CREDENTIAL_ACCESS = "Credential Access" + DISCOVERY = "Discovery" + LATERAL_MOVEMENT = "Lateral Movement" + COLLECTION = "Collection" + COMMAND_AND_CONTROL = "Command and Control" + EXFILTRATION = "Exfiltration" + IMPACT = "Impact" + +class Techniques(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): + """Techniques of the assessment + """ + + ABUSE_ELEVATION_CONTROL_MECHANISM = "Abuse Elevation Control Mechanism" + ACCESS_TOKEN_MANIPULATION = "Access Token Manipulation" + ACCOUNT_DISCOVERY = "Account Discovery" + ACCOUNT_MANIPULATION = "Account Manipulation" + ACTIVE_SCANNING = "Active Scanning" + APPLICATION_LAYER_PROTOCOL = "Application Layer Protocol" + AUDIO_CAPTURE = "Audio Capture" + BOOT_OR_LOGON_AUTOSTART_EXECUTION = "Boot or Logon Autostart Execution" + BOOT_OR_LOGON_INITIALIZATION_SCRIPTS = "Boot or Logon Initialization Scripts" + BRUTE_FORCE = "Brute Force" + CLOUD_INFRASTRUCTURE_DISCOVERY = "Cloud Infrastructure Discovery" + CLOUD_SERVICE_DASHBOARD = "Cloud Service Dashboard" + CLOUD_SERVICE_DISCOVERY = "Cloud Service Discovery" + COMMAND_AND_SCRIPTING_INTERPRETER = "Command and Scripting Interpreter" + COMPROMISE_CLIENT_SOFTWARE_BINARY = "Compromise Client Software Binary" + COMPROMISE_INFRASTRUCTURE = "Compromise Infrastructure" + CONTAINER_AND_RESOURCE_DISCOVERY = "Container and Resource Discovery" + CREATE_ACCOUNT = "Create Account" + CREATE_OR_MODIFY_SYSTEM_PROCESS = "Create or Modify System Process" + CREDENTIALS_FROM_PASSWORD_STORES = "Credentials from Password Stores" + DATA_DESTRUCTION = "Data Destruction" + DATA_ENCRYPTED_FOR_IMPACT = "Data Encrypted for Impact" + DATA_FROM_CLOUD_STORAGE_OBJECT = "Data from Cloud Storage Object" + DATA_FROM_CONFIGURATION_REPOSITORY = "Data from Configuration Repository" + DATA_FROM_INFORMATION_REPOSITORIES = "Data from Information Repositories" + DATA_FROM_LOCAL_SYSTEM = "Data from Local System" + DATA_MANIPULATION = "Data Manipulation" + DATA_STAGED = "Data Staged" + DEFACEMENT = "Defacement" + DEOBFUSCATE_DECODE_FILES_OR_INFORMATION = "Deobfuscate/Decode Files or Information" + DISK_WIPE = "Disk Wipe" + DOMAIN_TRUST_DISCOVERY = "Domain Trust Discovery" + DRIVE_BY_COMPROMISE = "Drive-by Compromise" + DYNAMIC_RESOLUTION = "Dynamic Resolution" + ENDPOINT_DENIAL_OF_SERVICE = "Endpoint Denial of Service" + EVENT_TRIGGERED_EXECUTION = "Event Triggered Execution" + EXFILTRATION_OVER_ALTERNATIVE_PROTOCOL = "Exfiltration Over Alternative Protocol" + EXPLOIT_PUBLIC_FACING_APPLICATION = "Exploit Public-Facing Application" + EXPLOITATION_FOR_CLIENT_EXECUTION = "Exploitation for Client Execution" + EXPLOITATION_FOR_CREDENTIAL_ACCESS = "Exploitation for Credential Access" + EXPLOITATION_FOR_DEFENSE_EVASION = "Exploitation for Defense Evasion" + EXPLOITATION_FOR_PRIVILEGE_ESCALATION = "Exploitation for Privilege Escalation" + EXPLOITATION_OF_REMOTE_SERVICES = "Exploitation of Remote Services" + EXTERNAL_REMOTE_SERVICES = "External Remote Services" + FALLBACK_CHANNELS = "Fallback Channels" + FILE_AND_DIRECTORY_DISCOVERY = "File and Directory Discovery" + GATHER_VICTIM_NETWORK_INFORMATION = "Gather Victim Network Information" + HIDE_ARTIFACTS = "Hide Artifacts" + HIJACK_EXECUTION_FLOW = "Hijack Execution Flow" + IMPAIR_DEFENSES = "Impair Defenses" + IMPLANT_CONTAINER_IMAGE = "Implant Container Image" + INDICATOR_REMOVAL_ON_HOST = "Indicator Removal on Host" + INDIRECT_COMMAND_EXECUTION = "Indirect Command Execution" + INGRESS_TOOL_TRANSFER = "Ingress Tool Transfer" + INPUT_CAPTURE = "Input Capture" + INTER_PROCESS_COMMUNICATION = "Inter-Process Communication" + LATERAL_TOOL_TRANSFER = "Lateral Tool Transfer" + MAN_IN_THE_MIDDLE = "Man-in-the-Middle" + MASQUERADING = "Masquerading" + MODIFY_AUTHENTICATION_PROCESS = "Modify Authentication Process" + MODIFY_REGISTRY = "Modify Registry" + NETWORK_DENIAL_OF_SERVICE = "Network Denial of Service" + NETWORK_SERVICE_SCANNING = "Network Service Scanning" + NETWORK_SNIFFING = "Network Sniffing" + NON_APPLICATION_LAYER_PROTOCOL = "Non-Application Layer Protocol" + NON_STANDARD_PORT = "Non-Standard Port" + OBTAIN_CAPABILITIES = "Obtain Capabilities" + OBFUSCATED_FILES_OR_INFORMATION = "Obfuscated Files or Information" + OFFICE_APPLICATION_STARTUP = "Office Application Startup" + OS_CREDENTIAL_DUMPING = "OS Credential Dumping" + PERMISSION_GROUPS_DISCOVERY = "Permission Groups Discovery" + PHISHING = "Phishing" + PRE_OS_BOOT = "Pre-OS Boot" + PROCESS_DISCOVERY = "Process Discovery" + PROCESS_INJECTION = "Process Injection" + PROTOCOL_TUNNELING = "Protocol Tunneling" + PROXY = "Proxy" + QUERY_REGISTRY = "Query Registry" + REMOTE_ACCESS_SOFTWARE = "Remote Access Software" + REMOTE_SERVICE_SESSION_HIJACKING = "Remote Service Session Hijacking" + REMOTE_SERVICES = "Remote Services" + REMOTE_SYSTEM_DISCOVERY = "Remote System Discovery" + RESOURCE_HIJACKING = "Resource Hijacking" + SCHEDULED_TASK_JOB = "Scheduled Task/Job" + SCREEN_CAPTURE = "Screen Capture" + SEARCH_VICTIM_OWNED_WEBSITES = "Search Victim-Owned Websites" + SERVER_SOFTWARE_COMPONENT = "Server Software Component" + SERVICE_STOP = "Service Stop" + SIGNED_BINARY_PROXY_EXECUTION = "Signed Binary Proxy Execution" + SOFTWARE_DEPLOYMENT_TOOLS = "Software Deployment Tools" + SQL_STORED_PROCEDURES = "SQL Stored Procedures" + STEAL_OR_FORGE_KERBEROS_TICKETS = "Steal or Forge Kerberos Tickets" + SUBVERT_TRUST_CONTROLS = "Subvert Trust Controls" + SUPPLY_CHAIN_COMPROMISE = "Supply Chain Compromise" + SYSTEM_INFORMATION_DISCOVERY = "System Information Discovery" + TAINT_SHARED_CONTENT = "Taint Shared Content" + TRAFFIC_SIGNALING = "Traffic Signaling" + TRANSFER_DATA_TO_CLOUD_ACCOUNT = "Transfer Data to Cloud Account" + TRUSTED_RELATIONSHIP = "Trusted Relationship" + UNSECURED_CREDENTIALS = "Unsecured Credentials" + USER_EXECUTION = "User Execution" + VALID_ACCOUNTS = "Valid Accounts" + WINDOWS_MANAGEMENT_INSTRUMENTATION = "Windows Management Instrumentation" + FILE_AND_DIRECTORY_PERMISSIONS_MODIFICATION = "File and Directory Permissions Modification" + +class Threats(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Threats impact of the assessment """ @@ -797,12 +949,12 @@ class Threats(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): MISSING_COVERAGE = "missingCoverage" DENIAL_OF_SERVICE = "denialOfService" -class TransportProtocol(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class TransportProtocol(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): TCP = "TCP" UDP = "UDP" -class UnmaskedIpLoggingStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class UnmaskedIpLoggingStatus(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """Unmasked IP address logging status """ @@ -811,7 +963,7 @@ class UnmaskedIpLoggingStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum #: Unmasked IP logging is enabled. ENABLED = "Enabled" -class UserImpact(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class UserImpact(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The user impact of the assessment """ @@ -819,7 +971,7 @@ class UserImpact(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): MODERATE = "Moderate" HIGH = "High" -class ValueType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): +class ValueType(with_metaclass(CaseInsensitiveEnumMeta, str, Enum)): """The value type of the items in the list. """ diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/__init__.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/__init__.py index 65df4fa5ef359..e0a22181ca006 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/__init__.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/__init__.py @@ -6,6 +6,9 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from ._mde_onboardings_operations import MdeOnboardingsOperations +from ._custom_assessment_automations_operations import CustomAssessmentAutomationsOperations +from ._custom_entity_store_assignments_operations import CustomEntityStoreAssignmentsOperations from ._compliance_results_operations import ComplianceResultsOperations from ._pricings_operations import PricingsOperations from ._advanced_threat_protection_operations import AdvancedThreatProtectionOperations @@ -51,8 +54,12 @@ from ._settings_operations import SettingsOperations from ._ingestion_settings_operations import IngestionSettingsOperations from ._software_inventories_operations import SoftwareInventoriesOperations +from ._security_connectors_operations import SecurityConnectorsOperations __all__ = [ + 'MdeOnboardingsOperations', + 'CustomAssessmentAutomationsOperations', + 'CustomEntityStoreAssignmentsOperations', 'ComplianceResultsOperations', 'PricingsOperations', 'AdvancedThreatProtectionOperations', @@ -98,4 +105,5 @@ 'SettingsOperations', 'IngestionSettingsOperations', 'SoftwareInventoriesOperations', + 'SecurityConnectorsOperations', ] diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_application_controls_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_application_controls_operations.py index ff1fdae054806..ccc2f318893c2 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_application_controls_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_application_controls_operations.py @@ -5,22 +5,177 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + *, + include_path_recommendations: Optional[bool] = None, + summary: Optional[bool] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/applicationWhitelistings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if include_path_recommendations is not None: + query_parameters['includePathRecommendations'] = _SERIALIZER.query("include_path_recommendations", include_path_recommendations, 'bool') + if summary is not None: + query_parameters['summary'] = _SERIALIZER.query("summary", summary, 'bool') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + asc_location: str, + group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "groupName": _SERIALIZER.url("group_name", group_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_put_request( + subscription_id: str, + asc_location: str, + group_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "groupName": _SERIALIZER.url("group_name", group_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + asc_location: str, + group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "groupName": _SERIALIZER.url("group_name", group_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class AdaptiveApplicationControlsOperations(object): """AdaptiveApplicationControlsOperations operations. @@ -44,13 +199,13 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - include_path_recommendations=None, # type: Optional[bool] - summary=None, # type: Optional[bool] - **kwargs # type: Any - ): - # type: (...) -> "_models.AdaptiveApplicationControlGroups" + include_path_recommendations: Optional[bool] = None, + summary: Optional[bool] = None, + **kwargs: Any + ) -> "_models.AdaptiveApplicationControlGroups": """Gets a list of application control machine groups for the subscription. :param include_path_recommendations: Include the policy rules. @@ -67,29 +222,17 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if include_path_recommendations is not None: - query_parameters['includePathRecommendations'] = self._serialize.query("include_path_recommendations", include_path_recommendations, 'bool') - if summary is not None: - query_parameters['summary'] = self._serialize.query("summary", summary, 'bool') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + subscription_id=self._config.subscription_id, + include_path_recommendations=include_path_recommendations, + summary=summary, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -103,14 +246,16 @@ def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/applicationWhitelistings'} # type: ignore + + @distributed_trace def get( self, - group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.AdaptiveApplicationControlGroup" + group_name: str, + **kwargs: Any + ) -> "_models.AdaptiveApplicationControlGroup": """Gets an application control VM/server group. :param group_name: Name of an application control machine group. @@ -125,27 +270,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'groupName': self._serialize.url("group_name", group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + group_name=group_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -159,15 +294,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}'} # type: ignore + + @distributed_trace def put( self, - group_name, # type: str - body, # type: "_models.AdaptiveApplicationControlGroup" - **kwargs # type: Any - ): - # type: (...) -> "_models.AdaptiveApplicationControlGroup" + group_name: str, + body: "_models.AdaptiveApplicationControlGroup", + **kwargs: Any + ) -> "_models.AdaptiveApplicationControlGroup": """Update an application control machine group. :param group_name: Name of an application control machine group. @@ -184,32 +321,22 @@ def put( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.put.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'groupName': self._serialize.url("group_name", group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'AdaptiveApplicationControlGroup') + + request = build_put_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + group_name=group_name, + content_type=content_type, + json=_json, + template_url=self.put.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'AdaptiveApplicationControlGroup') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -223,14 +350,16 @@ def put( return cls(pipeline_response, deserialized, {}) return deserialized + put.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}'} # type: ignore + + @distributed_trace def delete( self, - group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + group_name: str, + **kwargs: Any + ) -> None: """Delete an application control machine group. :param group_name: Name of an application control machine group. @@ -245,27 +374,17 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'groupName': self._serialize.url("group_name", group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + group_name=group_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -277,3 +396,4 @@ def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/applicationWhitelistings/{groupName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_network_hardenings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_network_hardenings_operations.py index 9bb7da3b503fb..29e3d71ddf964 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_network_hardenings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_adaptive_network_hardenings_operations.py @@ -5,25 +5,160 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat from azure.mgmt.core.polling.arm_polling import ARMPolling +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_by_extended_resource_request( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + adaptive_network_hardening_resource_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + "adaptiveNetworkHardeningResourceName": _SERIALIZER.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_enforce_request_initial( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + adaptive_network_hardening_resource_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + adaptive_network_hardening_enforce_action = "enforce" + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}/{adaptiveNetworkHardeningEnforceAction}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + "adaptiveNetworkHardeningResourceName": _SERIALIZER.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), + "adaptiveNetworkHardeningEnforceAction": _SERIALIZER.url("adaptive_network_hardening_enforce_action", adaptive_network_hardening_enforce_action, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class AdaptiveNetworkHardeningsOperations(object): """AdaptiveNetworkHardeningsOperations operations. @@ -47,15 +182,15 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list_by_extended_resource( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AdaptiveNetworkHardeningsList"] + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any + ) -> Iterable["_models.AdaptiveNetworkHardeningsList"]: """Gets a list of Adaptive Network Hardenings resources in scope of an extended resource. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -68,7 +203,8 @@ def list_by_extended_resource( :param resource_name: Name of the resource. :type resource_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AdaptiveNetworkHardeningsList or the result of cls(response) + :return: An iterator like instance of either AdaptiveNetworkHardeningsList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.AdaptiveNetworkHardeningsList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -77,38 +213,37 @@ def list_by_extended_resource( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_extended_resource.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.list_by_extended_resource.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AdaptiveNetworkHardeningsList', pipeline_response) + deserialized = self._deserialize("AdaptiveNetworkHardeningsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -126,21 +261,22 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_extended_resource.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - adaptive_network_hardening_resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.AdaptiveNetworkHardening" + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + adaptive_network_hardening_resource_name: str, + **kwargs: Any + ) -> "_models.AdaptiveNetworkHardening": """Gets a single Adaptive Network Hardening resource. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -165,30 +301,20 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'adaptiveNetworkHardeningResourceName': self._serialize.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + adaptive_network_hardening_resource_name=adaptive_network_hardening_resource_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -202,55 +328,44 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}'} # type: ignore + def _enforce_initial( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - adaptive_network_hardening_resource_name, # type: str - body, # type: "_models.AdaptiveNetworkHardeningEnforceRequest" - **kwargs # type: Any - ): - # type: (...) -> None + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + adaptive_network_hardening_resource_name: str, + body: "_models.AdaptiveNetworkHardeningEnforceRequest", + **kwargs: Any + ) -> None: cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - adaptive_network_hardening_enforce_action = "enforce" - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._enforce_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'adaptiveNetworkHardeningResourceName': self._serialize.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), - 'adaptiveNetworkHardeningEnforceAction': self._serialize.url("adaptive_network_hardening_enforce_action", adaptive_network_hardening_enforce_action, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'AdaptiveNetworkHardeningEnforceRequest') + + request = build_enforce_request_initial( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + adaptive_network_hardening_resource_name=adaptive_network_hardening_resource_name, + content_type=content_type, + json=_json, + template_url=self._enforce_initial.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'AdaptiveNetworkHardeningEnforceRequest') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -263,17 +378,18 @@ def _enforce_initial( _enforce_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}/{adaptiveNetworkHardeningEnforceAction}'} # type: ignore + + @distributed_trace def begin_enforce( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - adaptive_network_hardening_resource_name, # type: str - body, # type: "_models.AdaptiveNetworkHardeningEnforceRequest" - **kwargs # type: Any - ): - # type: (...) -> LROPoller[None] + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + adaptive_network_hardening_resource_name: str, + body: "_models.AdaptiveNetworkHardeningEnforceRequest", + **kwargs: Any + ) -> LROPoller[None]: """Enforces the given rules on the NSG(s) listed in the request. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -292,15 +408,18 @@ def begin_enforce( :type body: ~azure.mgmt.security.models.AdaptiveNetworkHardeningEnforceRequest :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. - Pass in False for this operation to not poll, or pass in your own initialized polling object for a personal polling strategy. + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: + :raises: ~azure.core.exceptions.HttpResponseError """ - polling = kwargs.pop('polling', True) # type: Union[bool, PollingMethod] + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + polling = kwargs.pop('polling', True) # type: Union[bool, azure.core.polling.PollingMethod] cls = kwargs.pop('cls', None) # type: ClsType[None] lro_delay = kwargs.pop( 'polling_interval', @@ -315,29 +434,18 @@ def begin_enforce( resource_name=resource_name, adaptive_network_hardening_resource_name=adaptive_network_hardening_resource_name, body=body, + content_type=content_type, cls=lambda x,y,z: x, **kwargs ) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) def get_long_running_output(pipeline_response): if cls: return cls(pipeline_response, None, {}) - adaptive_network_hardening_enforce_action = "enforce" - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'adaptiveNetworkHardeningResourceName': self._serialize.url("adaptive_network_hardening_resource_name", adaptive_network_hardening_resource_name, 'str'), - 'adaptiveNetworkHardeningEnforceAction': self._serialize.url("adaptive_network_hardening_enforce_action", adaptive_network_hardening_enforce_action, 'str'), - } - if polling is True: polling_method = ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + if polling is True: polling_method = ARMPolling(lro_delay, **kwargs) elif polling is False: polling_method = NoPolling() else: polling_method = polling if cont_token: @@ -349,4 +457,5 @@ def get_long_running_output(pipeline_response): ) else: return LROPoller(self._client, raw_result, get_long_running_output, polling_method) + begin_enforce.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/adaptiveNetworkHardenings/{adaptiveNetworkHardeningResourceName}/{adaptiveNetworkHardeningEnforceAction}'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_advanced_threat_protection_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_advanced_threat_protection_operations.py index 359abd268523d..1cd899b6876b5 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_advanced_threat_protection_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_advanced_threat_protection_operations.py @@ -5,22 +5,100 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_get_request( + resource_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01" + setting_name = "current" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/advancedThreatProtectionSettings/{settingName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "settingName": _SERIALIZER.url("setting_name", setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_request( + resource_id: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2019-01-01" + setting_name = "current" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/advancedThreatProtectionSettings/{settingName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "settingName": _SERIALIZER.url("setting_name", setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class AdvancedThreatProtectionOperations(object): """AdvancedThreatProtectionOperations operations. @@ -44,12 +122,12 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def get( self, - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.AdvancedThreatProtectionSetting" + resource_id: str, + **kwargs: Any + ) -> "_models.AdvancedThreatProtectionSetting": """Gets the Advanced Threat Protection settings for the specified resource. :param resource_id: The identifier of the resource. @@ -64,27 +142,15 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01" - setting_name = "current" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_get_request( + resource_id=resource_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -98,21 +164,24 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/advancedThreatProtectionSettings/{settingName}'} # type: ignore + + @distributed_trace def create( self, - resource_id, # type: str - advanced_threat_protection_setting, # type: "_models.AdvancedThreatProtectionSetting" - **kwargs # type: Any - ): - # type: (...) -> "_models.AdvancedThreatProtectionSetting" + resource_id: str, + advanced_threat_protection_setting: "_models.AdvancedThreatProtectionSetting", + **kwargs: Any + ) -> "_models.AdvancedThreatProtectionSetting": """Creates or updates the Advanced Threat Protection settings on a specified resource. :param resource_id: The identifier of the resource. :type resource_id: str :param advanced_threat_protection_setting: Advanced Threat Protection Settings. - :type advanced_threat_protection_setting: ~azure.mgmt.security.models.AdvancedThreatProtectionSetting + :type advanced_threat_protection_setting: + ~azure.mgmt.security.models.AdvancedThreatProtectionSetting :keyword callable cls: A custom type or function that will be passed the direct response :return: AdvancedThreatProtectionSetting, or the result of cls(response) :rtype: ~azure.mgmt.security.models.AdvancedThreatProtectionSetting @@ -123,32 +192,20 @@ def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01" - setting_name = "current" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(advanced_threat_protection_setting, 'AdvancedThreatProtectionSetting') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_request( + resource_id=resource_id, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(advanced_threat_protection_setting, 'AdvancedThreatProtectionSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -162,4 +219,6 @@ def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/advancedThreatProtectionSettings/{settingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_operations.py index 531b783650816..2dcbefd71494d 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_operations.py @@ -5,25 +5,490 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat from azure.mgmt.core.polling.arm_polling import ARMPolling +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alerts') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/alerts') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_subscription_level_by_region_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_resource_group_level_by_region_request( + subscription_id: str, + asc_location: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_subscription_level_request( + subscription_id: str, + asc_location: str, + alert_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_resource_group_level_request( + subscription_id: str, + asc_location: str, + alert_name: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_subscription_level_state_to_dismiss_request( + subscription_id: str, + asc_location: str, + alert_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/dismiss') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_subscription_level_state_to_resolve_request( + subscription_id: str, + asc_location: str, + alert_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/resolve') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_subscription_level_state_to_activate_request( + subscription_id: str, + asc_location: str, + alert_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/activate') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_resource_group_level_state_to_resolve_request( + subscription_id: str, + asc_location: str, + alert_name: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/resolve') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_resource_group_level_state_to_dismiss_request( + subscription_id: str, + asc_location: str, + alert_name: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/dismiss') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_resource_group_level_state_to_activate_request( + subscription_id: str, + asc_location: str, + alert_name: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/activate') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "alertName": _SERIALIZER.url("alert_name", alert_name, 'str'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_simulate_request_initial( + subscription_id: str, + asc_location: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/default/simulate') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class AlertsOperations(object): """AlertsOperations operations. @@ -47,11 +512,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AlertList"] + **kwargs: Any + ) -> Iterable["_models.AlertList"]: """List all the alerts that are associated with the subscription. :keyword callable cls: A custom type or function that will be passed the direct response @@ -64,34 +529,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -109,17 +569,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alerts'} # type: ignore + @distributed_trace def list_by_resource_group( self, - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AlertList"] + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.AlertList"]: """List all the alerts that are associated with the resource group. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -135,35 +596,31 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -181,16 +638,17 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/alerts'} # type: ignore + @distributed_trace def list_subscription_level_by_region( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AlertList"] + **kwargs: Any + ) -> Iterable["_models.AlertList"]: """List all the alerts that are associated with the subscription that are stored in a specific location. @@ -204,35 +662,31 @@ def list_subscription_level_by_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_subscription_level_by_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_subscription_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_subscription_level_by_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_subscription_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -250,17 +704,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_subscription_level_by_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts'} # type: ignore + @distributed_trace def list_resource_group_level_by_region( self, - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AlertList"] + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.AlertList"]: """List all the alerts that are associated with the resource group that are stored in a specific location. @@ -277,36 +732,33 @@ def list_resource_group_level_by_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_resource_group_level_by_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_resource_group_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + resource_group_name=resource_group_name, + template_url=self.list_resource_group_level_by_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_resource_group_level_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AlertList', pipeline_response) + deserialized = self._deserialize("AlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -324,17 +776,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_resource_group_level_by_region.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts'} # type: ignore + @distributed_trace def get_subscription_level( self, - alert_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Alert" + alert_name: str, + **kwargs: Any + ) -> "_models.Alert": """Get an alert that is associated with a subscription. :param alert_name: Name of the alert object. @@ -349,27 +802,17 @@ def get_subscription_level( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.get_subscription_level.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_subscription_level_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.get_subscription_level.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -383,15 +826,17 @@ def get_subscription_level( return cls(pipeline_response, deserialized, {}) return deserialized + get_subscription_level.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}'} # type: ignore + + @distributed_trace def get_resource_group_level( self, - alert_name, # type: str - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Alert" + alert_name: str, + resource_group_name: str, + **kwargs: Any + ) -> "_models.Alert": """Get an alert that is associated a resource group or a resource in a resource group. :param alert_name: Name of the alert object. @@ -409,28 +854,18 @@ def get_resource_group_level( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.get_resource_group_level.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_resource_group_level_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.get_resource_group_level.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -444,14 +879,16 @@ def get_resource_group_level( return cls(pipeline_response, deserialized, {}) return deserialized + get_resource_group_level.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}'} # type: ignore + + @distributed_trace def update_subscription_level_state_to_dismiss( self, - alert_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + alert_name: str, + **kwargs: Any + ) -> None: """Update the alert's state. :param alert_name: Name of the alert object. @@ -466,27 +903,17 @@ def update_subscription_level_state_to_dismiss( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_state_to_dismiss.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_state_to_dismiss_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.update_subscription_level_state_to_dismiss.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -499,12 +926,13 @@ def update_subscription_level_state_to_dismiss( update_subscription_level_state_to_dismiss.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/dismiss'} # type: ignore + + @distributed_trace def update_subscription_level_state_to_resolve( self, - alert_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + alert_name: str, + **kwargs: Any + ) -> None: """Update the alert's state. :param alert_name: Name of the alert object. @@ -519,27 +947,17 @@ def update_subscription_level_state_to_resolve( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_state_to_resolve.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_state_to_resolve_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.update_subscription_level_state_to_resolve.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -552,12 +970,13 @@ def update_subscription_level_state_to_resolve( update_subscription_level_state_to_resolve.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/resolve'} # type: ignore + + @distributed_trace def update_subscription_level_state_to_activate( self, - alert_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + alert_name: str, + **kwargs: Any + ) -> None: """Update the alert's state. :param alert_name: Name of the alert object. @@ -572,27 +991,17 @@ def update_subscription_level_state_to_activate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_state_to_activate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_state_to_activate_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + template_url=self.update_subscription_level_state_to_activate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -605,13 +1014,14 @@ def update_subscription_level_state_to_activate( update_subscription_level_state_to_activate.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/activate'} # type: ignore + + @distributed_trace def update_resource_group_level_state_to_resolve( self, - alert_name, # type: str - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + alert_name: str, + resource_group_name: str, + **kwargs: Any + ) -> None: """Update the alert's state. :param alert_name: Name of the alert object. @@ -629,28 +1039,18 @@ def update_resource_group_level_state_to_resolve( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_state_to_resolve.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_state_to_resolve_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.update_resource_group_level_state_to_resolve.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -663,13 +1063,14 @@ def update_resource_group_level_state_to_resolve( update_resource_group_level_state_to_resolve.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/resolve'} # type: ignore + + @distributed_trace def update_resource_group_level_state_to_dismiss( self, - alert_name, # type: str - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + alert_name: str, + resource_group_name: str, + **kwargs: Any + ) -> None: """Update the alert's state. :param alert_name: Name of the alert object. @@ -687,28 +1088,18 @@ def update_resource_group_level_state_to_dismiss( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_state_to_dismiss.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_state_to_dismiss_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.update_resource_group_level_state_to_dismiss.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -721,13 +1112,14 @@ def update_resource_group_level_state_to_dismiss( update_resource_group_level_state_to_dismiss.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/dismiss'} # type: ignore + + @distributed_trace def update_resource_group_level_state_to_activate( self, - alert_name, # type: str - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + alert_name: str, + resource_group_name: str, + **kwargs: Any + ) -> None: """Update the alert's state. :param alert_name: Name of the alert object. @@ -745,28 +1137,18 @@ def update_resource_group_level_state_to_activate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_state_to_activate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'alertName': self._serialize.url("alert_name", alert_name, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_state_to_activate_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + alert_name=alert_name, + resource_group_name=resource_group_name, + template_url=self.update_resource_group_level_state_to_activate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -779,42 +1161,32 @@ def update_resource_group_level_state_to_activate( update_resource_group_level_state_to_activate.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/alerts/{alertName}/activate'} # type: ignore + def _simulate_initial( self, - alert_simulator_request_body, # type: "_models.AlertSimulatorRequestBody" - **kwargs # type: Any - ): - # type: (...) -> None + alert_simulator_request_body: "_models.AlertSimulatorRequestBody", + **kwargs: Any + ) -> None: cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._simulate_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(alert_simulator_request_body, 'AlertSimulatorRequestBody') + + request = build_simulate_request_initial( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + content_type=content_type, + json=_json, + template_url=self._simulate_initial.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(alert_simulator_request_body, 'AlertSimulatorRequestBody') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -827,27 +1199,31 @@ def _simulate_initial( _simulate_initial.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/default/simulate'} # type: ignore + + @distributed_trace def begin_simulate( self, - alert_simulator_request_body, # type: "_models.AlertSimulatorRequestBody" - **kwargs # type: Any - ): - # type: (...) -> LROPoller[None] + alert_simulator_request_body: "_models.AlertSimulatorRequestBody", + **kwargs: Any + ) -> LROPoller[None]: """Simulate security alerts. :param alert_simulator_request_body: Alert Simulator Request Properties. :type alert_simulator_request_body: ~azure.mgmt.security.models.AlertSimulatorRequestBody :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. - Pass in False for this operation to not poll, or pass in your own initialized polling object for a personal polling strategy. + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: + :raises: ~azure.core.exceptions.HttpResponseError """ - polling = kwargs.pop('polling', True) # type: Union[bool, PollingMethod] + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + polling = kwargs.pop('polling', True) # type: Union[bool, azure.core.polling.PollingMethod] cls = kwargs.pop('cls', None) # type: ClsType[None] lro_delay = kwargs.pop( 'polling_interval', @@ -857,23 +1233,18 @@ def begin_simulate( if cont_token is None: raw_result = self._simulate_initial( alert_simulator_request_body=alert_simulator_request_body, + content_type=content_type, cls=lambda x,y,z: x, **kwargs ) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) def get_long_running_output(pipeline_response): if cls: return cls(pipeline_response, None, {}) - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'original-uri'}, path_format_arguments=path_format_arguments, **kwargs) + if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'original-uri'}, **kwargs) elif polling is False: polling_method = NoPolling() else: polling_method = polling if cont_token: @@ -885,4 +1256,5 @@ def get_long_running_output(pipeline_response): ) else: return LROPoller(self._client, raw_result, get_long_running_output, polling_method) + begin_simulate.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/alerts/default/simulate'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_suppression_rules_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_suppression_rules_operations.py index 21a4be9ec1028..f4860e099ea83 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_suppression_rules_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_alerts_suppression_rules_operations.py @@ -5,23 +5,169 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + *, + alert_type: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if alert_type is not None: + query_parameters['AlertType'] = _SERIALIZER.query("alert_type", alert_type, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + alerts_suppression_rule_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "alertsSuppressionRuleName": _SERIALIZER.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_request( + subscription_id: str, + alerts_suppression_rule_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "alertsSuppressionRuleName": _SERIALIZER.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + alerts_suppression_rule_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "alertsSuppressionRuleName": _SERIALIZER.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class AlertsSuppressionRulesOperations(object): """AlertsSuppressionRulesOperations operations. @@ -45,18 +191,19 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - alert_type=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AlertsSuppressionRulesList"] + alert_type: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.AlertsSuppressionRulesList"]: """List of all the dismiss rules for the given subscription. :param alert_type: Type of the alert to get rules for. :type alert_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AlertsSuppressionRulesList or the result of cls(response) + :return: An iterator like instance of either AlertsSuppressionRulesList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.AlertsSuppressionRulesList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -65,36 +212,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if alert_type is not None: - query_parameters['AlertType'] = self._serialize.query("alert_type", alert_type, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + alert_type=alert_type, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + alert_type=alert_type, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AlertsSuppressionRulesList', pipeline_response) + deserialized = self._deserialize("AlertsSuppressionRulesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,17 +254,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules'} # type: ignore + @distributed_trace def get( self, - alerts_suppression_rule_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.AlertsSuppressionRule" + alerts_suppression_rule_name: str, + **kwargs: Any + ) -> "_models.AlertsSuppressionRule": """Get dismiss rule, with name: {alertsSuppressionRuleName}, for the given subscription. :param alerts_suppression_rule_name: The unique name of the suppression alert rule. @@ -137,26 +280,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'alertsSuppressionRuleName': self._serialize.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + alerts_suppression_rule_name=alerts_suppression_rule_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -170,15 +303,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}'} # type: ignore + + @distributed_trace def update( self, - alerts_suppression_rule_name, # type: str - alerts_suppression_rule, # type: "_models.AlertsSuppressionRule" - **kwargs # type: Any - ): - # type: (...) -> "_models.AlertsSuppressionRule" + alerts_suppression_rule_name: str, + alerts_suppression_rule: "_models.AlertsSuppressionRule", + **kwargs: Any + ) -> "_models.AlertsSuppressionRule": """Update existing rule or create new rule if it doesn't exist. :param alerts_suppression_rule_name: The unique name of the suppression alert rule. @@ -195,31 +330,21 @@ def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'alertsSuppressionRuleName': self._serialize.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(alerts_suppression_rule, 'AlertsSuppressionRule') + + request = build_update_request( + subscription_id=self._config.subscription_id, + alerts_suppression_rule_name=alerts_suppression_rule_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(alerts_suppression_rule, 'AlertsSuppressionRule') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -233,14 +358,16 @@ def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}'} # type: ignore + + @distributed_trace def delete( self, - alerts_suppression_rule_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + alerts_suppression_rule_name: str, + **kwargs: Any + ) -> None: """Delete dismiss alert rule for this subscription. :param alerts_suppression_rule_name: The unique name of the suppression alert rule. @@ -255,26 +382,16 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'alertsSuppressionRuleName': self._serialize.url("alerts_suppression_rule_name", alerts_suppression_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + alerts_suppression_rule_name=alerts_suppression_rule_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -286,3 +403,4 @@ def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/alertsSuppressionRules/{alertsSuppressionRuleName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_allowed_connections_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_allowed_connections_operations.py index 5efee309b1895..dd05a7a33992c 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_allowed_connections_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_allowed_connections_operations.py @@ -5,23 +5,126 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/allowedConnections') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_home_region_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/allowedConnections') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + connection_type: Union[str, "_models.ConnectionType"], + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/allowedConnections/{connectionType}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "connectionType": _SERIALIZER.url("connection_type", connection_type, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class AllowedConnectionsOperations(object): """AllowedConnectionsOperations operations. @@ -45,15 +148,16 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AllowedConnectionsList"] + **kwargs: Any + ) -> Iterable["_models.AllowedConnectionsList"]: """Gets the list of all possible traffic between resources for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AllowedConnectionsList or the result of cls(response) + :return: An iterator like instance of either AllowedConnectionsList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.AllowedConnectionsList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -62,34 +166,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AllowedConnectionsList', pipeline_response) + deserialized = self._deserialize("AllowedConnectionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,20 +206,22 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/allowedConnections'} # type: ignore + @distributed_trace def list_by_home_region( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AllowedConnectionsList"] + **kwargs: Any + ) -> Iterable["_models.AllowedConnectionsList"]: """Gets the list of all possible traffic between resources for the subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AllowedConnectionsList or the result of cls(response) + :return: An iterator like instance of either AllowedConnectionsList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.AllowedConnectionsList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -129,35 +230,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AllowedConnectionsList', pipeline_response) + deserialized = self._deserialize("AllowedConnectionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -175,18 +272,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/allowedConnections'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - connection_type, # type: Union[str, "_models.ConnectionType"] - **kwargs # type: Any - ): - # type: (...) -> "_models.AllowedConnectionsResource" + resource_group_name: str, + connection_type: Union[str, "_models.ConnectionType"], + **kwargs: Any + ) -> "_models.AllowedConnectionsResource": """Gets the list of all possible traffic between resources for the subscription and location, based on connection type. @@ -205,28 +303,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'connectionType': self._serialize.url("connection_type", connection_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + connection_type=connection_type, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -240,4 +328,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/allowedConnections/{connectionType}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_metadata_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_metadata_operations.py index df24fe791662d..588a609dcaea2 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_metadata_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_metadata_operations.py @@ -5,23 +5,221 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/providers/Microsoft.Security/assessmentMetadata') + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + assessment_metadata_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}') + path_format_arguments = { + "assessmentMetadataName": _SERIALIZER.url("assessment_metadata_name", assessment_metadata_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_subscription_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_in_subscription_request( + assessment_metadata_name: str, + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}') + path_format_arguments = { + "assessmentMetadataName": _SERIALIZER.url("assessment_metadata_name", assessment_metadata_name, 'str'), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_in_subscription_request( + assessment_metadata_name: str, + subscription_id: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}') + path_format_arguments = { + "assessmentMetadataName": _SERIALIZER.url("assessment_metadata_name", assessment_metadata_name, 'str'), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_in_subscription_request( + assessment_metadata_name: str, + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}') + path_format_arguments = { + "assessmentMetadataName": _SERIALIZER.url("assessment_metadata_name", assessment_metadata_name, 'str'), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class AssessmentsMetadataOperations(object): """AssessmentsMetadataOperations operations. @@ -45,47 +243,46 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecurityAssessmentMetadataList"] + **kwargs: Any + ) -> Iterable["_models.SecurityAssessmentMetadataResponseList"]: """Get metadata information on all assessment types. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecurityAssessmentMetadataList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataList] + :return: An iterator like instance of either SecurityAssessmentMetadataResponseList or the + result of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataResponseList] :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataList"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponseList"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityAssessmentMetadataList', pipeline_response) + deserialized = self._deserialize("SecurityAssessmentMetadataResponseList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -103,50 +300,41 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/providers/Microsoft.Security/assessmentMetadata'} # type: ignore + @distributed_trace def get( self, - assessment_metadata_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityAssessmentMetadata" + assessment_metadata_name: str, + **kwargs: Any + ) -> "_models.SecurityAssessmentMetadataResponse": """Get metadata information on an assessment type. :param assessment_metadata_name: The Assessment Key - Unique key for the assessment type. :type assessment_metadata_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessmentMetadata, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :return: SecurityAssessmentMetadataResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadata"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + assessment_metadata_name=assessment_metadata_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -154,59 +342,58 @@ def get( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessmentMetadata', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentMetadataResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + + @distributed_trace def list_by_subscription( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecurityAssessmentMetadataList"] + **kwargs: Any + ) -> Iterable["_models.SecurityAssessmentMetadataResponseList"]: """Get metadata information on all assessment types in a specific subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecurityAssessmentMetadataList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataList] + :return: An iterator like instance of either SecurityAssessmentMetadataResponseList or the + result of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecurityAssessmentMetadataResponseList] :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataList"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponseList"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityAssessmentMetadataList', pipeline_response) + deserialized = self._deserialize("SecurityAssessmentMetadataResponseList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -224,51 +411,42 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata'} # type: ignore + @distributed_trace def get_in_subscription( self, - assessment_metadata_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityAssessmentMetadata" + assessment_metadata_name: str, + **kwargs: Any + ) -> "_models.SecurityAssessmentMetadataResponse": """Get metadata information on an assessment type in a specific subscription. :param assessment_metadata_name: The Assessment Key - Unique key for the assessment type. :type assessment_metadata_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessmentMetadata, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :return: SecurityAssessmentMetadataResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadata"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get_in_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_in_subscription_request( + assessment_metadata_name=assessment_metadata_name, + subscription_id=self._config.subscription_id, + template_url=self.get_in_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -276,62 +454,54 @@ def get_in_subscription( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessmentMetadata', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentMetadataResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + get_in_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + + @distributed_trace def create_in_subscription( self, - assessment_metadata_name, # type: str - assessment_metadata, # type: "_models.SecurityAssessmentMetadata" - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityAssessmentMetadata" + assessment_metadata_name: str, + assessment_metadata: "_models.SecurityAssessmentMetadataResponse", + **kwargs: Any + ) -> "_models.SecurityAssessmentMetadataResponse": """Create metadata information on an assessment type in a specific subscription. :param assessment_metadata_name: The Assessment Key - Unique key for the assessment type. :type assessment_metadata_name: str :param assessment_metadata: AssessmentMetadata object. - :type assessment_metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :type assessment_metadata: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessmentMetadata, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadata + :return: SecurityAssessmentMetadataResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentMetadataResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadata"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentMetadataResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_in_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(assessment_metadata, 'SecurityAssessmentMetadataResponse') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_in_subscription_request( + assessment_metadata_name=assessment_metadata_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + json=_json, + template_url=self.create_in_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(assessment_metadata, 'SecurityAssessmentMetadata') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -339,20 +509,22 @@ def create_in_subscription( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessmentMetadata', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentMetadataResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + create_in_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + + @distributed_trace def delete_in_subscription( self, - assessment_metadata_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + assessment_metadata_name: str, + **kwargs: Any + ) -> None: """Delete metadata information on an assessment type in a specific subscription, will cause the deletion of all the assessments of that type in that subscription. @@ -368,26 +540,16 @@ def delete_in_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete_in_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'assessmentMetadataName': self._serialize.url("assessment_metadata_name", assessment_metadata_name, 'str'), - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_in_subscription_request( + assessment_metadata_name=assessment_metadata_name, + subscription_id=self._config.subscription_id, + template_url=self.delete_in_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -399,3 +561,4 @@ def delete_in_subscription( return cls(pipeline_response, None, {}) delete_in_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/assessmentMetadata/{assessmentMetadataName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_operations.py index 0b7b28c2f6a3a..20598a0df6253 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_assessments_operations.py @@ -5,23 +5,169 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + scope: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/assessments') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + resource_id: str, + assessment_name: str, + *, + expand: Optional[Union[str, "_models.ExpandEnum"]] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "assessmentName": _SERIALIZER.url("assessment_name", assessment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if expand is not None: + query_parameters['$expand'] = _SERIALIZER.query("expand", expand, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + resource_id: str, + assessment_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "assessmentName": _SERIALIZER.url("assessment_name", assessment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + resource_id: str, + assessment_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "assessmentName": _SERIALIZER.url("assessment_name", assessment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class AssessmentsOperations(object): """AssessmentsOperations operations. @@ -45,12 +191,12 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - scope, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecurityAssessmentList"] + scope: str, + **kwargs: Any + ) -> Iterable["_models.SecurityAssessmentList"]: """Get security assessments on all your scanned resources inside a scope. :param scope: Scope of the query, can be subscription @@ -58,7 +204,8 @@ def list( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecurityAssessmentList or the result of cls(response) + :return: An iterator like instance of either SecurityAssessmentList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecurityAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -67,34 +214,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityAssessmentList', pipeline_response) + deserialized = self._deserialize("SecurityAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,19 +254,20 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/assessments'} # type: ignore + @distributed_trace def get( self, - resource_id, # type: str - assessment_name, # type: str - expand=None, # type: Optional[Union[str, "_models.ExpandEnum"]] - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityAssessment" + resource_id: str, + assessment_name: str, + expand: Optional[Union[str, "_models.ExpandEnum"]] = None, + **kwargs: Any + ) -> "_models.SecurityAssessmentResponse": """Get a security assessment on your scanned resource. :param resource_id: The identifier of the resource. @@ -134,37 +277,26 @@ def get( :param expand: OData expand. Optional. :type expand: str or ~azure.mgmt.security.models.ExpandEnum :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessment, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessment + :return: SecurityAssessmentResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessment"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if expand is not None: - query_parameters['$expand'] = self._serialize.query("expand", expand, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + resource_id=resource_id, + assessment_name=assessment_name, + expand=expand, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -172,22 +304,24 @@ def get( map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('SecurityAssessment', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}'} # type: ignore + + @distributed_trace def create_or_update( self, - resource_id, # type: str - assessment_name, # type: str - assessment, # type: "_models.SecurityAssessment" - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityAssessment" + resource_id: str, + assessment_name: str, + assessment: "_models.SecurityAssessment", + **kwargs: Any + ) -> "_models.SecurityAssessmentResponse": """Create a security assessment on your resource. An assessment metadata that describes this assessment must be predefined with the same name before inserting the assessment result. @@ -198,40 +332,30 @@ def create_or_update( :param assessment: Calculated assessment on a pre-defined assessment metadata. :type assessment: ~azure.mgmt.security.models.SecurityAssessment :keyword callable cls: A custom type or function that will be passed the direct response - :return: SecurityAssessment, or the result of cls(response) - :rtype: ~azure.mgmt.security.models.SecurityAssessment + :return: SecurityAssessmentResponse, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityAssessmentResponse :raises: ~azure.core.exceptions.HttpResponseError """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessment"] + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityAssessmentResponse"] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(assessment, 'SecurityAssessment') + + request = build_create_or_update_request( + resource_id=resource_id, + assessment_name=assessment_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(assessment, 'SecurityAssessment') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -240,24 +364,26 @@ def create_or_update( raise HttpResponseError(response=response, error_format=ARMErrorFormat) if response.status_code == 200: - deserialized = self._deserialize('SecurityAssessment', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentResponse', pipeline_response) if response.status_code == 201: - deserialized = self._deserialize('SecurityAssessment', pipeline_response) + deserialized = self._deserialize('SecurityAssessmentResponse', pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}'} # type: ignore + + @distributed_trace def delete( self, - resource_id, # type: str - assessment_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + resource_id: str, + assessment_name: str, + **kwargs: Any + ) -> None: """Delete a security assessment on your resource. An assessment metadata that describes this assessment must be predefined with the same name before inserting the assessment result. @@ -275,26 +401,16 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + resource_id=resource_id, + assessment_name=assessment_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -306,3 +422,4 @@ def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/assessments/{assessmentName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_auto_provisioning_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_auto_provisioning_settings_operations.py index 415b662697abe..478f63677d0a3 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_auto_provisioning_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_auto_provisioning_settings_operations.py @@ -5,23 +5,132 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + setting_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings/{settingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "settingName": _SERIALIZER.url("setting_name", setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_request( + subscription_id: str, + setting_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings/{settingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "settingName": _SERIALIZER.url("setting_name", setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class AutoProvisioningSettingsOperations(object): """AutoProvisioningSettingsOperations operations. @@ -45,15 +154,16 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AutoProvisioningSettingList"] + **kwargs: Any + ) -> Iterable["_models.AutoProvisioningSettingList"]: """Exposes the auto provisioning settings of the subscriptions. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either AutoProvisioningSettingList or the result of cls(response) + :return: An iterator like instance of either AutoProvisioningSettingList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.AutoProvisioningSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -62,34 +172,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AutoProvisioningSettingList', pipeline_response) + deserialized = self._deserialize("AutoProvisioningSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,17 +212,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings'} # type: ignore + @distributed_trace def get( self, - setting_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.AutoProvisioningSetting" + setting_name: str, + **kwargs: Any + ) -> "_models.AutoProvisioningSetting": """Details of a specific setting. :param setting_name: Auto provisioning setting key. @@ -132,26 +238,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -165,15 +261,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings/{settingName}'} # type: ignore + + @distributed_trace def create( self, - setting_name, # type: str - setting, # type: "_models.AutoProvisioningSetting" - **kwargs # type: Any - ): - # type: (...) -> "_models.AutoProvisioningSetting" + setting_name: str, + setting: "_models.AutoProvisioningSetting", + **kwargs: Any + ) -> "_models.AutoProvisioningSetting": """Details of a specific setting. :param setting_name: Auto provisioning setting key. @@ -190,31 +288,21 @@ def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(setting, 'AutoProvisioningSetting') + + request = build_create_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(setting, 'AutoProvisioningSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -228,4 +316,6 @@ def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/autoProvisioningSettings/{settingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_automations_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_automations_operations.py index 97727d493b0e2..205e8854ffc3b 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_automations_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_automations_operations.py @@ -5,23 +5,248 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/automations') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + automation_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "automationName": _SERIALIZER.url("automation_name", automation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + subscription_id: str, + resource_group_name: str, + automation_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "automationName": _SERIALIZER.url("automation_name", automation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + resource_group_name: str, + automation_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "automationName": _SERIALIZER.url("automation_name", automation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_validate_request( + subscription_id: str, + resource_group_name: str, + automation_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}/validate') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "automationName": _SERIALIZER.url("automation_name", automation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class AutomationsOperations(object): """AutomationsOperations operations. @@ -45,11 +270,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AutomationList"] + **kwargs: Any + ) -> Iterable["_models.AutomationList"]: """Lists all the security automations in the specified subscription. Use the 'nextLink' property in the response to get the next page of security automations for the specified subscription. @@ -63,34 +288,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AutomationList', pipeline_response) + deserialized = self._deserialize("AutomationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -108,17 +328,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/automations'} # type: ignore + @distributed_trace def list_by_resource_group( self, - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AutomationList"] + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.AutomationList"]: """Lists all the security automations in the specified resource group. Use the 'nextLink' property in the response to get the next page of security automations for the specified resource group. @@ -135,35 +356,31 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AutomationList', pipeline_response) + deserialized = self._deserialize("AutomationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -181,18 +398,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - automation_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Automation" + resource_group_name: str, + automation_name: str, + **kwargs: Any + ) -> "_models.Automation": """Retrieves information about the model of a security automation. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -210,27 +428,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -244,16 +452,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}'} # type: ignore + + @distributed_trace def create_or_update( self, - resource_group_name, # type: str - automation_name, # type: str - automation, # type: "_models.Automation" - **kwargs # type: Any - ): - # type: (...) -> "_models.Automation" + resource_group_name: str, + automation_name: str, + automation: "_models.Automation", + **kwargs: Any + ) -> "_models.Automation": """Creates or updates a security automation. If a security automation is already created and a subsequent request is issued for the same automation id, then it will be updated. @@ -274,32 +484,22 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(automation, 'Automation') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(automation, 'Automation') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -317,15 +517,17 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}'} # type: ignore + + @distributed_trace def delete( self, - resource_group_name, # type: str - automation_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + resource_group_name: str, + automation_name: str, + **kwargs: Any + ) -> None: """Deletes a security automation. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -343,27 +545,17 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -376,14 +568,15 @@ def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}'} # type: ignore + + @distributed_trace def validate( self, - resource_group_name, # type: str - automation_name, # type: str - automation, # type: "_models.Automation" - **kwargs # type: Any - ): - # type: (...) -> "_models.AutomationValidationStatus" + resource_group_name: str, + automation_name: str, + automation: "_models.Automation", + **kwargs: Any + ) -> "_models.AutomationValidationStatus": """Validates the security automation model before create or update. Any validation errors are returned to the client. @@ -404,32 +597,22 @@ def validate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.validate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'automationName': self._serialize.url("automation_name", automation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(automation, 'Automation') + + request = build_validate_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + automation_name=automation_name, + content_type=content_type, + json=_json, + template_url=self.validate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(automation, 'Automation') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -443,4 +626,6 @@ def validate( return cls(pipeline_response, deserialized, {}) return deserialized + validate.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/automations/{automationName}/validate'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliance_results_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliance_results_operations.py index 19c0725e83ecf..7d8cd644ab459 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliance_results_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliance_results_operations.py @@ -5,23 +5,89 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + scope: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/complianceResults') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + resource_id: str, + compliance_result_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/complianceResults/{complianceResultName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "complianceResultName": _SERIALIZER.url("compliance_result_name", compliance_result_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class ComplianceResultsOperations(object): """ComplianceResultsOperations operations. @@ -45,12 +111,12 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - scope, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.ComplianceResultList"] + scope: str, + **kwargs: Any + ) -> Iterable["_models.ComplianceResultList"]: """Security compliance results in the subscription. :param scope: Scope of the query, can be subscription @@ -58,7 +124,8 @@ def list( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ComplianceResultList or the result of cls(response) + :return: An iterator like instance of either ComplianceResultList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.ComplianceResultList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -67,34 +134,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('ComplianceResultList', pipeline_response) + deserialized = self._deserialize("ComplianceResultList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,18 +174,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/complianceResults'} # type: ignore + @distributed_trace def get( self, - resource_id, # type: str - compliance_result_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ComplianceResult" + resource_id: str, + compliance_result_name: str, + **kwargs: Any + ) -> "_models.ComplianceResult": """Security Compliance Result. :param resource_id: The identifier of the resource. @@ -140,26 +203,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'complianceResultName': self._serialize.url("compliance_result_name", compliance_result_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + resource_id=resource_id, + compliance_result_name=compliance_result_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -173,4 +226,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/complianceResults/{complianceResultName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliances_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliances_operations.py index c8b529670f496..d2cc049899127 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliances_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_compliances_operations.py @@ -5,23 +5,89 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + scope: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/compliances') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + scope: str, + compliance_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/compliances/{complianceName}') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + "complianceName": _SERIALIZER.url("compliance_name", compliance_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class CompliancesOperations(object): """CompliancesOperations operations. @@ -45,12 +111,12 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - scope, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.ComplianceList"] + scope: str, + **kwargs: Any + ) -> Iterable["_models.ComplianceList"]: """The Compliance scores of the specific management group. :param scope: Scope of the query, can be subscription @@ -67,34 +133,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('ComplianceList', pipeline_response) + deserialized = self._deserialize("ComplianceList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,18 +173,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/compliances'} # type: ignore + @distributed_trace def get( self, - scope, # type: str - compliance_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Compliance" + scope: str, + compliance_name: str, + **kwargs: Any + ) -> "_models.Compliance": """Details of a specific Compliance. :param scope: Scope of the query, can be subscription @@ -142,26 +204,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'complianceName': self._serialize.url("compliance_name", compliance_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scope=scope, + compliance_name=compliance_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -175,4 +227,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{scope}/providers/Microsoft.Security/compliances/{complianceName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_connectors_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_connectors_operations.py index 53148912e51fd..f1d85077001e7 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_connectors_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_connectors_operations.py @@ -5,23 +5,165 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + connector_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "connectorName": _SERIALIZER.url("connector_name", connector_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + subscription_id: str, + connector_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2020-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "connectorName": _SERIALIZER.url("connector_name", connector_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + connector_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "connectorName": _SERIALIZER.url("connector_name", connector_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class ConnectorsOperations(object): """ConnectorsOperations operations. @@ -45,15 +187,16 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.ConnectorSettingList"] + **kwargs: Any + ) -> Iterable["_models.ConnectorSettingList"]: """Cloud accounts connectors of a subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ConnectorSettingList or the result of cls(response) + :return: An iterator like instance of either ConnectorSettingList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.ConnectorSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -62,34 +205,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('ConnectorSettingList', pipeline_response) + deserialized = self._deserialize("ConnectorSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,17 +245,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors'} # type: ignore + @distributed_trace def get( self, - connector_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ConnectorSetting" + connector_name: str, + **kwargs: Any + ) -> "_models.ConnectorSetting": """Details of a specific cloud account connector. :param connector_name: Name of the cloud account connector. @@ -132,26 +271,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'connectorName': self._serialize.url("connector_name", connector_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + connector_name=connector_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -165,15 +294,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}'} # type: ignore + + @distributed_trace def create_or_update( self, - connector_name, # type: str - connector_setting, # type: "_models.ConnectorSetting" - **kwargs # type: Any - ): - # type: (...) -> "_models.ConnectorSetting" + connector_name: str, + connector_setting: "_models.ConnectorSetting", + **kwargs: Any + ) -> "_models.ConnectorSetting": """Create a cloud account connector or update an existing one. Connect to your cloud account. For AWS, use either account credentials or role-based authentication. For GCP, use account organization credentials. @@ -192,31 +323,21 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'connectorName': self._serialize.url("connector_name", connector_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(connector_setting, 'ConnectorSetting') + + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + connector_name=connector_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(connector_setting, 'ConnectorSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -230,14 +351,16 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}'} # type: ignore + + @distributed_trace def delete( self, - connector_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + connector_name: str, + **kwargs: Any + ) -> None: """Delete a cloud account connector from a subscription. :param connector_name: Name of the cloud account connector. @@ -252,26 +375,16 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'connectorName': self._serialize.url("connector_name", connector_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + connector_name=connector_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -283,3 +396,4 @@ def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/connectors/{connectorName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_assessment_automations_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_assessment_automations_operations.py new file mode 100644 index 0000000000000..47c434f47478a --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_assessment_automations_operations.py @@ -0,0 +1,540 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union +import warnings + +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.paging import ItemPaged +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer + +from .. import models as _models +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_get_request( + subscription_id: str, + resource_group_name: str, + custom_assessment_automation_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "customAssessmentAutomationName": _SERIALIZER.url("custom_assessment_automation_name", custom_assessment_automation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_request( + subscription_id: str, + resource_group_name: str, + custom_assessment_automation_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "customAssessmentAutomationName": _SERIALIZER.url("custom_assessment_automation_name", custom_assessment_automation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + resource_group_name: str, + custom_assessment_automation_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "customAssessmentAutomationName": _SERIALIZER.url("custom_assessment_automation_name", custom_assessment_automation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_subscription_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/customAssessmentAutomations') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + +class CustomAssessmentAutomationsOperations(object): + """CustomAssessmentAutomationsOperations operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer): + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace + def get( + self, + resource_group_name: str, + custom_assessment_automation_name: str, + **kwargs: Any + ) -> "_models.CustomAssessmentAutomation": + """Gets a custom assessment automation. + + Gets a single custom assessment automation by name for the provided subscription and resource + group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_assessment_automation_name: Name of the Custom Assessment Automation. + :type custom_assessment_automation_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomAssessmentAutomation, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomAssessmentAutomation + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomation"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_assessment_automation_name=custom_assessment_automation_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('CustomAssessmentAutomation', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}'} # type: ignore + + + @distributed_trace + def create( + self, + resource_group_name: str, + custom_assessment_automation_name: str, + custom_assessment_automation_body: "_models.CustomAssessmentAutomationRequest", + **kwargs: Any + ) -> "_models.CustomAssessmentAutomation": + """Creates a custom assessment automation. + + Creates or updates a custom assessment automation for the provided subscription. Please note + that providing an existing custom assessment automation will replace the existing record. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_assessment_automation_name: Name of the Custom Assessment Automation. + :type custom_assessment_automation_name: str + :param custom_assessment_automation_body: Custom Assessment Automation body. + :type custom_assessment_automation_body: + ~azure.mgmt.security.models.CustomAssessmentAutomationRequest + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomAssessmentAutomation, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomAssessmentAutomation + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomation"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(custom_assessment_automation_body, 'CustomAssessmentAutomationRequest') + + request = build_create_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_assessment_automation_name=custom_assessment_automation_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if response.status_code == 200: + deserialized = self._deserialize('CustomAssessmentAutomation', pipeline_response) + + if response.status_code == 201: + deserialized = self._deserialize('CustomAssessmentAutomation', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + create.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}'} # type: ignore + + + @distributed_trace + def delete( + self, + resource_group_name: str, + custom_assessment_automation_name: str, + **kwargs: Any + ) -> None: + """Deletes a custom assessment automation. + + Deletes a custom assessment automation by name for a provided subscription. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_assessment_automation_name: Name of the Custom Assessment Automation. + :type custom_assessment_automation_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: None, or the result of cls(response) + :rtype: None + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType[None] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_assessment_automation_name=custom_assessment_automation_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) + + delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations/{customAssessmentAutomationName}'} # type: ignore + + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.CustomAssessmentAutomationsListResult"]: + """List custom assessment automations in a subscription and a resource group. + + List custom assessment automations by provided subscription and resource group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomAssessmentAutomationsListResult or the + result of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.CustomAssessmentAutomationsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomationsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + def extract_data(pipeline_response): + deserialized = self._deserialize("CustomAssessmentAutomationsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, iter(list_of_elem) + + def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return ItemPaged( + get_next, extract_data + ) + list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customAssessmentAutomations'} # type: ignore + + @distributed_trace + def list_by_subscription( + self, + **kwargs: Any + ) -> Iterable["_models.CustomAssessmentAutomationsListResult"]: + """List custom assessment automations in a subscription. + + List custom assessment automations by provided subscription. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomAssessmentAutomationsListResult or the + result of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.CustomAssessmentAutomationsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomAssessmentAutomationsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + def extract_data(pipeline_response): + deserialized = self._deserialize("CustomAssessmentAutomationsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, iter(list_of_elem) + + def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return ItemPaged( + get_next, extract_data + ) + list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/customAssessmentAutomations'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_entity_store_assignments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_entity_store_assignments_operations.py new file mode 100644 index 0000000000000..fed46f64b4a46 --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_custom_entity_store_assignments_operations.py @@ -0,0 +1,542 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union +import warnings + +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.paging import ItemPaged +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer + +from .. import models as _models +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_get_request( + subscription_id: str, + resource_group_name: str, + custom_entity_store_assignment_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "customEntityStoreAssignmentName": _SERIALIZER.url("custom_entity_store_assignment_name", custom_entity_store_assignment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_request( + subscription_id: str, + resource_group_name: str, + custom_entity_store_assignment_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "customEntityStoreAssignmentName": _SERIALIZER.url("custom_entity_store_assignment_name", custom_entity_store_assignment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + resource_group_name: str, + custom_entity_store_assignment_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "customEntityStoreAssignmentName": _SERIALIZER.url("custom_entity_store_assignment_name", custom_entity_store_assignment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_subscription_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/customEntityStoreAssignments') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + +class CustomEntityStoreAssignmentsOperations(object): + """CustomEntityStoreAssignmentsOperations operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer): + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace + def get( + self, + resource_group_name: str, + custom_entity_store_assignment_name: str, + **kwargs: Any + ) -> "_models.CustomEntityStoreAssignment": + """Gets a custom entity store assignment. + + Gets a single custom entity store assignment by name for the provided subscription and resource + group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_entity_store_assignment_name: Name of the custom entity store assignment. + Generated name is GUID. + :type custom_entity_store_assignment_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomEntityStoreAssignment, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomEntityStoreAssignment + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignment"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_entity_store_assignment_name=custom_entity_store_assignment_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('CustomEntityStoreAssignment', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}'} # type: ignore + + + @distributed_trace + def create( + self, + resource_group_name: str, + custom_entity_store_assignment_name: str, + custom_entity_store_assignment_request_body: "_models.CustomEntityStoreAssignmentRequest", + **kwargs: Any + ) -> "_models.CustomEntityStoreAssignment": + """Creates a custom entity store assignment. + + Creates a custom entity store assignment for the provided subscription, if not already exists. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_entity_store_assignment_name: Name of the custom entity store assignment. + Generated name is GUID. + :type custom_entity_store_assignment_name: str + :param custom_entity_store_assignment_request_body: Custom entity store assignment body. + :type custom_entity_store_assignment_request_body: + ~azure.mgmt.security.models.CustomEntityStoreAssignmentRequest + :keyword callable cls: A custom type or function that will be passed the direct response + :return: CustomEntityStoreAssignment, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.CustomEntityStoreAssignment + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignment"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(custom_entity_store_assignment_request_body, 'CustomEntityStoreAssignmentRequest') + + request = build_create_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_entity_store_assignment_name=custom_entity_store_assignment_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if response.status_code == 200: + deserialized = self._deserialize('CustomEntityStoreAssignment', pipeline_response) + + if response.status_code == 201: + deserialized = self._deserialize('CustomEntityStoreAssignment', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + create.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}'} # type: ignore + + + @distributed_trace + def delete( + self, + resource_group_name: str, + custom_entity_store_assignment_name: str, + **kwargs: Any + ) -> None: + """Deleted a custom entity store assignment. + + Delete a custom entity store assignment by name for a provided subscription. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param custom_entity_store_assignment_name: Name of the custom entity store assignment. + Generated name is GUID. + :type custom_entity_store_assignment_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: None, or the result of cls(response) + :rtype: None + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType[None] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + custom_entity_store_assignment_name=custom_entity_store_assignment_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) + + delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments/{customEntityStoreAssignmentName}'} # type: ignore + + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.CustomEntityStoreAssignmentsListResult"]: + """List custom entity store assignments in a subscription and a resource group. + + List custom entity store assignments by a provided subscription and resource group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomEntityStoreAssignmentsListResult or the + result of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.CustomEntityStoreAssignmentsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignmentsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + def extract_data(pipeline_response): + deserialized = self._deserialize("CustomEntityStoreAssignmentsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, iter(list_of_elem) + + def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return ItemPaged( + get_next, extract_data + ) + list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Security/customEntityStoreAssignments'} # type: ignore + + @distributed_trace + def list_by_subscription( + self, + **kwargs: Any + ) -> Iterable["_models.CustomEntityStoreAssignmentsListResult"]: + """List custom entity store assignments in a subscription. + + List custom entity store assignments by provided subscription. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either CustomEntityStoreAssignmentsListResult or the + result of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.CustomEntityStoreAssignmentsListResult] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.CustomEntityStoreAssignmentsListResult"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + def extract_data(pipeline_response): + deserialized = self._deserialize("CustomEntityStoreAssignmentsListResult", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, iter(list_of_elem) + + def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return ItemPaged( + get_next, extract_data + ) + list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/customEntityStoreAssignments'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_device_security_groups_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_device_security_groups_operations.py index 35044ec43bdfa..7150f09792bd0 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_device_security_groups_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_device_security_groups_operations.py @@ -5,23 +5,165 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + resource_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + resource_id: str, + device_security_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "deviceSecurityGroupName": _SERIALIZER.url("device_security_group_name", device_security_group_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + resource_id: str, + device_security_group_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "deviceSecurityGroupName": _SERIALIZER.url("device_security_group_name", device_security_group_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + resource_id: str, + device_security_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + "deviceSecurityGroupName": _SERIALIZER.url("device_security_group_name", device_security_group_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class DeviceSecurityGroupsOperations(object): """DeviceSecurityGroupsOperations operations. @@ -45,18 +187,19 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.DeviceSecurityGroupList"] + resource_id: str, + **kwargs: Any + ) -> Iterable["_models.DeviceSecurityGroupList"]: """Use this method get the list of device security groups for the specified IoT Hub resource. :param resource_id: The identifier of the resource. :type resource_id: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DeviceSecurityGroupList or the result of cls(response) + :return: An iterator like instance of either DeviceSecurityGroupList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.DeviceSecurityGroupList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -65,34 +208,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + resource_id=resource_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + resource_id=resource_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('DeviceSecurityGroupList', pipeline_response) + deserialized = self._deserialize("DeviceSecurityGroupList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -110,18 +248,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups'} # type: ignore + @distributed_trace def get( self, - resource_id, # type: str - device_security_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.DeviceSecurityGroup" + resource_id: str, + device_security_group_name: str, + **kwargs: Any + ) -> "_models.DeviceSecurityGroup": """Use this method to get the device security group for the specified IoT Hub resource. :param resource_id: The identifier of the resource. @@ -139,26 +278,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'deviceSecurityGroupName': self._serialize.url("device_security_group_name", device_security_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + resource_id=resource_id, + device_security_group_name=device_security_group_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -172,16 +301,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}'} # type: ignore + + @distributed_trace def create_or_update( self, - resource_id, # type: str - device_security_group_name, # type: str - device_security_group, # type: "_models.DeviceSecurityGroup" - **kwargs # type: Any - ): - # type: (...) -> "_models.DeviceSecurityGroup" + resource_id: str, + device_security_group_name: str, + device_security_group: "_models.DeviceSecurityGroup", + **kwargs: Any + ) -> "_models.DeviceSecurityGroup": """Use this method to creates or updates the device security group on a specified IoT Hub resource. @@ -202,31 +333,21 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'deviceSecurityGroupName': self._serialize.url("device_security_group_name", device_security_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(device_security_group, 'DeviceSecurityGroup') + + request = build_create_or_update_request( + resource_id=resource_id, + device_security_group_name=device_security_group_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(device_security_group, 'DeviceSecurityGroup') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -244,15 +365,17 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}'} # type: ignore + + @distributed_trace def delete( self, - resource_id, # type: str - device_security_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + resource_id: str, + device_security_group_name: str, + **kwargs: Any + ) -> None: """User this method to deletes the device security group. :param resource_id: The identifier of the resource. @@ -270,26 +393,16 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - 'deviceSecurityGroupName': self._serialize.url("device_security_group_name", device_security_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + resource_id=resource_id, + device_security_group_name=device_security_group_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -301,3 +414,4 @@ def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/deviceSecurityGroups/{deviceSecurityGroupName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_discovered_security_solutions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_discovered_security_solutions_operations.py index 5a628e8a11230..987b38a2699c1 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_discovered_security_solutions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_discovered_security_solutions_operations.py @@ -5,23 +5,126 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/discoveredSecuritySolutions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_home_region_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/discoveredSecuritySolutions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + discovered_security_solution_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/discoveredSecuritySolutions/{discoveredSecuritySolutionName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "discoveredSecuritySolutionName": _SERIALIZER.url("discovered_security_solution_name", discovered_security_solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class DiscoveredSecuritySolutionsOperations(object): """DiscoveredSecuritySolutionsOperations operations. @@ -45,16 +148,18 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.DiscoveredSecuritySolutionList"] + **kwargs: Any + ) -> Iterable["_models.DiscoveredSecuritySolutionList"]: """Gets a list of discovered Security Solutions for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] + :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.DiscoveredSecuritySolutionList"] @@ -62,34 +167,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('DiscoveredSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("DiscoveredSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,21 +207,24 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/discoveredSecuritySolutions'} # type: ignore + @distributed_trace def list_by_home_region( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.DiscoveredSecuritySolutionList"] + **kwargs: Any + ) -> Iterable["_models.DiscoveredSecuritySolutionList"]: """Gets a list of discovered Security Solutions for the subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] + :return: An iterator like instance of either DiscoveredSecuritySolutionList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.DiscoveredSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.DiscoveredSecuritySolutionList"] @@ -129,35 +232,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('DiscoveredSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("DiscoveredSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -175,18 +274,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/discoveredSecuritySolutions'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - discovered_security_solution_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.DiscoveredSecuritySolution" + resource_group_name: str, + discovered_security_solution_name: str, + **kwargs: Any + ) -> "_models.DiscoveredSecuritySolution": """Gets a specific discovered Security Solution. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -204,28 +304,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'discoveredSecuritySolutionName': self._serialize.url("discovered_security_solution_name", discovered_security_solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + discovered_security_solution_name=discovered_security_solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -239,4 +329,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/discoveredSecuritySolutions/{discoveredSecuritySolutionName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_external_security_solutions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_external_security_solutions_operations.py index 6a9375abd1a93..12a4c0ab4ed05 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_external_security_solutions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_external_security_solutions_operations.py @@ -5,23 +5,126 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/externalSecuritySolutions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_home_region_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/ExternalSecuritySolutions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + external_security_solutions_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/ExternalSecuritySolutions/{externalSecuritySolutionsName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "externalSecuritySolutionsName": _SERIALIZER.url("external_security_solutions_name", external_security_solutions_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class ExternalSecuritySolutionsOperations(object): """ExternalSecuritySolutionsOperations operations. @@ -45,15 +148,16 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.ExternalSecuritySolutionList"] + **kwargs: Any + ) -> Iterable["_models.ExternalSecuritySolutionList"]: """Gets a list of external security solutions for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ExternalSecuritySolutionList or the result of cls(response) + :return: An iterator like instance of either ExternalSecuritySolutionList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.ExternalSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -62,34 +166,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('ExternalSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("ExternalSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,20 +206,22 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/externalSecuritySolutions'} # type: ignore + @distributed_trace def list_by_home_region( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.ExternalSecuritySolutionList"] + **kwargs: Any + ) -> Iterable["_models.ExternalSecuritySolutionList"]: """Gets a list of external Security Solutions for the subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either ExternalSecuritySolutionList or the result of cls(response) + :return: An iterator like instance of either ExternalSecuritySolutionList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.ExternalSecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -129,35 +230,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('ExternalSecuritySolutionList', pipeline_response) + deserialized = self._deserialize("ExternalSecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -175,18 +272,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/ExternalSecuritySolutions'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - external_security_solutions_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ExternalSecuritySolution" + resource_group_name: str, + external_security_solutions_name: str, + **kwargs: Any + ) -> "_models.ExternalSecuritySolution": """Gets a specific external Security Solution. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -204,28 +302,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'externalSecuritySolutionsName': self._serialize.url("external_security_solutions_name", external_security_solutions_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + external_security_solutions_name=external_security_solutions_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -239,4 +327,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/ExternalSecuritySolutions/{externalSecuritySolutionsName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_information_protection_policies_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_information_protection_policies_operations.py index 891af46bac018..88b3e29f0fb58 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_information_protection_policies_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_information_protection_policies_operations.py @@ -5,23 +5,132 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_get_request( + scope: str, + information_protection_policy_name: Union[str, "_models.Enum17"], + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/informationProtectionPolicies/{informationProtectionPolicyName}') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + "informationProtectionPolicyName": _SERIALIZER.url("information_protection_policy_name", information_protection_policy_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + scope: str, + information_protection_policy_name: Union[str, "_models.Enum17"], + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/informationProtectionPolicies/{informationProtectionPolicyName}') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + "informationProtectionPolicyName": _SERIALIZER.url("information_protection_policy_name", information_protection_policy_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_list_request( + scope: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/informationProtectionPolicies') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class InformationProtectionPoliciesOperations(object): """InformationProtectionPoliciesOperations operations. @@ -45,13 +154,13 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def get( self, - scope, # type: str - information_protection_policy_name, # type: Union[str, "_models.Enum15"] - **kwargs # type: Any - ): - # type: (...) -> "_models.InformationProtectionPolicy" + scope: str, + information_protection_policy_name: Union[str, "_models.Enum17"], + **kwargs: Any + ) -> "_models.InformationProtectionPolicy": """Details of the information protection policy. :param scope: Scope of the query, can be subscription @@ -59,7 +168,7 @@ def get( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :param information_protection_policy_name: Name of the information protection policy. - :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum15 + :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum17 :keyword callable cls: A custom type or function that will be passed the direct response :return: InformationProtectionPolicy, or the result of cls(response) :rtype: ~azure.mgmt.security.models.InformationProtectionPolicy @@ -70,26 +179,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'informationProtectionPolicyName': self._serialize.url("information_protection_policy_name", information_protection_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scope=scope, + information_protection_policy_name=information_protection_policy_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -103,16 +202,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{scope}/providers/Microsoft.Security/informationProtectionPolicies/{informationProtectionPolicyName}'} # type: ignore + + @distributed_trace def create_or_update( self, - scope, # type: str - information_protection_policy_name, # type: Union[str, "_models.Enum15"] - information_protection_policy, # type: "_models.InformationProtectionPolicy" - **kwargs # type: Any - ): - # type: (...) -> "_models.InformationProtectionPolicy" + scope: str, + information_protection_policy_name: Union[str, "_models.Enum17"], + information_protection_policy: "_models.InformationProtectionPolicy", + **kwargs: Any + ) -> "_models.InformationProtectionPolicy": """Details of the information protection policy. :param scope: Scope of the query, can be subscription @@ -120,7 +221,7 @@ def create_or_update( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :param information_protection_policy_name: Name of the information protection policy. - :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum15 + :type information_protection_policy_name: str or ~azure.mgmt.security.models.Enum17 :param information_protection_policy: Information protection policy. :type information_protection_policy: ~azure.mgmt.security.models.InformationProtectionPolicy :keyword callable cls: A custom type or function that will be passed the direct response @@ -133,31 +234,21 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'informationProtectionPolicyName': self._serialize.url("information_protection_policy_name", information_protection_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(information_protection_policy, 'InformationProtectionPolicy') + + request = build_create_or_update_request( + scope=scope, + information_protection_policy_name=information_protection_policy_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(information_protection_policy, 'InformationProtectionPolicy') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -175,14 +266,16 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{scope}/providers/Microsoft.Security/informationProtectionPolicies/{informationProtectionPolicyName}'} # type: ignore + + @distributed_trace def list( self, - scope, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.InformationProtectionPolicyList"] + scope: str, + **kwargs: Any + ) -> Iterable["_models.InformationProtectionPolicyList"]: """Information protection policies of a specific management group. :param scope: Scope of the query, can be subscription @@ -190,8 +283,10 @@ def list( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either InformationProtectionPolicyList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.InformationProtectionPolicyList] + :return: An iterator like instance of either InformationProtectionPolicyList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.InformationProtectionPolicyList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.InformationProtectionPolicyList"] @@ -199,34 +294,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('InformationProtectionPolicyList', pipeline_response) + deserialized = self._deserialize("InformationProtectionPolicyList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -244,6 +334,7 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_ingestion_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_ingestion_settings_operations.py index 0e8366ffbd0e9..5c6e7e64d6472 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_ingestion_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_ingestion_settings_operations.py @@ -5,23 +5,231 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-15-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + ingestion_setting_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-15-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ingestionSettingName": _SERIALIZER.url("ingestion_setting_name", ingestion_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_request( + subscription_id: str, + ingestion_setting_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-01-15-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ingestionSettingName": _SERIALIZER.url("ingestion_setting_name", ingestion_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + ingestion_setting_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-15-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ingestionSettingName": _SERIALIZER.url("ingestion_setting_name", ingestion_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_tokens_request( + subscription_id: str, + ingestion_setting_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-15-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}/listTokens') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ingestionSettingName": _SERIALIZER.url("ingestion_setting_name", ingestion_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_connection_strings_request( + subscription_id: str, + ingestion_setting_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-01-15-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}/listConnectionStrings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ingestionSettingName": _SERIALIZER.url("ingestion_setting_name", ingestion_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class IngestionSettingsOperations(object): """IngestionSettingsOperations operations. @@ -45,16 +253,17 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.IngestionSettingList"] + **kwargs: Any + ) -> Iterable["_models.IngestionSettingList"]: """Settings for ingesting security data and logs to correlate with resources associated with the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IngestionSettingList or the result of cls(response) + :return: An iterator like instance of either IngestionSettingList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.IngestionSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -63,34 +272,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('IngestionSettingList', pipeline_response) + deserialized = self._deserialize("IngestionSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -108,17 +312,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings'} # type: ignore + @distributed_trace def get( self, - ingestion_setting_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.IngestionSetting" + ingestion_setting_name: str, + **kwargs: Any + ) -> "_models.IngestionSetting": """Settings for ingesting security data and logs to correlate with resources associated with the subscription. @@ -134,26 +339,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -167,15 +362,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}'} # type: ignore + + @distributed_trace def create( self, - ingestion_setting_name, # type: str - ingestion_setting, # type: "_models.IngestionSetting" - **kwargs # type: Any - ): - # type: (...) -> "_models.IngestionSetting" + ingestion_setting_name: str, + ingestion_setting: "_models.IngestionSetting", + **kwargs: Any + ) -> "_models.IngestionSetting": """Create setting for ingesting security data and logs to correlate with resources associated with the subscription. @@ -193,31 +390,21 @@ def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(ingestion_setting, 'IngestionSetting') + + request = build_create_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(ingestion_setting, 'IngestionSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -231,14 +418,16 @@ def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}'} # type: ignore + + @distributed_trace def delete( self, - ingestion_setting_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + ingestion_setting_name: str, + **kwargs: Any + ) -> None: """Deletes the ingestion settings for this subscription. :param ingestion_setting_name: Name of the ingestion setting. @@ -253,26 +442,16 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -285,12 +464,13 @@ def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}'} # type: ignore + + @distributed_trace def list_tokens( self, - ingestion_setting_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.IngestionSettingToken" + ingestion_setting_name: str, + **kwargs: Any + ) -> "_models.IngestionSettingToken": """Returns the token that is used for correlating ingested telemetry with the resources in the subscription. @@ -306,26 +486,16 @@ def list_tokens( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.list_tokens.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_tokens_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.list_tokens.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -339,14 +509,16 @@ def list_tokens( return cls(pipeline_response, deserialized, {}) return deserialized + list_tokens.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}/listTokens'} # type: ignore + + @distributed_trace def list_connection_strings( self, - ingestion_setting_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ConnectionStrings" + ingestion_setting_name: str, + **kwargs: Any + ) -> "_models.ConnectionStrings": """Connection strings for ingesting security scan logs and data. :param ingestion_setting_name: Name of the ingestion setting. @@ -361,26 +533,16 @@ def list_connection_strings( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-01-15-preview" - accept = "application/json" - - # Construct URL - url = self.list_connection_strings.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ingestionSettingName': self._serialize.url("ingestion_setting_name", ingestion_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_connection_strings_request( + subscription_id=self._config.subscription_id, + ingestion_setting_name=ingestion_setting_name, + template_url=self.list_connection_strings.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -394,4 +556,6 @@ def list_connection_strings( return cls(pipeline_response, deserialized, {}) return deserialized + list_connection_strings.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/ingestionSettings/{ingestionSettingName}/listConnectionStrings'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_analytics_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_analytics_operations.py index 298a0942fff64..53a6320a14f00 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_analytics_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_analytics_operations.py @@ -5,22 +5,94 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class IotSecuritySolutionAnalyticsOperations(object): """IotSecuritySolutionAnalyticsOperations operations. @@ -44,13 +116,13 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - resource_group_name, # type: str - solution_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.IoTSecuritySolutionAnalyticsModelList" + resource_group_name: str, + solution_name: str, + **kwargs: Any + ) -> "_models.IoTSecuritySolutionAnalyticsModelList": """Use this method to get IoT security Analytics metrics in an array. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -68,27 +140,17 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -102,15 +164,17 @@ def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels'} # type: ignore + + @distributed_trace def get( self, - resource_group_name, # type: str - solution_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.IoTSecuritySolutionAnalyticsModel" + resource_group_name: str, + solution_name: str, + **kwargs: Any + ) -> "_models.IoTSecuritySolutionAnalyticsModel": """Use this method to get IoT Security Analytics metrics. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -128,27 +192,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -162,4 +216,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_operations.py index 61ddebb955185..397164c818584 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solution_operations.py @@ -5,23 +5,256 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_by_subscription_request( + subscription_id: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/iotSecuritySolutions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_update_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PATCH", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class IotSecuritySolutionOperations(object): """IotSecuritySolutionOperations operations. @@ -45,19 +278,20 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list_by_subscription( self, - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.IoTSecuritySolutionsList"] + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.IoTSecuritySolutionsList"]: """Use this method to get the list of IoT Security solutions by subscription. :param filter: Filter the IoT Security solution with OData syntax. Supports filtering by iotHubs. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecuritySolutionsList or the result of cls(response) + :return: An iterator like instance of either IoTSecuritySolutionsList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.IoTSecuritySolutionsList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -66,36 +300,31 @@ def list_by_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecuritySolutionsList', pipeline_response) + deserialized = self._deserialize("IoTSecuritySolutionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -113,18 +342,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/iotSecuritySolutions'} # type: ignore + @distributed_trace def list_by_resource_group( self, - resource_group_name, # type: str - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.IoTSecuritySolutionsList"] + resource_group_name: str, + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.IoTSecuritySolutionsList"]: """Use this method to get the list IoT Security solutions organized by resource group. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -134,7 +364,8 @@ def list_by_resource_group( iotHubs. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecuritySolutionsList or the result of cls(response) + :return: An iterator like instance of either IoTSecuritySolutionsList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.IoTSecuritySolutionsList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -143,37 +374,33 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + filter=filter, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecuritySolutionsList', pipeline_response) + deserialized = self._deserialize("IoTSecuritySolutionsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -191,18 +418,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - solution_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.IoTSecuritySolutionModel" + resource_group_name: str, + solution_name: str, + **kwargs: Any + ) -> "_models.IoTSecuritySolutionModel": """User this method to get details of a specific IoT Security solution based on solution name. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -220,27 +448,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -254,16 +472,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + + @distributed_trace def create_or_update( self, - resource_group_name, # type: str - solution_name, # type: str - iot_security_solution_data, # type: "_models.IoTSecuritySolutionModel" - **kwargs # type: Any - ): - # type: (...) -> "_models.IoTSecuritySolutionModel" + resource_group_name: str, + solution_name: str, + iot_security_solution_data: "_models.IoTSecuritySolutionModel", + **kwargs: Any + ) -> "_models.IoTSecuritySolutionModel": """Use this method to create or update yours IoT Security solution. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -283,32 +503,22 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(iot_security_solution_data, 'IoTSecuritySolutionModel') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(iot_security_solution_data, 'IoTSecuritySolutionModel') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -326,16 +536,18 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + + @distributed_trace def update( self, - resource_group_name, # type: str - solution_name, # type: str - update_iot_security_solution_data, # type: "_models.UpdateIotSecuritySolutionData" - **kwargs # type: Any - ): - # type: (...) -> "_models.IoTSecuritySolutionModel" + resource_group_name: str, + solution_name: str, + update_iot_security_solution_data: "_models.UpdateIotSecuritySolutionData", + **kwargs: Any + ) -> "_models.IoTSecuritySolutionModel": """Use this method to update existing IoT Security solution tags or user defined resources. To update other fields use the CreateOrUpdate method. @@ -345,7 +557,8 @@ def update( :param solution_name: The name of the IoT Security solution. :type solution_name: str :param update_iot_security_solution_data: The security solution data. - :type update_iot_security_solution_data: ~azure.mgmt.security.models.UpdateIotSecuritySolutionData + :type update_iot_security_solution_data: + ~azure.mgmt.security.models.UpdateIotSecuritySolutionData :keyword callable cls: A custom type or function that will be passed the direct response :return: IoTSecuritySolutionModel, or the result of cls(response) :rtype: ~azure.mgmt.security.models.IoTSecuritySolutionModel @@ -356,32 +569,22 @@ def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(update_iot_security_solution_data, 'UpdateIotSecuritySolutionData') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(update_iot_security_solution_data, 'UpdateIotSecuritySolutionData') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -395,15 +598,17 @@ def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + + @distributed_trace def delete( self, - resource_group_name, # type: str - solution_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + resource_group_name: str, + solution_name: str, + **kwargs: Any + ) -> None: """Use this method to delete yours IoT Security solution. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -421,27 +626,17 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -453,3 +648,4 @@ def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py index 65f9c6b520184..70b72da6d41bc 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_aggregated_alert_operations.py @@ -5,23 +5,138 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + *, + top: Optional[int] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if top is not None: + query_parameters['$top'] = _SERIALIZER.query("top", top, 'int') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + aggregated_alert_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts/{aggregatedAlertName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + "aggregatedAlertName": _SERIALIZER.url("aggregated_alert_name", aggregated_alert_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_dismiss_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + aggregated_alert_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts/{aggregatedAlertName}/dismiss') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + "aggregatedAlertName": _SERIALIZER.url("aggregated_alert_name", aggregated_alert_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class IotSecuritySolutionsAnalyticsAggregatedAlertOperations(object): """IotSecuritySolutionsAnalyticsAggregatedAlertOperations operations. @@ -45,14 +160,14 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - resource_group_name, # type: str - solution_name, # type: str - top=None, # type: Optional[int] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.IoTSecurityAggregatedAlertList"] + resource_group_name: str, + solution_name: str, + top: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.IoTSecurityAggregatedAlertList"]: """Use this method to get the aggregated alert list of yours IoT Security solution. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -63,8 +178,10 @@ def list( :param top: Number of results to retrieve. :type top: int :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecurityAggregatedAlertList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedAlertList] + :return: An iterator like instance of either IoTSecurityAggregatedAlertList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedAlertList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.IoTSecurityAggregatedAlertList"] @@ -72,38 +189,35 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecurityAggregatedAlertList', pipeline_response) + deserialized = self._deserialize("IoTSecurityAggregatedAlertList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -121,19 +235,20 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - solution_name, # type: str - aggregated_alert_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.IoTSecurityAggregatedAlert" + resource_group_name: str, + solution_name: str, + aggregated_alert_name: str, + **kwargs: Any + ) -> "_models.IoTSecurityAggregatedAlert": """Use this method to get a single the aggregated alert of yours IoT Security solution. This aggregation is performed by alert name. @@ -154,28 +269,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - 'aggregatedAlertName': self._serialize.url("aggregated_alert_name", aggregated_alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + aggregated_alert_name=aggregated_alert_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -189,16 +294,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts/{aggregatedAlertName}'} # type: ignore + + @distributed_trace def dismiss( self, - resource_group_name, # type: str - solution_name, # type: str - aggregated_alert_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + resource_group_name: str, + solution_name: str, + aggregated_alert_name: str, + **kwargs: Any + ) -> None: """Use this method to dismiss an aggregated IoT Security Solution Alert. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -218,28 +325,18 @@ def dismiss( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.dismiss.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - 'aggregatedAlertName': self._serialize.url("aggregated_alert_name", aggregated_alert_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_dismiss_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + aggregated_alert_name=aggregated_alert_name, + template_url=self.dismiss.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -251,3 +348,4 @@ def dismiss( return cls(pipeline_response, None, {}) dismiss.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedAlerts/{aggregatedAlertName}/dismiss'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_recommendation_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_recommendation_operations.py index 6b59514a802aa..1d44d86fc4934 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_recommendation_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_iot_security_solutions_analytics_recommendation_operations.py @@ -5,23 +5,101 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +def build_get_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + aggregated_recommendation_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedRecommendations/{aggregatedRecommendationName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + "aggregatedRecommendationName": _SERIALIZER.url("aggregated_recommendation_name", aggregated_recommendation_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_request( + subscription_id: str, + resource_group_name: str, + solution_name: str, + *, + top: Optional[int] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-08-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedRecommendations') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "solutionName": _SERIALIZER.url("solution_name", solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if top is not None: + query_parameters['$top'] = _SERIALIZER.query("top", top, 'int') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class IotSecuritySolutionsAnalyticsRecommendationOperations(object): """IotSecuritySolutionsAnalyticsRecommendationOperations operations. @@ -45,14 +123,14 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def get( self, - resource_group_name, # type: str - solution_name, # type: str - aggregated_recommendation_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.IoTSecurityAggregatedRecommendation" + resource_group_name: str, + solution_name: str, + aggregated_recommendation_name: str, + **kwargs: Any + ) -> "_models.IoTSecurityAggregatedRecommendation": """Use this method to get the aggregated security analytics recommendation of yours IoT Security solution. This aggregation is performed by recommendation name. @@ -73,28 +151,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - 'aggregatedRecommendationName': self._serialize.url("aggregated_recommendation_name", aggregated_recommendation_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + aggregated_recommendation_name=aggregated_recommendation_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -108,16 +176,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/iotSecuritySolutions/{solutionName}/analyticsModels/default/aggregatedRecommendations/{aggregatedRecommendationName}'} # type: ignore + + @distributed_trace def list( self, - resource_group_name, # type: str - solution_name, # type: str - top=None, # type: Optional[int] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.IoTSecurityAggregatedRecommendationList"] + resource_group_name: str, + solution_name: str, + top: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.IoTSecurityAggregatedRecommendationList"]: """Use this method to get the list of aggregated security analytics recommendations of yours IoT Security solution. @@ -129,8 +199,10 @@ def list( :param top: Number of results to retrieve. :type top: int :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either IoTSecurityAggregatedRecommendationList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendationList] + :return: An iterator like instance of either IoTSecurityAggregatedRecommendationList or the + result of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.IoTSecurityAggregatedRecommendationList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.IoTSecurityAggregatedRecommendationList"] @@ -138,38 +210,35 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-08-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'solutionName': self._serialize.url("solution_name", solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + solution_name=solution_name, + top=top, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('IoTSecurityAggregatedRecommendationList', pipeline_response) + deserialized = self._deserialize("IoTSecurityAggregatedRecommendationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -187,6 +256,7 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_jit_network_access_policies_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_jit_network_access_policies_operations.py index 83347ca88ca1f..f384c84b1c3e7 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_jit_network_access_policies_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_jit_network_access_policies_operations.py @@ -5,23 +5,326 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/jitNetworkAccessPolicies') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_region_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/jitNetworkAccessPolicies') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_and_region_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + jit_network_access_policy_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "jitNetworkAccessPolicyName": _SERIALIZER.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + jit_network_access_policy_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "jitNetworkAccessPolicyName": _SERIALIZER.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + jit_network_access_policy_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "jitNetworkAccessPolicyName": _SERIALIZER.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_initiate_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + jit_network_access_policy_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + jit_network_access_policy_initiate_type = "initiate" + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}/{jitNetworkAccessPolicyInitiateType}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "jitNetworkAccessPolicyName": _SERIALIZER.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), + "jitNetworkAccessPolicyInitiateType": _SERIALIZER.url("jit_network_access_policy_initiate_type", jit_network_access_policy_initiate_type, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class JitNetworkAccessPoliciesOperations(object): """JitNetworkAccessPoliciesOperations operations. @@ -45,15 +348,16 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.JitNetworkAccessPoliciesList"] + **kwargs: Any + ) -> Iterable["_models.JitNetworkAccessPoliciesList"]: """Policies for protecting resources using Just-in-Time access control. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -62,34 +366,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,21 +406,23 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace def list_by_region( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.JitNetworkAccessPoliciesList"] + **kwargs: Any + ) -> Iterable["_models.JitNetworkAccessPoliciesList"]: """Policies for protecting resources using Just-in-Time access control for the subscription, location. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -130,35 +431,31 @@ def list_by_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -176,17 +473,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace def list_by_resource_group( self, - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.JitNetworkAccessPoliciesList"] + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.JitNetworkAccessPoliciesList"]: """Policies for protecting resources using Just-in-Time access control for the subscription, location. @@ -194,7 +492,8 @@ def list_by_resource_group( name is case insensitive. :type resource_group_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -203,35 +502,31 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -249,17 +544,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace def list_by_resource_group_and_region( self, - resource_group_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.JitNetworkAccessPoliciesList"] + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.JitNetworkAccessPoliciesList"]: """Policies for protecting resources using Just-in-Time access control for the subscription, location. @@ -267,7 +563,8 @@ def list_by_resource_group_and_region( name is case insensitive. :type resource_group_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of cls(response) + :return: An iterator like instance of either JitNetworkAccessPoliciesList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.JitNetworkAccessPoliciesList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -276,36 +573,33 @@ def list_by_resource_group_and_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group_and_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_and_region_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + template_url=self.list_by_resource_group_and_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_and_region_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('JitNetworkAccessPoliciesList', pipeline_response) + deserialized = self._deserialize("JitNetworkAccessPoliciesList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -323,18 +617,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_resource_group_and_region.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - jit_network_access_policy_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.JitNetworkAccessPolicy" + resource_group_name: str, + jit_network_access_policy_name: str, + **kwargs: Any + ) -> "_models.JitNetworkAccessPolicy": """Policies for protecting resources using Just-in-Time access control for the subscription, location. @@ -353,28 +648,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -388,16 +673,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}'} # type: ignore + + @distributed_trace def create_or_update( self, - resource_group_name, # type: str - jit_network_access_policy_name, # type: str - body, # type: "_models.JitNetworkAccessPolicy" - **kwargs # type: Any - ): - # type: (...) -> "_models.JitNetworkAccessPolicy" + resource_group_name: str, + jit_network_access_policy_name: str, + body: "_models.JitNetworkAccessPolicy", + **kwargs: Any + ) -> "_models.JitNetworkAccessPolicy": """Create a policy for protecting resources using Just-in-Time access control. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -417,33 +704,23 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'JitNetworkAccessPolicy') + + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'JitNetworkAccessPolicy') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -457,15 +734,17 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}'} # type: ignore + + @distributed_trace def delete( self, - resource_group_name, # type: str - jit_network_access_policy_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + resource_group_name: str, + jit_network_access_policy_name: str, + **kwargs: Any + ) -> None: """Delete a Just-in-Time access control policy. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -483,28 +762,18 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -517,14 +786,15 @@ def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}'} # type: ignore + + @distributed_trace def initiate( self, - resource_group_name, # type: str - jit_network_access_policy_name, # type: str - body, # type: "_models.JitNetworkAccessPolicyInitiateRequest" - **kwargs # type: Any - ): - # type: (...) -> "_models.JitNetworkAccessRequest" + resource_group_name: str, + jit_network_access_policy_name: str, + body: "_models.JitNetworkAccessPolicyInitiateRequest", + **kwargs: Any + ) -> "_models.JitNetworkAccessRequest": """Initiate a JIT access from a specific Just-in-Time policy configuration. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -544,35 +814,23 @@ def initiate( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - jit_network_access_policy_initiate_type = "initiate" - api_version = "2020-01-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.initiate.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'jitNetworkAccessPolicyName': self._serialize.url("jit_network_access_policy_name", jit_network_access_policy_name, 'str'), - 'jitNetworkAccessPolicyInitiateType': self._serialize.url("jit_network_access_policy_initiate_type", jit_network_access_policy_initiate_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(body, 'JitNetworkAccessPolicyInitiateRequest') + + request = build_initiate_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + jit_network_access_policy_name=jit_network_access_policy_name, + content_type=content_type, + json=_json, + template_url=self.initiate.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(body, 'JitNetworkAccessPolicyInitiateRequest') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -586,4 +844,6 @@ def initiate( return cls(pipeline_response, deserialized, {}) return deserialized + initiate.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/jitNetworkAccessPolicies/{jitNetworkAccessPolicyName}/{jitNetworkAccessPolicyInitiateType}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_locations_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_locations_operations.py index 13291ea538d64..7042beba20f05 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_locations_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_locations_operations.py @@ -5,23 +5,89 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class LocationsOperations(object): """LocationsOperations operations. @@ -45,11 +111,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.AscLocationList"] + **kwargs: Any + ) -> Iterable["_models.AscLocationList"]: """The location of the responsible ASC of the specific subscription (home region). For each subscription there is only one responsible location. The location in the response should be used to read or write other resources in ASC according to their ID. @@ -64,34 +130,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('AscLocationList', pipeline_response) + deserialized = self._deserialize("AscLocationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -109,16 +170,17 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations'} # type: ignore + @distributed_trace def get( self, - **kwargs # type: Any - ): - # type: (...) -> "_models.AscLocation" + **kwargs: Any + ) -> "_models.AscLocation": """Details of a specific location. :keyword callable cls: A custom type or function that will be passed the direct response @@ -131,26 +193,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -164,4 +216,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_mde_onboardings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_mde_onboardings_operations.py new file mode 100644 index 0000000000000..ec50d7e9b651a --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_mde_onboardings_operations.py @@ -0,0 +1,195 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar +import warnings + +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer + +from .. import models as _models +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-10-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/mdeOnboardings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-10-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/mdeOnboardings/default') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + +class MdeOnboardingsOperations(object): + """MdeOnboardingsOperations operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer): + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace + def list( + self, + **kwargs: Any + ) -> "_models.MdeOnboardingDataList": + """The configuration or data needed to onboard the machine to MDE. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: MdeOnboardingDataList, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.MdeOnboardingDataList + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.MdeOnboardingDataList"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('MdeOnboardingDataList', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/mdeOnboardings'} # type: ignore + + + @distributed_trace + def get( + self, + **kwargs: Any + ) -> "_models.MdeOnboardingData": + """The default configuration or data needed to onboard the machine to MDE. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: MdeOnboardingData, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.MdeOnboardingData + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.MdeOnboardingData"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('MdeOnboardingData', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/mdeOnboardings/default'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_operations.py index 7021bf97353cd..965c409554539 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_operations.py @@ -5,23 +5,50 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/providers/Microsoft.Security/operations') + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class Operations(object): """Operations operations. @@ -45,11 +72,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.OperationList"] + **kwargs: Any + ) -> Iterable["_models.OperationList"]: """Exposes all available operations for discovery purposes. :keyword callable cls: A custom type or function that will be passed the direct response @@ -62,30 +89,27 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('OperationList', pipeline_response) + deserialized = self._deserialize("OperationList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -103,6 +127,7 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_pricings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_pricings_operations.py index a52746e23b278..916fee85adff4 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_pricings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_pricings_operations.py @@ -5,22 +5,131 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2018-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + pricing_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2018-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings/{pricingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "pricingName": _SERIALIZER.url("pricing_name", pricing_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_request( + subscription_id: str, + pricing_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2018-06-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings/{pricingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "pricingName": _SERIALIZER.url("pricing_name", pricing_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class PricingsOperations(object): """PricingsOperations operations. @@ -44,11 +153,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> "_models.PricingList" + **kwargs: Any + ) -> "_models.PricingList": """Lists Security Center pricing configurations in the subscription. :keyword callable cls: A custom type or function that will be passed the direct response @@ -61,25 +170,15 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2018-06-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -93,14 +192,16 @@ def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings'} # type: ignore + + @distributed_trace def get( self, - pricing_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Pricing" + pricing_name: str, + **kwargs: Any + ) -> "_models.Pricing": """Gets a provided Security Center pricing configuration in the subscription. :param pricing_name: name of the pricing configuration. @@ -115,26 +216,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2018-06-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'pricingName': self._serialize.url("pricing_name", pricing_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + pricing_name=pricing_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -148,15 +239,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings/{pricingName}'} # type: ignore + + @distributed_trace def update( self, - pricing_name, # type: str - pricing, # type: "_models.Pricing" - **kwargs # type: Any - ): - # type: (...) -> "_models.Pricing" + pricing_name: str, + pricing: "_models.Pricing", + **kwargs: Any + ) -> "_models.Pricing": """Updates a provided Security Center pricing configuration in the subscription. :param pricing_name: name of the pricing configuration. @@ -173,31 +266,21 @@ def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2018-06-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'pricingName': self._serialize.url("pricing_name", pricing_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(pricing, 'Pricing') + + request = build_update_request( + subscription_id=self._config.subscription_id, + pricing_name=pricing_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(pricing, 'Pricing') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -211,4 +294,6 @@ def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/pricings/{pricingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_assessments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_assessments_operations.py index 1510be8b7bea0..ea44d665e6734 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_assessments_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_assessments_operations.py @@ -5,23 +5,101 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +def build_list_request( + subscription_id: str, + regulatory_compliance_standard_name: str, + regulatory_compliance_control_name: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}/regulatoryComplianceAssessments') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "regulatoryComplianceStandardName": _SERIALIZER.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), + "regulatoryComplianceControlName": _SERIALIZER.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + regulatory_compliance_standard_name: str, + regulatory_compliance_control_name: str, + regulatory_compliance_assessment_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}/regulatoryComplianceAssessments/{regulatoryComplianceAssessmentName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "regulatoryComplianceStandardName": _SERIALIZER.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), + "regulatoryComplianceControlName": _SERIALIZER.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), + "regulatoryComplianceAssessmentName": _SERIALIZER.url("regulatory_compliance_assessment_name", regulatory_compliance_assessment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class RegulatoryComplianceAssessmentsOperations(object): """RegulatoryComplianceAssessmentsOperations operations. @@ -45,14 +123,14 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - regulatory_compliance_standard_name, # type: str - regulatory_compliance_control_name, # type: str - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.RegulatoryComplianceAssessmentList"] + regulatory_compliance_standard_name: str, + regulatory_compliance_control_name: str, + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.RegulatoryComplianceAssessmentList"]: """Details and state of assessments mapped to selected regulatory compliance control. :param regulatory_compliance_standard_name: Name of the regulatory compliance standard object. @@ -62,8 +140,10 @@ def list( :param filter: OData filter. Optional. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either RegulatoryComplianceAssessmentList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.RegulatoryComplianceAssessmentList] + :return: An iterator like instance of either RegulatoryComplianceAssessmentList or the result + of cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.RegulatoryComplianceAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.RegulatoryComplianceAssessmentList"] @@ -71,38 +151,35 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - 'regulatoryComplianceControlName': self._serialize.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('RegulatoryComplianceAssessmentList', pipeline_response) + deserialized = self._deserialize("RegulatoryComplianceAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -120,19 +197,20 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}/regulatoryComplianceAssessments'} # type: ignore + @distributed_trace def get( self, - regulatory_compliance_standard_name, # type: str - regulatory_compliance_control_name, # type: str - regulatory_compliance_assessment_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.RegulatoryComplianceAssessment" + regulatory_compliance_standard_name: str, + regulatory_compliance_control_name: str, + regulatory_compliance_assessment_name: str, + **kwargs: Any + ) -> "_models.RegulatoryComplianceAssessment": """Supported regulatory compliance details and state for selected assessment. :param regulatory_compliance_standard_name: Name of the regulatory compliance standard object. @@ -152,28 +230,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - 'regulatoryComplianceControlName': self._serialize.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), - 'regulatoryComplianceAssessmentName': self._serialize.url("regulatory_compliance_assessment_name", regulatory_compliance_assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + regulatory_compliance_assessment_name=regulatory_compliance_assessment_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -187,4 +255,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}/regulatoryComplianceAssessments/{regulatoryComplianceAssessmentName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_controls_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_controls_operations.py index 181e4a4f49058..34ac3ca67beab 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_controls_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_controls_operations.py @@ -5,23 +5,97 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + regulatory_compliance_standard_name: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "regulatoryComplianceStandardName": _SERIALIZER.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + regulatory_compliance_standard_name: str, + regulatory_compliance_control_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "regulatoryComplianceStandardName": _SERIALIZER.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), + "regulatoryComplianceControlName": _SERIALIZER.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class RegulatoryComplianceControlsOperations(object): """RegulatoryComplianceControlsOperations operations. @@ -45,13 +119,13 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - regulatory_compliance_standard_name, # type: str - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.RegulatoryComplianceControlList"] + regulatory_compliance_standard_name: str, + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.RegulatoryComplianceControlList"]: """All supported regulatory compliance controls details and state for selected standard. :param regulatory_compliance_standard_name: Name of the regulatory compliance standard object. @@ -59,8 +133,10 @@ def list( :param filter: OData filter. Optional. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either RegulatoryComplianceControlList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.RegulatoryComplianceControlList] + :return: An iterator like instance of either RegulatoryComplianceControlList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.RegulatoryComplianceControlList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.RegulatoryComplianceControlList"] @@ -68,37 +144,33 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('RegulatoryComplianceControlList', pipeline_response) + deserialized = self._deserialize("RegulatoryComplianceControlList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -116,18 +188,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls'} # type: ignore + @distributed_trace def get( self, - regulatory_compliance_standard_name, # type: str - regulatory_compliance_control_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.RegulatoryComplianceControl" + regulatory_compliance_standard_name: str, + regulatory_compliance_control_name: str, + **kwargs: Any + ) -> "_models.RegulatoryComplianceControl": """Selected regulatory compliance control details and state. :param regulatory_compliance_standard_name: Name of the regulatory compliance standard object. @@ -144,27 +217,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - 'regulatoryComplianceControlName': self._serialize.url("regulatory_compliance_control_name", regulatory_compliance_control_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + regulatory_compliance_control_name=regulatory_compliance_control_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -178,4 +241,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}/regulatoryComplianceControls/{regulatoryComplianceControlName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_standards_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_standards_operations.py index 506a32bc4add2..a0466d38e1eb4 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_standards_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_regulatory_compliance_standards_operations.py @@ -5,23 +5,93 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + regulatory_compliance_standard_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "regulatoryComplianceStandardName": _SERIALIZER.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class RegulatoryComplianceStandardsOperations(object): """RegulatoryComplianceStandardsOperations operations. @@ -45,19 +115,21 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.RegulatoryComplianceStandardList"] + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.RegulatoryComplianceStandardList"]: """Supported regulatory compliance standards details and state. :param filter: OData filter. Optional. :type filter: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either RegulatoryComplianceStandardList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.RegulatoryComplianceStandardList] + :return: An iterator like instance of either RegulatoryComplianceStandardList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.RegulatoryComplianceStandardList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.RegulatoryComplianceStandardList"] @@ -65,36 +137,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('RegulatoryComplianceStandardList', pipeline_response) + deserialized = self._deserialize("RegulatoryComplianceStandardList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,17 +179,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards'} # type: ignore + @distributed_trace def get( self, - regulatory_compliance_standard_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.RegulatoryComplianceStandard" + regulatory_compliance_standard_name: str, + **kwargs: Any + ) -> "_models.RegulatoryComplianceStandard": """Supported regulatory compliance details state for selected standard. :param regulatory_compliance_standard_name: Name of the regulatory compliance standard object. @@ -137,26 +205,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'regulatoryComplianceStandardName': self._serialize.url("regulatory_compliance_standard_name", regulatory_compliance_standard_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + regulatory_compliance_standard_name=regulatory_compliance_standard_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -170,4 +228,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/regulatoryComplianceStandards/{regulatoryComplianceStandardName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_control_definitions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_control_definitions_operations.py index 8d0d8eab0fd3e..1c4b3a84c7102 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_control_definitions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_control_definitions_operations.py @@ -5,23 +5,81 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/providers/Microsoft.Security/secureScoreControlDefinitions') + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_subscription_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScoreControlDefinitions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SecureScoreControlDefinitionsOperations(object): """SecureScoreControlDefinitionsOperations operations. @@ -45,16 +103,18 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecureScoreControlDefinitionList"] + **kwargs: Any + ) -> Iterable["_models.SecureScoreControlDefinitionList"]: """List the available security controls, their assessments, and the max score. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] + :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecureScoreControlDefinitionList"] @@ -62,30 +122,27 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_list_request( + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlDefinitionList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlDefinitionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -103,22 +160,25 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/providers/Microsoft.Security/secureScoreControlDefinitions'} # type: ignore + @distributed_trace def list_by_subscription( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecureScoreControlDefinitionList"] + **kwargs: Any + ) -> Iterable["_models.SecureScoreControlDefinitionList"]: """For a specified subscription, list the available security controls, their assessments, and the max score. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] + :return: An iterator like instance of either SecureScoreControlDefinitionList or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecureScoreControlDefinitionList] :raises: ~azure.core.exceptions.HttpResponseError """ cls = kwargs.pop('cls', None) # type: ClsType["_models.SecureScoreControlDefinitionList"] @@ -126,34 +186,29 @@ def list_by_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlDefinitionList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlDefinitionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -171,6 +226,7 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_controls_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_controls_operations.py index 096bc5571b5c1..b4a0cbaacd881 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_controls_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_score_controls_operations.py @@ -5,23 +5,97 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +def build_list_by_secure_score_request( + subscription_id: str, + secure_score_name: str, + *, + expand: Optional[Union[str, "_models.ExpandControlsEnum"]] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores/{secureScoreName}/secureScoreControls') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "secureScoreName": _SERIALIZER.url("secure_score_name", secure_score_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if expand is not None: + query_parameters['$expand'] = _SERIALIZER.query("expand", expand, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_request( + subscription_id: str, + *, + expand: Optional[Union[str, "_models.ExpandControlsEnum"]] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScoreControls') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if expand is not None: + query_parameters['$expand'] = _SERIALIZER.query("expand", expand, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SecureScoreControlsOperations(object): """SecureScoreControlsOperations operations. @@ -45,13 +119,13 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list_by_secure_score( self, - secure_score_name, # type: str - expand=None, # type: Optional[Union[str, "_models.ExpandControlsEnum"]] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecureScoreControlList"] + secure_score_name: str, + expand: Optional[Union[str, "_models.ExpandControlsEnum"]] = None, + **kwargs: Any + ) -> Iterable["_models.SecureScoreControlList"]: """Get all security controls for a specific initiative within a scope. :param secure_score_name: The initiative name. For the ASC Default initiative, use 'ascScore' @@ -60,7 +134,8 @@ def list_by_secure_score( :param expand: OData expand. Optional. :type expand: str or ~azure.mgmt.security.models.ExpandControlsEnum :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlList or the result of cls(response) + :return: An iterator like instance of either SecureScoreControlList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecureScoreControlList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -69,37 +144,33 @@ def list_by_secure_score( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_secure_score.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'secureScoreName': self._serialize.url("secure_score_name", secure_score_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if expand is not None: - query_parameters['$expand'] = self._serialize.query("expand", expand, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_secure_score_request( + subscription_id=self._config.subscription_id, + secure_score_name=secure_score_name, + expand=expand, + template_url=self.list_by_secure_score.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_secure_score_request( + subscription_id=self._config.subscription_id, + secure_score_name=secure_score_name, + expand=expand, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -117,23 +188,25 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_secure_score.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores/{secureScoreName}/secureScoreControls'} # type: ignore + @distributed_trace def list( self, - expand=None, # type: Optional[Union[str, "_models.ExpandControlsEnum"]] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecureScoreControlList"] + expand: Optional[Union[str, "_models.ExpandControlsEnum"]] = None, + **kwargs: Any + ) -> Iterable["_models.SecureScoreControlList"]: """Get all security controls within a scope. :param expand: OData expand. Optional. :type expand: str or ~azure.mgmt.security.models.ExpandControlsEnum :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecureScoreControlList or the result of cls(response) + :return: An iterator like instance of either SecureScoreControlList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecureScoreControlList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -142,36 +215,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if expand is not None: - query_parameters['$expand'] = self._serialize.query("expand", expand, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + expand=expand, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + expand=expand, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoreControlList', pipeline_response) + deserialized = self._deserialize("SecureScoreControlList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -189,6 +257,7 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_scores_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_scores_operations.py index cd3e5835e36e3..3a45c1bdaf095 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_scores_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_secure_scores_operations.py @@ -5,23 +5,89 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + secure_score_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores/{secureScoreName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "secureScoreName": _SERIALIZER.url("secure_score_name", secure_score_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SecureScoresOperations(object): """SecureScoresOperations operations. @@ -45,11 +111,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecureScoresList"] + **kwargs: Any + ) -> Iterable["_models.SecureScoresList"]: """List secure scores for all your Security Center initiatives within your current scope. :keyword callable cls: A custom type or function that will be passed the direct response @@ -62,34 +128,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecureScoresList', pipeline_response) + deserialized = self._deserialize("SecureScoresList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,17 +168,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores'} # type: ignore + @distributed_trace def get( self, - secure_score_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecureScoreItem" + secure_score_name: str, + **kwargs: Any + ) -> "_models.SecureScoreItem": """Get secure score for a specific Security Center initiative within your current scope. For the ASC Default initiative, use 'ascScore'. @@ -134,26 +196,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'secureScoreName': self._serialize.url("secure_score_name", secure_score_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + secure_score_name=secure_score_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -167,4 +219,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/secureScores/{secureScoreName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_connectors_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_connectors_operations.py new file mode 100644 index 0000000000000..4cabc0abe6530 --- /dev/null +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_connectors_operations.py @@ -0,0 +1,632 @@ +# 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. +# -------------------------------------------------------------------------- +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union +import warnings + +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.paging import ItemPaged +from azure.core.pipeline import PipelineResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer + +from .. import models as _models +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityConnectors') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + security_connector_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "securityConnectorName": _SERIALIZER.url("security_connector_name", security_connector_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + subscription_id: str, + resource_group_name: str, + security_connector_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "securityConnectorName": _SERIALIZER.url("security_connector_name", security_connector_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_update_request( + subscription_id: str, + resource_group_name: str, + security_connector_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "securityConnectorName": _SERIALIZER.url("security_connector_name", security_connector_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PATCH", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + resource_group_name: str, + security_connector_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "securityConnectorName": _SERIALIZER.url("security_connector_name", security_connector_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + +class SecurityConnectorsOperations(object): + """SecurityConnectorsOperations operations. + + You should not instantiate this class directly. Instead, you should create a Client instance that + instantiates it for you and attaches it as an attribute. + + :ivar models: Alias to model classes used in this operation group. + :type models: ~azure.mgmt.security.models + :param client: Client for service requests. + :param config: Configuration of service client. + :param serializer: An object model serializer. + :param deserializer: An object model deserializer. + """ + + models = _models + + def __init__(self, client, config, serializer, deserializer): + self._client = client + self._serialize = serializer + self._deserialize = deserializer + self._config = config + + @distributed_trace + def list( + self, + **kwargs: Any + ) -> Iterable["_models.SecurityConnectorsList"]: + """Lists all the security connectors in the specified subscription. Use the 'nextLink' property in + the response to get the next page of security connectors for the specified subscription. + + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either SecurityConnectorsList or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecurityConnectorsList] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnectorsList"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + def extract_data(pipeline_response): + deserialized = self._deserialize("SecurityConnectorsList", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, iter(list_of_elem) + + def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return ItemPaged( + get_next, extract_data + ) + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityConnectors'} # type: ignore + + @distributed_trace + def list_by_resource_group( + self, + resource_group_name: str, + **kwargs: Any + ) -> Iterable["_models.SecurityConnectorsList"]: + """Lists all the security connectors in the specified resource group. Use the 'nextLink' property + in the response to get the next page of security connectors for the specified resource group. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: An iterator like instance of either SecurityConnectorsList or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecurityConnectorsList] + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnectorsList"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + def prepare_request(next_link=None): + if not next_link: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + else: + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" + return request + + def extract_data(pipeline_response): + deserialized = self._deserialize("SecurityConnectorsList", pipeline_response) + list_of_elem = deserialized.value + if cls: + list_of_elem = cls(list_of_elem) + return deserialized.next_link or None, iter(list_of_elem) + + def get_next(next_link=None): + request = prepare_request(next_link) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + return pipeline_response + + + return ItemPaged( + get_next, extract_data + ) + list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors'} # type: ignore + + @distributed_trace + def get( + self, + resource_group_name: str, + security_connector_name: str, + **kwargs: Any + ) -> "_models.SecurityConnector": + """Retrieves details of a specific security connector. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: SecurityConnector, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityConnector + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnector"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + security_connector_name: str, + security_connector: "_models.SecurityConnector", + **kwargs: Any + ) -> "_models.SecurityConnector": + """Creates or updates a security connector. If a security connector is already created and a + subsequent request is issued for the same security connector id, then it will be updated. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :param security_connector: The security connector resource. + :type security_connector: ~azure.mgmt.security.models.SecurityConnector + :keyword callable cls: A custom type or function that will be passed the direct response + :return: SecurityConnector, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityConnector + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnector"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(security_connector, 'SecurityConnector') + + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + content_type=content_type, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if response.status_code == 200: + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if response.status_code == 201: + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + + + @distributed_trace + def update( + self, + resource_group_name: str, + security_connector_name: str, + security_connector: "_models.SecurityConnector", + **kwargs: Any + ) -> "_models.SecurityConnector": + """Updates a security connector. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :param security_connector: The security connector resource. + :type security_connector: ~azure.mgmt.security.models.SecurityConnector + :keyword callable cls: A custom type or function that will be passed the direct response + :return: SecurityConnector, or the result of cls(response) + :rtype: ~azure.mgmt.security.models.SecurityConnector + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType["_models.SecurityConnector"] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(security_connector, 'SecurityConnector') + + request = build_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + deserialized = self._deserialize('SecurityConnector', pipeline_response) + + if cls: + return cls(pipeline_response, deserialized, {}) + + return deserialized + + update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + + + @distributed_trace + def delete( + self, + resource_group_name: str, + security_connector_name: str, + **kwargs: Any + ) -> None: + """Deletes a security connector. + + :param resource_group_name: The name of the resource group within the user's subscription. The + name is case insensitive. + :type resource_group_name: str + :param security_connector_name: The security connector name. + :type security_connector_name: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: None, or the result of cls(response) + :rtype: None + :raises: ~azure.core.exceptions.HttpResponseError + """ + cls = kwargs.pop('cls', None) # type: ClsType[None] + error_map = { + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + } + error_map.update(kwargs.pop('error_map', {})) + + + request = build_delete_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + security_connector_name=security_connector_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + response = pipeline_response.http_response + + if response.status_code not in [200, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + raise HttpResponseError(response=response, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) + + delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/securityConnectors/{securityConnectorName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_contacts_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_contacts_operations.py index 6f48ee66bb1c4..df9ff852a5cc9 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_contacts_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_contacts_operations.py @@ -5,23 +5,207 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + security_contact_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "securityContactName": _SERIALIZER.url("security_contact_name", security_contact_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_request( + subscription_id: str, + security_contact_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "securityContactName": _SERIALIZER.url("security_contact_name", security_contact_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + security_contact_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "securityContactName": _SERIALIZER.url("security_contact_name", security_contact_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_request( + subscription_id: str, + security_contact_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "securityContactName": _SERIALIZER.url("security_contact_name", security_contact_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PATCH", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class SecurityContactsOperations(object): """SecurityContactsOperations operations. @@ -45,11 +229,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecurityContactList"] + **kwargs: Any + ) -> Iterable["_models.SecurityContactList"]: """Security contact configurations for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response @@ -62,34 +246,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityContactList', pipeline_response) + deserialized = self._deserialize("SecurityContactList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,17 +286,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts'} # type: ignore + @distributed_trace def get( self, - security_contact_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityContact" + security_contact_name: str, + **kwargs: Any + ) -> "_models.SecurityContact": """Security contact configurations for the subscription. :param security_contact_name: Name of the security contact object. @@ -132,26 +312,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -165,15 +335,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + + @distributed_trace def create( self, - security_contact_name, # type: str - security_contact, # type: "_models.SecurityContact" - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityContact" + security_contact_name: str, + security_contact: "_models.SecurityContact", + **kwargs: Any + ) -> "_models.SecurityContact": """Security contact configurations for the subscription. :param security_contact_name: Name of the security contact object. @@ -190,31 +362,21 @@ def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(security_contact, 'SecurityContact') + + request = build_create_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(security_contact, 'SecurityContact') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -228,14 +390,16 @@ def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + + @distributed_trace def delete( self, - security_contact_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + security_contact_name: str, + **kwargs: Any + ) -> None: """Security contact configurations for the subscription. :param security_contact_name: Name of the security contact object. @@ -250,26 +414,16 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -282,13 +436,14 @@ def delete( delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + + @distributed_trace def update( self, - security_contact_name, # type: str - security_contact, # type: "_models.SecurityContact" - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityContact" + security_contact_name: str, + security_contact: "_models.SecurityContact", + **kwargs: Any + ) -> "_models.SecurityContact": """Security contact configurations for the subscription. :param security_contact_name: Name of the security contact object. @@ -305,31 +460,21 @@ def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'securityContactName': self._serialize.url("security_contact_name", security_contact_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(security_contact, 'SecurityContact') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_update_request( + subscription_id=self._config.subscription_id, + security_contact_name=security_contact_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(security_contact, 'SecurityContact') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -343,4 +488,6 @@ def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securityContacts/{securityContactName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_operations.py index 8bce5009e5ab4..d822a389e14ec 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_operations.py @@ -5,23 +5,93 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securitySolutions') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + security_solution_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/securitySolutions/{securitySolutionName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "securitySolutionName": _SERIALIZER.url("security_solution_name", security_solution_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SecuritySolutionsOperations(object): """SecuritySolutionsOperations operations. @@ -45,15 +115,16 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecuritySolutionList"] + **kwargs: Any + ) -> Iterable["_models.SecuritySolutionList"]: """Gets a list of Security Solutions for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecuritySolutionList or the result of cls(response) + :return: An iterator like instance of either SecuritySolutionList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecuritySolutionList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -62,34 +133,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecuritySolutionList', pipeline_response) + deserialized = self._deserialize("SecuritySolutionList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,18 +173,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securitySolutions'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - security_solution_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecuritySolution" + resource_group_name: str, + security_solution_name: str, + **kwargs: Any + ) -> "_models.SecuritySolution": """Gets a specific Security Solution. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -136,28 +203,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'securitySolutionName': self._serialize.url("security_solution_name", security_solution_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + security_solution_name=security_solution_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -171,4 +228,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/securitySolutions/{securitySolutionName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_reference_data_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_reference_data_operations.py index 513b9aa41c019..7a99df9803685 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_reference_data_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_security_solutions_reference_data_operations.py @@ -5,22 +5,88 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securitySolutionsReferenceData') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_home_region_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/securitySolutionsReferenceData') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SecuritySolutionsReferenceDataOperations(object): """SecuritySolutionsReferenceDataOperations operations. @@ -44,11 +110,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> "_models.SecuritySolutionsReferenceDataList" + **kwargs: Any + ) -> "_models.SecuritySolutionsReferenceDataList": """Gets a list of all supported Security Solutions for the subscription. :keyword callable cls: A custom type or function that will be passed the direct response @@ -61,25 +127,15 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -93,13 +149,15 @@ def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/securitySolutionsReferenceData'} # type: ignore + + @distributed_trace def list_by_home_region( self, - **kwargs # type: Any - ): - # type: (...) -> "_models.SecuritySolutionsReferenceDataList" + **kwargs: Any + ) -> "_models.SecuritySolutionsReferenceDataList": """Gets list of all supported Security Solutions for subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response @@ -112,26 +170,16 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -145,4 +193,6 @@ def list_by_home_region( return cls(pipeline_response, deserialized, {}) return deserialized + list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/securitySolutionsReferenceData'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_server_vulnerability_assessment_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_server_vulnerability_assessment_operations.py index 8d66fb87767ea..8b27a1551713c 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_server_vulnerability_assessment_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_server_vulnerability_assessment_operations.py @@ -5,24 +5,188 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat from azure.mgmt.core.polling.arm_polling import ARMPolling +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_by_extended_resource_request( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any +) -> HttpRequest: + server_vulnerability_assessment = "default" + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + "serverVulnerabilityAssessment": _SERIALIZER.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_or_update_request( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any +) -> HttpRequest: + server_vulnerability_assessment = "default" + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + "serverVulnerabilityAssessment": _SERIALIZER.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_delete_request_initial( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any +) -> HttpRequest: + server_vulnerability_assessment = "default" + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + "serverVulnerabilityAssessment": _SERIALIZER.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class ServerVulnerabilityAssessmentOperations(object): """ServerVulnerabilityAssessmentOperations operations. @@ -46,15 +210,15 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list_by_extended_resource( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ServerVulnerabilityAssessmentsList" + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any + ) -> "_models.ServerVulnerabilityAssessmentsList": """Gets a list of server vulnerability assessment onboarding statuses on a given resource. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -76,29 +240,19 @@ def list_by_extended_resource( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.list_by_extended_resource.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.list_by_extended_resource.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -112,17 +266,19 @@ def list_by_extended_resource( return cls(pipeline_response, deserialized, {}) return deserialized + list_by_extended_resource.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments'} # type: ignore + + @distributed_trace def get( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ServerVulnerabilityAssessment" + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any + ) -> "_models.ServerVulnerabilityAssessment": """Gets a server vulnerability assessment onboarding statuses on a given resource. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -144,31 +300,19 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - server_vulnerability_assessment = "default" - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -182,17 +326,19 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore + + @distributed_trace def create_or_update( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ServerVulnerabilityAssessment" + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any + ) -> "_models.ServerVulnerabilityAssessment": """Creating a server vulnerability assessment on a resource, which will onboard a resource for having a vulnerability assessment on it. @@ -215,31 +361,19 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - server_vulnerability_assessment = "default" - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_create_or_update_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.put(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -253,47 +387,36 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore + def _delete_initial( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any + ) -> None: cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - server_vulnerability_assessment = "default" - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self._delete_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request_initial( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self._delete_initial.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -306,15 +429,16 @@ def _delete_initial( _delete_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore + + @distributed_trace def begin_delete( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> LROPoller[None] + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any + ) -> LROPoller[None]: """Removing server vulnerability assessment from a resource. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -328,15 +452,17 @@ def begin_delete( :type resource_name: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. - Pass in False for this operation to not poll, or pass in your own initialized polling object for a personal polling strategy. + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: + :raises: ~azure.core.exceptions.HttpResponseError """ - polling = kwargs.pop('polling', True) # type: Union[bool, PollingMethod] + polling = kwargs.pop('polling', True) # type: Union[bool, azure.core.polling.PollingMethod] cls = kwargs.pop('cls', None) # type: ClsType[None] lro_delay = kwargs.pop( 'polling_interval', @@ -352,25 +478,14 @@ def begin_delete( cls=lambda x,y,z: x, **kwargs ) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) def get_long_running_output(pipeline_response): if cls: return cls(pipeline_response, None, {}) - server_vulnerability_assessment = "default" - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'serverVulnerabilityAssessment': self._serialize.url("server_vulnerability_assessment", server_vulnerability_assessment, 'str'), - } - if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'location'}, path_format_arguments=path_format_arguments, **kwargs) + if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'location'}, **kwargs) elif polling is False: polling_method = NoPolling() else: polling_method = polling if cont_token: @@ -382,4 +497,5 @@ def get_long_running_output(pipeline_response): ) else: return LROPoller(self._client, raw_result, get_long_running_output, polling_method) + begin_delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/serverVulnerabilityAssessments/{serverVulnerabilityAssessment}'} # type: ignore diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_settings_operations.py index 6d8ccf6bd0442..c091f97cfc921 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_settings_operations.py @@ -5,23 +5,132 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + setting_name: Union[str, "_models.Enum73"], + **kwargs: Any +) -> HttpRequest: + api_version = "2021-07-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings/{settingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "settingName": _SERIALIZER.url("setting_name", setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_request( + subscription_id: str, + setting_name: Union[str, "_models.Enum73"], + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2021-07-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings/{settingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "settingName": _SERIALIZER.url("setting_name", setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class SettingsOperations(object): """SettingsOperations operations. @@ -45,11 +154,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SettingsList"] + **kwargs: Any + ) -> Iterable["_models.SettingsList"]: """Settings about different configurations in security center. :keyword callable cls: A custom type or function that will be passed the direct response @@ -62,34 +171,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-07-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SettingsList', pipeline_response) + deserialized = self._deserialize("SettingsList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,21 +211,22 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings'} # type: ignore + @distributed_trace def get( self, - setting_name, # type: Union[str, "_models.Enum69"] - **kwargs # type: Any - ): - # type: (...) -> "_models.Setting" + setting_name: Union[str, "_models.Enum73"], + **kwargs: Any + ) -> "_models.Setting": """Settings of different configurations in security center. :param setting_name: The name of the setting. - :type setting_name: str or ~azure.mgmt.security.models.Enum69 + :type setting_name: str or ~azure.mgmt.security.models.Enum73 :keyword callable cls: A custom type or function that will be passed the direct response :return: Setting, or the result of cls(response) :rtype: ~azure.mgmt.security.models.Setting @@ -132,26 +237,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-07-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -165,19 +260,21 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings/{settingName}'} # type: ignore + + @distributed_trace def update( self, - setting_name, # type: Union[str, "_models.Enum69"] - setting, # type: "_models.Setting" - **kwargs # type: Any - ): - # type: (...) -> "_models.Setting" + setting_name: Union[str, "_models.Enum73"], + setting: "_models.Setting", + **kwargs: Any + ) -> "_models.Setting": """updating settings about different configurations in security center. :param setting_name: The name of the setting. - :type setting_name: str or ~azure.mgmt.security.models.Enum69 + :type setting_name: str or ~azure.mgmt.security.models.Enum73 :param setting: Setting object. :type setting: ~azure.mgmt.security.models.Setting :keyword callable cls: A custom type or function that will be passed the direct response @@ -190,31 +287,21 @@ def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-07-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'settingName': self._serialize.url("setting_name", setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(setting, 'Setting') + + request = build_update_request( + subscription_id=self._config.subscription_id, + setting_name=setting_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(setting, 'Setting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -228,4 +315,6 @@ def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/settings/{settingName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_software_inventories_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_software_inventories_operations.py index 410d6ff4e69f5..698d3b36a54c7 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_software_inventories_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_software_inventories_operations.py @@ -5,23 +5,136 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_by_extended_resource_request( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-05-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/softwareInventories') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_subscription_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-05-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/softwareInventories') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + software_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2021-05-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/softwareInventories/{softwareName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "resourceNamespace": _SERIALIZER.url("resource_namespace", resource_namespace, 'str'), + "resourceType": _SERIALIZER.url("resource_type", resource_type, 'str'), + "resourceName": _SERIALIZER.url("resource_name", resource_name, 'str'), + "softwareName": _SERIALIZER.url("software_name", software_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SoftwareInventoriesOperations(object): """SoftwareInventoriesOperations operations. @@ -45,15 +158,15 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list_by_extended_resource( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SoftwaresList"] + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + **kwargs: Any + ) -> Iterable["_models.SoftwaresList"]: """Gets the software inventory of the virtual machine. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -75,38 +188,37 @@ def list_by_extended_resource( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-05-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_extended_resource.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=self.list_by_extended_resource.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_extended_resource_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SoftwaresList', pipeline_response) + deserialized = self._deserialize("SoftwaresList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -124,16 +236,17 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_extended_resource.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/softwareInventories'} # type: ignore + @distributed_trace def list_by_subscription( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SoftwaresList"] + **kwargs: Any + ) -> Iterable["_models.SoftwaresList"]: """Gets the software inventory of all virtual machines in the subscriptions. :keyword callable cls: A custom type or function that will be passed the direct response @@ -146,34 +259,29 @@ def list_by_subscription( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-05-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_subscription.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=self.list_by_subscription.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_subscription_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SoftwaresList', pipeline_response) + deserialized = self._deserialize("SoftwaresList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -191,21 +299,22 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_subscription.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/softwareInventories'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - resource_namespace, # type: str - resource_type, # type: str - resource_name, # type: str - software_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Software" + resource_group_name: str, + resource_namespace: str, + resource_type: str, + resource_name: str, + software_name: str, + **kwargs: Any + ) -> "_models.Software": """Gets a single software data of the virtual machine. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -229,30 +338,20 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2021-05-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'resourceNamespace': self._serialize.url("resource_namespace", resource_namespace, 'str'), - 'resourceType': self._serialize.url("resource_type", resource_type, 'str'), - 'resourceName': self._serialize.url("resource_name", resource_name, 'str'), - 'softwareName': self._serialize.url("software_name", software_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + resource_namespace=resource_namespace, + resource_type=resource_type, + resource_name=resource_name, + software_name=software_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -266,4 +365,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceNamespace}/{resourceType}/{resourceName}/providers/Microsoft.Security/softwareInventories/{softwareName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_baseline_rules_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_baseline_rules_operations.py index 7830d316df6fd..8b9990c11f7ae 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_baseline_rules_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_baseline_rules_operations.py @@ -5,22 +5,217 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_create_or_update_request( + rule_id: str, + resource_id: str, + *, + workspace_id: str, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}') + path_format_arguments = { + "ruleId": _SERIALIZER.url("rule_id", rule_id, 'str'), + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_get_request( + rule_id: str, + resource_id: str, + *, + workspace_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}') + path_format_arguments = { + "ruleId": _SERIALIZER.url("rule_id", rule_id, 'str'), + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_delete_request( + rule_id: str, + resource_id: str, + *, + workspace_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}') + path_format_arguments = { + "ruleId": _SERIALIZER.url("rule_id", rule_id, 'str'), + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_request( + resource_id: str, + *, + workspace_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_add_request( + resource_id: str, + *, + workspace_id: str, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) class SqlVulnerabilityAssessmentBaselineRulesOperations(object): """SqlVulnerabilityAssessmentBaselineRulesOperations operations. @@ -44,15 +239,15 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def create_or_update( self, - rule_id, # type: str - workspace_id, # type: str - resource_id, # type: str - body=None, # type: Optional["_models.RuleResultsInput"] - **kwargs # type: Any - ): - # type: (...) -> "_models.RuleResults" + rule_id: str, + workspace_id: str, + resource_id: str, + body: Optional["_models.RuleResultsInput"] = None, + **kwargs: Any + ) -> "_models.RuleResults": """Creates a Baseline for a rule in a database. Will overwrite any previously existing results. Creates a Baseline for a rule in a database. Will overwrite any previously existing results. @@ -75,35 +270,25 @@ def create_or_update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'ruleId': self._serialize.url("rule_id", rule_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] if body is not None: - body_content = self._serialize.body(body, 'RuleResultsInput') + _json = self._serialize.body(body, 'RuleResultsInput') else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) + _json = None + + request = build_create_or_update_request( + rule_id=rule_id, + resource_id=resource_id, + content_type=content_type, + workspace_id=workspace_id, + json=_json, + template_url=self.create_or_update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -117,16 +302,18 @@ def create_or_update( return cls(pipeline_response, deserialized, {}) return deserialized + create_or_update.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'} # type: ignore + + @distributed_trace def get( self, - rule_id, # type: str - workspace_id, # type: str - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.RuleResults" + rule_id: str, + workspace_id: str, + resource_id: str, + **kwargs: Any + ) -> "_models.RuleResults": """Gets the results for a given rule in the Baseline. Gets the results for a given rule in the Baseline. @@ -147,27 +334,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'ruleId': self._serialize.url("rule_id", rule_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + rule_id=rule_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -181,16 +358,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'} # type: ignore + + @distributed_trace def delete( self, - rule_id, # type: str - workspace_id, # type: str - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + rule_id: str, + workspace_id: str, + resource_id: str, + **kwargs: Any + ) -> None: """Deletes a rule from the Baseline of a given database. Deletes a rule from the Baseline of a given database. @@ -211,27 +390,17 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'ruleId': self._serialize.url("rule_id", rule_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_delete_request( + rule_id=rule_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -244,13 +413,14 @@ def delete( delete.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules/{ruleId}'} # type: ignore + + @distributed_trace def list( self, - workspace_id, # type: str - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.RulesResults" + workspace_id: str, + resource_id: str, + **kwargs: Any + ) -> "_models.RulesResults": """Gets the results for all rules in the Baseline. Gets the results for all rules in the Baseline. @@ -269,26 +439,16 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -302,16 +462,18 @@ def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules'} # type: ignore + + @distributed_trace def add( self, - workspace_id, # type: str - resource_id, # type: str - body=None, # type: Optional["_models.RulesResultsInput"] - **kwargs # type: Any - ): - # type: (...) -> "_models.RulesResults" + workspace_id: str, + resource_id: str, + body: Optional["_models.RulesResultsInput"] = None, + **kwargs: Any + ) -> "_models.RulesResults": """Add a list of baseline rules. Will overwrite any previously existing results (for all rules). Add a list of baseline rules. Will overwrite any previously existing results (for all rules). @@ -332,34 +494,24 @@ def add( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.add.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - body_content_kwargs = {} # type: Dict[str, Any] if body is not None: - body_content = self._serialize.body(body, 'RulesResultsInput') + _json = self._serialize.body(body, 'RulesResultsInput') else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) + _json = None + + request = build_add_request( + resource_id=resource_id, + content_type=content_type, + workspace_id=workspace_id, + json=_json, + template_url=self.add.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -373,4 +525,6 @@ def add( return cls(pipeline_response, deserialized, {}) return deserialized + add.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/baselineRules'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scan_results_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scan_results_operations.py index b768ddc1fa64e..e13bd7e054af8 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scan_results_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scan_results_operations.py @@ -5,22 +5,98 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_get_request( + scan_id: str, + scan_result_id: str, + resource_id: str, + *, + workspace_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}/scanResults/{scanResultId}') + path_format_arguments = { + "scanId": _SERIALIZER.url("scan_id", scan_id, 'str'), + "scanResultId": _SERIALIZER.url("scan_result_id", scan_result_id, 'str'), + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_request( + scan_id: str, + resource_id: str, + *, + workspace_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}/scanResults') + path_format_arguments = { + "scanId": _SERIALIZER.url("scan_id", scan_id, 'str'), + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SqlVulnerabilityAssessmentScanResultsOperations(object): """SqlVulnerabilityAssessmentScanResultsOperations operations. @@ -44,15 +120,15 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def get( self, - scan_id, # type: str - scan_result_id, # type: str - workspace_id, # type: str - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ScanResult" + scan_id: str, + scan_result_id: str, + workspace_id: str, + resource_id: str, + **kwargs: Any + ) -> "_models.ScanResult": """Gets the scan results of a single rule in a scan record. Gets the scan results of a single rule in a scan record. @@ -75,28 +151,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scanId': self._serialize.url("scan_id", scan_id, 'str'), - 'scanResultId': self._serialize.url("scan_result_id", scan_result_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_get_request( + scan_id=scan_id, + scan_result_id=scan_result_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -110,16 +176,18 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}/scanResults/{scanResultId}'} # type: ignore + + @distributed_trace def list( self, - scan_id, # type: str - workspace_id, # type: str - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.ScanResults" + scan_id: str, + workspace_id: str, + resource_id: str, + **kwargs: Any + ) -> "_models.ScanResults": """Gets a list of scan results for a single scan record. Gets a list of scan results for a single scan record. @@ -140,27 +208,17 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scanId': self._serialize.url("scan_id", scan_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + scan_id=scan_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -174,4 +232,6 @@ def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}/scanResults'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scans_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scans_operations.py index 715745a22c636..66e832b4f8dd4 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scans_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sql_vulnerability_assessment_scans_operations.py @@ -5,22 +5,94 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_get_request( + scan_id: str, + resource_id: str, + *, + workspace_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}') + path_format_arguments = { + "scanId": _SERIALIZER.url("scan_id", scan_id, 'str'), + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_request( + resource_id: str, + *, + workspace_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-07-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans') + path_format_arguments = { + "resourceId": _SERIALIZER.url("resource_id", resource_id, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['workspaceId'] = _SERIALIZER.query("workspace_id", workspace_id, 'str') + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SqlVulnerabilityAssessmentScansOperations(object): """SqlVulnerabilityAssessmentScansOperations operations. @@ -44,14 +116,14 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def get( self, - scan_id, # type: str - workspace_id, # type: str - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Scan" + scan_id: str, + workspace_id: str, + resource_id: str, + **kwargs: Any + ) -> "_models.Scan": """Gets the scan details of a single scan record. Gets the scan details of a single scan record. @@ -72,27 +144,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scanId': self._serialize.url("scan_id", scan_id, 'str'), - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + + request = build_get_request( + scan_id=scan_id, + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -106,15 +168,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans/{scanId}'} # type: ignore + + @distributed_trace def list( self, - workspace_id, # type: str - resource_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.Scans" + workspace_id: str, + resource_id: str, + **kwargs: Any + ) -> "_models.Scans": """Gets a list of scan records. Gets a list of scan records. @@ -133,26 +197,16 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-07-01-preview" - accept = "application/json" - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'resourceId': self._serialize.url("resource_id", resource_id, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['workspaceId'] = self._serialize.query("workspace_id", workspace_id, 'str') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_list_request( + resource_id=resource_id, + workspace_id=workspace_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -166,4 +220,6 @@ def list( return cls(pipeline_response, deserialized, {}) return deserialized + list.metadata = {'url': '/{resourceId}/providers/Microsoft.Security/sqlVulnerabilityAssessments/default/scans'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sub_assessments_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sub_assessments_operations.py index 98bf219f2712a..c9f4a4e02ab72 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sub_assessments_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_sub_assessments_operations.py @@ -5,23 +5,124 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_all_request( + scope: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/subAssessments') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_request( + scope: str, + assessment_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/assessments/{assessmentName}/subAssessments') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + "assessmentName": _SERIALIZER.url("assessment_name", assessment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + scope: str, + assessment_name: str, + sub_assessment_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2019-01-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/{scope}/providers/Microsoft.Security/assessments/{assessmentName}/subAssessments/{subAssessmentName}') + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, 'str', skip_quote=True), + "assessmentName": _SERIALIZER.url("assessment_name", assessment_name, 'str'), + "subAssessmentName": _SERIALIZER.url("sub_assessment_name", sub_assessment_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class SubAssessmentsOperations(object): """SubAssessmentsOperations operations. @@ -45,12 +146,12 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list_all( self, - scope, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecuritySubAssessmentList"] + scope: str, + **kwargs: Any + ) -> Iterable["_models.SecuritySubAssessmentList"]: """Get security sub-assessments on all your scanned resources inside a subscription scope. :param scope: Scope of the query, can be subscription @@ -58,7 +159,8 @@ def list_all( (/providers/Microsoft.Management/managementGroups/mgName). :type scope: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecuritySubAssessmentList or the result of cls(response) + :return: An iterator like instance of either SecuritySubAssessmentList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecuritySubAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -67,34 +169,29 @@ def list_all( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_all.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_all_request( + scope=scope, + template_url=self.list_all.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_all_request( + scope=scope, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecuritySubAssessmentList', pipeline_response) + deserialized = self._deserialize("SecuritySubAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,18 +209,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_all.metadata = {'url': '/{scope}/providers/Microsoft.Security/subAssessments'} # type: ignore + @distributed_trace def list( self, - scope, # type: str - assessment_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecuritySubAssessmentList"] + scope: str, + assessment_name: str, + **kwargs: Any + ) -> Iterable["_models.SecuritySubAssessmentList"]: """Get security sub-assessments on all your scanned resources inside a scope. :param scope: Scope of the query, can be subscription @@ -133,7 +231,8 @@ def list( :param assessment_name: The Assessment Key - Unique key for the assessment type. :type assessment_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either SecuritySubAssessmentList or the result of cls(response) + :return: An iterator like instance of either SecuritySubAssessmentList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.SecuritySubAssessmentList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -142,35 +241,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + assessment_name=assessment_name, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + scope=scope, + assessment_name=assessment_name, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecuritySubAssessmentList', pipeline_response) + deserialized = self._deserialize("SecuritySubAssessmentList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -188,19 +283,20 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/{scope}/providers/Microsoft.Security/assessments/{assessmentName}/subAssessments'} # type: ignore + @distributed_trace def get( self, - scope, # type: str - assessment_name, # type: str - sub_assessment_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecuritySubAssessment" + scope: str, + assessment_name: str, + sub_assessment_name: str, + **kwargs: Any + ) -> "_models.SecuritySubAssessment": """Get a security sub-assessment on your scanned resource. :param scope: Scope of the query, can be subscription @@ -221,27 +317,17 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2019-01-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'scope': self._serialize.url("scope", scope, 'str', skip_quote=True), - 'assessmentName': self._serialize.url("assessment_name", assessment_name, 'str'), - 'subAssessmentName': self._serialize.url("sub_assessment_name", sub_assessment_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + scope=scope, + assessment_name=assessment_name, + sub_assessment_name=sub_assessment_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -255,4 +341,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/{scope}/providers/Microsoft.Security/assessments/{assessmentName}/subAssessments/{subAssessmentName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_tasks_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_tasks_operations.py index 387baf6aafbd7..426a8c1c951ed 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_tasks_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_tasks_operations.py @@ -5,23 +5,284 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/tasks') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_home_region_request( + subscription_id: str, + asc_location: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_subscription_level_task_request( + subscription_id: str, + asc_location: str, + task_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "taskName": _SERIALIZER.url("task_name", task_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_subscription_level_task_state_request( + subscription_id: str, + asc_location: str, + task_name: str, + task_update_action_type: Union[str, "_models.Enum15"], + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}/{taskUpdateActionType}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "taskName": _SERIALIZER.url("task_name", task_name, 'str'), + "taskUpdateActionType": _SERIALIZER.url("task_update_action_type", task_update_action_type, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_resource_group_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + *, + filter: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + if filter is not None: + query_parameters['$filter'] = _SERIALIZER.query("filter", filter, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_resource_group_level_task_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + task_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "taskName": _SERIALIZER.url("task_name", task_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_update_resource_group_level_task_state_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + task_name: str, + task_update_action_type: Union[str, "_models.Enum15"], + **kwargs: Any +) -> HttpRequest: + api_version = "2015-06-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}/{taskUpdateActionType}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "taskName": _SERIALIZER.url("task_name", task_name, 'str'), + "taskUpdateActionType": _SERIALIZER.url("task_update_action_type", task_update_action_type, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class TasksOperations(object): """TasksOperations operations. @@ -45,12 +306,12 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecurityTaskList"] + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.SecurityTaskList"]: """Recommended tasks that will help improve the security of the subscription proactively. :param filter: OData filter. Optional. @@ -65,36 +326,31 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityTaskList', pipeline_response) + deserialized = self._deserialize("SecurityTaskList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -112,17 +368,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/tasks'} # type: ignore + @distributed_trace def list_by_home_region( self, - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecurityTaskList"] + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.SecurityTaskList"]: """Recommended tasks that will help improve the security of the subscription proactively. :param filter: OData filter. Optional. @@ -137,37 +394,33 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + filter=filter, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityTaskList', pipeline_response) + deserialized = self._deserialize("SecurityTaskList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -185,17 +438,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks'} # type: ignore + @distributed_trace def get_subscription_level_task( self, - task_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityTask" + task_name: str, + **kwargs: Any + ) -> "_models.SecurityTask": """Recommended tasks that will help improve the security of the subscription proactively. :param task_name: Name of the task object, will be a GUID. @@ -210,27 +464,17 @@ def get_subscription_level_task( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.get_subscription_level_task.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_subscription_level_task_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + task_name=task_name, + template_url=self.get_subscription_level_task.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -244,21 +488,23 @@ def get_subscription_level_task( return cls(pipeline_response, deserialized, {}) return deserialized + get_subscription_level_task.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}'} # type: ignore + + @distributed_trace def update_subscription_level_task_state( self, - task_name, # type: str - task_update_action_type, # type: Union[str, "_models.Enum13"] - **kwargs # type: Any - ): - # type: (...) -> None + task_name: str, + task_update_action_type: Union[str, "_models.Enum15"], + **kwargs: Any + ) -> None: """Recommended tasks that will help improve the security of the subscription proactively. :param task_name: Name of the task object, will be a GUID. :type task_name: str :param task_update_action_type: Type of the action to do on the task. - :type task_update_action_type: str or ~azure.mgmt.security.models.Enum13 + :type task_update_action_type: str or ~azure.mgmt.security.models.Enum15 :keyword callable cls: A custom type or function that will be passed the direct response :return: None, or the result of cls(response) :rtype: None @@ -269,28 +515,18 @@ def update_subscription_level_task_state( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.update_subscription_level_task_state.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - 'taskUpdateActionType': self._serialize.url("task_update_action_type", task_update_action_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_subscription_level_task_state_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + task_name=task_name, + task_update_action_type=task_update_action_type, + template_url=self.update_subscription_level_task_state.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -303,13 +539,14 @@ def update_subscription_level_task_state( update_subscription_level_task_state.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}/{taskUpdateActionType}'} # type: ignore + + @distributed_trace def list_by_resource_group( self, - resource_group_name, # type: str - filter=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.SecurityTaskList"] + resource_group_name: str, + filter: Optional[str] = None, + **kwargs: Any + ) -> Iterable["_models.SecurityTaskList"]: """Recommended tasks that will help improve the security of the subscription proactively. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -327,38 +564,35 @@ def list_by_resource_group( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + filter=filter, + template_url=self.list_by_resource_group.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + filter=filter, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('SecurityTaskList', pipeline_response) + deserialized = self._deserialize("SecurityTaskList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -376,18 +610,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks'} # type: ignore + @distributed_trace def get_resource_group_level_task( self, - resource_group_name, # type: str - task_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.SecurityTask" + resource_group_name: str, + task_name: str, + **kwargs: Any + ) -> "_models.SecurityTask": """Recommended tasks that will help improve the security of the subscription proactively. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -405,28 +640,18 @@ def get_resource_group_level_task( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.get_resource_group_level_task.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_resource_group_level_task_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + task_name=task_name, + template_url=self.get_resource_group_level_task.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -440,16 +665,18 @@ def get_resource_group_level_task( return cls(pipeline_response, deserialized, {}) return deserialized + get_resource_group_level_task.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}'} # type: ignore + + @distributed_trace def update_resource_group_level_task_state( self, - resource_group_name, # type: str - task_name, # type: str - task_update_action_type, # type: Union[str, "_models.Enum13"] - **kwargs # type: Any - ): - # type: (...) -> None + resource_group_name: str, + task_name: str, + task_update_action_type: Union[str, "_models.Enum15"], + **kwargs: Any + ) -> None: """Recommended tasks that will help improve the security of the subscription proactively. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -458,7 +685,7 @@ def update_resource_group_level_task_state( :param task_name: Name of the task object, will be a GUID. :type task_name: str :param task_update_action_type: Type of the action to do on the task. - :type task_update_action_type: str or ~azure.mgmt.security.models.Enum13 + :type task_update_action_type: str or ~azure.mgmt.security.models.Enum15 :keyword callable cls: A custom type or function that will be passed the direct response :return: None, or the result of cls(response) :rtype: None @@ -469,29 +696,19 @@ def update_resource_group_level_task_state( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2015-06-01-preview" - accept = "application/json" - - # Construct URL - url = self.update_resource_group_level_task_state.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'taskName': self._serialize.url("task_name", task_name, 'str'), - 'taskUpdateActionType': self._serialize.url("task_update_action_type", task_update_action_type, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_update_resource_group_level_task_state_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + task_name=task_name, + task_update_action_type=task_update_action_type, + template_url=self.update_resource_group_level_task_state.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.post(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -503,3 +720,4 @@ def update_resource_group_level_task_state( return cls(pipeline_response, None, {}) update_resource_group_level_task_state.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/tasks/{taskName}/{taskUpdateActionType}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_topology_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_topology_operations.py index b016e0dfe85d4..7af88296065eb 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_topology_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_topology_operations.py @@ -5,23 +5,126 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/topologies') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_list_by_home_region_request( + subscription_id: str, + asc_location: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/topologies') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + resource_group_name: str, + asc_location: str, + topology_resource_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2020-01-01" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/topologies/{topologyResourceName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + "ascLocation": _SERIALIZER.url("asc_location", asc_location, 'str'), + "topologyResourceName": _SERIALIZER.url("topology_resource_name", topology_resource_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class TopologyOperations(object): """TopologyOperations operations. @@ -45,11 +148,11 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.TopologyList"] + **kwargs: Any + ) -> Iterable["_models.TopologyList"]: """Gets a list that allows to build a topology view of a subscription. :keyword callable cls: A custom type or function that will be passed the direct response @@ -62,34 +165,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('TopologyList', pipeline_response) + deserialized = self._deserialize("TopologyList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -107,16 +205,17 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/topologies'} # type: ignore + @distributed_trace def list_by_home_region( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.TopologyList"] + **kwargs: Any + ) -> Iterable["_models.TopologyList"]: """Gets a list that allows to build a topology view of a subscription and location. :keyword callable cls: A custom type or function that will be passed the direct response @@ -129,35 +228,31 @@ def list_by_home_region( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_home_region.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=self.list_by_home_region.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_home_region_request( + subscription_id=self._config.subscription_id, + asc_location=self._config.asc_location, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('TopologyList', pipeline_response) + deserialized = self._deserialize("TopologyList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -175,18 +270,19 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list_by_home_region.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/locations/{ascLocation}/topologies'} # type: ignore + @distributed_trace def get( self, - resource_group_name, # type: str - topology_resource_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.TopologyResource" + resource_group_name: str, + topology_resource_name: str, + **kwargs: Any + ) -> "_models.TopologyResource": """Gets a specific topology component. :param resource_group_name: The name of the resource group within the user's subscription. The @@ -204,28 +300,18 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2020-01-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), - 'ascLocation': self._serialize.url("self._config.asc_location", self._config.asc_location, 'str'), - 'topologyResourceName': self._serialize.url("topology_resource_name", topology_resource_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + resource_group_name=resource_group_name, + asc_location=self._config.asc_location, + topology_resource_name=topology_resource_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -239,4 +325,6 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Security/locations/{ascLocation}/topologies/{topologyResourceName}'} # type: ignore + diff --git a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_workspace_settings_operations.py b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_workspace_settings_operations.py index 44fdc6da27c8f..e373177745acd 100644 --- a/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_workspace_settings_operations.py +++ b/sdk/security/azure-mgmt-security/azure/mgmt/security/operations/_workspace_settings_operations.py @@ -5,23 +5,207 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import functools +from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar import warnings from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace from azure.mgmt.core.exceptions import ARMErrorFormat +from msrest import Serializer from .. import models as _models - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar - - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +from .._vendor import _convert_request, _format_url_section +T = TypeVar('T') +JSONType = Any +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + +def build_list_request( + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_get_request( + subscription_id: str, + workspace_setting_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "workspaceSettingName": _SERIALIZER.url("workspace_setting_name", workspace_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) + + +def build_create_request( + subscription_id: str, + workspace_setting_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "workspaceSettingName": _SERIALIZER.url("workspace_setting_name", workspace_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_update_request( + subscription_id: str, + workspace_setting_name: str, + *, + json: JSONType = None, + content: Any = None, + **kwargs: Any +) -> HttpRequest: + content_type = kwargs.pop('content_type', None) # type: Optional[str] + + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "workspaceSettingName": _SERIALIZER.url("workspace_setting_name", workspace_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + if content_type is not None: + header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PATCH", + url=url, + params=query_parameters, + headers=header_parameters, + json=json, + content=content, + **kwargs + ) + + +def build_delete_request( + subscription_id: str, + workspace_setting_name: str, + **kwargs: Any +) -> HttpRequest: + api_version = "2017-08-01-preview" + accept = "application/json" + # Construct URL + url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}') + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), + "workspaceSettingName": _SERIALIZER.url("workspace_setting_name", workspace_setting_name, 'str'), + } + + url = _format_url_section(url, **path_format_arguments) + + # Construct parameters + query_parameters = kwargs.pop("params", {}) # type: Dict[str, Any] + query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + + # Construct headers + header_parameters = kwargs.pop("headers", {}) # type: Dict[str, Any] + header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="DELETE", + url=url, + params=query_parameters, + headers=header_parameters, + **kwargs + ) class WorkspaceSettingsOperations(object): """WorkspaceSettingsOperations operations. @@ -45,16 +229,17 @@ def __init__(self, client, config, serializer, deserializer): self._deserialize = deserializer self._config = config + @distributed_trace def list( self, - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.WorkspaceSettingList"] + **kwargs: Any + ) -> Iterable["_models.WorkspaceSettingList"]: """Settings about where we should store your security data and logs. If the result is empty, it means that no custom-workspace configuration was set. :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either WorkspaceSettingList or the result of cls(response) + :return: An iterator like instance of either WorkspaceSettingList or the result of + cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.security.models.WorkspaceSettingList] :raises: ~azure.core.exceptions.HttpResponseError """ @@ -63,34 +248,29 @@ def list( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=self.list.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + template_url=next_link, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('WorkspaceSettingList', pipeline_response) + deserialized = self._deserialize("WorkspaceSettingList", pipeline_response) list_of_elem = deserialized.value if cls: list_of_elem = cls(list_of_elem) @@ -108,17 +288,18 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged( get_next, extract_data ) list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings'} # type: ignore + @distributed_trace def get( self, - workspace_setting_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.WorkspaceSetting" + workspace_setting_name: str, + **kwargs: Any + ) -> "_models.WorkspaceSetting": """Settings about where we should store your security data and logs. If the result is empty, it means that no custom-workspace configuration was set. @@ -134,26 +315,16 @@ def get( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_get_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + template_url=self.get.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.get(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -167,15 +338,17 @@ def get( return cls(pipeline_response, deserialized, {}) return deserialized + get.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore + + @distributed_trace def create( self, - workspace_setting_name, # type: str - workspace_setting, # type: "_models.WorkspaceSetting" - **kwargs # type: Any - ): - # type: (...) -> "_models.WorkspaceSetting" + workspace_setting_name: str, + workspace_setting: "_models.WorkspaceSetting", + **kwargs: Any + ) -> "_models.WorkspaceSetting": """creating settings about where we should store your security data and logs. :param workspace_setting_name: Name of the security setting. @@ -192,31 +365,21 @@ def create( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _json = self._serialize.body(workspace_setting, 'WorkspaceSetting') + + request = build_create_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + content_type=content_type, + json=_json, + template_url=self.create.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(workspace_setting, 'WorkspaceSetting') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -230,15 +393,17 @@ def create( return cls(pipeline_response, deserialized, {}) return deserialized + create.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore + + @distributed_trace def update( self, - workspace_setting_name, # type: str - workspace_setting, # type: "_models.WorkspaceSetting" - **kwargs # type: Any - ): - # type: (...) -> "_models.WorkspaceSetting" + workspace_setting_name: str, + workspace_setting: "_models.WorkspaceSetting", + **kwargs: Any + ) -> "_models.WorkspaceSetting": """Settings about where we should store your security data and logs. :param workspace_setting_name: Name of the security setting. @@ -255,31 +420,21 @@ def update( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + content_type = kwargs.pop('content_type', "application/json") # type: Optional[str] + + _json = self._serialize.body(workspace_setting, 'WorkspaceSetting') - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + request = build_update_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + content_type=content_type, + json=_json, + template_url=self.update.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(workspace_setting, 'WorkspaceSetting') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -293,14 +448,16 @@ def update( return cls(pipeline_response, deserialized, {}) return deserialized + update.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore + + @distributed_trace def delete( self, - workspace_setting_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + workspace_setting_name: str, + **kwargs: Any + ) -> None: """Deletes the custom workspace settings for this subscription. new VMs will report to the default workspace. @@ -316,26 +473,16 @@ def delete( 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError } error_map.update(kwargs.pop('error_map', {})) - api_version = "2017-08-01-preview" - accept = "application/json" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', pattern=r'^[0-9A-Fa-f]{8}-([0-9A-Fa-f]{4}-){3}[0-9A-Fa-f]{12}$'), - 'workspaceSettingName': self._serialize.url("workspace_setting_name", workspace_setting_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + + request = build_delete_request( + subscription_id=self._config.subscription_id, + workspace_setting_name=workspace_setting_name, + template_url=self.delete.metadata['url'], + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - request = self._client.delete(url, query_parameters, header_parameters) pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response @@ -347,3 +494,4 @@ def delete( return cls(pipeline_response, None, {}) delete.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.Security/workspaceSettings/{workspaceSettingName}'} # type: ignore +