From 98cd7680c6151e85c351971bebb1b79b01d48cb4 Mon Sep 17 00:00:00 2001 From: SDKAuto Date: Tue, 13 May 2025 01:20:52 +0000 Subject: [PATCH] CodeGen from PR 34324 in Azure/azure-rest-api-specs Merge 2e8f925516048a901587c68abc3e66477e371151 into c64457b3dcc103d93c378ca1ec58fbf683a7f163 --- sdk/elastic/azure-mgmt-elastic/CHANGELOG.md | 4 + sdk/elastic/azure-mgmt-elastic/README.md | 6 +- sdk/elastic/azure-mgmt-elastic/_meta.json | 11 +- .../apiview-properties.json | 172 + .../azure/mgmt/elastic/__init__.py | 16 +- .../azure/mgmt/elastic/_client.py | 164 + .../azure/mgmt/elastic/_configuration.py | 27 +- .../azure/mgmt/elastic/_microsoft_elastic.py | 239 - .../azure/mgmt/elastic/_patch.py | 9 +- .../azure/mgmt/elastic/_utils/__init__.py | 6 + .../azure/mgmt/elastic/_utils/model_base.py | 1232 ++++ .../serialization.py} | 566 +- .../azure/mgmt/elastic/_version.py | 4 +- .../azure/mgmt/elastic/aio/__init__.py | 16 +- .../azure/mgmt/elastic/aio/_client.py | 168 + .../azure/mgmt/elastic/aio/_configuration.py | 27 +- .../mgmt/elastic/aio/_microsoft_elastic.py | 241 - .../azure/mgmt/elastic/aio/_patch.py | 9 +- .../mgmt/elastic/aio/operations/__init__.py | 70 +- .../_all_traffic_filters_operations.py | 116 - .../_associate_traffic_filter_operations.py | 174 - .../operations/_billing_info_operations.py | 114 - ..._connected_partner_resources_operations.py | 148 - ...eate_and_associate_ip_filter_operations.py | 188 - ...eate_and_associate_pl_filter_operations.py | 194 - .../operations/_deployment_info_operations.py | 116 - ...ch_and_delete_traffic_filter_operations.py | 115 - .../_detach_traffic_filter_operations.py | 174 - .../_elastic_versions_operations.py | 140 - .../operations/_external_user_operations.py | 203 - ...t_associated_traffic_filters_operations.py | 116 - .../aio/operations/_monitor_operations.py | 256 - .../_monitored_resources_operations.py | 144 - .../_monitored_subscriptions_operations.py | 773 -- .../aio/operations/_monitors_operations.py | 818 --- .../aio/operations/_open_ai_operations.py | 497 -- .../elastic/aio/operations/_operations.py | 5193 ++++++++++++- .../operations/_organizations_operations.py | 445 -- .../mgmt/elastic/aio/operations/_patch.py | 9 +- .../aio/operations/_tag_rules_operations.py | 486 -- .../operations/_traffic_filters_operations.py | 115 - .../_upgradable_versions_operations.py | 116 - .../operations/_vm_collection_operations.py | 193 - .../aio/operations/_vm_host_operations.py | 142 - .../operations/_vm_ingestion_operations.py | 116 - .../azure/mgmt/elastic/models/__init__.py | 197 +- ...{_microsoft_elastic_enums.py => _enums.py} | 35 +- .../azure/mgmt/elastic/models/_models.py | 2609 +++++++ .../azure/mgmt/elastic/models/_models_py3.py | 2568 ------- .../azure/mgmt/elastic/models/_patch.py | 9 +- .../azure/mgmt/elastic/operations/__init__.py | 70 +- .../_all_traffic_filters_operations.py | 148 - .../_associate_traffic_filter_operations.py | 217 - .../operations/_billing_info_operations.py | 148 - ..._connected_partner_resources_operations.py | 182 - ...eate_and_associate_ip_filter_operations.py | 234 - ...eate_and_associate_pl_filter_operations.py | 243 - .../operations/_deployment_info_operations.py | 150 - ...ch_and_delete_traffic_filter_operations.py | 158 - .../_detach_traffic_filter_operations.py | 217 - .../_elastic_versions_operations.py | 167 - .../operations/_external_user_operations.py | 242 - ...t_associated_traffic_filters_operations.py | 148 - .../elastic/operations/_monitor_operations.py | 295 - .../_monitored_resources_operations.py | 176 - .../_monitored_subscriptions_operations.py | 942 --- .../operations/_monitors_operations.py | 1000 --- .../elastic/operations/_open_ai_operations.py | 662 -- .../mgmt/elastic/operations/_operations.py | 6420 ++++++++++++++++- .../operations/_organizations_operations.py | 538 -- .../azure/mgmt/elastic/operations/_patch.py | 9 +- .../operations/_tag_rules_operations.py | 619 -- .../operations/_traffic_filters_operations.py | 158 - .../_upgradable_versions_operations.py | 150 - .../operations/_vm_collection_operations.py | 232 - .../elastic/operations/_vm_host_operations.py | 176 - .../operations/_vm_ingestion_operations.py | 150 - .../all_traffic_filters_list.py | 12 +- .../associate_traffic_filter_update.py | 12 +- .../associated_filters_for_deployment_list.py | 12 +- .../generated_samples/billing_info_get.py | 12 +- .../connected_partner_resources_list.py | 12 +- .../generated_samples/deployment_info_list.py | 12 +- ...detach_and_delete_traffic_filter_delete.py | 12 +- .../detach_traffic_filters_update.py | 12 +- .../elastic_versions_list.py | 12 +- .../generated_samples/external_user_info.py | 12 +- .../ip_traffic_filter_create.py | 13 +- .../generated_samples/monitor_upgrade.py | 12 +- .../monitored_resources_list.py | 12 +- ...monitored_subscriptions_createor_update.py | 43 - .../monitored_subscriptions_delete.py | 10 +- .../monitored_subscriptions_get.py | 10 +- .../monitored_subscriptions_list.py | 10 +- .../monitored_subscriptions_update.py | 43 - .../generated_samples/monitors_create.py | 42 - .../generated_samples/monitors_delete.py | 12 +- .../generated_samples/monitors_get.py | 12 +- .../generated_samples/monitors_list.py | 12 +- .../monitors_list_by_resource_group.py | 12 +- .../generated_samples/monitors_update.py | 42 - .../open_ai_create_or_update.py | 43 - .../generated_samples/open_ai_delete.py | 12 +- .../generated_samples/open_ai_get.py | 12 +- .../generated_samples/open_ai_get_status.py | 12 +- .../generated_samples/open_ai_list.py | 12 +- .../generated_samples/operations_list.py | 8 +- .../organizations_get_api_key.py | 39 - ...t_elastic_to_azure_subscription_mapping.py | 12 +- .../organizations_resubscribe.py | 12 +- .../private_link_traffic_filters_create.py | 13 +- .../tag_rules_create_or_update.py | 43 - .../generated_samples/tag_rules_delete.py | 10 +- .../generated_samples/tag_rules_get.py | 10 +- .../generated_samples/tag_rules_list.py | 10 +- .../traffic_filters_delete.py | 12 +- .../upgradable_versions_details.py | 12 +- .../generated_samples/vm_collection_update.py | 12 +- .../generated_samples/vm_host_list.py | 12 +- .../generated_samples/vm_ingestion_details.py | 12 +- .../generated_tests/conftest.py | 20 +- ...ic_elastic_monitor_resources_operations.py | 355 + ...stic_monitor_resources_operations_async.py | 378 + ...ic_versions_operation_group_operations.py} | 13 +- ...sions_operation_group_operations_async.py} | 13 +- ...stic_monitored_subscriptions_operations.py | 144 + ...onitored_subscriptions_operations_async.py | 151 + ...en_ai_integration_rp_models_operations.py} | 59 +- ...integration_rp_models_operations_async.py} | 59 +- ...erations.py => test_elastic_operations.py} | 14 +- ...nc.py => test_elastic_operations_async.py} | 14 +- ...ganizations_operation_group_operations.py} | 24 +- ...tions_operation_group_operations_async.py} | 24 +- ...y => test_elastic_tag_rules_operations.py} | 58 +- ...est_elastic_tag_rules_operations_async.py} | 66 +- ..._elastic_all_traffic_filters_operations.py | 31 - ...ic_all_traffic_filters_operations_async.py | 32 - ...tic_associate_traffic_filter_operations.py | 31 - ...sociate_traffic_filter_operations_async.py | 34 - ...crosoft_elastic_billing_info_operations.py | 31 - ...t_elastic_billing_info_operations_async.py | 32 - ..._connected_partner_resources_operations.py | 31 - ...cted_partner_resources_operations_async.py | 32 - ...eate_and_associate_ip_filter_operations.py | 31 - ...nd_associate_ip_filter_operations_async.py | 34 - ...eate_and_associate_pl_filter_operations.py | 31 - ...nd_associate_pl_filter_operations_async.py | 34 - ...soft_elastic_deployment_info_operations.py | 31 - ...lastic_deployment_info_operations_async.py | 32 - ...ch_and_delete_traffic_filter_operations.py | 31 - ..._delete_traffic_filter_operations_async.py | 32 - ...lastic_detach_traffic_filter_operations.py | 31 - ..._detach_traffic_filter_operations_async.py | 34 - ...rosoft_elastic_external_user_operations.py | 31 - ..._elastic_external_user_operations_async.py | 32 - ...t_associated_traffic_filters_operations.py | 31 - ...ciated_traffic_filters_operations_async.py | 32 - ...st_microsoft_elastic_monitor_operations.py | 31 - ...rosoft_elastic_monitor_operations_async.py | 34 - ..._elastic_monitored_resources_operations.py | 31 - ...ic_monitored_resources_operations_async.py | 32 - ...stic_monitored_subscriptions_operations.py | 83 - ...onitored_subscriptions_operations_async.py | 90 - ...t_microsoft_elastic_monitors_operations.py | 88 - ...osoft_elastic_monitors_operations_async.py | 95 - ...rosoft_elastic_organizations_operations.py | 51 - ..._elastic_organizations_operations_async.py | 54 - ...soft_elastic_traffic_filters_operations.py | 31 - ...lastic_traffic_filters_operations_async.py | 32 - ...rosoft_elastic_vm_collection_operations.py | 31 - ..._elastic_vm_collection_operations_async.py | 32 - ...st_microsoft_elastic_vm_host_operations.py | 31 - ...rosoft_elastic_vm_host_operations_async.py | 32 - ...crosoft_elastic_vm_ingestion_operations.py | 31 - ...t_elastic_vm_ingestion_operations_async.py | 32 - sdk/elastic/azure-mgmt-elastic/setup.py | 5 +- ..._elastic_monitors_operations_async_test.py | 1 - ...rosoft_elastic_monitors_operations_test.py | 1 - ...microsoft_elastic_operations_async_test.py | 1 - .../test_microsoft_elastic_operations_test.py | 1 - .../azure-mgmt-elastic/tsp-location.yaml | 4 + 181 files changed, 17926 insertions(+), 19091 deletions(-) create mode 100644 sdk/elastic/azure-mgmt-elastic/apiview-properties.json create mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_client.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_microsoft_elastic.py create mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/__init__.py create mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/model_base.py rename sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/{_serialization.py => _utils/serialization.py} (83%) create mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_client.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_microsoft_elastic.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_all_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_associate_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_billing_info_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_connected_partner_resources_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_ip_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_pl_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_deployment_info_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_and_delete_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_elastic_versions_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_external_user_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_list_associated_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitor_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_resources_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_subscriptions_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitors_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_open_ai_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_organizations_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_tag_rules_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_upgradable_versions_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_collection_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_host_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_ingestion_operations.py rename sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/{_microsoft_elastic_enums.py => _enums.py} (72%) create mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models_py3.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_all_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_associate_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_billing_info_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_connected_partner_resources_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_ip_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_pl_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_deployment_info_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_and_delete_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_elastic_versions_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_external_user_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_list_associated_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitor_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_resources_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_subscriptions_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitors_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_open_ai_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_organizations_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_tag_rules_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_upgradable_versions_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_collection_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_host_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_ingestion_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_createor_update.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_update.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_create.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_update.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_create_or_update.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_api_key.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_create_or_update.py create mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations.py create mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations_async.py rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_elastic_versions_operations.py => test_elastic_elastic_versions_operation_group_operations.py} (69%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_elastic_versions_operations_async.py => test_elastic_elastic_versions_operation_group_operations_async.py} (69%) create mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations.py create mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations_async.py rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_open_ai_operations.py => test_elastic_open_ai_integration_rp_models_operations.py} (54%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_open_ai_operations_async.py => test_elastic_open_ai_integration_rp_models_operations_async.py} (54%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_operations.py => test_elastic_operations.py} (69%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_operations_async.py => test_elastic_operations_async.py} (69%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_upgradable_versions_operations.py => test_elastic_organizations_operation_group_operations.py} (52%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_upgradable_versions_operations_async.py => test_elastic_organizations_operation_group_operations_async.py} (52%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_tag_rules_operations.py => test_elastic_tag_rules_operations.py} (61%) rename sdk/elastic/azure-mgmt-elastic/generated_tests/{test_microsoft_elastic_tag_rules_operations_async.py => test_elastic_tag_rules_operations_async.py} (62%) delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations_async.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations.py delete mode 100644 sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations_async.py create mode 100644 sdk/elastic/azure-mgmt-elastic/tsp-location.yaml diff --git a/sdk/elastic/azure-mgmt-elastic/CHANGELOG.md b/sdk/elastic/azure-mgmt-elastic/CHANGELOG.md index e5057eba9df2..9b2f1980320f 100644 --- a/sdk/elastic/azure-mgmt-elastic/CHANGELOG.md +++ b/sdk/elastic/azure-mgmt-elastic/CHANGELOG.md @@ -1,5 +1,9 @@ # Release History +## 1.1.0b5 (2025-05-13) + +change log generation failed!!! + ## 1.1.0b4 (2024-10-23) ### Features Added diff --git a/sdk/elastic/azure-mgmt-elastic/README.md b/sdk/elastic/azure-mgmt-elastic/README.md index ebb727a84a64..12b63a231f5b 100644 --- a/sdk/elastic/azure-mgmt-elastic/README.md +++ b/sdk/elastic/azure-mgmt-elastic/README.md @@ -1,7 +1,7 @@ # Microsoft Azure SDK for Python This is the Microsoft Azure Elastic Management Client Library. -This package has been tested with Python 3.8+. +This package has been tested with Python 3.9+. For a more complete view of Azure libraries, see the [azure sdk python release](https://aka.ms/azsdk/python/all). ## _Disclaimer_ @@ -12,7 +12,7 @@ _Azure SDK Python packages support for Python 2.7 has ended 01 January 2022. For ### Prerequisites -- Python 3.8+ is required to use this package. +- Python 3.9+ is required to use this package. - [Azure subscription](https://azure.microsoft.com/free/) ### Install the package @@ -24,7 +24,7 @@ pip install azure-identity ### Authentication -By default, [Azure Active Directory](https://aka.ms/awps/aad) token authentication depends on correct configure of following environment variables. +By default, [Azure Active Directory](https://aka.ms/awps/aad) token authentication depends on correct configuration of the following environment variables. - `AZURE_CLIENT_ID` for Azure client ID. - `AZURE_TENANT_ID` for Azure tenant ID. diff --git a/sdk/elastic/azure-mgmt-elastic/_meta.json b/sdk/elastic/azure-mgmt-elastic/_meta.json index 0823592354b1..34f59f452917 100644 --- a/sdk/elastic/azure-mgmt-elastic/_meta.json +++ b/sdk/elastic/azure-mgmt-elastic/_meta.json @@ -1,11 +1,6 @@ { - "commit": "6c709e5a3325eaa862649acee2252ce1c8166042", + "commit": "5c92dda155f958f8acd565f5ececf28dd940b560", "repository_url": "https://github.com/Azure/azure-rest-api-specs", - "autorest": "3.10.2", - "use": [ - "@autorest/python@6.19.0", - "@autorest/modelerfour@4.27.0" - ], - "autorest_command": "autorest specification/elastic/resource-manager/readme.md --generate-sample=True --generate-test=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.19.0 --use=@autorest/modelerfour@4.27.0 --version=3.10.2 --version-tolerant=False", - "readme": "specification/elastic/resource-manager/readme.md" + "typespec_src": "specification/elastic/Elastic.Management", + "@azure-tools/typespec-python": "0.44.2" } \ No newline at end of file diff --git a/sdk/elastic/azure-mgmt-elastic/apiview-properties.json b/sdk/elastic/azure-mgmt-elastic/apiview-properties.json new file mode 100644 index 000000000000..9aec30b5627d --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/apiview-properties.json @@ -0,0 +1,172 @@ +{ + "CrossLanguagePackageId": "Microsoft.Elastic", + "CrossLanguageDefinitionId": { + "azure.mgmt.elastic.models.ArmResponseElasticMonitorResource": "Azure.ResourceManager.ArmResponse", + "azure.mgmt.elastic.models.BillingInfoResponse": "Microsoft.Elastic.BillingInfoResponse", + "azure.mgmt.elastic.models.CompanyInfo": "Microsoft.Elastic.CompanyInfo", + "azure.mgmt.elastic.models.ConnectedPartnerResourceProperties": "Microsoft.Elastic.ConnectedPartnerResourceProperties", + "azure.mgmt.elastic.models.ConnectedPartnerResourcesListFormat": "Microsoft.Elastic.ConnectedPartnerResourcesListFormat", + "azure.mgmt.elastic.models.CreateOrUpdateParameterBody": "Microsoft.Elastic.createOrUpdate.Parameter.body.anonymous", + "azure.mgmt.elastic.models.DeploymentInfoResponse": "Microsoft.Elastic.DeploymentInfoResponse", + "azure.mgmt.elastic.models.ElasticCloudDeployment": "Microsoft.Elastic.ElasticCloudDeployment", + "azure.mgmt.elastic.models.ElasticCloudUser": "Microsoft.Elastic.ElasticCloudUser", + "azure.mgmt.elastic.models.Resource": "Azure.ResourceManager.CommonTypes.Resource", + "azure.mgmt.elastic.models.TrackedResource": "Azure.ResourceManager.CommonTypes.TrackedResource", + "azure.mgmt.elastic.models.ElasticMonitorResource": "Microsoft.Elastic.ElasticMonitorResource", + "azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters": "Microsoft.Elastic.ElasticMonitorResourceUpdateParameters", + "azure.mgmt.elastic.models.ElasticMonitorUpgrade": "Microsoft.Elastic.ElasticMonitorUpgrade", + "azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponse": "Microsoft.Elastic.ElasticOrganizationToAzureSubscriptionMappingResponse", + "azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponseProperties": "Microsoft.Elastic.ElasticOrganizationToAzureSubscriptionMappingResponseProperties", + "azure.mgmt.elastic.models.ElasticProperties": "Microsoft.Elastic.ElasticProperties", + "azure.mgmt.elastic.models.ElasticTrafficFilter": "Microsoft.Elastic.ElasticTrafficFilter", + "azure.mgmt.elastic.models.ElasticTrafficFilterResponse": "Microsoft.Elastic.ElasticTrafficFilterResponse", + "azure.mgmt.elastic.models.ElasticTrafficFilterRule": "Microsoft.Elastic.ElasticTrafficFilterRule", + "azure.mgmt.elastic.models.ElasticVersionListFormat": "Microsoft.Elastic.ElasticVersionListFormat", + "azure.mgmt.elastic.models.ElasticVersionListProperties": "Microsoft.Elastic.ElasticVersionListProperties", + "azure.mgmt.elastic.models.ErrorAdditionalInfo": "Azure.ResourceManager.CommonTypes.ErrorAdditionalInfo", + "azure.mgmt.elastic.models.ErrorDetail": "Azure.ResourceManager.CommonTypes.ErrorDetail", + "azure.mgmt.elastic.models.ErrorResponse": "Azure.ResourceManager.CommonTypes.ErrorResponse", + "azure.mgmt.elastic.models.ErrorResponseBody": "Microsoft.Elastic.ErrorResponseBody", + "azure.mgmt.elastic.models.ExternalUserCreationResponse": "Microsoft.Elastic.ExternalUserCreationResponse", + "azure.mgmt.elastic.models.ExternalUserInfo": "Microsoft.Elastic.ExternalUserInfo", + "azure.mgmt.elastic.models.FilteringTag": "Microsoft.Elastic.FilteringTag", + "azure.mgmt.elastic.models.IdentityProperties": "Microsoft.Elastic.IdentityProperties", + "azure.mgmt.elastic.models.LogRules": "Microsoft.Elastic.LogRules", + "azure.mgmt.elastic.models.MarketplaceSaaSInfo": "Microsoft.Elastic.MarketplaceSaaSInfo", + "azure.mgmt.elastic.models.MarketplaceSaaSInfoMarketplaceSubscription": "Microsoft.Elastic.MarketplaceSaaSInfoMarketplaceSubscription", + "azure.mgmt.elastic.models.MonitoredResource": "Microsoft.Elastic.MonitoredResource", + "azure.mgmt.elastic.models.MonitoredSubscription": "Microsoft.Elastic.MonitoredSubscription", + "azure.mgmt.elastic.models.ProxyResource": "Azure.ResourceManager.CommonTypes.ProxyResource", + "azure.mgmt.elastic.models.MonitoredSubscriptionProperties": "Microsoft.Elastic.MonitoredSubscriptionProperties", + "azure.mgmt.elastic.models.MonitoringTagRules": "Microsoft.Elastic.MonitoringTagRules", + "azure.mgmt.elastic.models.MonitoringTagRulesProperties": "Microsoft.Elastic.MonitoringTagRulesProperties", + "azure.mgmt.elastic.models.MonitorProperties": "Microsoft.Elastic.MonitorProperties", + "azure.mgmt.elastic.models.OkResponse": "TypeSpec.Http.OkResponse", + "azure.mgmt.elastic.models.OpenAIIntegrationProperties": "Microsoft.Elastic.OpenAIIntegrationProperties", + "azure.mgmt.elastic.models.OpenAIIntegrationRPModel": "Microsoft.Elastic.OpenAIIntegrationRPModel", + "azure.mgmt.elastic.models.OpenAIIntegrationStatusResponse": "Microsoft.Elastic.OpenAIIntegrationStatusResponse", + "azure.mgmt.elastic.models.OpenAIIntegrationStatusResponseProperties": "Microsoft.Elastic.OpenAIIntegrationStatusResponseProperties", + "azure.mgmt.elastic.models.OperationDisplay": "Microsoft.Elastic.OperationDisplay", + "azure.mgmt.elastic.models.OperationResult": "Microsoft.Elastic.OperationResult", + "azure.mgmt.elastic.models.PartnerBillingEntity": "Microsoft.Elastic.PartnerBillingEntity", + "azure.mgmt.elastic.models.PlanDetails": "Microsoft.Elastic.PlanDetails", + "azure.mgmt.elastic.models.ProjectDetails": "Microsoft.Elastic.ProjectDetails", + "azure.mgmt.elastic.models.ResourceProviderDefaultErrorResponse": "Microsoft.Elastic.ResourceProviderDefaultErrorResponse", + "azure.mgmt.elastic.models.ResourceSku": "Microsoft.Elastic.ResourceSku", + "azure.mgmt.elastic.models.ResubscribeParameterBody": "Microsoft.Elastic.resubscribe.Parameter.body.anonymous", + "azure.mgmt.elastic.models.ResubscribeProperties": "Microsoft.Elastic.ResubscribeProperties", + "azure.mgmt.elastic.models.SubscriptionList": "Microsoft.Elastic.SubscriptionList", + "azure.mgmt.elastic.models.SystemData": "Azure.ResourceManager.CommonTypes.SystemData", + "azure.mgmt.elastic.models.UpgradableVersionsList": "Microsoft.Elastic.UpgradableVersionsList", + "azure.mgmt.elastic.models.UpgradeParameterBody": "Microsoft.Elastic.upgrade.Parameter.body.anonymous", + "azure.mgmt.elastic.models.UserApiKeyResponse": "Microsoft.Elastic.UserApiKeyResponse", + "azure.mgmt.elastic.models.UserApiKeyResponseProperties": "Microsoft.Elastic.UserApiKeyResponseProperties", + "azure.mgmt.elastic.models.UserEmailId": "Microsoft.Elastic.UserEmailId", + "azure.mgmt.elastic.models.UserInfo": "Microsoft.Elastic.UserInfo", + "azure.mgmt.elastic.models.VMCollectionUpdate": "Microsoft.Elastic.VMCollectionUpdate", + "azure.mgmt.elastic.models.VMCollectionUpdateParameterBody": "Microsoft.Elastic.vMCollectionUpdate.Parameter.body.anonymous", + "azure.mgmt.elastic.models.VMIngestionDetailsResponse": "Microsoft.Elastic.VMIngestionDetailsResponse", + "azure.mgmt.elastic.models.VMResources": "Microsoft.Elastic.VMResources", + "azure.mgmt.elastic.models.CreatedByType": "Azure.ResourceManager.CommonTypes.createdByType", + "azure.mgmt.elastic.models.ProvisioningState": "Microsoft.Elastic.ProvisioningState", + "azure.mgmt.elastic.models.MonitoringStatus": "Microsoft.Elastic.MonitoringStatus", + "azure.mgmt.elastic.models.LiftrResourceCategories": "Microsoft.Elastic.LiftrResourceCategories", + "azure.mgmt.elastic.models.HostingType": "Microsoft.Elastic.HostingType", + "azure.mgmt.elastic.models.ProjectType": "Microsoft.Elastic.ProjectType", + "azure.mgmt.elastic.models.ConfigurationType": "Microsoft.Elastic.ConfigurationType", + "azure.mgmt.elastic.models.ManagedIdentityTypes": "Microsoft.Elastic.ManagedIdentityTypes", + "azure.mgmt.elastic.models.SendingLogs": "Microsoft.Elastic.SendingLogs", + "azure.mgmt.elastic.models.ElasticDeploymentStatus": "Microsoft.Elastic.ElasticDeploymentStatus", + "azure.mgmt.elastic.models.OperationName": "Microsoft.Elastic.OperationName", + "azure.mgmt.elastic.models.Type": "Microsoft.Elastic.Type", + "azure.mgmt.elastic.models.Operation": "Microsoft.Elastic.Operation", + "azure.mgmt.elastic.models.Status": "Microsoft.Elastic.Status", + "azure.mgmt.elastic.models.TagAction": "Microsoft.Elastic.TagAction", + "azure.mgmt.elastic.operations.Operations.list": "Microsoft.Elastic.Operations.list", + "azure.mgmt.elastic.aio.operations.Operations.list": "Microsoft.Elastic.Operations.list", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.get": "Microsoft.Elastic.ElasticMonitorResources.get", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.get": "Microsoft.Elastic.ElasticMonitorResources.get", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_create": "Microsoft.Elastic.ElasticMonitorResources.create", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_create": "Microsoft.Elastic.ElasticMonitorResources.create", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_update": "Microsoft.Elastic.ElasticMonitorResources.update", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_update": "Microsoft.Elastic.ElasticMonitorResources.update", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_delete": "Microsoft.Elastic.ElasticMonitorResources.delete", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_delete": "Microsoft.Elastic.ElasticMonitorResources.delete", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.list_by_resource_group": "Microsoft.Elastic.ElasticMonitorResources.listByResourceGroup", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.list_by_resource_group": "Microsoft.Elastic.ElasticMonitorResources.listByResourceGroup", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.list": "Microsoft.Elastic.ElasticMonitorResources.list", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.list": "Microsoft.Elastic.ElasticMonitorResources.list", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.monitored_resources_list": "Microsoft.Elastic.ElasticMonitorResources.monitoredResourcesList", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.monitored_resources_list": "Microsoft.Elastic.ElasticMonitorResources.monitoredResourcesList", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.deployment_info_list": "Microsoft.Elastic.ElasticMonitorResources.deploymentInfoList", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.deployment_info_list": "Microsoft.Elastic.ElasticMonitorResources.deploymentInfoList", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.create_or_update": "Microsoft.Elastic.ElasticMonitorResources.createOrUpdate", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.create_or_update": "Microsoft.Elastic.ElasticMonitorResources.createOrUpdate", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.billing_info_get": "Microsoft.Elastic.ElasticMonitorResources.billingInfoGet", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.billing_info_get": "Microsoft.Elastic.ElasticMonitorResources.billingInfoGet", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.connected_partner_resources_list": "Microsoft.Elastic.ElasticMonitorResources.connectedPartnerResourcesList", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.connected_partner_resources_list": "Microsoft.Elastic.ElasticMonitorResources.connectedPartnerResourcesList", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.v_m_host_list": "Microsoft.Elastic.ElasticMonitorResources.vMHostList", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.v_m_host_list": "Microsoft.Elastic.ElasticMonitorResources.vMHostList", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.details": "Microsoft.Elastic.ElasticMonitorResources.details", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.details": "Microsoft.Elastic.ElasticMonitorResources.details", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.v_m_collection_update": "Microsoft.Elastic.ElasticMonitorResources.vMCollectionUpdate", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.v_m_collection_update": "Microsoft.Elastic.ElasticMonitorResources.vMCollectionUpdate", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.upgradable_versions_details": "Microsoft.Elastic.ElasticMonitorResources.upgradableVersionsDetails", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.upgradable_versions_details": "Microsoft.Elastic.ElasticMonitorResources.upgradableVersionsDetails", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_upgrade": "Microsoft.Elastic.ElasticMonitorResources.upgrade", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_upgrade": "Microsoft.Elastic.ElasticMonitorResources.upgrade", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.all_traffic_filters_list": "Microsoft.Elastic.ElasticMonitorResources.allTrafficFiltersList", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.all_traffic_filters_list": "Microsoft.Elastic.ElasticMonitorResources.allTrafficFiltersList", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.list_associated_traffic_filters_list": "Microsoft.Elastic.ElasticMonitorResources.listAssociatedTrafficFiltersList", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.list_associated_traffic_filters_list": "Microsoft.Elastic.ElasticMonitorResources.listAssociatedTrafficFiltersList", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_create_and_associate_ip_filter_create": "Microsoft.Elastic.ElasticMonitorResources.createAndAssociateIPFilterCreate", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_create_and_associate_ip_filter_create": "Microsoft.Elastic.ElasticMonitorResources.createAndAssociateIPFilterCreate", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_create_and_associate_pl_filter_create": "Microsoft.Elastic.ElasticMonitorResources.createAndAssociatePLFilterCreate", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_create_and_associate_pl_filter_create": "Microsoft.Elastic.ElasticMonitorResources.createAndAssociatePLFilterCreate", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_associate": "Microsoft.Elastic.ElasticMonitorResources.associate", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_associate": "Microsoft.Elastic.ElasticMonitorResources.associate", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.detach_and_delete_traffic_filter_delete": "Microsoft.Elastic.ElasticMonitorResources.detachAndDeleteTrafficFilterDelete", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.detach_and_delete_traffic_filter_delete": "Microsoft.Elastic.ElasticMonitorResources.detachAndDeleteTrafficFilterDelete", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_detach_traffic_filter_update": "Microsoft.Elastic.ElasticMonitorResources.detachTrafficFilterUpdate", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_detach_traffic_filter_update": "Microsoft.Elastic.ElasticMonitorResources.detachTrafficFilterUpdate", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.traffic_filters_delete": "Microsoft.Elastic.ElasticMonitorResources.trafficFiltersDelete", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.traffic_filters_delete": "Microsoft.Elastic.ElasticMonitorResources.trafficFiltersDelete", + "azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations.begin_resubscribe": "Microsoft.Elastic.ElasticMonitorResources.resubscribe", + "azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations.begin_resubscribe": "Microsoft.Elastic.ElasticMonitorResources.resubscribe", + "azure.mgmt.elastic.operations.MonitoredSubscriptionsOperations.get": "Microsoft.Elastic.MonitoredSubscriptions.get", + "azure.mgmt.elastic.aio.operations.MonitoredSubscriptionsOperations.get": "Microsoft.Elastic.MonitoredSubscriptions.get", + "azure.mgmt.elastic.operations.MonitoredSubscriptionsOperations.begin_createor_update": "Microsoft.Elastic.MonitoredSubscriptions.createorUpdate", + "azure.mgmt.elastic.aio.operations.MonitoredSubscriptionsOperations.begin_createor_update": "Microsoft.Elastic.MonitoredSubscriptions.createorUpdate", + "azure.mgmt.elastic.operations.MonitoredSubscriptionsOperations.begin_update": "Microsoft.Elastic.MonitoredSubscriptions.update", + "azure.mgmt.elastic.aio.operations.MonitoredSubscriptionsOperations.begin_update": "Microsoft.Elastic.MonitoredSubscriptions.update", + "azure.mgmt.elastic.operations.MonitoredSubscriptionsOperations.begin_delete": "Microsoft.Elastic.MonitoredSubscriptions.delete", + "azure.mgmt.elastic.aio.operations.MonitoredSubscriptionsOperations.begin_delete": "Microsoft.Elastic.MonitoredSubscriptions.delete", + "azure.mgmt.elastic.operations.MonitoredSubscriptionsOperations.list": "Microsoft.Elastic.MonitoredSubscriptions.list", + "azure.mgmt.elastic.aio.operations.MonitoredSubscriptionsOperations.list": "Microsoft.Elastic.MonitoredSubscriptions.list", + "azure.mgmt.elastic.operations.OpenAIIntegrationRPModelsOperations.get": "Microsoft.Elastic.OpenAIIntegrationRPModels.get", + "azure.mgmt.elastic.aio.operations.OpenAIIntegrationRPModelsOperations.get": "Microsoft.Elastic.OpenAIIntegrationRPModels.get", + "azure.mgmt.elastic.operations.OpenAIIntegrationRPModelsOperations.create_or_update": "Microsoft.Elastic.OpenAIIntegrationRPModels.createOrUpdate", + "azure.mgmt.elastic.aio.operations.OpenAIIntegrationRPModelsOperations.create_or_update": "Microsoft.Elastic.OpenAIIntegrationRPModels.createOrUpdate", + "azure.mgmt.elastic.operations.OpenAIIntegrationRPModelsOperations.delete": "Microsoft.Elastic.OpenAIIntegrationRPModels.delete", + "azure.mgmt.elastic.aio.operations.OpenAIIntegrationRPModelsOperations.delete": "Microsoft.Elastic.OpenAIIntegrationRPModels.delete", + "azure.mgmt.elastic.operations.OpenAIIntegrationRPModelsOperations.list": "Microsoft.Elastic.OpenAIIntegrationRPModels.list", + "azure.mgmt.elastic.aio.operations.OpenAIIntegrationRPModelsOperations.list": "Microsoft.Elastic.OpenAIIntegrationRPModels.list", + "azure.mgmt.elastic.operations.OpenAIIntegrationRPModelsOperations.get_status": "Microsoft.Elastic.OpenAIIntegrationRPModels.getStatus", + "azure.mgmt.elastic.aio.operations.OpenAIIntegrationRPModelsOperations.get_status": "Microsoft.Elastic.OpenAIIntegrationRPModels.getStatus", + "azure.mgmt.elastic.operations.TagRulesOperations.get": "Microsoft.Elastic.TagRules.get", + "azure.mgmt.elastic.aio.operations.TagRulesOperations.get": "Microsoft.Elastic.TagRules.get", + "azure.mgmt.elastic.operations.TagRulesOperations.create_or_update": "Microsoft.Elastic.TagRules.createOrUpdate", + "azure.mgmt.elastic.aio.operations.TagRulesOperations.create_or_update": "Microsoft.Elastic.TagRules.createOrUpdate", + "azure.mgmt.elastic.operations.TagRulesOperations.begin_delete": "Microsoft.Elastic.TagRules.delete", + "azure.mgmt.elastic.aio.operations.TagRulesOperations.begin_delete": "Microsoft.Elastic.TagRules.delete", + "azure.mgmt.elastic.operations.TagRulesOperations.list": "Microsoft.Elastic.TagRules.list", + "azure.mgmt.elastic.aio.operations.TagRulesOperations.list": "Microsoft.Elastic.TagRules.list", + "azure.mgmt.elastic.operations.ElasticVersionsOperationGroupOperations.list": "Microsoft.Elastic.ElasticVersionsOperationGroup.list", + "azure.mgmt.elastic.aio.operations.ElasticVersionsOperationGroupOperations.list": "Microsoft.Elastic.ElasticVersionsOperationGroup.list", + "azure.mgmt.elastic.operations.OrganizationsOperationGroupOperations.get_api_key": "Microsoft.Elastic.OrganizationsOperationGroup.getApiKey", + "azure.mgmt.elastic.aio.operations.OrganizationsOperationGroupOperations.get_api_key": "Microsoft.Elastic.OrganizationsOperationGroup.getApiKey", + "azure.mgmt.elastic.operations.OrganizationsOperationGroupOperations.get_elastic_to_azure_subscription_mapping": "Microsoft.Elastic.OrganizationsOperationGroup.getElasticToAzureSubscriptionMapping", + "azure.mgmt.elastic.aio.operations.OrganizationsOperationGroupOperations.get_elastic_to_azure_subscription_mapping": "Microsoft.Elastic.OrganizationsOperationGroup.getElasticToAzureSubscriptionMapping" + } +} \ No newline at end of file diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/__init__.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/__init__.py index a9ab412a5f83..aaa24ba07ed5 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/__init__.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/__init__.py @@ -2,25 +2,31 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +# pylint: disable=wrong-import-position -from ._microsoft_elastic import MicrosoftElastic +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from ._patch import * # pylint: disable=unused-wildcard-import + +from ._client import ElasticClient # type: ignore from ._version import VERSION __version__ = VERSION try: from ._patch import __all__ as _patch_all - from ._patch import * # pylint: disable=unused-wildcard-import + from ._patch import * except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk __all__ = [ - "MicrosoftElastic", + "ElasticClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_client.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_client.py new file mode 100644 index 000000000000..803f97bfd3d4 --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_client.py @@ -0,0 +1,164 @@ +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from copy import deepcopy +from typing import Any, Optional, TYPE_CHECKING, cast +from typing_extensions import Self + +from azure.core.pipeline import policies +from azure.core.rest import HttpRequest, HttpResponse +from azure.core.settings import settings +from azure.mgmt.core import ARMPipelineClient +from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy +from azure.mgmt.core.tools import get_arm_endpoints + +from ._configuration import ElasticClientConfiguration +from ._utils.serialization import Deserializer, Serializer +from .operations import ( + ElasticMonitorResourcesOperations, + ElasticVersionsOperationGroupOperations, + MonitoredSubscriptionsOperations, + OpenAIIntegrationRPModelsOperations, + Operations, + OrganizationsOperationGroupOperations, + TagRulesOperations, +) + +if TYPE_CHECKING: + from azure.core.credentials import TokenCredential + + +class ElasticClient: # pylint: disable=too-many-instance-attributes + """ElasticClient. + + :ivar operations: Operations operations + :vartype operations: azure.mgmt.elastic.operations.Operations + :ivar elastic_monitor_resources: ElasticMonitorResourcesOperations operations + :vartype elastic_monitor_resources: + azure.mgmt.elastic.operations.ElasticMonitorResourcesOperations + :ivar monitored_subscriptions: MonitoredSubscriptionsOperations operations + :vartype monitored_subscriptions: + azure.mgmt.elastic.operations.MonitoredSubscriptionsOperations + :ivar open_ai_integration_rp_models: OpenAIIntegrationRPModelsOperations operations + :vartype open_ai_integration_rp_models: + azure.mgmt.elastic.operations.OpenAIIntegrationRPModelsOperations + :ivar tag_rules: TagRulesOperations operations + :vartype tag_rules: azure.mgmt.elastic.operations.TagRulesOperations + :ivar elastic_versions_operation_group: ElasticVersionsOperationGroupOperations operations + :vartype elastic_versions_operation_group: + azure.mgmt.elastic.operations.ElasticVersionsOperationGroupOperations + :ivar organizations_operation_group: OrganizationsOperationGroupOperations operations + :vartype organizations_operation_group: + azure.mgmt.elastic.operations.OrganizationsOperationGroupOperations + :param credential: Credential used to authenticate requests to the service. Required. + :type credential: ~azure.core.credentials.TokenCredential + :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. + :type subscription_id: str + :param base_url: Service host. Default value is None. + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is + "2025-01-15-preview". Note that overriding this default value may result in unsupported + behavior. + :paramtype api_version: 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: "TokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any + ) -> None: + _endpoint = "{endpoint}" + _cloud = kwargs.pop("cloud_setting", None) or settings.current.azure_cloud # type: ignore + _endpoints = get_arm_endpoints(_cloud) + if not base_url: + base_url = _endpoints["resource_manager"] + credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) + self._config = ElasticClientConfiguration( + credential=credential, + subscription_id=subscription_id, + base_url=cast(str, base_url), + credential_scopes=credential_scopes, + **kwargs + ) + + _policies = kwargs.pop("policies", None) + if _policies is None: + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] + self._client: ARMPipelineClient = ARMPipelineClient(base_url=cast(str, _endpoint), policies=_policies, **kwargs) + + self._serialize = Serializer() + self._deserialize = Deserializer() + self._serialize.client_side_validation = False + self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) + self.elastic_monitor_resources = ElasticMonitorResourcesOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.monitored_subscriptions = MonitoredSubscriptionsOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.open_ai_integration_rp_models = OpenAIIntegrationRPModelsOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.tag_rules = TagRulesOperations(self._client, self._config, self._serialize, self._deserialize) + self.elastic_versions_operation_group = ElasticVersionsOperationGroupOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.organizations_operation_group = OrganizationsOperationGroupOperations( + self._client, self._config, self._serialize, self._deserialize + ) + + def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + """Runs the network request through the client's chained policies. + + >>> 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/dpcodegen/python/send_request + + :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.rest.HttpResponse + """ + + request_copy = deepcopy(request) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore + + def close(self) -> None: + self._client.close() + + def __enter__(self) -> Self: + self._client.__enter__() + return self + + def __exit__(self, *exc_details: Any) -> None: + self._client.__exit__(*exc_details) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_configuration.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_configuration.py index 7e7a23a868b0..cb47ab10d652 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_configuration.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_configuration.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -14,27 +14,35 @@ from ._version import VERSION if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports from azure.core.credentials import TokenCredential -class MicrosoftElasticConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long - """Configuration for MicrosoftElastic. +class ElasticClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for ElasticClient. Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :keyword api_version: Api Version. Default value is "2024-06-15-preview". Note that overriding - this default value may result in unsupported behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is + "2025-01-15-preview". Note that overriding this default value may result in unsupported + behavior. :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2024-06-15-preview") + def __init__( + self, + credential: "TokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop("api_version", "2025-01-15-preview") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -43,6 +51,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-elastic/{}".format(VERSION)) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_microsoft_elastic.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_microsoft_elastic.py deleted file mode 100644 index 80947d3c4e99..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_microsoft_elastic.py +++ /dev/null @@ -1,239 +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. -# -------------------------------------------------------------------------- - -from copy import deepcopy -from typing import Any, TYPE_CHECKING -from typing_extensions import Self - -from azure.core.pipeline import policies -from azure.core.rest import HttpRequest, HttpResponse -from azure.mgmt.core import ARMPipelineClient -from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy - -from . import models as _models -from ._configuration import MicrosoftElasticConfiguration -from ._serialization import Deserializer, Serializer -from .operations import ( - AllTrafficFiltersOperations, - AssociateTrafficFilterOperations, - BillingInfoOperations, - ConnectedPartnerResourcesOperations, - CreateAndAssociateIPFilterOperations, - CreateAndAssociatePLFilterOperations, - DeploymentInfoOperations, - DetachAndDeleteTrafficFilterOperations, - DetachTrafficFilterOperations, - ElasticVersionsOperations, - ExternalUserOperations, - ListAssociatedTrafficFiltersOperations, - MonitorOperations, - MonitoredResourcesOperations, - MonitoredSubscriptionsOperations, - MonitorsOperations, - OpenAIOperations, - Operations, - OrganizationsOperations, - TagRulesOperations, - TrafficFiltersOperations, - UpgradableVersionsOperations, - VMCollectionOperations, - VMHostOperations, - VMIngestionOperations, -) - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from azure.core.credentials import TokenCredential - - -class MicrosoftElastic: # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes - """MicrosoftElastic. - - :ivar operations: Operations operations - :vartype operations: azure.mgmt.elastic.operations.Operations - :ivar monitors: MonitorsOperations operations - :vartype monitors: azure.mgmt.elastic.operations.MonitorsOperations - :ivar elastic_versions: ElasticVersionsOperations operations - :vartype elastic_versions: azure.mgmt.elastic.operations.ElasticVersionsOperations - :ivar monitored_subscriptions: MonitoredSubscriptionsOperations operations - :vartype monitored_subscriptions: - azure.mgmt.elastic.operations.MonitoredSubscriptionsOperations - :ivar monitored_resources: MonitoredResourcesOperations operations - :vartype monitored_resources: azure.mgmt.elastic.operations.MonitoredResourcesOperations - :ivar deployment_info: DeploymentInfoOperations operations - :vartype deployment_info: azure.mgmt.elastic.operations.DeploymentInfoOperations - :ivar external_user: ExternalUserOperations operations - :vartype external_user: azure.mgmt.elastic.operations.ExternalUserOperations - :ivar billing_info: BillingInfoOperations operations - :vartype billing_info: azure.mgmt.elastic.operations.BillingInfoOperations - :ivar connected_partner_resources: ConnectedPartnerResourcesOperations operations - :vartype connected_partner_resources: - azure.mgmt.elastic.operations.ConnectedPartnerResourcesOperations - :ivar open_ai: OpenAIOperations operations - :vartype open_ai: azure.mgmt.elastic.operations.OpenAIOperations - :ivar tag_rules: TagRulesOperations operations - :vartype tag_rules: azure.mgmt.elastic.operations.TagRulesOperations - :ivar vm_host: VMHostOperations operations - :vartype vm_host: azure.mgmt.elastic.operations.VMHostOperations - :ivar vm_ingestion: VMIngestionOperations operations - :vartype vm_ingestion: azure.mgmt.elastic.operations.VMIngestionOperations - :ivar vm_collection: VMCollectionOperations operations - :vartype vm_collection: azure.mgmt.elastic.operations.VMCollectionOperations - :ivar upgradable_versions: UpgradableVersionsOperations operations - :vartype upgradable_versions: azure.mgmt.elastic.operations.UpgradableVersionsOperations - :ivar monitor: MonitorOperations operations - :vartype monitor: azure.mgmt.elastic.operations.MonitorOperations - :ivar all_traffic_filters: AllTrafficFiltersOperations operations - :vartype all_traffic_filters: azure.mgmt.elastic.operations.AllTrafficFiltersOperations - :ivar list_associated_traffic_filters: ListAssociatedTrafficFiltersOperations operations - :vartype list_associated_traffic_filters: - azure.mgmt.elastic.operations.ListAssociatedTrafficFiltersOperations - :ivar create_and_associate_ip_filter: CreateAndAssociateIPFilterOperations operations - :vartype create_and_associate_ip_filter: - azure.mgmt.elastic.operations.CreateAndAssociateIPFilterOperations - :ivar create_and_associate_pl_filter: CreateAndAssociatePLFilterOperations operations - :vartype create_and_associate_pl_filter: - azure.mgmt.elastic.operations.CreateAndAssociatePLFilterOperations - :ivar associate_traffic_filter: AssociateTrafficFilterOperations operations - :vartype associate_traffic_filter: - azure.mgmt.elastic.operations.AssociateTrafficFilterOperations - :ivar detach_and_delete_traffic_filter: DetachAndDeleteTrafficFilterOperations operations - :vartype detach_and_delete_traffic_filter: - azure.mgmt.elastic.operations.DetachAndDeleteTrafficFilterOperations - :ivar detach_traffic_filter: DetachTrafficFilterOperations operations - :vartype detach_traffic_filter: azure.mgmt.elastic.operations.DetachTrafficFilterOperations - :ivar traffic_filters: TrafficFiltersOperations operations - :vartype traffic_filters: azure.mgmt.elastic.operations.TrafficFiltersOperations - :ivar organizations: OrganizationsOperations operations - :vartype organizations: azure.mgmt.elastic.operations.OrganizationsOperations - :param credential: Credential needed for the client to connect to Azure. Required. - :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. - :type subscription_id: str - :param base_url: Service URL. Default value is "https://management.azure.com". - :type base_url: str - :keyword api_version: Api Version. Default value is "2024-06-15-preview". Note that overriding - this default value may result in unsupported behavior. - :paramtype api_version: 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: "TokenCredential", - subscription_id: str, - base_url: str = "https://management.azure.com", - **kwargs: Any - ) -> None: - self._config = MicrosoftElasticConfiguration(credential=credential, subscription_id=subscription_id, **kwargs) - _policies = kwargs.pop("policies", None) - if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} - self._serialize = Serializer(client_models) - self._deserialize = Deserializer(client_models) - self._serialize.client_side_validation = False - self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) - self.monitors = MonitorsOperations(self._client, self._config, self._serialize, self._deserialize) - self.elastic_versions = ElasticVersionsOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.monitored_subscriptions = MonitoredSubscriptionsOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.monitored_resources = MonitoredResourcesOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.deployment_info = DeploymentInfoOperations(self._client, self._config, self._serialize, self._deserialize) - self.external_user = ExternalUserOperations(self._client, self._config, self._serialize, self._deserialize) - self.billing_info = BillingInfoOperations(self._client, self._config, self._serialize, self._deserialize) - self.connected_partner_resources = ConnectedPartnerResourcesOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.open_ai = OpenAIOperations(self._client, self._config, self._serialize, self._deserialize) - self.tag_rules = TagRulesOperations(self._client, self._config, self._serialize, self._deserialize) - self.vm_host = VMHostOperations(self._client, self._config, self._serialize, self._deserialize) - self.vm_ingestion = VMIngestionOperations(self._client, self._config, self._serialize, self._deserialize) - self.vm_collection = VMCollectionOperations(self._client, self._config, self._serialize, self._deserialize) - self.upgradable_versions = UpgradableVersionsOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.monitor = MonitorOperations(self._client, self._config, self._serialize, self._deserialize) - self.all_traffic_filters = AllTrafficFiltersOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.list_associated_traffic_filters = ListAssociatedTrafficFiltersOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.create_and_associate_ip_filter = CreateAndAssociateIPFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.create_and_associate_pl_filter = CreateAndAssociatePLFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.associate_traffic_filter = AssociateTrafficFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.detach_and_delete_traffic_filter = DetachAndDeleteTrafficFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.detach_traffic_filter = DetachTrafficFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.traffic_filters = TrafficFiltersOperations(self._client, self._config, self._serialize, self._deserialize) - self.organizations = OrganizationsOperations(self._client, self._config, self._serialize, self._deserialize) - - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: - """Runs the network request through the client's chained policies. - - >>> 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/dpcodegen/python/send_request - - :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.rest.HttpResponse - """ - - request_copy = deepcopy(request) - request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore - - def close(self) -> None: - self._client.close() - - def __enter__(self) -> Self: - self._client.__enter__() - return self - - def __exit__(self, *exc_details: Any) -> None: - self._client.__exit__(*exc_details) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_patch.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_patch.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/__init__.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/__init__.py new file mode 100644 index 000000000000..8026245c2abc --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/__init__.py @@ -0,0 +1,6 @@ +# -------------------------------------------------------------------------- +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/model_base.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/model_base.py new file mode 100644 index 000000000000..49d5c7259389 --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/model_base.py @@ -0,0 +1,1232 @@ +# pylint: disable=too-many-lines +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +# pylint: disable=protected-access, broad-except + +import copy +import calendar +import decimal +import functools +import sys +import logging +import base64 +import re +import typing +import enum +import email.utils +from datetime import datetime, date, time, timedelta, timezone +from json import JSONEncoder +import xml.etree.ElementTree as ET +from collections.abc import MutableMapping +from typing_extensions import Self +import isodate +from azure.core.exceptions import DeserializationError +from azure.core import CaseInsensitiveEnumMeta +from azure.core.pipeline import PipelineResponse +from azure.core.serialization import _Null + +_LOGGER = logging.getLogger(__name__) + +__all__ = ["SdkJSONEncoder", "Model", "rest_field", "rest_discriminator"] + +TZ_UTC = timezone.utc +_T = typing.TypeVar("_T") + + +def _timedelta_as_isostr(td: timedelta) -> str: + """Converts a datetime.timedelta object into an ISO 8601 formatted string, e.g. 'P4DT12H30M05S' + + Function adapted from the Tin Can Python project: https://github.com/RusticiSoftware/TinCanPython + + :param timedelta td: The timedelta to convert + :rtype: str + :return: ISO8601 version of this timedelta + """ + + # Split seconds to larger units + seconds = td.total_seconds() + minutes, seconds = divmod(seconds, 60) + hours, minutes = divmod(minutes, 60) + days, hours = divmod(hours, 24) + + days, hours, minutes = list(map(int, (days, hours, minutes))) + seconds = round(seconds, 6) + + # Build date + date_str = "" + if days: + date_str = "%sD" % days + + if hours or minutes or seconds: + # Build time + time_str = "T" + + # Hours + bigger_exists = date_str or hours + if bigger_exists: + time_str += "{:02}H".format(hours) + + # Minutes + bigger_exists = bigger_exists or minutes + if bigger_exists: + time_str += "{:02}M".format(minutes) + + # Seconds + try: + if seconds.is_integer(): + seconds_string = "{:02}".format(int(seconds)) + else: + # 9 chars long w/ leading 0, 6 digits after decimal + seconds_string = "%09.6f" % seconds + # Remove trailing zeros + seconds_string = seconds_string.rstrip("0") + except AttributeError: # int.is_integer() raises + seconds_string = "{:02}".format(seconds) + + time_str += "{}S".format(seconds_string) + else: + time_str = "" + + return "P" + date_str + time_str + + +def _serialize_bytes(o, format: typing.Optional[str] = None) -> str: + encoded = base64.b64encode(o).decode() + if format == "base64url": + return encoded.strip("=").replace("+", "-").replace("/", "_") + return encoded + + +def _serialize_datetime(o, format: typing.Optional[str] = None): + if hasattr(o, "year") and hasattr(o, "hour"): + if format == "rfc7231": + return email.utils.format_datetime(o, usegmt=True) + if format == "unix-timestamp": + return int(calendar.timegm(o.utctimetuple())) + + # astimezone() fails for naive times in Python 2.7, so make make sure o is aware (tzinfo is set) + if not o.tzinfo: + iso_formatted = o.replace(tzinfo=TZ_UTC).isoformat() + else: + iso_formatted = o.astimezone(TZ_UTC).isoformat() + # Replace the trailing "+00:00" UTC offset with "Z" (RFC 3339: https://www.ietf.org/rfc/rfc3339.txt) + return iso_formatted.replace("+00:00", "Z") + # Next try datetime.date or datetime.time + return o.isoformat() + + +def _is_readonly(p): + try: + return p._visibility == ["read"] + except AttributeError: + return False + + +class SdkJSONEncoder(JSONEncoder): + """A JSON encoder that's capable of serializing datetime objects and bytes.""" + + def __init__(self, *args, exclude_readonly: bool = False, format: typing.Optional[str] = None, **kwargs): + super().__init__(*args, **kwargs) + self.exclude_readonly = exclude_readonly + self.format = format + + def default(self, o): # pylint: disable=too-many-return-statements + if _is_model(o): + if self.exclude_readonly: + readonly_props = [p._rest_name for p in o._attr_to_rest_field.values() if _is_readonly(p)] + return {k: v for k, v in o.items() if k not in readonly_props} + return dict(o.items()) + try: + return super(SdkJSONEncoder, self).default(o) + except TypeError: + if isinstance(o, _Null): + return None + if isinstance(o, decimal.Decimal): + return float(o) + if isinstance(o, (bytes, bytearray)): + return _serialize_bytes(o, self.format) + try: + # First try datetime.datetime + return _serialize_datetime(o, self.format) + except AttributeError: + pass + # Last, try datetime.timedelta + try: + return _timedelta_as_isostr(o) + except AttributeError: + # This will be raised when it hits value.total_seconds in the method above + pass + return super(SdkJSONEncoder, self).default(o) + + +_VALID_DATE = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}" + r"\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") +_VALID_RFC7231 = re.compile( + r"(Mon|Tue|Wed|Thu|Fri|Sat|Sun),\s\d{2}\s" + r"(Jan|Feb|Mar|Apr|May|Jun|Jul|Aug|Sep|Oct|Nov|Dec)\s\d{4}\s\d{2}:\d{2}:\d{2}\sGMT" +) + + +def _deserialize_datetime(attr: typing.Union[str, datetime]) -> datetime: + """Deserialize ISO-8601 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + attr = attr.upper() + match = _VALID_DATE.match(attr) + if not match: + raise ValueError("Invalid datetime string: " + attr) + + check_decimal = attr.split(".") + if len(check_decimal) > 1: + decimal_str = "" + for digit in check_decimal[1]: + if digit.isdigit(): + decimal_str += digit + else: + break + if len(decimal_str) > 6: + attr = attr.replace(decimal_str, decimal_str[0:6]) + + date_obj = isodate.parse_datetime(attr) + test_utc = date_obj.utctimetuple() + if test_utc.tm_year > 9999 or test_utc.tm_year < 1: + raise OverflowError("Hit max or min date") + return date_obj + + +def _deserialize_datetime_rfc7231(attr: typing.Union[str, datetime]) -> datetime: + """Deserialize RFC7231 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + match = _VALID_RFC7231.match(attr) + if not match: + raise ValueError("Invalid datetime string: " + attr) + + return email.utils.parsedate_to_datetime(attr) + + +def _deserialize_datetime_unix_timestamp(attr: typing.Union[float, datetime]) -> datetime: + """Deserialize unix timestamp into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + return datetime.fromtimestamp(attr, TZ_UTC) + + +def _deserialize_date(attr: typing.Union[str, date]) -> date: + """Deserialize ISO-8601 formatted string into Date object. + :param str attr: response string to be deserialized. + :rtype: date + :returns: The date object from that input + """ + # This must NOT use defaultmonth/defaultday. Using None ensure this raises an exception. + if isinstance(attr, date): + return attr + return isodate.parse_date(attr, defaultmonth=None, defaultday=None) # type: ignore + + +def _deserialize_time(attr: typing.Union[str, time]) -> time: + """Deserialize ISO-8601 formatted string into time object. + + :param str attr: response string to be deserialized. + :rtype: datetime.time + :returns: The time object from that input + """ + if isinstance(attr, time): + return attr + return isodate.parse_time(attr) + + +def _deserialize_bytes(attr): + if isinstance(attr, (bytes, bytearray)): + return attr + return bytes(base64.b64decode(attr)) + + +def _deserialize_bytes_base64(attr): + if isinstance(attr, (bytes, bytearray)): + return attr + padding = "=" * (3 - (len(attr) + 3) % 4) # type: ignore + attr = attr + padding # type: ignore + encoded = attr.replace("-", "+").replace("_", "/") + return bytes(base64.b64decode(encoded)) + + +def _deserialize_duration(attr): + if isinstance(attr, timedelta): + return attr + return isodate.parse_duration(attr) + + +def _deserialize_decimal(attr): + if isinstance(attr, decimal.Decimal): + return attr + return decimal.Decimal(str(attr)) + + +def _deserialize_int_as_str(attr): + if isinstance(attr, int): + return attr + return int(attr) + + +_DESERIALIZE_MAPPING = { + datetime: _deserialize_datetime, + date: _deserialize_date, + time: _deserialize_time, + bytes: _deserialize_bytes, + bytearray: _deserialize_bytes, + timedelta: _deserialize_duration, + typing.Any: lambda x: x, + decimal.Decimal: _deserialize_decimal, +} + +_DESERIALIZE_MAPPING_WITHFORMAT = { + "rfc3339": _deserialize_datetime, + "rfc7231": _deserialize_datetime_rfc7231, + "unix-timestamp": _deserialize_datetime_unix_timestamp, + "base64": _deserialize_bytes, + "base64url": _deserialize_bytes_base64, +} + + +def get_deserializer(annotation: typing.Any, rf: typing.Optional["_RestField"] = None): + if annotation is int and rf and rf._format == "str": + return _deserialize_int_as_str + if rf and rf._format: + return _DESERIALIZE_MAPPING_WITHFORMAT.get(rf._format) + return _DESERIALIZE_MAPPING.get(annotation) # pyright: ignore + + +def _get_type_alias_type(module_name: str, alias_name: str): + types = { + k: v + for k, v in sys.modules[module_name].__dict__.items() + if isinstance(v, typing._GenericAlias) # type: ignore + } + if alias_name not in types: + return alias_name + return types[alias_name] + + +def _get_model(module_name: str, model_name: str): + models = {k: v for k, v in sys.modules[module_name].__dict__.items() if isinstance(v, type)} + module_end = module_name.rsplit(".", 1)[0] + models.update({k: v for k, v in sys.modules[module_end].__dict__.items() if isinstance(v, type)}) + if isinstance(model_name, str): + model_name = model_name.split(".")[-1] + if model_name not in models: + return model_name + return models[model_name] + + +_UNSET = object() + + +class _MyMutableMapping(MutableMapping[str, typing.Any]): + def __init__(self, data: typing.Dict[str, typing.Any]) -> None: + self._data = data + + def __contains__(self, key: typing.Any) -> bool: + return key in self._data + + def __getitem__(self, key: str) -> typing.Any: + return self._data.__getitem__(key) + + def __setitem__(self, key: str, value: typing.Any) -> None: + self._data.__setitem__(key, value) + + def __delitem__(self, key: str) -> None: + self._data.__delitem__(key) + + def __iter__(self) -> typing.Iterator[typing.Any]: + return self._data.__iter__() + + def __len__(self) -> int: + return self._data.__len__() + + def __ne__(self, other: typing.Any) -> bool: + return not self.__eq__(other) + + def keys(self) -> typing.KeysView[str]: + """ + :returns: a set-like object providing a view on D's keys + :rtype: ~typing.KeysView + """ + return self._data.keys() + + def values(self) -> typing.ValuesView[typing.Any]: + """ + :returns: an object providing a view on D's values + :rtype: ~typing.ValuesView + """ + return self._data.values() + + def items(self) -> typing.ItemsView[str, typing.Any]: + """ + :returns: set-like object providing a view on D's items + :rtype: ~typing.ItemsView + """ + return self._data.items() + + def get(self, key: str, default: typing.Any = None) -> typing.Any: + """ + Get the value for key if key is in the dictionary, else default. + :param str key: The key to look up. + :param any default: The value to return if key is not in the dictionary. Defaults to None + :returns: D[k] if k in D, else d. + :rtype: any + """ + try: + return self[key] + except KeyError: + return default + + @typing.overload + def pop(self, key: str) -> typing.Any: ... # pylint: disable=arguments-differ + + @typing.overload + def pop(self, key: str, default: _T) -> _T: ... # pylint: disable=signature-differs + + @typing.overload + def pop(self, key: str, default: typing.Any) -> typing.Any: ... # pylint: disable=signature-differs + + def pop(self, key: str, default: typing.Any = _UNSET) -> typing.Any: + """ + Removes specified key and return the corresponding value. + :param str key: The key to pop. + :param any default: The value to return if key is not in the dictionary + :returns: The value corresponding to the key. + :rtype: any + :raises KeyError: If key is not found and default is not given. + """ + if default is _UNSET: + return self._data.pop(key) + return self._data.pop(key, default) + + def popitem(self) -> typing.Tuple[str, typing.Any]: + """ + Removes and returns some (key, value) pair + :returns: The (key, value) pair. + :rtype: tuple + :raises KeyError: if D is empty. + """ + return self._data.popitem() + + def clear(self) -> None: + """ + Remove all items from D. + """ + self._data.clear() + + def update(self, *args: typing.Any, **kwargs: typing.Any) -> None: # pylint: disable=arguments-differ + """ + Updates D from mapping/iterable E and F. + :param any args: Either a mapping object or an iterable of key-value pairs. + """ + self._data.update(*args, **kwargs) + + @typing.overload + def setdefault(self, key: str, default: None = None) -> None: ... + + @typing.overload + def setdefault(self, key: str, default: typing.Any) -> typing.Any: ... # pylint: disable=signature-differs + + def setdefault(self, key: str, default: typing.Any = _UNSET) -> typing.Any: + """ + Same as calling D.get(k, d), and setting D[k]=d if k not found + :param str key: The key to look up. + :param any default: The value to set if key is not in the dictionary + :returns: D[k] if k in D, else d. + :rtype: any + """ + if default is _UNSET: + return self._data.setdefault(key) + return self._data.setdefault(key, default) + + def __eq__(self, other: typing.Any) -> bool: + try: + other_model = self.__class__(other) + except Exception: + return False + return self._data == other_model._data + + def __repr__(self) -> str: + return str(self._data) + + +def _is_model(obj: typing.Any) -> bool: + return getattr(obj, "_is_model", False) + + +def _serialize(o, format: typing.Optional[str] = None): # pylint: disable=too-many-return-statements + if isinstance(o, list): + return [_serialize(x, format) for x in o] + if isinstance(o, dict): + return {k: _serialize(v, format) for k, v in o.items()} + if isinstance(o, set): + return {_serialize(x, format) for x in o} + if isinstance(o, tuple): + return tuple(_serialize(x, format) for x in o) + if isinstance(o, (bytes, bytearray)): + return _serialize_bytes(o, format) + if isinstance(o, decimal.Decimal): + return float(o) + if isinstance(o, enum.Enum): + return o.value + if isinstance(o, int): + if format == "str": + return str(o) + return o + try: + # First try datetime.datetime + return _serialize_datetime(o, format) + except AttributeError: + pass + # Last, try datetime.timedelta + try: + return _timedelta_as_isostr(o) + except AttributeError: + # This will be raised when it hits value.total_seconds in the method above + pass + return o + + +def _get_rest_field( + attr_to_rest_field: typing.Dict[str, "_RestField"], rest_name: str +) -> typing.Optional["_RestField"]: + try: + return next(rf for rf in attr_to_rest_field.values() if rf._rest_name == rest_name) + except StopIteration: + return None + + +def _create_value(rf: typing.Optional["_RestField"], value: typing.Any) -> typing.Any: + if not rf: + return _serialize(value, None) + if rf._is_multipart_file_input: + return value + if rf._is_model: + return _deserialize(rf._type, value) + if isinstance(value, ET.Element): + value = _deserialize(rf._type, value) + return _serialize(value, rf._format) + + +class Model(_MyMutableMapping): + _is_model = True + # label whether current class's _attr_to_rest_field has been calculated + # could not see _attr_to_rest_field directly because subclass inherits it from parent class + _calculated: typing.Set[str] = set() + + def __init__(self, *args: typing.Any, **kwargs: typing.Any) -> None: + class_name = self.__class__.__name__ + if len(args) > 1: + raise TypeError(f"{class_name}.__init__() takes 2 positional arguments but {len(args) + 1} were given") + dict_to_pass = { + rest_field._rest_name: rest_field._default + for rest_field in self._attr_to_rest_field.values() + if rest_field._default is not _UNSET + } + if args: # pylint: disable=too-many-nested-blocks + if isinstance(args[0], ET.Element): + existed_attr_keys = [] + model_meta = getattr(self, "_xml", {}) + + for rf in self._attr_to_rest_field.values(): + prop_meta = getattr(rf, "_xml", {}) + xml_name = prop_meta.get("name", rf._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + # attribute + if prop_meta.get("attribute", False) and args[0].get(xml_name) is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].get(xml_name)) + continue + + # unwrapped element is array + if prop_meta.get("unwrapped", False): + # unwrapped array could either use prop items meta/prop meta + if prop_meta.get("itemsName"): + xml_name = prop_meta.get("itemsName") + xml_ns = prop_meta.get("itemNs") + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + items = args[0].findall(xml_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, items) + continue + + # text element is primitive type + if prop_meta.get("text", False): + if args[0].text is not None: + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].text) + continue + + # wrapped element could be normal property or array, it should only have one element + item = args[0].find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, item) + + # rest thing is additional properties + for e in args[0]: + if e.tag not in existed_attr_keys: + dict_to_pass[e.tag] = _convert_element(e) + else: + dict_to_pass.update( + {k: _create_value(_get_rest_field(self._attr_to_rest_field, k), v) for k, v in args[0].items()} + ) + else: + non_attr_kwargs = [k for k in kwargs if k not in self._attr_to_rest_field] + if non_attr_kwargs: + # actual type errors only throw the first wrong keyword arg they see, so following that. + raise TypeError(f"{class_name}.__init__() got an unexpected keyword argument '{non_attr_kwargs[0]}'") + dict_to_pass.update( + { + self._attr_to_rest_field[k]._rest_name: _create_value(self._attr_to_rest_field[k], v) + for k, v in kwargs.items() + if v is not None + } + ) + super().__init__(dict_to_pass) + + def copy(self) -> "Model": + return Model(self.__dict__) + + def __new__(cls, *args: typing.Any, **kwargs: typing.Any) -> Self: + if f"{cls.__module__}.{cls.__qualname__}" not in cls._calculated: + # we know the last nine classes in mro are going to be 'Model', '_MyMutableMapping', 'MutableMapping', + # 'Mapping', 'Collection', 'Sized', 'Iterable', 'Container' and 'object' + mros = cls.__mro__[:-9][::-1] # ignore parents, and reverse the mro order + attr_to_rest_field: typing.Dict[str, _RestField] = { # map attribute name to rest_field property + k: v for mro_class in mros for k, v in mro_class.__dict__.items() if k[0] != "_" and hasattr(v, "_type") + } + annotations = { + k: v + for mro_class in mros + if hasattr(mro_class, "__annotations__") + for k, v in mro_class.__annotations__.items() + } + for attr, rf in attr_to_rest_field.items(): + rf._module = cls.__module__ + if not rf._type: + rf._type = rf._get_deserialize_callable_from_annotation(annotations.get(attr, None)) + if not rf._rest_name_input: + rf._rest_name_input = attr + cls._attr_to_rest_field: typing.Dict[str, _RestField] = dict(attr_to_rest_field.items()) + cls._calculated.add(f"{cls.__module__}.{cls.__qualname__}") + + return super().__new__(cls) + + def __init_subclass__(cls, discriminator: typing.Optional[str] = None) -> None: + for base in cls.__bases__: + if hasattr(base, "__mapping__"): + base.__mapping__[discriminator or cls.__name__] = cls # type: ignore + + @classmethod + def _get_discriminator(cls, exist_discriminators) -> typing.Optional["_RestField"]: + for v in cls.__dict__.values(): + if isinstance(v, _RestField) and v._is_discriminator and v._rest_name not in exist_discriminators: + return v + return None + + @classmethod + def _deserialize(cls, data, exist_discriminators): + if not hasattr(cls, "__mapping__"): + return cls(data) + discriminator = cls._get_discriminator(exist_discriminators) + if discriminator is None: + return cls(data) + exist_discriminators.append(discriminator._rest_name) + if isinstance(data, ET.Element): + model_meta = getattr(cls, "_xml", {}) + prop_meta = getattr(discriminator, "_xml", {}) + xml_name = prop_meta.get("name", discriminator._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + if data.get(xml_name) is not None: + discriminator_value = data.get(xml_name) + else: + discriminator_value = data.find(xml_name).text # pyright: ignore + else: + discriminator_value = data.get(discriminator._rest_name) + mapped_cls = cls.__mapping__.get(discriminator_value, cls) # pyright: ignore # pylint: disable=no-member + return mapped_cls._deserialize(data, exist_discriminators) + + def as_dict(self, *, exclude_readonly: bool = False) -> typing.Dict[str, typing.Any]: + """Return a dict that can be turned into json using json.dump. + + :keyword bool exclude_readonly: Whether to remove the readonly properties. + :returns: A dict JSON compatible object + :rtype: dict + """ + + result = {} + readonly_props = [] + if exclude_readonly: + readonly_props = [p._rest_name for p in self._attr_to_rest_field.values() if _is_readonly(p)] + for k, v in self.items(): + if exclude_readonly and k in readonly_props: # pyright: ignore + continue + is_multipart_file_input = False + try: + is_multipart_file_input = next( + rf for rf in self._attr_to_rest_field.values() if rf._rest_name == k + )._is_multipart_file_input + except StopIteration: + pass + result[k] = v if is_multipart_file_input else Model._as_dict_value(v, exclude_readonly=exclude_readonly) + return result + + @staticmethod + def _as_dict_value(v: typing.Any, exclude_readonly: bool = False) -> typing.Any: + if v is None or isinstance(v, _Null): + return None + if isinstance(v, (list, tuple, set)): + return type(v)(Model._as_dict_value(x, exclude_readonly=exclude_readonly) for x in v) + if isinstance(v, dict): + return {dk: Model._as_dict_value(dv, exclude_readonly=exclude_readonly) for dk, dv in v.items()} + return v.as_dict(exclude_readonly=exclude_readonly) if hasattr(v, "as_dict") else v + + +def _deserialize_model(model_deserializer: typing.Optional[typing.Callable], obj): + if _is_model(obj): + return obj + return _deserialize(model_deserializer, obj) + + +def _deserialize_with_optional(if_obj_deserializer: typing.Optional[typing.Callable], obj): + if obj is None: + return obj + return _deserialize_with_callable(if_obj_deserializer, obj) + + +def _deserialize_with_union(deserializers, obj): + for deserializer in deserializers: + try: + return _deserialize(deserializer, obj) + except DeserializationError: + pass + raise DeserializationError() + + +def _deserialize_dict( + value_deserializer: typing.Optional[typing.Callable], + module: typing.Optional[str], + obj: typing.Dict[typing.Any, typing.Any], +): + if obj is None: + return obj + if isinstance(obj, ET.Element): + obj = {child.tag: child for child in obj} + return {k: _deserialize(value_deserializer, v, module) for k, v in obj.items()} + + +def _deserialize_multiple_sequence( + entry_deserializers: typing.List[typing.Optional[typing.Callable]], + module: typing.Optional[str], + obj, +): + if obj is None: + return obj + return type(obj)(_deserialize(deserializer, entry, module) for entry, deserializer in zip(obj, entry_deserializers)) + + +def _deserialize_sequence( + deserializer: typing.Optional[typing.Callable], + module: typing.Optional[str], + obj, +): + if obj is None: + return obj + if isinstance(obj, ET.Element): + obj = list(obj) + return type(obj)(_deserialize(deserializer, entry, module) for entry in obj) + + +def _sorted_annotations(types: typing.List[typing.Any]) -> typing.List[typing.Any]: + return sorted( + types, + key=lambda x: hasattr(x, "__name__") and x.__name__.lower() in ("str", "float", "int", "bool"), + ) + + +def _get_deserialize_callable_from_annotation( # pylint: disable=too-many-return-statements, too-many-branches + annotation: typing.Any, + module: typing.Optional[str], + rf: typing.Optional["_RestField"] = None, +) -> typing.Optional[typing.Callable[[typing.Any], typing.Any]]: + if not annotation: + return None + + # is it a type alias? + if isinstance(annotation, str): + if module is not None: + annotation = _get_type_alias_type(module, annotation) + + # is it a forward ref / in quotes? + if isinstance(annotation, (str, typing.ForwardRef)): + try: + model_name = annotation.__forward_arg__ # type: ignore + except AttributeError: + model_name = annotation + if module is not None: + annotation = _get_model(module, model_name) # type: ignore + + try: + if module and _is_model(annotation): + if rf: + rf._is_model = True + + return functools.partial(_deserialize_model, annotation) # pyright: ignore + except Exception: + pass + + # is it a literal? + try: + if annotation.__origin__ is typing.Literal: # pyright: ignore + return None + except AttributeError: + pass + + # is it optional? + try: + if any(a for a in annotation.__args__ if a == type(None)): # pyright: ignore + if len(annotation.__args__) <= 2: # pyright: ignore + if_obj_deserializer = _get_deserialize_callable_from_annotation( + next(a for a in annotation.__args__ if a != type(None)), module, rf # pyright: ignore + ) + + return functools.partial(_deserialize_with_optional, if_obj_deserializer) + # the type is Optional[Union[...]], we need to remove the None type from the Union + annotation_copy = copy.copy(annotation) + annotation_copy.__args__ = [a for a in annotation_copy.__args__ if a != type(None)] # pyright: ignore + return _get_deserialize_callable_from_annotation(annotation_copy, module, rf) + except AttributeError: + pass + + # is it union? + if getattr(annotation, "__origin__", None) is typing.Union: + # initial ordering is we make `string` the last deserialization option, because it is often them most generic + deserializers = [ + _get_deserialize_callable_from_annotation(arg, module, rf) + for arg in _sorted_annotations(annotation.__args__) # pyright: ignore + ] + + return functools.partial(_deserialize_with_union, deserializers) + + try: + if annotation._name == "Dict": # pyright: ignore + value_deserializer = _get_deserialize_callable_from_annotation( + annotation.__args__[1], module, rf # pyright: ignore + ) + + return functools.partial( + _deserialize_dict, + value_deserializer, + module, + ) + except (AttributeError, IndexError): + pass + try: + if annotation._name in ["List", "Set", "Tuple", "Sequence"]: # pyright: ignore + if len(annotation.__args__) > 1: # pyright: ignore + entry_deserializers = [ + _get_deserialize_callable_from_annotation(dt, module, rf) + for dt in annotation.__args__ # pyright: ignore + ] + return functools.partial(_deserialize_multiple_sequence, entry_deserializers, module) + deserializer = _get_deserialize_callable_from_annotation( + annotation.__args__[0], module, rf # pyright: ignore + ) + + return functools.partial(_deserialize_sequence, deserializer, module) + except (TypeError, IndexError, AttributeError, SyntaxError): + pass + + def _deserialize_default( + deserializer, + obj, + ): + if obj is None: + return obj + try: + return _deserialize_with_callable(deserializer, obj) + except Exception: + pass + return obj + + if get_deserializer(annotation, rf): + return functools.partial(_deserialize_default, get_deserializer(annotation, rf)) + + return functools.partial(_deserialize_default, annotation) + + +def _deserialize_with_callable( + deserializer: typing.Optional[typing.Callable[[typing.Any], typing.Any]], + value: typing.Any, +): # pylint: disable=too-many-return-statements + try: + if value is None or isinstance(value, _Null): + return None + if isinstance(value, ET.Element): + if deserializer is str: + return value.text or "" + if deserializer is int: + return int(value.text) if value.text else None + if deserializer is float: + return float(value.text) if value.text else None + if deserializer is bool: + return value.text == "true" if value.text else None + if deserializer is None: + return value + if deserializer in [int, float, bool]: + return deserializer(value) + if isinstance(deserializer, CaseInsensitiveEnumMeta): + try: + return deserializer(value) + except ValueError: + # for unknown value, return raw value + return value + if isinstance(deserializer, type) and issubclass(deserializer, Model): + return deserializer._deserialize(value, []) + return typing.cast(typing.Callable[[typing.Any], typing.Any], deserializer)(value) + except Exception as e: + raise DeserializationError() from e + + +def _deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + if isinstance(value, PipelineResponse): + value = value.http_response.json() + if rf is None and format: + rf = _RestField(format=format) + if not isinstance(deserializer, functools.partial): + deserializer = _get_deserialize_callable_from_annotation(deserializer, module, rf) + return _deserialize_with_callable(deserializer, value) + + +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + +def _failsafe_deserialize_xml( + deserializer: typing.Any, + value: typing.Any, +) -> typing.Any: + try: + return _deserialize_xml(deserializer, value) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + +class _RestField: + def __init__( + self, + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + is_discriminator: bool = False, + visibility: typing.Optional[typing.List[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, + ): + self._type = type + self._rest_name_input = name + self._module: typing.Optional[str] = None + self._is_discriminator = is_discriminator + self._visibility = visibility + self._is_model = False + self._default = default + self._format = format + self._is_multipart_file_input = is_multipart_file_input + self._xml = xml if xml is not None else {} + + @property + def _class_type(self) -> typing.Any: + return getattr(self._type, "args", [None])[0] + + @property + def _rest_name(self) -> str: + if self._rest_name_input is None: + raise ValueError("Rest name was never set") + return self._rest_name_input + + def __get__(self, obj: Model, type=None): # pylint: disable=redefined-builtin + # by this point, type and rest_name will have a value bc we default + # them in __new__ of the Model class + item = obj.get(self._rest_name) + if item is None: + return item + if self._is_model: + return item + return _deserialize(self._type, _serialize(item, self._format), rf=self) + + def __set__(self, obj: Model, value) -> None: + if value is None: + # we want to wipe out entries if users set attr to None + try: + obj.__delitem__(self._rest_name) + except KeyError: + pass + return + if self._is_model: + if not _is_model(value): + value = _deserialize(self._type, value) + obj.__setitem__(self._rest_name, value) + return + obj.__setitem__(self._rest_name, _serialize(value, self._format)) + + def _get_deserialize_callable_from_annotation( + self, annotation: typing.Any + ) -> typing.Optional[typing.Callable[[typing.Any], typing.Any]]: + return _get_deserialize_callable_from_annotation(annotation, self._module, self) + + +def rest_field( + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + visibility: typing.Optional[typing.List[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, +) -> typing.Any: + return _RestField( + name=name, + type=type, + visibility=visibility, + default=default, + format=format, + is_multipart_file_input=is_multipart_file_input, + xml=xml, + ) + + +def rest_discriminator( + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + visibility: typing.Optional[typing.List[str]] = None, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, +) -> typing.Any: + return _RestField(name=name, type=type, is_discriminator=True, visibility=visibility, xml=xml) + + +def serialize_xml(model: Model, exclude_readonly: bool = False) -> str: + """Serialize a model to XML. + + :param Model model: The model to serialize. + :param bool exclude_readonly: Whether to exclude readonly properties. + :returns: The XML representation of the model. + :rtype: str + """ + return ET.tostring(_get_element(model, exclude_readonly), encoding="unicode") # type: ignore + + +def _get_element( + o: typing.Any, + exclude_readonly: bool = False, + parent_meta: typing.Optional[typing.Dict[str, typing.Any]] = None, + wrapped_element: typing.Optional[ET.Element] = None, +) -> typing.Union[ET.Element, typing.List[ET.Element]]: + if _is_model(o): + model_meta = getattr(o, "_xml", {}) + + # if prop is a model, then use the prop element directly, else generate a wrapper of model + if wrapped_element is None: + wrapped_element = _create_xml_element( + model_meta.get("name", o.__class__.__name__), + model_meta.get("prefix"), + model_meta.get("ns"), + ) + + readonly_props = [] + if exclude_readonly: + readonly_props = [p._rest_name for p in o._attr_to_rest_field.values() if _is_readonly(p)] + + for k, v in o.items(): + # do not serialize readonly properties + if exclude_readonly and k in readonly_props: + continue + + prop_rest_field = _get_rest_field(o._attr_to_rest_field, k) + if prop_rest_field: + prop_meta = getattr(prop_rest_field, "_xml").copy() + # use the wire name as xml name if no specific name is set + if prop_meta.get("name") is None: + prop_meta["name"] = k + else: + # additional properties will not have rest field, use the wire name as xml name + prop_meta = {"name": k} + + # if no ns for prop, use model's + if prop_meta.get("ns") is None and model_meta.get("ns"): + prop_meta["ns"] = model_meta.get("ns") + prop_meta["prefix"] = model_meta.get("prefix") + + if prop_meta.get("unwrapped", False): + # unwrapped could only set on array + wrapped_element.extend(_get_element(v, exclude_readonly, prop_meta)) + elif prop_meta.get("text", False): + # text could only set on primitive type + wrapped_element.text = _get_primitive_type_value(v) + elif prop_meta.get("attribute", False): + xml_name = prop_meta.get("name", k) + if prop_meta.get("ns"): + ET.register_namespace(prop_meta.get("prefix"), prop_meta.get("ns")) # pyright: ignore + xml_name = "{" + prop_meta.get("ns") + "}" + xml_name # pyright: ignore + # attribute should be primitive type + wrapped_element.set(xml_name, _get_primitive_type_value(v)) + else: + # other wrapped prop element + wrapped_element.append(_get_wrapped_element(v, exclude_readonly, prop_meta)) + return wrapped_element + if isinstance(o, list): + return [_get_element(x, exclude_readonly, parent_meta) for x in o] # type: ignore + if isinstance(o, dict): + result = [] + for k, v in o.items(): + result.append( + _get_wrapped_element( + v, + exclude_readonly, + { + "name": k, + "ns": parent_meta.get("ns") if parent_meta else None, + "prefix": parent_meta.get("prefix") if parent_meta else None, + }, + ) + ) + return result + + # primitive case need to create element based on parent_meta + if parent_meta: + return _get_wrapped_element( + o, + exclude_readonly, + { + "name": parent_meta.get("itemsName", parent_meta.get("name")), + "prefix": parent_meta.get("itemsPrefix", parent_meta.get("prefix")), + "ns": parent_meta.get("itemsNs", parent_meta.get("ns")), + }, + ) + + raise ValueError("Could not serialize value into xml: " + o) + + +def _get_wrapped_element( + v: typing.Any, + exclude_readonly: bool, + meta: typing.Optional[typing.Dict[str, typing.Any]], +) -> ET.Element: + wrapped_element = _create_xml_element( + meta.get("name") if meta else None, meta.get("prefix") if meta else None, meta.get("ns") if meta else None + ) + if isinstance(v, (dict, list)): + wrapped_element.extend(_get_element(v, exclude_readonly, meta)) + elif _is_model(v): + _get_element(v, exclude_readonly, meta, wrapped_element) + else: + wrapped_element.text = _get_primitive_type_value(v) + return wrapped_element + + +def _get_primitive_type_value(v) -> str: + if v is True: + return "true" + if v is False: + return "false" + if isinstance(v, _Null): + return "" + return str(v) + + +def _create_xml_element(tag, prefix=None, ns=None): + if prefix and ns: + ET.register_namespace(prefix, ns) + if ns: + return ET.Element("{" + ns + "}" + tag) + return ET.Element(tag) + + +def _deserialize_xml( + deserializer: typing.Any, + value: str, +) -> typing.Any: + element = ET.fromstring(value) # nosec + return _deserialize(deserializer, element) + + +def _convert_element(e: ET.Element): + # dict case + if len(e.attrib) > 0 or len({child.tag for child in e}) > 1: + dict_result: typing.Dict[str, typing.Any] = {} + for child in e: + if dict_result.get(child.tag) is not None: + if isinstance(dict_result[child.tag], list): + dict_result[child.tag].append(_convert_element(child)) + else: + dict_result[child.tag] = [dict_result[child.tag], _convert_element(child)] + else: + dict_result[child.tag] = _convert_element(child) + dict_result.update(e.attrib) + return dict_result + # array case + if len(e) > 0: + array_result: typing.List[typing.Any] = [] + for child in e: + array_result.append(_convert_element(child)) + return array_result + # primitive case + return e.text diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_serialization.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/serialization.py similarity index 83% rename from sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_serialization.py rename to sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/serialization.py index 8139854b97bb..eb86ea23c965 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_serialization.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_utils/serialization.py @@ -1,30 +1,12 @@ +# pylint: disable=line-too-long,useless-suppression,too-many-lines +# coding=utf-8 # -------------------------------------------------------------------------- -# # Copyright (c) Microsoft Corporation. All rights reserved. -# -# The MIT License (MIT) -# -# Permission is hereby granted, free of charge, to any person obtaining a copy -# of this software and associated documentation files (the ""Software""), to -# deal in the Software without restriction, including without limitation the -# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or -# sell copies of the Software, and to permit persons to whom the Software is -# furnished to do so, subject to the following conditions: -# -# The above copyright notice and this permission notice shall be included in -# all copies or substantial portions of the Software. -# -# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR -# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, -# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE -# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER -# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING -# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS -# IN THE SOFTWARE. -# +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -# pylint: skip-file # pyright: reportUnnecessaryTypeIgnoreComment=false from base64 import b64decode, b64encode @@ -48,11 +30,8 @@ IO, Mapping, Callable, - TypeVar, MutableMapping, - Type, List, - Mapping, ) try: @@ -62,13 +41,13 @@ import xml.etree.ElementTree as ET import isodate # type: ignore +from typing_extensions import Self from azure.core.exceptions import DeserializationError, SerializationError from azure.core.serialization import NULL as CoreNull _BOM = codecs.BOM_UTF8.decode(encoding="utf-8") -ModelType = TypeVar("ModelType", bound="Model") JSON = MutableMapping[str, Any] @@ -91,6 +70,8 @@ def deserialize_from_text(cls, data: Optional[Union[AnyStr, IO]], content_type: :param data: Input, could be bytes or stream (will be decoded with UTF8) or text :type data: str or bytes or IO :param str content_type: The content type. + :return: The deserialized data. + :rtype: object """ if hasattr(data, "read"): # Assume a stream @@ -112,7 +93,7 @@ def deserialize_from_text(cls, data: Optional[Union[AnyStr, IO]], content_type: try: return json.loads(data_as_str) except ValueError as err: - raise DeserializationError("JSON is invalid: {}".format(err), err) + raise DeserializationError("JSON is invalid: {}".format(err), err) from err elif "xml" in (content_type or []): try: @@ -155,6 +136,11 @@ def deserialize_from_http_generics(cls, body_bytes: Optional[Union[AnyStr, IO]], Use bytes and headers to NOT use any requests/aiohttp or whatever specific implementation. Headers will tested for "content-type" + + :param bytes body_bytes: The body of the response. + :param dict headers: The headers of the response. + :returns: The deserialized data. + :rtype: object """ # Try to use content-type from headers if available content_type = None @@ -179,80 +165,31 @@ def deserialize_from_http_generics(cls, body_bytes: Optional[Union[AnyStr, IO]], except NameError: _long_type = int - -class UTC(datetime.tzinfo): - """Time Zone info for handling UTC""" - - def utcoffset(self, dt): - """UTF offset for UTC is 0.""" - return datetime.timedelta(0) - - def tzname(self, dt): - """Timestamp representation.""" - return "Z" - - def dst(self, dt): - """No daylight saving for UTC.""" - return datetime.timedelta(hours=1) - - -try: - from datetime import timezone as _FixedOffset # type: ignore -except ImportError: # Python 2.7 - - class _FixedOffset(datetime.tzinfo): # type: ignore - """Fixed offset in minutes east from UTC. - Copy/pasted from Python doc - :param datetime.timedelta offset: offset in timedelta format - """ - - def __init__(self, offset): - self.__offset = offset - - def utcoffset(self, dt): - return self.__offset - - def tzname(self, dt): - return str(self.__offset.total_seconds() / 3600) - - def __repr__(self): - return "".format(self.tzname(None)) - - def dst(self, dt): - return datetime.timedelta(0) - - def __getinitargs__(self): - return (self.__offset,) - - -try: - from datetime import timezone - - TZ_UTC = timezone.utc -except ImportError: - TZ_UTC = UTC() # type: ignore +TZ_UTC = datetime.timezone.utc _FLATTEN = re.compile(r"(? None: self.additional_properties: Optional[Dict[str, Any]] = {} - for k in kwargs: + for k in kwargs: # pylint: disable=consider-using-dict-items if k not in self._attribute_map: _LOGGER.warning("%s is not a known attribute of class %s and will be ignored", k, self.__class__) elif k in self._validation and self._validation[k].get("readonly", False): @@ -300,13 +244,23 @@ def __init__(self, **kwargs: Any) -> None: setattr(self, k, kwargs[k]) def __eq__(self, other: Any) -> bool: - """Compare objects by comparing all attributes.""" + """Compare objects by comparing all attributes. + + :param object other: The object to compare + :returns: True if objects are equal + :rtype: bool + """ if isinstance(other, self.__class__): return self.__dict__ == other.__dict__ return False def __ne__(self, other: Any) -> bool: - """Compare objects by comparing all attributes.""" + """Compare objects by comparing all attributes. + + :param object other: The object to compare + :returns: True if objects are not equal + :rtype: bool + """ return not self.__eq__(other) def __str__(self) -> str: @@ -326,7 +280,11 @@ def is_xml_model(cls) -> bool: @classmethod def _create_xml_node(cls): - """Create XML node.""" + """Create XML node. + + :returns: The XML node + :rtype: xml.etree.ElementTree.Element + """ try: xml_map = cls._xml_map # type: ignore except AttributeError: @@ -346,7 +304,9 @@ def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: :rtype: dict """ serializer = Serializer(self._infer_class_models()) - return serializer._serialize(self, keep_readonly=keep_readonly, **kwargs) # type: ignore + return serializer._serialize( # type: ignore # pylint: disable=protected-access + self, keep_readonly=keep_readonly, **kwargs + ) def as_dict( self, @@ -380,12 +340,15 @@ def my_key_transformer(key, attr_desc, value): If you want XML serialization, you can pass the kwargs is_xml=True. + :param bool keep_readonly: If you want to serialize the readonly attributes :param function key_transformer: A key transformer function. :returns: A dict JSON compatible object :rtype: dict """ serializer = Serializer(self._infer_class_models()) - return serializer._serialize(self, key_transformer=key_transformer, keep_readonly=keep_readonly, **kwargs) # type: ignore + return serializer._serialize( # type: ignore # pylint: disable=protected-access + self, key_transformer=key_transformer, keep_readonly=keep_readonly, **kwargs + ) @classmethod def _infer_class_models(cls): @@ -395,30 +358,31 @@ def _infer_class_models(cls): client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)} if cls.__name__ not in client_models: raise ValueError("Not Autorest generated code") - except Exception: + except Exception: # pylint: disable=broad-exception-caught # Assume it's not Autorest generated (tests?). Add ourselves as dependencies. client_models = {cls.__name__: cls} return client_models @classmethod - def deserialize(cls: Type[ModelType], data: Any, content_type: Optional[str] = None) -> ModelType: + def deserialize(cls, data: Any, content_type: Optional[str] = None) -> Self: """Parse a str using the RestAPI syntax and return a model. :param str data: A str using RestAPI structure. JSON by default. :param str content_type: JSON by default, set application/xml if XML. :returns: An instance of this model - :raises: DeserializationError if something went wrong + :raises DeserializationError: if something went wrong + :rtype: Self """ deserializer = Deserializer(cls._infer_class_models()) return deserializer(cls.__name__, data, content_type=content_type) # type: ignore @classmethod def from_dict( - cls: Type[ModelType], + cls, data: Any, key_extractors: Optional[Callable[[str, Dict[str, Any], Any], Any]] = None, content_type: Optional[str] = None, - ) -> ModelType: + ) -> Self: """Parse a dict using given key extractor return a model. By default consider key @@ -426,9 +390,11 @@ def from_dict( and last_rest_key_case_insensitive_extractor) :param dict data: A dict using RestAPI structure + :param function key_extractors: A key extractor function. :param str content_type: JSON by default, set application/xml if XML. :returns: An instance of this model - :raises: DeserializationError if something went wrong + :raises DeserializationError: if something went wrong + :rtype: Self """ deserializer = Deserializer(cls._infer_class_models()) deserializer.key_extractors = ( # type: ignore @@ -448,21 +414,25 @@ def _flatten_subtype(cls, key, objects): return {} result = dict(cls._subtype_map[key]) for valuetype in cls._subtype_map[key].values(): - result.update(objects[valuetype]._flatten_subtype(key, objects)) + result.update(objects[valuetype]._flatten_subtype(key, objects)) # pylint: disable=protected-access return result @classmethod def _classify(cls, response, objects): """Check the class _subtype_map for any child classes. We want to ignore any inherited _subtype_maps. - Remove the polymorphic key from the initial data. + + :param dict response: The initial data + :param dict objects: The class objects + :returns: The class to be used + :rtype: class """ for subtype_key in cls.__dict__.get("_subtype_map", {}).keys(): subtype_value = None if not isinstance(response, ET.Element): rest_api_response_key = cls._get_rest_key_parts(subtype_key)[-1] - subtype_value = response.pop(rest_api_response_key, None) or response.pop(subtype_key, None) + subtype_value = response.get(rest_api_response_key, None) or response.get(subtype_key, None) else: subtype_value = xml_key_extractor(subtype_key, cls._attribute_map[subtype_key], response) if subtype_value: @@ -501,11 +471,13 @@ def _decode_attribute_map_key(key): inside the received data. :param str key: A key string from the generated code + :returns: The decoded key + :rtype: str """ return key.replace("\\.", ".") -class Serializer(object): +class Serializer: # pylint: disable=too-many-public-methods """Request object model serializer.""" basic_types = {str: "str", int: "int", bool: "bool", float: "float"} @@ -540,7 +512,7 @@ class Serializer(object): "multiple": lambda x, y: x % y != 0, } - def __init__(self, classes: Optional[Mapping[str, type]] = None): + def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: self.serialize_type = { "iso-8601": Serializer.serialize_iso, "rfc-1123": Serializer.serialize_rfc, @@ -560,13 +532,16 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None): self.key_transformer = full_restapi_key_transformer self.client_side_validation = True - def _serialize(self, target_obj, data_type=None, **kwargs): + def _serialize( # pylint: disable=too-many-nested-blocks, too-many-branches, too-many-statements, too-many-locals + self, target_obj, data_type=None, **kwargs + ): """Serialize data into a string according to type. - :param target_obj: The data to be serialized. + :param object target_obj: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str, dict - :raises: SerializationError if serialization fails. + :raises SerializationError: if serialization fails. + :returns: The serialized data. """ key_transformer = kwargs.get("key_transformer", self.key_transformer) keep_readonly = kwargs.get("keep_readonly", False) @@ -592,12 +567,14 @@ def _serialize(self, target_obj, data_type=None, **kwargs): serialized = {} if is_xml_model_serialization: - serialized = target_obj._create_xml_node() + serialized = target_obj._create_xml_node() # pylint: disable=protected-access try: - attributes = target_obj._attribute_map + attributes = target_obj._attribute_map # pylint: disable=protected-access for attr, attr_desc in attributes.items(): attr_name = attr - if not keep_readonly and target_obj._validation.get(attr_name, {}).get("readonly", False): + if not keep_readonly and target_obj._validation.get( # pylint: disable=protected-access + attr_name, {} + ).get("readonly", False): continue if attr_name == "additional_properties" and attr_desc["key"] == "": @@ -633,7 +610,8 @@ def _serialize(self, target_obj, data_type=None, **kwargs): if isinstance(new_attr, list): serialized.extend(new_attr) # type: ignore elif isinstance(new_attr, ET.Element): - # If the down XML has no XML/Name, we MUST replace the tag with the local tag. But keeping the namespaces. + # If the down XML has no XML/Name, + # we MUST replace the tag with the local tag. But keeping the namespaces. if "name" not in getattr(orig_attr, "_xml_map", {}): splitted_tag = new_attr.tag.split("}") if len(splitted_tag) == 2: # Namespace @@ -664,17 +642,17 @@ def _serialize(self, target_obj, data_type=None, **kwargs): except (AttributeError, KeyError, TypeError) as err: msg = "Attribute {} in object {} cannot be serialized.\n{}".format(attr_name, class_name, str(target_obj)) raise SerializationError(msg) from err - else: - return serialized + return serialized def body(self, data, data_type, **kwargs): """Serialize data intended for a request body. - :param data: The data to be serialized. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: dict - :raises: SerializationError if serialization fails. - :raises: ValueError if data is None + :raises SerializationError: if serialization fails. + :raises ValueError: if data is None + :returns: The serialized request body """ # Just in case this is a dict @@ -703,7 +681,7 @@ def body(self, data, data_type, **kwargs): attribute_key_case_insensitive_extractor, last_rest_key_case_insensitive_extractor, ] - data = deserializer._deserialize(data_type, data) + data = deserializer._deserialize(data_type, data) # pylint: disable=protected-access except DeserializationError as err: raise SerializationError("Unable to build a model: " + str(err)) from err @@ -712,11 +690,13 @@ def body(self, data, data_type, **kwargs): def url(self, name, data, data_type, **kwargs): """Serialize data intended for a URL path. - :param data: The data to be serialized. + :param str name: The name of the URL path parameter. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str - :raises: TypeError if serialization fails. - :raises: ValueError if data is None + :returns: The serialized URL path + :raises TypeError: if serialization fails. + :raises ValueError: if data is None """ try: output = self.serialize_data(data, data_type, **kwargs) @@ -728,21 +708,20 @@ def url(self, name, data, data_type, **kwargs): output = output.replace("{", quote("{")).replace("}", quote("}")) else: output = quote(str(output), safe="") - except SerializationError: - raise TypeError("{} must be type {}.".format(name, data_type)) - else: - return output + except SerializationError as exc: + raise TypeError("{} must be type {}.".format(name, data_type)) from exc + return output def query(self, name, data, data_type, **kwargs): """Serialize data intended for a URL query. - :param data: The data to be serialized. + :param str name: The name of the query parameter. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. - :keyword bool skip_quote: Whether to skip quote the serialized result. - Defaults to False. :rtype: str, list - :raises: TypeError if serialization fails. - :raises: ValueError if data is None + :raises TypeError: if serialization fails. + :raises ValueError: if data is None + :returns: The serialized query parameter """ try: # Treat the list aside, since we don't want to encode the div separator @@ -759,19 +738,20 @@ def query(self, name, data, data_type, **kwargs): output = str(output) else: output = quote(str(output), safe="") - except SerializationError: - raise TypeError("{} must be type {}.".format(name, data_type)) - else: - return str(output) + except SerializationError as exc: + raise TypeError("{} must be type {}.".format(name, data_type)) from exc + return str(output) def header(self, name, data, data_type, **kwargs): """Serialize data intended for a request header. - :param data: The data to be serialized. + :param str name: The name of the header. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str - :raises: TypeError if serialization fails. - :raises: ValueError if data is None + :raises TypeError: if serialization fails. + :raises ValueError: if data is None + :returns: The serialized header """ try: if data_type in ["[str]"]: @@ -780,21 +760,20 @@ def header(self, name, data, data_type, **kwargs): output = self.serialize_data(data, data_type, **kwargs) if data_type == "bool": output = json.dumps(output) - except SerializationError: - raise TypeError("{} must be type {}.".format(name, data_type)) - else: - return str(output) + except SerializationError as exc: + raise TypeError("{} must be type {}.".format(name, data_type)) from exc + return str(output) def serialize_data(self, data, data_type, **kwargs): """Serialize generic data according to supplied data type. - :param data: The data to be serialized. + :param object data: The data to be serialized. :param str data_type: The type to be serialized from. - :param bool required: Whether it's essential that the data not be - empty or None - :raises: AttributeError if required data is None. - :raises: ValueError if data is None - :raises: SerializationError if serialization fails. + :raises AttributeError: if required data is None. + :raises ValueError: if data is None + :raises SerializationError: if serialization fails. + :returns: The serialized data. + :rtype: str, int, float, bool, dict, list """ if data is None: raise ValueError("No value for given attribute") @@ -805,7 +784,7 @@ def serialize_data(self, data, data_type, **kwargs): if data_type in self.basic_types.values(): return self.serialize_basic(data, data_type, **kwargs) - elif data_type in self.serialize_type: + if data_type in self.serialize_type: return self.serialize_type[data_type](data, **kwargs) # If dependencies is empty, try with current data class @@ -821,11 +800,10 @@ def serialize_data(self, data, data_type, **kwargs): except (ValueError, TypeError) as err: msg = "Unable to serialize value: {!r} as type: {!r}." raise SerializationError(msg.format(data, data_type)) from err - else: - return self._serialize(data, **kwargs) + return self._serialize(data, **kwargs) @classmethod - def _get_custom_serializers(cls, data_type, **kwargs): + def _get_custom_serializers(cls, data_type, **kwargs): # pylint: disable=inconsistent-return-statements custom_serializer = kwargs.get("basic_types_serializers", {}).get(data_type) if custom_serializer: return custom_serializer @@ -841,23 +819,26 @@ def serialize_basic(cls, data, data_type, **kwargs): - basic_types_serializers dict[str, callable] : If set, use the callable as serializer - is_xml bool : If set, use xml_basic_types_serializers - :param data: Object to be serialized. + :param obj data: Object to be serialized. :param str data_type: Type of object in the iterable. + :rtype: str, int, float, bool + :return: serialized object """ custom_serializer = cls._get_custom_serializers(data_type, **kwargs) if custom_serializer: return custom_serializer(data) if data_type == "str": return cls.serialize_unicode(data) - return eval(data_type)(data) # nosec + return eval(data_type)(data) # nosec # pylint: disable=eval-used @classmethod def serialize_unicode(cls, data): """Special handling for serializing unicode strings in Py2. Encode to UTF-8 if unicode, otherwise handle as a str. - :param data: Object to be serialized. + :param str data: Object to be serialized. :rtype: str + :return: serialized object """ try: # If I received an enum, return its value return data.value @@ -871,8 +852,7 @@ def serialize_unicode(cls, data): return data except NameError: return str(data) - else: - return str(data) + return str(data) def serialize_iter(self, data, iter_type, div=None, **kwargs): """Serialize iterable. @@ -882,15 +862,13 @@ def serialize_iter(self, data, iter_type, div=None, **kwargs): serialization_ctxt['type'] should be same as data_type. - is_xml bool : If set, serialize as XML - :param list attr: Object to be serialized. + :param list data: Object to be serialized. :param str iter_type: Type of object in the iterable. - :param bool required: Whether the objects in the iterable must - not be None or empty. :param str div: If set, this str will be used to combine the elements in the iterable into a combined string. Default is 'None'. - :keyword bool do_quote: Whether to quote the serialized result of each iterable element. Defaults to False. :rtype: list, str + :return: serialized iterable """ if isinstance(data, str): raise SerializationError("Refuse str type as a valid iter type.") @@ -945,9 +923,8 @@ def serialize_dict(self, attr, dict_type, **kwargs): :param dict attr: Object to be serialized. :param str dict_type: Type of object in the dictionary. - :param bool required: Whether the objects in the dictionary must - not be None or empty. :rtype: dict + :return: serialized dictionary """ serialization_ctxt = kwargs.get("serialization_ctxt", {}) serialized = {} @@ -971,7 +948,7 @@ def serialize_dict(self, attr, dict_type, **kwargs): return serialized - def serialize_object(self, attr, **kwargs): + def serialize_object(self, attr, **kwargs): # pylint: disable=too-many-return-statements """Serialize a generic object. This will be handled as a dictionary. If object passed in is not a basic type (str, int, float, dict, list) it will simply be @@ -979,6 +956,7 @@ def serialize_object(self, attr, **kwargs): :param dict attr: Object to be serialized. :rtype: dict or str + :return: serialized object """ if attr is None: return None @@ -1003,7 +981,7 @@ def serialize_object(self, attr, **kwargs): return self.serialize_decimal(attr) # If it's a model or I know this dependency, serialize as a Model - elif obj_type in self.dependencies.values() or isinstance(attr, Model): + if obj_type in self.dependencies.values() or isinstance(attr, Model): return self._serialize(attr) if obj_type == dict: @@ -1034,56 +1012,61 @@ def serialize_enum(attr, enum_obj=None): try: enum_obj(result) # type: ignore return result - except ValueError: + except ValueError as exc: for enum_value in enum_obj: # type: ignore if enum_value.value.lower() == str(attr).lower(): return enum_value.value error = "{!r} is not valid value for enum {!r}" - raise SerializationError(error.format(attr, enum_obj)) + raise SerializationError(error.format(attr, enum_obj)) from exc @staticmethod - def serialize_bytearray(attr, **kwargs): + def serialize_bytearray(attr, **kwargs): # pylint: disable=unused-argument """Serialize bytearray into base-64 string. - :param attr: Object to be serialized. + :param str attr: Object to be serialized. :rtype: str + :return: serialized base64 """ return b64encode(attr).decode() @staticmethod - def serialize_base64(attr, **kwargs): + def serialize_base64(attr, **kwargs): # pylint: disable=unused-argument """Serialize str into base-64 string. - :param attr: Object to be serialized. + :param str attr: Object to be serialized. :rtype: str + :return: serialized base64 """ encoded = b64encode(attr).decode("ascii") return encoded.strip("=").replace("+", "-").replace("/", "_") @staticmethod - def serialize_decimal(attr, **kwargs): + def serialize_decimal(attr, **kwargs): # pylint: disable=unused-argument """Serialize Decimal object to float. - :param attr: Object to be serialized. + :param decimal attr: Object to be serialized. :rtype: float + :return: serialized decimal """ return float(attr) @staticmethod - def serialize_long(attr, **kwargs): + def serialize_long(attr, **kwargs): # pylint: disable=unused-argument """Serialize long (Py2) or int (Py3). - :param attr: Object to be serialized. + :param int attr: Object to be serialized. :rtype: int/long + :return: serialized long """ return _long_type(attr) @staticmethod - def serialize_date(attr, **kwargs): + def serialize_date(attr, **kwargs): # pylint: disable=unused-argument """Serialize Date object into ISO-8601 formatted string. :param Date attr: Object to be serialized. :rtype: str + :return: serialized date """ if isinstance(attr, str): attr = isodate.parse_date(attr) @@ -1091,11 +1074,12 @@ def serialize_date(attr, **kwargs): return t @staticmethod - def serialize_time(attr, **kwargs): + def serialize_time(attr, **kwargs): # pylint: disable=unused-argument """Serialize Time object into ISO-8601 formatted string. :param datetime.time attr: Object to be serialized. :rtype: str + :return: serialized time """ if isinstance(attr, str): attr = isodate.parse_time(attr) @@ -1105,30 +1089,32 @@ def serialize_time(attr, **kwargs): return t @staticmethod - def serialize_duration(attr, **kwargs): + def serialize_duration(attr, **kwargs): # pylint: disable=unused-argument """Serialize TimeDelta object into ISO-8601 formatted string. :param TimeDelta attr: Object to be serialized. :rtype: str + :return: serialized duration """ if isinstance(attr, str): attr = isodate.parse_duration(attr) return isodate.duration_isoformat(attr) @staticmethod - def serialize_rfc(attr, **kwargs): + def serialize_rfc(attr, **kwargs): # pylint: disable=unused-argument """Serialize Datetime object into RFC-1123 formatted string. :param Datetime attr: Object to be serialized. :rtype: str - :raises: TypeError if format invalid. + :raises TypeError: if format invalid. + :return: serialized rfc """ try: if not attr.tzinfo: _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") utc = attr.utctimetuple() - except AttributeError: - raise TypeError("RFC1123 object must be valid Datetime object.") + except AttributeError as exc: + raise TypeError("RFC1123 object must be valid Datetime object.") from exc return "{}, {:02} {} {:04} {:02}:{:02}:{:02} GMT".format( Serializer.days[utc.tm_wday], @@ -1141,12 +1127,13 @@ def serialize_rfc(attr, **kwargs): ) @staticmethod - def serialize_iso(attr, **kwargs): + def serialize_iso(attr, **kwargs): # pylint: disable=unused-argument """Serialize Datetime object into ISO-8601 formatted string. :param Datetime attr: Object to be serialized. :rtype: str - :raises: SerializationError if format invalid. + :raises SerializationError: if format invalid. + :return: serialized iso """ if isinstance(attr, str): attr = isodate.parse_datetime(attr) @@ -1172,13 +1159,14 @@ def serialize_iso(attr, **kwargs): raise TypeError(msg) from err @staticmethod - def serialize_unix(attr, **kwargs): + def serialize_unix(attr, **kwargs): # pylint: disable=unused-argument """Serialize Datetime object into IntTime format. This is represented as seconds. :param Datetime attr: Object to be serialized. :rtype: int - :raises: SerializationError if format invalid + :raises SerializationError: if format invalid + :return: serialied unix """ if isinstance(attr, int): return attr @@ -1186,11 +1174,11 @@ def serialize_unix(attr, **kwargs): if not attr.tzinfo: _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") return int(calendar.timegm(attr.utctimetuple())) - except AttributeError: - raise TypeError("Unix time object must be valid Datetime object.") + except AttributeError as exc: + raise TypeError("Unix time object must be valid Datetime object.") from exc -def rest_key_extractor(attr, attr_desc, data): +def rest_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argument key = attr_desc["key"] working_data = data @@ -1211,7 +1199,9 @@ def rest_key_extractor(attr, attr_desc, data): return working_data.get(key) -def rest_key_case_insensitive_extractor(attr, attr_desc, data): +def rest_key_case_insensitive_extractor( # pylint: disable=unused-argument, inconsistent-return-statements + attr, attr_desc, data +): key = attr_desc["key"] working_data = data @@ -1232,17 +1222,29 @@ def rest_key_case_insensitive_extractor(attr, attr_desc, data): return attribute_key_case_insensitive_extractor(key, None, working_data) -def last_rest_key_extractor(attr, attr_desc, data): - """Extract the attribute in "data" based on the last part of the JSON path key.""" +def last_rest_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argument + """Extract the attribute in "data" based on the last part of the JSON path key. + + :param str attr: The attribute to extract + :param dict attr_desc: The attribute description + :param dict data: The data to extract from + :rtype: object + :returns: The extracted attribute + """ key = attr_desc["key"] dict_keys = _FLATTEN.split(key) return attribute_key_extractor(dict_keys[-1], None, data) -def last_rest_key_case_insensitive_extractor(attr, attr_desc, data): +def last_rest_key_case_insensitive_extractor(attr, attr_desc, data): # pylint: disable=unused-argument """Extract the attribute in "data" based on the last part of the JSON path key. This is the case insensitive version of "last_rest_key_extractor" + :param str attr: The attribute to extract + :param dict attr_desc: The attribute description + :param dict data: The data to extract from + :rtype: object + :returns: The extracted attribute """ key = attr_desc["key"] dict_keys = _FLATTEN.split(key) @@ -1279,7 +1281,7 @@ def _extract_name_from_internal_type(internal_type): return xml_name -def xml_key_extractor(attr, attr_desc, data): +def xml_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argument,too-many-return-statements if isinstance(data, dict): return None @@ -1331,22 +1333,21 @@ def xml_key_extractor(attr, attr_desc, data): if is_iter_type: if is_wrapped: return None # is_wrapped no node, we want None - else: - return [] # not wrapped, assume empty list + return [] # not wrapped, assume empty list return None # Assume it's not there, maybe an optional node. # If is_iter_type and not wrapped, return all found children if is_iter_type: if not is_wrapped: return children - else: # Iter and wrapped, should have found one node only (the wrap one) - if len(children) != 1: - raise DeserializationError( - "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format( - xml_name - ) + # Iter and wrapped, should have found one node only (the wrap one) + if len(children) != 1: + raise DeserializationError( + "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format( + xml_name ) - return list(children[0]) # Might be empty list and that's ok. + ) + return list(children[0]) # Might be empty list and that's ok. # Here it's not a itertype, we should have found one element only or empty if len(children) > 1: @@ -1354,7 +1355,7 @@ def xml_key_extractor(attr, attr_desc, data): return children[0] -class Deserializer(object): +class Deserializer: """Response object model deserializer. :param dict classes: Class type dictionary for deserializing complex types. @@ -1363,9 +1364,9 @@ class Deserializer(object): basic_types = {str: "str", int: "int", bool: "bool", float: "float"} - valid_date = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}" r"\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") + valid_date = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") - def __init__(self, classes: Optional[Mapping[str, type]] = None): + def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: self.deserialize_type = { "iso-8601": Deserializer.deserialize_iso, "rfc-1123": Deserializer.deserialize_rfc, @@ -1401,27 +1402,29 @@ def __call__(self, target_obj, response_data, content_type=None): :param str target_obj: Target data type to deserialize to. :param requests.Response response_data: REST response object. :param str content_type: Swagger "produces" if available. - :raises: DeserializationError if deserialization fails. + :raises DeserializationError: if deserialization fails. :return: Deserialized object. + :rtype: object """ data = self._unpack_content(response_data, content_type) return self._deserialize(target_obj, data) - def _deserialize(self, target_obj, data): + def _deserialize(self, target_obj, data): # pylint: disable=inconsistent-return-statements """Call the deserializer on a model. Data needs to be already deserialized as JSON or XML ElementTree :param str target_obj: Target data type to deserialize to. :param object data: Object to deserialize. - :raises: DeserializationError if deserialization fails. + :raises DeserializationError: if deserialization fails. :return: Deserialized object. + :rtype: object """ # This is already a model, go recursive just in case if hasattr(data, "_attribute_map"): constants = [name for name, config in getattr(data, "_validation", {}).items() if config.get("constant")] try: - for attr, mapconfig in data._attribute_map.items(): + for attr, mapconfig in data._attribute_map.items(): # pylint: disable=protected-access if attr in constants: continue value = getattr(data, attr) @@ -1440,13 +1443,13 @@ def _deserialize(self, target_obj, data): if isinstance(response, str): return self.deserialize_data(data, response) - elif isinstance(response, type) and issubclass(response, Enum): + if isinstance(response, type) and issubclass(response, Enum): return self.deserialize_enum(data, response) if data is None or data is CoreNull: return data try: - attributes = response._attribute_map # type: ignore + attributes = response._attribute_map # type: ignore # pylint: disable=protected-access d_attrs = {} for attr, attr_desc in attributes.items(): # Check empty string. If it's not empty, someone has a real "additionalProperties"... @@ -1476,9 +1479,8 @@ def _deserialize(self, target_obj, data): except (AttributeError, TypeError, KeyError) as err: msg = "Unable to deserialize to object: " + class_name # type: ignore raise DeserializationError(msg) from err - else: - additional_properties = self._build_additional_properties(attributes, data) - return self._instantiate_model(response, d_attrs, additional_properties) + additional_properties = self._build_additional_properties(attributes, data) + return self._instantiate_model(response, d_attrs, additional_properties) def _build_additional_properties(self, attribute_map, data): if not self.additional_properties_detection: @@ -1505,6 +1507,8 @@ def _classify_target(self, target, data): :param str target: The target object type to deserialize to. :param str/dict data: The response data to deserialize. + :return: The classified target object and its class name. + :rtype: tuple """ if target is None: return None, None @@ -1516,7 +1520,7 @@ def _classify_target(self, target, data): return target, target try: - target = target._classify(data, self.dependencies) # type: ignore + target = target._classify(data, self.dependencies) # type: ignore # pylint: disable=protected-access except AttributeError: pass # Target is not a Model, no classify return target, target.__class__.__name__ # type: ignore @@ -1531,10 +1535,12 @@ def failsafe_deserialize(self, target_obj, data, content_type=None): :param str target_obj: The target object type to deserialize to. :param str/dict data: The response data to deserialize. :param str content_type: Swagger "produces" if available. + :return: Deserialized object. + :rtype: object """ try: return self(target_obj, data, content_type=content_type) - except: + except: # pylint: disable=bare-except _LOGGER.debug( "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True ) @@ -1552,10 +1558,12 @@ def _unpack_content(raw_data, content_type=None): If raw_data is something else, bypass all logic and return it directly. - :param raw_data: Data to be processed. - :param content_type: How to parse if raw_data is a string/bytes. + :param obj raw_data: Data to be processed. + :param str content_type: How to parse if raw_data is a string/bytes. :raises JSONDecodeError: If JSON is requested and parsing is impossible. :raises UnicodeDecodeError: If bytes is not UTF8 + :rtype: object + :return: Unpacked content. """ # Assume this is enough to detect a Pipeline Response without importing it context = getattr(raw_data, "context", {}) @@ -1579,24 +1587,35 @@ def _unpack_content(raw_data, content_type=None): def _instantiate_model(self, response, attrs, additional_properties=None): """Instantiate a response model passing in deserialized args. - :param response: The response model class. - :param d_attrs: The deserialized response attributes. + :param Response response: The response model class. + :param dict attrs: The deserialized response attributes. + :param dict additional_properties: Additional properties to be set. + :rtype: Response + :return: The instantiated response model. """ if callable(response): subtype = getattr(response, "_subtype_map", {}) try: - readonly = [k for k, v in response._validation.items() if v.get("readonly")] - const = [k for k, v in response._validation.items() if v.get("constant")] + readonly = [ + k + for k, v in response._validation.items() # pylint: disable=protected-access # type: ignore + if v.get("readonly") + ] + const = [ + k + for k, v in response._validation.items() # pylint: disable=protected-access # type: ignore + if v.get("constant") + ] kwargs = {k: v for k, v in attrs.items() if k not in subtype and k not in readonly + const} response_obj = response(**kwargs) for attr in readonly: setattr(response_obj, attr, attrs.get(attr)) if additional_properties: - response_obj.additional_properties = additional_properties + response_obj.additional_properties = additional_properties # type: ignore return response_obj except TypeError as err: msg = "Unable to deserialize {} into model {}. ".format(kwargs, response) # type: ignore - raise DeserializationError(msg + str(err)) + raise DeserializationError(msg + str(err)) from err else: try: for attr, value in attrs.items(): @@ -1605,15 +1624,16 @@ def _instantiate_model(self, response, attrs, additional_properties=None): except Exception as exp: msg = "Unable to populate response model. " msg += "Type: {}, Error: {}".format(type(response), exp) - raise DeserializationError(msg) + raise DeserializationError(msg) from exp - def deserialize_data(self, data, data_type): + def deserialize_data(self, data, data_type): # pylint: disable=too-many-return-statements """Process data for deserialization according to data type. :param str data: The response string to be deserialized. :param str data_type: The type to deserialize to. - :raises: DeserializationError if deserialization fails. + :raises DeserializationError: if deserialization fails. :return: Deserialized object. + :rtype: object """ if data is None: return data @@ -1627,7 +1647,11 @@ def deserialize_data(self, data, data_type): if isinstance(data, self.deserialize_expected_types.get(data_type, tuple())): return data - is_a_text_parsing_type = lambda x: x not in ["object", "[]", r"{}"] + is_a_text_parsing_type = lambda x: x not in [ # pylint: disable=unnecessary-lambda-assignment + "object", + "[]", + r"{}", + ] if isinstance(data, ET.Element) and is_a_text_parsing_type(data_type) and not data.text: return None data_val = self.deserialize_type[data_type](data) @@ -1647,14 +1671,14 @@ def deserialize_data(self, data, data_type): msg = "Unable to deserialize response data." msg += " Data: {}, {}".format(data, data_type) raise DeserializationError(msg) from err - else: - return self._deserialize(obj_type, data) + return self._deserialize(obj_type, data) def deserialize_iter(self, attr, iter_type): """Deserialize an iterable. :param list attr: Iterable to be deserialized. :param str iter_type: The type of object in the iterable. + :return: Deserialized iterable. :rtype: list """ if attr is None: @@ -1671,6 +1695,7 @@ def deserialize_dict(self, attr, dict_type): :param dict/list attr: Dictionary to be deserialized. Also accepts a list of key, value pairs. :param str dict_type: The object type of the items in the dictionary. + :return: Deserialized dictionary. :rtype: dict """ if isinstance(attr, list): @@ -1681,13 +1706,14 @@ def deserialize_dict(self, attr, dict_type): attr = {el.tag: el.text for el in attr} return {k: self.deserialize_data(v, dict_type) for k, v in attr.items()} - def deserialize_object(self, attr, **kwargs): + def deserialize_object(self, attr, **kwargs): # pylint: disable=too-many-return-statements """Deserialize a generic object. This will be handled as a dictionary. :param dict attr: Dictionary to be deserialized. + :return: Deserialized object. :rtype: dict - :raises: TypeError if non-builtin datatype encountered. + :raises TypeError: if non-builtin datatype encountered. """ if attr is None: return None @@ -1720,11 +1746,10 @@ def deserialize_object(self, attr, **kwargs): pass return deserialized - else: - error = "Cannot deserialize generic object with type: " - raise TypeError(error + str(obj_type)) + error = "Cannot deserialize generic object with type: " + raise TypeError(error + str(obj_type)) - def deserialize_basic(self, attr, data_type): + def deserialize_basic(self, attr, data_type): # pylint: disable=too-many-return-statements """Deserialize basic builtin data type from string. Will attempt to convert to str, int, float and bool. This function will also accept '1', '0', 'true' and 'false' as @@ -1732,8 +1757,9 @@ def deserialize_basic(self, attr, data_type): :param str attr: response string to be deserialized. :param str data_type: deserialization data type. + :return: Deserialized basic type. :rtype: str, int, float or bool - :raises: TypeError if string format is not valid. + :raises TypeError: if string format is not valid. """ # If we're here, data is supposed to be a basic type. # If it's still an XML node, take the text @@ -1743,24 +1769,23 @@ def deserialize_basic(self, attr, data_type): if data_type == "str": # None or '', node is empty string. return "" - else: - # None or '', node with a strong type is None. - # Don't try to model "empty bool" or "empty int" - return None + # None or '', node with a strong type is None. + # Don't try to model "empty bool" or "empty int" + return None if data_type == "bool": if attr in [True, False, 1, 0]: return bool(attr) - elif isinstance(attr, str): + if isinstance(attr, str): if attr.lower() in ["true", "1"]: return True - elif attr.lower() in ["false", "0"]: + if attr.lower() in ["false", "0"]: return False raise TypeError("Invalid boolean value: {}".format(attr)) if data_type == "str": return self.deserialize_unicode(attr) - return eval(data_type)(attr) # nosec + return eval(data_type)(attr) # nosec # pylint: disable=eval-used @staticmethod def deserialize_unicode(data): @@ -1768,6 +1793,7 @@ def deserialize_unicode(data): as a string. :param str data: response string to be deserialized. + :return: Deserialized string. :rtype: str or unicode """ # We might be here because we have an enum modeled as string, @@ -1781,8 +1807,7 @@ def deserialize_unicode(data): return data except NameError: return str(data) - else: - return str(data) + return str(data) @staticmethod def deserialize_enum(data, enum_obj): @@ -1794,6 +1819,7 @@ def deserialize_enum(data, enum_obj): :param str data: Response string to be deserialized. If this value is None or invalid it will be returned as-is. :param Enum enum_obj: Enum object to deserialize to. + :return: Deserialized enum object. :rtype: Enum """ if isinstance(data, enum_obj) or data is None: @@ -1804,9 +1830,9 @@ def deserialize_enum(data, enum_obj): # Workaround. We might consider remove it in the future. try: return list(enum_obj.__members__.values())[data] - except IndexError: + except IndexError as exc: error = "{!r} is not a valid index for enum {!r}" - raise DeserializationError(error.format(data, enum_obj)) + raise DeserializationError(error.format(data, enum_obj)) from exc try: return enum_obj(str(data)) except ValueError: @@ -1822,8 +1848,9 @@ def deserialize_bytearray(attr): """Deserialize string into bytearray. :param str attr: response string to be deserialized. + :return: Deserialized bytearray :rtype: bytearray - :raises: TypeError if string format invalid. + :raises TypeError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1834,8 +1861,9 @@ def deserialize_base64(attr): """Deserialize base64 encoded string into string. :param str attr: response string to be deserialized. + :return: Deserialized base64 string :rtype: bytearray - :raises: TypeError if string format invalid. + :raises TypeError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1849,8 +1877,9 @@ def deserialize_decimal(attr): """Deserialize string into Decimal object. :param str attr: response string to be deserialized. - :rtype: Decimal - :raises: DeserializationError if string format invalid. + :return: Deserialized decimal + :raises DeserializationError: if string format invalid. + :rtype: decimal """ if isinstance(attr, ET.Element): attr = attr.text @@ -1865,8 +1894,9 @@ def deserialize_long(attr): """Deserialize string into long (Py2) or int (Py3). :param str attr: response string to be deserialized. + :return: Deserialized int :rtype: long or int - :raises: ValueError if string format invalid. + :raises ValueError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1877,8 +1907,9 @@ def deserialize_duration(attr): """Deserialize ISO-8601 formatted string into TimeDelta object. :param str attr: response string to be deserialized. + :return: Deserialized duration :rtype: TimeDelta - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1887,16 +1918,16 @@ def deserialize_duration(attr): except (ValueError, OverflowError, AttributeError) as err: msg = "Cannot deserialize duration object." raise DeserializationError(msg) from err - else: - return duration + return duration @staticmethod def deserialize_date(attr): """Deserialize ISO-8601 formatted string into Date object. :param str attr: response string to be deserialized. + :return: Deserialized date :rtype: Date - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1910,8 +1941,9 @@ def deserialize_time(attr): """Deserialize ISO-8601 formatted string into time object. :param str attr: response string to be deserialized. + :return: Deserialized time :rtype: datetime.time - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1924,31 +1956,32 @@ def deserialize_rfc(attr): """Deserialize RFC-1123 formatted string into Datetime object. :param str attr: response string to be deserialized. + :return: Deserialized RFC datetime :rtype: Datetime - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text try: parsed_date = email.utils.parsedate_tz(attr) # type: ignore date_obj = datetime.datetime( - *parsed_date[:6], tzinfo=_FixedOffset(datetime.timedelta(minutes=(parsed_date[9] or 0) / 60)) + *parsed_date[:6], tzinfo=datetime.timezone(datetime.timedelta(minutes=(parsed_date[9] or 0) / 60)) ) if not date_obj.tzinfo: date_obj = date_obj.astimezone(tz=TZ_UTC) except ValueError as err: msg = "Cannot deserialize to rfc datetime object." raise DeserializationError(msg) from err - else: - return date_obj + return date_obj @staticmethod def deserialize_iso(attr): """Deserialize ISO-8601 formatted string into Datetime object. :param str attr: response string to be deserialized. + :return: Deserialized ISO datetime :rtype: Datetime - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1976,8 +2009,7 @@ def deserialize_iso(attr): except (ValueError, OverflowError, AttributeError) as err: msg = "Cannot deserialize datetime object." raise DeserializationError(msg) from err - else: - return date_obj + return date_obj @staticmethod def deserialize_unix(attr): @@ -1985,8 +2017,9 @@ def deserialize_unix(attr): This is represented as seconds. :param int attr: Object to be serialized. + :return: Deserialized datetime :rtype: Datetime - :raises: DeserializationError if format invalid + :raises DeserializationError: if format invalid """ if isinstance(attr, ET.Element): attr = int(attr.text) # type: ignore @@ -1996,5 +2029,4 @@ def deserialize_unix(attr): except ValueError as err: msg = "Cannot deserialize to unix datetime object." raise DeserializationError(msg) from err - else: - return date_obj + return date_obj diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_version.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_version.py index a741dc7b69b5..cffb9c489194 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_version.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/_version.py @@ -2,8 +2,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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "1.1.0b4" +VERSION = "1.1.0b5" diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/__init__.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/__init__.py index f09378ce2fff..3f9a984f2dcc 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/__init__.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/__init__.py @@ -2,22 +2,28 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +# pylint: disable=wrong-import-position -from ._microsoft_elastic import MicrosoftElastic +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from ._patch import * # pylint: disable=unused-wildcard-import + +from ._client import ElasticClient # type: ignore try: from ._patch import __all__ as _patch_all - from ._patch import * # pylint: disable=unused-wildcard-import + from ._patch import * except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk __all__ = [ - "MicrosoftElastic", + "ElasticClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_client.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_client.py new file mode 100644 index 000000000000..3a92b1e711ec --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_client.py @@ -0,0 +1,168 @@ +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from copy import deepcopy +from typing import Any, Awaitable, Optional, TYPE_CHECKING, cast +from typing_extensions import Self + +from azure.core.pipeline import policies +from azure.core.rest import AsyncHttpResponse, HttpRequest +from azure.core.settings import settings +from azure.mgmt.core import AsyncARMPipelineClient +from azure.mgmt.core.policies import AsyncARMAutoResourceProviderRegistrationPolicy +from azure.mgmt.core.tools import get_arm_endpoints + +from .._utils.serialization import Deserializer, Serializer +from ._configuration import ElasticClientConfiguration +from .operations import ( + ElasticMonitorResourcesOperations, + ElasticVersionsOperationGroupOperations, + MonitoredSubscriptionsOperations, + OpenAIIntegrationRPModelsOperations, + Operations, + OrganizationsOperationGroupOperations, + TagRulesOperations, +) + +if TYPE_CHECKING: + from azure.core.credentials_async import AsyncTokenCredential + + +class ElasticClient: # pylint: disable=too-many-instance-attributes + """ElasticClient. + + :ivar operations: Operations operations + :vartype operations: azure.mgmt.elastic.aio.operations.Operations + :ivar elastic_monitor_resources: ElasticMonitorResourcesOperations operations + :vartype elastic_monitor_resources: + azure.mgmt.elastic.aio.operations.ElasticMonitorResourcesOperations + :ivar monitored_subscriptions: MonitoredSubscriptionsOperations operations + :vartype monitored_subscriptions: + azure.mgmt.elastic.aio.operations.MonitoredSubscriptionsOperations + :ivar open_ai_integration_rp_models: OpenAIIntegrationRPModelsOperations operations + :vartype open_ai_integration_rp_models: + azure.mgmt.elastic.aio.operations.OpenAIIntegrationRPModelsOperations + :ivar tag_rules: TagRulesOperations operations + :vartype tag_rules: azure.mgmt.elastic.aio.operations.TagRulesOperations + :ivar elastic_versions_operation_group: ElasticVersionsOperationGroupOperations operations + :vartype elastic_versions_operation_group: + azure.mgmt.elastic.aio.operations.ElasticVersionsOperationGroupOperations + :ivar organizations_operation_group: OrganizationsOperationGroupOperations operations + :vartype organizations_operation_group: + azure.mgmt.elastic.aio.operations.OrganizationsOperationGroupOperations + :param credential: Credential used to authenticate requests to the service. Required. + :type credential: ~azure.core.credentials_async.AsyncTokenCredential + :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. + :type subscription_id: str + :param base_url: Service host. Default value is None. + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is + "2025-01-15-preview". Note that overriding this default value may result in unsupported + behavior. + :paramtype api_version: 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: "AsyncTokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any + ) -> None: + _endpoint = "{endpoint}" + _cloud = kwargs.pop("cloud_setting", None) or settings.current.azure_cloud # type: ignore + _endpoints = get_arm_endpoints(_cloud) + if not base_url: + base_url = _endpoints["resource_manager"] + credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) + self._config = ElasticClientConfiguration( + credential=credential, + subscription_id=subscription_id, + base_url=cast(str, base_url), + credential_scopes=credential_scopes, + **kwargs + ) + + _policies = kwargs.pop("policies", None) + if _policies is None: + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] + self._client: AsyncARMPipelineClient = AsyncARMPipelineClient( + base_url=cast(str, _endpoint), policies=_policies, **kwargs + ) + + self._serialize = Serializer() + self._deserialize = Deserializer() + self._serialize.client_side_validation = False + self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) + self.elastic_monitor_resources = ElasticMonitorResourcesOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.monitored_subscriptions = MonitoredSubscriptionsOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.open_ai_integration_rp_models = OpenAIIntegrationRPModelsOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.tag_rules = TagRulesOperations(self._client, self._config, self._serialize, self._deserialize) + self.elastic_versions_operation_group = ElasticVersionsOperationGroupOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.organizations_operation_group = OrganizationsOperationGroupOperations( + self._client, self._config, self._serialize, self._deserialize + ) + + def send_request( + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + ) -> Awaitable[AsyncHttpResponse]: + """Runs the network request through the client's chained policies. + + >>> 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/dpcodegen/python/send_request + + :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.rest.AsyncHttpResponse + """ + + request_copy = deepcopy(request) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore + + async def close(self) -> None: + await self._client.close() + + async def __aenter__(self) -> Self: + await self._client.__aenter__() + return self + + async def __aexit__(self, *exc_details: Any) -> None: + await self._client.__aexit__(*exc_details) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_configuration.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_configuration.py index 91988f778fcb..620610146c5f 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_configuration.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_configuration.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -14,27 +14,35 @@ from .._version import VERSION if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports from azure.core.credentials_async import AsyncTokenCredential -class MicrosoftElasticConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long - """Configuration for MicrosoftElastic. +class ElasticClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for ElasticClient. Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :keyword api_version: Api Version. Default value is "2024-06-15-preview". Note that overriding - this default value may result in unsupported behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is + "2025-01-15-preview". Note that overriding this default value may result in unsupported + behavior. :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2024-06-15-preview") + def __init__( + self, + credential: "AsyncTokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop("api_version", "2025-01-15-preview") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -43,6 +51,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-elastic/{}".format(VERSION)) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_microsoft_elastic.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_microsoft_elastic.py deleted file mode 100644 index eb1747f32c25..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_microsoft_elastic.py +++ /dev/null @@ -1,241 +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. -# -------------------------------------------------------------------------- - -from copy import deepcopy -from typing import Any, Awaitable, TYPE_CHECKING -from typing_extensions import Self - -from azure.core.pipeline import policies -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.mgmt.core import AsyncARMPipelineClient -from azure.mgmt.core.policies import AsyncARMAutoResourceProviderRegistrationPolicy - -from .. import models as _models -from .._serialization import Deserializer, Serializer -from ._configuration import MicrosoftElasticConfiguration -from .operations import ( - AllTrafficFiltersOperations, - AssociateTrafficFilterOperations, - BillingInfoOperations, - ConnectedPartnerResourcesOperations, - CreateAndAssociateIPFilterOperations, - CreateAndAssociatePLFilterOperations, - DeploymentInfoOperations, - DetachAndDeleteTrafficFilterOperations, - DetachTrafficFilterOperations, - ElasticVersionsOperations, - ExternalUserOperations, - ListAssociatedTrafficFiltersOperations, - MonitorOperations, - MonitoredResourcesOperations, - MonitoredSubscriptionsOperations, - MonitorsOperations, - OpenAIOperations, - Operations, - OrganizationsOperations, - TagRulesOperations, - TrafficFiltersOperations, - UpgradableVersionsOperations, - VMCollectionOperations, - VMHostOperations, - VMIngestionOperations, -) - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from azure.core.credentials_async import AsyncTokenCredential - - -class MicrosoftElastic: # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes - """MicrosoftElastic. - - :ivar operations: Operations operations - :vartype operations: azure.mgmt.elastic.aio.operations.Operations - :ivar monitors: MonitorsOperations operations - :vartype monitors: azure.mgmt.elastic.aio.operations.MonitorsOperations - :ivar elastic_versions: ElasticVersionsOperations operations - :vartype elastic_versions: azure.mgmt.elastic.aio.operations.ElasticVersionsOperations - :ivar monitored_subscriptions: MonitoredSubscriptionsOperations operations - :vartype monitored_subscriptions: - azure.mgmt.elastic.aio.operations.MonitoredSubscriptionsOperations - :ivar monitored_resources: MonitoredResourcesOperations operations - :vartype monitored_resources: azure.mgmt.elastic.aio.operations.MonitoredResourcesOperations - :ivar deployment_info: DeploymentInfoOperations operations - :vartype deployment_info: azure.mgmt.elastic.aio.operations.DeploymentInfoOperations - :ivar external_user: ExternalUserOperations operations - :vartype external_user: azure.mgmt.elastic.aio.operations.ExternalUserOperations - :ivar billing_info: BillingInfoOperations operations - :vartype billing_info: azure.mgmt.elastic.aio.operations.BillingInfoOperations - :ivar connected_partner_resources: ConnectedPartnerResourcesOperations operations - :vartype connected_partner_resources: - azure.mgmt.elastic.aio.operations.ConnectedPartnerResourcesOperations - :ivar open_ai: OpenAIOperations operations - :vartype open_ai: azure.mgmt.elastic.aio.operations.OpenAIOperations - :ivar tag_rules: TagRulesOperations operations - :vartype tag_rules: azure.mgmt.elastic.aio.operations.TagRulesOperations - :ivar vm_host: VMHostOperations operations - :vartype vm_host: azure.mgmt.elastic.aio.operations.VMHostOperations - :ivar vm_ingestion: VMIngestionOperations operations - :vartype vm_ingestion: azure.mgmt.elastic.aio.operations.VMIngestionOperations - :ivar vm_collection: VMCollectionOperations operations - :vartype vm_collection: azure.mgmt.elastic.aio.operations.VMCollectionOperations - :ivar upgradable_versions: UpgradableVersionsOperations operations - :vartype upgradable_versions: azure.mgmt.elastic.aio.operations.UpgradableVersionsOperations - :ivar monitor: MonitorOperations operations - :vartype monitor: azure.mgmt.elastic.aio.operations.MonitorOperations - :ivar all_traffic_filters: AllTrafficFiltersOperations operations - :vartype all_traffic_filters: azure.mgmt.elastic.aio.operations.AllTrafficFiltersOperations - :ivar list_associated_traffic_filters: ListAssociatedTrafficFiltersOperations operations - :vartype list_associated_traffic_filters: - azure.mgmt.elastic.aio.operations.ListAssociatedTrafficFiltersOperations - :ivar create_and_associate_ip_filter: CreateAndAssociateIPFilterOperations operations - :vartype create_and_associate_ip_filter: - azure.mgmt.elastic.aio.operations.CreateAndAssociateIPFilterOperations - :ivar create_and_associate_pl_filter: CreateAndAssociatePLFilterOperations operations - :vartype create_and_associate_pl_filter: - azure.mgmt.elastic.aio.operations.CreateAndAssociatePLFilterOperations - :ivar associate_traffic_filter: AssociateTrafficFilterOperations operations - :vartype associate_traffic_filter: - azure.mgmt.elastic.aio.operations.AssociateTrafficFilterOperations - :ivar detach_and_delete_traffic_filter: DetachAndDeleteTrafficFilterOperations operations - :vartype detach_and_delete_traffic_filter: - azure.mgmt.elastic.aio.operations.DetachAndDeleteTrafficFilterOperations - :ivar detach_traffic_filter: DetachTrafficFilterOperations operations - :vartype detach_traffic_filter: azure.mgmt.elastic.aio.operations.DetachTrafficFilterOperations - :ivar traffic_filters: TrafficFiltersOperations operations - :vartype traffic_filters: azure.mgmt.elastic.aio.operations.TrafficFiltersOperations - :ivar organizations: OrganizationsOperations operations - :vartype organizations: azure.mgmt.elastic.aio.operations.OrganizationsOperations - :param credential: Credential needed for the client to connect to Azure. Required. - :type credential: ~azure.core.credentials_async.AsyncTokenCredential - :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. - :type subscription_id: str - :param base_url: Service URL. Default value is "https://management.azure.com". - :type base_url: str - :keyword api_version: Api Version. Default value is "2024-06-15-preview". Note that overriding - this default value may result in unsupported behavior. - :paramtype api_version: 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: "AsyncTokenCredential", - subscription_id: str, - base_url: str = "https://management.azure.com", - **kwargs: Any - ) -> None: - self._config = MicrosoftElasticConfiguration(credential=credential, subscription_id=subscription_id, **kwargs) - _policies = kwargs.pop("policies", None) - if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} - self._serialize = Serializer(client_models) - self._deserialize = Deserializer(client_models) - self._serialize.client_side_validation = False - self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) - self.monitors = MonitorsOperations(self._client, self._config, self._serialize, self._deserialize) - self.elastic_versions = ElasticVersionsOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.monitored_subscriptions = MonitoredSubscriptionsOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.monitored_resources = MonitoredResourcesOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.deployment_info = DeploymentInfoOperations(self._client, self._config, self._serialize, self._deserialize) - self.external_user = ExternalUserOperations(self._client, self._config, self._serialize, self._deserialize) - self.billing_info = BillingInfoOperations(self._client, self._config, self._serialize, self._deserialize) - self.connected_partner_resources = ConnectedPartnerResourcesOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.open_ai = OpenAIOperations(self._client, self._config, self._serialize, self._deserialize) - self.tag_rules = TagRulesOperations(self._client, self._config, self._serialize, self._deserialize) - self.vm_host = VMHostOperations(self._client, self._config, self._serialize, self._deserialize) - self.vm_ingestion = VMIngestionOperations(self._client, self._config, self._serialize, self._deserialize) - self.vm_collection = VMCollectionOperations(self._client, self._config, self._serialize, self._deserialize) - self.upgradable_versions = UpgradableVersionsOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.monitor = MonitorOperations(self._client, self._config, self._serialize, self._deserialize) - self.all_traffic_filters = AllTrafficFiltersOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.list_associated_traffic_filters = ListAssociatedTrafficFiltersOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.create_and_associate_ip_filter = CreateAndAssociateIPFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.create_and_associate_pl_filter = CreateAndAssociatePLFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.associate_traffic_filter = AssociateTrafficFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.detach_and_delete_traffic_filter = DetachAndDeleteTrafficFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.detach_traffic_filter = DetachTrafficFilterOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.traffic_filters = TrafficFiltersOperations(self._client, self._config, self._serialize, self._deserialize) - self.organizations = OrganizationsOperations(self._client, self._config, self._serialize, self._deserialize) - - def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any - ) -> Awaitable[AsyncHttpResponse]: - """Runs the network request through the client's chained policies. - - >>> 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/dpcodegen/python/send_request - - :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.rest.AsyncHttpResponse - """ - - request_copy = deepcopy(request) - request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore - - async def close(self) -> None: - await self._client.close() - - async def __aenter__(self) -> Self: - await self._client.__aenter__() - return self - - async def __aexit__(self, *exc_details: Any) -> None: - await self._client.__aexit__(*exc_details) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_patch.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_patch.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/__init__.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/__init__.py index 1cdfeb9a7b42..7754ab6a86e0 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/__init__.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/__init__.py @@ -2,66 +2,36 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +# pylint: disable=wrong-import-position -from ._operations import Operations -from ._monitors_operations import MonitorsOperations -from ._elastic_versions_operations import ElasticVersionsOperations -from ._monitored_subscriptions_operations import MonitoredSubscriptionsOperations -from ._monitored_resources_operations import MonitoredResourcesOperations -from ._deployment_info_operations import DeploymentInfoOperations -from ._external_user_operations import ExternalUserOperations -from ._billing_info_operations import BillingInfoOperations -from ._connected_partner_resources_operations import ConnectedPartnerResourcesOperations -from ._open_ai_operations import OpenAIOperations -from ._tag_rules_operations import TagRulesOperations -from ._vm_host_operations import VMHostOperations -from ._vm_ingestion_operations import VMIngestionOperations -from ._vm_collection_operations import VMCollectionOperations -from ._upgradable_versions_operations import UpgradableVersionsOperations -from ._monitor_operations import MonitorOperations -from ._all_traffic_filters_operations import AllTrafficFiltersOperations -from ._list_associated_traffic_filters_operations import ListAssociatedTrafficFiltersOperations -from ._create_and_associate_ip_filter_operations import CreateAndAssociateIPFilterOperations -from ._create_and_associate_pl_filter_operations import CreateAndAssociatePLFilterOperations -from ._associate_traffic_filter_operations import AssociateTrafficFilterOperations -from ._detach_and_delete_traffic_filter_operations import DetachAndDeleteTrafficFilterOperations -from ._detach_traffic_filter_operations import DetachTrafficFilterOperations -from ._traffic_filters_operations import TrafficFiltersOperations -from ._organizations_operations import OrganizationsOperations +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from ._patch import * # pylint: disable=unused-wildcard-import + +from ._operations import Operations # type: ignore +from ._operations import ElasticMonitorResourcesOperations # type: ignore +from ._operations import MonitoredSubscriptionsOperations # type: ignore +from ._operations import OpenAIIntegrationRPModelsOperations # type: ignore +from ._operations import TagRulesOperations # type: ignore +from ._operations import ElasticVersionsOperationGroupOperations # type: ignore +from ._operations import OrganizationsOperationGroupOperations # type: ignore from ._patch import __all__ as _patch_all -from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import * from ._patch import patch_sdk as _patch_sdk __all__ = [ "Operations", - "MonitorsOperations", - "ElasticVersionsOperations", + "ElasticMonitorResourcesOperations", "MonitoredSubscriptionsOperations", - "MonitoredResourcesOperations", - "DeploymentInfoOperations", - "ExternalUserOperations", - "BillingInfoOperations", - "ConnectedPartnerResourcesOperations", - "OpenAIOperations", + "OpenAIIntegrationRPModelsOperations", "TagRulesOperations", - "VMHostOperations", - "VMIngestionOperations", - "VMCollectionOperations", - "UpgradableVersionsOperations", - "MonitorOperations", - "AllTrafficFiltersOperations", - "ListAssociatedTrafficFiltersOperations", - "CreateAndAssociateIPFilterOperations", - "CreateAndAssociatePLFilterOperations", - "AssociateTrafficFilterOperations", - "DetachAndDeleteTrafficFilterOperations", - "DetachTrafficFilterOperations", - "TrafficFiltersOperations", - "OrganizationsOperations", + "ElasticVersionsOperationGroupOperations", + "OrganizationsOperationGroupOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_all_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_all_traffic_filters_operations.py deleted file mode 100644 index 861ff93eff4e..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_all_traffic_filters_operations.py +++ /dev/null @@ -1,116 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._all_traffic_filters_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class AllTrafficFiltersOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`all_traffic_filters` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> _models.ElasticTrafficFilterResponse: - """Get the list of all traffic filters for the account. - - Get the list of all traffic filters for the account. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: ElasticTrafficFilterResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ElasticTrafficFilterResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_associate_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_associate_traffic_filter_operations.py deleted file mode 100644 index 7f5521f5ebc8..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_associate_traffic_filter_operations.py +++ /dev/null @@ -1,174 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterator, Callable, Dict, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._associate_traffic_filter_operations import build_associate_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class AssociateTrafficFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`associate_traffic_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - async def _associate_initial( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_associate_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_associate( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> AsyncLROPoller[None]: - """Associate traffic filter for the given deployment. - - Associate traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._associate_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - ruleset_id=ruleset_id, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_billing_info_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_billing_info_operations.py deleted file mode 100644 index f3106e9bb129..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_billing_info_operations.py +++ /dev/null @@ -1,114 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._billing_info_operations import build_get_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class BillingInfoOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`billing_info` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def get(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.BillingInfoResponse: - """Get marketplace and organization info mapped to the given monitor. - - Get marketplace and organization info mapped to the given monitor. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: BillingInfoResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.BillingInfoResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.BillingInfoResponse] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("BillingInfoResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_connected_partner_resources_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_connected_partner_resources_operations.py deleted file mode 100644 index b9053d30bbb5..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_connected_partner_resources_operations.py +++ /dev/null @@ -1,148 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, Type, TypeVar -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._connected_partner_resources_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class ConnectedPartnerResourcesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`connected_partner_resources` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> AsyncIterable["_models.ConnectedPartnerResourcesListFormat"]: - """List of all active deployments that are associated with the marketplace subscription linked to - the given monitor. - - List of all active deployments that are associated with the marketplace subscription linked to - the given monitor. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either ConnectedPartnerResourcesListFormat or the result - of cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ConnectedPartnerResourcesListFormat] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ConnectedPartnerResourcesListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("ConnectedPartnerResourcesListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_ip_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_ip_filter_operations.py deleted file mode 100644 index a0c6b8261e62..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_ip_filter_operations.py +++ /dev/null @@ -1,188 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterator, Callable, Dict, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._create_and_associate_ip_filter_operations import build_create_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class CreateAndAssociateIPFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`create_and_associate_ip_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - async def _create_initial( - self, - resource_group_name: str, - monitor_name: str, - ips: Optional[str] = None, - name: Optional[str] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_create_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ips=ips, - name=name, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_create( - self, - resource_group_name: str, - monitor_name: str, - ips: Optional[str] = None, - name: Optional[str] = None, - **kwargs: Any - ) -> AsyncLROPoller[None]: - """Create and Associate IP traffic filter for the given deployment. - - Create and Associate IP traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ips: List of ips. Default value is None. - :type ips: str - :param name: Name of the traffic filter. Default value is None. - :type name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._create_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - ips=ips, - name=name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_pl_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_pl_filter_operations.py deleted file mode 100644 index 0484a59d57f6..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_create_and_associate_pl_filter_operations.py +++ /dev/null @@ -1,194 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterator, Callable, Dict, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._create_and_associate_pl_filter_operations import build_create_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class CreateAndAssociatePLFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`create_and_associate_pl_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - async def _create_initial( - self, - resource_group_name: str, - monitor_name: str, - name: Optional[str] = None, - private_endpoint_guid: Optional[str] = None, - private_endpoint_name: Optional[str] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_create_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - name=name, - private_endpoint_guid=private_endpoint_guid, - private_endpoint_name=private_endpoint_name, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_create( - self, - resource_group_name: str, - monitor_name: str, - name: Optional[str] = None, - private_endpoint_guid: Optional[str] = None, - private_endpoint_name: Optional[str] = None, - **kwargs: Any - ) -> AsyncLROPoller[None]: - """Create and Associate private link traffic filter for the given deployment. - - Create and Associate private link traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param name: Name of the traffic filter. Default value is None. - :type name: str - :param private_endpoint_guid: Guid of the private endpoint. Default value is None. - :type private_endpoint_guid: str - :param private_endpoint_name: Name of the private endpoint. Default value is None. - :type private_endpoint_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._create_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - name=name, - private_endpoint_guid=private_endpoint_guid, - private_endpoint_name=private_endpoint_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_deployment_info_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_deployment_info_operations.py deleted file mode 100644 index 67a8b2965a4f..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_deployment_info_operations.py +++ /dev/null @@ -1,116 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._deployment_info_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class DeploymentInfoOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`deployment_info` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def list(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.DeploymentInfoResponse: - """Fetch information regarding Elastic cloud deployment corresponding to the Elastic monitor - resource. - - Fetch information regarding Elastic cloud deployment corresponding to the Elastic monitor - resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: DeploymentInfoResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.DeploymentInfoResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.DeploymentInfoResponse] = kwargs.pop("cls", None) - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("DeploymentInfoResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_and_delete_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_and_delete_traffic_filter_operations.py deleted file mode 100644 index 44f0c46daf60..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_and_delete_traffic_filter_operations.py +++ /dev/null @@ -1,115 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._detach_and_delete_traffic_filter_operations import build_delete_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class DetachAndDeleteTrafficFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`detach_and_delete_traffic_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> None: - """Detach and Delete traffic filter from the given deployment. - - Detach and Delete traffic filter from the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_traffic_filter_operations.py deleted file mode 100644 index 1d7ca8c993d1..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_detach_traffic_filter_operations.py +++ /dev/null @@ -1,174 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterator, Callable, Dict, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._detach_traffic_filter_operations import build_update_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class DetachTrafficFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`detach_traffic_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - async def _update_initial( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_update( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> AsyncLROPoller[None]: - """Detach traffic filter for the given deployment. - - Detach traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - ruleset_id=ruleset_id, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_elastic_versions_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_elastic_versions_operations.py deleted file mode 100644 index c55d5473f9ef..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_elastic_versions_operations.py +++ /dev/null @@ -1,140 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, Type, TypeVar -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._elastic_versions_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class ElasticVersionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`elastic_versions` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, region: str, **kwargs: Any) -> AsyncIterable["_models.ElasticVersionListFormat"]: - """Get a list of available versions for a region. - - Get a list of available versions for a region. - - :param region: Region where elastic deployment will take place. Required. - :type region: str - :return: An iterator like instance of either ElasticVersionListFormat or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ElasticVersionListFormat] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticVersionsListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - subscription_id=self._config.subscription_id, - region=region, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("ElasticVersionsListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_external_user_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_external_user_operations.py deleted file mode 100644 index bbe670027f5e..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_external_user_operations.py +++ /dev/null @@ -1,203 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Optional, Type, TypeVar, Union, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._external_user_operations import build_create_or_update_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class ExternalUserOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`external_user` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ExternalUserInfo] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.ExternalUserCreationResponse: - """Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic External User Creation Parameters. Default value is None. - :type body: ~azure.mgmt.elastic.models.ExternalUserInfo - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: ExternalUserCreationResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.ExternalUserCreationResponse: - """Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic External User Creation Parameters. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: ExternalUserCreationResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ExternalUserInfo, IO[bytes]]] = None, - **kwargs: Any - ) -> _models.ExternalUserCreationResponse: - """Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic External User Creation Parameters. Is either a ExternalUserInfo type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ExternalUserInfo or IO[bytes] - :return: ExternalUserCreationResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ExternalUserCreationResponse] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ExternalUserInfo") - else: - _json = None - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ExternalUserCreationResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_list_associated_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_list_associated_traffic_filters_operations.py deleted file mode 100644 index dc896a6f4a9b..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_list_associated_traffic_filters_operations.py +++ /dev/null @@ -1,116 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._list_associated_traffic_filters_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class ListAssociatedTrafficFiltersOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`list_associated_traffic_filters` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> _models.ElasticTrafficFilterResponse: - """Get the list of all associated traffic filters for the given deployment. - - Get the list of all associated traffic filters for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: ElasticTrafficFilterResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ElasticTrafficFilterResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitor_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitor_operations.py deleted file mode 100644 index ddd1e5cd0380..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitor_operations.py +++ /dev/null @@ -1,256 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, AsyncIterator, Callable, Dict, IO, Optional, Type, TypeVar, Union, cast, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._monitor_operations import build_upgrade_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class MonitorOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`monitor` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - async def _upgrade_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorUpgrade, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ElasticMonitorUpgrade") - else: - _json = None - - _request = build_upgrade_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_upgrade( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ElasticMonitorUpgrade] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[None]: - """Upgradable version for a monitor resource. - - Upgradable version for a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic Monitor Upgrade Parameters. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :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: - """ - - @overload - async def begin_upgrade( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[None]: - """Upgradable version for a monitor resource. - - Upgradable version for a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic Monitor Upgrade Parameters. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :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: - """ - - @distributed_trace_async - async def begin_upgrade( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorUpgrade, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncLROPoller[None]: - """Upgradable version for a monitor resource. - - Upgradable version for a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic Monitor Upgrade Parameters. Is either a ElasticMonitorUpgrade type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade or IO[bytes] - :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: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._upgrade_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_resources_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_resources_operations.py deleted file mode 100644 index 376efeccb0bc..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_resources_operations.py +++ /dev/null @@ -1,144 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, Type, TypeVar -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._monitored_resources_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class MonitoredResourcesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`monitored_resources` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> AsyncIterable["_models.MonitoredResource"]: - """List the resources currently being monitored by the Elastic monitor resource. - - List the resources currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either MonitoredResource or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.MonitoredResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoredResourceListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("MonitoredResourceListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_subscriptions_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_subscriptions_operations.py deleted file mode 100644 index 17954a231f92..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitored_subscriptions_operations.py +++ /dev/null @@ -1,773 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, Optional, Type, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._monitored_subscriptions_operations import ( - build_createor_update_request, - build_delete_request, - build_get_request, - build_list_request, - build_update_request, -) - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class MonitoredSubscriptionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`monitored_subscriptions` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> AsyncIterable["_models.MonitoredSubscriptionProperties"]: - """List all the subscriptions currently being monitored by the Elastic monitor resource. - - List all the subscriptions currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either MonitoredSubscriptionProperties or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoredSubscriptionPropertiesList] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("MonitoredSubscriptionPropertiesList", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @distributed_trace_async - async def get( - self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any - ) -> _models.MonitoredSubscriptionProperties: - """List all the subscriptions currently being monitored by the Elastic monitor resource. - - List all the subscriptions currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :return: MonitoredSubscriptionProperties or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("MonitoredSubscriptionProperties", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - async def _createor_update_initial( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "MonitoredSubscriptionProperties") - else: - _json = None - - _request = build_createor_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 201]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_createor_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[_models.MonitoredSubscriptionProperties] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: - """Add the subscriptions that should be monitored by the Elastic monitor resource. - - Add the subscriptions that should be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either MonitoredSubscriptionProperties or - the result of cls(response) - :rtype: - ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_createor_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: - """Add the subscriptions that should be monitored by the Elastic monitor resource. - - Add the subscriptions that should be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either MonitoredSubscriptionProperties or - the result of cls(response) - :rtype: - ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_createor_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: - """Add the subscriptions that should be monitored by the Elastic monitor resource. - - Add the subscriptions that should be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Is either a MonitoredSubscriptionProperties type or a IO[bytes] type. Default - value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or IO[bytes] - :return: An instance of AsyncLROPoller that returns either MonitoredSubscriptionProperties or - the result of cls(response) - :rtype: - ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._createor_update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("MonitoredSubscriptionProperties", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.MonitoredSubscriptionProperties].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.MonitoredSubscriptionProperties]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - async def _update_initial( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "MonitoredSubscriptionProperties") - else: - _json = None - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[_models.MonitoredSubscriptionProperties] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either MonitoredSubscriptionProperties or - the result of cls(response) - :rtype: - ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either MonitoredSubscriptionProperties or - the result of cls(response) - :rtype: - ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Is either a MonitoredSubscriptionProperties type or a IO[bytes] type. Default - value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or IO[bytes] - :return: An instance of AsyncLROPoller that returns either MonitoredSubscriptionProperties or - the result of cls(response) - :rtype: - ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("MonitoredSubscriptionProperties", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.MonitoredSubscriptionProperties].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.MonitoredSubscriptionProperties]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - async def _delete_initial( - self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202, 204]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_delete( - self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any - ) -> AsyncLROPoller[None]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._delete_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitors_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitors_operations.py deleted file mode 100644 index ed1b3a8d47f7..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_monitors_operations.py +++ /dev/null @@ -1,818 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, Optional, Type, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._monitors_operations import ( - build_create_request, - build_delete_request, - build_get_request, - build_list_by_resource_group_request, - build_list_request, - build_update_request, -) - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class MonitorsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`monitors` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, **kwargs: Any) -> AsyncIterable["_models.ElasticMonitorResource"]: - """List all monitors under the specified subscription. - - List all monitors under the specified subscription. - - :return: An iterator like instance of either ElasticMonitorResource or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticMonitorResourceListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResourceListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @distributed_trace - def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any - ) -> AsyncIterable["_models.ElasticMonitorResource"]: - """List all monitors under the specified resource group. - - List all monitors under the specified resource group. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :return: An iterator like instance of either ElasticMonitorResource or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticMonitorResourceListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_by_resource_group_request( - resource_group_name=resource_group_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResourceListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @distributed_trace_async - async def get(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.ElasticMonitorResource: - """Get the properties of a specific monitor resource. - - Get the properties of a specific monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: ElasticMonitorResource or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticMonitorResource - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - async def _create_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResource, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ElasticMonitorResource") - else: - _json = None - - _request = build_create_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 201]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_create( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ElasticMonitorResource] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Create a monitor resource. - - Create a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic monitor resource model. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_create( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Create a monitor resource. - - Create a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic monitor resource model. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_create( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResource, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Create a monitor resource. - - Create a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic monitor resource model. Is either a ElasticMonitorResource type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource or IO[bytes] - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._create_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, - AsyncARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs), - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.ElasticMonitorResource].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.ElasticMonitorResource]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - async def _update_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResourceUpdateParameters, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ElasticMonitorResourceUpdateParameters") - else: - _json = None - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ElasticMonitorResourceUpdateParameters] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Update a monitor resource. - - Update a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic resource model update parameters. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Update a monitor resource. - - Update a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic resource model update parameters. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResourceUpdateParameters, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Update a monitor resource. - - Update a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic resource model update parameters. Is either a - ElasticMonitorResourceUpdateParameters type or a IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters or IO[bytes] - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.ElasticMonitorResource].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.ElasticMonitorResource]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - async def _delete_initial(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202, 204]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_delete(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> AsyncLROPoller[None]: - """Delete a monitor resource. - - Delete a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._delete_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_open_ai_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_open_ai_operations.py deleted file mode 100644 index 5537bc725259..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_open_ai_operations.py +++ /dev/null @@ -1,497 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, AsyncIterable, Callable, Dict, IO, Optional, Type, TypeVar, Union, overload -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._open_ai_operations import ( - build_create_or_update_request, - build_delete_request, - build_get_request, - build_get_status_request, - build_list_request, -) - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class OpenAIOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`open_ai` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> AsyncIterable["_models.OpenAIIntegrationRPModel"]: - """List OpenAI integration rule for a given monitor resource. - - List OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either OpenAIIntegrationRPModel or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.OpenAIIntegrationRPModel] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OpenAIIntegrationRPModelListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("OpenAIIntegrationRPModelListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @distributed_trace_async - async def get( - self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Get OpenAI integration rule for a given monitor resource. - - Get OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("OpenAIIntegrationRPModel", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - integration_name: str, - body: Optional[_models.OpenAIIntegrationRPModel] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Create or update a OpenAI integration rule for a given monitor resource. - - Create or update a OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :param body: Default value is None. - :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - integration_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Create or update a OpenAI integration rule for a given monitor resource. - - Create or update a OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :param body: Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - integration_name: str, - body: Optional[Union[_models.OpenAIIntegrationRPModel, IO[bytes]]] = None, - **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Create or update a OpenAI integration rule for a given monitor resource. - - Create or update a OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :param body: Is either a OpenAIIntegrationRPModel type or a IO[bytes] type. Default value is - None. - :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel or IO[bytes] - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "OpenAIIntegrationRPModel") - else: - _json = None - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("OpenAIIntegrationRPModel", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any - ) -> None: - """Delete OpenAI integration rule for a given monitor resource. - - Delete OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace_async - async def get_status( - self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any - ) -> _models.OpenAIIntegrationStatusResponse: - """Get OpenAI integration status for a given integration. - - Get OpenAI integration status for a given integration. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :return: OpenAIIntegrationStatusResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationStatusResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OpenAIIntegrationStatusResponse] = kwargs.pop("cls", None) - - _request = build_get_status_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("OpenAIIntegrationStatusResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_operations.py index 4ef4c0f0d0c8..5b331aada72f 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_operations.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_operations.py @@ -1,15 +1,18 @@ -# pylint: disable=too-many-lines,too-many-statements +# pylint: disable=too-many-lines # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, Type, TypeVar +from collections.abc import MutableMapping +from io import IOBase +import json +from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, List, Optional, TypeVar, Union, cast, overload import urllib.parse +from azure.core import AsyncPipelineClient from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ( ClientAuthenticationError, @@ -17,23 +20,72 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.pipeline import PipelineResponse +from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat +from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._operations import build_list_request +from ..._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize +from ..._utils.serialization import Deserializer, Serializer +from ...operations._operations import ( + build_elastic_monitor_resources_all_traffic_filters_list_request, + build_elastic_monitor_resources_associate_request, + build_elastic_monitor_resources_billing_info_get_request, + build_elastic_monitor_resources_connected_partner_resources_list_request, + build_elastic_monitor_resources_create_and_associate_ip_filter_create_request, + build_elastic_monitor_resources_create_and_associate_pl_filter_create_request, + build_elastic_monitor_resources_create_or_update_request, + build_elastic_monitor_resources_create_request, + build_elastic_monitor_resources_delete_request, + build_elastic_monitor_resources_deployment_info_list_request, + build_elastic_monitor_resources_detach_and_delete_traffic_filter_delete_request, + build_elastic_monitor_resources_detach_traffic_filter_update_request, + build_elastic_monitor_resources_details_request, + build_elastic_monitor_resources_get_request, + build_elastic_monitor_resources_list_associated_traffic_filters_list_request, + build_elastic_monitor_resources_list_by_resource_group_request, + build_elastic_monitor_resources_list_request, + build_elastic_monitor_resources_monitored_resources_list_request, + build_elastic_monitor_resources_resubscribe_request, + build_elastic_monitor_resources_traffic_filters_delete_request, + build_elastic_monitor_resources_update_request, + build_elastic_monitor_resources_upgradable_versions_details_request, + build_elastic_monitor_resources_upgrade_request, + build_elastic_monitor_resources_v_m_collection_update_request, + build_elastic_monitor_resources_v_m_host_list_request, + build_elastic_versions_operation_group_list_request, + build_monitored_subscriptions_createor_update_request, + build_monitored_subscriptions_delete_request, + build_monitored_subscriptions_get_request, + build_monitored_subscriptions_list_request, + build_monitored_subscriptions_update_request, + build_open_ai_integration_rp_models_create_or_update_request, + build_open_ai_integration_rp_models_delete_request, + build_open_ai_integration_rp_models_get_request, + build_open_ai_integration_rp_models_get_status_request, + build_open_ai_integration_rp_models_list_request, + build_operations_list_request, + build_organizations_operation_group_get_api_key_request, + build_organizations_operation_group_get_elastic_to_azure_subscription_mapping_request, + build_tag_rules_create_or_update_request, + build_tag_rules_delete_request, + build_tag_rules_get_request, + build_tag_rules_list_request, +) +from .._configuration import ElasticClientConfiguration -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] +JSON = MutableMapping[str, Any] class Operations: @@ -42,36 +94,31 @@ class Operations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s + :class:`~azure.mgmt.elastic.aio.ElasticClient`'s :attr:`operations` attribute. """ - models = _models - def __init__(self, *args, **kwargs) -> None: input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.OperationResult"]: - """List all operations provided by Microsoft.Elastic. + """List the operations for the provider. - List all operations provided by Microsoft.Elastic. - - :return: An iterator like instance of either OperationResult or the result of cls(response) + :return: An iterator like instance of OperationResult :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.OperationResult] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OperationListResult] = kwargs.pop("cls", None) + cls: ClsType[List[_models.OperationResult]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -82,12 +129,17 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.OperationResult"]: def prepare_request(next_link=None): if not next_link: - _request = build_list_request( - api_version=api_version, + _request = build_operations_list_request( + api_version=self._config.api_version, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -102,16 +154,21 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("OperationListResult", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.OperationResult], deserialized.get("value", [])) if cls: list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) @@ -124,11 +181,5079 @@ async def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class ElasticMonitorResourcesOperations: # pylint: disable=too-many-public-methods + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.aio.ElasticClient`'s + :attr:`elastic_monitor_resources` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.ElasticMonitorResource: + """Get the properties of a specific monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: ElasticMonitorResource. The ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticMonitorResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticMonitorResource, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _create_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResource, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_elastic_monitor_resources_create_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: _models.ElasticMonitorResource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResource, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Is one of the following types: + ElasticMonitorResource, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource or JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._create_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response = pipeline_response.http_response + deserialized = _deserialize(_models.ElasticMonitorResource, response.json()) + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[_models.ElasticMonitorResource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.ElasticMonitorResource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _update_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResourceUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_elastic_monitor_resources_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: _models.ElasticMonitorResourceUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResourceUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Is one of the following types: + ElasticMonitorResourceUpdateParameters, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters or JSON or + IO[bytes] + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _delete_initial(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202, 204]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_delete(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> AsyncLROPoller[None]: + """Delete a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._delete_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list_by_resource_group( + self, resource_group_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ElasticMonitorResource"]: + """List all monitors under the specified resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :return: An iterator like instance of ElasticMonitorResource + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ElasticMonitorResource]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_list_by_resource_group_request( + resource_group_name=resource_group_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.ElasticMonitorResource], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response return AsyncItemPaged(get_next, extract_data) + + @distributed_trace + def list(self, **kwargs: Any) -> AsyncIterable["_models.ElasticMonitorResource"]: + """List all monitors under the specified subscription. + + :return: An iterator like instance of ElasticMonitorResource + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ElasticMonitorResource]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_list_request( + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.ElasticMonitorResource], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace + def monitored_resources_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> AsyncIterable["_models.MonitoredResource"]: + """List the resources currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of MonitoredResource + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.MonitoredResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.MonitoredResource]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_monitored_resources_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.MonitoredResource], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace_async + async def deployment_info_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.DeploymentInfoResponse: + """Fetch information regarding Elastic cloud deployment corresponding to the Elastic monitor + resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: DeploymentInfoResponse. The DeploymentInfoResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.DeploymentInfoResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.DeploymentInfoResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_deployment_info_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.DeploymentInfoResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.ExternalUserInfo] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Default value is None. + :type body: ~azure.mgmt.elastic.models.ExternalUserInfo + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ExternalUserInfo, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Is one of the following types: + ExternalUserInfo, JSON, IO[bytes] Default value is None. + :type body: ~azure.mgmt.elastic.models.ExternalUserInfo or JSON or IO[bytes] + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ExternalUserCreationResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_create_or_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ExternalUserCreationResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def billing_info_get( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.BillingInfoResponse: + """Get marketplace and organization info mapped to the given monitor. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: BillingInfoResponse. The BillingInfoResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.BillingInfoResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.BillingInfoResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_billing_info_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.BillingInfoResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def connected_partner_resources_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ConnectedPartnerResourcesListFormat"]: + """List of all active deployments that are associated with the marketplace subscription linked to + the given monitor. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of ConnectedPartnerResourcesListFormat + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ConnectedPartnerResourcesListFormat] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ConnectedPartnerResourcesListFormat]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_connected_partner_resources_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.ConnectedPartnerResourcesListFormat], deserialized.get("value", []) + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace + def v_m_host_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> AsyncIterable["_models.VMResources"]: + """List the vm resources currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of VMResources + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.VMResources] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.VMResources]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_v_m_host_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.VMResources], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace_async + async def details( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.VMIngestionDetailsResponse: + """List the vm ingestion details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: VMIngestionDetailsResponse. The VMIngestionDetailsResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.VMIngestionDetailsResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.VMIngestionDetailsResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_details_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.VMIngestionDetailsResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def v_m_collection_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.VMCollectionUpdate] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Default value is None. + :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def v_m_collection_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def v_m_collection_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def v_m_collection_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.VMCollectionUpdate, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Is one of the following types: VMCollectionUpdate, JSON, IO[bytes] + Default value is None. + :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate or JSON or IO[bytes] + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_v_m_collection_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace_async + async def upgradable_versions_details( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.UpgradableVersionsList: + """List of upgradable versions for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: UpgradableVersionsList. The UpgradableVersionsList is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UpgradableVersionsList + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.UpgradableVersionsList] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_upgradable_versions_details_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.UpgradableVersionsList, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _upgrade_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ElasticMonitorUpgrade, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_upgrade_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.ElasticMonitorUpgrade] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Default value is None. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ElasticMonitorUpgrade, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Is one of the following types: + ElasticMonitorUpgrade, JSON, IO[bytes] Default value is None. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade or JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._upgrade_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def all_traffic_filters_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.ElasticTrafficFilterResponse: + """Get the list of all traffic filters for the account. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: ElasticTrafficFilterResponse. The ElasticTrafficFilterResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_all_traffic_filters_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticTrafficFilterResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def list_associated_traffic_filters_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.ElasticTrafficFilterResponse: + """Get the list of all associated traffic filters for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: ElasticTrafficFilterResponse. The ElasticTrafficFilterResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_list_associated_traffic_filters_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticTrafficFilterResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _create_and_associate_ip_filter_create_initial( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + ips: Optional[str] = None, + name: Optional[str] = None, + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_create_and_associate_ip_filter_create_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ips=ips, + name=name, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_create_and_associate_ip_filter_create( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + ips: Optional[str] = None, + name: Optional[str] = None, + **kwargs: Any + ) -> AsyncLROPoller[_models.OkResponse]: + """Create and Associate IP traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ips: List of ips. Default value is None. + :paramtype ips: str + :keyword name: Name of the traffic filter. Default value is None. + :paramtype name: str + :return: An instance of AsyncLROPoller that returns OkResponse. The OkResponse is compatible + with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.OkResponse] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OkResponse] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._create_and_associate_ip_filter_create_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + ips=ips, + name=name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response_headers = {} + response = pipeline_response.http_response + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = _deserialize(_models.OkResponse, response.json()) + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[_models.OkResponse].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.OkResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _create_and_associate_pl_filter_create_initial( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + name: Optional[str] = None, + private_endpoint_guid: Optional[str] = None, + private_endpoint_name: Optional[str] = None, + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_create_and_associate_pl_filter_create_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + name=name, + private_endpoint_guid=private_endpoint_guid, + private_endpoint_name=private_endpoint_name, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_create_and_associate_pl_filter_create( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + name: Optional[str] = None, + private_endpoint_guid: Optional[str] = None, + private_endpoint_name: Optional[str] = None, + **kwargs: Any + ) -> AsyncLROPoller[_models.OkResponse]: + """Create and Associate private link traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword name: Name of the traffic filter. Default value is None. + :paramtype name: str + :keyword private_endpoint_guid: Guid of the private endpoint. Default value is None. + :paramtype private_endpoint_guid: str + :keyword private_endpoint_name: Name of the private endpoint. Default value is None. + :paramtype private_endpoint_name: str + :return: An instance of AsyncLROPoller that returns OkResponse. The OkResponse is compatible + with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.OkResponse] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OkResponse] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._create_and_associate_pl_filter_create_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + name=name, + private_endpoint_guid=private_endpoint_guid, + private_endpoint_name=private_endpoint_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response_headers = {} + response = pipeline_response.http_response + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = _deserialize(_models.OkResponse, response.json()) + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[_models.OkResponse].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.OkResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _associate_initial( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_associate_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_associate( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Associate traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._associate_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + ruleset_id=ruleset_id, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def detach_and_delete_traffic_filter_delete( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> None: + """Detach and Delete traffic filter from the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_detach_and_delete_traffic_filter_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + async def _detach_traffic_filter_update_initial( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_detach_traffic_filter_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_detach_traffic_filter_update( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Detach traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._detach_traffic_filter_update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + ruleset_id=ruleset_id, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def traffic_filters_delete( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> None: + """Delete traffic filter from the account. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_traffic_filters_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + async def _resubscribe_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ResubscribeProperties, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_resubscribe_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.ResubscribeProperties] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Default value is None. + :type body: ~azure.mgmt.elastic.models.ResubscribeProperties + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ResubscribeProperties, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> AsyncLROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Is one of the following types: ResubscribeProperties, + JSON, IO[bytes] Default value is None. + :type body: ~azure.mgmt.elastic.models.ResubscribeProperties or JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ArmResponseElasticMonitorResource] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._resubscribe_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response_headers = {} + response = pipeline_response.http_response + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = _deserialize(_models.ArmResponseElasticMonitorResource, response.json()) + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[_models.ArmResponseElasticMonitorResource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.ArmResponseElasticMonitorResource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + +class MonitoredSubscriptionsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.aio.ElasticClient`'s + :attr:`monitored_subscriptions` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get( + self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any + ) -> _models.MonitoredSubscriptionProperties: + """List all the subscriptions currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :return: MonitoredSubscriptionProperties. The MonitoredSubscriptionProperties is compatible + with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) + + _request = build_monitored_subscriptions_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.MonitoredSubscriptionProperties, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _createor_update_initial( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_monitored_subscriptions_createor_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: _models.MonitoredSubscriptionProperties, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Is one of the following types: MonitoredSubscriptionProperties, JSON, IO[bytes] + Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._createor_update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _update_initial( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_monitored_subscriptions_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: _models.MonitoredSubscriptionProperties, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Is one of the following types: MonitoredSubscriptionProperties, JSON, IO[bytes] + Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response = pipeline_response.http_response + deserialized = _deserialize(_models.MonitoredSubscriptionProperties, response.json()) + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[_models.MonitoredSubscriptionProperties].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.MonitoredSubscriptionProperties]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _delete_initial( + self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_monitored_subscriptions_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202, 204]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_delete( + self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._delete_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> AsyncIterable["_models.MonitoredSubscriptionProperties"]: + """List all the subscriptions currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of MonitoredSubscriptionProperties + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.MonitoredSubscriptionProperties]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_monitored_subscriptions_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.MonitoredSubscriptionProperties], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class OpenAIIntegrationRPModelsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.aio.ElasticClient`'s + :attr:`open_ai_integration_rp_models` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get( + self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Get OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) + + _request = build_open_ai_integration_rp_models_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.OpenAIIntegrationRPModel, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: _models.OpenAIIntegrationRPModel, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Required. + :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: Union[_models.OpenAIIntegrationRPModel, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Is one of the following types: OpenAIIntegrationRPModel, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel or JSON or IO[bytes] + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_open_ai_integration_rp_models_create_or_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.OpenAIIntegrationRPModel, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete(self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any) -> None: + """Delete OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_open_ai_integration_rp_models_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> AsyncIterable["_models.OpenAIIntegrationRPModel"]: + """List OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of OpenAIIntegrationRPModel + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.OpenAIIntegrationRPModel] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.OpenAIIntegrationRPModel]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_open_ai_integration_rp_models_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.OpenAIIntegrationRPModel], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace_async + async def get_status( + self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any + ) -> _models.OpenAIIntegrationStatusResponse: + """Get OpenAI integration status for a given integration. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :return: OpenAIIntegrationStatusResponse. The OpenAIIntegrationStatusResponse is compatible + with MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationStatusResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OpenAIIntegrationStatusResponse] = kwargs.pop("cls", None) + + _request = build_open_ai_integration_rp_models_get_status_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.OpenAIIntegrationStatusResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + +class TagRulesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.aio.ElasticClient`'s + :attr:`tag_rules` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get( + self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any + ) -> _models.MonitoringTagRules: + """Get a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) + + _request = build_tag_rules_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.MonitoringTagRules, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: _models.MonitoringTagRules, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Required. + :type body: ~azure.mgmt.elastic.models.MonitoringTagRules + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: Union[_models.MonitoringTagRules, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Is one of the following types: + MonitoringTagRules, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.MonitoringTagRules or JSON or IO[bytes] + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_tag_rules_create_or_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.MonitoringTagRules, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _delete_initial( + self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_tag_rules_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202, 204]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_delete( + self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Delete a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._delete_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> AsyncIterable["_models.MonitoringTagRules"]: + """List the tag rules for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of MonitoringTagRules + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.MonitoringTagRules] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.MonitoringTagRules]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_tag_rules_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.MonitoringTagRules], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class ElasticVersionsOperationGroupOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.aio.ElasticClient`'s + :attr:`elastic_versions_operation_group` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def list(self, *, region: str, **kwargs: Any) -> AsyncIterable["_models.ElasticVersionListFormat"]: + """Get a list of available versions for a region. + + Get a list of available versions for a region. + + :keyword region: Region where elastic deployment will take place. Required. + :paramtype region: str + :return: An iterator like instance of ElasticVersionListFormat + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.ElasticVersionListFormat] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ElasticVersionListFormat]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_versions_operation_group_list_request( + subscription_id=self._config.subscription_id, + region=region, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.ElasticVersionListFormat], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class OrganizationsOperationGroupOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.aio.ElasticClient`'s + :attr:`organizations_operation_group` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @overload + async def get_api_key( + self, body: _models.UserEmailId, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Required. + :type body: ~azure.mgmt.elastic.models.UserEmailId + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def get_api_key( + self, body: JSON, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def get_api_key( + self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def get_api_key( + self, body: Union[_models.UserEmailId, JSON, IO[bytes]], **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Is one of the following types: UserEmailId, JSON, IO[bytes] + Required. + :type body: ~azure.mgmt.elastic.models.UserEmailId or JSON or IO[bytes] + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.UserApiKeyResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_organizations_operation_group_get_api_key_request( + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.UserApiKeyResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def get_elastic_to_azure_subscription_mapping( # pylint: disable=name-too-long + self, **kwargs: Any + ) -> _models.ElasticOrganizationToAzureSubscriptionMappingResponse: + """Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. + + ; + /** + Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. + + . + + :return: ElasticOrganizationToAzureSubscriptionMappingResponse. The + ElasticOrganizationToAzureSubscriptionMappingResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticOrganizationToAzureSubscriptionMappingResponse] = kwargs.pop("cls", None) + + _request = build_organizations_operation_group_get_elastic_to_azure_subscription_mapping_request( + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticOrganizationToAzureSubscriptionMappingResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_organizations_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_organizations_operations.py deleted file mode 100644 index fbfa8eafb705..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_organizations_operations.py +++ /dev/null @@ -1,445 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, AsyncIterator, Callable, Dict, IO, Optional, Type, TypeVar, Union, cast, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._organizations_operations import ( - build_get_api_key_request, - build_get_elastic_to_azure_subscription_mapping_request, - build_resubscribe_request, -) - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class OrganizationsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`organizations` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - async def get_api_key( - self, body: Optional[_models.UserEmailId] = None, *, content_type: str = "application/json", **kwargs: Any - ) -> _models.UserApiKeyResponse: - """Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - :param body: Email Id parameter of the User Organization, of which the API Key must be - returned. Default value is None. - :type body: ~azure.mgmt.elastic.models.UserEmailId - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: UserApiKeyResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def get_api_key( - self, body: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any - ) -> _models.UserApiKeyResponse: - """Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - :param body: Email Id parameter of the User Organization, of which the API Key must be - returned. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: UserApiKeyResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def get_api_key( - self, body: Optional[Union[_models.UserEmailId, IO[bytes]]] = None, **kwargs: Any - ) -> _models.UserApiKeyResponse: - """Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - :param body: Email Id parameter of the User Organization, of which the API Key must be - returned. Is either a UserEmailId type or a IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.UserEmailId or IO[bytes] - :return: UserApiKeyResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.UserApiKeyResponse] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "UserEmailId") - else: - _json = None - - _request = build_get_api_key_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("UserApiKeyResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def get_elastic_to_azure_subscription_mapping( # pylint: disable=name-too-long - self, **kwargs: Any - ) -> _models.ElasticOrganizationToAzureSubscriptionMappingResponse: - """Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. - - Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. - - :return: ElasticOrganizationToAzureSubscriptionMappingResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticOrganizationToAzureSubscriptionMappingResponse] = kwargs.pop("cls", None) - - _request = build_get_elastic_to_azure_subscription_mapping_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize( - "ElasticOrganizationToAzureSubscriptionMappingResponse", pipeline_response.http_response - ) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - async def _resubscribe_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ResubscribeProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ResubscribeProperties") - else: - _json = None - - _request = build_resubscribe_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_resubscribe( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ResubscribeProperties] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Resubscribe the Elasticsearch Organization. - - Resubscribe the Elasticsearch Organization. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Resubscribe Properties. Default value is None. - :type body: ~azure.mgmt.elastic.models.ResubscribeProperties - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_resubscribe( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Resubscribe the Elasticsearch Organization. - - Resubscribe the Elasticsearch Organization. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Resubscribe Properties. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_resubscribe( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ResubscribeProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> AsyncLROPoller[_models.ElasticMonitorResource]: - """Resubscribe the Elasticsearch Organization. - - Resubscribe the Elasticsearch Organization. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Resubscribe Properties. Is either a ResubscribeProperties type or a IO[bytes] - type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ResubscribeProperties or IO[bytes] - :return: An instance of AsyncLROPoller that returns either ElasticMonitorResource or the result - of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._resubscribe_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.ElasticMonitorResource].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.ElasticMonitorResource]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_patch.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_patch.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_tag_rules_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_tag_rules_operations.py deleted file mode 100644 index 5997a419ea8f..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_tag_rules_operations.py +++ /dev/null @@ -1,486 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, Optional, Type, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._tag_rules_operations import ( - build_create_or_update_request, - build_delete_request, - build_get_request, - build_list_request, -) - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class TagRulesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`tag_rules` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> AsyncIterable["_models.MonitoringTagRules"]: - """List the tag rules for a given monitor resource. - - List the tag rules for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either MonitoringTagRules or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.MonitoringTagRules] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoringTagRulesListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("MonitoringTagRulesListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @overload - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - rule_set_name: str, - body: Optional[_models.MonitoringTagRules] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.MonitoringTagRules: - """Create or update a tag rule set for a given monitor resource. - - Create or update a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :param body: request body of MonitoringTagRules. Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoringTagRules - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - rule_set_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.MonitoringTagRules: - """Create or update a tag rule set for a given monitor resource. - - Create or update a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :param body: request body of MonitoringTagRules. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - rule_set_name: str, - body: Optional[Union[_models.MonitoringTagRules, IO[bytes]]] = None, - **kwargs: Any - ) -> _models.MonitoringTagRules: - """Create or update a tag rule set for a given monitor resource. - - Create or update a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :param body: request body of MonitoringTagRules. Is either a MonitoringTagRules type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoringTagRules or IO[bytes] - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "MonitoringTagRules") - else: - _json = None - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("MonitoringTagRules", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def get( - self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any - ) -> _models.MonitoringTagRules: - """Get a tag rule set for a given monitor resource. - - Get a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("MonitoringTagRules", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - async def _delete_initial( - self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202, 204]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_delete( - self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any - ) -> AsyncLROPoller[None]: - """Delete a tag rule set for a given monitor resource. - - Delete a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._delete_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_traffic_filters_operations.py deleted file mode 100644 index 5ed5c2ee55bf..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_traffic_filters_operations.py +++ /dev/null @@ -1,115 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._traffic_filters_operations import build_delete_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class TrafficFiltersOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`traffic_filters` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> None: - """Delete traffic filter from the account. - - Delete traffic filter from the account. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_upgradable_versions_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_upgradable_versions_operations.py deleted file mode 100644 index 61fc5593e891..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_upgradable_versions_operations.py +++ /dev/null @@ -1,116 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._upgradable_versions_operations import build_details_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class UpgradableVersionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`upgradable_versions` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def details( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> _models.UpgradableVersionsList: - """List of upgradable versions for a given monitor resource. - - List of upgradable versions for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: UpgradableVersionsList or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UpgradableVersionsList - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.UpgradableVersionsList] = kwargs.pop("cls", None) - - _request = build_details_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("UpgradableVersionsList", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_collection_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_collection_operations.py deleted file mode 100644 index 8b0ccf3c67d1..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_collection_operations.py +++ /dev/null @@ -1,193 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Optional, Type, TypeVar, Union, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._vm_collection_operations import build_update_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class VMCollectionOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`vm_collection` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - async def update( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.VMCollectionUpdate] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> None: - """Update the vm details that will be monitored by the Elastic monitor resource. - - Update the vm details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: VM resource Id. Default value is None. - :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def update( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> None: - """Update the vm details that will be monitored by the Elastic monitor resource. - - Update the vm details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: VM resource Id. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def update( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.VMCollectionUpdate, IO[bytes]]] = None, - **kwargs: Any - ) -> None: - """Update the vm details that will be monitored by the Elastic monitor resource. - - Update the vm details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: VM resource Id. Is either a VMCollectionUpdate type or a IO[bytes] type. Default - value is None. - :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate or IO[bytes] - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "VMCollectionUpdate") - else: - _json = None - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_host_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_host_operations.py deleted file mode 100644 index 05240605c320..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_host_operations.py +++ /dev/null @@ -1,142 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, Type, TypeVar -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._vm_host_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class VMHostOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`vm_host` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> AsyncIterable["_models.VMResources"]: - """List the vm resources currently being monitored by the Elastic monitor resource. - - List the vm resources currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either VMResources or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.elastic.models.VMResources] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.VMHostListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("VMHostListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_ingestion_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_ingestion_operations.py deleted file mode 100644 index d0a726a92fb5..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/aio/operations/_vm_ingestion_operations.py +++ /dev/null @@ -1,116 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._vm_ingestion_operations import build_details_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class VMIngestionOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.aio.MicrosoftElastic`'s - :attr:`vm_ingestion` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def details( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> _models.VMIngestionDetailsResponse: - """List the vm ingestion details that will be monitored by the Elastic monitor resource. - - List the vm ingestion details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: VMIngestionDetailsResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.VMIngestionDetailsResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.VMIngestionDetailsResponse] = kwargs.pop("cls", None) - - _request = build_details_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("VMIngestionDetailsResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/__init__.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/__init__.py index e7cb4bb33964..54b88b188459 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/__init__.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/__init__.py @@ -2,103 +2,118 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +# pylint: disable=wrong-import-position -from ._models_py3 import BillingInfoResponse -from ._models_py3 import CompanyInfo -from ._models_py3 import ConnectedPartnerResourceProperties -from ._models_py3 import ConnectedPartnerResourcesListFormat -from ._models_py3 import ConnectedPartnerResourcesListResponse -from ._models_py3 import DeploymentInfoResponse -from ._models_py3 import ElasticCloudDeployment -from ._models_py3 import ElasticCloudUser -from ._models_py3 import ElasticMonitorResource -from ._models_py3 import ElasticMonitorResourceListResponse -from ._models_py3 import ElasticMonitorResourceUpdateParameters -from ._models_py3 import ElasticMonitorUpgrade -from ._models_py3 import ElasticOrganizationToAzureSubscriptionMappingResponse -from ._models_py3 import ElasticOrganizationToAzureSubscriptionMappingResponseProperties -from ._models_py3 import ElasticProperties -from ._models_py3 import ElasticTrafficFilter -from ._models_py3 import ElasticTrafficFilterResponse -from ._models_py3 import ElasticTrafficFilterRule -from ._models_py3 import ElasticVersionListFormat -from ._models_py3 import ElasticVersionListProperties -from ._models_py3 import ElasticVersionsListResponse -from ._models_py3 import ErrorAdditionalInfo -from ._models_py3 import ErrorDetail -from ._models_py3 import ErrorResponse -from ._models_py3 import ErrorResponseBody -from ._models_py3 import ExternalUserCreationResponse -from ._models_py3 import ExternalUserInfo -from ._models_py3 import FilteringTag -from ._models_py3 import IdentityProperties -from ._models_py3 import LogRules -from ._models_py3 import MarketplaceSaaSInfo -from ._models_py3 import MarketplaceSaaSInfoMarketplaceSubscription -from ._models_py3 import MonitorProperties -from ._models_py3 import MonitoredResource -from ._models_py3 import MonitoredResourceListResponse -from ._models_py3 import MonitoredSubscription -from ._models_py3 import MonitoredSubscriptionProperties -from ._models_py3 import MonitoredSubscriptionPropertiesList -from ._models_py3 import MonitoringTagRules -from ._models_py3 import MonitoringTagRulesListResponse -from ._models_py3 import MonitoringTagRulesProperties -from ._models_py3 import OpenAIIntegrationProperties -from ._models_py3 import OpenAIIntegrationRPModel -from ._models_py3 import OpenAIIntegrationRPModelListResponse -from ._models_py3 import OpenAIIntegrationStatusResponse -from ._models_py3 import OpenAIIntegrationStatusResponseProperties -from ._models_py3 import OperationDisplay -from ._models_py3 import OperationListResult -from ._models_py3 import OperationResult -from ._models_py3 import PartnerBillingEntity -from ._models_py3 import PlanDetails -from ._models_py3 import ResourceProviderDefaultErrorResponse -from ._models_py3 import ResourceSku -from ._models_py3 import ResubscribeProperties -from ._models_py3 import SubscriptionList -from ._models_py3 import SystemData -from ._models_py3 import UpgradableVersionsList -from ._models_py3 import UserApiKeyResponse -from ._models_py3 import UserApiKeyResponseProperties -from ._models_py3 import UserEmailId -from ._models_py3 import UserInfo -from ._models_py3 import VMCollectionUpdate -from ._models_py3 import VMHostListResponse -from ._models_py3 import VMIngestionDetailsResponse -from ._models_py3 import VMResources +from typing import TYPE_CHECKING -from ._microsoft_elastic_enums import CreatedByType -from ._microsoft_elastic_enums import ElasticDeploymentStatus -from ._microsoft_elastic_enums import LiftrResourceCategories -from ._microsoft_elastic_enums import ManagedIdentityTypes -from ._microsoft_elastic_enums import MonitoringStatus -from ._microsoft_elastic_enums import Operation -from ._microsoft_elastic_enums import OperationName -from ._microsoft_elastic_enums import ProvisioningState -from ._microsoft_elastic_enums import SendingLogs -from ._microsoft_elastic_enums import Status -from ._microsoft_elastic_enums import TagAction -from ._microsoft_elastic_enums import Type +if TYPE_CHECKING: + from ._patch import * # pylint: disable=unused-wildcard-import + + +from ._models import ( # type: ignore + ArmResponseElasticMonitorResource, + BillingInfoResponse, + CompanyInfo, + ConnectedPartnerResourceProperties, + ConnectedPartnerResourcesListFormat, + CreateOrUpdateParameterBody, + DeploymentInfoResponse, + ElasticCloudDeployment, + ElasticCloudUser, + ElasticMonitorResource, + ElasticMonitorResourceUpdateParameters, + ElasticMonitorUpgrade, + ElasticOrganizationToAzureSubscriptionMappingResponse, + ElasticOrganizationToAzureSubscriptionMappingResponseProperties, + ElasticProperties, + ElasticTrafficFilter, + ElasticTrafficFilterResponse, + ElasticTrafficFilterRule, + ElasticVersionListFormat, + ElasticVersionListProperties, + ErrorAdditionalInfo, + ErrorDetail, + ErrorResponse, + ErrorResponseBody, + ExternalUserCreationResponse, + ExternalUserInfo, + FilteringTag, + IdentityProperties, + LogRules, + MarketplaceSaaSInfo, + MarketplaceSaaSInfoMarketplaceSubscription, + MonitorProperties, + MonitoredResource, + MonitoredSubscription, + MonitoredSubscriptionProperties, + MonitoringTagRules, + MonitoringTagRulesProperties, + OkResponse, + OpenAIIntegrationProperties, + OpenAIIntegrationRPModel, + OpenAIIntegrationStatusResponse, + OpenAIIntegrationStatusResponseProperties, + OperationDisplay, + OperationResult, + PartnerBillingEntity, + PlanDetails, + ProjectDetails, + ProxyResource, + Resource, + ResourceProviderDefaultErrorResponse, + ResourceSku, + ResubscribeParameterBody, + ResubscribeProperties, + SubscriptionList, + SystemData, + TrackedResource, + UpgradableVersionsList, + UpgradeParameterBody, + UserApiKeyResponse, + UserApiKeyResponseProperties, + UserEmailId, + UserInfo, + VMCollectionUpdate, + VMCollectionUpdateParameterBody, + VMIngestionDetailsResponse, + VMResources, +) + +from ._enums import ( # type: ignore + ConfigurationType, + CreatedByType, + ElasticDeploymentStatus, + HostingType, + LiftrResourceCategories, + ManagedIdentityTypes, + MonitoringStatus, + Operation, + OperationName, + ProjectType, + ProvisioningState, + SendingLogs, + Status, + TagAction, + Type, +) from ._patch import __all__ as _patch_all -from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import * from ._patch import patch_sdk as _patch_sdk __all__ = [ + "ArmResponseElasticMonitorResource", "BillingInfoResponse", "CompanyInfo", "ConnectedPartnerResourceProperties", "ConnectedPartnerResourcesListFormat", - "ConnectedPartnerResourcesListResponse", + "CreateOrUpdateParameterBody", "DeploymentInfoResponse", "ElasticCloudDeployment", "ElasticCloudUser", "ElasticMonitorResource", - "ElasticMonitorResourceListResponse", "ElasticMonitorResourceUpdateParameters", "ElasticMonitorUpgrade", "ElasticOrganizationToAzureSubscriptionMappingResponse", @@ -109,7 +124,6 @@ "ElasticTrafficFilterRule", "ElasticVersionListFormat", "ElasticVersionListProperties", - "ElasticVersionsListResponse", "ErrorAdditionalInfo", "ErrorDetail", "ErrorResponse", @@ -123,49 +137,54 @@ "MarketplaceSaaSInfoMarketplaceSubscription", "MonitorProperties", "MonitoredResource", - "MonitoredResourceListResponse", "MonitoredSubscription", "MonitoredSubscriptionProperties", - "MonitoredSubscriptionPropertiesList", "MonitoringTagRules", - "MonitoringTagRulesListResponse", "MonitoringTagRulesProperties", + "OkResponse", "OpenAIIntegrationProperties", "OpenAIIntegrationRPModel", - "OpenAIIntegrationRPModelListResponse", "OpenAIIntegrationStatusResponse", "OpenAIIntegrationStatusResponseProperties", "OperationDisplay", - "OperationListResult", "OperationResult", "PartnerBillingEntity", "PlanDetails", + "ProjectDetails", + "ProxyResource", + "Resource", "ResourceProviderDefaultErrorResponse", "ResourceSku", + "ResubscribeParameterBody", "ResubscribeProperties", "SubscriptionList", "SystemData", + "TrackedResource", "UpgradableVersionsList", + "UpgradeParameterBody", "UserApiKeyResponse", "UserApiKeyResponseProperties", "UserEmailId", "UserInfo", "VMCollectionUpdate", - "VMHostListResponse", + "VMCollectionUpdateParameterBody", "VMIngestionDetailsResponse", "VMResources", + "ConfigurationType", "CreatedByType", "ElasticDeploymentStatus", + "HostingType", "LiftrResourceCategories", "ManagedIdentityTypes", "MonitoringStatus", "Operation", "OperationName", + "ProjectType", "ProvisioningState", "SendingLogs", "Status", "TagAction", "Type", ] -__all__.extend([p for p in _patch_all if p not in __all__]) +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_microsoft_elastic_enums.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_enums.py similarity index 72% rename from sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_microsoft_elastic_enums.py rename to sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_enums.py index ddf60fa4b1b8..da70765ea5eb 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_microsoft_elastic_enums.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_enums.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -10,13 +10,26 @@ from azure.core import CaseInsensitiveEnumMeta +class ConfigurationType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Configuration type of the Elasticsearch project.""" + + GENERAL_PURPOSE = "GeneralPurpose" + VECTOR = "Vector" + TIME_SERIES = "TimeSeries" + NOT_APPLICABLE = "NotApplicable" + + class CreatedByType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """The type of identity that created the resource.""" + """The kind of entity that created the resource.""" USER = "User" + """The entity was created by a user.""" APPLICATION = "Application" + """The entity was created by an application.""" MANAGED_IDENTITY = "ManagedIdentity" + """The entity was created by a managed identity.""" KEY = "Key" + """The entity was created by a key.""" class ElasticDeploymentStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -26,8 +39,15 @@ class ElasticDeploymentStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): UNHEALTHY = "Unhealthy" +class HostingType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Hosting type of the monitor resource - either Hosted deployments or Serverless Projects.""" + + HOSTED = "Hosted" + SERVERLESS = "Serverless" + + class LiftrResourceCategories(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """LiftrResourceCategories.""" + """Type of LiftrResourceCategories.""" UNKNOWN = "Unknown" MONITOR_LOGS = "MonitorLogs" @@ -63,6 +83,15 @@ class OperationName(str, Enum, metaclass=CaseInsensitiveEnumMeta): DELETE = "Delete" +class ProjectType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Project type; ex: Elasticsearch / Observability / Security.""" + + ELASTICSEARCH = "Elasticsearch" + OBSERVABILITY = "Observability" + SECURITY = "Security" + NOT_APPLICABLE = "NotApplicable" + + class ProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Provisioning state of Elastic resource.""" diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models.py new file mode 100644 index 000000000000..ecc057e6cb19 --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models.py @@ -0,0 +1,2609 @@ +# pylint: disable=line-too-long,useless-suppression,too-many-lines +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +# pylint: disable=useless-super-delegation + +import datetime +from typing import Any, Dict, List, Mapping, Optional, TYPE_CHECKING, Union, overload + +from .._utils.model_base import Model as _Model, rest_field + +if TYPE_CHECKING: + from .. import models as _models + + +class ArmResponseElasticMonitorResource(_Model): + """Azure operation completed successfully. + + :ivar body: The body type of the operation request or response. Required. + :vartype body: ~azure.mgmt.elastic.models.ElasticMonitorResource + """ + + body: "_models.ElasticMonitorResource" = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The body type of the operation request or response. Required.""" + + @overload + def __init__( + self, + *, + body: "_models.ElasticMonitorResource", + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class BillingInfoResponse(_Model): + """Marketplace Subscription and Organization details to which resource gets billed into. + + :ivar marketplace_saas_info: Marketplace Subscription details. + :vartype marketplace_saas_info: ~azure.mgmt.elastic.models.MarketplaceSaaSInfo + :ivar partner_billing_entity: Partner Billing Entity details: Organization Info. + :vartype partner_billing_entity: ~azure.mgmt.elastic.models.PartnerBillingEntity + """ + + marketplace_saas_info: Optional["_models.MarketplaceSaaSInfo"] = rest_field( + name="marketplaceSaasInfo", visibility=["read", "create", "update", "delete", "query"] + ) + """Marketplace Subscription details.""" + partner_billing_entity: Optional["_models.PartnerBillingEntity"] = rest_field( + name="partnerBillingEntity", visibility=["read", "create", "update", "delete", "query"] + ) + """Partner Billing Entity details: Organization Info.""" + + @overload + def __init__( + self, + *, + marketplace_saas_info: Optional["_models.MarketplaceSaaSInfo"] = None, + partner_billing_entity: Optional["_models.PartnerBillingEntity"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class CompanyInfo(_Model): + """Company information of the user to be passed to partners. + + :ivar domain: Domain of the company. + :vartype domain: str + :ivar business: Business of the company. + :vartype business: str + :ivar employees_number: Number of employees in the company. + :vartype employees_number: str + :ivar state: State of the company location. + :vartype state: str + :ivar country: Country of the company location. + :vartype country: str + """ + + domain: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Domain of the company.""" + business: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Business of the company.""" + employees_number: Optional[str] = rest_field( + name="employeesNumber", visibility=["read", "create", "update", "delete", "query"] + ) + """Number of employees in the company.""" + state: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """State of the company location.""" + country: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Country of the company location.""" + + @overload + def __init__( + self, + *, + domain: Optional[str] = None, + business: Optional[str] = None, + employees_number: Optional[str] = None, + state: Optional[str] = None, + country: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ConnectedPartnerResourceProperties(_Model): + """Connected Partner Resource Properties. + + :ivar partner_deployment_name: Elastic resource name. + :vartype partner_deployment_name: str + :ivar partner_deployment_uri: URL of the resource in Elastic cloud. + :vartype partner_deployment_uri: str + :ivar azure_resource_id: The azure resource Id of the resource. + :vartype azure_resource_id: str + :ivar location: The location of the resource. + :vartype location: str + :ivar type: The hosting type of the resource. + :vartype type: str + """ + + partner_deployment_name: Optional[str] = rest_field( + name="partnerDeploymentName", visibility=["read", "create", "update", "delete", "query"] + ) + """Elastic resource name.""" + partner_deployment_uri: Optional[str] = rest_field( + name="partnerDeploymentUri", visibility=["read", "create", "update", "delete", "query"] + ) + """URL of the resource in Elastic cloud.""" + azure_resource_id: Optional[str] = rest_field( + name="azureResourceId", visibility=["read", "create", "update", "delete", "query"] + ) + """The azure resource Id of the resource.""" + location: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The location of the resource.""" + type: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The hosting type of the resource.""" + + @overload + def __init__( + self, + *, + partner_deployment_name: Optional[str] = None, + partner_deployment_uri: Optional[str] = None, + azure_resource_id: Optional[str] = None, + location: Optional[str] = None, + type: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ConnectedPartnerResourcesListFormat(_Model): + """Connected Partner Resources List Format. + + :ivar properties: Connected Partner Resource Properties. + :vartype properties: ~azure.mgmt.elastic.models.ConnectedPartnerResourceProperties + """ + + properties: Optional["_models.ConnectedPartnerResourceProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Connected Partner Resource Properties.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.ConnectedPartnerResourceProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class CreateOrUpdateParameterBody(_Model): + """CreateOrUpdateParameterBody. + + :ivar body: Elastic External User Creation Parameters. + :vartype body: ~azure.mgmt.elastic.models.ExternalUserInfo + """ + + body: Optional["_models.ExternalUserInfo"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Elastic External User Creation Parameters.""" + + @overload + def __init__( + self, + *, + body: Optional["_models.ExternalUserInfo"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class DeploymentInfoResponse(_Model): + """The properties of deployment in Elastic cloud corresponding to the Elastic monitor resource. + + :ivar status: The Elastic deployment status. Known values are: "Healthy" and "Unhealthy". + :vartype status: str or ~azure.mgmt.elastic.models.ElasticDeploymentStatus + :ivar version: Version of the elasticsearch in Elastic cloud deployment. + :vartype version: str + :ivar memory_capacity: RAM capacity of the elasticsearch in Elastic cloud deployment. + :vartype memory_capacity: str + :ivar disk_capacity: Disk capacity of the elasticsearch in Elastic cloud deployment. + :vartype disk_capacity: str + :ivar elasticsearch_end_point: Elasticsearch endpoint in Elastic cloud deployment. This is + either the aliased_endpoint if available, or the service_url otherwise. + :vartype elasticsearch_end_point: str + :ivar deployment_url: Deployment URL of the elasticsearch in Elastic cloud deployment. + :vartype deployment_url: str + :ivar marketplace_saas_info: Marketplace SaaS Info of the resource. + :vartype marketplace_saas_info: ~azure.mgmt.elastic.models.MarketplaceSaaSInfo + :ivar project_type: Project Type - Applicable for Serverless only. + :vartype project_type: str + :ivar configuration_type: ConfigurationType Type - Applicable for Serverless only. + :vartype configuration_type: str + """ + + status: Optional[Union[str, "_models.ElasticDeploymentStatus"]] = rest_field(visibility=["read"]) + """The Elastic deployment status. Known values are: \"Healthy\" and \"Unhealthy\".""" + version: Optional[str] = rest_field(visibility=["read"]) + """Version of the elasticsearch in Elastic cloud deployment.""" + memory_capacity: Optional[str] = rest_field(name="memoryCapacity", visibility=["read"]) + """RAM capacity of the elasticsearch in Elastic cloud deployment.""" + disk_capacity: Optional[str] = rest_field(name="diskCapacity", visibility=["read"]) + """Disk capacity of the elasticsearch in Elastic cloud deployment.""" + elasticsearch_end_point: Optional[str] = rest_field(name="elasticsearchEndPoint", visibility=["read"]) + """Elasticsearch endpoint in Elastic cloud deployment. This is either the aliased_endpoint if + available, or the service_url otherwise.""" + deployment_url: Optional[str] = rest_field(name="deploymentUrl", visibility=["read"]) + """Deployment URL of the elasticsearch in Elastic cloud deployment.""" + marketplace_saas_info: Optional["_models.MarketplaceSaaSInfo"] = rest_field( + name="marketplaceSaasInfo", visibility=["read"] + ) + """Marketplace SaaS Info of the resource.""" + project_type: Optional[str] = rest_field(name="projectType", visibility=["read"]) + """Project Type - Applicable for Serverless only.""" + configuration_type: Optional[str] = rest_field(name="configurationType", visibility=["read"]) + """ConfigurationType Type - Applicable for Serverless only.""" + + +class ElasticCloudDeployment(_Model): + """Details of the user's elastic deployment associated with the monitor resource. + + :ivar name: Elastic deployment name. + :vartype name: str + :ivar deployment_id: Elastic deployment Id. + :vartype deployment_id: str + :ivar azure_subscription_id: Associated Azure subscription Id for the elastic deployment. + :vartype azure_subscription_id: str + :ivar elasticsearch_region: Region where Deployment at Elastic side took place. + :vartype elasticsearch_region: str + :ivar elasticsearch_service_url: Elasticsearch ingestion endpoint of the Elastic deployment. + :vartype elasticsearch_service_url: str + :ivar kibana_service_url: Kibana endpoint of the Elastic deployment. + :vartype kibana_service_url: str + :ivar kibana_sso_url: Kibana dashboard sso URL of the Elastic deployment. + :vartype kibana_sso_url: str + """ + + name: Optional[str] = rest_field(visibility=["read"]) + """Elastic deployment name.""" + deployment_id: Optional[str] = rest_field(name="deploymentId", visibility=["read"]) + """Elastic deployment Id.""" + azure_subscription_id: Optional[str] = rest_field(name="azureSubscriptionId", visibility=["read"]) + """Associated Azure subscription Id for the elastic deployment.""" + elasticsearch_region: Optional[str] = rest_field(name="elasticsearchRegion", visibility=["read"]) + """Region where Deployment at Elastic side took place.""" + elasticsearch_service_url: Optional[str] = rest_field(name="elasticsearchServiceUrl", visibility=["read"]) + """Elasticsearch ingestion endpoint of the Elastic deployment.""" + kibana_service_url: Optional[str] = rest_field(name="kibanaServiceUrl", visibility=["read"]) + """Kibana endpoint of the Elastic deployment.""" + kibana_sso_url: Optional[str] = rest_field(name="kibanaSsoUrl", visibility=["read"]) + """Kibana dashboard sso URL of the Elastic deployment.""" + + +class ElasticCloudUser(_Model): + """Details of the user's elastic account. + + :ivar email_address: Email of the Elastic User Account. + :vartype email_address: str + :ivar id: User Id of the elastic account of the User. + :vartype id: str + :ivar elastic_cloud_sso_default_url: Elastic cloud default dashboard sso URL of the Elastic + user account. + :vartype elastic_cloud_sso_default_url: str + """ + + email_address: Optional[str] = rest_field(name="emailAddress", visibility=["read"]) + """Email of the Elastic User Account.""" + id: Optional[str] = rest_field(visibility=["read"]) + """User Id of the elastic account of the User.""" + elastic_cloud_sso_default_url: Optional[str] = rest_field(name="elasticCloudSsoDefaultUrl", visibility=["read"]) + """Elastic cloud default dashboard sso URL of the Elastic user account.""" + + +class Resource(_Model): + """Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.elastic.models.SystemData + """ + + id: Optional[str] = rest_field(visibility=["read"]) + """Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}.""" + name: Optional[str] = rest_field(visibility=["read"]) + """The name of the resource.""" + type: Optional[str] = rest_field(visibility=["read"]) + """The type of the resource. E.g. \"Microsoft.Compute/virtualMachines\" or + \"Microsoft.Storage/storageAccounts\".""" + system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"]) + """Azure Resource Manager metadata containing createdBy and modifiedBy information.""" + + +class TrackedResource(Resource): + """Tracked Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.elastic.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + """ + + tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Resource tags.""" + location: str = rest_field(visibility=["read", "create"]) + """The geo-location where the resource lives. Required.""" + + @overload + def __init__( + self, + *, + location: str, + tags: Optional[Dict[str, str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticMonitorResource(TrackedResource): + """Monitor resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.elastic.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + :ivar properties: Properties of the monitor resource. + :vartype properties: ~azure.mgmt.elastic.models.MonitorProperties + :ivar kind: The kind of the Elastic resource - observability, security, search etc. + :vartype kind: str + :ivar sku: SKU of the monitor resource. + :vartype sku: ~azure.mgmt.elastic.models.ResourceSku + :ivar identity: Identity properties of the monitor resource. + :vartype identity: ~azure.mgmt.elastic.models.IdentityProperties + """ + + properties: Optional["_models.MonitorProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of the monitor resource.""" + kind: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The kind of the Elastic resource - observability, security, search etc.""" + sku: Optional["_models.ResourceSku"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """SKU of the monitor resource.""" + identity: Optional["_models.IdentityProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Identity properties of the monitor resource.""" + + @overload + def __init__( + self, + *, + location: str, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.MonitorProperties"] = None, + kind: Optional[str] = None, + sku: Optional["_models.ResourceSku"] = None, + identity: Optional["_models.IdentityProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticMonitorResourceUpdateParameters(_Model): + """Monitor resource update parameters. + + :ivar tags: elastic monitor resource tags. + :vartype tags: dict[str, str] + """ + + tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """elastic monitor resource tags.""" + + @overload + def __init__( + self, + *, + tags: Optional[Dict[str, str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticMonitorUpgrade(_Model): + """Upgrade elastic monitor version. + + :ivar version: Version to which the elastic monitor should be upgraded to. + :vartype version: str + """ + + version: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Version to which the elastic monitor should be upgraded to.""" + + @overload + def __init__( + self, + *, + version: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticOrganizationToAzureSubscriptionMappingResponse(_Model): # pylint: disable=name-too-long + """The Azure Subscription ID to which the Organization of the logged in user belongs and gets + billed into. + + :ivar properties: The properties of Azure Subscription ID to which the Organization of the + logged in user belongs and gets billed into. + :vartype properties: + ~azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponseProperties + """ + + properties: Optional["_models.ElasticOrganizationToAzureSubscriptionMappingResponseProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The properties of Azure Subscription ID to which the Organization of the logged in user belongs + and gets billed into.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.ElasticOrganizationToAzureSubscriptionMappingResponseProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticOrganizationToAzureSubscriptionMappingResponseProperties(_Model): # pylint: disable=name-too-long + """The properties of Azure Subscription ID to which the Organization of the logged in user belongs + and gets billed into. + + :ivar billed_azure_subscription_id: The Azure Subscription ID to which the Organization belongs + and gets billed into. This is empty for a new user OR a user without an Elastic Organization. + :vartype billed_azure_subscription_id: str + :ivar marketplace_saas_info: Marketplace SaaS Info of the resource. + :vartype marketplace_saas_info: ~azure.mgmt.elastic.models.MarketplaceSaaSInfo + :ivar elastic_organization_id: The Elastic Organization Id. + :vartype elastic_organization_id: str + :ivar elastic_organization_name: The Elastic Organization Name. + :vartype elastic_organization_name: str + """ + + billed_azure_subscription_id: Optional[str] = rest_field( + name="billedAzureSubscriptionId", visibility=["read", "create", "update", "delete", "query"] + ) + """The Azure Subscription ID to which the Organization belongs and gets billed into. This is empty + for a new user OR a user without an Elastic Organization.""" + marketplace_saas_info: Optional["_models.MarketplaceSaaSInfo"] = rest_field( + name="marketplaceSaasInfo", visibility=["read"] + ) + """Marketplace SaaS Info of the resource.""" + elastic_organization_id: Optional[str] = rest_field( + name="elasticOrganizationId", visibility=["read", "create", "update", "delete", "query"] + ) + """The Elastic Organization Id.""" + elastic_organization_name: Optional[str] = rest_field( + name="elasticOrganizationName", visibility=["read", "create", "update", "delete", "query"] + ) + """The Elastic Organization Name.""" + + @overload + def __init__( + self, + *, + billed_azure_subscription_id: Optional[str] = None, + elastic_organization_id: Optional[str] = None, + elastic_organization_name: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticProperties(_Model): + """Elastic Resource Properties. + + :ivar elastic_cloud_user: Details of the user's elastic account. + :vartype elastic_cloud_user: ~azure.mgmt.elastic.models.ElasticCloudUser + :ivar elastic_cloud_deployment: Details of the elastic cloud deployment. + :vartype elastic_cloud_deployment: ~azure.mgmt.elastic.models.ElasticCloudDeployment + """ + + elastic_cloud_user: Optional["_models.ElasticCloudUser"] = rest_field( + name="elasticCloudUser", visibility=["read", "create", "update", "delete", "query"] + ) + """Details of the user's elastic account.""" + elastic_cloud_deployment: Optional["_models.ElasticCloudDeployment"] = rest_field( + name="elasticCloudDeployment", visibility=["read", "create", "update", "delete", "query"] + ) + """Details of the elastic cloud deployment.""" + + @overload + def __init__( + self, + *, + elastic_cloud_user: Optional["_models.ElasticCloudUser"] = None, + elastic_cloud_deployment: Optional["_models.ElasticCloudDeployment"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticTrafficFilter(_Model): + """Elastic traffic filter object. + + :ivar id: Id of the elastic filter. + :vartype id: str + :ivar name: Name of the elastic filter. + :vartype name: str + :ivar description: Description of the elastic filter. + :vartype description: str + :ivar region: Region of the elastic filter. + :vartype region: str + :ivar type: Type of the elastic filter. Known values are: "ip" and "azure_private_endpoint". + :vartype type: str or ~azure.mgmt.elastic.models.Type + :ivar include_by_default: IncludeByDefault for the elastic filter. + :vartype include_by_default: bool + :ivar rules: Rules in the elastic filter. + :vartype rules: list[~azure.mgmt.elastic.models.ElasticTrafficFilterRule] + """ + + id: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Id of the elastic filter.""" + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Name of the elastic filter.""" + description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Description of the elastic filter.""" + region: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Region of the elastic filter.""" + type: Optional[Union[str, "_models.Type"]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Type of the elastic filter. Known values are: \"ip\" and \"azure_private_endpoint\".""" + include_by_default: Optional[bool] = rest_field( + name="includeByDefault", visibility=["read", "create", "update", "delete", "query"] + ) + """IncludeByDefault for the elastic filter.""" + rules: Optional[List["_models.ElasticTrafficFilterRule"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Rules in the elastic filter.""" + + @overload + def __init__( + self, + *, + id: Optional[str] = None, # pylint: disable=redefined-builtin + name: Optional[str] = None, + description: Optional[str] = None, + region: Optional[str] = None, + type: Optional[Union[str, "_models.Type"]] = None, + include_by_default: Optional[bool] = None, + rules: Optional[List["_models.ElasticTrafficFilterRule"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticTrafficFilterResponse(_Model): + """List of elastic traffic filters in the account. + + :ivar rulesets: List of elastic traffic filters in the account. + :vartype rulesets: list[~azure.mgmt.elastic.models.ElasticTrafficFilter] + """ + + rulesets: Optional[List["_models.ElasticTrafficFilter"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """List of elastic traffic filters in the account.""" + + @overload + def __init__( + self, + *, + rulesets: Optional[List["_models.ElasticTrafficFilter"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticTrafficFilterRule(_Model): + """Elastic traffic filter rule object. + + :ivar source: IP of the elastic filter rule. + :vartype source: str + :ivar description: Description of the elastic filter rule. + :vartype description: str + :ivar azure_endpoint_guid: Guid of Private Endpoint in the elastic filter rule. + :vartype azure_endpoint_guid: str + :ivar azure_endpoint_name: Name of the Private Endpoint in the elastic filter rule. + :vartype azure_endpoint_name: str + :ivar id: Id of the elastic filter rule. + :vartype id: str + """ + + source: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """IP of the elastic filter rule.""" + description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Description of the elastic filter rule.""" + azure_endpoint_guid: Optional[str] = rest_field( + name="azureEndpointGuid", visibility=["read", "create", "update", "delete", "query"] + ) + """Guid of Private Endpoint in the elastic filter rule.""" + azure_endpoint_name: Optional[str] = rest_field( + name="azureEndpointName", visibility=["read", "create", "update", "delete", "query"] + ) + """Name of the Private Endpoint in the elastic filter rule.""" + id: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Id of the elastic filter rule.""" + + @overload + def __init__( + self, + *, + source: Optional[str] = None, + description: Optional[str] = None, + azure_endpoint_guid: Optional[str] = None, + azure_endpoint_name: Optional[str] = None, + id: Optional[str] = None, # pylint: disable=redefined-builtin + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticVersionListFormat(_Model): + """Elastic Version List Format. + + :ivar properties: Elastic Version Properties. + :vartype properties: ~azure.mgmt.elastic.models.ElasticVersionListProperties + """ + + properties: Optional["_models.ElasticVersionListProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Elastic Version Properties.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.ElasticVersionListProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ElasticVersionListProperties(_Model): + """Elastic Version Properties. + + :ivar version: Available elastic version of the given region. + :vartype version: str + """ + + version: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Available elastic version of the given region.""" + + @overload + def __init__( + self, + *, + version: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ErrorAdditionalInfo(_Model): + """The resource management error additional info. + + :ivar type: The additional info type. + :vartype type: str + :ivar info: The additional info. + :vartype info: any + """ + + type: Optional[str] = rest_field(visibility=["read"]) + """The additional info type.""" + info: Optional[Any] = rest_field(visibility=["read"]) + """The additional info.""" + + +class ErrorDetail(_Model): + """The error detail. + + :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.elastic.models.ErrorDetail] + :ivar additional_info: The error additional info. + :vartype additional_info: list[~azure.mgmt.elastic.models.ErrorAdditionalInfo] + """ + + code: Optional[str] = rest_field(visibility=["read"]) + """The error code.""" + message: Optional[str] = rest_field(visibility=["read"]) + """The error message.""" + target: Optional[str] = rest_field(visibility=["read"]) + """The error target.""" + details: Optional[List["_models.ErrorDetail"]] = rest_field(visibility=["read"]) + """The error details.""" + additional_info: Optional[List["_models.ErrorAdditionalInfo"]] = rest_field( + name="additionalInfo", visibility=["read"] + ) + """The error additional info.""" + + +class ErrorResponse(_Model): + """Error response. + + :ivar error: The error object. + :vartype error: ~azure.mgmt.elastic.models.ErrorDetail + """ + + error: Optional["_models.ErrorDetail"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The error object.""" + + @overload + def __init__( + self, + *, + error: Optional["_models.ErrorDetail"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ErrorResponseBody(_Model): + """Error response body. + + :ivar code: Error code. + :vartype code: str + :ivar message: Error message. + :vartype message: str + :ivar target: Error target. + :vartype target: str + :ivar details: Error details. + :vartype details: list[~azure.mgmt.elastic.models.ErrorResponseBody] + """ + + code: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Error code.""" + message: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Error message.""" + target: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Error target.""" + details: Optional[List["_models.ErrorResponseBody"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Error details.""" + + @overload + def __init__( + self, + *, + code: Optional[str] = None, + message: Optional[str] = None, + target: Optional[str] = None, + details: Optional[List["_models.ErrorResponseBody"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ExternalUserCreationResponse(_Model): + """The properties of the response we got from elastic while creating external user. + + :ivar created: Shows if user is created or updated. + :vartype created: bool + """ + + created: Optional[bool] = rest_field(visibility=["read"]) + """Shows if user is created or updated.""" + + +class ExternalUserInfo(_Model): + """The properties of the request required for creating user on elastic side. + + :ivar user_name: Username of the user to be created or updated. + :vartype user_name: str + :ivar full_name: Full name of the user to be created or updated. + :vartype full_name: str + :ivar password: Password of the user to be created or updated. + :vartype password: str + :ivar email_id: Email id of the user to be created or updated. + :vartype email_id: str + :ivar roles: Roles to be assigned for created or updated user. + :vartype roles: list[str] + """ + + user_name: Optional[str] = rest_field(name="userName", visibility=["read", "create", "update", "delete", "query"]) + """Username of the user to be created or updated.""" + full_name: Optional[str] = rest_field(name="fullName", visibility=["read", "create", "update", "delete", "query"]) + """Full name of the user to be created or updated.""" + password: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Password of the user to be created or updated.""" + email_id: Optional[str] = rest_field(name="emailId", visibility=["read", "create", "update", "delete", "query"]) + """Email id of the user to be created or updated.""" + roles: Optional[List[str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Roles to be assigned for created or updated user.""" + + @overload + def __init__( + self, + *, + user_name: Optional[str] = None, + full_name: Optional[str] = None, + password: Optional[str] = None, + email_id: Optional[str] = None, + roles: Optional[List[str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class FilteringTag(_Model): + """The definition of a filtering tag. Filtering tags are used for capturing resources and + include/exclude them from being monitored. + + :ivar name: The name (also known as the key) of the tag. + :vartype name: str + :ivar value: The value of the tag. + :vartype value: str + :ivar action: Valid actions for a filtering tag. Known values are: "Include" and "Exclude". + :vartype action: str or ~azure.mgmt.elastic.models.TagAction + """ + + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The name (also known as the key) of the tag.""" + value: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The value of the tag.""" + action: Optional[Union[str, "_models.TagAction"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Valid actions for a filtering tag. Known values are: \"Include\" and \"Exclude\".""" + + @overload + def __init__( + self, + *, + name: Optional[str] = None, + value: Optional[str] = None, + action: Optional[Union[str, "_models.TagAction"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class IdentityProperties(_Model): + """Identity properties. + + :ivar principal_id: The identity ID. + :vartype principal_id: str + :ivar tenant_id: The tenant ID of resource. + :vartype tenant_id: str + :ivar type: Managed identity type. "SystemAssigned" + :vartype type: str or ~azure.mgmt.elastic.models.ManagedIdentityTypes + """ + + principal_id: Optional[str] = rest_field(name="principalId", visibility=["read"]) + """The identity ID.""" + tenant_id: Optional[str] = rest_field(name="tenantId", visibility=["read"]) + """The tenant ID of resource.""" + type: Optional[Union[str, "_models.ManagedIdentityTypes"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Managed identity type. \"SystemAssigned\"""" + + @overload + def __init__( + self, + *, + type: Optional[Union[str, "_models.ManagedIdentityTypes"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class LogRules(_Model): + """Set of rules for sending logs for the Monitor resource. + + :ivar send_aad_logs: Flag specifying if AAD logs should be sent for the Monitor resource. + :vartype send_aad_logs: bool + :ivar send_subscription_logs: Flag specifying if subscription logs should be sent for the + Monitor resource. + :vartype send_subscription_logs: bool + :ivar send_activity_logs: Flag specifying if activity logs from Azure resources should be sent + for the Monitor resource. + :vartype send_activity_logs: bool + :ivar filtering_tags: List of filtering tags to be used for capturing logs. This only takes + effect if SendActivityLogs flag is enabled. If empty, all resources will be captured. If only + Exclude action is specified, the rules will apply to the list of all available resources. If + Include actions are specified, the rules will only include resources with the associated tags. + :vartype filtering_tags: list[~azure.mgmt.elastic.models.FilteringTag] + """ + + send_aad_logs: Optional[bool] = rest_field( + name="sendAadLogs", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag specifying if AAD logs should be sent for the Monitor resource.""" + send_subscription_logs: Optional[bool] = rest_field( + name="sendSubscriptionLogs", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag specifying if subscription logs should be sent for the Monitor resource.""" + send_activity_logs: Optional[bool] = rest_field( + name="sendActivityLogs", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag specifying if activity logs from Azure resources should be sent for the Monitor resource.""" + filtering_tags: Optional[List["_models.FilteringTag"]] = rest_field( + name="filteringTags", visibility=["read", "create", "update", "delete", "query"] + ) + """List of filtering tags to be used for capturing logs. This only takes effect if + SendActivityLogs flag is enabled. If empty, all resources will be captured. If only Exclude + action is specified, the rules will apply to the list of all available resources. If Include + actions are specified, the rules will only include resources with the associated tags.""" + + @overload + def __init__( + self, + *, + send_aad_logs: Optional[bool] = None, + send_subscription_logs: Optional[bool] = None, + send_activity_logs: Optional[bool] = None, + filtering_tags: Optional[List["_models.FilteringTag"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class MarketplaceSaaSInfo(_Model): + """Marketplace SAAS Info of the resource. + + :ivar marketplace_subscription: Marketplace Subscription. + :vartype marketplace_subscription: + ~azure.mgmt.elastic.models.MarketplaceSaaSInfoMarketplaceSubscription + :ivar marketplace_name: Marketplace Subscription Details: SAAS Name. + :vartype marketplace_name: str + :ivar marketplace_resource_id: Marketplace Subscription Details: Resource URI. + :vartype marketplace_resource_id: str + :ivar marketplace_status: Marketplace Subscription Details: SaaS Subscription Status. + :vartype marketplace_status: str + :ivar billed_azure_subscription_id: The Azure Subscription ID to which the Marketplace + Subscription belongs and gets billed into. + :vartype billed_azure_subscription_id: str + :ivar subscribed: Flag specifying if the Marketplace status is subscribed or not. + :vartype subscribed: bool + """ + + marketplace_subscription: Optional["_models.MarketplaceSaaSInfoMarketplaceSubscription"] = rest_field( + name="marketplaceSubscription", visibility=["read", "create", "update", "delete", "query"] + ) + """Marketplace Subscription.""" + marketplace_name: Optional[str] = rest_field( + name="marketplaceName", visibility=["read", "create", "update", "delete", "query"] + ) + """Marketplace Subscription Details: SAAS Name.""" + marketplace_resource_id: Optional[str] = rest_field( + name="marketplaceResourceId", visibility=["read", "create", "update", "delete", "query"] + ) + """Marketplace Subscription Details: Resource URI.""" + marketplace_status: Optional[str] = rest_field( + name="marketplaceStatus", visibility=["read", "create", "update", "delete", "query"] + ) + """Marketplace Subscription Details: SaaS Subscription Status.""" + billed_azure_subscription_id: Optional[str] = rest_field( + name="billedAzureSubscriptionId", visibility=["read", "create", "update", "delete", "query"] + ) + """The Azure Subscription ID to which the Marketplace Subscription belongs and gets billed into.""" + subscribed: Optional[bool] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Flag specifying if the Marketplace status is subscribed or not.""" + + @overload + def __init__( + self, + *, + marketplace_subscription: Optional["_models.MarketplaceSaaSInfoMarketplaceSubscription"] = None, + marketplace_name: Optional[str] = None, + marketplace_resource_id: Optional[str] = None, + marketplace_status: Optional[str] = None, + billed_azure_subscription_id: Optional[str] = None, + subscribed: Optional[bool] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class MarketplaceSaaSInfoMarketplaceSubscription(_Model): # pylint: disable=name-too-long + """Marketplace Subscription. + + :ivar id: Marketplace Subscription Id. This is a GUID-formatted string. + :vartype id: str + :ivar publisher_id: Publisher Id of the Marketplace offer. + :vartype publisher_id: str + :ivar offer_id: Offer Id of the Marketplace offer,. + :vartype offer_id: str + """ + + id: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Marketplace Subscription Id. This is a GUID-formatted string.""" + publisher_id: Optional[str] = rest_field( + name="publisherId", visibility=["read", "create", "update", "delete", "query"] + ) + """Publisher Id of the Marketplace offer.""" + offer_id: Optional[str] = rest_field(name="offerId", visibility=["read", "create", "update", "delete", "query"]) + """Offer Id of the Marketplace offer,.""" + + @overload + def __init__( + self, + *, + id: Optional[str] = None, # pylint: disable=redefined-builtin + publisher_id: Optional[str] = None, + offer_id: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class MonitoredResource(_Model): + """The properties of a resource currently being monitored by the Elastic monitor resource. + + :ivar id: The ARM id of the resource. + :vartype id: str + :ivar sending_logs: Flag indicating the status of the resource for sending logs operation to + Elastic. Known values are: "True" and "False". + :vartype sending_logs: str or ~azure.mgmt.elastic.models.SendingLogs + :ivar reason_for_logs_status: Reason for why the resource is sending logs (or why it is not + sending). + :vartype reason_for_logs_status: str + """ + + id: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The ARM id of the resource.""" + sending_logs: Optional[Union[str, "_models.SendingLogs"]] = rest_field( + name="sendingLogs", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag indicating the status of the resource for sending logs operation to Elastic. Known values + are: \"True\" and \"False\".""" + reason_for_logs_status: Optional[str] = rest_field( + name="reasonForLogsStatus", visibility=["read", "create", "update", "delete", "query"] + ) + """Reason for why the resource is sending logs (or why it is not sending).""" + + @overload + def __init__( + self, + *, + id: Optional[str] = None, # pylint: disable=redefined-builtin + sending_logs: Optional[Union[str, "_models.SendingLogs"]] = None, + reason_for_logs_status: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class MonitoredSubscription(_Model): + """The list of subscriptions and it's monitoring status by current Elastic monitor. + + :ivar subscription_id: The subscriptionId to be monitored. Required. + :vartype subscription_id: str + :ivar status: The state of monitoring. Known values are: "InProgress", "Active", "Failed", and + "Deleting". + :vartype status: str or ~azure.mgmt.elastic.models.Status + :ivar error: The reason of not monitoring the subscription. + :vartype error: str + :ivar tag_rules: Definition of the properties for a TagRules resource. + :vartype tag_rules: ~azure.mgmt.elastic.models.MonitoringTagRulesProperties + """ + + subscription_id: str = rest_field(name="subscriptionId", visibility=["read", "create", "update", "delete", "query"]) + """The subscriptionId to be monitored. Required.""" + status: Optional[Union[str, "_models.Status"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The state of monitoring. Known values are: \"InProgress\", \"Active\", \"Failed\", and + \"Deleting\".""" + error: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The reason of not monitoring the subscription.""" + tag_rules: Optional["_models.MonitoringTagRulesProperties"] = rest_field( + name="tagRules", visibility=["read", "create", "update", "delete", "query"] + ) + """Definition of the properties for a TagRules resource.""" + + @overload + def __init__( + self, + *, + subscription_id: str, + status: Optional[Union[str, "_models.Status"]] = None, + error: Optional[str] = None, + tag_rules: Optional["_models.MonitoringTagRulesProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ProxyResource(Resource): + """Proxy Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.elastic.models.SystemData + """ + + +class MonitoredSubscriptionProperties(ProxyResource): + """The request to update subscriptions needed to be monitored by the Elastic monitor resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.elastic.models.SystemData + :ivar properties: The request to update subscriptions needed to be monitored by the Elastic + monitor resource. + :vartype properties: ~azure.mgmt.elastic.models.SubscriptionList + """ + + properties: Optional["_models.SubscriptionList"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The request to update subscriptions needed to be monitored by the Elastic monitor resource.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.SubscriptionList"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class MonitoringTagRules(ProxyResource): + """Capture logs and metrics of Azure resources based on ARM tags. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.elastic.models.SystemData + :ivar properties: Properties of the monitoring tag rules. + :vartype properties: ~azure.mgmt.elastic.models.MonitoringTagRulesProperties + """ + + properties: Optional["_models.MonitoringTagRulesProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of the monitoring tag rules.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.MonitoringTagRulesProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class MonitoringTagRulesProperties(_Model): + """Definition of the properties for a TagRules resource. + + :ivar provisioning_state: Provisioning state of the monitoring tag rules. Known values are: + "Accepted", "Creating", "Updating", "Deleting", "Succeeded", "Failed", "Canceled", "Deleted", + and "NotSpecified". + :vartype provisioning_state: str or ~azure.mgmt.elastic.models.ProvisioningState + :ivar log_rules: Rules for sending logs. + :vartype log_rules: ~azure.mgmt.elastic.models.LogRules + """ + + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = rest_field( + name="provisioningState", visibility=["read"] + ) + """Provisioning state of the monitoring tag rules. Known values are: \"Accepted\", \"Creating\", + \"Updating\", \"Deleting\", \"Succeeded\", \"Failed\", \"Canceled\", \"Deleted\", and + \"NotSpecified\".""" + log_rules: Optional["_models.LogRules"] = rest_field( + name="logRules", visibility=["read", "create", "update", "delete", "query"] + ) + """Rules for sending logs.""" + + @overload + def __init__( + self, + *, + log_rules: Optional["_models.LogRules"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class MonitorProperties(_Model): + """Properties specific to the monitor resource. + + :ivar provisioning_state: Provisioning state of the monitor resource. Known values are: + "Accepted", "Creating", "Updating", "Deleting", "Succeeded", "Failed", "Canceled", "Deleted", + and "NotSpecified". + :vartype provisioning_state: str or ~azure.mgmt.elastic.models.ProvisioningState + :ivar monitoring_status: Flag specifying if the resource monitoring is enabled or disabled. + Known values are: "Enabled" and "Disabled". + :vartype monitoring_status: str or ~azure.mgmt.elastic.models.MonitoringStatus + :ivar elastic_properties: Elastic cloud properties. + :vartype elastic_properties: ~azure.mgmt.elastic.models.ElasticProperties + :ivar user_info: User information. + :vartype user_info: ~azure.mgmt.elastic.models.UserInfo + :ivar plan_details: Plan details of the monitor resource. + :vartype plan_details: ~azure.mgmt.elastic.models.PlanDetails + :ivar version: Version of elastic of the monitor resource. + :vartype version: str + :ivar subscription_state: State of the Azure Subscription containing the monitor resource. + :vartype subscription_state: str + :ivar saa_s_azure_subscription_status: Status of Azure Subscription where Marketplace SaaS is + located. + :vartype saa_s_azure_subscription_status: str + :ivar source_campaign_name: Name of the marketing campaign. + :vartype source_campaign_name: str + :ivar source_campaign_id: A unique identifier associated with the campaign. + :vartype source_campaign_id: str + :ivar liftr_resource_category: Known values are: "Unknown" and "MonitorLogs". + :vartype liftr_resource_category: str or ~azure.mgmt.elastic.models.LiftrResourceCategories + :ivar liftr_resource_preference: The priority of the resource. + :vartype liftr_resource_preference: int + :ivar generate_api_key: Flag to determine if User API Key has to be generated and shared. + :vartype generate_api_key: bool + :ivar hosting_type: Hosting type of the monitor resource - either Hosted deployments OR + Serverless Projects. Known values are: "Hosted" and "Serverless". + :vartype hosting_type: str or ~azure.mgmt.elastic.models.HostingType + :ivar project_details: Project details of the monitor resource IF it belongs to Serverless + offer kind. + :vartype project_details: ~azure.mgmt.elastic.models.ProjectDetails + """ + + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = rest_field( + name="provisioningState", visibility=["read"] + ) + """Provisioning state of the monitor resource. Known values are: \"Accepted\", \"Creating\", + \"Updating\", \"Deleting\", \"Succeeded\", \"Failed\", \"Canceled\", \"Deleted\", and + \"NotSpecified\".""" + monitoring_status: Optional[Union[str, "_models.MonitoringStatus"]] = rest_field( + name="monitoringStatus", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag specifying if the resource monitoring is enabled or disabled. Known values are: + \"Enabled\" and \"Disabled\".""" + elastic_properties: Optional["_models.ElasticProperties"] = rest_field( + name="elasticProperties", visibility=["read", "create", "update", "delete", "query"] + ) + """Elastic cloud properties.""" + user_info: Optional["_models.UserInfo"] = rest_field(name="userInfo", visibility=["create"]) + """User information.""" + plan_details: Optional["_models.PlanDetails"] = rest_field( + name="planDetails", visibility=["read", "create", "update", "delete", "query"] + ) + """Plan details of the monitor resource.""" + version: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Version of elastic of the monitor resource.""" + subscription_state: Optional[str] = rest_field( + name="subscriptionState", visibility=["read", "create", "update", "delete", "query"] + ) + """State of the Azure Subscription containing the monitor resource.""" + saa_s_azure_subscription_status: Optional[str] = rest_field( + name="saaSAzureSubscriptionStatus", visibility=["read", "create", "update", "delete", "query"] + ) + """Status of Azure Subscription where Marketplace SaaS is located.""" + source_campaign_name: Optional[str] = rest_field( + name="sourceCampaignName", visibility=["read", "create", "update", "delete", "query"] + ) + """Name of the marketing campaign.""" + source_campaign_id: Optional[str] = rest_field( + name="sourceCampaignId", visibility=["read", "create", "update", "delete", "query"] + ) + """A unique identifier associated with the campaign.""" + liftr_resource_category: Optional[Union[str, "_models.LiftrResourceCategories"]] = rest_field( + name="liftrResourceCategory", visibility=["read"] + ) + """Known values are: \"Unknown\" and \"MonitorLogs\".""" + liftr_resource_preference: Optional[int] = rest_field(name="liftrResourcePreference", visibility=["read"]) + """The priority of the resource.""" + generate_api_key: Optional[bool] = rest_field( + name="generateApiKey", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag to determine if User API Key has to be generated and shared.""" + hosting_type: Optional[Union[str, "_models.HostingType"]] = rest_field( + name="hostingType", visibility=["read", "create", "update", "delete", "query"] + ) + """Hosting type of the monitor resource - either Hosted deployments OR Serverless Projects. Known + values are: \"Hosted\" and \"Serverless\".""" + project_details: Optional["_models.ProjectDetails"] = rest_field( + name="projectDetails", visibility=["read", "create", "update", "delete", "query"] + ) + """Project details of the monitor resource IF it belongs to Serverless offer kind.""" + + @overload + def __init__( + self, + *, + monitoring_status: Optional[Union[str, "_models.MonitoringStatus"]] = None, + elastic_properties: Optional["_models.ElasticProperties"] = None, + user_info: Optional["_models.UserInfo"] = None, + plan_details: Optional["_models.PlanDetails"] = None, + version: Optional[str] = None, + subscription_state: Optional[str] = None, + saa_s_azure_subscription_status: Optional[str] = None, + source_campaign_name: Optional[str] = None, + source_campaign_id: Optional[str] = None, + generate_api_key: Optional[bool] = None, + hosting_type: Optional[Union[str, "_models.HostingType"]] = None, + project_details: Optional["_models.ProjectDetails"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OkResponse(_Model): + """The request has succeeded.""" + + +class OpenAIIntegrationProperties(_Model): + """Open AI Integration details. + + :ivar open_ai_resource_id: The resource name of Open AI resource. + :vartype open_ai_resource_id: str + :ivar open_ai_resource_endpoint: The API endpoint for Open AI resource. + :vartype open_ai_resource_endpoint: str + :ivar open_ai_connector_id: The connector id of Open AI resource. + :vartype open_ai_connector_id: str + :ivar key: Value of API key for Open AI resource. + :vartype key: str + :ivar last_refresh_at: Last Update Timestamp for key updation. + :vartype last_refresh_at: ~datetime.datetime + """ + + open_ai_resource_id: Optional[str] = rest_field( + name="openAIResourceId", visibility=["read", "create", "update", "delete", "query"] + ) + """The resource name of Open AI resource.""" + open_ai_resource_endpoint: Optional[str] = rest_field( + name="openAIResourceEndpoint", visibility=["read", "create", "update", "delete", "query"] + ) + """The API endpoint for Open AI resource.""" + open_ai_connector_id: Optional[str] = rest_field( + name="openAIConnectorId", visibility=["read", "create", "update", "delete", "query"] + ) + """The connector id of Open AI resource.""" + key: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Value of API key for Open AI resource.""" + last_refresh_at: Optional[datetime.datetime] = rest_field( + name="lastRefreshAt", visibility=["read"], format="rfc3339" + ) + """Last Update Timestamp for key updation.""" + + @overload + def __init__( + self, + *, + open_ai_resource_id: Optional[str] = None, + open_ai_resource_endpoint: Optional[str] = None, + open_ai_connector_id: Optional[str] = None, + key: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OpenAIIntegrationRPModel(ProxyResource): + """Capture properties of Open AI resource Integration. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.elastic.models.SystemData + :ivar properties: Open AI Integration details. + :vartype properties: ~azure.mgmt.elastic.models.OpenAIIntegrationProperties + """ + + properties: Optional["_models.OpenAIIntegrationProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Open AI Integration details.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.OpenAIIntegrationProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OpenAIIntegrationStatusResponse(_Model): + """Status of the OpenAI Integration. + + :ivar properties: Status of the OpenAI Integration. + :vartype properties: ~azure.mgmt.elastic.models.OpenAIIntegrationStatusResponseProperties + """ + + properties: Optional["_models.OpenAIIntegrationStatusResponseProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Status of the OpenAI Integration.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.OpenAIIntegrationStatusResponseProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OpenAIIntegrationStatusResponseProperties(_Model): # pylint: disable=name-too-long + """Status of the OpenAI Integration. + + :ivar status: Status of the OpenAI Integration. + :vartype status: str + """ + + status: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Status of the OpenAI Integration.""" + + @overload + def __init__( + self, + *, + status: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OperationDisplay(_Model): + """Represents the display information for an operation. + + :ivar provider: The service provider of the operation. + :vartype provider: str + :ivar resource: The resource type of the operation. + :vartype resource: str + :ivar operation: The name of the operation. + :vartype operation: str + :ivar description: A description of the operation. + :vartype description: str + """ + + provider: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The service provider of the operation.""" + resource: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The resource type of the operation.""" + operation: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The name of the operation.""" + description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """A description of the operation.""" + + @overload + def __init__( + self, + *, + provider: Optional[str] = None, + resource: Optional[str] = None, + operation: Optional[str] = None, + description: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OperationResult(_Model): + """A Microsoft.Elastic REST API operation. + + :ivar name: Operation name, i.e., {provider}/{resource}/{operation}. + :vartype name: str + :ivar is_data_action: Indicates whether the operation is a data action. + :vartype is_data_action: bool + :ivar display: The object that represents the operation. + :vartype display: ~azure.mgmt.elastic.models.OperationDisplay + :ivar origin: Origin of the operation. + :vartype origin: str + """ + + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Operation name, i.e., {provider}/{resource}/{operation}.""" + is_data_action: Optional[bool] = rest_field( + name="isDataAction", visibility=["read", "create", "update", "delete", "query"] + ) + """Indicates whether the operation is a data action.""" + display: Optional["_models.OperationDisplay"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The object that represents the operation.""" + origin: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Origin of the operation.""" + + @overload + def __init__( + self, + *, + name: Optional[str] = None, + is_data_action: Optional[bool] = None, + display: Optional["_models.OperationDisplay"] = None, + origin: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class PartnerBillingEntity(_Model): + """Partner Billing details associated with the resource. + + :ivar id: The Elastic Organization Id. + :vartype id: str + :ivar name: The Elastic Organization Name. + :vartype name: str + :ivar partner_entity_uri: Link to the elastic organization page. + :vartype partner_entity_uri: str + """ + + id: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The Elastic Organization Id.""" + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The Elastic Organization Name.""" + partner_entity_uri: Optional[str] = rest_field( + name="partnerEntityUri", visibility=["read", "create", "update", "delete", "query"] + ) + """Link to the elastic organization page.""" + + @overload + def __init__( + self, + *, + id: Optional[str] = None, # pylint: disable=redefined-builtin + name: Optional[str] = None, + partner_entity_uri: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class PlanDetails(_Model): + """Plan details of the monitor resource. + + :ivar offer_id: Offer ID of the plan. + :vartype offer_id: str + :ivar publisher_id: Publisher ID of the plan. + :vartype publisher_id: str + :ivar term_id: Term ID of the plan. + :vartype term_id: str + :ivar plan_id: Plan ID. + :vartype plan_id: str + :ivar plan_name: Plan Name. + :vartype plan_name: str + """ + + offer_id: Optional[str] = rest_field(name="offerID", visibility=["read", "create", "update", "delete", "query"]) + """Offer ID of the plan.""" + publisher_id: Optional[str] = rest_field( + name="publisherID", visibility=["read", "create", "update", "delete", "query"] + ) + """Publisher ID of the plan.""" + term_id: Optional[str] = rest_field(name="termID", visibility=["read", "create", "update", "delete", "query"]) + """Term ID of the plan.""" + plan_id: Optional[str] = rest_field(name="planID", visibility=["read", "create", "update", "delete", "query"]) + """Plan ID.""" + plan_name: Optional[str] = rest_field(name="planName", visibility=["read", "create", "update", "delete", "query"]) + """Plan Name.""" + + @overload + def __init__( + self, + *, + offer_id: Optional[str] = None, + publisher_id: Optional[str] = None, + term_id: Optional[str] = None, + plan_id: Optional[str] = None, + plan_name: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ProjectDetails(_Model): + """Project details of the monitor resource IF it belongs to Serverless offer kind. + + :ivar project_type: Project type; ex: Elasticsearch / Observability / Security. Known values + are: "Elasticsearch", "Observability", "Security", and "NotApplicable". + :vartype project_type: str or ~azure.mgmt.elastic.models.ProjectType + :ivar configuration_type: Configuration type of the Elasticsearch project. Known values are: + "GeneralPurpose", "Vector", "TimeSeries", and "NotApplicable". + :vartype configuration_type: str or ~azure.mgmt.elastic.models.ConfigurationType + """ + + project_type: Optional[Union[str, "_models.ProjectType"]] = rest_field( + name="projectType", visibility=["read", "create", "update", "delete", "query"] + ) + """Project type; ex: Elasticsearch / Observability / Security. Known values are: + \"Elasticsearch\", \"Observability\", \"Security\", and \"NotApplicable\".""" + configuration_type: Optional[Union[str, "_models.ConfigurationType"]] = rest_field( + name="configurationType", visibility=["read", "create", "update", "delete", "query"] + ) + """Configuration type of the Elasticsearch project. Known values are: \"GeneralPurpose\", + \"Vector\", \"TimeSeries\", and \"NotApplicable\".""" + + @overload + def __init__( + self, + *, + project_type: Optional[Union[str, "_models.ProjectType"]] = None, + configuration_type: Optional[Union[str, "_models.ConfigurationType"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ResourceProviderDefaultErrorResponse(_Model): + """RP default error response. + + :ivar error: Response body of Error. + :vartype error: ~azure.mgmt.elastic.models.ErrorResponseBody + """ + + error: Optional["_models.ErrorResponseBody"] = rest_field(visibility=["read"]) + """Response body of Error.""" + + +class ResourceSku(_Model): + """Represents the SKU of a resource. + + :ivar name: The name of the SKU. Required. + :vartype name: str + """ + + name: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The name of the SKU. Required.""" + + @overload + def __init__( + self, + *, + name: str, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ResubscribeParameterBody(_Model): + """ResubscribeParameterBody. + + :ivar body: Resubscribe Properties. + :vartype body: ~azure.mgmt.elastic.models.ResubscribeProperties + """ + + body: Optional["_models.ResubscribeProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Resubscribe Properties.""" + + @overload + def __init__( + self, + *, + body: Optional["_models.ResubscribeProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ResubscribeProperties(_Model): + """Resubscribe Properties. + + :ivar plan_id: Newly selected plan Id to create the new Marketplace subscription for + Resubscribe. + :vartype plan_id: str + :ivar term: Newly selected term to create the new Marketplace subscription for Resubscribe. + :vartype term: str + :ivar subscription_id: Newly selected Azure Subscription Id in which the new Marketplace + subscription will be created for Resubscribe. + :vartype subscription_id: str + :ivar resource_group: Newly selected Azure resource group in which the new Marketplace + subscription will be created for Resubscribe. + :vartype resource_group: str + :ivar organization_id: Organization Id of the Elastic Organization that needs to be + resubscribed. + :vartype organization_id: str + """ + + plan_id: Optional[str] = rest_field(name="planId", visibility=["read", "create", "update", "delete", "query"]) + """Newly selected plan Id to create the new Marketplace subscription for Resubscribe.""" + term: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Newly selected term to create the new Marketplace subscription for Resubscribe.""" + subscription_id: Optional[str] = rest_field( + name="subscriptionId", visibility=["read", "create", "update", "delete", "query"] + ) + """Newly selected Azure Subscription Id in which the new Marketplace subscription will be created + for Resubscribe.""" + resource_group: Optional[str] = rest_field( + name="resourceGroup", visibility=["read", "create", "update", "delete", "query"] + ) + """Newly selected Azure resource group in which the new Marketplace subscription will be created + for Resubscribe.""" + organization_id: Optional[str] = rest_field( + name="organizationId", visibility=["read", "create", "update", "delete", "query"] + ) + """Organization Id of the Elastic Organization that needs to be resubscribed.""" + + @overload + def __init__( + self, + *, + plan_id: Optional[str] = None, + term: Optional[str] = None, + subscription_id: Optional[str] = None, + resource_group: Optional[str] = None, + organization_id: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class SubscriptionList(_Model): + """The request to update subscriptions needed to be monitored by the Elastic monitor resource. + + :ivar operation: The operation for the patch on the resource. Known values are: "AddBegin", + "AddComplete", "DeleteBegin", "DeleteComplete", and "Active". + :vartype operation: str or ~azure.mgmt.elastic.models.Operation + :ivar monitored_subscription_list: List of subscriptions and the state of the monitoring. + :vartype monitored_subscription_list: list[~azure.mgmt.elastic.models.MonitoredSubscription] + :ivar provisioning_state: Provisioning State of the resource. Known values are: "Accepted", + "Creating", "Updating", "Deleting", "Succeeded", "Failed", "Canceled", "Deleted", and + "NotSpecified". + :vartype provisioning_state: str or ~azure.mgmt.elastic.models.ProvisioningState + """ + + operation: Optional[Union[str, "_models.Operation"]] = rest_field(visibility=["create", "update"]) + """The operation for the patch on the resource. Known values are: \"AddBegin\", \"AddComplete\", + \"DeleteBegin\", \"DeleteComplete\", and \"Active\".""" + monitored_subscription_list: Optional[List["_models.MonitoredSubscription"]] = rest_field( + name="monitoredSubscriptionList", visibility=["read", "create", "update", "delete", "query"] + ) + """List of subscriptions and the state of the monitoring.""" + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = rest_field( + name="provisioningState", visibility=["read"] + ) + """Provisioning State of the resource. Known values are: \"Accepted\", \"Creating\", \"Updating\", + \"Deleting\", \"Succeeded\", \"Failed\", \"Canceled\", \"Deleted\", and \"NotSpecified\".""" + + @overload + def __init__( + self, + *, + operation: Optional[Union[str, "_models.Operation"]] = None, + monitored_subscription_list: Optional[List["_models.MonitoredSubscription"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class SystemData(_Model): + """Metadata pertaining to creation and last modification of the resource. + + :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. Known values are: + "User", "Application", "ManagedIdentity", and "Key". + :vartype created_by_type: str or ~azure.mgmt.elastic.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. Known values + are: "User", "Application", "ManagedIdentity", and "Key". + :vartype last_modified_by_type: str or ~azure.mgmt.elastic.models.CreatedByType + :ivar last_modified_at: The timestamp of resource last modification (UTC). + :vartype last_modified_at: ~datetime.datetime + """ + + created_by: Optional[str] = rest_field(name="createdBy", visibility=["read", "create", "update", "delete", "query"]) + """The identity that created the resource.""" + created_by_type: Optional[Union[str, "_models.CreatedByType"]] = rest_field( + name="createdByType", visibility=["read", "create", "update", "delete", "query"] + ) + """The type of identity that created the resource. Known values are: \"User\", \"Application\", + \"ManagedIdentity\", and \"Key\".""" + created_at: Optional[datetime.datetime] = rest_field( + name="createdAt", visibility=["read", "create", "update", "delete", "query"], format="rfc3339" + ) + """The timestamp of resource creation (UTC).""" + last_modified_by: Optional[str] = rest_field( + name="lastModifiedBy", visibility=["read", "create", "update", "delete", "query"] + ) + """The identity that last modified the resource.""" + last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = rest_field( + name="lastModifiedByType", visibility=["read", "create", "update", "delete", "query"] + ) + """The type of identity that last modified the resource. Known values are: \"User\", + \"Application\", \"ManagedIdentity\", and \"Key\".""" + last_modified_at: Optional[datetime.datetime] = rest_field( + name="lastModifiedAt", visibility=["read", "create", "update", "delete", "query"], format="rfc3339" + ) + """The timestamp of resource last modification (UTC).""" + + @overload + def __init__( + self, + *, + created_by: Optional[str] = None, + created_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, + created_at: Optional[datetime.datetime] = None, + last_modified_by: Optional[str] = None, + last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, + last_modified_at: Optional[datetime.datetime] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class UpgradableVersionsList(_Model): + """Stack Versions that this version can upgrade to. + + :ivar current_version: Current version of the elastic monitor. + :vartype current_version: str + :ivar upgradable_versions: Stack Versions that this version can upgrade to. + :vartype upgradable_versions: list[str] + """ + + current_version: Optional[str] = rest_field( + name="currentVersion", visibility=["read", "create", "update", "delete", "query"] + ) + """Current version of the elastic monitor.""" + upgradable_versions: Optional[List[str]] = rest_field( + name="upgradableVersions", visibility=["read", "create", "update", "delete", "query"] + ) + """Stack Versions that this version can upgrade to.""" + + @overload + def __init__( + self, + *, + current_version: Optional[str] = None, + upgradable_versions: Optional[List[str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class UpgradeParameterBody(_Model): + """UpgradeParameterBody. + + :ivar body: Elastic Monitor Upgrade Parameters. + :vartype body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade + """ + + body: Optional["_models.ElasticMonitorUpgrade"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Elastic Monitor Upgrade Parameters.""" + + @overload + def __init__( + self, + *, + body: Optional["_models.ElasticMonitorUpgrade"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class UserApiKeyResponse(_Model): + """The User Api Key created for the Organization associated with the User Email Id that was passed + in the request. + + :ivar properties: + :vartype properties: ~azure.mgmt.elastic.models.UserApiKeyResponseProperties + """ + + properties: Optional["_models.UserApiKeyResponseProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + + @overload + def __init__( + self, + *, + properties: Optional["_models.UserApiKeyResponseProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class UserApiKeyResponseProperties(_Model): + """UserApiKeyResponseProperties. + + :ivar api_key: The User Api Key Generated based on GenerateApiKey flag. This is applicable for + non-Portal clients only. + :vartype api_key: str + """ + + api_key: Optional[str] = rest_field(name="apiKey", visibility=["read", "create", "update", "delete", "query"]) + """The User Api Key Generated based on GenerateApiKey flag. This is applicable for non-Portal + clients only.""" + + @overload + def __init__( + self, + *, + api_key: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class UserEmailId(_Model): + """Email Id of the User Organization, of which the API Key must be returned. + + :ivar email_id: The User email Id. + :vartype email_id: str + """ + + email_id: Optional[str] = rest_field(name="emailId", visibility=["read", "create", "update", "delete", "query"]) + """The User email Id.""" + + @overload + def __init__( + self, + *, + email_id: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class UserInfo(_Model): + """User Information to be passed to partners. + + :ivar first_name: First name of the user. + :vartype first_name: str + :ivar last_name: Last name of the user. + :vartype last_name: str + :ivar company_name: Company name of the user. + :vartype company_name: str + :ivar email_address: Email of the user used by Elastic for contacting them if needed. + :vartype email_address: str + :ivar company_info: Company information of the user to be passed to partners. + :vartype company_info: ~azure.mgmt.elastic.models.CompanyInfo + """ + + first_name: Optional[str] = rest_field(name="firstName", visibility=["read", "create", "update", "delete", "query"]) + """First name of the user.""" + last_name: Optional[str] = rest_field(name="lastName", visibility=["read", "create", "update", "delete", "query"]) + """Last name of the user.""" + company_name: Optional[str] = rest_field( + name="companyName", visibility=["read", "create", "update", "delete", "query"] + ) + """Company name of the user.""" + email_address: Optional[str] = rest_field( + name="emailAddress", visibility=["read", "create", "update", "delete", "query"] + ) + """Email of the user used by Elastic for contacting them if needed.""" + company_info: Optional["_models.CompanyInfo"] = rest_field( + name="companyInfo", visibility=["read", "create", "update", "delete", "query"] + ) + """Company information of the user to be passed to partners.""" + + @overload + def __init__( + self, + *, + first_name: Optional[str] = None, + last_name: Optional[str] = None, + company_name: Optional[str] = None, + email_address: Optional[str] = None, + company_info: Optional["_models.CompanyInfo"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class VMCollectionUpdate(_Model): + """Update VM resource collection. + + :ivar vm_resource_id: ARM id of the VM resource. + :vartype vm_resource_id: str + :ivar operation_name: Operation to be performed for given VM. Known values are: "Add" and + "Delete". + :vartype operation_name: str or ~azure.mgmt.elastic.models.OperationName + """ + + vm_resource_id: Optional[str] = rest_field( + name="vmResourceId", visibility=["read", "create", "update", "delete", "query"] + ) + """ARM id of the VM resource.""" + operation_name: Optional[Union[str, "_models.OperationName"]] = rest_field( + name="operationName", visibility=["read", "create", "update", "delete", "query"] + ) + """Operation to be performed for given VM. Known values are: \"Add\" and \"Delete\".""" + + @overload + def __init__( + self, + *, + vm_resource_id: Optional[str] = None, + operation_name: Optional[Union[str, "_models.OperationName"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class VMCollectionUpdateParameterBody(_Model): + """VMCollectionUpdateParameterBody. + + :ivar body: VM resource Id. + :vartype body: ~azure.mgmt.elastic.models.VMCollectionUpdate + """ + + body: Optional["_models.VMCollectionUpdate"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """VM resource Id.""" + + @overload + def __init__( + self, + *, + body: Optional["_models.VMCollectionUpdate"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class VMIngestionDetailsResponse(_Model): + """The vm ingestion details to install an agent. + + :ivar cloud_id: The cloudId of given Elastic monitor resource. + :vartype cloud_id: str + :ivar ingestion_key: Ingestion details to install agent on given VM. + :vartype ingestion_key: str + """ + + cloud_id: Optional[str] = rest_field(name="cloudId", visibility=["read", "create", "update", "delete", "query"]) + """The cloudId of given Elastic monitor resource.""" + ingestion_key: Optional[str] = rest_field( + name="ingestionKey", visibility=["read", "create", "update", "delete", "query"] + ) + """Ingestion details to install agent on given VM.""" + + @overload + def __init__( + self, + *, + cloud_id: Optional[str] = None, + ingestion_key: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class VMResources(_Model): + """The vm resource properties that is currently being monitored by the Elastic monitor resource. + + :ivar vm_resource_id: The ARM id of the VM resource. + :vartype vm_resource_id: str + """ + + vm_resource_id: Optional[str] = rest_field( + name="vmResourceId", visibility=["read", "create", "update", "delete", "query"] + ) + """The ARM id of the VM resource.""" + + @overload + def __init__( + self, + *, + vm_resource_id: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models_py3.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models_py3.py deleted file mode 100644 index d47d3629939b..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_models_py3.py +++ /dev/null @@ -1,2568 +0,0 @@ -# coding=utf-8 -# pylint: disable=too-many-lines -# -------------------------------------------------------------------------- -# 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 datetime -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union - -from .. import _serialization - -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from .. import models as _models - - -class BillingInfoResponse(_serialization.Model): - """Marketplace Subscription and Organization details to which resource gets billed into. - - :ivar marketplace_saas_info: Marketplace Subscription details. - :vartype marketplace_saas_info: ~azure.mgmt.elastic.models.MarketplaceSaaSInfo - :ivar partner_billing_entity: Partner Billing Entity details: Organization Info. - :vartype partner_billing_entity: ~azure.mgmt.elastic.models.PartnerBillingEntity - """ - - _attribute_map = { - "marketplace_saas_info": {"key": "marketplaceSaasInfo", "type": "MarketplaceSaaSInfo"}, - "partner_billing_entity": {"key": "partnerBillingEntity", "type": "PartnerBillingEntity"}, - } - - def __init__( - self, - *, - marketplace_saas_info: Optional["_models.MarketplaceSaaSInfo"] = None, - partner_billing_entity: Optional["_models.PartnerBillingEntity"] = None, - **kwargs: Any - ) -> None: - """ - :keyword marketplace_saas_info: Marketplace Subscription details. - :paramtype marketplace_saas_info: ~azure.mgmt.elastic.models.MarketplaceSaaSInfo - :keyword partner_billing_entity: Partner Billing Entity details: Organization Info. - :paramtype partner_billing_entity: ~azure.mgmt.elastic.models.PartnerBillingEntity - """ - super().__init__(**kwargs) - self.marketplace_saas_info = marketplace_saas_info - self.partner_billing_entity = partner_billing_entity - - -class CompanyInfo(_serialization.Model): - """Company information of the user to be passed to partners. - - :ivar domain: Domain of the company. - :vartype domain: str - :ivar business: Business of the company. - :vartype business: str - :ivar employees_number: Number of employees in the company. - :vartype employees_number: str - :ivar state: State of the company location. - :vartype state: str - :ivar country: Country of the company location. - :vartype country: str - """ - - _validation = { - "domain": {"max_length": 250}, - "business": {"max_length": 64}, - "employees_number": {"max_length": 20}, - "state": {"max_length": 64}, - "country": {"max_length": 64}, - } - - _attribute_map = { - "domain": {"key": "domain", "type": "str"}, - "business": {"key": "business", "type": "str"}, - "employees_number": {"key": "employeesNumber", "type": "str"}, - "state": {"key": "state", "type": "str"}, - "country": {"key": "country", "type": "str"}, - } - - def __init__( - self, - *, - domain: Optional[str] = None, - business: Optional[str] = None, - employees_number: Optional[str] = None, - state: Optional[str] = None, - country: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword domain: Domain of the company. - :paramtype domain: str - :keyword business: Business of the company. - :paramtype business: str - :keyword employees_number: Number of employees in the company. - :paramtype employees_number: str - :keyword state: State of the company location. - :paramtype state: str - :keyword country: Country of the company location. - :paramtype country: str - """ - super().__init__(**kwargs) - self.domain = domain - self.business = business - self.employees_number = employees_number - self.state = state - self.country = country - - -class ConnectedPartnerResourceProperties(_serialization.Model): - """Connected Partner Resource Properties. - - :ivar partner_deployment_name: Elastic deployment name. - :vartype partner_deployment_name: str - :ivar partner_deployment_uri: Deployment URL of the elasticsearch in Elastic cloud deployment. - :vartype partner_deployment_uri: str - :ivar azure_resource_id: The azure resource Id of the deployment. - :vartype azure_resource_id: str - :ivar location: The location of the deployment. - :vartype location: str - """ - - _attribute_map = { - "partner_deployment_name": {"key": "partnerDeploymentName", "type": "str"}, - "partner_deployment_uri": {"key": "partnerDeploymentUri", "type": "str"}, - "azure_resource_id": {"key": "azureResourceId", "type": "str"}, - "location": {"key": "location", "type": "str"}, - } - - def __init__( - self, - *, - partner_deployment_name: Optional[str] = None, - partner_deployment_uri: Optional[str] = None, - azure_resource_id: Optional[str] = None, - location: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword partner_deployment_name: Elastic deployment name. - :paramtype partner_deployment_name: str - :keyword partner_deployment_uri: Deployment URL of the elasticsearch in Elastic cloud - deployment. - :paramtype partner_deployment_uri: str - :keyword azure_resource_id: The azure resource Id of the deployment. - :paramtype azure_resource_id: str - :keyword location: The location of the deployment. - :paramtype location: str - """ - super().__init__(**kwargs) - self.partner_deployment_name = partner_deployment_name - self.partner_deployment_uri = partner_deployment_uri - self.azure_resource_id = azure_resource_id - self.location = location - - -class ConnectedPartnerResourcesListFormat(_serialization.Model): - """Connected Partner Resources List Format. - - :ivar properties: Connected Partner Resource Properties. - :vartype properties: ~azure.mgmt.elastic.models.ConnectedPartnerResourceProperties - """ - - _attribute_map = { - "properties": {"key": "properties", "type": "ConnectedPartnerResourceProperties"}, - } - - def __init__( - self, *, properties: Optional["_models.ConnectedPartnerResourceProperties"] = None, **kwargs: Any - ) -> None: - """ - :keyword properties: Connected Partner Resource Properties. - :paramtype properties: ~azure.mgmt.elastic.models.ConnectedPartnerResourceProperties - """ - super().__init__(**kwargs) - self.properties = properties - - -class ConnectedPartnerResourcesListResponse(_serialization.Model): - """List of all active elastic deployments. - - :ivar value: Results of a list operation. - :vartype value: list[~azure.mgmt.elastic.models.ConnectedPartnerResourcesListFormat] - :ivar next_link: Link to the next set of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[ConnectedPartnerResourcesListFormat]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.ConnectedPartnerResourcesListFormat"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Results of a list operation. - :paramtype value: list[~azure.mgmt.elastic.models.ConnectedPartnerResourcesListFormat] - :keyword next_link: Link to the next set of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class DeploymentInfoResponse(_serialization.Model): - """The properties of deployment in Elastic cloud corresponding to the Elastic monitor resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar status: The Elastic deployment status. Known values are: "Healthy" and "Unhealthy". - :vartype status: str or ~azure.mgmt.elastic.models.ElasticDeploymentStatus - :ivar version: Version of the elasticsearch in Elastic cloud deployment. - :vartype version: str - :ivar memory_capacity: RAM capacity of the elasticsearch in Elastic cloud deployment. - :vartype memory_capacity: str - :ivar disk_capacity: Disk capacity of the elasticsearch in Elastic cloud deployment. - :vartype disk_capacity: str - :ivar elasticsearch_end_point: Elasticsearch endpoint in Elastic cloud deployment. This is - either the aliased_endpoint if available, or the service_url otherwise. - :vartype elasticsearch_end_point: str - :ivar deployment_url: Deployment URL of the elasticsearch in Elastic cloud deployment. - :vartype deployment_url: str - :ivar marketplace_saas_info: Marketplace SaaS Info of the resource. - :vartype marketplace_saas_info: ~azure.mgmt.elastic.models.MarketplaceSaaSInfo - """ - - _validation = { - "status": {"readonly": True}, - "version": {"readonly": True}, - "memory_capacity": {"readonly": True}, - "disk_capacity": {"readonly": True}, - "elasticsearch_end_point": {"readonly": True}, - "deployment_url": {"readonly": True}, - "marketplace_saas_info": {"readonly": True}, - } - - _attribute_map = { - "status": {"key": "status", "type": "str"}, - "version": {"key": "version", "type": "str"}, - "memory_capacity": {"key": "memoryCapacity", "type": "str"}, - "disk_capacity": {"key": "diskCapacity", "type": "str"}, - "elasticsearch_end_point": {"key": "elasticsearchEndPoint", "type": "str"}, - "deployment_url": {"key": "deploymentUrl", "type": "str"}, - "marketplace_saas_info": {"key": "marketplaceSaasInfo", "type": "MarketplaceSaaSInfo"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.status = None - self.version = None - self.memory_capacity = None - self.disk_capacity = None - self.elasticsearch_end_point = None - self.deployment_url = None - self.marketplace_saas_info = None - - -class ElasticCloudDeployment(_serialization.Model): - """Details of the user's elastic deployment associated with the monitor resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: Elastic deployment name. - :vartype name: str - :ivar deployment_id: Elastic deployment Id. - :vartype deployment_id: str - :ivar azure_subscription_id: Associated Azure subscription Id for the elastic deployment. - :vartype azure_subscription_id: str - :ivar elasticsearch_region: Region where Deployment at Elastic side took place. - :vartype elasticsearch_region: str - :ivar elasticsearch_service_url: Elasticsearch ingestion endpoint of the Elastic deployment. - :vartype elasticsearch_service_url: str - :ivar kibana_service_url: Kibana endpoint of the Elastic deployment. - :vartype kibana_service_url: str - :ivar kibana_sso_url: Kibana dashboard sso URL of the Elastic deployment. - :vartype kibana_sso_url: str - """ - - _validation = { - "name": {"readonly": True}, - "deployment_id": {"readonly": True}, - "azure_subscription_id": {"readonly": True}, - "elasticsearch_region": {"readonly": True}, - "elasticsearch_service_url": {"readonly": True}, - "kibana_service_url": {"readonly": True}, - "kibana_sso_url": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "deployment_id": {"key": "deploymentId", "type": "str"}, - "azure_subscription_id": {"key": "azureSubscriptionId", "type": "str"}, - "elasticsearch_region": {"key": "elasticsearchRegion", "type": "str"}, - "elasticsearch_service_url": {"key": "elasticsearchServiceUrl", "type": "str"}, - "kibana_service_url": {"key": "kibanaServiceUrl", "type": "str"}, - "kibana_sso_url": {"key": "kibanaSsoUrl", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.name = None - self.deployment_id = None - self.azure_subscription_id = None - self.elasticsearch_region = None - self.elasticsearch_service_url = None - self.kibana_service_url = None - self.kibana_sso_url = None - - -class ElasticCloudUser(_serialization.Model): - """Details of the user's elastic account. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar email_address: Email of the Elastic User Account. - :vartype email_address: str - :ivar id: User Id of the elastic account of the User. - :vartype id: str - :ivar elastic_cloud_sso_default_url: Elastic cloud default dashboard sso URL of the Elastic - user account. - :vartype elastic_cloud_sso_default_url: str - """ - - _validation = { - "email_address": {"readonly": True}, - "id": {"readonly": True}, - "elastic_cloud_sso_default_url": {"readonly": True}, - } - - _attribute_map = { - "email_address": {"key": "emailAddress", "type": "str"}, - "id": {"key": "id", "type": "str"}, - "elastic_cloud_sso_default_url": {"key": "elasticCloudSsoDefaultUrl", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.email_address = None - self.id = None - self.elastic_cloud_sso_default_url = None - - -class ElasticMonitorResource(_serialization.Model): - """Monitor 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 server. - - :ivar id: ARM id of the monitor resource. - :vartype id: str - :ivar name: Name of the monitor resource. - :vartype name: str - :ivar type: The type of the monitor resource. - :vartype type: str - :ivar sku: SKU of the monitor resource. - :vartype sku: ~azure.mgmt.elastic.models.ResourceSku - :ivar properties: Properties of the monitor resource. - :vartype properties: ~azure.mgmt.elastic.models.MonitorProperties - :ivar identity: Identity properties of the monitor resource. - :vartype identity: ~azure.mgmt.elastic.models.IdentityProperties - :ivar tags: The tags of the monitor resource. - :vartype tags: dict[str, str] - :ivar location: The location of the monitor resource. Required. - :vartype location: str - :ivar system_data: The system metadata relating to this resource. - :vartype system_data: ~azure.mgmt.elastic.models.SystemData - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "location": {"required": True}, - "system_data": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "sku": {"key": "sku", "type": "ResourceSku"}, - "properties": {"key": "properties", "type": "MonitorProperties"}, - "identity": {"key": "identity", "type": "IdentityProperties"}, - "tags": {"key": "tags", "type": "{str}"}, - "location": {"key": "location", "type": "str"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - } - - def __init__( - self, - *, - location: str, - sku: Optional["_models.ResourceSku"] = None, - properties: Optional["_models.MonitorProperties"] = None, - identity: Optional["_models.IdentityProperties"] = None, - tags: Optional[Dict[str, str]] = None, - **kwargs: Any - ) -> None: - """ - :keyword sku: SKU of the monitor resource. - :paramtype sku: ~azure.mgmt.elastic.models.ResourceSku - :keyword properties: Properties of the monitor resource. - :paramtype properties: ~azure.mgmt.elastic.models.MonitorProperties - :keyword identity: Identity properties of the monitor resource. - :paramtype identity: ~azure.mgmt.elastic.models.IdentityProperties - :keyword tags: The tags of the monitor resource. - :paramtype tags: dict[str, str] - :keyword location: The location of the monitor resource. Required. - :paramtype location: str - """ - super().__init__(**kwargs) - self.id = None - self.name = None - self.type = None - self.sku = sku - self.properties = properties - self.identity = identity - self.tags = tags - self.location = location - self.system_data = None - - -class ElasticMonitorResourceListResponse(_serialization.Model): - """Response of a list operation. - - :ivar value: Results of a list operation. - :vartype value: list[~azure.mgmt.elastic.models.ElasticMonitorResource] - :ivar next_link: Link to the next set of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[ElasticMonitorResource]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.ElasticMonitorResource"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Results of a list operation. - :paramtype value: list[~azure.mgmt.elastic.models.ElasticMonitorResource] - :keyword next_link: Link to the next set of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class ElasticMonitorResourceUpdateParameters(_serialization.Model): - """Monitor resource update parameters. - - :ivar tags: elastic monitor resource tags. - :vartype tags: dict[str, str] - """ - - _attribute_map = { - "tags": {"key": "tags", "type": "{str}"}, - } - - def __init__(self, *, tags: Optional[Dict[str, str]] = None, **kwargs: Any) -> None: - """ - :keyword tags: elastic monitor resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(**kwargs) - self.tags = tags - - -class ElasticMonitorUpgrade(_serialization.Model): - """Upgrade elastic monitor version. - - :ivar version: Version to which the elastic monitor should be upgraded to. - :vartype version: str - """ - - _attribute_map = { - "version": {"key": "version", "type": "str"}, - } - - def __init__(self, *, version: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword version: Version to which the elastic monitor should be upgraded to. - :paramtype version: str - """ - super().__init__(**kwargs) - self.version = version - - -class ElasticOrganizationToAzureSubscriptionMappingResponse(_serialization.Model): # pylint: disable=name-too-long - """The Azure Subscription ID to which the Organization of the logged in user belongs and gets - billed into. - - :ivar properties: The properties of Azure Subscription ID to which the Organization of the - logged in user belongs and gets billed into. - :vartype properties: - ~azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponseProperties - """ - - _attribute_map = { - "properties": {"key": "properties", "type": "ElasticOrganizationToAzureSubscriptionMappingResponseProperties"}, - } - - def __init__( - self, - *, - properties: Optional["_models.ElasticOrganizationToAzureSubscriptionMappingResponseProperties"] = None, - **kwargs: Any - ) -> None: - """ - :keyword properties: The properties of Azure Subscription ID to which the Organization of the - logged in user belongs and gets billed into. - :paramtype properties: - ~azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponseProperties - """ - super().__init__(**kwargs) - self.properties = properties - - -class ElasticOrganizationToAzureSubscriptionMappingResponseProperties( - _serialization.Model -): # pylint: disable=name-too-long - """The properties of Azure Subscription ID to which the Organization of the logged in user belongs - and gets billed into. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar billed_azure_subscription_id: The Azure Subscription ID to which the Organization belongs - and gets billed into. This is empty for a new user OR a user without an Elastic Organization. - :vartype billed_azure_subscription_id: str - :ivar marketplace_saas_info: Marketplace SaaS Info of the resource. - :vartype marketplace_saas_info: ~azure.mgmt.elastic.models.MarketplaceSaaSInfo - :ivar elastic_organization_id: The Elastic Organization Id. - :vartype elastic_organization_id: str - :ivar elastic_organization_name: The Elastic Organization Name. - :vartype elastic_organization_name: str - """ - - _validation = { - "marketplace_saas_info": {"readonly": True}, - } - - _attribute_map = { - "billed_azure_subscription_id": {"key": "billedAzureSubscriptionId", "type": "str"}, - "marketplace_saas_info": {"key": "marketplaceSaasInfo", "type": "MarketplaceSaaSInfo"}, - "elastic_organization_id": {"key": "elasticOrganizationId", "type": "str"}, - "elastic_organization_name": {"key": "elasticOrganizationName", "type": "str"}, - } - - def __init__( - self, - *, - billed_azure_subscription_id: Optional[str] = None, - elastic_organization_id: Optional[str] = None, - elastic_organization_name: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword billed_azure_subscription_id: The Azure Subscription ID to which the Organization - belongs and gets billed into. This is empty for a new user OR a user without an Elastic - Organization. - :paramtype billed_azure_subscription_id: str - :keyword elastic_organization_id: The Elastic Organization Id. - :paramtype elastic_organization_id: str - :keyword elastic_organization_name: The Elastic Organization Name. - :paramtype elastic_organization_name: str - """ - super().__init__(**kwargs) - self.billed_azure_subscription_id = billed_azure_subscription_id - self.marketplace_saas_info = None - self.elastic_organization_id = elastic_organization_id - self.elastic_organization_name = elastic_organization_name - - -class ElasticProperties(_serialization.Model): - """Elastic Resource Properties. - - :ivar elastic_cloud_user: Details of the user's elastic account. - :vartype elastic_cloud_user: ~azure.mgmt.elastic.models.ElasticCloudUser - :ivar elastic_cloud_deployment: Details of the elastic cloud deployment. - :vartype elastic_cloud_deployment: ~azure.mgmt.elastic.models.ElasticCloudDeployment - """ - - _attribute_map = { - "elastic_cloud_user": {"key": "elasticCloudUser", "type": "ElasticCloudUser"}, - "elastic_cloud_deployment": {"key": "elasticCloudDeployment", "type": "ElasticCloudDeployment"}, - } - - def __init__( - self, - *, - elastic_cloud_user: Optional["_models.ElasticCloudUser"] = None, - elastic_cloud_deployment: Optional["_models.ElasticCloudDeployment"] = None, - **kwargs: Any - ) -> None: - """ - :keyword elastic_cloud_user: Details of the user's elastic account. - :paramtype elastic_cloud_user: ~azure.mgmt.elastic.models.ElasticCloudUser - :keyword elastic_cloud_deployment: Details of the elastic cloud deployment. - :paramtype elastic_cloud_deployment: ~azure.mgmt.elastic.models.ElasticCloudDeployment - """ - super().__init__(**kwargs) - self.elastic_cloud_user = elastic_cloud_user - self.elastic_cloud_deployment = elastic_cloud_deployment - - -class ElasticTrafficFilter(_serialization.Model): - """Elastic traffic filter object. - - :ivar id: Id of the elastic filter. - :vartype id: str - :ivar name: Name of the elastic filter. - :vartype name: str - :ivar description: Description of the elastic filter. - :vartype description: str - :ivar region: Region of the elastic filter. - :vartype region: str - :ivar type: Type of the elastic filter. Known values are: "ip" and "azure_private_endpoint". - :vartype type: str or ~azure.mgmt.elastic.models.Type - :ivar include_by_default: IncludeByDefault for the elastic filter. - :vartype include_by_default: bool - :ivar rules: Rules in the elastic filter. - :vartype rules: list[~azure.mgmt.elastic.models.ElasticTrafficFilterRule] - """ - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "description": {"key": "description", "type": "str"}, - "region": {"key": "region", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "include_by_default": {"key": "includeByDefault", "type": "bool"}, - "rules": {"key": "rules", "type": "[ElasticTrafficFilterRule]"}, - } - - def __init__( - self, - *, - id: Optional[str] = None, # pylint: disable=redefined-builtin - name: Optional[str] = None, - description: Optional[str] = None, - region: Optional[str] = None, - type: Optional[Union[str, "_models.Type"]] = None, - include_by_default: Optional[bool] = None, - rules: Optional[List["_models.ElasticTrafficFilterRule"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword id: Id of the elastic filter. - :paramtype id: str - :keyword name: Name of the elastic filter. - :paramtype name: str - :keyword description: Description of the elastic filter. - :paramtype description: str - :keyword region: Region of the elastic filter. - :paramtype region: str - :keyword type: Type of the elastic filter. Known values are: "ip" and "azure_private_endpoint". - :paramtype type: str or ~azure.mgmt.elastic.models.Type - :keyword include_by_default: IncludeByDefault for the elastic filter. - :paramtype include_by_default: bool - :keyword rules: Rules in the elastic filter. - :paramtype rules: list[~azure.mgmt.elastic.models.ElasticTrafficFilterRule] - """ - super().__init__(**kwargs) - self.id = id - self.name = name - self.description = description - self.region = region - self.type = type - self.include_by_default = include_by_default - self.rules = rules - - -class ElasticTrafficFilterResponse(_serialization.Model): - """List of elastic traffic filters in the account. - - :ivar rulesets: List of elastic traffic filters in the account. - :vartype rulesets: list[~azure.mgmt.elastic.models.ElasticTrafficFilter] - """ - - _attribute_map = { - "rulesets": {"key": "rulesets", "type": "[ElasticTrafficFilter]"}, - } - - def __init__(self, *, rulesets: Optional[List["_models.ElasticTrafficFilter"]] = None, **kwargs: Any) -> None: - """ - :keyword rulesets: List of elastic traffic filters in the account. - :paramtype rulesets: list[~azure.mgmt.elastic.models.ElasticTrafficFilter] - """ - super().__init__(**kwargs) - self.rulesets = rulesets - - -class ElasticTrafficFilterRule(_serialization.Model): - """Elastic traffic filter rule object. - - :ivar source: IP of the elastic filter rule. - :vartype source: str - :ivar description: Description of the elastic filter rule. - :vartype description: str - :ivar azure_endpoint_guid: Guid of Private Endpoint in the elastic filter rule. - :vartype azure_endpoint_guid: str - :ivar azure_endpoint_name: Name of the Private Endpoint in the elastic filter rule. - :vartype azure_endpoint_name: str - :ivar id: Id of the elastic filter rule. - :vartype id: str - """ - - _attribute_map = { - "source": {"key": "source", "type": "str"}, - "description": {"key": "description", "type": "str"}, - "azure_endpoint_guid": {"key": "azureEndpointGuid", "type": "str"}, - "azure_endpoint_name": {"key": "azureEndpointName", "type": "str"}, - "id": {"key": "id", "type": "str"}, - } - - def __init__( - self, - *, - source: Optional[str] = None, - description: Optional[str] = None, - azure_endpoint_guid: Optional[str] = None, - azure_endpoint_name: Optional[str] = None, - id: Optional[str] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: - """ - :keyword source: IP of the elastic filter rule. - :paramtype source: str - :keyword description: Description of the elastic filter rule. - :paramtype description: str - :keyword azure_endpoint_guid: Guid of Private Endpoint in the elastic filter rule. - :paramtype azure_endpoint_guid: str - :keyword azure_endpoint_name: Name of the Private Endpoint in the elastic filter rule. - :paramtype azure_endpoint_name: str - :keyword id: Id of the elastic filter rule. - :paramtype id: str - """ - super().__init__(**kwargs) - self.source = source - self.description = description - self.azure_endpoint_guid = azure_endpoint_guid - self.azure_endpoint_name = azure_endpoint_name - self.id = id - - -class ElasticVersionListFormat(_serialization.Model): - """Elastic Version List Format. - - :ivar properties: Elastic Version Properties. - :vartype properties: ~azure.mgmt.elastic.models.ElasticVersionListProperties - """ - - _attribute_map = { - "properties": {"key": "properties", "type": "ElasticVersionListProperties"}, - } - - def __init__(self, *, properties: Optional["_models.ElasticVersionListProperties"] = None, **kwargs: Any) -> None: - """ - :keyword properties: Elastic Version Properties. - :paramtype properties: ~azure.mgmt.elastic.models.ElasticVersionListProperties - """ - super().__init__(**kwargs) - self.properties = properties - - -class ElasticVersionListProperties(_serialization.Model): - """Elastic Version Properties. - - :ivar version: Available elastic version of the given region. - :vartype version: str - """ - - _attribute_map = { - "version": {"key": "version", "type": "str"}, - } - - def __init__(self, *, version: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword version: Available elastic version of the given region. - :paramtype version: str - """ - super().__init__(**kwargs) - self.version = version - - -class ElasticVersionsListResponse(_serialization.Model): - """List of elastic versions available in a region. - - :ivar value: Results of a list operation. - :vartype value: list[~azure.mgmt.elastic.models.ElasticVersionListFormat] - :ivar next_link: Link to the next set of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[ElasticVersionListFormat]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.ElasticVersionListFormat"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Results of a list operation. - :paramtype value: list[~azure.mgmt.elastic.models.ElasticVersionListFormat] - :keyword next_link: Link to the next set of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class ErrorAdditionalInfo(_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: JSON - """ - - _validation = { - "type": {"readonly": True}, - "info": {"readonly": True}, - } - - _attribute_map = { - "type": {"key": "type", "type": "str"}, - "info": {"key": "info", "type": "object"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.type = None - self.info = None - - -class ErrorDetail(_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.elastic.models.ErrorDetail] - :ivar additional_info: The error additional info. - :vartype additional_info: list[~azure.mgmt.elastic.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": "[ErrorDetail]"}, - "additional_info": {"key": "additionalInfo", "type": "[ErrorAdditionalInfo]"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.code = None - self.message = None - self.target = None - self.details = None - self.additional_info = None - - -class ErrorResponse(_serialization.Model): - """Common error response for all Azure Resource Manager APIs to return error details for failed - operations. (This also follows the OData error response format.). - - :ivar error: The error object. - :vartype error: ~azure.mgmt.elastic.models.ErrorDetail - """ - - _attribute_map = { - "error": {"key": "error", "type": "ErrorDetail"}, - } - - def __init__(self, *, error: Optional["_models.ErrorDetail"] = None, **kwargs: Any) -> None: - """ - :keyword error: The error object. - :paramtype error: ~azure.mgmt.elastic.models.ErrorDetail - """ - super().__init__(**kwargs) - self.error = error - - -class ErrorResponseBody(_serialization.Model): - """Error response body. - - :ivar code: Error code. - :vartype code: str - :ivar message: Error message. - :vartype message: str - :ivar target: Error target. - :vartype target: str - :ivar details: Error details. - :vartype details: list[~azure.mgmt.elastic.models.ErrorResponseBody] - """ - - _attribute_map = { - "code": {"key": "code", "type": "str"}, - "message": {"key": "message", "type": "str"}, - "target": {"key": "target", "type": "str"}, - "details": {"key": "details", "type": "[ErrorResponseBody]"}, - } - - def __init__( - self, - *, - code: Optional[str] = None, - message: Optional[str] = None, - target: Optional[str] = None, - details: Optional[List["_models.ErrorResponseBody"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword code: Error code. - :paramtype code: str - :keyword message: Error message. - :paramtype message: str - :keyword target: Error target. - :paramtype target: str - :keyword details: Error details. - :paramtype details: list[~azure.mgmt.elastic.models.ErrorResponseBody] - """ - super().__init__(**kwargs) - self.code = code - self.message = message - self.target = target - self.details = details - - -class ExternalUserCreationResponse(_serialization.Model): - """The properties of the response we got from elastic while creating external user. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar created: Shows if user is created or updated. - :vartype created: bool - """ - - _validation = { - "created": {"readonly": True}, - } - - _attribute_map = { - "created": {"key": "created", "type": "bool"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.created = None - - -class ExternalUserInfo(_serialization.Model): - """The properties of the request required for creating user on elastic side. - - :ivar user_name: Username of the user to be created or updated. - :vartype user_name: str - :ivar full_name: Full name of the user to be created or updated. - :vartype full_name: str - :ivar password: Password of the user to be created or updated. - :vartype password: str - :ivar email_id: Email id of the user to be created or updated. - :vartype email_id: str - :ivar roles: Roles to be assigned for created or updated user. - :vartype roles: list[str] - """ - - _attribute_map = { - "user_name": {"key": "userName", "type": "str"}, - "full_name": {"key": "fullName", "type": "str"}, - "password": {"key": "password", "type": "str"}, - "email_id": {"key": "emailId", "type": "str"}, - "roles": {"key": "roles", "type": "[str]"}, - } - - def __init__( - self, - *, - user_name: Optional[str] = None, - full_name: Optional[str] = None, - password: Optional[str] = None, - email_id: Optional[str] = None, - roles: Optional[List[str]] = None, - **kwargs: Any - ) -> None: - """ - :keyword user_name: Username of the user to be created or updated. - :paramtype user_name: str - :keyword full_name: Full name of the user to be created or updated. - :paramtype full_name: str - :keyword password: Password of the user to be created or updated. - :paramtype password: str - :keyword email_id: Email id of the user to be created or updated. - :paramtype email_id: str - :keyword roles: Roles to be assigned for created or updated user. - :paramtype roles: list[str] - """ - super().__init__(**kwargs) - self.user_name = user_name - self.full_name = full_name - self.password = password - self.email_id = email_id - self.roles = roles - - -class FilteringTag(_serialization.Model): - """The definition of a filtering tag. Filtering tags are used for capturing resources and - include/exclude them from being monitored. - - :ivar name: The name (also known as the key) of the tag. - :vartype name: str - :ivar value: The value of the tag. - :vartype value: str - :ivar action: Valid actions for a filtering tag. Known values are: "Include" and "Exclude". - :vartype action: str or ~azure.mgmt.elastic.models.TagAction - """ - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "value": {"key": "value", "type": "str"}, - "action": {"key": "action", "type": "str"}, - } - - def __init__( - self, - *, - name: Optional[str] = None, - value: Optional[str] = None, - action: Optional[Union[str, "_models.TagAction"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword name: The name (also known as the key) of the tag. - :paramtype name: str - :keyword value: The value of the tag. - :paramtype value: str - :keyword action: Valid actions for a filtering tag. Known values are: "Include" and "Exclude". - :paramtype action: str or ~azure.mgmt.elastic.models.TagAction - """ - super().__init__(**kwargs) - self.name = name - self.value = value - self.action = action - - -class IdentityProperties(_serialization.Model): - """Identity properties. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar principal_id: The identity ID. - :vartype principal_id: str - :ivar tenant_id: The tenant ID of resource. - :vartype tenant_id: str - :ivar type: Managed identity type. "SystemAssigned" - :vartype type: str or ~azure.mgmt.elastic.models.ManagedIdentityTypes - """ - - _validation = { - "principal_id": {"readonly": True}, - "tenant_id": {"readonly": True}, - } - - _attribute_map = { - "principal_id": {"key": "principalId", "type": "str"}, - "tenant_id": {"key": "tenantId", "type": "str"}, - "type": {"key": "type", "type": "str"}, - } - - def __init__(self, *, type: Optional[Union[str, "_models.ManagedIdentityTypes"]] = None, **kwargs: Any) -> None: - """ - :keyword type: Managed identity type. "SystemAssigned" - :paramtype type: str or ~azure.mgmt.elastic.models.ManagedIdentityTypes - """ - super().__init__(**kwargs) - self.principal_id = None - self.tenant_id = None - self.type = type - - -class LogRules(_serialization.Model): - """Set of rules for sending logs for the Monitor resource. - - :ivar send_aad_logs: Flag specifying if AAD logs should be sent for the Monitor resource. - :vartype send_aad_logs: bool - :ivar send_subscription_logs: Flag specifying if subscription logs should be sent for the - Monitor resource. - :vartype send_subscription_logs: bool - :ivar send_activity_logs: Flag specifying if activity logs from Azure resources should be sent - for the Monitor resource. - :vartype send_activity_logs: bool - :ivar filtering_tags: List of filtering tags to be used for capturing logs. This only takes - effect if SendActivityLogs flag is enabled. If empty, all resources will be captured. If only - Exclude action is specified, the rules will apply to the list of all available resources. If - Include actions are specified, the rules will only include resources with the associated tags. - :vartype filtering_tags: list[~azure.mgmt.elastic.models.FilteringTag] - """ - - _attribute_map = { - "send_aad_logs": {"key": "sendAadLogs", "type": "bool"}, - "send_subscription_logs": {"key": "sendSubscriptionLogs", "type": "bool"}, - "send_activity_logs": {"key": "sendActivityLogs", "type": "bool"}, - "filtering_tags": {"key": "filteringTags", "type": "[FilteringTag]"}, - } - - def __init__( - self, - *, - send_aad_logs: Optional[bool] = None, - send_subscription_logs: Optional[bool] = None, - send_activity_logs: Optional[bool] = None, - filtering_tags: Optional[List["_models.FilteringTag"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword send_aad_logs: Flag specifying if AAD logs should be sent for the Monitor resource. - :paramtype send_aad_logs: bool - :keyword send_subscription_logs: Flag specifying if subscription logs should be sent for the - Monitor resource. - :paramtype send_subscription_logs: bool - :keyword send_activity_logs: Flag specifying if activity logs from Azure resources should be - sent for the Monitor resource. - :paramtype send_activity_logs: bool - :keyword filtering_tags: List of filtering tags to be used for capturing logs. This only takes - effect if SendActivityLogs flag is enabled. If empty, all resources will be captured. If only - Exclude action is specified, the rules will apply to the list of all available resources. If - Include actions are specified, the rules will only include resources with the associated tags. - :paramtype filtering_tags: list[~azure.mgmt.elastic.models.FilteringTag] - """ - super().__init__(**kwargs) - self.send_aad_logs = send_aad_logs - self.send_subscription_logs = send_subscription_logs - self.send_activity_logs = send_activity_logs - self.filtering_tags = filtering_tags - - -class MarketplaceSaaSInfo(_serialization.Model): - """Marketplace SAAS Info of the resource. - - :ivar marketplace_subscription: Marketplace Subscription. - :vartype marketplace_subscription: - ~azure.mgmt.elastic.models.MarketplaceSaaSInfoMarketplaceSubscription - :ivar marketplace_name: Marketplace Subscription Details: SAAS Name. - :vartype marketplace_name: str - :ivar marketplace_resource_id: Marketplace Subscription Details: Resource URI. - :vartype marketplace_resource_id: str - :ivar marketplace_status: Marketplace Subscription Details: SaaS Subscription Status. - :vartype marketplace_status: str - :ivar billed_azure_subscription_id: The Azure Subscription ID to which the Marketplace - Subscription belongs and gets billed into. - :vartype billed_azure_subscription_id: str - :ivar subscribed: Flag specifying if the Marketplace status is subscribed or not. - :vartype subscribed: bool - """ - - _attribute_map = { - "marketplace_subscription": { - "key": "marketplaceSubscription", - "type": "MarketplaceSaaSInfoMarketplaceSubscription", - }, - "marketplace_name": {"key": "marketplaceName", "type": "str"}, - "marketplace_resource_id": {"key": "marketplaceResourceId", "type": "str"}, - "marketplace_status": {"key": "marketplaceStatus", "type": "str"}, - "billed_azure_subscription_id": {"key": "billedAzureSubscriptionId", "type": "str"}, - "subscribed": {"key": "subscribed", "type": "bool"}, - } - - def __init__( - self, - *, - marketplace_subscription: Optional["_models.MarketplaceSaaSInfoMarketplaceSubscription"] = None, - marketplace_name: Optional[str] = None, - marketplace_resource_id: Optional[str] = None, - marketplace_status: Optional[str] = None, - billed_azure_subscription_id: Optional[str] = None, - subscribed: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword marketplace_subscription: Marketplace Subscription. - :paramtype marketplace_subscription: - ~azure.mgmt.elastic.models.MarketplaceSaaSInfoMarketplaceSubscription - :keyword marketplace_name: Marketplace Subscription Details: SAAS Name. - :paramtype marketplace_name: str - :keyword marketplace_resource_id: Marketplace Subscription Details: Resource URI. - :paramtype marketplace_resource_id: str - :keyword marketplace_status: Marketplace Subscription Details: SaaS Subscription Status. - :paramtype marketplace_status: str - :keyword billed_azure_subscription_id: The Azure Subscription ID to which the Marketplace - Subscription belongs and gets billed into. - :paramtype billed_azure_subscription_id: str - :keyword subscribed: Flag specifying if the Marketplace status is subscribed or not. - :paramtype subscribed: bool - """ - super().__init__(**kwargs) - self.marketplace_subscription = marketplace_subscription - self.marketplace_name = marketplace_name - self.marketplace_resource_id = marketplace_resource_id - self.marketplace_status = marketplace_status - self.billed_azure_subscription_id = billed_azure_subscription_id - self.subscribed = subscribed - - -class MarketplaceSaaSInfoMarketplaceSubscription(_serialization.Model): # pylint: disable=name-too-long - """Marketplace Subscription. - - :ivar id: Marketplace Subscription Id. This is a GUID-formatted string. - :vartype id: str - :ivar publisher_id: Publisher Id of the Marketplace offer. - :vartype publisher_id: str - :ivar offer_id: Offer Id of the Marketplace offer,. - :vartype offer_id: str - """ - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "publisher_id": {"key": "publisherId", "type": "str"}, - "offer_id": {"key": "offerId", "type": "str"}, - } - - def __init__( - self, - *, - id: Optional[str] = None, # pylint: disable=redefined-builtin - publisher_id: Optional[str] = None, - offer_id: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword id: Marketplace Subscription Id. This is a GUID-formatted string. - :paramtype id: str - :keyword publisher_id: Publisher Id of the Marketplace offer. - :paramtype publisher_id: str - :keyword offer_id: Offer Id of the Marketplace offer,. - :paramtype offer_id: str - """ - super().__init__(**kwargs) - self.id = id - self.publisher_id = publisher_id - self.offer_id = offer_id - - -class MonitoredResource(_serialization.Model): - """The properties of a resource currently being monitored by the Elastic monitor resource. - - :ivar id: The ARM id of the resource. - :vartype id: str - :ivar sending_logs: Flag indicating the status of the resource for sending logs operation to - Elastic. Known values are: "True" and "False". - :vartype sending_logs: str or ~azure.mgmt.elastic.models.SendingLogs - :ivar reason_for_logs_status: Reason for why the resource is sending logs (or why it is not - sending). - :vartype reason_for_logs_status: str - """ - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "sending_logs": {"key": "sendingLogs", "type": "str"}, - "reason_for_logs_status": {"key": "reasonForLogsStatus", "type": "str"}, - } - - def __init__( - self, - *, - id: Optional[str] = None, # pylint: disable=redefined-builtin - sending_logs: Optional[Union[str, "_models.SendingLogs"]] = None, - reason_for_logs_status: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword id: The ARM id of the resource. - :paramtype id: str - :keyword sending_logs: Flag indicating the status of the resource for sending logs operation to - Elastic. Known values are: "True" and "False". - :paramtype sending_logs: str or ~azure.mgmt.elastic.models.SendingLogs - :keyword reason_for_logs_status: Reason for why the resource is sending logs (or why it is not - sending). - :paramtype reason_for_logs_status: str - """ - super().__init__(**kwargs) - self.id = id - self.sending_logs = sending_logs - self.reason_for_logs_status = reason_for_logs_status - - -class MonitoredResourceListResponse(_serialization.Model): - """Response of a list operation. - - :ivar value: Results of a list operation. - :vartype value: list[~azure.mgmt.elastic.models.MonitoredResource] - :ivar next_link: Link to the next set of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[MonitoredResource]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.MonitoredResource"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Results of a list operation. - :paramtype value: list[~azure.mgmt.elastic.models.MonitoredResource] - :keyword next_link: Link to the next set of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class MonitoredSubscription(_serialization.Model): - """The list of subscriptions and it's monitoring status by current Elastic monitor. - - :ivar subscription_id: The subscriptionId to be monitored. - :vartype subscription_id: str - :ivar status: The state of monitoring. Known values are: "InProgress", "Active", "Failed", and - "Deleting". - :vartype status: str or ~azure.mgmt.elastic.models.Status - :ivar error: The reason of not monitoring the subscription. - :vartype error: str - :ivar tag_rules: Definition of the properties for a TagRules resource. - :vartype tag_rules: ~azure.mgmt.elastic.models.MonitoringTagRulesProperties - """ - - _attribute_map = { - "subscription_id": {"key": "subscriptionId", "type": "str"}, - "status": {"key": "status", "type": "str"}, - "error": {"key": "error", "type": "str"}, - "tag_rules": {"key": "tagRules", "type": "MonitoringTagRulesProperties"}, - } - - def __init__( - self, - *, - subscription_id: Optional[str] = None, - status: Optional[Union[str, "_models.Status"]] = None, - error: Optional[str] = None, - tag_rules: Optional["_models.MonitoringTagRulesProperties"] = None, - **kwargs: Any - ) -> None: - """ - :keyword subscription_id: The subscriptionId to be monitored. - :paramtype subscription_id: str - :keyword status: The state of monitoring. Known values are: "InProgress", "Active", "Failed", - and "Deleting". - :paramtype status: str or ~azure.mgmt.elastic.models.Status - :keyword error: The reason of not monitoring the subscription. - :paramtype error: str - :keyword tag_rules: Definition of the properties for a TagRules resource. - :paramtype tag_rules: ~azure.mgmt.elastic.models.MonitoringTagRulesProperties - """ - super().__init__(**kwargs) - self.subscription_id = subscription_id - self.status = status - self.error = error - self.tag_rules = tag_rules - - -class MonitoredSubscriptionProperties(_serialization.Model): - """The request to update subscriptions needed to be monitored by the Elastic monitor resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: Name of the monitored subscription resource. - :vartype name: str - :ivar id: The id of the monitored subscription resource. - :vartype id: str - :ivar type: The type of the monitored subscription resource. - :vartype type: str - :ivar properties: The request to update subscriptions needed to be monitored by the Elastic - monitor resource. - :vartype properties: ~azure.mgmt.elastic.models.SubscriptionList - """ - - _validation = { - "name": {"readonly": True}, - "id": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "id": {"key": "id", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "properties": {"key": "properties", "type": "SubscriptionList"}, - } - - def __init__(self, *, properties: Optional["_models.SubscriptionList"] = None, **kwargs: Any) -> None: - """ - :keyword properties: The request to update subscriptions needed to be monitored by the Elastic - monitor resource. - :paramtype properties: ~azure.mgmt.elastic.models.SubscriptionList - """ - super().__init__(**kwargs) - self.name = None - self.id = None - self.type = None - self.properties = properties - - -class MonitoredSubscriptionPropertiesList(_serialization.Model): - """MonitoredSubscriptionPropertiesList. - - :ivar value: - :vartype value: list[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :ivar next_link: The link to the next page of items. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[MonitoredSubscriptionProperties]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.MonitoredSubscriptionProperties"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: - :paramtype value: list[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :keyword next_link: The link to the next page of items. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class MonitoringTagRules(_serialization.Model): - """Capture logs and metrics of Azure resources based on ARM tags. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: Name of the rule set. - :vartype name: str - :ivar id: The id of the rule set. - :vartype id: str - :ivar type: The type of the rule set. - :vartype type: str - :ivar properties: Properties of the monitoring tag rules. - :vartype properties: ~azure.mgmt.elastic.models.MonitoringTagRulesProperties - :ivar system_data: The system metadata relating to this resource. - :vartype system_data: ~azure.mgmt.elastic.models.SystemData - """ - - _validation = { - "name": {"readonly": True}, - "id": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "id": {"key": "id", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "properties": {"key": "properties", "type": "MonitoringTagRulesProperties"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - } - - def __init__(self, *, properties: Optional["_models.MonitoringTagRulesProperties"] = None, **kwargs: Any) -> None: - """ - :keyword properties: Properties of the monitoring tag rules. - :paramtype properties: ~azure.mgmt.elastic.models.MonitoringTagRulesProperties - """ - super().__init__(**kwargs) - self.name = None - self.id = None - self.type = None - self.properties = properties - self.system_data = None - - -class MonitoringTagRulesListResponse(_serialization.Model): - """Response of a list operation. - - :ivar value: Results of a list operation. - :vartype value: list[~azure.mgmt.elastic.models.MonitoringTagRules] - :ivar next_link: Link to the next set of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[MonitoringTagRules]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.MonitoringTagRules"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Results of a list operation. - :paramtype value: list[~azure.mgmt.elastic.models.MonitoringTagRules] - :keyword next_link: Link to the next set of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class MonitoringTagRulesProperties(_serialization.Model): - """Definition of the properties for a TagRules resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar provisioning_state: Provisioning state of the monitoring tag rules. Known values are: - "Accepted", "Creating", "Updating", "Deleting", "Succeeded", "Failed", "Canceled", "Deleted", - and "NotSpecified". - :vartype provisioning_state: str or ~azure.mgmt.elastic.models.ProvisioningState - :ivar log_rules: Rules for sending logs. - :vartype log_rules: ~azure.mgmt.elastic.models.LogRules - """ - - _validation = { - "provisioning_state": {"readonly": True}, - } - - _attribute_map = { - "provisioning_state": {"key": "provisioningState", "type": "str"}, - "log_rules": {"key": "logRules", "type": "LogRules"}, - } - - def __init__(self, *, log_rules: Optional["_models.LogRules"] = None, **kwargs: Any) -> None: - """ - :keyword log_rules: Rules for sending logs. - :paramtype log_rules: ~azure.mgmt.elastic.models.LogRules - """ - super().__init__(**kwargs) - self.provisioning_state = None - self.log_rules = log_rules - - -class MonitorProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes - """Properties specific to the monitor resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar provisioning_state: Provisioning state of the monitor resource. Known values are: - "Accepted", "Creating", "Updating", "Deleting", "Succeeded", "Failed", "Canceled", "Deleted", - and "NotSpecified". - :vartype provisioning_state: str or ~azure.mgmt.elastic.models.ProvisioningState - :ivar monitoring_status: Flag specifying if the resource monitoring is enabled or disabled. - Known values are: "Enabled" and "Disabled". - :vartype monitoring_status: str or ~azure.mgmt.elastic.models.MonitoringStatus - :ivar elastic_properties: Elastic cloud properties. - :vartype elastic_properties: ~azure.mgmt.elastic.models.ElasticProperties - :ivar user_info: User information. - :vartype user_info: ~azure.mgmt.elastic.models.UserInfo - :ivar plan_details: Plan details of the monitor resource. - :vartype plan_details: ~azure.mgmt.elastic.models.PlanDetails - :ivar version: Version of elastic of the monitor resource. - :vartype version: str - :ivar subscription_state: State of the Azure Subscription containing the monitor resource. - :vartype subscription_state: str - :ivar saa_s_azure_subscription_status: Status of Azure Subscription where Marketplace SaaS is - located. - :vartype saa_s_azure_subscription_status: str - :ivar source_campaign_name: Name of the marketing campaign. - :vartype source_campaign_name: str - :ivar source_campaign_id: A unique identifier associated with the campaign. - :vartype source_campaign_id: str - :ivar liftr_resource_category: Known values are: "Unknown" and "MonitorLogs". - :vartype liftr_resource_category: str or ~azure.mgmt.elastic.models.LiftrResourceCategories - :ivar liftr_resource_preference: The priority of the resource. - :vartype liftr_resource_preference: int - :ivar generate_api_key: Flag to determine if User API Key has to be generated and shared. - :vartype generate_api_key: bool - """ - - _validation = { - "provisioning_state": {"readonly": True}, - "liftr_resource_category": {"readonly": True}, - "liftr_resource_preference": {"readonly": True}, - } - - _attribute_map = { - "provisioning_state": {"key": "provisioningState", "type": "str"}, - "monitoring_status": {"key": "monitoringStatus", "type": "str"}, - "elastic_properties": {"key": "elasticProperties", "type": "ElasticProperties"}, - "user_info": {"key": "userInfo", "type": "UserInfo"}, - "plan_details": {"key": "planDetails", "type": "PlanDetails"}, - "version": {"key": "version", "type": "str"}, - "subscription_state": {"key": "subscriptionState", "type": "str"}, - "saa_s_azure_subscription_status": {"key": "saaSAzureSubscriptionStatus", "type": "str"}, - "source_campaign_name": {"key": "sourceCampaignName", "type": "str"}, - "source_campaign_id": {"key": "sourceCampaignId", "type": "str"}, - "liftr_resource_category": {"key": "liftrResourceCategory", "type": "str"}, - "liftr_resource_preference": {"key": "liftrResourcePreference", "type": "int"}, - "generate_api_key": {"key": "generateApiKey", "type": "bool"}, - } - - def __init__( - self, - *, - monitoring_status: Optional[Union[str, "_models.MonitoringStatus"]] = None, - elastic_properties: Optional["_models.ElasticProperties"] = None, - user_info: Optional["_models.UserInfo"] = None, - plan_details: Optional["_models.PlanDetails"] = None, - version: Optional[str] = None, - subscription_state: Optional[str] = None, - saa_s_azure_subscription_status: Optional[str] = None, - source_campaign_name: Optional[str] = None, - source_campaign_id: Optional[str] = None, - generate_api_key: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword monitoring_status: Flag specifying if the resource monitoring is enabled or disabled. - Known values are: "Enabled" and "Disabled". - :paramtype monitoring_status: str or ~azure.mgmt.elastic.models.MonitoringStatus - :keyword elastic_properties: Elastic cloud properties. - :paramtype elastic_properties: ~azure.mgmt.elastic.models.ElasticProperties - :keyword user_info: User information. - :paramtype user_info: ~azure.mgmt.elastic.models.UserInfo - :keyword plan_details: Plan details of the monitor resource. - :paramtype plan_details: ~azure.mgmt.elastic.models.PlanDetails - :keyword version: Version of elastic of the monitor resource. - :paramtype version: str - :keyword subscription_state: State of the Azure Subscription containing the monitor resource. - :paramtype subscription_state: str - :keyword saa_s_azure_subscription_status: Status of Azure Subscription where Marketplace SaaS - is located. - :paramtype saa_s_azure_subscription_status: str - :keyword source_campaign_name: Name of the marketing campaign. - :paramtype source_campaign_name: str - :keyword source_campaign_id: A unique identifier associated with the campaign. - :paramtype source_campaign_id: str - :keyword generate_api_key: Flag to determine if User API Key has to be generated and shared. - :paramtype generate_api_key: bool - """ - super().__init__(**kwargs) - self.provisioning_state = None - self.monitoring_status = monitoring_status - self.elastic_properties = elastic_properties - self.user_info = user_info - self.plan_details = plan_details - self.version = version - self.subscription_state = subscription_state - self.saa_s_azure_subscription_status = saa_s_azure_subscription_status - self.source_campaign_name = source_campaign_name - self.source_campaign_id = source_campaign_id - self.liftr_resource_category = None - self.liftr_resource_preference = None - self.generate_api_key = generate_api_key - - -class OpenAIIntegrationProperties(_serialization.Model): - """Open AI Integration details. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar open_ai_resource_id: The resource name of Open AI resource. - :vartype open_ai_resource_id: str - :ivar open_ai_resource_endpoint: The API endpoint for Open AI resource. - :vartype open_ai_resource_endpoint: str - :ivar open_ai_connector_id: The connector id of Open AI resource. - :vartype open_ai_connector_id: str - :ivar key: Value of API key for Open AI resource. - :vartype key: str - :ivar last_refresh_at: Last Update Timestamp for key updation. - :vartype last_refresh_at: ~datetime.datetime - """ - - _validation = { - "last_refresh_at": {"readonly": True}, - } - - _attribute_map = { - "open_ai_resource_id": {"key": "openAIResourceId", "type": "str"}, - "open_ai_resource_endpoint": {"key": "openAIResourceEndpoint", "type": "str"}, - "open_ai_connector_id": {"key": "openAIConnectorId", "type": "str"}, - "key": {"key": "key", "type": "str"}, - "last_refresh_at": {"key": "lastRefreshAt", "type": "iso-8601"}, - } - - def __init__( - self, - *, - open_ai_resource_id: Optional[str] = None, - open_ai_resource_endpoint: Optional[str] = None, - open_ai_connector_id: Optional[str] = None, - key: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword open_ai_resource_id: The resource name of Open AI resource. - :paramtype open_ai_resource_id: str - :keyword open_ai_resource_endpoint: The API endpoint for Open AI resource. - :paramtype open_ai_resource_endpoint: str - :keyword open_ai_connector_id: The connector id of Open AI resource. - :paramtype open_ai_connector_id: str - :keyword key: Value of API key for Open AI resource. - :paramtype key: str - """ - super().__init__(**kwargs) - self.open_ai_resource_id = open_ai_resource_id - self.open_ai_resource_endpoint = open_ai_resource_endpoint - self.open_ai_connector_id = open_ai_connector_id - self.key = key - self.last_refresh_at = None - - -class OpenAIIntegrationRPModel(_serialization.Model): - """Capture properties of Open AI resource Integration. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: Name of the integration. - :vartype name: str - :ivar id: The id of the integration. - :vartype id: str - :ivar type: The type of the integration. - :vartype type: str - :ivar properties: Open AI Integration details. - :vartype properties: ~azure.mgmt.elastic.models.OpenAIIntegrationProperties - """ - - _validation = { - "name": {"readonly": True}, - "id": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "id": {"key": "id", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "properties": {"key": "properties", "type": "OpenAIIntegrationProperties"}, - } - - def __init__(self, *, properties: Optional["_models.OpenAIIntegrationProperties"] = None, **kwargs: Any) -> None: - """ - :keyword properties: Open AI Integration details. - :paramtype properties: ~azure.mgmt.elastic.models.OpenAIIntegrationProperties - """ - super().__init__(**kwargs) - self.name = None - self.id = None - self.type = None - self.properties = properties - - -class OpenAIIntegrationRPModelListResponse(_serialization.Model): - """Response of a list operation. - - :ivar value: Results of a list operation. - :vartype value: list[~azure.mgmt.elastic.models.OpenAIIntegrationRPModel] - :ivar next_link: Link to the next set of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[OpenAIIntegrationRPModel]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.OpenAIIntegrationRPModel"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: Results of a list operation. - :paramtype value: list[~azure.mgmt.elastic.models.OpenAIIntegrationRPModel] - :keyword next_link: Link to the next set of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class OpenAIIntegrationStatusResponse(_serialization.Model): - """Status of the OpenAI Integration. - - :ivar properties: Status of the OpenAI Integration. - :vartype properties: ~azure.mgmt.elastic.models.OpenAIIntegrationStatusResponseProperties - """ - - _attribute_map = { - "properties": {"key": "properties", "type": "OpenAIIntegrationStatusResponseProperties"}, - } - - def __init__( - self, *, properties: Optional["_models.OpenAIIntegrationStatusResponseProperties"] = None, **kwargs: Any - ) -> None: - """ - :keyword properties: Status of the OpenAI Integration. - :paramtype properties: ~azure.mgmt.elastic.models.OpenAIIntegrationStatusResponseProperties - """ - super().__init__(**kwargs) - self.properties = properties - - -class OpenAIIntegrationStatusResponseProperties(_serialization.Model): # pylint: disable=name-too-long - """Status of the OpenAI Integration. - - :ivar status: Status of the OpenAI Integration. - :vartype status: str - """ - - _attribute_map = { - "status": {"key": "status", "type": "str"}, - } - - def __init__(self, *, status: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword status: Status of the OpenAI Integration. - :paramtype status: str - """ - super().__init__(**kwargs) - self.status = status - - -class OperationDisplay(_serialization.Model): - """The object that represents the operation. - - :ivar provider: Service provider, i.e., Microsoft.Elastic. - :vartype provider: str - :ivar resource: Type on which the operation is performed, e.g., 'monitors'. - :vartype resource: str - :ivar operation: Operation type, e.g., read, write, delete, etc. - :vartype operation: str - :ivar description: Description of the operation, e.g., 'Write monitors'. - :vartype description: str - """ - - _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, - *, - provider: Optional[str] = None, - resource: Optional[str] = None, - operation: Optional[str] = None, - description: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword provider: Service provider, i.e., Microsoft.Elastic. - :paramtype provider: str - :keyword resource: Type on which the operation is performed, e.g., 'monitors'. - :paramtype resource: str - :keyword operation: Operation type, e.g., read, write, delete, etc. - :paramtype operation: str - :keyword description: Description of the operation, e.g., 'Write monitors'. - :paramtype description: str - """ - super().__init__(**kwargs) - self.provider = provider - self.resource = resource - self.operation = operation - self.description = description - - -class OperationListResult(_serialization.Model): - """Result of GET request to list the Microsoft.Elastic operations. - - :ivar value: List of operations supported by the Microsoft.Elastic provider. - :vartype value: list[~azure.mgmt.elastic.models.OperationResult] - :ivar next_link: URL to get the next set of operation list results if there are any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[OperationResult]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, *, value: Optional[List["_models.OperationResult"]] = None, next_link: Optional[str] = None, **kwargs: Any - ) -> None: - """ - :keyword value: List of operations supported by the Microsoft.Elastic provider. - :paramtype value: list[~azure.mgmt.elastic.models.OperationResult] - :keyword next_link: URL to get the next set of operation list results if there are any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class OperationResult(_serialization.Model): - """A Microsoft.Elastic REST API operation. - - :ivar name: Operation name, i.e., {provider}/{resource}/{operation}. - :vartype name: str - :ivar is_data_action: Indicates whether the operation is a data action. - :vartype is_data_action: bool - :ivar display: The object that represents the operation. - :vartype display: ~azure.mgmt.elastic.models.OperationDisplay - :ivar origin: Origin of the operation. - :vartype origin: str - """ - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "is_data_action": {"key": "isDataAction", "type": "bool"}, - "display": {"key": "display", "type": "OperationDisplay"}, - "origin": {"key": "origin", "type": "str"}, - } - - def __init__( - self, - *, - name: Optional[str] = None, - is_data_action: Optional[bool] = None, - display: Optional["_models.OperationDisplay"] = None, - origin: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword name: Operation name, i.e., {provider}/{resource}/{operation}. - :paramtype name: str - :keyword is_data_action: Indicates whether the operation is a data action. - :paramtype is_data_action: bool - :keyword display: The object that represents the operation. - :paramtype display: ~azure.mgmt.elastic.models.OperationDisplay - :keyword origin: Origin of the operation. - :paramtype origin: str - """ - super().__init__(**kwargs) - self.name = name - self.is_data_action = is_data_action - self.display = display - self.origin = origin - - -class PartnerBillingEntity(_serialization.Model): - """Partner Billing details associated with the resource. - - :ivar id: The Elastic Organization Id. - :vartype id: str - :ivar name: The Elastic Organization Name. - :vartype name: str - :ivar partner_entity_uri: Link to the elastic organization page. - :vartype partner_entity_uri: str - """ - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "partner_entity_uri": {"key": "partnerEntityUri", "type": "str"}, - } - - def __init__( - self, - *, - id: Optional[str] = None, # pylint: disable=redefined-builtin - name: Optional[str] = None, - partner_entity_uri: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword id: The Elastic Organization Id. - :paramtype id: str - :keyword name: The Elastic Organization Name. - :paramtype name: str - :keyword partner_entity_uri: Link to the elastic organization page. - :paramtype partner_entity_uri: str - """ - super().__init__(**kwargs) - self.id = id - self.name = name - self.partner_entity_uri = partner_entity_uri - - -class PlanDetails(_serialization.Model): - """Plan details of the monitor resource. - - :ivar offer_id: Offer ID of the plan. - :vartype offer_id: str - :ivar publisher_id: Publisher ID of the plan. - :vartype publisher_id: str - :ivar term_id: Term ID of the plan. - :vartype term_id: str - :ivar plan_id: Plan ID. - :vartype plan_id: str - :ivar plan_name: Plan Name. - :vartype plan_name: str - """ - - _attribute_map = { - "offer_id": {"key": "offerID", "type": "str"}, - "publisher_id": {"key": "publisherID", "type": "str"}, - "term_id": {"key": "termID", "type": "str"}, - "plan_id": {"key": "planID", "type": "str"}, - "plan_name": {"key": "planName", "type": "str"}, - } - - def __init__( - self, - *, - offer_id: Optional[str] = None, - publisher_id: Optional[str] = None, - term_id: Optional[str] = None, - plan_id: Optional[str] = None, - plan_name: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword offer_id: Offer ID of the plan. - :paramtype offer_id: str - :keyword publisher_id: Publisher ID of the plan. - :paramtype publisher_id: str - :keyword term_id: Term ID of the plan. - :paramtype term_id: str - :keyword plan_id: Plan ID. - :paramtype plan_id: str - :keyword plan_name: Plan Name. - :paramtype plan_name: str - """ - super().__init__(**kwargs) - self.offer_id = offer_id - self.publisher_id = publisher_id - self.term_id = term_id - self.plan_id = plan_id - self.plan_name = plan_name - - -class ResourceProviderDefaultErrorResponse(_serialization.Model): - """RP default error response. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar error: Response body of Error. - :vartype error: ~azure.mgmt.elastic.models.ErrorResponseBody - """ - - _validation = { - "error": {"readonly": True}, - } - - _attribute_map = { - "error": {"key": "error", "type": "ErrorResponseBody"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.error = None - - -class ResourceSku(_serialization.Model): - """Microsoft.Elastic SKU. - - All required parameters must be populated in order to send to server. - - :ivar name: Name of the SKU. Required. - :vartype name: str - """ - - _validation = { - "name": {"required": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - } - - def __init__(self, *, name: str, **kwargs: Any) -> None: - """ - :keyword name: Name of the SKU. Required. - :paramtype name: str - """ - super().__init__(**kwargs) - self.name = name - - -class ResubscribeProperties(_serialization.Model): - """Resubscribe Properties. - - :ivar plan_id: Newly selected plan Id to create the new Marketplace subscription for - Resubscribe. - :vartype plan_id: str - :ivar term: Newly selected term to create the new Marketplace subscription for Resubscribe. - :vartype term: str - :ivar subscription_id: Newly selected Azure Subscription Id in which the new Marketplace - subscription will be created for Resubscribe. - :vartype subscription_id: str - :ivar resource_group: Newly selected Azure resource group in which the new Marketplace - subscription will be created for Resubscribe. - :vartype resource_group: str - :ivar organization_id: Organization Id of the Elastic Organization that needs to be - resubscribed. - :vartype organization_id: str - """ - - _attribute_map = { - "plan_id": {"key": "planId", "type": "str"}, - "term": {"key": "term", "type": "str"}, - "subscription_id": {"key": "subscriptionId", "type": "str"}, - "resource_group": {"key": "resourceGroup", "type": "str"}, - "organization_id": {"key": "organizationId", "type": "str"}, - } - - def __init__( - self, - *, - plan_id: Optional[str] = None, - term: Optional[str] = None, - subscription_id: Optional[str] = None, - resource_group: Optional[str] = None, - organization_id: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword plan_id: Newly selected plan Id to create the new Marketplace subscription for - Resubscribe. - :paramtype plan_id: str - :keyword term: Newly selected term to create the new Marketplace subscription for Resubscribe. - :paramtype term: str - :keyword subscription_id: Newly selected Azure Subscription Id in which the new Marketplace - subscription will be created for Resubscribe. - :paramtype subscription_id: str - :keyword resource_group: Newly selected Azure resource group in which the new Marketplace - subscription will be created for Resubscribe. - :paramtype resource_group: str - :keyword organization_id: Organization Id of the Elastic Organization that needs to be - resubscribed. - :paramtype organization_id: str - """ - super().__init__(**kwargs) - self.plan_id = plan_id - self.term = term - self.subscription_id = subscription_id - self.resource_group = resource_group - self.organization_id = organization_id - - -class SubscriptionList(_serialization.Model): - """The request to update subscriptions needed to be monitored by the Elastic monitor resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar operation: The operation for the patch on the resource. Known values are: "AddBegin", - "AddComplete", "DeleteBegin", "DeleteComplete", and "Active". - :vartype operation: str or ~azure.mgmt.elastic.models.Operation - :ivar monitored_subscription_list: List of subscriptions and the state of the monitoring. - :vartype monitored_subscription_list: list[~azure.mgmt.elastic.models.MonitoredSubscription] - :ivar provisioning_state: Provisioning State of the resource. Known values are: "Accepted", - "Creating", "Updating", "Deleting", "Succeeded", "Failed", "Canceled", "Deleted", and - "NotSpecified". - :vartype provisioning_state: str or ~azure.mgmt.elastic.models.ProvisioningState - """ - - _validation = { - "provisioning_state": {"readonly": True}, - } - - _attribute_map = { - "operation": {"key": "operation", "type": "str"}, - "monitored_subscription_list": {"key": "monitoredSubscriptionList", "type": "[MonitoredSubscription]"}, - "provisioning_state": {"key": "provisioningState", "type": "str"}, - } - - def __init__( - self, - *, - operation: Optional[Union[str, "_models.Operation"]] = None, - monitored_subscription_list: Optional[List["_models.MonitoredSubscription"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword operation: The operation for the patch on the resource. Known values are: "AddBegin", - "AddComplete", "DeleteBegin", "DeleteComplete", and "Active". - :paramtype operation: str or ~azure.mgmt.elastic.models.Operation - :keyword monitored_subscription_list: List of subscriptions and the state of the monitoring. - :paramtype monitored_subscription_list: list[~azure.mgmt.elastic.models.MonitoredSubscription] - """ - super().__init__(**kwargs) - self.operation = operation - self.monitored_subscription_list = monitored_subscription_list - self.provisioning_state = None - - -class SystemData(_serialization.Model): - """Metadata pertaining to creation and last modification of the resource. - - :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. Known values are: - "User", "Application", "ManagedIdentity", and "Key". - :vartype created_by_type: str or ~azure.mgmt.elastic.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. Known values - are: "User", "Application", "ManagedIdentity", and "Key". - :vartype last_modified_by_type: str or ~azure.mgmt.elastic.models.CreatedByType - :ivar last_modified_at: The timestamp of resource last modification (UTC). - :vartype 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, - *, - created_by: Optional[str] = None, - created_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, - created_at: Optional[datetime.datetime] = None, - last_modified_by: Optional[str] = None, - last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, - last_modified_at: Optional[datetime.datetime] = None, - **kwargs: Any - ) -> None: - """ - :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. Known values are: - "User", "Application", "ManagedIdentity", and "Key". - :paramtype created_by_type: str or ~azure.mgmt.elastic.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. Known - values are: "User", "Application", "ManagedIdentity", and "Key". - :paramtype last_modified_by_type: str or ~azure.mgmt.elastic.models.CreatedByType - :keyword last_modified_at: The timestamp of resource last modification (UTC). - :paramtype last_modified_at: ~datetime.datetime - """ - super().__init__(**kwargs) - self.created_by = created_by - self.created_by_type = created_by_type - self.created_at = created_at - self.last_modified_by = last_modified_by - self.last_modified_by_type = last_modified_by_type - self.last_modified_at = last_modified_at - - -class UpgradableVersionsList(_serialization.Model): - """Stack Versions that this version can upgrade to. - - :ivar current_version: Current version of the elastic monitor. - :vartype current_version: str - :ivar upgradable_versions: Stack Versions that this version can upgrade to. - :vartype upgradable_versions: list[str] - """ - - _attribute_map = { - "current_version": {"key": "currentVersion", "type": "str"}, - "upgradable_versions": {"key": "upgradableVersions", "type": "[str]"}, - } - - def __init__( - self, *, current_version: Optional[str] = None, upgradable_versions: Optional[List[str]] = None, **kwargs: Any - ) -> None: - """ - :keyword current_version: Current version of the elastic monitor. - :paramtype current_version: str - :keyword upgradable_versions: Stack Versions that this version can upgrade to. - :paramtype upgradable_versions: list[str] - """ - super().__init__(**kwargs) - self.current_version = current_version - self.upgradable_versions = upgradable_versions - - -class UserApiKeyResponse(_serialization.Model): - """The User Api Key created for the Organization associated with the User Email Id that was passed - in the request. - - :ivar properties: - :vartype properties: ~azure.mgmt.elastic.models.UserApiKeyResponseProperties - """ - - _attribute_map = { - "properties": {"key": "properties", "type": "UserApiKeyResponseProperties"}, - } - - def __init__(self, *, properties: Optional["_models.UserApiKeyResponseProperties"] = None, **kwargs: Any) -> None: - """ - :keyword properties: - :paramtype properties: ~azure.mgmt.elastic.models.UserApiKeyResponseProperties - """ - super().__init__(**kwargs) - self.properties = properties - - -class UserApiKeyResponseProperties(_serialization.Model): - """UserApiKeyResponseProperties. - - :ivar api_key: The User Api Key Generated based on GenerateApiKey flag. This is applicable for - non-Portal clients only. - :vartype api_key: str - """ - - _attribute_map = { - "api_key": {"key": "apiKey", "type": "str"}, - } - - def __init__(self, *, api_key: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword api_key: The User Api Key Generated based on GenerateApiKey flag. This is applicable - for non-Portal clients only. - :paramtype api_key: str - """ - super().__init__(**kwargs) - self.api_key = api_key - - -class UserEmailId(_serialization.Model): - """Email Id of the User Organization, of which the API Key must be returned. - - :ivar email_id: The User email Id. - :vartype email_id: str - """ - - _attribute_map = { - "email_id": {"key": "emailId", "type": "str"}, - } - - def __init__(self, *, email_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword email_id: The User email Id. - :paramtype email_id: str - """ - super().__init__(**kwargs) - self.email_id = email_id - - -class UserInfo(_serialization.Model): - """User Information to be passed to partners. - - :ivar first_name: First name of the user. - :vartype first_name: str - :ivar last_name: Last name of the user. - :vartype last_name: str - :ivar company_name: Company name of the user. - :vartype company_name: str - :ivar email_address: Email of the user used by Elastic for contacting them if needed. - :vartype email_address: str - :ivar company_info: Company information of the user to be passed to partners. - :vartype company_info: ~azure.mgmt.elastic.models.CompanyInfo - """ - - _validation = { - "first_name": {"max_length": 50}, - "last_name": {"max_length": 50}, - "company_name": {"max_length": 64}, - "email_address": { - "pattern": r'^([^<>()\[\]\.,;:\s@"]+(\.[^<>()\[\]\.,;:\s@"]+)*)@(([a-zA-Z-_0-9]+\.)+[a-zA-Z]{2,})$' - }, - } - - _attribute_map = { - "first_name": {"key": "firstName", "type": "str"}, - "last_name": {"key": "lastName", "type": "str"}, - "company_name": {"key": "companyName", "type": "str"}, - "email_address": {"key": "emailAddress", "type": "str"}, - "company_info": {"key": "companyInfo", "type": "CompanyInfo"}, - } - - def __init__( - self, - *, - first_name: Optional[str] = None, - last_name: Optional[str] = None, - company_name: Optional[str] = None, - email_address: Optional[str] = None, - company_info: Optional["_models.CompanyInfo"] = None, - **kwargs: Any - ) -> None: - """ - :keyword first_name: First name of the user. - :paramtype first_name: str - :keyword last_name: Last name of the user. - :paramtype last_name: str - :keyword company_name: Company name of the user. - :paramtype company_name: str - :keyword email_address: Email of the user used by Elastic for contacting them if needed. - :paramtype email_address: str - :keyword company_info: Company information of the user to be passed to partners. - :paramtype company_info: ~azure.mgmt.elastic.models.CompanyInfo - """ - super().__init__(**kwargs) - self.first_name = first_name - self.last_name = last_name - self.company_name = company_name - self.email_address = email_address - self.company_info = company_info - - -class VMCollectionUpdate(_serialization.Model): - """Update VM resource collection. - - :ivar vm_resource_id: ARM id of the VM resource. - :vartype vm_resource_id: str - :ivar operation_name: Operation to be performed for given VM. Known values are: "Add" and - "Delete". - :vartype operation_name: str or ~azure.mgmt.elastic.models.OperationName - """ - - _attribute_map = { - "vm_resource_id": {"key": "vmResourceId", "type": "str"}, - "operation_name": {"key": "operationName", "type": "str"}, - } - - def __init__( - self, - *, - vm_resource_id: Optional[str] = None, - operation_name: Optional[Union[str, "_models.OperationName"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword vm_resource_id: ARM id of the VM resource. - :paramtype vm_resource_id: str - :keyword operation_name: Operation to be performed for given VM. Known values are: "Add" and - "Delete". - :paramtype operation_name: str or ~azure.mgmt.elastic.models.OperationName - """ - super().__init__(**kwargs) - self.vm_resource_id = vm_resource_id - self.operation_name = operation_name - - -class VMHostListResponse(_serialization.Model): - """Response of a list operation. - - :ivar value: Results of a list operation. - :vartype value: list[~azure.mgmt.elastic.models.VMResources] - :ivar next_link: Link to the next Vm resource Id, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[VMResources]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, *, value: Optional[List["_models.VMResources"]] = None, next_link: Optional[str] = None, **kwargs: Any - ) -> None: - """ - :keyword value: Results of a list operation. - :paramtype value: list[~azure.mgmt.elastic.models.VMResources] - :keyword next_link: Link to the next Vm resource Id, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class VMIngestionDetailsResponse(_serialization.Model): - """The vm ingestion details to install an agent. - - :ivar cloud_id: The cloudId of given Elastic monitor resource. - :vartype cloud_id: str - :ivar ingestion_key: Ingestion details to install agent on given VM. - :vartype ingestion_key: str - """ - - _attribute_map = { - "cloud_id": {"key": "cloudId", "type": "str"}, - "ingestion_key": {"key": "ingestionKey", "type": "str"}, - } - - def __init__(self, *, cloud_id: Optional[str] = None, ingestion_key: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword cloud_id: The cloudId of given Elastic monitor resource. - :paramtype cloud_id: str - :keyword ingestion_key: Ingestion details to install agent on given VM. - :paramtype ingestion_key: str - """ - super().__init__(**kwargs) - self.cloud_id = cloud_id - self.ingestion_key = ingestion_key - - -class VMResources(_serialization.Model): - """The vm resource properties that is currently being monitored by the Elastic monitor resource. - - :ivar vm_resource_id: The ARM id of the VM resource. - :vartype vm_resource_id: str - """ - - _attribute_map = { - "vm_resource_id": {"key": "vmResourceId", "type": "str"}, - } - - def __init__(self, *, vm_resource_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword vm_resource_id: The ARM id of the VM resource. - :paramtype vm_resource_id: str - """ - super().__init__(**kwargs) - self.vm_resource_id = vm_resource_id diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_patch.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_patch.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/models/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/__init__.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/__init__.py index 1cdfeb9a7b42..7754ab6a86e0 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/__init__.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/__init__.py @@ -2,66 +2,36 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +# pylint: disable=wrong-import-position -from ._operations import Operations -from ._monitors_operations import MonitorsOperations -from ._elastic_versions_operations import ElasticVersionsOperations -from ._monitored_subscriptions_operations import MonitoredSubscriptionsOperations -from ._monitored_resources_operations import MonitoredResourcesOperations -from ._deployment_info_operations import DeploymentInfoOperations -from ._external_user_operations import ExternalUserOperations -from ._billing_info_operations import BillingInfoOperations -from ._connected_partner_resources_operations import ConnectedPartnerResourcesOperations -from ._open_ai_operations import OpenAIOperations -from ._tag_rules_operations import TagRulesOperations -from ._vm_host_operations import VMHostOperations -from ._vm_ingestion_operations import VMIngestionOperations -from ._vm_collection_operations import VMCollectionOperations -from ._upgradable_versions_operations import UpgradableVersionsOperations -from ._monitor_operations import MonitorOperations -from ._all_traffic_filters_operations import AllTrafficFiltersOperations -from ._list_associated_traffic_filters_operations import ListAssociatedTrafficFiltersOperations -from ._create_and_associate_ip_filter_operations import CreateAndAssociateIPFilterOperations -from ._create_and_associate_pl_filter_operations import CreateAndAssociatePLFilterOperations -from ._associate_traffic_filter_operations import AssociateTrafficFilterOperations -from ._detach_and_delete_traffic_filter_operations import DetachAndDeleteTrafficFilterOperations -from ._detach_traffic_filter_operations import DetachTrafficFilterOperations -from ._traffic_filters_operations import TrafficFiltersOperations -from ._organizations_operations import OrganizationsOperations +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from ._patch import * # pylint: disable=unused-wildcard-import + +from ._operations import Operations # type: ignore +from ._operations import ElasticMonitorResourcesOperations # type: ignore +from ._operations import MonitoredSubscriptionsOperations # type: ignore +from ._operations import OpenAIIntegrationRPModelsOperations # type: ignore +from ._operations import TagRulesOperations # type: ignore +from ._operations import ElasticVersionsOperationGroupOperations # type: ignore +from ._operations import OrganizationsOperationGroupOperations # type: ignore from ._patch import __all__ as _patch_all -from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import * from ._patch import patch_sdk as _patch_sdk __all__ = [ "Operations", - "MonitorsOperations", - "ElasticVersionsOperations", + "ElasticMonitorResourcesOperations", "MonitoredSubscriptionsOperations", - "MonitoredResourcesOperations", - "DeploymentInfoOperations", - "ExternalUserOperations", - "BillingInfoOperations", - "ConnectedPartnerResourcesOperations", - "OpenAIOperations", + "OpenAIIntegrationRPModelsOperations", "TagRulesOperations", - "VMHostOperations", - "VMIngestionOperations", - "VMCollectionOperations", - "UpgradableVersionsOperations", - "MonitorOperations", - "AllTrafficFiltersOperations", - "ListAssociatedTrafficFiltersOperations", - "CreateAndAssociateIPFilterOperations", - "CreateAndAssociatePLFilterOperations", - "AssociateTrafficFilterOperations", - "DetachAndDeleteTrafficFilterOperations", - "DetachTrafficFilterOperations", - "TrafficFiltersOperations", - "OrganizationsOperations", + "ElasticVersionsOperationGroupOperations", + "OrganizationsOperationGroupOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_all_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_all_traffic_filters_operations.py deleted file mode 100644 index 247e471b0046..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_all_traffic_filters_operations.py +++ /dev/null @@ -1,148 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listAllTrafficFilters", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class AllTrafficFiltersOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`all_traffic_filters` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.ElasticTrafficFilterResponse: - """Get the list of all traffic filters for the account. - - Get the list of all traffic filters for the account. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: ElasticTrafficFilterResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ElasticTrafficFilterResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_associate_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_associate_traffic_filter_operations.py deleted file mode 100644 index 5e55c23410e5..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_associate_traffic_filter_operations.py +++ /dev/null @@ -1,217 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterator, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_associate_request( - resource_group_name: str, - monitor_name: str, - subscription_id: str, - *, - ruleset_id: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/associateTrafficFilter", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - if ruleset_id is not None: - _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class AssociateTrafficFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`associate_traffic_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - def _associate_initial( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_associate_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_associate( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> LROPoller[None]: - """Associate traffic filter for the given deployment. - - Associate traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._associate_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - ruleset_id=ruleset_id, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_billing_info_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_billing_info_operations.py deleted file mode 100644 index c9edfaf26c52..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_billing_info_operations.py +++ /dev/null @@ -1,148 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/getBillingInfo", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class BillingInfoOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`billing_info` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def get(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.BillingInfoResponse: - """Get marketplace and organization info mapped to the given monitor. - - Get marketplace and organization info mapped to the given monitor. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: BillingInfoResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.BillingInfoResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.BillingInfoResponse] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("BillingInfoResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_connected_partner_resources_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_connected_partner_resources_operations.py deleted file mode 100644 index 09c93caa2074..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_connected_partner_resources_operations.py +++ /dev/null @@ -1,182 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterable, Optional, Type, TypeVar -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listConnectedPartnerResources", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class ConnectedPartnerResourcesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`connected_partner_resources` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> Iterable["_models.ConnectedPartnerResourcesListFormat"]: - """List of all active deployments that are associated with the marketplace subscription linked to - the given monitor. - - List of all active deployments that are associated with the marketplace subscription linked to - the given monitor. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either ConnectedPartnerResourcesListFormat or the result - of cls(response) - :rtype: - ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ConnectedPartnerResourcesListFormat] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ConnectedPartnerResourcesListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("ConnectedPartnerResourcesListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_ip_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_ip_filter_operations.py deleted file mode 100644 index fc1856b098c6..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_ip_filter_operations.py +++ /dev/null @@ -1,234 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterator, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_create_request( - resource_group_name: str, - monitor_name: str, - subscription_id: str, - *, - ips: Optional[str] = None, - name: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/createAndAssociateIPFilter", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - if ips is not None: - _params["ips"] = _SERIALIZER.query("ips", ips, "str") - if name is not None: - _params["name"] = _SERIALIZER.query("name", name, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class CreateAndAssociateIPFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`create_and_associate_ip_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - def _create_initial( - self, - resource_group_name: str, - monitor_name: str, - ips: Optional[str] = None, - name: Optional[str] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_create_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ips=ips, - name=name, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_create( - self, - resource_group_name: str, - monitor_name: str, - ips: Optional[str] = None, - name: Optional[str] = None, - **kwargs: Any - ) -> LROPoller[None]: - """Create and Associate IP traffic filter for the given deployment. - - Create and Associate IP traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ips: List of ips. Default value is None. - :type ips: str - :param name: Name of the traffic filter. Default value is None. - :type name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._create_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - ips=ips, - name=name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_pl_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_pl_filter_operations.py deleted file mode 100644 index 62d8bb2a09d7..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_create_and_associate_pl_filter_operations.py +++ /dev/null @@ -1,243 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterator, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_create_request( - resource_group_name: str, - monitor_name: str, - subscription_id: str, - *, - name: Optional[str] = None, - private_endpoint_guid: Optional[str] = None, - private_endpoint_name: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/createAndAssociatePLFilter", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - if name is not None: - _params["name"] = _SERIALIZER.query("name", name, "str") - if private_endpoint_guid is not None: - _params["privateEndpointGuid"] = _SERIALIZER.query("private_endpoint_guid", private_endpoint_guid, "str") - if private_endpoint_name is not None: - _params["privateEndpointName"] = _SERIALIZER.query("private_endpoint_name", private_endpoint_name, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class CreateAndAssociatePLFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`create_and_associate_pl_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - def _create_initial( - self, - resource_group_name: str, - monitor_name: str, - name: Optional[str] = None, - private_endpoint_guid: Optional[str] = None, - private_endpoint_name: Optional[str] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_create_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - name=name, - private_endpoint_guid=private_endpoint_guid, - private_endpoint_name=private_endpoint_name, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_create( - self, - resource_group_name: str, - monitor_name: str, - name: Optional[str] = None, - private_endpoint_guid: Optional[str] = None, - private_endpoint_name: Optional[str] = None, - **kwargs: Any - ) -> LROPoller[None]: - """Create and Associate private link traffic filter for the given deployment. - - Create and Associate private link traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param name: Name of the traffic filter. Default value is None. - :type name: str - :param private_endpoint_guid: Guid of the private endpoint. Default value is None. - :type private_endpoint_guid: str - :param private_endpoint_name: Name of the private endpoint. Default value is None. - :type private_endpoint_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._create_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - name=name, - private_endpoint_guid=private_endpoint_guid, - private_endpoint_name=private_endpoint_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_deployment_info_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_deployment_info_operations.py deleted file mode 100644 index 02d51ddb261d..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_deployment_info_operations.py +++ /dev/null @@ -1,150 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listDeploymentInfo", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class DeploymentInfoOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`deployment_info` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.DeploymentInfoResponse: - """Fetch information regarding Elastic cloud deployment corresponding to the Elastic monitor - resource. - - Fetch information regarding Elastic cloud deployment corresponding to the Elastic monitor - resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: DeploymentInfoResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.DeploymentInfoResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.DeploymentInfoResponse] = kwargs.pop("cls", None) - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("DeploymentInfoResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_and_delete_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_and_delete_traffic_filter_operations.py deleted file mode 100644 index b9725a5a5916..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_and_delete_traffic_filter_operations.py +++ /dev/null @@ -1,158 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_delete_request( - resource_group_name: str, - monitor_name: str, - subscription_id: str, - *, - ruleset_id: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/detachAndDeleteTrafficFilter", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - if ruleset_id is not None: - _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class DetachAndDeleteTrafficFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`detach_and_delete_traffic_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> None: - """Detach and Delete traffic filter from the given deployment. - - Detach and Delete traffic filter from the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_traffic_filter_operations.py deleted file mode 100644 index 8f24ed5e6b7d..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_detach_traffic_filter_operations.py +++ /dev/null @@ -1,217 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterator, Optional, Type, TypeVar, Union, cast - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_update_request( - resource_group_name: str, - monitor_name: str, - subscription_id: str, - *, - ruleset_id: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/detachTrafficFilter", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - if ruleset_id is not None: - _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class DetachTrafficFilterOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`detach_traffic_filter` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - def _update_initial( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_update( - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> LROPoller[None]: - """Detach traffic filter for the given deployment. - - Detach traffic filter for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - ruleset_id=ruleset_id, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_elastic_versions_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_elastic_versions_operations.py deleted file mode 100644 index 24efb5446433..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_elastic_versions_operations.py +++ /dev/null @@ -1,167 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterable, Optional, Type, TypeVar -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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, *, region: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/elasticVersions") - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - _params["region"] = _SERIALIZER.query("region", region, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class ElasticVersionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`elastic_versions` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, region: str, **kwargs: Any) -> Iterable["_models.ElasticVersionListFormat"]: - """Get a list of available versions for a region. - - Get a list of available versions for a region. - - :param region: Region where elastic deployment will take place. Required. - :type region: str - :return: An iterator like instance of either ElasticVersionListFormat or the result of - cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ElasticVersionListFormat] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticVersionsListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - subscription_id=self._config.subscription_id, - region=region, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("ElasticVersionsListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_external_user_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_external_user_operations.py deleted file mode 100644 index e742701dd182..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_external_user_operations.py +++ /dev/null @@ -1,242 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Optional, Type, TypeVar, Union, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_create_or_update_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/createOrUpdateExternalUser", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class ExternalUserOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`external_user` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ExternalUserInfo] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.ExternalUserCreationResponse: - """Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic External User Creation Parameters. Default value is None. - :type body: ~azure.mgmt.elastic.models.ExternalUserInfo - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: ExternalUserCreationResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.ExternalUserCreationResponse: - """Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic External User Creation Parameters. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: ExternalUserCreationResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ExternalUserInfo, IO[bytes]]] = None, - **kwargs: Any - ) -> _models.ExternalUserCreationResponse: - """Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - Create User inside elastic deployment which are used by customers to perform operations on the - elastic deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic External User Creation Parameters. Is either a ExternalUserInfo type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ExternalUserInfo or IO[bytes] - :return: ExternalUserCreationResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ExternalUserCreationResponse] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ExternalUserInfo") - else: - _json = None - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ExternalUserCreationResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_list_associated_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_list_associated_traffic_filters_operations.py deleted file mode 100644 index 93cec5f78835..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_list_associated_traffic_filters_operations.py +++ /dev/null @@ -1,148 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listAssociatedTrafficFilters", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class ListAssociatedTrafficFiltersOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`list_associated_traffic_filters` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.ElasticTrafficFilterResponse: - """Get the list of all associated traffic filters for the given deployment. - - Get the list of all associated traffic filters for the given deployment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: ElasticTrafficFilterResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ElasticTrafficFilterResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitor_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitor_operations.py deleted file mode 100644 index f075ef028243..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitor_operations.py +++ /dev/null @@ -1,295 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Iterator, Optional, Type, TypeVar, Union, cast, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_upgrade_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/upgrade", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class MonitorOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`monitor` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - def _upgrade_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorUpgrade, IO[bytes]]] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ElasticMonitorUpgrade") - else: - _json = None - - _request = build_upgrade_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_upgrade( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ElasticMonitorUpgrade] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[None]: - """Upgradable version for a monitor resource. - - Upgradable version for a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic Monitor Upgrade Parameters. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :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: - """ - - @overload - def begin_upgrade( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[None]: - """Upgradable version for a monitor resource. - - Upgradable version for a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic Monitor Upgrade Parameters. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :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: - """ - - @distributed_trace - def begin_upgrade( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorUpgrade, IO[bytes]]] = None, - **kwargs: Any - ) -> LROPoller[None]: - """Upgradable version for a monitor resource. - - Upgradable version for a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic Monitor Upgrade Parameters. Is either a ElasticMonitorUpgrade type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade or IO[bytes] - :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: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._upgrade_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_resources_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_resources_operations.py deleted file mode 100644 index d63d85cc8f32..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_resources_operations.py +++ /dev/null @@ -1,176 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterable, Optional, Type, TypeVar -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listMonitoredResources", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class MonitoredResourcesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`monitored_resources` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> Iterable["_models.MonitoredResource"]: - """List the resources currently being monitored by the Elastic monitor resource. - - List the resources currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either MonitoredResource or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.MonitoredResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoredResourceListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("MonitoredResourceListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_subscriptions_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_subscriptions_operations.py deleted file mode 100644 index 69d5312700d4..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitored_subscriptions_operations.py +++ /dev/null @@ -1,942 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, Type, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_createor_update_request( - resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_update_request( - resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_request( - resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -class MonitoredSubscriptionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`monitored_subscriptions` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> Iterable["_models.MonitoredSubscriptionProperties"]: - """List all the subscriptions currently being monitored by the Elastic monitor resource. - - List all the subscriptions currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either MonitoredSubscriptionProperties or the result of - cls(response) - :rtype: - ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoredSubscriptionPropertiesList] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("MonitoredSubscriptionPropertiesList", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @distributed_trace - def get( - self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any - ) -> _models.MonitoredSubscriptionProperties: - """List all the subscriptions currently being monitored by the Elastic monitor resource. - - List all the subscriptions currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :return: MonitoredSubscriptionProperties or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("MonitoredSubscriptionProperties", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - def _createor_update_initial( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "MonitoredSubscriptionProperties") - else: - _json = None - - _request = build_createor_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 201]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_createor_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[_models.MonitoredSubscriptionProperties] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.MonitoredSubscriptionProperties]: - """Add the subscriptions that should be monitored by the Elastic monitor resource. - - Add the subscriptions that should be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either MonitoredSubscriptionProperties or the - result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_createor_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.MonitoredSubscriptionProperties]: - """Add the subscriptions that should be monitored by the Elastic monitor resource. - - Add the subscriptions that should be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either MonitoredSubscriptionProperties or the - result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_createor_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> LROPoller[_models.MonitoredSubscriptionProperties]: - """Add the subscriptions that should be monitored by the Elastic monitor resource. - - Add the subscriptions that should be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Is either a MonitoredSubscriptionProperties type or a IO[bytes] type. Default - value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or IO[bytes] - :return: An instance of LROPoller that returns either MonitoredSubscriptionProperties or the - result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._createor_update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("MonitoredSubscriptionProperties", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.MonitoredSubscriptionProperties].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.MonitoredSubscriptionProperties]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - def _update_initial( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "MonitoredSubscriptionProperties") - else: - _json = None - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[_models.MonitoredSubscriptionProperties] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.MonitoredSubscriptionProperties]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either MonitoredSubscriptionProperties or the - result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.MonitoredSubscriptionProperties]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either MonitoredSubscriptionProperties or the - result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_update( - self, - resource_group_name: str, - monitor_name: str, - configuration_name: str, - body: Optional[Union[_models.MonitoredSubscriptionProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> LROPoller[_models.MonitoredSubscriptionProperties]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :param body: Is either a MonitoredSubscriptionProperties type or a IO[bytes] type. Default - value is None. - :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or IO[bytes] - :return: An instance of LROPoller that returns either MonitoredSubscriptionProperties or the - result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("MonitoredSubscriptionProperties", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.MonitoredSubscriptionProperties].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.MonitoredSubscriptionProperties]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - def _delete_initial( - self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202, 204]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_delete( - self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any - ) -> LROPoller[None]: - """Updates the subscriptions that are being monitored by the Elastic monitor resource. - - Updates the subscriptions that are being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param configuration_name: The configuration name. Only 'default' value is supported. Required. - :type configuration_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._delete_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - configuration_name=configuration_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitors_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitors_operations.py deleted file mode 100644 index 06572eb6d65e..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_monitors_operations.py +++ /dev/null @@ -1,1000 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, Type, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/monitors") - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_list_by_resource_group_request(resource_group_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_create_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_update_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -class MonitorsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`monitors` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, **kwargs: Any) -> Iterable["_models.ElasticMonitorResource"]: - """List all monitors under the specified subscription. - - List all monitors under the specified subscription. - - :return: An iterator like instance of either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticMonitorResourceListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResourceListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @distributed_trace - def list_by_resource_group( - self, resource_group_name: str, **kwargs: Any - ) -> Iterable["_models.ElasticMonitorResource"]: - """List all monitors under the specified resource group. - - List all monitors under the specified resource group. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :return: An iterator like instance of either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticMonitorResourceListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_by_resource_group_request( - resource_group_name=resource_group_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResourceListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @distributed_trace - def get(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.ElasticMonitorResource: - """Get the properties of a specific monitor resource. - - Get the properties of a specific monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: ElasticMonitorResource or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticMonitorResource - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - def _create_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResource, IO[bytes]]] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ElasticMonitorResource") - else: - _json = None - - _request = build_create_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 201]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_create( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ElasticMonitorResource] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Create a monitor resource. - - Create a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic monitor resource model. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_create( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Create a monitor resource. - - Create a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic monitor resource model. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_create( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResource, IO[bytes]]] = None, - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Create a monitor resource. - - Create a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic monitor resource model. Is either a ElasticMonitorResource type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource or IO[bytes] - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._create_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.ElasticMonitorResource].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.ElasticMonitorResource]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - def _update_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResourceUpdateParameters, IO[bytes]]] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ElasticMonitorResourceUpdateParameters") - else: - _json = None - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ElasticMonitorResourceUpdateParameters] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Update a monitor resource. - - Update a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic resource model update parameters. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Update a monitor resource. - - Update a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic resource model update parameters. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_update( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ElasticMonitorResourceUpdateParameters, IO[bytes]]] = None, - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Update a monitor resource. - - Update a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Elastic resource model update parameters. Is either a - ElasticMonitorResourceUpdateParameters type or a IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters or IO[bytes] - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._update_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.ElasticMonitorResource].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.ElasticMonitorResource]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - def _delete_initial(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202, 204]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_delete(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> LROPoller[None]: - """Delete a monitor resource. - - Delete a monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._delete_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_open_ai_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_open_ai_operations.py deleted file mode 100644 index d882b6b5b1f3..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_open_ai_operations.py +++ /dev/null @@ -1,662 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Iterable, Optional, Type, TypeVar, Union, overload -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "integrationName": _SERIALIZER.url("integration_name", integration_name, "str", pattern=r"^[a-z][a-z0-9]*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_create_or_update_request( - resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "integrationName": _SERIALIZER.url("integration_name", integration_name, "str", pattern=r"^[a-z][a-z0-9]*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_request( - resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "integrationName": _SERIALIZER.url("integration_name", integration_name, "str", pattern=r"^[a-z][a-z0-9]*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_status_request( - resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}/getStatus", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "integrationName": _SERIALIZER.url("integration_name", integration_name, "str", pattern=r"^[a-z][a-z0-9]*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class OpenAIOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`open_ai` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> Iterable["_models.OpenAIIntegrationRPModel"]: - """List OpenAI integration rule for a given monitor resource. - - List OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either OpenAIIntegrationRPModel or the result of - cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.OpenAIIntegrationRPModel] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OpenAIIntegrationRPModelListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("OpenAIIntegrationRPModelListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @distributed_trace - def get( - self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Get OpenAI integration rule for a given monitor resource. - - Get OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("OpenAIIntegrationRPModel", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - integration_name: str, - body: Optional[_models.OpenAIIntegrationRPModel] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Create or update a OpenAI integration rule for a given monitor resource. - - Create or update a OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :param body: Default value is None. - :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - integration_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Create or update a OpenAI integration rule for a given monitor resource. - - Create or update a OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :param body: Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - integration_name: str, - body: Optional[Union[_models.OpenAIIntegrationRPModel, IO[bytes]]] = None, - **kwargs: Any - ) -> _models.OpenAIIntegrationRPModel: - """Create or update a OpenAI integration rule for a given monitor resource. - - Create or update a OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :param body: Is either a OpenAIIntegrationRPModel type or a IO[bytes] type. Default value is - None. - :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel or IO[bytes] - :return: OpenAIIntegrationRPModel or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "OpenAIIntegrationRPModel") - else: - _json = None - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("OpenAIIntegrationRPModel", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any - ) -> None: - """Delete OpenAI integration rule for a given monitor resource. - - Delete OpenAI integration rule for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - @distributed_trace - def get_status( - self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any - ) -> _models.OpenAIIntegrationStatusResponse: - """Get OpenAI integration status for a given integration. - - Get OpenAI integration status for a given integration. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param integration_name: OpenAI Integration name. Required. - :type integration_name: str - :return: OpenAIIntegrationStatusResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationStatusResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OpenAIIntegrationStatusResponse] = kwargs.pop("cls", None) - - _request = build_get_status_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - integration_name=integration_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("OpenAIIntegrationStatusResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_operations.py index 5811170a74f9..c2ff622e2d75 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_operations.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_operations.py @@ -1,53 +1,59 @@ -# pylint: disable=too-many-lines,too-many-statements +# pylint: disable=line-too-long,useless-suppression,too-many-lines # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -import sys -from typing import Any, Callable, Dict, Iterable, Optional, Type, TypeVar +from collections.abc import MutableMapping +from io import IOBase +import json +from typing import Any, Callable, Dict, IO, Iterable, Iterator, List, Optional, TypeVar, Union, cast, overload import urllib.parse +from azure.core import PipelineClient from azure.core.exceptions import ( ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse +from azure.core.polling import LROPoller, NoPolling, PollingMethod from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat +from azure.mgmt.core.polling.arm_polling import ARMPolling from .. import models as _models -from .._serialization import Serializer +from .._configuration import ElasticClientConfiguration +from .._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize +from .._utils.serialization import Deserializer, Serializer -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +JSON = MutableMapping[str, Any] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_list_request(**kwargs: Any) -> HttpRequest: +def build_operations_list_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop("template_url", "/providers/Microsoft.Elastic/operations") + _url = "/providers/Microsoft.Elastic/operations" # Construct parameters _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") @@ -58,42 +64,6034 @@ def build_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) +def build_elastic_monitor_resources_get_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_create_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_update_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_delete_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_list_by_resource_group_request( # pylint: disable=name-too-long + resource_group_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_list_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/monitors" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_monitored_resources_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listMonitoredResources" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_deployment_info_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listDeploymentInfo" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/createOrUpdateExternalUser" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_billing_info_get_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/getBillingInfo" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_connected_partner_resources_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listConnectedPartnerResources" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_v_m_host_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listVMHost" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_details_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/vmIngestionDetails" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_v_m_collection_update_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/vmCollectionUpdate" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_upgradable_versions_details_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listUpgradableVersions" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_upgrade_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/upgrade" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_all_traffic_filters_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listAllTrafficFilters" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_list_associated_traffic_filters_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listAssociatedTrafficFilters" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_create_and_associate_ip_filter_create_request( # pylint: disable=name-too-long + resource_group_name: str, + monitor_name: str, + subscription_id: str, + *, + ips: Optional[str] = None, + name: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/createAndAssociateIPFilter" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if ips is not None: + _params["ips"] = _SERIALIZER.query("ips", ips, "str") + if name is not None: + _params["name"] = _SERIALIZER.query("name", name, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_create_and_associate_pl_filter_create_request( # pylint: disable=name-too-long + resource_group_name: str, + monitor_name: str, + subscription_id: str, + *, + name: Optional[str] = None, + private_endpoint_guid: Optional[str] = None, + private_endpoint_name: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/createAndAssociatePLFilter" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if name is not None: + _params["name"] = _SERIALIZER.query("name", name, "str") + if private_endpoint_guid is not None: + _params["privateEndpointGuid"] = _SERIALIZER.query("private_endpoint_guid", private_endpoint_guid, "str") + if private_endpoint_name is not None: + _params["privateEndpointName"] = _SERIALIZER.query("private_endpoint_name", private_endpoint_name, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_associate_request( # pylint: disable=name-too-long + resource_group_name: str, + monitor_name: str, + subscription_id: str, + *, + ruleset_id: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/associateTrafficFilter" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if ruleset_id is not None: + _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_detach_and_delete_traffic_filter_delete_request( # pylint: disable=name-too-long + resource_group_name: str, + monitor_name: str, + subscription_id: str, + *, + ruleset_id: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/detachAndDeleteTrafficFilter" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if ruleset_id is not None: + _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_detach_traffic_filter_update_request( # pylint: disable=name-too-long + resource_group_name: str, + monitor_name: str, + subscription_id: str, + *, + ruleset_id: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/detachTrafficFilter" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if ruleset_id is not None: + _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_traffic_filters_delete_request( # pylint: disable=name-too-long + resource_group_name: str, + monitor_name: str, + subscription_id: str, + *, + ruleset_id: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/deleteTrafficFilter" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if ruleset_id is not None: + _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_monitor_resources_resubscribe_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/resubscribe" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_monitored_subscriptions_get_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_monitored_subscriptions_createor_update_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_monitored_subscriptions_update_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_monitored_subscriptions_delete_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, configuration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions/{configurationName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "configurationName": _SERIALIZER.url("configuration_name", configuration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_monitored_subscriptions_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/monitoredSubscriptions" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_open_ai_integration_rp_models_get_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "integrationName": _SERIALIZER.url("integration_name", integration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_open_ai_integration_rp_models_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "integrationName": _SERIALIZER.url("integration_name", integration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_open_ai_integration_rp_models_delete_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "integrationName": _SERIALIZER.url("integration_name", integration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_open_ai_integration_rp_models_list_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_open_ai_integration_rp_models_get_status_request( # pylint: disable=name-too-long + resource_group_name: str, monitor_name: str, integration_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/openAIIntegrations/{integrationName}/getStatus" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "integrationName": _SERIALIZER.url("integration_name", integration_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_tag_rules_get_request( + resource_group_name: str, monitor_name: str, rule_set_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules/{ruleSetName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "ruleSetName": _SERIALIZER.url("rule_set_name", rule_set_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_tag_rules_create_or_update_request( + resource_group_name: str, monitor_name: str, rule_set_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules/{ruleSetName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "ruleSetName": _SERIALIZER.url("rule_set_name", rule_set_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_tag_rules_delete_request( + resource_group_name: str, monitor_name: str, rule_set_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules/{ruleSetName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + "ruleSetName": _SERIALIZER.url("rule_set_name", rule_set_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_tag_rules_list_request( + resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_elastic_versions_operation_group_list_request( # pylint: disable=name-too-long + subscription_id: str, *, region: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/elasticVersions" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params["region"] = _SERIALIZER.query("region", region, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_organizations_operation_group_get_api_key_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/getOrganizationApiKey" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_organizations_operation_group_get_elastic_to_azure_subscription_mapping_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-01-15-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = ( + "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/getElasticOrganizationToAzureSubscriptionMapping" + ) + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + class Operations: """ .. warning:: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`operations` attribute. + :class:`~azure.mgmt.elastic.ElasticClient`'s + :attr:`operations` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def list(self, **kwargs: Any) -> Iterable["_models.OperationResult"]: + """List the operations for the provider. + + :return: An iterator like instance of OperationResult + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.OperationResult] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.OperationResult]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_operations_list_request( + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.OperationResult], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class ElasticMonitorResourcesOperations: # pylint: disable=too-many-public-methods + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.ElasticClient`'s + :attr:`elastic_monitor_resources` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.ElasticMonitorResource: + """Get the properties of a specific monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: ElasticMonitorResource. The ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticMonitorResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticMonitorResource, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _create_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResource, JSON, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_elastic_monitor_resources_create_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: _models.ElasticMonitorResource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResource, JSON, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.ElasticMonitorResource]: + """Create a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic monitor resource model. Is one of the following types: + ElasticMonitorResource, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResource or JSON or IO[bytes] + :return: An instance of LROPoller that returns ElasticMonitorResource. The + ElasticMonitorResource is compatible with MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._create_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response = pipeline_response.http_response + deserialized = _deserialize(_models.ElasticMonitorResource, response.json()) + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[_models.ElasticMonitorResource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.ElasticMonitorResource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _update_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResourceUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_elastic_monitor_resources_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: _models.ElasticMonitorResourceUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + body: Union[_models.ElasticMonitorResourceUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> LROPoller[None]: + """Update a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic resource model update parameters. Is one of the following types: + ElasticMonitorResourceUpdateParameters, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorResourceUpdateParameters or JSON or + IO[bytes] + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _delete_initial(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202, 204]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_delete(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> LROPoller[None]: + """Delete a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._delete_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list_by_resource_group( + self, resource_group_name: str, **kwargs: Any + ) -> Iterable["_models.ElasticMonitorResource"]: + """List all monitors under the specified resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :return: An iterator like instance of ElasticMonitorResource + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ElasticMonitorResource]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_list_by_resource_group_request( + resource_group_name=resource_group_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.ElasticMonitorResource], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def list(self, **kwargs: Any) -> Iterable["_models.ElasticMonitorResource"]: + """List all monitors under the specified subscription. + + :return: An iterator like instance of ElasticMonitorResource + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ElasticMonitorResource]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_list_request( + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.ElasticMonitorResource], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def monitored_resources_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> Iterable["_models.MonitoredResource"]: + """List the resources currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of MonitoredResource + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.MonitoredResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.MonitoredResource]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_monitored_resources_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.MonitoredResource], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def deployment_info_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.DeploymentInfoResponse: + """Fetch information regarding Elastic cloud deployment corresponding to the Elastic monitor + resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: DeploymentInfoResponse. The DeploymentInfoResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.DeploymentInfoResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.DeploymentInfoResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_deployment_info_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.DeploymentInfoResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.ExternalUserInfo] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Default value is None. + :type body: ~azure.mgmt.elastic.models.ExternalUserInfo + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ExternalUserInfo, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> _models.ExternalUserCreationResponse: + """Create User inside elastic deployment which are used by customers to perform operations on the + elastic deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic External User Creation Parameters. Is one of the following types: + ExternalUserInfo, JSON, IO[bytes] Default value is None. + :type body: ~azure.mgmt.elastic.models.ExternalUserInfo or JSON or IO[bytes] + :return: ExternalUserCreationResponse. The ExternalUserCreationResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ExternalUserCreationResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ExternalUserCreationResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_create_or_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ExternalUserCreationResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def billing_info_get( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.BillingInfoResponse: + """Get marketplace and organization info mapped to the given monitor. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: BillingInfoResponse. The BillingInfoResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.BillingInfoResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.BillingInfoResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_billing_info_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.BillingInfoResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def connected_partner_resources_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> Iterable["_models.ConnectedPartnerResourcesListFormat"]: + """List of all active deployments that are associated with the marketplace subscription linked to + the given monitor. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of ConnectedPartnerResourcesListFormat + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ConnectedPartnerResourcesListFormat] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ConnectedPartnerResourcesListFormat]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_connected_partner_resources_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.ConnectedPartnerResourcesListFormat], deserialized.get("value", []) + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def v_m_host_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> Iterable["_models.VMResources"]: + """List the vm resources currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of VMResources + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.VMResources] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.VMResources]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_monitor_resources_v_m_host_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.VMResources], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def details(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.VMIngestionDetailsResponse: + """List the vm ingestion details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: VMIngestionDetailsResponse. The VMIngestionDetailsResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.VMIngestionDetailsResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.VMIngestionDetailsResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_details_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.VMIngestionDetailsResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def v_m_collection_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.VMCollectionUpdate] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Default value is None. + :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def v_m_collection_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def v_m_collection_update( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def v_m_collection_update( # pylint: disable=inconsistent-return-statements + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.VMCollectionUpdate, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> None: + """Update the vm details that will be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: VM resource Id. Is one of the following types: VMCollectionUpdate, JSON, IO[bytes] + Default value is None. + :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate or JSON or IO[bytes] + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_v_m_collection_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def upgradable_versions_details( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.UpgradableVersionsList: + """List of upgradable versions for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: UpgradableVersionsList. The UpgradableVersionsList is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UpgradableVersionsList + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.UpgradableVersionsList] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_upgradable_versions_details_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.UpgradableVersionsList, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _upgrade_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ElasticMonitorUpgrade, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_upgrade_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.ElasticMonitorUpgrade] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Default value is None. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_upgrade( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ElasticMonitorUpgrade, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> LROPoller[None]: + """Upgradable version for a monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Elastic Monitor Upgrade Parameters. Is one of the following types: + ElasticMonitorUpgrade, JSON, IO[bytes] Default value is None. + :type body: ~azure.mgmt.elastic.models.ElasticMonitorUpgrade or JSON or IO[bytes] + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._upgrade_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def all_traffic_filters_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.ElasticTrafficFilterResponse: + """Get the list of all traffic filters for the account. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: ElasticTrafficFilterResponse. The ElasticTrafficFilterResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_all_traffic_filters_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticTrafficFilterResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_associated_traffic_filters_list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> _models.ElasticTrafficFilterResponse: + """Get the list of all associated traffic filters for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: ElasticTrafficFilterResponse. The ElasticTrafficFilterResponse is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticTrafficFilterResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticTrafficFilterResponse] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_list_associated_traffic_filters_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticTrafficFilterResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _create_and_associate_ip_filter_create_initial( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + ips: Optional[str] = None, + name: Optional[str] = None, + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_create_and_associate_ip_filter_create_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ips=ips, + name=name, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_create_and_associate_ip_filter_create( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + ips: Optional[str] = None, + name: Optional[str] = None, + **kwargs: Any + ) -> LROPoller[_models.OkResponse]: + """Create and Associate IP traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ips: List of ips. Default value is None. + :paramtype ips: str + :keyword name: Name of the traffic filter. Default value is None. + :paramtype name: str + :return: An instance of LROPoller that returns OkResponse. The OkResponse is compatible with + MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.OkResponse] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OkResponse] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._create_and_associate_ip_filter_create_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + ips=ips, + name=name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response_headers = {} + response = pipeline_response.http_response + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = _deserialize(_models.OkResponse, response.json()) + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[_models.OkResponse].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.OkResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _create_and_associate_pl_filter_create_initial( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + name: Optional[str] = None, + private_endpoint_guid: Optional[str] = None, + private_endpoint_name: Optional[str] = None, + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_create_and_associate_pl_filter_create_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + name=name, + private_endpoint_guid=private_endpoint_guid, + private_endpoint_name=private_endpoint_name, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_create_and_associate_pl_filter_create( # pylint: disable=name-too-long + self, + resource_group_name: str, + monitor_name: str, + *, + name: Optional[str] = None, + private_endpoint_guid: Optional[str] = None, + private_endpoint_name: Optional[str] = None, + **kwargs: Any + ) -> LROPoller[_models.OkResponse]: + """Create and Associate private link traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword name: Name of the traffic filter. Default value is None. + :paramtype name: str + :keyword private_endpoint_guid: Guid of the private endpoint. Default value is None. + :paramtype private_endpoint_guid: str + :keyword private_endpoint_name: Name of the private endpoint. Default value is None. + :paramtype private_endpoint_name: str + :return: An instance of LROPoller that returns OkResponse. The OkResponse is compatible with + MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.OkResponse] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OkResponse] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._create_and_associate_pl_filter_create_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + name=name, + private_endpoint_guid=private_endpoint_guid, + private_endpoint_name=private_endpoint_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response_headers = {} + response = pipeline_response.http_response + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = _deserialize(_models.OkResponse, response.json()) + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[_models.OkResponse].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.OkResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _associate_initial( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_associate_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_associate( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> LROPoller[None]: + """Associate traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._associate_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + ruleset_id=ruleset_id, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def detach_and_delete_traffic_filter_delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> None: + """Detach and Delete traffic filter from the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_detach_and_delete_traffic_filter_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + def _detach_traffic_filter_update_initial( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_detach_traffic_filter_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_detach_traffic_filter_update( + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> LROPoller[None]: + """Detach traffic filter for the given deployment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._detach_traffic_filter_update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + ruleset_id=ruleset_id, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def traffic_filters_delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, monitor_name: str, *, ruleset_id: Optional[str] = None, **kwargs: Any + ) -> None: + """Delete traffic filter from the account. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :keyword ruleset_id: Ruleset Id of the filter. Default value is None. + :paramtype ruleset_id: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_elastic_monitor_resources_traffic_filters_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + ruleset_id=ruleset_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + def _resubscribe_initial( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ResubscribeProperties, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + if body is not None: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + else: + _content = None + + _request = build_elastic_monitor_resources_resubscribe_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[_models.ResubscribeProperties] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Default value is None. + :type body: ~azure.mgmt.elastic.models.ResubscribeProperties + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[JSON] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Default value is None. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Default value is None. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_resubscribe( + self, + resource_group_name: str, + monitor_name: str, + body: Optional[Union[_models.ResubscribeProperties, JSON, IO[bytes]]] = None, + **kwargs: Any + ) -> LROPoller[_models.ArmResponseElasticMonitorResource]: + """Resubscribe the Elasticsearch Organization. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param body: Resubscribe Properties. Is one of the following types: ResubscribeProperties, + JSON, IO[bytes] Default value is None. + :type body: ~azure.mgmt.elastic.models.ResubscribeProperties or JSON or IO[bytes] + :return: An instance of LROPoller that returns ArmResponseElasticMonitorResource. The + ArmResponseElasticMonitorResource is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ArmResponseElasticMonitorResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ArmResponseElasticMonitorResource] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._resubscribe_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response_headers = {} + response = pipeline_response.http_response + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = _deserialize(_models.ArmResponseElasticMonitorResource, response.json()) + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[_models.ArmResponseElasticMonitorResource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.ArmResponseElasticMonitorResource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + +class MonitoredSubscriptionsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.ElasticClient`'s + :attr:`monitored_subscriptions` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get( + self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any + ) -> _models.MonitoredSubscriptionProperties: + """List all the subscriptions currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :return: MonitoredSubscriptionProperties. The MonitoredSubscriptionProperties is compatible + with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) + + _request = build_monitored_subscriptions_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.MonitoredSubscriptionProperties, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _createor_update_initial( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_monitored_subscriptions_createor_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: _models.MonitoredSubscriptionProperties, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_createor_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> LROPoller[None]: + """Add the subscriptions that should be monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Is one of the following types: MonitoredSubscriptionProperties, JSON, IO[bytes] + Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or JSON or IO[bytes] + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._createor_update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _update_initial( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_monitored_subscriptions_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: _models.MonitoredSubscriptionProperties, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update( + self, + resource_group_name: str, + monitor_name: str, + configuration_name: str, + body: Union[_models.MonitoredSubscriptionProperties, JSON, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.MonitoredSubscriptionProperties]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :param body: Is one of the following types: MonitoredSubscriptionProperties, JSON, IO[bytes] + Required. + :type body: ~azure.mgmt.elastic.models.MonitoredSubscriptionProperties or JSON or IO[bytes] + :return: An instance of LROPoller that returns MonitoredSubscriptionProperties. The + MonitoredSubscriptionProperties is compatible with MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.MonitoredSubscriptionProperties] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._update_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + body=body, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response = pipeline_response.http_response + deserialized = _deserialize(_models.MonitoredSubscriptionProperties, response.json()) + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[_models.MonitoredSubscriptionProperties].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.MonitoredSubscriptionProperties]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _delete_initial( + self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_monitored_subscriptions_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202, 204]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_delete( + self, resource_group_name: str, monitor_name: str, configuration_name: str, **kwargs: Any + ) -> LROPoller[None]: + """Updates the subscriptions that are being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param configuration_name: The configuration name. Only 'default' value is supported. Required. + :type configuration_name: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._delete_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + configuration_name=configuration_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> Iterable["_models.MonitoredSubscriptionProperties"]: + """List all the subscriptions currently being monitored by the Elastic monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of MonitoredSubscriptionProperties + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.MonitoredSubscriptionProperties] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.MonitoredSubscriptionProperties]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_monitored_subscriptions_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.MonitoredSubscriptionProperties], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class OpenAIIntegrationRPModelsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.ElasticClient`'s + :attr:`open_ai_integration_rp_models` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get( + self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Get OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) + + _request = build_open_ai_integration_rp_models_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.OpenAIIntegrationRPModel, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: _models.OpenAIIntegrationRPModel, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Required. + :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + integration_name: str, + body: Union[_models.OpenAIIntegrationRPModel, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.OpenAIIntegrationRPModel: + """Create or update a OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :param body: Is one of the following types: OpenAIIntegrationRPModel, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel or JSON or IO[bytes] + :return: OpenAIIntegrationRPModel. The OpenAIIntegrationRPModel is compatible with + MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationRPModel + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.OpenAIIntegrationRPModel] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_open_ai_integration_rp_models_create_or_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.OpenAIIntegrationRPModel, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any + ) -> None: + """Delete OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :return: None + :rtype: None + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + + _request = build_open_ai_integration_rp_models_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> Iterable["_models.OpenAIIntegrationRPModel"]: + """List OpenAI integration rule for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of OpenAIIntegrationRPModel + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.OpenAIIntegrationRPModel] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.OpenAIIntegrationRPModel]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_open_ai_integration_rp_models_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.OpenAIIntegrationRPModel], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def get_status( + self, resource_group_name: str, monitor_name: str, integration_name: str, **kwargs: Any + ) -> _models.OpenAIIntegrationStatusResponse: + """Get OpenAI integration status for a given integration. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param integration_name: OpenAI Integration name. Required. + :type integration_name: str + :return: OpenAIIntegrationStatusResponse. The OpenAIIntegrationStatusResponse is compatible + with MutableMapping + :rtype: ~azure.mgmt.elastic.models.OpenAIIntegrationStatusResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OpenAIIntegrationStatusResponse] = kwargs.pop("cls", None) + + _request = build_open_ai_integration_rp_models_get_status_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + integration_name=integration_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.OpenAIIntegrationStatusResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + +class TagRulesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.ElasticClient`'s + :attr:`tag_rules` attribute. """ - models = _models - - def __init__(self, *args, **kwargs): + def __init__(self, *args, **kwargs) -> None: input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") @distributed_trace - def list(self, **kwargs: Any) -> Iterable["_models.OperationResult"]: - """List all operations provided by Microsoft.Elastic. + def get( + self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any + ) -> _models.MonitoringTagRules: + """Get a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) - List all operations provided by Microsoft.Elastic. + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} - :return: An iterator like instance of either OperationResult or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.OperationResult] + cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) + + _request = build_tag_rules_get_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.MonitoringTagRules, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: _models.MonitoringTagRules, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Required. + :type body: ~azure.mgmt.elastic.models.MonitoringTagRules + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + monitor_name: str, + rule_set_name: str, + body: Union[_models.MonitoringTagRules, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.MonitoringTagRules: + """Create or update a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :param body: request body of MonitoringTagRules. Is one of the following types: + MonitoringTagRules, JSON, IO[bytes] Required. + :type body: ~azure.mgmt.elastic.models.MonitoringTagRules or JSON or IO[bytes] + :return: MonitoringTagRules. The MonitoringTagRules is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_tag_rules_create_or_update_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.MonitoringTagRules, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _delete_initial( + self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_tag_rules_delete_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202, 204]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_delete( + self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any + ) -> LROPoller[None]: + """Delete a tag rule set for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :param rule_set_name: Tag Rule Set resource name. Required. + :type rule_set_name: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._delete_initial( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + rule_set_name=rule_set_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list( + self, resource_group_name: str, monitor_name: str, **kwargs: Any + ) -> Iterable["_models.MonitoringTagRules"]: + """List the tag rules for a given monitor resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param monitor_name: Monitor resource name. Required. + :type monitor_name: str + :return: An iterator like instance of MonitoringTagRules + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.MonitoringTagRules] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OperationListResult] = kwargs.pop("cls", None) + cls: ClsType[List[_models.MonitoringTagRules]] = kwargs.pop("cls", None) - error_map: MutableMapping[int, Type[HttpResponseError]] = { + error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, @@ -104,12 +6102,20 @@ def list(self, **kwargs: Any) -> Iterable["_models.OperationResult"]: def prepare_request(next_link=None): if not next_link: - _request = build_list_request( - api_version=api_version, + _request = build_tag_rules_list_request( + resource_group_name=resource_group_name, + monitor_name=monitor_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -124,16 +6130,21 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + return _request def extract_data(pipeline_response): - deserialized = self._deserialize("OperationListResult", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.MonitoringTagRules], deserialized.get("value", [])) if cls: list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) + return deserialized.get("nextLink") or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) @@ -146,11 +6157,340 @@ def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class ElasticVersionsOperationGroupOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.ElasticClient`'s + :attr:`elastic_versions_operation_group` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def list(self, *, region: str, **kwargs: Any) -> Iterable["_models.ElasticVersionListFormat"]: + """Get a list of available versions for a region. + + Get a list of available versions for a region. + + :keyword region: Region where elastic deployment will take place. Required. + :paramtype region: str + :return: An iterator like instance of ElasticVersionListFormat + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.ElasticVersionListFormat] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.ElasticVersionListFormat]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_elastic_versions_operation_group_list_request( + subscription_id=self._config.subscription_id, + region=region, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.ElasticVersionListFormat], deserialized.get("value", [])) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **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) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response return ItemPaged(get_next, extract_data) + + +class OrganizationsOperationGroupOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.elastic.ElasticClient`'s + :attr:`organizations_operation_group` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: ElasticClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @overload + def get_api_key( + self, body: _models.UserEmailId, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Required. + :type body: ~azure.mgmt.elastic.models.UserEmailId + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def get_api_key( + self, body: JSON, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Required. + :type body: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def get_api_key( + self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Required. + :type body: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def get_api_key( + self, body: Union[_models.UserEmailId, JSON, IO[bytes]], **kwargs: Any + ) -> _models.UserApiKeyResponse: + """Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + Fetch User API Key from internal database, if it was generated and stored while creating the + Elasticsearch Organization. + + :param body: The request body. Is one of the following types: UserEmailId, JSON, IO[bytes] + Required. + :type body: ~azure.mgmt.elastic.models.UserEmailId or JSON or IO[bytes] + :return: UserApiKeyResponse. The UserApiKeyResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.UserApiKeyResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _content = json.dumps(body, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_organizations_operation_group_get_api_key_request( + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.UserApiKeyResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def get_elastic_to_azure_subscription_mapping( # pylint: disable=name-too-long + self, **kwargs: Any + ) -> _models.ElasticOrganizationToAzureSubscriptionMappingResponse: + """Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. + + ; + /** + Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. + + . + + :return: ElasticOrganizationToAzureSubscriptionMappingResponse. The + ElasticOrganizationToAzureSubscriptionMappingResponse is compatible with MutableMapping + :rtype: ~azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.ElasticOrganizationToAzureSubscriptionMappingResponse] = kwargs.pop("cls", None) + + _request = build_organizations_operation_group_get_elastic_to_azure_subscription_mapping_request( + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ResourceProviderDefaultErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.ElasticOrganizationToAzureSubscriptionMappingResponse, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_organizations_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_organizations_operations.py deleted file mode 100644 index 637c0aa8d3e2..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_organizations_operations.py +++ /dev/null @@ -1,538 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Iterator, Optional, Type, TypeVar, Union, cast, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_get_api_key_request(subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/getOrganizationApiKey" - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_elastic_to_azure_subscription_mapping_request( # pylint: disable=name-too-long - subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/providers/Microsoft.Elastic/getElasticOrganizationToAzureSubscriptionMapping", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_resubscribe_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/resubscribe", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class OrganizationsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`organizations` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - def get_api_key( - self, body: Optional[_models.UserEmailId] = None, *, content_type: str = "application/json", **kwargs: Any - ) -> _models.UserApiKeyResponse: - """Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - :param body: Email Id parameter of the User Organization, of which the API Key must be - returned. Default value is None. - :type body: ~azure.mgmt.elastic.models.UserEmailId - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: UserApiKeyResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def get_api_key( - self, body: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any - ) -> _models.UserApiKeyResponse: - """Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - :param body: Email Id parameter of the User Organization, of which the API Key must be - returned. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: UserApiKeyResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def get_api_key( - self, body: Optional[Union[_models.UserEmailId, IO[bytes]]] = None, **kwargs: Any - ) -> _models.UserApiKeyResponse: - """Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - Fetch User API Key from internal database, if it was generated and stored while creating the - Elasticsearch Organization. - - :param body: Email Id parameter of the User Organization, of which the API Key must be - returned. Is either a UserEmailId type or a IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.UserEmailId or IO[bytes] - :return: UserApiKeyResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UserApiKeyResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.UserApiKeyResponse] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "UserEmailId") - else: - _json = None - - _request = build_get_api_key_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("UserApiKeyResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def get_elastic_to_azure_subscription_mapping( # pylint: disable=name-too-long - self, **kwargs: Any - ) -> _models.ElasticOrganizationToAzureSubscriptionMappingResponse: - """Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. - - Get Elastic Organization To Azure Subscription Mapping details for the logged-in user. - - :return: ElasticOrganizationToAzureSubscriptionMappingResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.ElasticOrganizationToAzureSubscriptionMappingResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ElasticOrganizationToAzureSubscriptionMappingResponse] = kwargs.pop("cls", None) - - _request = build_get_elastic_to_azure_subscription_mapping_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize( - "ElasticOrganizationToAzureSubscriptionMappingResponse", pipeline_response.http_response - ) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - def _resubscribe_initial( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ResubscribeProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "ResubscribeProperties") - else: - _json = None - - _request = build_resubscribe_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_resubscribe( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.ResubscribeProperties] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Resubscribe the Elasticsearch Organization. - - Resubscribe the Elasticsearch Organization. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Resubscribe Properties. Default value is None. - :type body: ~azure.mgmt.elastic.models.ResubscribeProperties - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_resubscribe( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Resubscribe the Elasticsearch Organization. - - Resubscribe the Elasticsearch Organization. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Resubscribe Properties. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_resubscribe( - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.ResubscribeProperties, IO[bytes]]] = None, - **kwargs: Any - ) -> LROPoller[_models.ElasticMonitorResource]: - """Resubscribe the Elasticsearch Organization. - - Resubscribe the Elasticsearch Organization. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: Resubscribe Properties. Is either a ResubscribeProperties type or a IO[bytes] - type. Default value is None. - :type body: ~azure.mgmt.elastic.models.ResubscribeProperties or IO[bytes] - :return: An instance of LROPoller that returns either ElasticMonitorResource or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.elastic.models.ElasticMonitorResource] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ElasticMonitorResource] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._resubscribe_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - body=body, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("ElasticMonitorResource", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.ElasticMonitorResource].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.ElasticMonitorResource]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_patch.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_patch.py +++ b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_tag_rules_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_tag_rules_operations.py deleted file mode 100644 index c90ee68c915b..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_tag_rules_operations.py +++ /dev/null @@ -1,619 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, Type, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat -from azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_create_or_update_request( - resource_group_name: str, monitor_name: str, rule_set_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules/{ruleSetName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "ruleSetName": _SERIALIZER.url("rule_set_name", rule_set_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, monitor_name: str, rule_set_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules/{ruleSetName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "ruleSetName": _SERIALIZER.url("rule_set_name", rule_set_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_delete_request( - resource_group_name: str, monitor_name: str, rule_set_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/tagRules/{ruleSetName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - "ruleSetName": _SERIALIZER.url("rule_set_name", rule_set_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -class TagRulesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`tag_rules` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, monitor_name: str, **kwargs: Any - ) -> Iterable["_models.MonitoringTagRules"]: - """List the tag rules for a given monitor resource. - - List the tag rules for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either MonitoringTagRules or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.MonitoringTagRules] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoringTagRulesListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("MonitoringTagRulesListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - rule_set_name: str, - body: Optional[_models.MonitoringTagRules] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.MonitoringTagRules: - """Create or update a tag rule set for a given monitor resource. - - Create or update a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :param body: request body of MonitoringTagRules. Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoringTagRules - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - rule_set_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.MonitoringTagRules: - """Create or update a tag rule set for a given monitor resource. - - Create or update a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :param body: request body of MonitoringTagRules. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update( - self, - resource_group_name: str, - monitor_name: str, - rule_set_name: str, - body: Optional[Union[_models.MonitoringTagRules, IO[bytes]]] = None, - **kwargs: Any - ) -> _models.MonitoringTagRules: - """Create or update a tag rule set for a given monitor resource. - - Create or update a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :param body: request body of MonitoringTagRules. Is either a MonitoringTagRules type or a - IO[bytes] type. Default value is None. - :type body: ~azure.mgmt.elastic.models.MonitoringTagRules or IO[bytes] - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "MonitoringTagRules") - else: - _json = None - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("MonitoringTagRules", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def get( - self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any - ) -> _models.MonitoringTagRules: - """Get a tag rule set for a given monitor resource. - - Get a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :return: MonitoringTagRules or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.MonitoringTagRules - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.MonitoringTagRules] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("MonitoringTagRules", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - def _delete_initial( - self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202, 204]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_delete( - self, resource_group_name: str, monitor_name: str, rule_set_name: str, **kwargs: Any - ) -> LROPoller[None]: - """Delete a tag rule set for a given monitor resource. - - Delete a tag rule set for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param rule_set_name: Tag Rule Set resource name. Required. - :type rule_set_name: str - :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: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._delete_initial( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - rule_set_name=rule_set_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_traffic_filters_operations.py deleted file mode 100644 index da9e60c4ef68..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_traffic_filters_operations.py +++ /dev/null @@ -1,158 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_delete_request( - resource_group_name: str, - monitor_name: str, - subscription_id: str, - *, - ruleset_id: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/deleteTrafficFilter", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - if ruleset_id is not None: - _params["rulesetId"] = _SERIALIZER.query("ruleset_id", ruleset_id, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class TrafficFiltersOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`traffic_filters` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, monitor_name: str, ruleset_id: Optional[str] = None, **kwargs: Any - ) -> None: - """Delete traffic filter from the account. - - Delete traffic filter from the account. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param ruleset_id: Ruleset Id of the filter. Default value is None. - :type ruleset_id: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - ruleset_id=ruleset_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_upgradable_versions_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_upgradable_versions_operations.py deleted file mode 100644 index 97c59ff6a68b..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_upgradable_versions_operations.py +++ /dev/null @@ -1,150 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_details_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listUpgradableVersions", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class UpgradableVersionsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`upgradable_versions` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def details(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.UpgradableVersionsList: - """List of upgradable versions for a given monitor resource. - - List of upgradable versions for a given monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: UpgradableVersionsList or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.UpgradableVersionsList - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.UpgradableVersionsList] = kwargs.pop("cls", None) - - _request = build_details_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("UpgradableVersionsList", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_collection_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_collection_operations.py deleted file mode 100644 index 0e01c0b74d53..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_collection_operations.py +++ /dev/null @@ -1,232 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Optional, Type, TypeVar, Union, overload - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_update_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/vmCollectionUpdate", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class VMCollectionOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`vm_collection` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @overload - def update( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - monitor_name: str, - body: Optional[_models.VMCollectionUpdate] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> None: - """Update the vm details that will be monitored by the Elastic monitor resource. - - Update the vm details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: VM resource Id. Default value is None. - :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def update( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - monitor_name: str, - body: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> None: - """Update the vm details that will be monitored by the Elastic monitor resource. - - Update the vm details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: VM resource Id. Default value is None. - :type body: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def update( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - monitor_name: str, - body: Optional[Union[_models.VMCollectionUpdate, IO[bytes]]] = None, - **kwargs: Any - ) -> None: - """Update the vm details that will be monitored by the Elastic monitor resource. - - Update the vm details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :param body: VM resource Id. Is either a VMCollectionUpdate type or a IO[bytes] type. Default - value is None. - :type body: ~azure.mgmt.elastic.models.VMCollectionUpdate or IO[bytes] - :return: None or the result of cls(response) - :rtype: None - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(body, (IOBase, bytes)): - _content = body - else: - if body is not None: - _json = self._serialize.body(body, "VMCollectionUpdate") - else: - _json = None - - _request = build_update_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_host_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_host_operations.py deleted file mode 100644 index 44069753b840..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_host_operations.py +++ /dev/null @@ -1,176 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Iterable, Optional, Type, TypeVar -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -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(resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/listVMHost", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class VMHostOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`vm_host` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> Iterable["_models.VMResources"]: - """List the vm resources currently being monitored by the Elastic monitor resource. - - List the vm resources currently being monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: An iterator like instance of either VMResources or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.elastic.models.VMResources] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.VMHostListResponse] = kwargs.pop("cls", None) - - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("VMHostListResponse", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) diff --git a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_ingestion_operations.py b/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_ingestion_operations.py deleted file mode 100644 index 355848b89f65..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/azure/mgmt/elastic/operations/_vm_ingestion_operations.py +++ /dev/null @@ -1,150 +0,0 @@ -# pylint: disable=too-many-lines,too-many-statements -# 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 sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_details_request( - resource_group_name: str, monitor_name: str, subscription_id: str, **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-06-15-preview")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Elastic/monitors/{monitorName}/vmIngestionDetails", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "monitorName": _SERIALIZER.url("monitor_name", monitor_name, "str", pattern=r"^.*$"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class VMIngestionOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.elastic.MicrosoftElastic`'s - :attr:`vm_ingestion` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def details(self, resource_group_name: str, monitor_name: str, **kwargs: Any) -> _models.VMIngestionDetailsResponse: - """List the vm ingestion details that will be monitored by the Elastic monitor resource. - - List the vm ingestion details that will be monitored by the Elastic monitor resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param monitor_name: Monitor resource name. Required. - :type monitor_name: str - :return: VMIngestionDetailsResponse or the result of cls(response) - :rtype: ~azure.mgmt.elastic.models.VMIngestionDetailsResponse - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping[int, Type[HttpResponseError]] = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.VMIngestionDetailsResponse] = kwargs.pop("cls", None) - - _request = build_details_request( - resource_group_name=resource_group_name, - monitor_name=monitor_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **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) - error = self._deserialize.failsafe_deserialize( - _models.ResourceProviderDefaultErrorResponse, pipeline_response - ) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("VMIngestionDetailsResponse", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/all_traffic_filters_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/all_traffic_filters_list.py index 192c220eacfa..cb78b8b5f960 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/all_traffic_filters_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/all_traffic_filters_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.all_traffic_filters.list( + response = client.elastic_monitor_resources.all_traffic_filters_list( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/AllTrafficFilters_list.json +# x-ms-original-file: 2025-01-15-preview/AllTrafficFilters_list.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/associate_traffic_filter_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/associate_traffic_filter_update.py index 86d4e919a9c3..cfe72009df0b 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/associate_traffic_filter_update.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/associate_traffic_filter_update.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,17 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.associate_traffic_filter.begin_associate( + client.elastic_monitor_resources.begin_associate( resource_group_name="myResourceGroup", monitor_name="myMonitor", ).result() -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/AssociateTrafficFilter_Update.json +# x-ms-original-file: 2025-01-15-preview/AssociateTrafficFilter_Update.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/associated_filters_for_deployment_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/associated_filters_for_deployment_list.py index 94485706a700..e47a4577a244 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/associated_filters_for_deployment_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/associated_filters_for_deployment_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.list_associated_traffic_filters.list( + response = client.elastic_monitor_resources.list_associated_traffic_filters_list( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/AssociatedFiltersForDeployment_list.json +# x-ms-original-file: 2025-01-15-preview/AssociatedFiltersForDeployment_list.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/billing_info_get.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/billing_info_get.py index d63d5aff3400..b5e884bbc0b7 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/billing_info_get.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/billing_info_get.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.billing_info.get( + response = client.elastic_monitor_resources.billing_info_get( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/BillingInfo_Get.json +# x-ms-original-file: 2025-01-15-preview/BillingInfo_Get.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/connected_partner_resources_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/connected_partner_resources_list.py index d1291a1790b5..320fedcb9ff0 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/connected_partner_resources_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/connected_partner_resources_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,12 +25,12 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.connected_partner_resources.list( + response = client.elastic_monitor_resources.connected_partner_resources_list( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) @@ -38,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/ConnectedPartnerResources_List.json +# x-ms-original-file: 2025-01-15-preview/ConnectedPartnerResources_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/deployment_info_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/deployment_info_list.py index e72c44fd3a1b..1d8fbebf392c 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/deployment_info_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/deployment_info_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.deployment_info.list( + response = client.elastic_monitor_resources.deployment_info_list( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/DeploymentInfo_List.json +# x-ms-original-file: 2025-01-15-preview/DeploymentInfo_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_and_delete_traffic_filter_delete.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_and_delete_traffic_filter_delete.py index 4bb6081719f7..96fc7c5a098d 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_and_delete_traffic_filter_delete.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_and_delete_traffic_filter_delete.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,17 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.detach_and_delete_traffic_filter.delete( + client.elastic_monitor_resources.detach_and_delete_traffic_filter_delete( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/DetachAndDeleteTrafficFilter_Delete.json +# x-ms-original-file: 2025-01-15-preview/DetachAndDeleteTrafficFilter_Delete.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_traffic_filters_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_traffic_filters_update.py index f02a77efb667..5c9b56deddb3 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_traffic_filters_update.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/detach_traffic_filters_update.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,17 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.detach_traffic_filter.begin_update( + client.elastic_monitor_resources.begin_detach_traffic_filter_update( resource_group_name="myResourceGroup", monitor_name="myMonitor", ).result() -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/DetachTrafficFilters_Update.json +# x-ms-original-file: 2025-01-15-preview/DetachTrafficFilters_Update.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/elastic_versions_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/elastic_versions_list.py index 99f685505a33..c4539896b458 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/elastic_versions_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/elastic_versions_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.elastic_versions.list( + response = client.elastic_versions_operation_group.list( region="myregion", ) for item in response: print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/ElasticVersions_List.json +# x-ms-original-file: 2025-01-15-preview/ElasticVersions_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/external_user_info.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/external_user_info.py index 5898f821f03b..7ba0b9c164d1 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/external_user_info.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/external_user_info.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.external_user.create_or_update( + response = client.elastic_monitor_resources.create_or_update( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/ExternalUserInfo.json +# x-ms-original-file: 2025-01-15-preview/ExternalUserInfo.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/ip_traffic_filter_create.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/ip_traffic_filter_create.py index 0a6ad7a84643..62fb4b4b0382 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/ip_traffic_filter_create.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/ip_traffic_filter_create.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.create_and_associate_ip_filter.begin_create( + response = client.elastic_monitor_resources.begin_create_and_associate_ip_filter_create( resource_group_name="myResourceGroup", monitor_name="myMonitor", ).result() + print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/IPTrafficFilter_Create.json +# x-ms-original-file: 2025-01-15-preview/IPTrafficFilter_Create.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitor_upgrade.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitor_upgrade.py index 4342e54799c0..c2c965ab8d37 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitor_upgrade.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitor_upgrade.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,17 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.monitor.begin_upgrade( + client.elastic_monitor_resources.begin_upgrade( resource_group_name="myResourceGroup", monitor_name="myMonitor", ).result() -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Monitor_Upgrade.json +# x-ms-original-file: 2025-01-15-preview/Monitor_Upgrade.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_resources_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_resources_list.py index c7d6b86e09e2..7a8826d40cb7 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_resources_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_resources_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,12 +25,12 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.monitored_resources.list( + response = client.elastic_monitor_resources.monitored_resources_list( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) @@ -38,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/MonitoredResources_List.json +# x-ms-original-file: 2025-01-15-preview/MonitoredResources_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_createor_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_createor_update.py deleted file mode 100644 index 43b6c606ca26..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_createor_update.py +++ /dev/null @@ -1,43 +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. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from azure.mgmt.elastic import MicrosoftElastic - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-elastic -# USAGE - python monitored_subscriptions_createor_update.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = MicrosoftElastic( - credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", - ) - - response = client.monitored_subscriptions.begin_createor_update( - resource_group_name="myResourceGroup", - monitor_name="myMonitor", - configuration_name="default", - ).result() - print(response) - - -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/MonitoredSubscriptions_CreateorUpdate.json -if __name__ == "__main__": - main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_delete.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_delete.py index 748f23944922..352d0c023640 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_delete.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_delete.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,9 +25,9 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) client.monitored_subscriptions.begin_delete( @@ -37,6 +37,6 @@ def main(): ).result() -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/MonitoredSubscriptions_Delete.json +# x-ms-original-file: 2025-01-15-preview/MonitoredSubscriptions_Delete.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_get.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_get.py index 4023857d25cb..9451bc7aaf34 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_get.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_get.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,9 +25,9 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) response = client.monitored_subscriptions.get( @@ -38,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/MonitoredSubscriptions_Get.json +# x-ms-original-file: 2025-01-15-preview/MonitoredSubscriptions_Get.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_list.py index 76e5a8085a86..e9d041e875fb 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,9 +25,9 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) response = client.monitored_subscriptions.list( @@ -38,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/MonitoredSubscriptions_List.json +# x-ms-original-file: 2025-01-15-preview/MonitoredSubscriptions_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_update.py deleted file mode 100644 index 729b049942e7..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitored_subscriptions_update.py +++ /dev/null @@ -1,43 +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. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from azure.mgmt.elastic import MicrosoftElastic - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-elastic -# USAGE - python monitored_subscriptions_update.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = MicrosoftElastic( - credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", - ) - - response = client.monitored_subscriptions.begin_update( - resource_group_name="myResourceGroup", - monitor_name="myMonitor", - configuration_name="default", - ).result() - print(response) - - -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/MonitoredSubscriptions_Update.json -if __name__ == "__main__": - main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_create.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_create.py deleted file mode 100644 index 8b1f4ebd3fc9..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_create.py +++ /dev/null @@ -1,42 +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. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from azure.mgmt.elastic import MicrosoftElastic - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-elastic -# USAGE - python monitors_create.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = MicrosoftElastic( - credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", - ) - - response = client.monitors.begin_create( - resource_group_name="myResourceGroup", - monitor_name="myMonitor", - ).result() - print(response) - - -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Monitors_Create.json -if __name__ == "__main__": - main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_delete.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_delete.py index 4f8086aa0e8f..25deeeb7f821 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_delete.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_delete.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,17 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.monitors.begin_delete( + client.elastic_monitor_resources.begin_delete( resource_group_name="myResourceGroup", monitor_name="myMonitor", ).result() -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Monitors_Delete.json +# x-ms-original-file: 2025-01-15-preview/Monitors_Delete.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_get.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_get.py index 811546973bff..1cffb5412814 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_get.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_get.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.monitors.get( + response = client.elastic_monitor_resources.get( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Monitors_Get.json +# x-ms-original-file: 2025-01-15-preview/Monitors_Get.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list.py index 2c625cfdb3a4..0a4d46ca7c80 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,16 +25,16 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.monitors.list() + response = client.elastic_monitor_resources.list() for item in response: print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Monitors_List.json +# x-ms-original-file: 2025-01-15-preview/Monitors_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list_by_resource_group.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list_by_resource_group.py index f9cc17b0731e..ef901be7fc49 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list_by_resource_group.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_list_by_resource_group.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.monitors.list_by_resource_group( + response = client.elastic_monitor_resources.list_by_resource_group( resource_group_name="myResourceGroup", ) for item in response: print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Monitors_ListByResourceGroup.json +# x-ms-original-file: 2025-01-15-preview/Monitors_ListByResourceGroup.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_update.py deleted file mode 100644 index b88a09930f13..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/monitors_update.py +++ /dev/null @@ -1,42 +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. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from azure.mgmt.elastic import MicrosoftElastic - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-elastic -# USAGE - python monitors_update.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = MicrosoftElastic( - credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", - ) - - response = client.monitors.begin_update( - resource_group_name="myResourceGroup", - monitor_name="myMonitor", - ).result() - print(response) - - -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Monitors_Update.json -if __name__ == "__main__": - main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_create_or_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_create_or_update.py deleted file mode 100644 index 83ac9a43b3c5..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_create_or_update.py +++ /dev/null @@ -1,43 +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. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from azure.mgmt.elastic import MicrosoftElastic - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-elastic -# USAGE - python open_ai_create_or_update.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = MicrosoftElastic( - credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", - ) - - response = client.open_ai.create_or_update( - resource_group_name="myResourceGroup", - monitor_name="myMonitor", - integration_name="default", - ) - print(response) - - -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/OpenAI_CreateOrUpdate.json -if __name__ == "__main__": - main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_delete.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_delete.py index 0475f4f38535..6dbb8ebdf532 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_delete.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_delete.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.open_ai.delete( + client.open_ai_integration_rp_models.delete( resource_group_name="myResourceGroup", monitor_name="myMonitor", integration_name="default", ) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/OpenAI_Delete.json +# x-ms-original-file: 2025-01-15-preview/OpenAI_Delete.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get.py index 669ca691ff4e..6e48b06f5d2f 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,12 +25,12 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.open_ai.get( + response = client.open_ai_integration_rp_models.get( resource_group_name="myResourceGroup", monitor_name="myMonitor", integration_name="default", @@ -38,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/OpenAI_Get.json +# x-ms-original-file: 2025-01-15-preview/OpenAI_Get.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get_status.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get_status.py index 6dffd6a9c13f..f5b6b4d169f9 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get_status.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_get_status.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,12 +25,12 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.open_ai.get_status( + response = client.open_ai_integration_rp_models.get_status( resource_group_name="myResourceGroup", monitor_name="myMonitor", integration_name="default", @@ -38,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/OpenAI_GetStatus.json +# x-ms-original-file: 2025-01-15-preview/OpenAI_GetStatus.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_list.py index 4a61dc188645..fe999724a6f6 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/open_ai_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,12 +25,12 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.open_ai.list( + response = client.open_ai_integration_rp_models.list( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) @@ -38,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/OpenAI_List.json +# x-ms-original-file: 2025-01-15-preview/OpenAI_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/operations_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/operations_list.py index f958f28edbde..2c11663767a6 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/operations_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/operations_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) @@ -35,6 +35,6 @@ def main(): print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Operations_List.json +# x-ms-original-file: 2025-01-15-preview/Operations_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_api_key.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_api_key.py deleted file mode 100644 index 379c1d421a2c..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_api_key.py +++ /dev/null @@ -1,39 +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. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from azure.mgmt.elastic import MicrosoftElastic - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-elastic -# USAGE - python organizations_get_api_key.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = MicrosoftElastic( - credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", - ) - - response = client.organizations.get_api_key() - print(response) - - -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Organizations_GetApiKey.json -if __name__ == "__main__": - main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_elastic_to_azure_subscription_mapping.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_elastic_to_azure_subscription_mapping.py index ab851377bbe4..a08bc6be5b6a 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_elastic_to_azure_subscription_mapping.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_get_elastic_to_azure_subscription_mapping.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,15 +25,15 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.organizations.get_elastic_to_azure_subscription_mapping() + response = client.organizations_operation_group.get_elastic_to_azure_subscription_mapping() print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Organizations_GetElasticToAzureSubscriptionMapping.json +# x-ms-original-file: 2025-01-15-preview/Organizations_GetElasticToAzureSubscriptionMapping.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_resubscribe.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_resubscribe.py index d41596481c59..3f8d70dc351b 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_resubscribe.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/organizations_resubscribe.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.organizations.begin_resubscribe( + response = client.elastic_monitor_resources.begin_resubscribe( resource_group_name="myResourceGroup", monitor_name="myMonitor", ).result() print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/Organizations_Resubscribe.json +# x-ms-original-file: 2025-01-15-preview/Organizations_Resubscribe.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/private_link_traffic_filters_create.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/private_link_traffic_filters_create.py index 97ed397049dc..1f6c63c05656 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/private_link_traffic_filters_create.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/private_link_traffic_filters_create.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.create_and_associate_pl_filter.begin_create( + response = client.elastic_monitor_resources.begin_create_and_associate_pl_filter_create( resource_group_name="myResourceGroup", monitor_name="myMonitor", ).result() + print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/PrivateLinkTrafficFilters_Create.json +# x-ms-original-file: 2025-01-15-preview/PrivateLinkTrafficFilters_Create.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_create_or_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_create_or_update.py deleted file mode 100644 index c5ec10d30dca..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_create_or_update.py +++ /dev/null @@ -1,43 +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. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from azure.mgmt.elastic import MicrosoftElastic - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-elastic -# USAGE - python tag_rules_create_or_update.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = MicrosoftElastic( - credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", - ) - - response = client.tag_rules.create_or_update( - resource_group_name="myResourceGroup", - monitor_name="myMonitor", - rule_set_name="default", - ) - print(response) - - -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/TagRules_CreateOrUpdate.json -if __name__ == "__main__": - main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_delete.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_delete.py index 512759075338..5f0eb05b46c8 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_delete.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_delete.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,9 +25,9 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) client.tag_rules.begin_delete( @@ -37,6 +37,6 @@ def main(): ).result() -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/TagRules_Delete.json +# x-ms-original-file: 2025-01-15-preview/TagRules_Delete.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_get.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_get.py index 5827ddbab4f7..a9f779a6aa92 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_get.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_get.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,9 +25,9 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) response = client.tag_rules.get( @@ -38,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/TagRules_Get.json +# x-ms-original-file: 2025-01-15-preview/TagRules_Get.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_list.py index ee84a6dbbcb3..e7af6cee99e6 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/tag_rules_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,9 +25,9 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) response = client.tag_rules.list( @@ -38,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/TagRules_List.json +# x-ms-original-file: 2025-01-15-preview/TagRules_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/traffic_filters_delete.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/traffic_filters_delete.py index e90d069400f8..4bf41b2f4992 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/traffic_filters_delete.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/traffic_filters_delete.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,17 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.traffic_filters.delete( + client.elastic_monitor_resources.traffic_filters_delete( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/TrafficFilters_Delete.json +# x-ms-original-file: 2025-01-15-preview/TrafficFilters_Delete.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/upgradable_versions_details.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/upgradable_versions_details.py index c61e8a3f0b5a..92199ce8fbee 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/upgradable_versions_details.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/upgradable_versions_details.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.upgradable_versions.details( + response = client.elastic_monitor_resources.upgradable_versions_details( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/UpgradableVersions_Details.json +# x-ms-original-file: 2025-01-15-preview/UpgradableVersions_Details.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_collection_update.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_collection_update.py index e74269665eb1..b62868e33460 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_collection_update.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_collection_update.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,17 +25,17 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - client.vm_collection.update( + client.elastic_monitor_resources.v_m_collection_update( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/VMCollection_Update.json +# x-ms-original-file: 2025-01-15-preview/VMCollection_Update.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_host_list.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_host_list.py index 087403205d0f..e360b7ef2188 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_host_list.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_host_list.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,12 +25,12 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.vm_host.list( + response = client.elastic_monitor_resources.v_m_host_list( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) @@ -38,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/VMHost_List.json +# x-ms-original-file: 2025-01-15-preview/VMHost_List.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_ingestion_details.py b/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_ingestion_details.py index a7392fa83c4b..11a26633c2ff 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_ingestion_details.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_samples/vm_ingestion_details.py @@ -2,13 +2,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient """ # PREREQUISITES @@ -25,18 +25,18 @@ def main(): - client = MicrosoftElastic( + client = ElasticClient( credential=DefaultAzureCredential(), - subscription_id="00000000-0000-0000-0000-000000000000", + subscription_id="SUBSCRIPTION_ID", ) - response = client.vm_ingestion.details( + response = client.elastic_monitor_resources.details( resource_group_name="myResourceGroup", monitor_name="myMonitor", ) print(response) -# x-ms-original-file: specification/elastic/resource-manager/Microsoft.Elastic/preview/2024-06-15-preview/examples/VMIngestion_Details.json +# x-ms-original-file: 2025-01-15-preview/VMIngestion_Details.json if __name__ == "__main__": main() diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/conftest.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/conftest.py index d03a078ad93b..9e3e52e68876 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/conftest.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/conftest.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import os @@ -18,17 +18,17 @@ load_dotenv() -# aovid record sensitive identity information in recordings +# For security, please avoid record sensitive identity information in recordings @pytest.fixture(scope="session", autouse=True) def add_sanitizers(test_proxy): - microsoftelastic_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - microsoftelastic_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - microsoftelastic_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - microsoftelastic_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=microsoftelastic_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=microsoftelastic_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=microsoftelastic_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=microsoftelastic_client_secret, value="00000000-0000-0000-0000-000000000000") + elastic_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + elastic_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + elastic_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + elastic_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=elastic_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=elastic_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=elastic_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=elastic_client_secret, value="00000000-0000-0000-0000-000000000000") add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") add_header_regex_sanitizer(key="Cookie", value="cookie;") diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations.py new file mode 100644 index 000000000000..a6538c2f9dee --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations.py @@ -0,0 +1,355 @@ +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.elastic import ElasticClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestElasticElasticMonitorResourcesOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ElasticClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_get(self, resource_group): + response = self.client.elastic_monitor_resources.get( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_create(self, resource_group): + response = self.client.elastic_monitor_resources.begin_create( + resource_group_name=resource_group.name, + monitor_name="str", + body={ + "location": "str", + "id": "str", + "identity": {"principalId": "str", "tenantId": "str", "type": "str"}, + "kind": "str", + "name": "str", + "properties": { + "elasticProperties": { + "elasticCloudDeployment": { + "azureSubscriptionId": "str", + "deploymentId": "str", + "elasticsearchRegion": "str", + "elasticsearchServiceUrl": "str", + "kibanaServiceUrl": "str", + "kibanaSsoUrl": "str", + "name": "str", + }, + "elasticCloudUser": {"elasticCloudSsoDefaultUrl": "str", "emailAddress": "str", "id": "str"}, + }, + "generateApiKey": bool, + "hostingType": "str", + "liftrResourceCategory": "str", + "liftrResourcePreference": 0, + "monitoringStatus": "str", + "planDetails": { + "offerID": "str", + "planID": "str", + "planName": "str", + "publisherID": "str", + "termID": "str", + }, + "projectDetails": {"configurationType": "str", "projectType": "str"}, + "provisioningState": "str", + "saaSAzureSubscriptionStatus": "str", + "sourceCampaignId": "str", + "sourceCampaignName": "str", + "subscriptionState": "str", + "userInfo": { + "companyInfo": { + "business": "str", + "country": "str", + "domain": "str", + "employeesNumber": "str", + "state": "str", + }, + "companyName": "str", + "emailAddress": "str", + "firstName": "str", + "lastName": "str", + }, + "version": "str", + }, + "sku": {"name": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_update(self, resource_group): + response = self.client.elastic_monitor_resources.begin_update( + resource_group_name=resource_group.name, + monitor_name="str", + body={"tags": {"str": "str"}}, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_delete(self, resource_group): + response = self.client.elastic_monitor_resources.begin_delete( + resource_group_name=resource_group.name, + monitor_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_list_by_resource_group(self, resource_group): + response = self.client.elastic_monitor_resources.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_list(self, resource_group): + response = self.client.elastic_monitor_resources.list() + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_monitored_resources_list(self, resource_group): + response = self.client.elastic_monitor_resources.monitored_resources_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_deployment_info_list(self, resource_group): + response = self.client.elastic_monitor_resources.deployment_info_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_create_or_update(self, resource_group): + response = self.client.elastic_monitor_resources.create_or_update( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_billing_info_get(self, resource_group): + response = self.client.elastic_monitor_resources.billing_info_get( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_connected_partner_resources_list(self, resource_group): + response = self.client.elastic_monitor_resources.connected_partner_resources_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_v_m_host_list(self, resource_group): + response = self.client.elastic_monitor_resources.v_m_host_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_details(self, resource_group): + response = self.client.elastic_monitor_resources.details( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_v_m_collection_update(self, resource_group): + response = self.client.elastic_monitor_resources.v_m_collection_update( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_upgradable_versions_details(self, resource_group): + response = self.client.elastic_monitor_resources.upgradable_versions_details( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_upgrade(self, resource_group): + response = self.client.elastic_monitor_resources.begin_upgrade( + resource_group_name=resource_group.name, + monitor_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_all_traffic_filters_list(self, resource_group): + response = self.client.elastic_monitor_resources.all_traffic_filters_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_list_associated_traffic_filters_list(self, resource_group): + response = self.client.elastic_monitor_resources.list_associated_traffic_filters_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_create_and_associate_ip_filter_create(self, resource_group): + response = self.client.elastic_monitor_resources.begin_create_and_associate_ip_filter_create( + resource_group_name=resource_group.name, + monitor_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_create_and_associate_pl_filter_create(self, resource_group): + response = self.client.elastic_monitor_resources.begin_create_and_associate_pl_filter_create( + resource_group_name=resource_group.name, + monitor_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_associate(self, resource_group): + response = self.client.elastic_monitor_resources.begin_associate( + resource_group_name=resource_group.name, + monitor_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_detach_and_delete_traffic_filter_delete(self, resource_group): + response = self.client.elastic_monitor_resources.detach_and_delete_traffic_filter_delete( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_detach_traffic_filter_update(self, resource_group): + response = self.client.elastic_monitor_resources.begin_detach_traffic_filter_update( + resource_group_name=resource_group.name, + monitor_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_traffic_filters_delete(self, resource_group): + response = self.client.elastic_monitor_resources.traffic_filters_delete( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_elastic_monitor_resources_begin_resubscribe(self, resource_group): + response = self.client.elastic_monitor_resources.begin_resubscribe( + resource_group_name=resource_group.name, + monitor_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations_async.py new file mode 100644 index 000000000000..8af8557994f1 --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_monitor_resources_operations_async.py @@ -0,0 +1,378 @@ +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.elastic.aio import ElasticClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestElasticElasticMonitorResourcesOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ElasticClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_get(self, resource_group): + response = await self.client.elastic_monitor_resources.get( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_create(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_create( + resource_group_name=resource_group.name, + monitor_name="str", + body={ + "location": "str", + "id": "str", + "identity": {"principalId": "str", "tenantId": "str", "type": "str"}, + "kind": "str", + "name": "str", + "properties": { + "elasticProperties": { + "elasticCloudDeployment": { + "azureSubscriptionId": "str", + "deploymentId": "str", + "elasticsearchRegion": "str", + "elasticsearchServiceUrl": "str", + "kibanaServiceUrl": "str", + "kibanaSsoUrl": "str", + "name": "str", + }, + "elasticCloudUser": { + "elasticCloudSsoDefaultUrl": "str", + "emailAddress": "str", + "id": "str", + }, + }, + "generateApiKey": bool, + "hostingType": "str", + "liftrResourceCategory": "str", + "liftrResourcePreference": 0, + "monitoringStatus": "str", + "planDetails": { + "offerID": "str", + "planID": "str", + "planName": "str", + "publisherID": "str", + "termID": "str", + }, + "projectDetails": {"configurationType": "str", "projectType": "str"}, + "provisioningState": "str", + "saaSAzureSubscriptionStatus": "str", + "sourceCampaignId": "str", + "sourceCampaignName": "str", + "subscriptionState": "str", + "userInfo": { + "companyInfo": { + "business": "str", + "country": "str", + "domain": "str", + "employeesNumber": "str", + "state": "str", + }, + "companyName": "str", + "emailAddress": "str", + "firstName": "str", + "lastName": "str", + }, + "version": "str", + }, + "sku": {"name": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_update(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_update( + resource_group_name=resource_group.name, + monitor_name="str", + body={"tags": {"str": "str"}}, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_delete(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_delete( + resource_group_name=resource_group.name, + monitor_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_list_by_resource_group(self, resource_group): + response = self.client.elastic_monitor_resources.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_list(self, resource_group): + response = self.client.elastic_monitor_resources.list() + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_monitored_resources_list(self, resource_group): + response = self.client.elastic_monitor_resources.monitored_resources_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_deployment_info_list(self, resource_group): + response = await self.client.elastic_monitor_resources.deployment_info_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_create_or_update(self, resource_group): + response = await self.client.elastic_monitor_resources.create_or_update( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_billing_info_get(self, resource_group): + response = await self.client.elastic_monitor_resources.billing_info_get( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_connected_partner_resources_list(self, resource_group): + response = self.client.elastic_monitor_resources.connected_partner_resources_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_v_m_host_list(self, resource_group): + response = self.client.elastic_monitor_resources.v_m_host_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_details(self, resource_group): + response = await self.client.elastic_monitor_resources.details( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_v_m_collection_update(self, resource_group): + response = await self.client.elastic_monitor_resources.v_m_collection_update( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_upgradable_versions_details(self, resource_group): + response = await self.client.elastic_monitor_resources.upgradable_versions_details( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_upgrade(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_upgrade( + resource_group_name=resource_group.name, + monitor_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_all_traffic_filters_list(self, resource_group): + response = await self.client.elastic_monitor_resources.all_traffic_filters_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_list_associated_traffic_filters_list(self, resource_group): + response = await self.client.elastic_monitor_resources.list_associated_traffic_filters_list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_create_and_associate_ip_filter_create(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_create_and_associate_ip_filter_create( + resource_group_name=resource_group.name, + monitor_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_create_and_associate_pl_filter_create(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_create_and_associate_pl_filter_create( + resource_group_name=resource_group.name, + monitor_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_associate(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_associate( + resource_group_name=resource_group.name, + monitor_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_detach_and_delete_traffic_filter_delete(self, resource_group): + response = await self.client.elastic_monitor_resources.detach_and_delete_traffic_filter_delete( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_detach_traffic_filter_update(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_detach_traffic_filter_update( + resource_group_name=resource_group.name, + monitor_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_traffic_filters_delete(self, resource_group): + response = await self.client.elastic_monitor_resources.traffic_filters_delete( + resource_group_name=resource_group.name, + monitor_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_elastic_monitor_resources_begin_resubscribe(self, resource_group): + response = await ( + await self.client.elastic_monitor_resources.begin_resubscribe( + resource_group_name=resource_group.name, + monitor_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_elastic_versions_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_versions_operation_group_operations.py similarity index 69% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_elastic_versions_operations.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_versions_operation_group_operations.py index bc3f8467455f..ff01b18bcb97 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_elastic_versions_operations.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_versions_operation_group_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,16 +14,15 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticElasticVersionsOperations(AzureMgmtRecordedTestCase): +class TestElasticElasticVersionsOperationGroupOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) + self.client = self.create_mgmt_client(ElasticClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.elastic_versions.list( + def test_elastic_versions_operation_group_list(self, resource_group): + response = self.client.elastic_versions_operation_group.list( region="str", - api_version="2024-06-15-preview", ) result = [r for r in response] # please add some check logic here by yourself diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_elastic_versions_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_versions_operation_group_operations_async.py similarity index 69% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_elastic_versions_operations_async.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_versions_operation_group_operations_async.py index 4b092239c411..66936efb2763 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_elastic_versions_operations_async.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_elastic_versions_operation_group_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic.aio import MicrosoftElastic +from azure.mgmt.elastic.aio import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,16 +15,15 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticElasticVersionsOperationsAsync(AzureMgmtRecordedTestCase): +class TestElasticElasticVersionsOperationGroupOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) + self.client = self.create_mgmt_client(ElasticClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.elastic_versions.list( + async def test_elastic_versions_operation_group_list(self, resource_group): + response = self.client.elastic_versions_operation_group.list( region="str", - api_version="2024-06-15-preview", ) result = [r async for r in response] # please add some check logic here by yourself diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations.py new file mode 100644 index 000000000000..7363e195f2e7 --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations.py @@ -0,0 +1,144 @@ +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.elastic import ElasticClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestElasticMonitoredSubscriptionsOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ElasticClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_monitored_subscriptions_get(self, resource_group): + response = self.client.monitored_subscriptions.get( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_monitored_subscriptions_begin_createor_update(self, resource_group): + response = self.client.monitored_subscriptions.begin_createor_update( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "monitoredSubscriptionList": [ + { + "subscriptionId": "str", + "error": "str", + "status": "str", + "tagRules": { + "logRules": { + "filteringTags": [{"action": "str", "name": "str", "value": "str"}], + "sendAadLogs": bool, + "sendActivityLogs": bool, + "sendSubscriptionLogs": bool, + }, + "provisioningState": "str", + }, + } + ], + "operation": "str", + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_monitored_subscriptions_begin_update(self, resource_group): + response = self.client.monitored_subscriptions.begin_update( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "monitoredSubscriptionList": [ + { + "subscriptionId": "str", + "error": "str", + "status": "str", + "tagRules": { + "logRules": { + "filteringTags": [{"action": "str", "name": "str", "value": "str"}], + "sendAadLogs": bool, + "sendActivityLogs": bool, + "sendSubscriptionLogs": bool, + }, + "provisioningState": "str", + }, + } + ], + "operation": "str", + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_monitored_subscriptions_begin_delete(self, resource_group): + response = self.client.monitored_subscriptions.begin_delete( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_monitored_subscriptions_list(self, resource_group): + response = self.client.monitored_subscriptions.list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations_async.py new file mode 100644 index 000000000000..131d2080c3b3 --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_monitored_subscriptions_operations_async.py @@ -0,0 +1,151 @@ +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.elastic.aio import ElasticClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestElasticMonitoredSubscriptionsOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ElasticClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_monitored_subscriptions_get(self, resource_group): + response = await self.client.monitored_subscriptions.get( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_monitored_subscriptions_begin_createor_update(self, resource_group): + response = await ( + await self.client.monitored_subscriptions.begin_createor_update( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "monitoredSubscriptionList": [ + { + "subscriptionId": "str", + "error": "str", + "status": "str", + "tagRules": { + "logRules": { + "filteringTags": [{"action": "str", "name": "str", "value": "str"}], + "sendAadLogs": bool, + "sendActivityLogs": bool, + "sendSubscriptionLogs": bool, + }, + "provisioningState": "str", + }, + } + ], + "operation": "str", + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_monitored_subscriptions_begin_update(self, resource_group): + response = await ( + await self.client.monitored_subscriptions.begin_update( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "monitoredSubscriptionList": [ + { + "subscriptionId": "str", + "error": "str", + "status": "str", + "tagRules": { + "logRules": { + "filteringTags": [{"action": "str", "name": "str", "value": "str"}], + "sendAadLogs": bool, + "sendActivityLogs": bool, + "sendSubscriptionLogs": bool, + }, + "provisioningState": "str", + }, + } + ], + "operation": "str", + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_monitored_subscriptions_begin_delete(self, resource_group): + response = await ( + await self.client.monitored_subscriptions.begin_delete( + resource_group_name=resource_group.name, + monitor_name="str", + configuration_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_monitored_subscriptions_list(self, resource_group): + response = self.client.monitored_subscriptions.list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_open_ai_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_open_ai_integration_rp_models_operations.py similarity index 54% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_open_ai_operations.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_open_ai_integration_rp_models_operations.py index 03b3fa6a3b2a..388c97be0146 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_open_ai_operations.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_open_ai_integration_rp_models_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,30 +14,49 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticOpenAIOperations(AzureMgmtRecordedTestCase): +class TestElasticOpenAIIntegrationRPModelsOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) + self.client = self.create_mgmt_client(ElasticClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.open_ai.list( + def test_open_ai_integration_rp_models_get(self, resource_group): + response = self.client.open_ai_integration_rp_models.get( resource_group_name=resource_group.name, monitor_name="str", - api_version="2024-06-15-preview", + integration_name="str", ) - result = [r for r in response] + # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_get(self, resource_group): - response = self.client.open_ai.get( + def test_open_ai_integration_rp_models_create_or_update(self, resource_group): + response = self.client.open_ai_integration_rp_models.create_or_update( resource_group_name=resource_group.name, monitor_name="str", integration_name="str", - api_version="2024-06-15-preview", + body={ + "id": "str", + "name": "str", + "properties": { + "key": "str", + "lastRefreshAt": "2020-02-20 00:00:00", + "openAIConnectorId": "str", + "openAIResourceEndpoint": "str", + "openAIResourceId": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, ) # please add some check logic here by yourself @@ -45,12 +64,11 @@ def test_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_create_or_update(self, resource_group): - response = self.client.open_ai.create_or_update( + def test_open_ai_integration_rp_models_delete(self, resource_group): + response = self.client.open_ai_integration_rp_models.delete( resource_group_name=resource_group.name, monitor_name="str", integration_name="str", - api_version="2024-06-15-preview", ) # please add some check logic here by yourself @@ -58,25 +76,22 @@ def test_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_delete(self, resource_group): - response = self.client.open_ai.delete( + def test_open_ai_integration_rp_models_list(self, resource_group): + response = self.client.open_ai_integration_rp_models.list( resource_group_name=resource_group.name, monitor_name="str", - integration_name="str", - api_version="2024-06-15-preview", ) - + result = [r for r in response] # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_get_status(self, resource_group): - response = self.client.open_ai.get_status( + def test_open_ai_integration_rp_models_get_status(self, resource_group): + response = self.client.open_ai_integration_rp_models.get_status( resource_group_name=resource_group.name, monitor_name="str", integration_name="str", - api_version="2024-06-15-preview", ) # please add some check logic here by yourself diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_open_ai_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_open_ai_integration_rp_models_operations_async.py similarity index 54% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_open_ai_operations_async.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_open_ai_integration_rp_models_operations_async.py index 812b715a0c60..e181f3158c29 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_open_ai_operations_async.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_open_ai_integration_rp_models_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic.aio import MicrosoftElastic +from azure.mgmt.elastic.aio import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,30 +15,49 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticOpenAIOperationsAsync(AzureMgmtRecordedTestCase): +class TestElasticOpenAIIntegrationRPModelsOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) + self.client = self.create_mgmt_client(ElasticClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.open_ai.list( + async def test_open_ai_integration_rp_models_get(self, resource_group): + response = await self.client.open_ai_integration_rp_models.get( resource_group_name=resource_group.name, monitor_name="str", - api_version="2024-06-15-preview", + integration_name="str", ) - result = [r async for r in response] + # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_get(self, resource_group): - response = await self.client.open_ai.get( + async def test_open_ai_integration_rp_models_create_or_update(self, resource_group): + response = await self.client.open_ai_integration_rp_models.create_or_update( resource_group_name=resource_group.name, monitor_name="str", integration_name="str", - api_version="2024-06-15-preview", + body={ + "id": "str", + "name": "str", + "properties": { + "key": "str", + "lastRefreshAt": "2020-02-20 00:00:00", + "openAIConnectorId": "str", + "openAIResourceEndpoint": "str", + "openAIResourceId": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, ) # please add some check logic here by yourself @@ -46,12 +65,11 @@ async def test_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_create_or_update(self, resource_group): - response = await self.client.open_ai.create_or_update( + async def test_open_ai_integration_rp_models_delete(self, resource_group): + response = await self.client.open_ai_integration_rp_models.delete( resource_group_name=resource_group.name, monitor_name="str", integration_name="str", - api_version="2024-06-15-preview", ) # please add some check logic here by yourself @@ -59,25 +77,22 @@ async def test_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_delete(self, resource_group): - response = await self.client.open_ai.delete( + async def test_open_ai_integration_rp_models_list(self, resource_group): + response = self.client.open_ai_integration_rp_models.list( resource_group_name=resource_group.name, monitor_name="str", - integration_name="str", - api_version="2024-06-15-preview", ) - + result = [r async for r in response] # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_get_status(self, resource_group): - response = await self.client.open_ai.get_status( + async def test_open_ai_integration_rp_models_get_status(self, resource_group): + response = await self.client.open_ai_integration_rp_models.get_status( resource_group_name=resource_group.name, monitor_name="str", integration_name="str", - api_version="2024-06-15-preview", ) # please add some check logic here by yourself diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_operations.py similarity index 69% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_operations.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_operations.py index d476086e4f39..a307c8b250db 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_operations.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,16 +14,14 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticOperations(AzureMgmtRecordedTestCase): +class TestElasticOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) + self.client = self.create_mgmt_client(ElasticClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.operations.list( - api_version="2024-06-15-preview", - ) + def test_operations_list(self, resource_group): + response = self.client.operations.list() result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_operations_async.py similarity index 69% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_operations_async.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_operations_async.py index d61bd1cd7172..4e56671acce8 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_operations_async.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic.aio import MicrosoftElastic +from azure.mgmt.elastic.aio import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,16 +15,14 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticOperationsAsync(AzureMgmtRecordedTestCase): +class TestElasticOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) + self.client = self.create_mgmt_client(ElasticClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.operations.list( - api_version="2024-06-15-preview", - ) + async def test_operations_list(self, resource_group): + response = self.client.operations.list() result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_upgradable_versions_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_organizations_operation_group_operations.py similarity index 52% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_upgradable_versions_operations.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_organizations_operation_group_operations.py index 953d030f8991..f5980a687182 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_upgradable_versions_operations.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_organizations_operation_group_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,18 +14,24 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticUpgradableVersionsOperations(AzureMgmtRecordedTestCase): +class TestElasticOrganizationsOperationGroupOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) + self.client = self.create_mgmt_client(ElasticClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_details(self, resource_group): - response = self.client.upgradable_versions.details( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", + def test_organizations_operation_group_get_api_key(self, resource_group): + response = self.client.organizations_operation_group.get_api_key( + body={"emailId": "str"}, ) # please add some check logic here by yourself # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_organizations_operation_group_get_elastic_to_azure_subscription_mapping(self, resource_group): + response = self.client.organizations_operation_group.get_elastic_to_azure_subscription_mapping() + + # please add some check logic here by yourself + # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_upgradable_versions_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_organizations_operation_group_operations_async.py similarity index 52% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_upgradable_versions_operations_async.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_organizations_operation_group_operations_async.py index abea2df1aedf..0dff8394bfe3 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_upgradable_versions_operations_async.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_organizations_operation_group_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic.aio import MicrosoftElastic +from azure.mgmt.elastic.aio import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,18 +15,24 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticUpgradableVersionsOperationsAsync(AzureMgmtRecordedTestCase): +class TestElasticOrganizationsOperationGroupOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) + self.client = self.create_mgmt_client(ElasticClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_details(self, resource_group): - response = await self.client.upgradable_versions.details( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", + async def test_organizations_operation_group_get_api_key(self, resource_group): + response = await self.client.organizations_operation_group.get_api_key( + body={"emailId": "str"}, ) # please add some check logic here by yourself # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_organizations_operation_group_get_elastic_to_azure_subscription_mapping(self, resource_group): + response = await self.client.organizations_operation_group.get_elastic_to_azure_subscription_mapping() + + # please add some check logic here by yourself + # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_tag_rules_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_tag_rules_operations.py similarity index 61% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_tag_rules_operations.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_tag_rules_operations.py index a75af90ce3c6..3f60e265b74d 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_tag_rules_operations.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_tag_rules_operations.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic import MicrosoftElastic +from azure.mgmt.elastic import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,30 +14,51 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticTagRulesOperations(AzureMgmtRecordedTestCase): +class TestElasticTagRulesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) + self.client = self.create_mgmt_client(ElasticClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.tag_rules.list( + def test_tag_rules_get(self, resource_group): + response = self.client.tag_rules.get( resource_group_name=resource_group.name, monitor_name="str", - api_version="2024-06-15-preview", + rule_set_name="str", ) - result = [r for r in response] + # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_create_or_update(self, resource_group): + def test_tag_rules_create_or_update(self, resource_group): response = self.client.tag_rules.create_or_update( resource_group_name=resource_group.name, monitor_name="str", rule_set_name="str", - api_version="2024-06-15-preview", + body={ + "id": "str", + "name": "str", + "properties": { + "logRules": { + "filteringTags": [{"action": "str", "name": "str", "value": "str"}], + "sendAadLogs": bool, + "sendActivityLogs": bool, + "sendSubscriptionLogs": bool, + }, + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, ) # please add some check logic here by yourself @@ -45,26 +66,23 @@ def test_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_get(self, resource_group): - response = self.client.tag_rules.get( + def test_tag_rules_begin_delete(self, resource_group): + response = self.client.tag_rules.begin_delete( resource_group_name=resource_group.name, monitor_name="str", rule_set_name="str", - api_version="2024-06-15-preview", - ) + ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_begin_delete(self, resource_group): - response = self.client.tag_rules.begin_delete( + def test_tag_rules_list(self, resource_group): + response = self.client.tag_rules.list( resource_group_name=resource_group.name, monitor_name="str", - rule_set_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - + ) + result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_tag_rules_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_tag_rules_operations_async.py similarity index 62% rename from sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_tag_rules_operations_async.py rename to sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_tag_rules_operations_async.py index fd9075f04ec8..afff31e04f87 100644 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_tag_rules_operations_async.py +++ b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_elastic_tag_rules_operations_async.py @@ -2,11 +2,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.elastic.aio import MicrosoftElastic +from azure.mgmt.elastic.aio import ElasticClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,30 +15,17 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticTagRulesOperationsAsync(AzureMgmtRecordedTestCase): +class TestElasticTagRulesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) + self.client = self.create_mgmt_client(ElasticClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.tag_rules.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_create_or_update(self, resource_group): - response = await self.client.tag_rules.create_or_update( + async def test_tag_rules_get(self, resource_group): + response = await self.client.tag_rules.get( resource_group_name=resource_group.name, monitor_name="str", rule_set_name="str", - api_version="2024-06-15-preview", ) # please add some check logic here by yourself @@ -46,12 +33,33 @@ async def test_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_get(self, resource_group): - response = await self.client.tag_rules.get( + async def test_tag_rules_create_or_update(self, resource_group): + response = await self.client.tag_rules.create_or_update( resource_group_name=resource_group.name, monitor_name="str", rule_set_name="str", - api_version="2024-06-15-preview", + body={ + "id": "str", + "name": "str", + "properties": { + "logRules": { + "filteringTags": [{"action": "str", "name": "str", "value": "str"}], + "sendAadLogs": bool, + "sendActivityLogs": bool, + "sendSubscriptionLogs": bool, + }, + "provisioningState": "str", + }, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, ) # please add some check logic here by yourself @@ -59,15 +67,25 @@ async def test_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_begin_delete(self, resource_group): + async def test_tag_rules_begin_delete(self, resource_group): response = await ( await self.client.tag_rules.begin_delete( resource_group_name=resource_group.name, monitor_name="str", rule_set_name="str", - api_version="2024-06-15-preview", ) ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_tag_rules_list(self, resource_group): + response = self.client.tag_rules.list( + resource_group_name=resource_group.name, + monitor_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations.py deleted file mode 100644 index e4db0452e98f..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticAllTrafficFiltersOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.all_traffic_filters.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations_async.py deleted file mode 100644 index a3357ea7d5b6..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_all_traffic_filters_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticAllTrafficFiltersOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = await self.client.all_traffic_filters.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations.py deleted file mode 100644 index cb81bf7f2e82..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticAssociateTrafficFilterOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_associate(self, resource_group): - response = self.client.associate_traffic_filter.begin_associate( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations_async.py deleted file mode 100644 index 6873c474445c..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_associate_traffic_filter_operations_async.py +++ /dev/null @@ -1,34 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticAssociateTrafficFilterOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_associate(self, resource_group): - response = await ( - await self.client.associate_traffic_filter.begin_associate( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations.py deleted file mode 100644 index 7de79a9848b4..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticBillingInfoOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_get(self, resource_group): - response = self.client.billing_info.get( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations_async.py deleted file mode 100644 index 31247e4b6763..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_billing_info_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticBillingInfoOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_get(self, resource_group): - response = await self.client.billing_info.get( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations.py deleted file mode 100644 index fa57f45907a6..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticConnectedPartnerResourcesOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.connected_partner_resources.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations_async.py deleted file mode 100644 index b7563fb50da9..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_connected_partner_resources_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticConnectedPartnerResourcesOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.connected_partner_resources.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations.py deleted file mode 100644 index c8e0940e5ad3..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticCreateAndAssociateIPFilterOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_create(self, resource_group): - response = self.client.create_and_associate_ip_filter.begin_create( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations_async.py deleted file mode 100644 index 6f46465ad587..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_ip_filter_operations_async.py +++ /dev/null @@ -1,34 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticCreateAndAssociateIPFilterOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_create(self, resource_group): - response = await ( - await self.client.create_and_associate_ip_filter.begin_create( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations.py deleted file mode 100644 index 1c8d37c16dfe..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticCreateAndAssociatePLFilterOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_create(self, resource_group): - response = self.client.create_and_associate_pl_filter.begin_create( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations_async.py deleted file mode 100644 index 106640ce814f..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_create_and_associate_pl_filter_operations_async.py +++ /dev/null @@ -1,34 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticCreateAndAssociatePLFilterOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_create(self, resource_group): - response = await ( - await self.client.create_and_associate_pl_filter.begin_create( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations.py deleted file mode 100644 index 1c800b472472..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticDeploymentInfoOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.deployment_info.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations_async.py deleted file mode 100644 index 8aa7f9dbb4f0..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_deployment_info_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticDeploymentInfoOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = await self.client.deployment_info.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations.py deleted file mode 100644 index f65ec1a02ac0..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticDetachAndDeleteTrafficFilterOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_delete(self, resource_group): - response = self.client.detach_and_delete_traffic_filter.delete( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations_async.py deleted file mode 100644 index e908cf0c3f13..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_and_delete_traffic_filter_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticDetachAndDeleteTrafficFilterOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_delete(self, resource_group): - response = await self.client.detach_and_delete_traffic_filter.delete( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations.py deleted file mode 100644 index 46428fc636a1..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticDetachTrafficFilterOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_update(self, resource_group): - response = self.client.detach_traffic_filter.begin_update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations_async.py deleted file mode 100644 index dce5638c7df5..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_detach_traffic_filter_operations_async.py +++ /dev/null @@ -1,34 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticDetachTrafficFilterOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_update(self, resource_group): - response = await ( - await self.client.detach_traffic_filter.begin_update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations.py deleted file mode 100644 index f223f743532d..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticExternalUserOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_create_or_update(self, resource_group): - response = self.client.external_user.create_or_update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations_async.py deleted file mode 100644 index 1b2255806e51..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_external_user_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticExternalUserOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_create_or_update(self, resource_group): - response = await self.client.external_user.create_or_update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations.py deleted file mode 100644 index d2112aeca6db..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticListAssociatedTrafficFiltersOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.list_associated_traffic_filters.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations_async.py deleted file mode 100644 index 8a1607e917a7..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_list_associated_traffic_filters_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticListAssociatedTrafficFiltersOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = await self.client.list_associated_traffic_filters.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations.py deleted file mode 100644 index f2d187a15c3f..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitorOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_upgrade(self, resource_group): - response = self.client.monitor.begin_upgrade( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations_async.py deleted file mode 100644 index 6fecd4722342..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitor_operations_async.py +++ /dev/null @@ -1,34 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitorOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_upgrade(self, resource_group): - response = await ( - await self.client.monitor.begin_upgrade( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations.py deleted file mode 100644 index d48fda06cbf7..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitoredResourcesOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.monitored_resources.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations_async.py deleted file mode 100644 index 5fd3ed4e05d7..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_resources_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitoredResourcesOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.monitored_resources.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations.py deleted file mode 100644 index 7ac36dd351de..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations.py +++ /dev/null @@ -1,83 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitoredSubscriptionsOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.monitored_subscriptions.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_get(self, resource_group): - response = self.client.monitored_subscriptions.get( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_createor_update(self, resource_group): - response = self.client.monitored_subscriptions.begin_createor_update( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_update(self, resource_group): - response = self.client.monitored_subscriptions.begin_update( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_delete(self, resource_group): - response = self.client.monitored_subscriptions.begin_delete( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations_async.py deleted file mode 100644 index 0dc276dccafa..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitored_subscriptions_operations_async.py +++ /dev/null @@ -1,90 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitoredSubscriptionsOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.monitored_subscriptions.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_get(self, resource_group): - response = await self.client.monitored_subscriptions.get( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_createor_update(self, resource_group): - response = await ( - await self.client.monitored_subscriptions.begin_createor_update( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_update(self, resource_group): - response = await ( - await self.client.monitored_subscriptions.begin_update( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_delete(self, resource_group): - response = await ( - await self.client.monitored_subscriptions.begin_delete( - resource_group_name=resource_group.name, - monitor_name="str", - configuration_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations.py deleted file mode 100644 index 99b10372f793..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations.py +++ /dev/null @@ -1,88 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitorsOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.monitors.list( - api_version="2024-06-15-preview", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list_by_resource_group(self, resource_group): - response = self.client.monitors.list_by_resource_group( - resource_group_name=resource_group.name, - api_version="2024-06-15-preview", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_get(self, resource_group): - response = self.client.monitors.get( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_create(self, resource_group): - response = self.client.monitors.begin_create( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_update(self, resource_group): - response = self.client.monitors.begin_update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_delete(self, resource_group): - response = self.client.monitors.begin_delete( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations_async.py deleted file mode 100644 index c839ca62f464..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_monitors_operations_async.py +++ /dev/null @@ -1,95 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticMonitorsOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.monitors.list( - api_version="2024-06-15-preview", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list_by_resource_group(self, resource_group): - response = self.client.monitors.list_by_resource_group( - resource_group_name=resource_group.name, - api_version="2024-06-15-preview", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_get(self, resource_group): - response = await self.client.monitors.get( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_create(self, resource_group): - response = await ( - await self.client.monitors.begin_create( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_update(self, resource_group): - response = await ( - await self.client.monitors.begin_update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_delete(self, resource_group): - response = await ( - await self.client.monitors.begin_delete( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations.py deleted file mode 100644 index af42b24aedad..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations.py +++ /dev/null @@ -1,51 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticOrganizationsOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_get_api_key(self, resource_group): - response = self.client.organizations.get_api_key( - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_get_elastic_to_azure_subscription_mapping(self, resource_group): - response = self.client.organizations.get_elastic_to_azure_subscription_mapping( - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_begin_resubscribe(self, resource_group): - response = self.client.organizations.begin_resubscribe( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations_async.py deleted file mode 100644 index b73734e0416b..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_organizations_operations_async.py +++ /dev/null @@ -1,54 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticOrganizationsOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_get_api_key(self, resource_group): - response = await self.client.organizations.get_api_key( - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_get_elastic_to_azure_subscription_mapping(self, resource_group): - response = await self.client.organizations.get_elastic_to_azure_subscription_mapping( - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_begin_resubscribe(self, resource_group): - response = await ( - await self.client.organizations.begin_resubscribe( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations.py deleted file mode 100644 index 14ce49475bd8..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticTrafficFiltersOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_delete(self, resource_group): - response = self.client.traffic_filters.delete( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations_async.py deleted file mode 100644 index 59606a937e1e..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_traffic_filters_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticTrafficFiltersOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_delete(self, resource_group): - response = await self.client.traffic_filters.delete( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations.py deleted file mode 100644 index 483e36e3a77d..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticVMCollectionOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_update(self, resource_group): - response = self.client.vm_collection.update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations_async.py deleted file mode 100644 index 17f890e905b4..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_collection_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticVMCollectionOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_update(self, resource_group): - response = await self.client.vm_collection.update( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations.py deleted file mode 100644 index 2be8d092eff9..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticVMHostOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_list(self, resource_group): - response = self.client.vm_host.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations_async.py deleted file mode 100644 index 2cc969241443..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_host_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticVMHostOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_list(self, resource_group): - response = self.client.vm_host.list( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations.py deleted file mode 100644 index 220607428fce..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations.py +++ /dev/null @@ -1,31 +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 pytest -from azure.mgmt.elastic import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticVMIngestionOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_details(self, resource_group): - response = self.client.vm_ingestion.details( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations_async.py b/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations_async.py deleted file mode 100644 index d7a0911cef8f..000000000000 --- a/sdk/elastic/azure-mgmt-elastic/generated_tests/test_microsoft_elastic_vm_ingestion_operations_async.py +++ /dev/null @@ -1,32 +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 pytest -from azure.mgmt.elastic.aio import MicrosoftElastic - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMicrosoftElasticVMIngestionOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(MicrosoftElastic, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_details(self, resource_group): - response = await self.client.vm_ingestion.details( - resource_group_name=resource_group.name, - monitor_name="str", - api_version="2024-06-15-preview", - ) - - # please add some check logic here by yourself - # ... diff --git a/sdk/elastic/azure-mgmt-elastic/setup.py b/sdk/elastic/azure-mgmt-elastic/setup.py index 388f25f41b7a..f83915c86f5c 100644 --- a/sdk/elastic/azure-mgmt-elastic/setup.py +++ b/sdk/elastic/azure-mgmt-elastic/setup.py @@ -53,7 +53,6 @@ "Programming Language :: Python", "Programming Language :: Python :: 3 :: Only", "Programming Language :: Python :: 3", - "Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", @@ -77,7 +76,7 @@ "isodate>=0.6.1", "typing-extensions>=4.6.0", "azure-common>=1.1", - "azure-mgmt-core>=1.3.2", + "azure-mgmt-core>=1.5.0", ], - python_requires=">=3.8", + python_requires=">=3.9", ) diff --git a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_async_test.py b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_async_test.py index cc48bebaa8eb..c94d384637ea 100644 --- a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_async_test.py +++ b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_async_test.py @@ -27,4 +27,3 @@ async def test_list_by_resource_group(self, resource_group): ) result = [r async for r in response] assert result == [] - \ No newline at end of file diff --git a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_test.py b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_test.py index 7a077bc6e250..abf286bb1d9b 100644 --- a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_test.py +++ b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_monitors_operations_test.py @@ -26,4 +26,3 @@ def test_list_by_resource_group(self, resource_group): ) result = [r for r in response] assert result == [] - \ No newline at end of file diff --git a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_async_test.py b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_async_test.py index 30bf0dfda01c..fb5764f7c1e8 100644 --- a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_async_test.py +++ b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_async_test.py @@ -25,4 +25,3 @@ async def test_list(self, resource_group): response = self.client.operations.list() result = [r async for r in response] assert result - diff --git a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_test.py b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_test.py index 848cefdabc13..b3d705b311b5 100644 --- a/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_test.py +++ b/sdk/elastic/azure-mgmt-elastic/tests/test_microsoft_elastic_operations_test.py @@ -24,4 +24,3 @@ def test_list(self, resource_group): response = self.client.operations.list() result = [r for r in response] assert result - diff --git a/sdk/elastic/azure-mgmt-elastic/tsp-location.yaml b/sdk/elastic/azure-mgmt-elastic/tsp-location.yaml new file mode 100644 index 000000000000..341d765362a7 --- /dev/null +++ b/sdk/elastic/azure-mgmt-elastic/tsp-location.yaml @@ -0,0 +1,4 @@ +directory: specification/elastic/Elastic.Management +commit: 5c92dda155f958f8acd565f5ececf28dd940b560 +repo: Azure/azure-rest-api-specs +additionalDirectories: