From ba0d3321d02c16b90cac0a4c18eda4897c6d68e5 Mon Sep 17 00:00:00 2001 From: SDKAuto Date: Tue, 22 Oct 2024 10:43:37 +0000 Subject: [PATCH] CodeGen from PR 30931 in Azure/azure-rest-api-specs Merge 8c8fb6eb80c3fbe4cf0eb47f6fe074a569ef4bed into 6e95d1f68657c6b1abfae889b6a12fe12ce67eaa --- sdk/migrate/azure-mgmt-migrate/CHANGELOG.md | 5 + sdk/migrate/azure-mgmt-migrate/LICENSE | 21 + sdk/migrate/azure-mgmt-migrate/MANIFEST.in | 8 + sdk/migrate/azure-mgmt-migrate/README.md | 61 + sdk/migrate/azure-mgmt-migrate/_meta.json | 11 + .../azure-mgmt-migrate/azure/__init__.py | 1 + .../azure-mgmt-migrate/azure/mgmt/__init__.py | 1 + .../azure/mgmt/migrate/__init__.py | 26 + .../_azure_migrate_discovery_service.py | 472 + .../azure/mgmt/migrate/_configuration.py | 65 + .../azure/mgmt/migrate/_patch.py | 20 + .../azure/mgmt/migrate/_serialization.py | 2000 +++ .../azure/mgmt/migrate/_version.py | 9 + .../azure/mgmt/migrate/aio/__init__.py | 23 + .../aio/_azure_migrate_discovery_service.py | 485 + .../azure/mgmt/migrate/aio/_configuration.py | 65 + .../azure/mgmt/migrate/aio/_patch.py | 20 + .../mgmt/migrate/aio/operations/__init__.py | 127 + .../_dependency_map_controller_operations.py | 1057 ++ .../_hyperv_cluster_controller_operations.py | 483 + ...rv_dependency_map_controller_operations.py | 1256 ++ .../_hyperv_host_controller_operations.py | 541 + .../_hyperv_jobs_controller_operations.py | 200 + .../aio/operations/_hyperv_jobs_operations.py | 200 + .../_hyperv_machines_controller_operations.py | 572 + ...operations_status_controller_operations.py | 115 + ...v_run_as_accounts_controller_operations.py | 207 + .../_hyperv_sites_controller_operations.py | 1007 ++ .../operations/_hyperv_sites_operations.py | 217 + ...tware_inventories_controller_operations.py | 218 + ..._web_applications_controller_operations.py | 385 + .../_iis_web_servers_controller_operations.py | 229 + .../_import_jobs_controller_operations.py | 648 + .../_import_machines_controller_operations.py | 283 + .../_import_sites_controller_operations.py | 910 ++ .../_machines_controller_operations.py | 613 + .../_master_sites_controller_operations.py | 862 + ...operations_status_controller_operations.py | 117 + .../migrate/aio/operations/_operations.py | 130 + .../mgmt/migrate/aio/operations/_patch.py | 20 + ...dpoint_connection_controller_operations.py | 412 + ...te_link_resources_controller_operations.py | 205 + .../_run_as_accounts_controller_operations.py | 204 + ...er_dependency_map_controller_operations.py | 1057 ++ .../_server_jobs_controller_operations.py | 203 + ...operations_status_controller_operations.py | 117 + ...r_run_as_accounts_controller_operations.py | 205 + .../_server_sites_controller_operations.py | 1749 ++ ...tware_inventories_controller_operations.py | 218 + .../_servers_controller_operations.py | 426 + .../_sites_controller_operations.py | 1369 ++ ...ailability_groups_controller_operations.py | 239 + .../_sql_databases_controller_operations.py | 229 + ..._site_data_source_controller_operations.py | 516 + .../_sql_jobs_controller_operations.py | 212 + ...operations_status_controller_operations.py | 118 + ...l_run_as_accounts_controller_operations.py | 208 + .../_sql_servers_controller_operations.py | 385 + .../_sql_sites_controller_operations.py | 1376 ++ ..._web_applications_controller_operations.py | 313 + ...omcat_web_servers_controller_operations.py | 232 + .../_vcenter_controller_operations.py | 481 + .../_vmware_host_controller_operations.py | 200 + .../_vmware_operations_status_operations.py | 115 + ...vmware_properties_controller_operations.py | 859 + ...tware_inventories_controller_operations.py | 218 + ...site_data_sources_controller_operations.py | 579 + ...extended_machines_controller_operations.py | 239 + ...eb_app_properties_controller_operations.py | 270 + ...p_run_as_accounts_controller_operations.py | 211 + .../_web_app_sites_controller_operations.py | 1308 ++ ..._web_applications_controller_operations.py | 164 + .../_web_servers_controller_operations.py | 164 + .../azure/mgmt/migrate/models/__init__.py | 627 + .../_azure_migrate_discovery_service_enums.py | 654 + .../azure/mgmt/migrate/models/_models_py3.py | 13467 ++++++++++++++++ .../azure/mgmt/migrate/models/_patch.py | 20 + .../azure/mgmt/migrate/operations/__init__.py | 127 + .../_dependency_map_controller_operations.py | 1234 ++ .../_hyperv_cluster_controller_operations.py | 620 + ...rv_dependency_map_controller_operations.py | 1468 ++ .../_hyperv_host_controller_operations.py | 673 + .../_hyperv_jobs_controller_operations.py | 269 + .../operations/_hyperv_jobs_operations.py | 269 + .../_hyperv_machines_controller_operations.py | 725 + ...operations_status_controller_operations.py | 154 + ...v_run_as_accounts_controller_operations.py | 272 + .../_hyperv_sites_controller_operations.py | 1337 ++ .../operations/_hyperv_sites_operations.py | 271 + ...tware_inventories_controller_operations.py | 290 + ..._web_applications_controller_operations.py | 529 + .../_iis_web_servers_controller_operations.py | 328 + .../_import_jobs_controller_operations.py | 907 ++ .../_import_machines_controller_operations.py | 398 + .../_import_sites_controller_operations.py | 1194 ++ .../_machines_controller_operations.py | 797 + .../_master_sites_controller_operations.py | 1070 ++ ...operations_status_controller_operations.py | 154 + .../mgmt/migrate/operations/_operations.py | 152 + .../azure/mgmt/migrate/operations/_patch.py | 20 + ...dpoint_connection_controller_operations.py | 552 + ...te_link_resources_controller_operations.py | 273 + .../_run_as_accounts_controller_operations.py | 272 + ...er_dependency_map_controller_operations.py | 1234 ++ .../_server_jobs_controller_operations.py | 269 + ...operations_status_controller_operations.py | 154 + ...r_run_as_accounts_controller_operations.py | 271 + .../_server_sites_controller_operations.py | 2228 +++ ...tware_inventories_controller_operations.py | 290 + .../_servers_controller_operations.py | 577 + .../_sites_controller_operations.py | 1785 ++ ...ailability_groups_controller_operations.py | 332 + .../_sql_databases_controller_operations.py | 326 + ..._site_data_source_controller_operations.py | 672 + .../_sql_jobs_controller_operations.py | 283 + ...operations_status_controller_operations.py | 163 + ...l_run_as_accounts_controller_operations.py | 279 + .../_sql_servers_controller_operations.py | 519 + .../_sql_sites_controller_operations.py | 1721 ++ ..._web_applications_controller_operations.py | 458 + ...omcat_web_servers_controller_operations.py | 328 + .../_vcenter_controller_operations.py | 615 + .../_vmware_host_controller_operations.py | 269 + .../_vmware_operations_status_operations.py | 154 + ...vmware_properties_controller_operations.py | 1001 ++ ...tware_inventories_controller_operations.py | 290 + ...site_data_sources_controller_operations.py | 746 + ...extended_machines_controller_operations.py | 336 + ...eb_app_properties_controller_operations.py | 312 + ...p_run_as_accounts_controller_operations.py | 288 + .../_web_app_sites_controller_operations.py | 1633 ++ ..._web_applications_controller_operations.py | 222 + .../_web_servers_controller_operations.py | 222 + .../azure/mgmt/migrate/py.typed | 1 + ...er_client_group_members_maximum_set_gen.py | 50 + ...ler_export_dependencies_maximum_set_gen.py | 43 + ...ler_generate_coarse_map_maximum_set_gen.py | 47 + ...r_generate_detailed_map_maximum_set_gen.py | 48 + ...er_server_group_members_maximum_set_gen.py | 48 + ...ntroller_create_cluster_maximum_set_gen.py | 51 + ...uster_controller_delete_maximum_set_gen.py | 42 + ..._controller_get_cluster_maximum_set_gen.py | 43 + ...ler_list_by_hyperv_site_maximum_set_gen.py | 43 + ...er_client_group_members_maximum_set_gen.py | 50 + ...ler_export_dependencies_maximum_set_gen.py | 43 + ...ler_generate_coarse_map_maximum_set_gen.py | 47 + ...r_generate_detailed_map_maximum_set_gen.py | 48 + ...er_server_group_members_maximum_set_gen.py | 48 + ...e_dependency_map_status_maximum_set_gen.py | 43 + ..._host_controller_create_maximum_set_gen.py | 44 + ..._host_controller_delete_maximum_set_gen.py | 42 + ...erv_host_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_hyperv_site_maximum_set_gen.py | 43 + ...erv_jobs_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_vmware_site_maximum_set_gen.py | 43 + .../hyperv_jobs_get_maximum_set_gen.py | 43 + ...obs_list_by_hyperv_site_maximum_set_gen.py | 43 + ...machines_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_hyperv_site_maximum_set_gen.py | 43 + ...hines_controller_update_maximum_set_gen.py | 67 + ...oller_update_properties_maximum_set_gen.py | 51 + ...yperv_operations_status_maximum_set_gen.py | 43 + ...accounts_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_hyperv_site_maximum_set_gen.py | 43 + ...sites_controller_create_maximum_set_gen.py | 60 + ...sites_controller_delete_maximum_set_gen.py | 41 + ...r_export_machine_errors_maximum_set_gen.py | 43 + ...rv_sites_controller_get_maximum_set_gen.py | 42 + ...ler_list_health_summary_maximum_set_gen.py | 42 + ...ites_controller_summary_maximum_set_gen.py | 42 + ...sites_controller_update_maximum_set_gen.py | 59 + ..._list_by_resource_group_maximum_set_gen.py | 42 + ...es_list_by_subscription_maximum_set_gen.py | 40 + ...hine_software_inventory_maximum_set_gen.py | 44 + ..._list_by_hyperv_machine_maximum_set_gen.py | 44 + ...ications_controller_get_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ...tions_controller_update_maximum_set_gen.py | 45 + ..._servers_controller_get_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ...ontroller_get_deletejob_maximum_set_gen.py | 43 + ...ontroller_get_exportjob_maximum_set_gen.py | 43 + ...ontroller_get_importjob_maximum_set_gen.py | 43 + ...ort_jobs_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_import_site_maximum_set_gen.py | 43 + ...troller_list_deletejobs_maximum_set_gen.py | 43 + ...troller_list_exportjobs_maximum_set_gen.py | 43 + ...troller_list_importjobs_maximum_set_gen.py | 43 + ...hines_controller_delete_maximum_set_gen.py | 42 + ...machines_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_import_site_maximum_set_gen.py | 43 + ...sites_controller_create_maximum_set_gen.py | 47 + ...sites_controller_delete_maximum_set_gen.py | 41 + ...s_controller_export_uri_maximum_set_gen.py | 43 + ...rt_sites_controller_get_maximum_set_gen.py | 42 + ...s_controller_import_uri_maximum_set_gen.py | 43 + ..._list_by_resource_group_maximum_set_gen.py | 42 + ...er_list_by_subscription_maximum_set_gen.py | 40 + ...sites_controller_update_maximum_set_gen.py | 46 + ...machines_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_vmware_site_maximum_set_gen.py | 43 + ...chines_controller_start_maximum_set_gen.py | 43 + ...achines_controller_stop_maximum_set_gen.py | 43 + ...hines_controller_update_maximum_set_gen.py | 67 + ...sites_controller_create_maximum_set_gen.py | 53 + ...sites_controller_delete_maximum_set_gen.py | 41 + ...ontroller_error_summary_maximum_set_gen.py | 43 + ...er_sites_controller_get_maximum_set_gen.py | 42 + ..._list_by_resource_group_maximum_set_gen.py | 42 + ...er_list_by_subscription_maximum_set_gen.py | 40 + ...sites_controller_update_maximum_set_gen.py | 51 + ...vmware_operation_status_maximum_set_gen.py | 43 + .../operations_list_maximum_set_gen.py | 40 + ...ction_controller_create_maximum_set_gen.py | 54 + ...ction_controller_delete_maximum_set_gen.py | 42 + ...nnection_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_master_site_maximum_set_gen.py | 43 + ...esources_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_master_site_maximum_set_gen.py | 43 + ...accounts_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_vmware_site_maximum_set_gen.py | 43 + ...er_client_group_members_maximum_set_gen.py | 50 + ...ler_export_dependencies_maximum_set_gen.py | 43 + ...ler_generate_coarse_map_maximum_set_gen.py | 47 + ...r_generate_detailed_map_maximum_set_gen.py | 48 + ...er_server_group_members_maximum_set_gen.py | 48 + ...ver_jobs_controller_get_maximum_set_gen.py | 43 + ...by_server_site_resource_maximum_set_gen.py | 43 + ..._site_operations_status_maximum_set_gen.py | 43 + ...accounts_controller_get_maximum_set_gen.py | 43 + ...by_server_site_resource_maximum_set_gen.py | 43 + ...sites_controller_create_maximum_set_gen.py | 60 + ...sites_controller_delete_maximum_set_gen.py | 41 + ...r_export_machine_errors_maximum_set_gen.py | 43 + ...er_sites_controller_get_maximum_set_gen.py | 42 + ..._list_by_resource_group_maximum_set_gen.py | 42 + ...er_list_by_subscription_maximum_set_gen.py | 40 + ...ler_list_health_summary_maximum_set_gen.py | 42 + ...controller_refresh_site_maximum_set_gen.py | 42 + ...ites_controller_summary_maximum_set_gen.py | 42 + ...e_dependency_map_status_maximum_set_gen.py | 43 + ...sites_controller_update_maximum_set_gen.py | 58 + ...oller_update_properties_maximum_set_gen.py | 51 + ...hine_software_inventory_maximum_set_gen.py | 44 + ...ntroller_list_by_server_maximum_set_gen.py | 44 + ...ntroller_delete_machine_maximum_set_gen.py | 42 + ..._controller_get_machine_maximum_set_gen.py | 43 + ...by_server_site_resource_maximum_set_gen.py | 43 + ...ntroller_update_machine_maximum_set_gen.py | 90 + ...sites_controller_create_maximum_set_gen.py | 60 + ...sites_controller_delete_maximum_set_gen.py | 41 + ...r_export_machine_errors_maximum_set_gen.py | 43 + ...troller_export_machines_maximum_set_gen.py | 43 + .../sites_controller_get_maximum_set_gen.py | 42 + ..._list_by_resource_group_maximum_set_gen.py | 42 + ...er_list_by_subscription_maximum_set_gen.py | 40 + ...ler_list_health_summary_maximum_set_gen.py | 42 + ...ites_controller_summary_maximum_set_gen.py | 42 + ...sites_controller_update_maximum_set_gen.py | 59 + ...y_groups_controller_get_maximum_set_gen.py | 44 + ...roller_list_by_sql_site_maximum_set_gen.py | 44 + ...atabases_controller_get_maximum_set_gen.py | 44 + ...roller_list_by_sql_site_maximum_set_gen.py | 44 + ...ource_controller_create_maximum_set_gen.py | 45 + ...ource_controller_delete_maximum_set_gen.py | 43 + ...a_source_controller_get_maximum_set_gen.py | 44 + ...roller_list_by_sql_site_maximum_set_gen.py | 44 + ...sql_jobs_controller_get_maximum_set_gen.py | 44 + ...roller_list_by_sql_site_maximum_set_gen.py | 44 + ...et_sql_operation_status_maximum_set_gen.py | 44 + ...accounts_controller_get_maximum_set_gen.py | 44 + ...roller_list_by_sql_site_maximum_set_gen.py | 44 + ..._servers_controller_get_maximum_set_gen.py | 44 + ...roller_list_by_sql_site_maximum_set_gen.py | 44 + ...rvers_controller_update_maximum_set_gen.py | 88 + ...sites_controller_create_maximum_set_gen.py | 63 + ...sites_controller_delete_maximum_set_gen.py | 42 + ...ontroller_error_summary_maximum_set_gen.py | 44 + ...ller_export_sql_servers_maximum_set_gen.py | 44 + ...ql_sites_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_master_site_maximum_set_gen.py | 43 + ...ites_controller_refresh_maximum_set_gen.py | 44 + ...ites_controller_summary_maximum_set_gen.py | 43 + ...sites_controller_update_maximum_set_gen.py | 62 + ...ications_controller_get_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ..._servers_controller_get_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ...enter_controller_create_maximum_set_gen.py | 52 + ...enter_controller_delete_maximum_set_gen.py | 42 + .../vcenter_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_vmware_site_maximum_set_gen.py | 43 + ...are_host_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_vmware_site_maximum_set_gen.py | 43 + ...vmware_operation_status_maximum_set_gen.py | 43 + ...e_dependency_map_status_maximum_set_gen.py | 45 + ...oller_update_properties_maximum_set_gen.py | 51 + ..._controller_update_tags_maximum_set_gen.py | 43 + ...hine_software_inventory_maximum_set_gen.py | 44 + ...ist_by_machine_resource_maximum_set_gen.py | 44 + ...urces_controller_create_maximum_set_gen.py | 45 + ...urces_controller_delete_maximum_set_gen.py | 43 + ..._sources_controller_get_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ...machines_controller_get_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ...oller_update_properties_maximum_set_gen.py | 44 + ...accounts_controller_get_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ...sites_controller_create_maximum_set_gen.py | 63 + ...sites_controller_delete_maximum_set_gen.py | 42 + ...ontroller_error_summary_maximum_set_gen.py | 44 + ...roller_export_inventory_maximum_set_gen.py | 44 + ...pp_sites_controller_get_maximum_set_gen.py | 43 + ...ler_list_by_master_site_maximum_set_gen.py | 43 + ...ites_controller_refresh_maximum_set_gen.py | 44 + ...ites_controller_summary_maximum_set_gen.py | 43 + ...sites_controller_update_maximum_set_gen.py | 62 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + ...er_list_by_web_app_site_maximum_set_gen.py | 44 + .../generated_tests/conftest.py | 47 + ...ce_dependency_map_controller_operations.py | 105 + ...endency_map_controller_operations_async.py | 116 + ...ce_hyperv_cluster_controller_operations.py | 110 + ...erv_cluster_controller_operations_async.py | 113 + ...rv_dependency_map_controller_operations.py | 118 + ...endency_map_controller_operations_async.py | 131 + ...rvice_hyperv_host_controller_operations.py | 108 + ...hyperv_host_controller_operations_async.py | 113 + ...rvice_hyperv_jobs_controller_operations.py | 44 + ...hyperv_jobs_controller_operations_async.py | 45 + ...iscovery_service_hyperv_jobs_operations.py | 44 + ...ry_service_hyperv_jobs_operations_async.py | 45 + ...e_hyperv_machines_controller_operations.py | 94 + ...rv_machines_controller_operations_async.py | 97 + ...operations_status_controller_operations.py | 32 + ...ions_status_controller_operations_async.py | 33 + ...v_run_as_accounts_controller_operations.py | 44 + ...as_accounts_controller_operations_async.py | 45 + ...vice_hyperv_sites_controller_operations.py | 203 + ...yperv_sites_controller_operations_async.py | 208 + ...scovery_service_hyperv_sites_operations.py | 40 + ...y_service_hyperv_sites_operations_async.py | 41 + ...tware_inventories_controller_operations.py | 46 + ...inventories_controller_operations_async.py | 47 + ..._web_applications_controller_operations.py | 61 + ...pplications_controller_operations_async.py | 62 + ...e_iis_web_servers_controller_operations.py | 46 + ...web_servers_controller_operations_async.py | 47 + ...rvice_import_jobs_controller_operations.py | 119 + ...import_jobs_controller_operations_async.py | 120 + ...e_import_machines_controller_operations.py | 57 + ...rt_machines_controller_operations_async.py | 58 + ...vice_import_sites_controller_operations.py | 149 + ...mport_sites_controller_operations_async.py | 150 + ..._service_machines_controller_operations.py | 107 + ...ce_machines_controller_operations_async.py | 112 + ...vice_master_sites_controller_operations.py | 158 + ...aster_sites_controller_operations_async.py | 163 + ...operations_status_controller_operations.py | 32 + ...ions_status_controller_operations_async.py | 33 + ...re_migrate_discovery_service_operations.py | 29 + ...rate_discovery_service_operations_async.py | 30 + ...dpoint_connection_controller_operations.py | 93 + ..._connection_controller_operations_async.py | 94 + ...te_link_resources_controller_operations.py | 44 + ...k_resources_controller_operations_async.py | 45 + ...e_run_as_accounts_controller_operations.py | 44 + ...as_accounts_controller_operations_async.py | 45 + ...er_dependency_map_controller_operations.py | 105 + ...endency_map_controller_operations_async.py | 116 + ...rvice_server_jobs_controller_operations.py | 44 + ...server_jobs_controller_operations_async.py | 45 + ...operations_status_controller_operations.py | 32 + ...ions_status_controller_operations_async.py | 33 + ...r_run_as_accounts_controller_operations.py | 44 + ...as_accounts_controller_operations_async.py | 45 + ...vice_server_sites_controller_operations.py | 261 + ...erver_sites_controller_operations_async.py | 274 + ...tware_inventories_controller_operations.py | 46 + ...inventories_controller_operations_async.py | 47 + ...y_service_servers_controller_operations.py | 118 + ...ice_servers_controller_operations_async.py | 119 + ...ery_service_sites_controller_operations.py | 238 + ...rvice_sites_controller_operations_async.py | 245 + ...ailability_groups_controller_operations.py | 46 + ...lity_groups_controller_operations_async.py | 47 + ...ice_sql_databases_controller_operations.py | 46 + ...l_databases_controller_operations_async.py | 47 + ..._site_data_source_controller_operations.py | 88 + ...data_source_controller_operations_async.py | 91 + ..._service_sql_jobs_controller_operations.py | 46 + ...ce_sql_jobs_controller_operations_async.py | 47 + ...operations_status_controller_operations.py | 33 + ...ions_status_controller_operations_async.py | 34 + ...l_run_as_accounts_controller_operations.py | 46 + ...as_accounts_controller_operations_async.py | 47 + ...rvice_sql_servers_controller_operations.py | 104 + ...sql_servers_controller_operations_async.py | 105 + ...service_sql_sites_controller_operations.py | 215 + ...e_sql_sites_controller_operations_async.py | 224 + ..._web_applications_controller_operations.py | 61 + ...pplications_controller_operations_async.py | 62 + ...omcat_web_servers_controller_operations.py | 46 + ...web_servers_controller_operations_async.py | 47 + ...y_service_vcenter_controller_operations.py | 112 + ...ice_vcenter_controller_operations_async.py | 115 + ...rvice_vmware_host_controller_operations.py | 44 + ...vmware_host_controller_operations_async.py | 45 + ...ice_vmware_operations_status_operations.py | 32 + ...ware_operations_status_operations_async.py | 33 + ...vmware_properties_controller_operations.py | 71 + ..._properties_controller_operations_async.py | 80 + ...tware_inventories_controller_operations.py | 46 + ...inventories_controller_operations_async.py | 47 + ...site_data_sources_controller_operations.py | 88 + ...ata_sources_controller_operations_async.py | 95 + ...extended_machines_controller_operations.py | 46 + ...ed_machines_controller_operations_async.py | 47 + ...eb_app_properties_controller_operations.py | 33 + ..._properties_controller_operations_async.py | 36 + ...p_run_as_accounts_controller_operations.py | 46 + ...as_accounts_controller_operations_async.py | 47 + ...ice_web_app_sites_controller_operations.py | 201 + ...b_app_sites_controller_operations_async.py | 210 + ..._web_applications_controller_operations.py | 32 + ...pplications_controller_operations_async.py | 33 + ...rvice_web_servers_controller_operations.py | 32 + ...web_servers_controller_operations_async.py | 33 + .../azure-mgmt-migrate/sdk_packaging.toml | 11 + sdk/migrate/azure-mgmt-migrate/setup.py | 83 + sdk/migrate/ci.yml | 34 + 432 files changed, 93037 insertions(+) create mode 100644 sdk/migrate/azure-mgmt-migrate/CHANGELOG.md create mode 100644 sdk/migrate/azure-mgmt-migrate/LICENSE create mode 100644 sdk/migrate/azure-mgmt-migrate/MANIFEST.in create mode 100644 sdk/migrate/azure-mgmt-migrate/README.md create mode 100644 sdk/migrate/azure-mgmt-migrate/_meta.json create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/__init__.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/__init__.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/__init__.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_azure_migrate_discovery_service.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_configuration.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_patch.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_serialization.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_version.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/__init__.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_azure_migrate_discovery_service.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_configuration.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_patch.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/__init__.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_cluster_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_host_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_patch.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_endpoint_connection_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_link_resources_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_availability_groups_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_databases_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_discovery_site_data_source_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vcenter_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_host_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_operations_status_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_properties_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_discovery_site_data_sources_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_extended_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_properties_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/__init__.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_azure_migrate_discovery_service_enums.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_models_py3.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_patch.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/__init__.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_cluster_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_host_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_patch.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_endpoint_connection_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_link_resources_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_availability_groups_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_databases_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_discovery_site_data_source_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vcenter_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_host_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_operations_status_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_properties_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_discovery_site_data_sources_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_extended_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_properties_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/py.typed create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_client_group_members_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_export_dependencies_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_coarse_map_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_detailed_map_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_server_group_members_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_create_cluster_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_get_cluster_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_list_by_hyperv_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_client_group_members_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_export_dependencies_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_coarse_map_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_detailed_map_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_server_group_members_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_update_dependency_map_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_list_by_hyperv_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_list_by_vmware_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_list_by_hyperv_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_list_by_hyperv_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_properties_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_operations_status_controller_get_hyperv_operations_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_list_by_hyperv_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_export_machine_errors_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_list_health_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_resource_group_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_subscription_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_list_by_hyperv_machine_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_deletejob_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_exportjob_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_importjob_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_by_import_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_deletejobs_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_exportjobs_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_importjobs_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_list_by_import_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_export_uri_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_import_uri_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_resource_group_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_subscription_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_list_by_vmware_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_start_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_stop_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_error_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_resource_group_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_subscription_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_operations_status_controller_get_vmware_operation_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/operations_list_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_list_by_master_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_list_by_master_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_list_by_vmware_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_client_group_members_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_export_dependencies_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_coarse_map_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_detailed_map_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_server_group_members_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_list_by_server_site_resource_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_operations_status_controller_get_server_site_operations_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_list_by_server_site_resource_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_export_machine_errors_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_resource_group_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_subscription_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_health_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_refresh_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_dependency_map_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_properties_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_list_by_server_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_delete_machine_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_get_machine_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_list_by_server_site_resource_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_update_machine_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machine_errors_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machines_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_resource_group_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_subscription_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_health_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_list_by_sql_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_list_by_sql_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_list_by_sql_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_list_by_sql_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_operations_status_controller_get_sql_operation_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_list_by_sql_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_list_by_sql_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_error_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_export_sql_servers_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_list_by_master_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_refresh_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_list_by_vmware_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_list_by_vmware_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_operations_status_get_vmware_operation_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_dependency_map_status_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_properties_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_tags_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_list_by_machine_resource_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_properties_controller_update_properties_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_create_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_delete_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_error_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_export_inventory_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_get_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_list_by_master_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_refresh_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_summary_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_update_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_applications_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_samples/web_servers_controller_list_by_web_app_site_maximum_set_gen.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/conftest.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations.py create mode 100644 sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations_async.py create mode 100644 sdk/migrate/azure-mgmt-migrate/sdk_packaging.toml create mode 100644 sdk/migrate/azure-mgmt-migrate/setup.py create mode 100644 sdk/migrate/ci.yml diff --git a/sdk/migrate/azure-mgmt-migrate/CHANGELOG.md b/sdk/migrate/azure-mgmt-migrate/CHANGELOG.md new file mode 100644 index 000000000000..578ed6acf479 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/CHANGELOG.md @@ -0,0 +1,5 @@ +# Release History + +## 0.1.0 (1970-01-01) + +* Initial Release diff --git a/sdk/migrate/azure-mgmt-migrate/LICENSE b/sdk/migrate/azure-mgmt-migrate/LICENSE new file mode 100644 index 000000000000..b2f52a2bad4e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/LICENSE @@ -0,0 +1,21 @@ +Copyright (c) Microsoft Corporation. + +MIT License + +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. diff --git a/sdk/migrate/azure-mgmt-migrate/MANIFEST.in b/sdk/migrate/azure-mgmt-migrate/MANIFEST.in new file mode 100644 index 000000000000..0e40dbbfc8e5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/MANIFEST.in @@ -0,0 +1,8 @@ +include _meta.json +recursive-include tests *.py *.json +recursive-include samples *.py *.md +include *.md +include azure/__init__.py +include azure/mgmt/__init__.py +include LICENSE +include azure/mgmt/migrate/py.typed diff --git a/sdk/migrate/azure-mgmt-migrate/README.md b/sdk/migrate/azure-mgmt-migrate/README.md new file mode 100644 index 000000000000..6bff49c0344a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/README.md @@ -0,0 +1,61 @@ +# Microsoft Azure SDK for Python + +This is the Microsoft Azure MyService Management Client Library. +This package has been tested with Python 3.8+. +For a more complete view of Azure libraries, see the [azure sdk python release](https://aka.ms/azsdk/python/all). + +## _Disclaimer_ + +_Azure SDK Python packages support for Python 2.7 has ended 01 January 2022. For more information and questions, please refer to https://github.com/Azure/azure-sdk-for-python/issues/20691_ + +## Getting started + +### Prerequisites + +- Python 3.8+ is required to use this package. +- [Azure subscription](https://azure.microsoft.com/free/) + +### Install the package + +```bash +pip install azure-mgmt-migrate +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. + +- `AZURE_CLIENT_ID` for Azure client ID. +- `AZURE_TENANT_ID` for Azure tenant ID. +- `AZURE_CLIENT_SECRET` for Azure client secret. + +In addition, Azure subscription ID can be configured via environment variable `AZURE_SUBSCRIPTION_ID`. + +With above configuration, client can be authenticated by following code: + +```python +from azure.identity import DefaultAzureCredential +from azure.mgmt.migrate import +import os + +sub_id = os.getenv("AZURE_SUBSCRIPTION_ID") +client = (credential=DefaultAzureCredential(), subscription_id=sub_id) +``` + +## Examples + +Code samples for this package can be found at: +- [Search MyService Management](https://docs.microsoft.com/samples/browse/?languages=python&term=Getting%20started%20-%20Managing&terms=Getting%20started%20-%20Managing) on docs.microsoft.com +- [Azure Python Mgmt SDK Samples Repo](https://aka.ms/azsdk/python/mgmt/samples) + + +## Troubleshooting + +## Next steps + +## Provide Feedback + +If you encounter any bugs or have suggestions, please file an issue in the +[Issues](https://github.com/Azure/azure-sdk-for-python/issues) +section of the project. diff --git a/sdk/migrate/azure-mgmt-migrate/_meta.json b/sdk/migrate/azure-mgmt-migrate/_meta.json new file mode 100644 index 000000000000..7f0ce2798ef0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/_meta.json @@ -0,0 +1,11 @@ +{ + "commit": "efc9894a519b31e579ec928c02faba0d0141935b", + "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/migrate/resource-manager/readme.md --generate-sample=True --generate-test=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/mnt/vss/_work/1/s/azure-sdk-for-python/sdk --use=@autorest/python@6.19.0 --use=@autorest/modelerfour@4.27.0 --version=3.10.2 --version-tolerant=False", + "readme": "specification/migrate/resource-manager/readme.md" +} \ No newline at end of file diff --git a/sdk/migrate/azure-mgmt-migrate/azure/__init__.py b/sdk/migrate/azure-mgmt-migrate/azure/__init__.py new file mode 100644 index 000000000000..8db66d3d0f0f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/__init__.py @@ -0,0 +1 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/__init__.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/__init__.py new file mode 100644 index 000000000000..8db66d3d0f0f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/__init__.py @@ -0,0 +1 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/__init__.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/__init__.py new file mode 100644 index 000000000000..0399e981e6ae --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/__init__.py @@ -0,0 +1,26 @@ +# 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_migrate_discovery_service import AzureMigrateDiscoveryService +from ._version import VERSION + +__version__ = VERSION + +try: + from ._patch import __all__ as _patch_all + from ._patch import * # pylint: disable=unused-wildcard-import +except ImportError: + _patch_all = [] +from ._patch import patch_sdk as _patch_sdk + +__all__ = [ + "AzureMigrateDiscoveryService", +] +__all__.extend([p for p in _patch_all if p not in __all__]) + +_patch_sdk() diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_azure_migrate_discovery_service.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_azure_migrate_discovery_service.py new file mode 100644 index 000000000000..c357d6188968 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_azure_migrate_discovery_service.py @@ -0,0 +1,472 @@ +# 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 AzureMigrateDiscoveryServiceConfiguration +from ._serialization import Deserializer, Serializer +from .operations import ( + DependencyMapControllerOperations, + HypervClusterControllerOperations, + HypervDependencyMapControllerOperations, + HypervHostControllerOperations, + HypervJobsControllerOperations, + HypervJobsOperations, + HypervMachinesControllerOperations, + HypervOperationsStatusControllerOperations, + HypervRunAsAccountsControllerOperations, + HypervSitesControllerOperations, + HypervSitesOperations, + HypervSoftwareInventoriesControllerOperations, + IisWebApplicationsControllerOperations, + IisWebServersControllerOperations, + ImportJobsControllerOperations, + ImportMachinesControllerOperations, + ImportSitesControllerOperations, + MachinesControllerOperations, + MasterSitesControllerOperations, + MasterSitesOperationsStatusControllerOperations, + Operations, + PrivateEndpointConnectionControllerOperations, + PrivateLinkResourcesControllerOperations, + RunAsAccountsControllerOperations, + ServerDependencyMapControllerOperations, + ServerJobsControllerOperations, + ServerOperationsStatusControllerOperations, + ServerRunAsAccountsControllerOperations, + ServerSitesControllerOperations, + ServerSoftwareInventoriesControllerOperations, + ServersControllerOperations, + SitesControllerOperations, + SqlAvailabilityGroupsControllerOperations, + SqlDatabasesControllerOperations, + SqlDiscoverySiteDataSourceControllerOperations, + SqlJobsControllerOperations, + SqlOperationsStatusControllerOperations, + SqlRunAsAccountsControllerOperations, + SqlServersControllerOperations, + SqlSitesControllerOperations, + TomcatWebApplicationsControllerOperations, + TomcatWebServersControllerOperations, + VcenterControllerOperations, + VmwareHostControllerOperations, + VmwareOperationsStatusOperations, + VmwarePropertiesControllerOperations, + VmwareSoftwareInventoriesControllerOperations, + WebAppDiscoverySiteDataSourcesControllerOperations, + WebAppExtendedMachinesControllerOperations, + WebAppPropertiesControllerOperations, + WebAppRunAsAccountsControllerOperations, + WebAppSitesControllerOperations, + WebApplicationsControllerOperations, + WebServersControllerOperations, +) + +if TYPE_CHECKING: + # pylint: disable=unused-import,ungrouped-imports + from azure.core.credentials import TokenCredential + + +class AzureMigrateDiscoveryService: # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes + """Azure OffAzure Resource Provider management API. + + :ivar operations: Operations operations + :vartype operations: azure.mgmt.migrate.operations.Operations + :ivar hyperv_sites: HypervSitesOperations operations + :vartype hyperv_sites: azure.mgmt.migrate.operations.HypervSitesOperations + :ivar import_sites_controller: ImportSitesControllerOperations operations + :vartype import_sites_controller: azure.mgmt.migrate.operations.ImportSitesControllerOperations + :ivar master_sites_controller: MasterSitesControllerOperations operations + :vartype master_sites_controller: azure.mgmt.migrate.operations.MasterSitesControllerOperations + :ivar server_sites_controller: ServerSitesControllerOperations operations + :vartype server_sites_controller: azure.mgmt.migrate.operations.ServerSitesControllerOperations + :ivar sites_controller: SitesControllerOperations operations + :vartype sites_controller: azure.mgmt.migrate.operations.SitesControllerOperations + :ivar hyperv_sites_controller: HypervSitesControllerOperations operations + :vartype hyperv_sites_controller: azure.mgmt.migrate.operations.HypervSitesControllerOperations + :ivar hyperv_dependency_map_controller: HypervDependencyMapControllerOperations operations + :vartype hyperv_dependency_map_controller: + azure.mgmt.migrate.operations.HypervDependencyMapControllerOperations + :ivar hyperv_cluster_controller: HypervClusterControllerOperations operations + :vartype hyperv_cluster_controller: + azure.mgmt.migrate.operations.HypervClusterControllerOperations + :ivar hyperv_host_controller: HypervHostControllerOperations operations + :vartype hyperv_host_controller: azure.mgmt.migrate.operations.HypervHostControllerOperations + :ivar hyperv_jobs: HypervJobsOperations operations + :vartype hyperv_jobs: azure.mgmt.migrate.operations.HypervJobsOperations + :ivar hyperv_machines_controller: HypervMachinesControllerOperations operations + :vartype hyperv_machines_controller: + azure.mgmt.migrate.operations.HypervMachinesControllerOperations + :ivar hyperv_software_inventories_controller: HypervSoftwareInventoriesControllerOperations + operations + :vartype hyperv_software_inventories_controller: + azure.mgmt.migrate.operations.HypervSoftwareInventoriesControllerOperations + :ivar hyperv_operations_status_controller: HypervOperationsStatusControllerOperations + operations + :vartype hyperv_operations_status_controller: + azure.mgmt.migrate.operations.HypervOperationsStatusControllerOperations + :ivar hyperv_run_as_accounts_controller: HypervRunAsAccountsControllerOperations operations + :vartype hyperv_run_as_accounts_controller: + azure.mgmt.migrate.operations.HypervRunAsAccountsControllerOperations + :ivar import_jobs_controller: ImportJobsControllerOperations operations + :vartype import_jobs_controller: azure.mgmt.migrate.operations.ImportJobsControllerOperations + :ivar import_machines_controller: ImportMachinesControllerOperations operations + :vartype import_machines_controller: + azure.mgmt.migrate.operations.ImportMachinesControllerOperations + :ivar master_sites_operations_status_controller: + MasterSitesOperationsStatusControllerOperations operations + :vartype master_sites_operations_status_controller: + azure.mgmt.migrate.operations.MasterSitesOperationsStatusControllerOperations + :ivar private_endpoint_connection_controller: PrivateEndpointConnectionControllerOperations + operations + :vartype private_endpoint_connection_controller: + azure.mgmt.migrate.operations.PrivateEndpointConnectionControllerOperations + :ivar private_link_resources_controller: PrivateLinkResourcesControllerOperations operations + :vartype private_link_resources_controller: + azure.mgmt.migrate.operations.PrivateLinkResourcesControllerOperations + :ivar sql_sites_controller: SqlSitesControllerOperations operations + :vartype sql_sites_controller: azure.mgmt.migrate.operations.SqlSitesControllerOperations + :ivar sql_discovery_site_data_source_controller: SqlDiscoverySiteDataSourceControllerOperations + operations + :vartype sql_discovery_site_data_source_controller: + azure.mgmt.migrate.operations.SqlDiscoverySiteDataSourceControllerOperations + :ivar sql_jobs_controller: SqlJobsControllerOperations operations + :vartype sql_jobs_controller: azure.mgmt.migrate.operations.SqlJobsControllerOperations + :ivar sql_operations_status_controller: SqlOperationsStatusControllerOperations operations + :vartype sql_operations_status_controller: + azure.mgmt.migrate.operations.SqlOperationsStatusControllerOperations + :ivar sql_run_as_accounts_controller: SqlRunAsAccountsControllerOperations operations + :vartype sql_run_as_accounts_controller: + azure.mgmt.migrate.operations.SqlRunAsAccountsControllerOperations + :ivar sql_availability_groups_controller: SqlAvailabilityGroupsControllerOperations operations + :vartype sql_availability_groups_controller: + azure.mgmt.migrate.operations.SqlAvailabilityGroupsControllerOperations + :ivar sql_databases_controller: SqlDatabasesControllerOperations operations + :vartype sql_databases_controller: + azure.mgmt.migrate.operations.SqlDatabasesControllerOperations + :ivar sql_servers_controller: SqlServersControllerOperations operations + :vartype sql_servers_controller: azure.mgmt.migrate.operations.SqlServersControllerOperations + :ivar web_app_sites_controller: WebAppSitesControllerOperations operations + :vartype web_app_sites_controller: + azure.mgmt.migrate.operations.WebAppSitesControllerOperations + :ivar web_app_discovery_site_data_sources_controller: + WebAppDiscoverySiteDataSourcesControllerOperations operations + :vartype web_app_discovery_site_data_sources_controller: + azure.mgmt.migrate.operations.WebAppDiscoverySiteDataSourcesControllerOperations + :ivar web_app_extended_machines_controller: WebAppExtendedMachinesControllerOperations + operations + :vartype web_app_extended_machines_controller: + azure.mgmt.migrate.operations.WebAppExtendedMachinesControllerOperations + :ivar iis_web_applications_controller: IisWebApplicationsControllerOperations operations + :vartype iis_web_applications_controller: + azure.mgmt.migrate.operations.IisWebApplicationsControllerOperations + :ivar iis_web_servers_controller: IisWebServersControllerOperations operations + :vartype iis_web_servers_controller: + azure.mgmt.migrate.operations.IisWebServersControllerOperations + :ivar web_app_run_as_accounts_controller: WebAppRunAsAccountsControllerOperations operations + :vartype web_app_run_as_accounts_controller: + azure.mgmt.migrate.operations.WebAppRunAsAccountsControllerOperations + :ivar tomcat_web_applications_controller: TomcatWebApplicationsControllerOperations operations + :vartype tomcat_web_applications_controller: + azure.mgmt.migrate.operations.TomcatWebApplicationsControllerOperations + :ivar tomcat_web_servers_controller: TomcatWebServersControllerOperations operations + :vartype tomcat_web_servers_controller: + azure.mgmt.migrate.operations.TomcatWebServersControllerOperations + :ivar web_app_properties_controller: WebAppPropertiesControllerOperations operations + :vartype web_app_properties_controller: + azure.mgmt.migrate.operations.WebAppPropertiesControllerOperations + :ivar web_applications_controller: WebApplicationsControllerOperations operations + :vartype web_applications_controller: + azure.mgmt.migrate.operations.WebApplicationsControllerOperations + :ivar web_servers_controller: WebServersControllerOperations operations + :vartype web_servers_controller: azure.mgmt.migrate.operations.WebServersControllerOperations + :ivar server_dependency_map_controller: ServerDependencyMapControllerOperations operations + :vartype server_dependency_map_controller: + azure.mgmt.migrate.operations.ServerDependencyMapControllerOperations + :ivar server_jobs_controller: ServerJobsControllerOperations operations + :vartype server_jobs_controller: azure.mgmt.migrate.operations.ServerJobsControllerOperations + :ivar servers_controller: ServersControllerOperations operations + :vartype servers_controller: azure.mgmt.migrate.operations.ServersControllerOperations + :ivar server_software_inventories_controller: ServerSoftwareInventoriesControllerOperations + operations + :vartype server_software_inventories_controller: + azure.mgmt.migrate.operations.ServerSoftwareInventoriesControllerOperations + :ivar server_operations_status_controller: ServerOperationsStatusControllerOperations + operations + :vartype server_operations_status_controller: + azure.mgmt.migrate.operations.ServerOperationsStatusControllerOperations + :ivar server_run_as_accounts_controller: ServerRunAsAccountsControllerOperations operations + :vartype server_run_as_accounts_controller: + azure.mgmt.migrate.operations.ServerRunAsAccountsControllerOperations + :ivar dependency_map_controller: DependencyMapControllerOperations operations + :vartype dependency_map_controller: + azure.mgmt.migrate.operations.DependencyMapControllerOperations + :ivar vmware_host_controller: VmwareHostControllerOperations operations + :vartype vmware_host_controller: azure.mgmt.migrate.operations.VmwareHostControllerOperations + :ivar hyperv_jobs_controller: HypervJobsControllerOperations operations + :vartype hyperv_jobs_controller: azure.mgmt.migrate.operations.HypervJobsControllerOperations + :ivar machines_controller: MachinesControllerOperations operations + :vartype machines_controller: azure.mgmt.migrate.operations.MachinesControllerOperations + :ivar vmware_software_inventories_controller: VmwareSoftwareInventoriesControllerOperations + operations + :vartype vmware_software_inventories_controller: + azure.mgmt.migrate.operations.VmwareSoftwareInventoriesControllerOperations + :ivar vmware_operations_status: VmwareOperationsStatusOperations operations + :vartype vmware_operations_status: + azure.mgmt.migrate.operations.VmwareOperationsStatusOperations + :ivar run_as_accounts_controller: RunAsAccountsControllerOperations operations + :vartype run_as_accounts_controller: + azure.mgmt.migrate.operations.RunAsAccountsControllerOperations + :ivar vmware_properties_controller: VmwarePropertiesControllerOperations operations + :vartype vmware_properties_controller: + azure.mgmt.migrate.operations.VmwarePropertiesControllerOperations + :ivar vcenter_controller: VcenterControllerOperations operations + :vartype vcenter_controller: azure.mgmt.migrate.operations.VcenterControllerOperations + :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. 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 "2023-10-01-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 = AzureMigrateDiscoveryServiceConfiguration( + 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.hyperv_sites = HypervSitesOperations(self._client, self._config, self._serialize, self._deserialize) + self.import_sites_controller = ImportSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.master_sites_controller = MasterSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_sites_controller = ServerSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sites_controller = SitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_sites_controller = HypervSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_dependency_map_controller = HypervDependencyMapControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_cluster_controller = HypervClusterControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_host_controller = HypervHostControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_jobs = HypervJobsOperations(self._client, self._config, self._serialize, self._deserialize) + self.hyperv_machines_controller = HypervMachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_software_inventories_controller = HypervSoftwareInventoriesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_operations_status_controller = HypervOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_run_as_accounts_controller = HypervRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.import_jobs_controller = ImportJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.import_machines_controller = ImportMachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.master_sites_operations_status_controller = MasterSitesOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.private_endpoint_connection_controller = PrivateEndpointConnectionControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.private_link_resources_controller = PrivateLinkResourcesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_sites_controller = SqlSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_discovery_site_data_source_controller = SqlDiscoverySiteDataSourceControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_jobs_controller = SqlJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_operations_status_controller = SqlOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_run_as_accounts_controller = SqlRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_availability_groups_controller = SqlAvailabilityGroupsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_databases_controller = SqlDatabasesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_servers_controller = SqlServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_sites_controller = WebAppSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_discovery_site_data_sources_controller = WebAppDiscoverySiteDataSourcesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_extended_machines_controller = WebAppExtendedMachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.iis_web_applications_controller = IisWebApplicationsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.iis_web_servers_controller = IisWebServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_run_as_accounts_controller = WebAppRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.tomcat_web_applications_controller = TomcatWebApplicationsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.tomcat_web_servers_controller = TomcatWebServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_properties_controller = WebAppPropertiesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_applications_controller = WebApplicationsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_servers_controller = WebServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_dependency_map_controller = ServerDependencyMapControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_jobs_controller = ServerJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.servers_controller = ServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_software_inventories_controller = ServerSoftwareInventoriesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_operations_status_controller = ServerOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_run_as_accounts_controller = ServerRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.dependency_map_controller = DependencyMapControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_host_controller = VmwareHostControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_jobs_controller = HypervJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.machines_controller = MachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_software_inventories_controller = VmwareSoftwareInventoriesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_operations_status = VmwareOperationsStatusOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.run_as_accounts_controller = RunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_properties_controller = VmwarePropertiesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vcenter_controller = VcenterControllerOperations( + 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/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_configuration.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_configuration.py new file mode 100644 index 000000000000..426b49471874 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_configuration.py @@ -0,0 +1,65 @@ +# 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 typing import Any, TYPE_CHECKING + +from azure.core.pipeline import policies +from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy + +from ._version import VERSION + +if TYPE_CHECKING: + # pylint: disable=unused-import,ungrouped-imports + from azure.core.credentials import TokenCredential + + +class AzureMigrateDiscoveryServiceConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + """Configuration for AzureMigrateDiscoveryService. + + 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. + :type credential: ~azure.core.credentials.TokenCredential + :param subscription_id: The ID of the target subscription. Required. + :type subscription_id: str + :keyword api_version: Api Version. Default value is "2023-10-01-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", "2023-10-01-preview") + + if credential is None: + raise ValueError("Parameter 'credential' must not be None.") + if subscription_id is None: + raise ValueError("Parameter 'subscription_id' must not be None.") + + self.credential = credential + self.subscription_id = subscription_id + self.api_version = api_version + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "mgmt-migrate/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) + self._configure(**kwargs) + + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") + if self.credential and not self.authentication_policy: + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_patch.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_serialization.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_serialization.py new file mode 100644 index 000000000000..8139854b97bb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_serialization.py @@ -0,0 +1,2000 @@ +# -------------------------------------------------------------------------- +# +# 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. +# +# -------------------------------------------------------------------------- + +# pylint: skip-file +# pyright: reportUnnecessaryTypeIgnoreComment=false + +from base64 import b64decode, b64encode +import calendar +import datetime +import decimal +import email +from enum import Enum +import json +import logging +import re +import sys +import codecs +from typing import ( + Dict, + Any, + cast, + Optional, + Union, + AnyStr, + IO, + Mapping, + Callable, + TypeVar, + MutableMapping, + Type, + List, + Mapping, +) + +try: + from urllib import quote # type: ignore +except ImportError: + from urllib.parse import quote +import xml.etree.ElementTree as ET + +import isodate # type: ignore + +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] + + +class RawDeserializer: + + # Accept "text" because we're open minded people... + JSON_REGEXP = re.compile(r"^(application|text)/([a-z+.]+\+)?json$") + + # Name used in context + CONTEXT_NAME = "deserialized_data" + + @classmethod + def deserialize_from_text(cls, data: Optional[Union[AnyStr, IO]], content_type: Optional[str] = None) -> Any: + """Decode data according to content-type. + + Accept a stream of data as well, but will be load at once in memory for now. + + If no content-type, will return the string version (not bytes, not stream) + + :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. + """ + if hasattr(data, "read"): + # Assume a stream + data = cast(IO, data).read() + + if isinstance(data, bytes): + data_as_str = data.decode(encoding="utf-8-sig") + else: + # Explain to mypy the correct type. + data_as_str = cast(str, data) + + # Remove Byte Order Mark if present in string + data_as_str = data_as_str.lstrip(_BOM) + + if content_type is None: + return data + + if cls.JSON_REGEXP.match(content_type): + try: + return json.loads(data_as_str) + except ValueError as err: + raise DeserializationError("JSON is invalid: {}".format(err), err) + elif "xml" in (content_type or []): + try: + + try: + if isinstance(data, unicode): # type: ignore + # If I'm Python 2.7 and unicode XML will scream if I try a "fromstring" on unicode string + data_as_str = data_as_str.encode(encoding="utf-8") # type: ignore + except NameError: + pass + + return ET.fromstring(data_as_str) # nosec + except ET.ParseError as err: + # It might be because the server has an issue, and returned JSON with + # content-type XML.... + # So let's try a JSON load, and if it's still broken + # let's flow the initial exception + def _json_attemp(data): + try: + return True, json.loads(data) + except ValueError: + return False, None # Don't care about this one + + success, json_result = _json_attemp(data) + if success: + return json_result + # If i'm here, it's not JSON, it's not XML, let's scream + # and raise the last context in this block (the XML exception) + # The function hack is because Py2.7 messes up with exception + # context otherwise. + _LOGGER.critical("Wasn't XML not JSON, failing") + raise DeserializationError("XML is invalid") from err + elif content_type.startswith("text/"): + return data_as_str + raise DeserializationError("Cannot deserialize content-type: {}".format(content_type)) + + @classmethod + def deserialize_from_http_generics(cls, body_bytes: Optional[Union[AnyStr, IO]], headers: Mapping) -> Any: + """Deserialize from HTTP response. + + Use bytes and headers to NOT use any requests/aiohttp or whatever + specific implementation. + Headers will tested for "content-type" + """ + # Try to use content-type from headers if available + content_type = None + if "content-type" in headers: + content_type = headers["content-type"].split(";")[0].strip().lower() + # Ouch, this server did not declare what it sent... + # Let's guess it's JSON... + # Also, since Autorest was considering that an empty body was a valid JSON, + # need that test as well.... + else: + content_type = "application/json" + + if body_bytes: + return cls.deserialize_from_text(body_bytes, content_type) + return None + + +_LOGGER = logging.getLogger(__name__) + +try: + _long_type = long # type: ignore +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 + +_FLATTEN = re.compile(r"(? None: + self.additional_properties: Optional[Dict[str, Any]] = {} + for k in kwargs: + if k not in self._attribute_map: + _LOGGER.warning("%s is not a known attribute of class %s and will be ignored", k, self.__class__) + elif k in self._validation and self._validation[k].get("readonly", False): + _LOGGER.warning("Readonly attribute %s will be ignored in class %s", k, self.__class__) + else: + setattr(self, k, kwargs[k]) + + def __eq__(self, other: Any) -> bool: + """Compare objects by comparing all attributes.""" + if isinstance(other, self.__class__): + return self.__dict__ == other.__dict__ + return False + + def __ne__(self, other: Any) -> bool: + """Compare objects by comparing all attributes.""" + return not self.__eq__(other) + + def __str__(self) -> str: + return str(self.__dict__) + + @classmethod + def enable_additional_properties_sending(cls) -> None: + cls._attribute_map["additional_properties"] = {"key": "", "type": "{object}"} + + @classmethod + def is_xml_model(cls) -> bool: + try: + cls._xml_map # type: ignore + except AttributeError: + return False + return True + + @classmethod + def _create_xml_node(cls): + """Create XML node.""" + try: + xml_map = cls._xml_map # type: ignore + except AttributeError: + xml_map = {} + + return _create_xml_node(xml_map.get("name", cls.__name__), xml_map.get("prefix", None), xml_map.get("ns", None)) + + def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: + """Return the JSON that would be sent to server from this model. + + This is an alias to `as_dict(full_restapi_key_transformer, keep_readonly=False)`. + + 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 + :returns: A dict JSON compatible object + :rtype: dict + """ + serializer = Serializer(self._infer_class_models()) + return serializer._serialize(self, keep_readonly=keep_readonly, **kwargs) # type: ignore + + def as_dict( + self, + keep_readonly: bool = True, + key_transformer: Callable[[str, Dict[str, Any], Any], Any] = attribute_transformer, + **kwargs: Any + ) -> JSON: + """Return a dict that can be serialized using json.dump. + + Advanced usage might optionally use a callback as parameter: + + .. code::python + + def my_key_transformer(key, attr_desc, value): + return key + + Key is the attribute name used in Python. Attr_desc + is a dict of metadata. Currently contains 'type' with the + msrest type and 'key' with the RestAPI encoded key. + Value is the current value in this object. + + The string returned will be used to serialize the key. + If the return type is a list, this is considered hierarchical + result dict. + + See the three examples in this file: + + - attribute_transformer + - full_restapi_key_transformer + - last_restapi_key_transformer + + If you want XML serialization, you can pass the kwargs is_xml=True. + + :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 + + @classmethod + def _infer_class_models(cls): + try: + str_models = cls.__module__.rsplit(".", 1)[0] + models = sys.modules[str_models] + 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: + # 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: + """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 + """ + deserializer = Deserializer(cls._infer_class_models()) + return deserializer(cls.__name__, data, content_type=content_type) # type: ignore + + @classmethod + def from_dict( + cls: Type[ModelType], + data: Any, + key_extractors: Optional[Callable[[str, Dict[str, Any], Any], Any]] = None, + content_type: Optional[str] = None, + ) -> ModelType: + """Parse a dict using given key extractor return a model. + + By default consider key + extractors (rest_key_case_insensitive_extractor, attribute_key_case_insensitive_extractor + and last_rest_key_case_insensitive_extractor) + + :param dict data: A dict using RestAPI structure + :param str content_type: JSON by default, set application/xml if XML. + :returns: An instance of this model + :raises: DeserializationError if something went wrong + """ + deserializer = Deserializer(cls._infer_class_models()) + deserializer.key_extractors = ( # type: ignore + [ # type: ignore + attribute_key_case_insensitive_extractor, + rest_key_case_insensitive_extractor, + last_rest_key_case_insensitive_extractor, + ] + if key_extractors is None + else key_extractors + ) + return deserializer(cls.__name__, data, content_type=content_type) # type: ignore + + @classmethod + def _flatten_subtype(cls, key, objects): + if "_subtype_map" not in cls.__dict__: + return {} + result = dict(cls._subtype_map[key]) + for valuetype in cls._subtype_map[key].values(): + result.update(objects[valuetype]._flatten_subtype(key, objects)) + 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. + """ + 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) + else: + subtype_value = xml_key_extractor(subtype_key, cls._attribute_map[subtype_key], response) + if subtype_value: + # Try to match base class. Can be class name only + # (bug to fix in Autorest to support x-ms-discriminator-name) + if cls.__name__ == subtype_value: + return cls + flatten_mapping_type = cls._flatten_subtype(subtype_key, objects) + try: + return objects[flatten_mapping_type[subtype_value]] # type: ignore + except KeyError: + _LOGGER.warning( + "Subtype value %s has no mapping, use base class %s.", + subtype_value, + cls.__name__, + ) + break + else: + _LOGGER.warning("Discriminator %s is absent or null, use base class %s.", subtype_key, cls.__name__) + break + return cls + + @classmethod + def _get_rest_key_parts(cls, attr_key): + """Get the RestAPI key of this attr, split it and decode part + :param str attr_key: Attribute key must be in attribute_map. + :returns: A list of RestAPI part + :rtype: list + """ + rest_split_key = _FLATTEN.split(cls._attribute_map[attr_key]["key"]) + return [_decode_attribute_map_key(key_part) for key_part in rest_split_key] + + +def _decode_attribute_map_key(key): + """This decode a key in an _attribute_map to the actual key we want to look at + inside the received data. + + :param str key: A key string from the generated code + """ + return key.replace("\\.", ".") + + +class Serializer(object): + """Request object model serializer.""" + + basic_types = {str: "str", int: "int", bool: "bool", float: "float"} + + _xml_basic_types_serializers = {"bool": lambda x: str(x).lower()} + days = {0: "Mon", 1: "Tue", 2: "Wed", 3: "Thu", 4: "Fri", 5: "Sat", 6: "Sun"} + months = { + 1: "Jan", + 2: "Feb", + 3: "Mar", + 4: "Apr", + 5: "May", + 6: "Jun", + 7: "Jul", + 8: "Aug", + 9: "Sep", + 10: "Oct", + 11: "Nov", + 12: "Dec", + } + validation = { + "min_length": lambda x, y: len(x) < y, + "max_length": lambda x, y: len(x) > y, + "minimum": lambda x, y: x < y, + "maximum": lambda x, y: x > y, + "minimum_ex": lambda x, y: x <= y, + "maximum_ex": lambda x, y: x >= y, + "min_items": lambda x, y: len(x) < y, + "max_items": lambda x, y: len(x) > y, + "pattern": lambda x, y: not re.match(y, x, re.UNICODE), + "unique": lambda x, y: len(x) != len(set(x)), + "multiple": lambda x, y: x % y != 0, + } + + def __init__(self, classes: Optional[Mapping[str, type]] = None): + self.serialize_type = { + "iso-8601": Serializer.serialize_iso, + "rfc-1123": Serializer.serialize_rfc, + "unix-time": Serializer.serialize_unix, + "duration": Serializer.serialize_duration, + "date": Serializer.serialize_date, + "time": Serializer.serialize_time, + "decimal": Serializer.serialize_decimal, + "long": Serializer.serialize_long, + "bytearray": Serializer.serialize_bytearray, + "base64": Serializer.serialize_base64, + "object": self.serialize_object, + "[]": self.serialize_iter, + "{}": self.serialize_dict, + } + self.dependencies: Dict[str, type] = dict(classes) if classes else {} + self.key_transformer = full_restapi_key_transformer + self.client_side_validation = True + + def _serialize(self, target_obj, data_type=None, **kwargs): + """Serialize data into a string according to type. + + :param 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. + """ + key_transformer = kwargs.get("key_transformer", self.key_transformer) + keep_readonly = kwargs.get("keep_readonly", False) + if target_obj is None: + return None + + attr_name = None + class_name = target_obj.__class__.__name__ + + if data_type: + return self.serialize_data(target_obj, data_type, **kwargs) + + if not hasattr(target_obj, "_attribute_map"): + data_type = type(target_obj).__name__ + if data_type in self.basic_types.values(): + return self.serialize_data(target_obj, data_type, **kwargs) + + # Force "is_xml" kwargs if we detect a XML model + try: + is_xml_model_serialization = kwargs["is_xml"] + except KeyError: + is_xml_model_serialization = kwargs.setdefault("is_xml", target_obj.is_xml_model()) + + serialized = {} + if is_xml_model_serialization: + serialized = target_obj._create_xml_node() + try: + attributes = target_obj._attribute_map + for attr, attr_desc in attributes.items(): + attr_name = attr + if not keep_readonly and target_obj._validation.get(attr_name, {}).get("readonly", False): + continue + + if attr_name == "additional_properties" and attr_desc["key"] == "": + if target_obj.additional_properties is not None: + serialized.update(target_obj.additional_properties) + continue + try: + + orig_attr = getattr(target_obj, attr) + if is_xml_model_serialization: + pass # Don't provide "transformer" for XML for now. Keep "orig_attr" + else: # JSON + keys, orig_attr = key_transformer(attr, attr_desc.copy(), orig_attr) + keys = keys if isinstance(keys, list) else [keys] + + kwargs["serialization_ctxt"] = attr_desc + new_attr = self.serialize_data(orig_attr, attr_desc["type"], **kwargs) + + if is_xml_model_serialization: + xml_desc = attr_desc.get("xml", {}) + xml_name = xml_desc.get("name", attr_desc["key"]) + xml_prefix = xml_desc.get("prefix", None) + xml_ns = xml_desc.get("ns", None) + if xml_desc.get("attr", False): + if xml_ns: + ET.register_namespace(xml_prefix, xml_ns) + xml_name = "{{{}}}{}".format(xml_ns, xml_name) + serialized.set(xml_name, new_attr) # type: ignore + continue + if xml_desc.get("text", False): + serialized.text = new_attr # type: ignore + continue + 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 "name" not in getattr(orig_attr, "_xml_map", {}): + splitted_tag = new_attr.tag.split("}") + if len(splitted_tag) == 2: # Namespace + new_attr.tag = "}".join([splitted_tag[0], xml_name]) + else: + new_attr.tag = xml_name + serialized.append(new_attr) # type: ignore + else: # That's a basic type + # Integrate namespace if necessary + local_node = _create_xml_node(xml_name, xml_prefix, xml_ns) + local_node.text = str(new_attr) + serialized.append(local_node) # type: ignore + else: # JSON + for k in reversed(keys): # type: ignore + new_attr = {k: new_attr} + + _new_attr = new_attr + _serialized = serialized + for k in keys: # type: ignore + if k not in _serialized: + _serialized.update(_new_attr) # type: ignore + _new_attr = _new_attr[k] # type: ignore + _serialized = _serialized[k] + except ValueError as err: + if isinstance(err, SerializationError): + raise + + 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 + + def body(self, data, data_type, **kwargs): + """Serialize data intended for a request body. + + :param 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 + """ + + # Just in case this is a dict + internal_data_type_str = data_type.strip("[]{}") + internal_data_type = self.dependencies.get(internal_data_type_str, None) + try: + is_xml_model_serialization = kwargs["is_xml"] + except KeyError: + if internal_data_type and issubclass(internal_data_type, Model): + is_xml_model_serialization = kwargs.setdefault("is_xml", internal_data_type.is_xml_model()) + else: + is_xml_model_serialization = False + if internal_data_type and not isinstance(internal_data_type, Enum): + try: + deserializer = Deserializer(self.dependencies) + # Since it's on serialization, it's almost sure that format is not JSON REST + # We're not able to deal with additional properties for now. + deserializer.additional_properties_detection = False + if is_xml_model_serialization: + deserializer.key_extractors = [ # type: ignore + attribute_key_case_insensitive_extractor, + ] + else: + deserializer.key_extractors = [ + rest_key_case_insensitive_extractor, + attribute_key_case_insensitive_extractor, + last_rest_key_case_insensitive_extractor, + ] + data = deserializer._deserialize(data_type, data) + except DeserializationError as err: + raise SerializationError("Unable to build a model: " + str(err)) from err + + return self._serialize(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 data_type: The type to be serialized from. + :rtype: str + :raises: TypeError if serialization fails. + :raises: ValueError if data is None + """ + try: + output = self.serialize_data(data, data_type, **kwargs) + if data_type == "bool": + output = json.dumps(output) + + if kwargs.get("skip_quote") is True: + output = str(output) + 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 + + def query(self, name, data, data_type, **kwargs): + """Serialize data intended for a URL query. + + :param 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 + """ + try: + # Treat the list aside, since we don't want to encode the div separator + if data_type.startswith("["): + internal_data_type = data_type[1:-1] + do_quote = not kwargs.get("skip_quote", False) + return self.serialize_iter(data, internal_data_type, do_quote=do_quote, **kwargs) + + # Not a list, regular serialization + output = self.serialize_data(data, data_type, **kwargs) + if data_type == "bool": + output = json.dumps(output) + if kwargs.get("skip_quote") is True: + output = str(output) + else: + output = quote(str(output), safe="") + except SerializationError: + raise TypeError("{} must be type {}.".format(name, data_type)) + else: + 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 data_type: The type to be serialized from. + :rtype: str + :raises: TypeError if serialization fails. + :raises: ValueError if data is None + """ + try: + if data_type in ["[str]"]: + data = ["" if d is None else d for d in data] + + 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) + + def serialize_data(self, data, data_type, **kwargs): + """Serialize generic data according to supplied data type. + + :param 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. + """ + if data is None: + raise ValueError("No value for given attribute") + + try: + if data is CoreNull: + return None + if data_type in self.basic_types.values(): + return self.serialize_basic(data, data_type, **kwargs) + + elif data_type in self.serialize_type: + return self.serialize_type[data_type](data, **kwargs) + + # If dependencies is empty, try with current data class + # It has to be a subclass of Enum anyway + enum_type = self.dependencies.get(data_type, data.__class__) + if issubclass(enum_type, Enum): + return Serializer.serialize_enum(data, enum_obj=enum_type) + + iter_type = data_type[0] + data_type[-1] + if iter_type in self.serialize_type: + return self.serialize_type[iter_type](data, data_type[1:-1], **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) + + @classmethod + def _get_custom_serializers(cls, data_type, **kwargs): + custom_serializer = kwargs.get("basic_types_serializers", {}).get(data_type) + if custom_serializer: + return custom_serializer + if kwargs.get("is_xml", False): + return cls._xml_basic_types_serializers.get(data_type) + + @classmethod + def serialize_basic(cls, data, data_type, **kwargs): + """Serialize basic builting data type. + Serializes objects to str, int, float or bool. + + Possible 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 str data_type: Type of object in the iterable. + """ + 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 + + @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. + :rtype: str + """ + try: # If I received an enum, return its value + return data.value + except AttributeError: + pass + + try: + if isinstance(data, unicode): # type: ignore + # Don't change it, JSON and XML ElementTree are totally able + # to serialize correctly u'' strings + return data + except NameError: + return str(data) + else: + return str(data) + + def serialize_iter(self, data, iter_type, div=None, **kwargs): + """Serialize iterable. + + Supported kwargs: + - serialization_ctxt dict : The current entry of _attribute_map, or same format. + 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 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 + """ + if isinstance(data, str): + raise SerializationError("Refuse str type as a valid iter type.") + + serialization_ctxt = kwargs.get("serialization_ctxt", {}) + is_xml = kwargs.get("is_xml", False) + + serialized = [] + for d in data: + try: + serialized.append(self.serialize_data(d, iter_type, **kwargs)) + except ValueError as err: + if isinstance(err, SerializationError): + raise + serialized.append(None) + + if kwargs.get("do_quote", False): + serialized = ["" if s is None else quote(str(s), safe="") for s in serialized] + + if div: + serialized = ["" if s is None else str(s) for s in serialized] + serialized = div.join(serialized) + + if "xml" in serialization_ctxt or is_xml: + # XML serialization is more complicated + xml_desc = serialization_ctxt.get("xml", {}) + xml_name = xml_desc.get("name") + if not xml_name: + xml_name = serialization_ctxt["key"] + + # Create a wrap node if necessary (use the fact that Element and list have "append") + is_wrapped = xml_desc.get("wrapped", False) + node_name = xml_desc.get("itemsName", xml_name) + if is_wrapped: + final_result = _create_xml_node(xml_name, xml_desc.get("prefix", None), xml_desc.get("ns", None)) + else: + final_result = [] + # All list elements to "local_node" + for el in serialized: + if isinstance(el, ET.Element): + el_node = el + else: + el_node = _create_xml_node(node_name, xml_desc.get("prefix", None), xml_desc.get("ns", None)) + if el is not None: # Otherwise it writes "None" :-p + el_node.text = str(el) + final_result.append(el_node) + return final_result + return serialized + + def serialize_dict(self, attr, dict_type, **kwargs): + """Serialize a dictionary of objects. + + :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 + """ + serialization_ctxt = kwargs.get("serialization_ctxt", {}) + serialized = {} + for key, value in attr.items(): + try: + serialized[self.serialize_unicode(key)] = self.serialize_data(value, dict_type, **kwargs) + except ValueError as err: + if isinstance(err, SerializationError): + raise + serialized[self.serialize_unicode(key)] = None + + if "xml" in serialization_ctxt: + # XML serialization is more complicated + xml_desc = serialization_ctxt["xml"] + xml_name = xml_desc["name"] + + final_result = _create_xml_node(xml_name, xml_desc.get("prefix", None), xml_desc.get("ns", None)) + for key, value in serialized.items(): + ET.SubElement(final_result, key).text = value + return final_result + + return serialized + + def serialize_object(self, attr, **kwargs): + """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 + cast to str. + + :param dict attr: Object to be serialized. + :rtype: dict or str + """ + if attr is None: + return None + if isinstance(attr, ET.Element): + return attr + obj_type = type(attr) + if obj_type in self.basic_types: + return self.serialize_basic(attr, self.basic_types[obj_type], **kwargs) + if obj_type is _long_type: + return self.serialize_long(attr) + if obj_type is str: + return self.serialize_unicode(attr) + if obj_type is datetime.datetime: + return self.serialize_iso(attr) + if obj_type is datetime.date: + return self.serialize_date(attr) + if obj_type is datetime.time: + return self.serialize_time(attr) + if obj_type is datetime.timedelta: + return self.serialize_duration(attr) + if obj_type is decimal.Decimal: + 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): + return self._serialize(attr) + + if obj_type == dict: + serialized = {} + for key, value in attr.items(): + try: + serialized[self.serialize_unicode(key)] = self.serialize_object(value, **kwargs) + except ValueError: + serialized[self.serialize_unicode(key)] = None + return serialized + + if obj_type == list: + serialized = [] + for obj in attr: + try: + serialized.append(self.serialize_object(obj, **kwargs)) + except ValueError: + pass + return serialized + return str(attr) + + @staticmethod + def serialize_enum(attr, enum_obj=None): + try: + result = attr.value + except AttributeError: + result = attr + try: + enum_obj(result) # type: ignore + return result + except ValueError: + 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)) + + @staticmethod + def serialize_bytearray(attr, **kwargs): + """Serialize bytearray into base-64 string. + + :param attr: Object to be serialized. + :rtype: str + """ + return b64encode(attr).decode() + + @staticmethod + def serialize_base64(attr, **kwargs): + """Serialize str into base-64 string. + + :param attr: Object to be serialized. + :rtype: str + """ + encoded = b64encode(attr).decode("ascii") + return encoded.strip("=").replace("+", "-").replace("/", "_") + + @staticmethod + def serialize_decimal(attr, **kwargs): + """Serialize Decimal object to float. + + :param attr: Object to be serialized. + :rtype: float + """ + return float(attr) + + @staticmethod + def serialize_long(attr, **kwargs): + """Serialize long (Py2) or int (Py3). + + :param attr: Object to be serialized. + :rtype: int/long + """ + return _long_type(attr) + + @staticmethod + def serialize_date(attr, **kwargs): + """Serialize Date object into ISO-8601 formatted string. + + :param Date attr: Object to be serialized. + :rtype: str + """ + if isinstance(attr, str): + attr = isodate.parse_date(attr) + t = "{:04}-{:02}-{:02}".format(attr.year, attr.month, attr.day) + return t + + @staticmethod + def serialize_time(attr, **kwargs): + """Serialize Time object into ISO-8601 formatted string. + + :param datetime.time attr: Object to be serialized. + :rtype: str + """ + if isinstance(attr, str): + attr = isodate.parse_time(attr) + t = "{:02}:{:02}:{:02}".format(attr.hour, attr.minute, attr.second) + if attr.microsecond: + t += ".{:02}".format(attr.microsecond) + return t + + @staticmethod + def serialize_duration(attr, **kwargs): + """Serialize TimeDelta object into ISO-8601 formatted string. + + :param TimeDelta attr: Object to be serialized. + :rtype: str + """ + if isinstance(attr, str): + attr = isodate.parse_duration(attr) + return isodate.duration_isoformat(attr) + + @staticmethod + def serialize_rfc(attr, **kwargs): + """Serialize Datetime object into RFC-1123 formatted string. + + :param Datetime attr: Object to be serialized. + :rtype: str + :raises: TypeError if format invalid. + """ + 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.") + + return "{}, {:02} {} {:04} {:02}:{:02}:{:02} GMT".format( + Serializer.days[utc.tm_wday], + utc.tm_mday, + Serializer.months[utc.tm_mon], + utc.tm_year, + utc.tm_hour, + utc.tm_min, + utc.tm_sec, + ) + + @staticmethod + def serialize_iso(attr, **kwargs): + """Serialize Datetime object into ISO-8601 formatted string. + + :param Datetime attr: Object to be serialized. + :rtype: str + :raises: SerializationError if format invalid. + """ + if isinstance(attr, str): + attr = isodate.parse_datetime(attr) + try: + if not attr.tzinfo: + _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") + utc = attr.utctimetuple() + if utc.tm_year > 9999 or utc.tm_year < 1: + raise OverflowError("Hit max or min date") + + microseconds = str(attr.microsecond).rjust(6, "0").rstrip("0").ljust(3, "0") + if microseconds: + microseconds = "." + microseconds + date = "{:04}-{:02}-{:02}T{:02}:{:02}:{:02}".format( + utc.tm_year, utc.tm_mon, utc.tm_mday, utc.tm_hour, utc.tm_min, utc.tm_sec + ) + return date + microseconds + "Z" + except (ValueError, OverflowError) as err: + msg = "Unable to serialize datetime object." + raise SerializationError(msg) from err + except AttributeError as err: + msg = "ISO-8601 object must be valid Datetime object." + raise TypeError(msg) from err + + @staticmethod + def serialize_unix(attr, **kwargs): + """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 + """ + if isinstance(attr, int): + return attr + try: + 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.") + + +def rest_key_extractor(attr, attr_desc, data): + key = attr_desc["key"] + working_data = data + + while "." in key: + # Need the cast, as for some reasons "split" is typed as list[str | Any] + dict_keys = cast(List[str], _FLATTEN.split(key)) + if len(dict_keys) == 1: + key = _decode_attribute_map_key(dict_keys[0]) + break + working_key = _decode_attribute_map_key(dict_keys[0]) + working_data = working_data.get(working_key, data) + if working_data is None: + # If at any point while following flatten JSON path see None, it means + # that all properties under are None as well + return None + key = ".".join(dict_keys[1:]) + + return working_data.get(key) + + +def rest_key_case_insensitive_extractor(attr, attr_desc, data): + key = attr_desc["key"] + working_data = data + + while "." in key: + dict_keys = _FLATTEN.split(key) + if len(dict_keys) == 1: + key = _decode_attribute_map_key(dict_keys[0]) + break + working_key = _decode_attribute_map_key(dict_keys[0]) + working_data = attribute_key_case_insensitive_extractor(working_key, None, working_data) + if working_data is None: + # If at any point while following flatten JSON path see None, it means + # that all properties under are None as well + return None + key = ".".join(dict_keys[1:]) + + if working_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.""" + 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): + """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" + """ + key = attr_desc["key"] + dict_keys = _FLATTEN.split(key) + return attribute_key_case_insensitive_extractor(dict_keys[-1], None, data) + + +def attribute_key_extractor(attr, _, data): + return data.get(attr) + + +def attribute_key_case_insensitive_extractor(attr, _, data): + found_key = None + lower_attr = attr.lower() + for key in data: + if lower_attr == key.lower(): + found_key = key + break + + return data.get(found_key) + + +def _extract_name_from_internal_type(internal_type): + """Given an internal type XML description, extract correct XML name with namespace. + + :param dict internal_type: An model type + :rtype: tuple + :returns: A tuple XML name + namespace dict + """ + internal_type_xml_map = getattr(internal_type, "_xml_map", {}) + xml_name = internal_type_xml_map.get("name", internal_type.__name__) + xml_ns = internal_type_xml_map.get("ns", None) + if xml_ns: + xml_name = "{{{}}}{}".format(xml_ns, xml_name) + return xml_name + + +def xml_key_extractor(attr, attr_desc, data): + if isinstance(data, dict): + return None + + # Test if this model is XML ready first + if not isinstance(data, ET.Element): + return None + + xml_desc = attr_desc.get("xml", {}) + xml_name = xml_desc.get("name", attr_desc["key"]) + + # Look for a children + is_iter_type = attr_desc["type"].startswith("[") + is_wrapped = xml_desc.get("wrapped", False) + internal_type = attr_desc.get("internalType", None) + internal_type_xml_map = getattr(internal_type, "_xml_map", {}) + + # Integrate namespace if necessary + xml_ns = xml_desc.get("ns", internal_type_xml_map.get("ns", None)) + if xml_ns: + xml_name = "{{{}}}{}".format(xml_ns, xml_name) + + # If it's an attribute, that's simple + if xml_desc.get("attr", False): + return data.get(xml_name) + + # If it's x-ms-text, that's simple too + if xml_desc.get("text", False): + return data.text + + # Scenario where I take the local name: + # - Wrapped node + # - Internal type is an enum (considered basic types) + # - Internal type has no XML/Name node + if is_wrapped or (internal_type and (issubclass(internal_type, Enum) or "name" not in internal_type_xml_map)): + children = data.findall(xml_name) + # If internal type has a local name and it's not a list, I use that name + elif not is_iter_type and internal_type and "name" in internal_type_xml_map: + xml_name = _extract_name_from_internal_type(internal_type) + children = data.findall(xml_name) + # That's an array + else: + if internal_type: # Complex type, ignore itemsName and use the complex type name + items_name = _extract_name_from_internal_type(internal_type) + else: + items_name = xml_desc.get("itemsName", xml_name) + children = data.findall(items_name) + + if len(children) == 0: + if is_iter_type: + if is_wrapped: + return None # is_wrapped no node, we want None + else: + 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 + ) + ) + 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: + raise DeserializationError("Find several XML '{}' where it was not expected".format(xml_name)) + return children[0] + + +class Deserializer(object): + """Response object model deserializer. + + :param dict classes: Class type dictionary for deserializing complex types. + :ivar list key_extractors: Ordered list of extractors to be used by this deserializer. + """ + + 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}]?") + + def __init__(self, classes: Optional[Mapping[str, type]] = None): + self.deserialize_type = { + "iso-8601": Deserializer.deserialize_iso, + "rfc-1123": Deserializer.deserialize_rfc, + "unix-time": Deserializer.deserialize_unix, + "duration": Deserializer.deserialize_duration, + "date": Deserializer.deserialize_date, + "time": Deserializer.deserialize_time, + "decimal": Deserializer.deserialize_decimal, + "long": Deserializer.deserialize_long, + "bytearray": Deserializer.deserialize_bytearray, + "base64": Deserializer.deserialize_base64, + "object": self.deserialize_object, + "[]": self.deserialize_iter, + "{}": self.deserialize_dict, + } + self.deserialize_expected_types = { + "duration": (isodate.Duration, datetime.timedelta), + "iso-8601": (datetime.datetime), + } + self.dependencies: Dict[str, type] = dict(classes) if classes else {} + self.key_extractors = [rest_key_extractor, xml_key_extractor] + # Additional properties only works if the "rest_key_extractor" is used to + # extract the keys. Making it to work whatever the key extractor is too much + # complicated, with no real scenario for now. + # So adding a flag to disable additional properties detection. This flag should be + # used if your expect the deserialization to NOT come from a JSON REST syntax. + # Otherwise, result are unexpected + self.additional_properties_detection = True + + def __call__(self, target_obj, response_data, content_type=None): + """Call the deserializer to process a REST response. + + :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. + :return: Deserialized object. + """ + data = self._unpack_content(response_data, content_type) + return self._deserialize(target_obj, data) + + def _deserialize(self, target_obj, data): + """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. + :return: Deserialized 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(): + if attr in constants: + continue + value = getattr(data, attr) + if value is None: + continue + local_type = mapconfig["type"] + internal_data_type = local_type.strip("[]{}") + if internal_data_type not in self.dependencies or isinstance(internal_data_type, Enum): + continue + setattr(data, attr, self._deserialize(local_type, value)) + return data + except AttributeError: + return + + response, class_name = self._classify_target(target_obj, data) + + if isinstance(response, str): + return self.deserialize_data(data, response) + elif 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 + d_attrs = {} + for attr, attr_desc in attributes.items(): + # Check empty string. If it's not empty, someone has a real "additionalProperties"... + if attr == "additional_properties" and attr_desc["key"] == "": + continue + raw_value = None + # Enhance attr_desc with some dynamic data + attr_desc = attr_desc.copy() # Do a copy, do not change the real one + internal_data_type = attr_desc["type"].strip("[]{}") + if internal_data_type in self.dependencies: + attr_desc["internalType"] = self.dependencies[internal_data_type] + + for key_extractor in self.key_extractors: + found_value = key_extractor(attr, attr_desc, data) + if found_value is not None: + if raw_value is not None and raw_value != found_value: + msg = ( + "Ignoring extracted value '%s' from %s for key '%s'" + " (duplicate extraction, follow extractors order)" + ) + _LOGGER.warning(msg, found_value, key_extractor, attr) + continue + raw_value = found_value + + value = self.deserialize_data(raw_value, attr_desc["type"]) + d_attrs[attr] = value + 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) + + def _build_additional_properties(self, attribute_map, data): + if not self.additional_properties_detection: + return None + if "additional_properties" in attribute_map and attribute_map.get("additional_properties", {}).get("key") != "": + # Check empty string. If it's not empty, someone has a real "additionalProperties" + return None + if isinstance(data, ET.Element): + data = {el.tag: el.text for el in data} + + known_keys = { + _decode_attribute_map_key(_FLATTEN.split(desc["key"])[0]) + for desc in attribute_map.values() + if desc["key"] != "" + } + present_keys = set(data.keys()) + missing_keys = present_keys - known_keys + return {key: data[key] for key in missing_keys} + + def _classify_target(self, target, data): + """Check to see whether the deserialization target object can + be classified into a subclass. + Once classification has been determined, initialize object. + + :param str target: The target object type to deserialize to. + :param str/dict data: The response data to deserialize. + """ + if target is None: + return None, None + + if isinstance(target, str): + try: + target = self.dependencies[target] + except KeyError: + return target, target + + try: + target = target._classify(data, self.dependencies) # type: ignore + except AttributeError: + pass # Target is not a Model, no classify + return target, target.__class__.__name__ # type: ignore + + def failsafe_deserialize(self, target_obj, data, content_type=None): + """Ignores any errors encountered in deserialization, + and falls back to not deserializing the object. Recommended + for use in error deserialization, as we want to return the + HttpResponseError to users, and not have them deal with + a deserialization error. + + :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. + """ + try: + return self(target_obj, data, content_type=content_type) + except: + _LOGGER.debug( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + @staticmethod + def _unpack_content(raw_data, content_type=None): + """Extract the correct structure for deserialization. + + If raw_data is a PipelineResponse, try to extract the result of RawDeserializer. + if we can't, raise. Your Pipeline should have a RawDeserializer. + + If not a pipeline response and raw_data is bytes or string, use content-type + to decode it. If no content-type, try JSON. + + 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. + :raises JSONDecodeError: If JSON is requested and parsing is impossible. + :raises UnicodeDecodeError: If bytes is not UTF8 + """ + # Assume this is enough to detect a Pipeline Response without importing it + context = getattr(raw_data, "context", {}) + if context: + if RawDeserializer.CONTEXT_NAME in context: + return context[RawDeserializer.CONTEXT_NAME] + raise ValueError("This pipeline didn't have the RawDeserializer policy; can't deserialize") + + # Assume this is enough to recognize universal_http.ClientResponse without importing it + if hasattr(raw_data, "body"): + return RawDeserializer.deserialize_from_http_generics(raw_data.text(), raw_data.headers) + + # Assume this enough to recognize requests.Response without importing it. + if hasattr(raw_data, "_content_consumed"): + return RawDeserializer.deserialize_from_http_generics(raw_data.text, raw_data.headers) + + if isinstance(raw_data, (str, bytes)) or hasattr(raw_data, "read"): + return RawDeserializer.deserialize_from_text(raw_data, content_type) # type: ignore + return raw_data + + 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. + """ + 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")] + 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 + return response_obj + except TypeError as err: + msg = "Unable to deserialize {} into model {}. ".format(kwargs, response) # type: ignore + raise DeserializationError(msg + str(err)) + else: + try: + for attr, value in attrs.items(): + setattr(response, attr, value) + return response + except Exception as exp: + msg = "Unable to populate response model. " + msg += "Type: {}, Error: {}".format(type(response), exp) + raise DeserializationError(msg) + + def deserialize_data(self, data, data_type): + """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. + :return: Deserialized object. + """ + if data is None: + return data + + try: + if not data_type: + return data + if data_type in self.basic_types.values(): + return self.deserialize_basic(data, data_type) + if data_type in self.deserialize_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"{}"] + 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) + return data_val + + iter_type = data_type[0] + data_type[-1] + if iter_type in self.deserialize_type: + return self.deserialize_type[iter_type](data, data_type[1:-1]) + + obj_type = self.dependencies[data_type] + if issubclass(obj_type, Enum): + if isinstance(data, ET.Element): + data = data.text + return self.deserialize_enum(data, obj_type) + + except (ValueError, TypeError, AttributeError) as err: + msg = "Unable to deserialize response data." + msg += " Data: {}, {}".format(data, data_type) + raise DeserializationError(msg) from err + else: + 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. + :rtype: list + """ + if attr is None: + return None + if isinstance(attr, ET.Element): # If I receive an element here, get the children + attr = list(attr) + if not isinstance(attr, (list, set)): + raise DeserializationError("Cannot deserialize as [{}] an object of type {}".format(iter_type, type(attr))) + return [self.deserialize_data(a, iter_type) for a in attr] + + def deserialize_dict(self, attr, dict_type): + """Deserialize a dictionary. + + :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. + :rtype: dict + """ + if isinstance(attr, list): + return {x["key"]: self.deserialize_data(x["value"], dict_type) for x in attr} + + if isinstance(attr, ET.Element): + # Transform value into {"Key": "value"} + 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): + """Deserialize a generic object. + This will be handled as a dictionary. + + :param dict attr: Dictionary to be deserialized. + :rtype: dict + :raises: TypeError if non-builtin datatype encountered. + """ + if attr is None: + return None + if isinstance(attr, ET.Element): + # Do no recurse on XML, just return the tree as-is + return attr + if isinstance(attr, str): + return self.deserialize_basic(attr, "str") + obj_type = type(attr) + if obj_type in self.basic_types: + return self.deserialize_basic(attr, self.basic_types[obj_type]) + if obj_type is _long_type: + return self.deserialize_long(attr) + + if obj_type == dict: + deserialized = {} + for key, value in attr.items(): + try: + deserialized[key] = self.deserialize_object(value, **kwargs) + except ValueError: + deserialized[key] = None + return deserialized + + if obj_type == list: + deserialized = [] + for obj in attr: + try: + deserialized.append(self.deserialize_object(obj, **kwargs)) + except ValueError: + pass + return deserialized + + else: + error = "Cannot deserialize generic object with type: " + raise TypeError(error + str(obj_type)) + + def deserialize_basic(self, attr, data_type): + """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 + valid bool values. + + :param str attr: response string to be deserialized. + :param str data_type: deserialization data type. + :rtype: str, int, float or bool + :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 + if isinstance(attr, ET.Element): + attr = attr.text + if not attr: + 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 + + if data_type == "bool": + if attr in [True, False, 1, 0]: + return bool(attr) + elif isinstance(attr, str): + if attr.lower() in ["true", "1"]: + return True + elif 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 + + @staticmethod + def deserialize_unicode(data): + """Preserve unicode objects in Python 2, otherwise return data + as a string. + + :param str data: response string to be deserialized. + :rtype: str or unicode + """ + # We might be here because we have an enum modeled as string, + # and we try to deserialize a partial dict with enum inside + if isinstance(data, Enum): + return data + + # Consider this is real string + try: + if isinstance(data, unicode): # type: ignore + return data + except NameError: + return str(data) + else: + return str(data) + + @staticmethod + def deserialize_enum(data, enum_obj): + """Deserialize string into enum object. + + If the string is not a valid enum value it will be returned as-is + and a warning will be logged. + + :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. + :rtype: Enum + """ + if isinstance(data, enum_obj) or data is None: + return data + if isinstance(data, Enum): + data = data.value + if isinstance(data, int): + # Workaround. We might consider remove it in the future. + try: + return list(enum_obj.__members__.values())[data] + except IndexError: + error = "{!r} is not a valid index for enum {!r}" + raise DeserializationError(error.format(data, enum_obj)) + try: + return enum_obj(str(data)) + except ValueError: + for enum_value in enum_obj: + if enum_value.value.lower() == str(data).lower(): + return enum_value + # We don't fail anymore for unknown value, we deserialize as a string + _LOGGER.warning("Deserializer is not able to find %s as valid enum in %s", data, enum_obj) + return Deserializer.deserialize_unicode(data) + + @staticmethod + def deserialize_bytearray(attr): + """Deserialize string into bytearray. + + :param str attr: response string to be deserialized. + :rtype: bytearray + :raises: TypeError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + return bytearray(b64decode(attr)) # type: ignore + + @staticmethod + def deserialize_base64(attr): + """Deserialize base64 encoded string into string. + + :param str attr: response string to be deserialized. + :rtype: bytearray + :raises: TypeError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + padding = "=" * (3 - (len(attr) + 3) % 4) # type: ignore + attr = attr + padding # type: ignore + encoded = attr.replace("-", "+").replace("_", "/") + return b64decode(encoded) + + @staticmethod + 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. + """ + if isinstance(attr, ET.Element): + attr = attr.text + try: + return decimal.Decimal(str(attr)) # type: ignore + except decimal.DecimalException as err: + msg = "Invalid decimal {}".format(attr) + raise DeserializationError(msg) from err + + @staticmethod + def deserialize_long(attr): + """Deserialize string into long (Py2) or int (Py3). + + :param str attr: response string to be deserialized. + :rtype: long or int + :raises: ValueError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + return _long_type(attr) # type: ignore + + @staticmethod + def deserialize_duration(attr): + """Deserialize ISO-8601 formatted string into TimeDelta object. + + :param str attr: response string to be deserialized. + :rtype: TimeDelta + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + try: + duration = isodate.parse_duration(attr) + except (ValueError, OverflowError, AttributeError) as err: + msg = "Cannot deserialize duration object." + raise DeserializationError(msg) from err + else: + return duration + + @staticmethod + def deserialize_date(attr): + """Deserialize ISO-8601 formatted string into Date object. + + :param str attr: response string to be deserialized. + :rtype: Date + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + if re.search(r"[^\W\d_]", attr, re.I + re.U): # type: ignore + raise DeserializationError("Date must have only digits and -. Received: %s" % attr) + # This must NOT use defaultmonth/defaultday. Using None ensure this raises an exception. + return isodate.parse_date(attr, defaultmonth=0, defaultday=0) + + @staticmethod + def deserialize_time(attr): + """Deserialize ISO-8601 formatted string into time object. + + :param str attr: response string to be deserialized. + :rtype: datetime.time + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + if re.search(r"[^\W\d_]", attr, re.I + re.U): # type: ignore + raise DeserializationError("Date must have only digits and -. Received: %s" % attr) + return isodate.parse_time(attr) + + @staticmethod + def deserialize_rfc(attr): + """Deserialize RFC-1123 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: Datetime + :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)) + ) + 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 + + @staticmethod + def deserialize_iso(attr): + """Deserialize ISO-8601 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: Datetime + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + try: + attr = attr.upper() # type: ignore + match = Deserializer.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") + except (ValueError, OverflowError, AttributeError) as err: + msg = "Cannot deserialize datetime object." + raise DeserializationError(msg) from err + else: + return date_obj + + @staticmethod + def deserialize_unix(attr): + """Serialize Datetime object into IntTime format. + This is represented as seconds. + + :param int attr: Object to be serialized. + :rtype: Datetime + :raises: DeserializationError if format invalid + """ + if isinstance(attr, ET.Element): + attr = int(attr.text) # type: ignore + try: + attr = int(attr) + date_obj = datetime.datetime.fromtimestamp(attr, TZ_UTC) + except ValueError as err: + msg = "Cannot deserialize to unix datetime object." + raise DeserializationError(msg) from err + else: + return date_obj diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_version.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_version.py new file mode 100644 index 000000000000..e5754a47ce68 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/_version.py @@ -0,0 +1,9 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +VERSION = "1.0.0b1" diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/__init__.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/__init__.py new file mode 100644 index 000000000000..f7cbd76ac3cb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/__init__.py @@ -0,0 +1,23 @@ +# 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_migrate_discovery_service import AzureMigrateDiscoveryService + +try: + from ._patch import __all__ as _patch_all + from ._patch import * # pylint: disable=unused-wildcard-import +except ImportError: + _patch_all = [] +from ._patch import patch_sdk as _patch_sdk + +__all__ = [ + "AzureMigrateDiscoveryService", +] +__all__.extend([p for p in _patch_all if p not in __all__]) + +_patch_sdk() diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_azure_migrate_discovery_service.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_azure_migrate_discovery_service.py new file mode 100644 index 000000000000..e76f0a48b44a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_azure_migrate_discovery_service.py @@ -0,0 +1,485 @@ +# 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 AzureMigrateDiscoveryServiceConfiguration +from .operations import ( + DependencyMapControllerOperations, + HypervClusterControllerOperations, + HypervDependencyMapControllerOperations, + HypervHostControllerOperations, + HypervJobsControllerOperations, + HypervJobsOperations, + HypervMachinesControllerOperations, + HypervOperationsStatusControllerOperations, + HypervRunAsAccountsControllerOperations, + HypervSitesControllerOperations, + HypervSitesOperations, + HypervSoftwareInventoriesControllerOperations, + IisWebApplicationsControllerOperations, + IisWebServersControllerOperations, + ImportJobsControllerOperations, + ImportMachinesControllerOperations, + ImportSitesControllerOperations, + MachinesControllerOperations, + MasterSitesControllerOperations, + MasterSitesOperationsStatusControllerOperations, + Operations, + PrivateEndpointConnectionControllerOperations, + PrivateLinkResourcesControllerOperations, + RunAsAccountsControllerOperations, + ServerDependencyMapControllerOperations, + ServerJobsControllerOperations, + ServerOperationsStatusControllerOperations, + ServerRunAsAccountsControllerOperations, + ServerSitesControllerOperations, + ServerSoftwareInventoriesControllerOperations, + ServersControllerOperations, + SitesControllerOperations, + SqlAvailabilityGroupsControllerOperations, + SqlDatabasesControllerOperations, + SqlDiscoverySiteDataSourceControllerOperations, + SqlJobsControllerOperations, + SqlOperationsStatusControllerOperations, + SqlRunAsAccountsControllerOperations, + SqlServersControllerOperations, + SqlSitesControllerOperations, + TomcatWebApplicationsControllerOperations, + TomcatWebServersControllerOperations, + VcenterControllerOperations, + VmwareHostControllerOperations, + VmwareOperationsStatusOperations, + VmwarePropertiesControllerOperations, + VmwareSoftwareInventoriesControllerOperations, + WebAppDiscoverySiteDataSourcesControllerOperations, + WebAppExtendedMachinesControllerOperations, + WebAppPropertiesControllerOperations, + WebAppRunAsAccountsControllerOperations, + WebAppSitesControllerOperations, + WebApplicationsControllerOperations, + WebServersControllerOperations, +) + +if TYPE_CHECKING: + # pylint: disable=unused-import,ungrouped-imports + from azure.core.credentials_async import AsyncTokenCredential + + +class AzureMigrateDiscoveryService: # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes + """Azure OffAzure Resource Provider management API. + + :ivar operations: Operations operations + :vartype operations: azure.mgmt.migrate.aio.operations.Operations + :ivar hyperv_sites: HypervSitesOperations operations + :vartype hyperv_sites: azure.mgmt.migrate.aio.operations.HypervSitesOperations + :ivar import_sites_controller: ImportSitesControllerOperations operations + :vartype import_sites_controller: + azure.mgmt.migrate.aio.operations.ImportSitesControllerOperations + :ivar master_sites_controller: MasterSitesControllerOperations operations + :vartype master_sites_controller: + azure.mgmt.migrate.aio.operations.MasterSitesControllerOperations + :ivar server_sites_controller: ServerSitesControllerOperations operations + :vartype server_sites_controller: + azure.mgmt.migrate.aio.operations.ServerSitesControllerOperations + :ivar sites_controller: SitesControllerOperations operations + :vartype sites_controller: azure.mgmt.migrate.aio.operations.SitesControllerOperations + :ivar hyperv_sites_controller: HypervSitesControllerOperations operations + :vartype hyperv_sites_controller: + azure.mgmt.migrate.aio.operations.HypervSitesControllerOperations + :ivar hyperv_dependency_map_controller: HypervDependencyMapControllerOperations operations + :vartype hyperv_dependency_map_controller: + azure.mgmt.migrate.aio.operations.HypervDependencyMapControllerOperations + :ivar hyperv_cluster_controller: HypervClusterControllerOperations operations + :vartype hyperv_cluster_controller: + azure.mgmt.migrate.aio.operations.HypervClusterControllerOperations + :ivar hyperv_host_controller: HypervHostControllerOperations operations + :vartype hyperv_host_controller: + azure.mgmt.migrate.aio.operations.HypervHostControllerOperations + :ivar hyperv_jobs: HypervJobsOperations operations + :vartype hyperv_jobs: azure.mgmt.migrate.aio.operations.HypervJobsOperations + :ivar hyperv_machines_controller: HypervMachinesControllerOperations operations + :vartype hyperv_machines_controller: + azure.mgmt.migrate.aio.operations.HypervMachinesControllerOperations + :ivar hyperv_software_inventories_controller: HypervSoftwareInventoriesControllerOperations + operations + :vartype hyperv_software_inventories_controller: + azure.mgmt.migrate.aio.operations.HypervSoftwareInventoriesControllerOperations + :ivar hyperv_operations_status_controller: HypervOperationsStatusControllerOperations + operations + :vartype hyperv_operations_status_controller: + azure.mgmt.migrate.aio.operations.HypervOperationsStatusControllerOperations + :ivar hyperv_run_as_accounts_controller: HypervRunAsAccountsControllerOperations operations + :vartype hyperv_run_as_accounts_controller: + azure.mgmt.migrate.aio.operations.HypervRunAsAccountsControllerOperations + :ivar import_jobs_controller: ImportJobsControllerOperations operations + :vartype import_jobs_controller: + azure.mgmt.migrate.aio.operations.ImportJobsControllerOperations + :ivar import_machines_controller: ImportMachinesControllerOperations operations + :vartype import_machines_controller: + azure.mgmt.migrate.aio.operations.ImportMachinesControllerOperations + :ivar master_sites_operations_status_controller: + MasterSitesOperationsStatusControllerOperations operations + :vartype master_sites_operations_status_controller: + azure.mgmt.migrate.aio.operations.MasterSitesOperationsStatusControllerOperations + :ivar private_endpoint_connection_controller: PrivateEndpointConnectionControllerOperations + operations + :vartype private_endpoint_connection_controller: + azure.mgmt.migrate.aio.operations.PrivateEndpointConnectionControllerOperations + :ivar private_link_resources_controller: PrivateLinkResourcesControllerOperations operations + :vartype private_link_resources_controller: + azure.mgmt.migrate.aio.operations.PrivateLinkResourcesControllerOperations + :ivar sql_sites_controller: SqlSitesControllerOperations operations + :vartype sql_sites_controller: azure.mgmt.migrate.aio.operations.SqlSitesControllerOperations + :ivar sql_discovery_site_data_source_controller: SqlDiscoverySiteDataSourceControllerOperations + operations + :vartype sql_discovery_site_data_source_controller: + azure.mgmt.migrate.aio.operations.SqlDiscoverySiteDataSourceControllerOperations + :ivar sql_jobs_controller: SqlJobsControllerOperations operations + :vartype sql_jobs_controller: azure.mgmt.migrate.aio.operations.SqlJobsControllerOperations + :ivar sql_operations_status_controller: SqlOperationsStatusControllerOperations operations + :vartype sql_operations_status_controller: + azure.mgmt.migrate.aio.operations.SqlOperationsStatusControllerOperations + :ivar sql_run_as_accounts_controller: SqlRunAsAccountsControllerOperations operations + :vartype sql_run_as_accounts_controller: + azure.mgmt.migrate.aio.operations.SqlRunAsAccountsControllerOperations + :ivar sql_availability_groups_controller: SqlAvailabilityGroupsControllerOperations operations + :vartype sql_availability_groups_controller: + azure.mgmt.migrate.aio.operations.SqlAvailabilityGroupsControllerOperations + :ivar sql_databases_controller: SqlDatabasesControllerOperations operations + :vartype sql_databases_controller: + azure.mgmt.migrate.aio.operations.SqlDatabasesControllerOperations + :ivar sql_servers_controller: SqlServersControllerOperations operations + :vartype sql_servers_controller: + azure.mgmt.migrate.aio.operations.SqlServersControllerOperations + :ivar web_app_sites_controller: WebAppSitesControllerOperations operations + :vartype web_app_sites_controller: + azure.mgmt.migrate.aio.operations.WebAppSitesControllerOperations + :ivar web_app_discovery_site_data_sources_controller: + WebAppDiscoverySiteDataSourcesControllerOperations operations + :vartype web_app_discovery_site_data_sources_controller: + azure.mgmt.migrate.aio.operations.WebAppDiscoverySiteDataSourcesControllerOperations + :ivar web_app_extended_machines_controller: WebAppExtendedMachinesControllerOperations + operations + :vartype web_app_extended_machines_controller: + azure.mgmt.migrate.aio.operations.WebAppExtendedMachinesControllerOperations + :ivar iis_web_applications_controller: IisWebApplicationsControllerOperations operations + :vartype iis_web_applications_controller: + azure.mgmt.migrate.aio.operations.IisWebApplicationsControllerOperations + :ivar iis_web_servers_controller: IisWebServersControllerOperations operations + :vartype iis_web_servers_controller: + azure.mgmt.migrate.aio.operations.IisWebServersControllerOperations + :ivar web_app_run_as_accounts_controller: WebAppRunAsAccountsControllerOperations operations + :vartype web_app_run_as_accounts_controller: + azure.mgmt.migrate.aio.operations.WebAppRunAsAccountsControllerOperations + :ivar tomcat_web_applications_controller: TomcatWebApplicationsControllerOperations operations + :vartype tomcat_web_applications_controller: + azure.mgmt.migrate.aio.operations.TomcatWebApplicationsControllerOperations + :ivar tomcat_web_servers_controller: TomcatWebServersControllerOperations operations + :vartype tomcat_web_servers_controller: + azure.mgmt.migrate.aio.operations.TomcatWebServersControllerOperations + :ivar web_app_properties_controller: WebAppPropertiesControllerOperations operations + :vartype web_app_properties_controller: + azure.mgmt.migrate.aio.operations.WebAppPropertiesControllerOperations + :ivar web_applications_controller: WebApplicationsControllerOperations operations + :vartype web_applications_controller: + azure.mgmt.migrate.aio.operations.WebApplicationsControllerOperations + :ivar web_servers_controller: WebServersControllerOperations operations + :vartype web_servers_controller: + azure.mgmt.migrate.aio.operations.WebServersControllerOperations + :ivar server_dependency_map_controller: ServerDependencyMapControllerOperations operations + :vartype server_dependency_map_controller: + azure.mgmt.migrate.aio.operations.ServerDependencyMapControllerOperations + :ivar server_jobs_controller: ServerJobsControllerOperations operations + :vartype server_jobs_controller: + azure.mgmt.migrate.aio.operations.ServerJobsControllerOperations + :ivar servers_controller: ServersControllerOperations operations + :vartype servers_controller: azure.mgmt.migrate.aio.operations.ServersControllerOperations + :ivar server_software_inventories_controller: ServerSoftwareInventoriesControllerOperations + operations + :vartype server_software_inventories_controller: + azure.mgmt.migrate.aio.operations.ServerSoftwareInventoriesControllerOperations + :ivar server_operations_status_controller: ServerOperationsStatusControllerOperations + operations + :vartype server_operations_status_controller: + azure.mgmt.migrate.aio.operations.ServerOperationsStatusControllerOperations + :ivar server_run_as_accounts_controller: ServerRunAsAccountsControllerOperations operations + :vartype server_run_as_accounts_controller: + azure.mgmt.migrate.aio.operations.ServerRunAsAccountsControllerOperations + :ivar dependency_map_controller: DependencyMapControllerOperations operations + :vartype dependency_map_controller: + azure.mgmt.migrate.aio.operations.DependencyMapControllerOperations + :ivar vmware_host_controller: VmwareHostControllerOperations operations + :vartype vmware_host_controller: + azure.mgmt.migrate.aio.operations.VmwareHostControllerOperations + :ivar hyperv_jobs_controller: HypervJobsControllerOperations operations + :vartype hyperv_jobs_controller: + azure.mgmt.migrate.aio.operations.HypervJobsControllerOperations + :ivar machines_controller: MachinesControllerOperations operations + :vartype machines_controller: azure.mgmt.migrate.aio.operations.MachinesControllerOperations + :ivar vmware_software_inventories_controller: VmwareSoftwareInventoriesControllerOperations + operations + :vartype vmware_software_inventories_controller: + azure.mgmt.migrate.aio.operations.VmwareSoftwareInventoriesControllerOperations + :ivar vmware_operations_status: VmwareOperationsStatusOperations operations + :vartype vmware_operations_status: + azure.mgmt.migrate.aio.operations.VmwareOperationsStatusOperations + :ivar run_as_accounts_controller: RunAsAccountsControllerOperations operations + :vartype run_as_accounts_controller: + azure.mgmt.migrate.aio.operations.RunAsAccountsControllerOperations + :ivar vmware_properties_controller: VmwarePropertiesControllerOperations operations + :vartype vmware_properties_controller: + azure.mgmt.migrate.aio.operations.VmwarePropertiesControllerOperations + :ivar vcenter_controller: VcenterControllerOperations operations + :vartype vcenter_controller: azure.mgmt.migrate.aio.operations.VcenterControllerOperations + :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. 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 "2023-10-01-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 = AzureMigrateDiscoveryServiceConfiguration( + 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.hyperv_sites = HypervSitesOperations(self._client, self._config, self._serialize, self._deserialize) + self.import_sites_controller = ImportSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.master_sites_controller = MasterSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_sites_controller = ServerSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sites_controller = SitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_sites_controller = HypervSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_dependency_map_controller = HypervDependencyMapControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_cluster_controller = HypervClusterControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_host_controller = HypervHostControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_jobs = HypervJobsOperations(self._client, self._config, self._serialize, self._deserialize) + self.hyperv_machines_controller = HypervMachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_software_inventories_controller = HypervSoftwareInventoriesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_operations_status_controller = HypervOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_run_as_accounts_controller = HypervRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.import_jobs_controller = ImportJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.import_machines_controller = ImportMachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.master_sites_operations_status_controller = MasterSitesOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.private_endpoint_connection_controller = PrivateEndpointConnectionControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.private_link_resources_controller = PrivateLinkResourcesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_sites_controller = SqlSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_discovery_site_data_source_controller = SqlDiscoverySiteDataSourceControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_jobs_controller = SqlJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_operations_status_controller = SqlOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_run_as_accounts_controller = SqlRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_availability_groups_controller = SqlAvailabilityGroupsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_databases_controller = SqlDatabasesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.sql_servers_controller = SqlServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_sites_controller = WebAppSitesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_discovery_site_data_sources_controller = WebAppDiscoverySiteDataSourcesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_extended_machines_controller = WebAppExtendedMachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.iis_web_applications_controller = IisWebApplicationsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.iis_web_servers_controller = IisWebServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_run_as_accounts_controller = WebAppRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.tomcat_web_applications_controller = TomcatWebApplicationsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.tomcat_web_servers_controller = TomcatWebServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_app_properties_controller = WebAppPropertiesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_applications_controller = WebApplicationsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.web_servers_controller = WebServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_dependency_map_controller = ServerDependencyMapControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_jobs_controller = ServerJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.servers_controller = ServersControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_software_inventories_controller = ServerSoftwareInventoriesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_operations_status_controller = ServerOperationsStatusControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.server_run_as_accounts_controller = ServerRunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.dependency_map_controller = DependencyMapControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_host_controller = VmwareHostControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.hyperv_jobs_controller = HypervJobsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.machines_controller = MachinesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_software_inventories_controller = VmwareSoftwareInventoriesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_operations_status = VmwareOperationsStatusOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.run_as_accounts_controller = RunAsAccountsControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vmware_properties_controller = VmwarePropertiesControllerOperations( + self._client, self._config, self._serialize, self._deserialize + ) + self.vcenter_controller = VcenterControllerOperations( + 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/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_configuration.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_configuration.py new file mode 100644 index 000000000000..babfd38390fd --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_configuration.py @@ -0,0 +1,65 @@ +# 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 typing import Any, TYPE_CHECKING + +from azure.core.pipeline import policies +from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy + +from .._version import VERSION + +if TYPE_CHECKING: + # pylint: disable=unused-import,ungrouped-imports + from azure.core.credentials_async import AsyncTokenCredential + + +class AzureMigrateDiscoveryServiceConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + """Configuration for AzureMigrateDiscoveryService. + + 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. + :type credential: ~azure.core.credentials_async.AsyncTokenCredential + :param subscription_id: The ID of the target subscription. Required. + :type subscription_id: str + :keyword api_version: Api Version. Default value is "2023-10-01-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", "2023-10-01-preview") + + if credential is None: + raise ValueError("Parameter 'credential' must not be None.") + if subscription_id is None: + raise ValueError("Parameter 'subscription_id' must not be None.") + + self.credential = credential + self.subscription_id = subscription_id + self.api_version = api_version + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "mgmt-migrate/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) + self._configure(**kwargs) + + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") + if self.credential and not self.authentication_policy: + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_patch.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/__init__.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/__init__.py new file mode 100644 index 000000000000..4f913fc50d15 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/__init__.py @@ -0,0 +1,127 @@ +# 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 ._operations import Operations +from ._hyperv_sites_operations import HypervSitesOperations +from ._import_sites_controller_operations import ImportSitesControllerOperations +from ._master_sites_controller_operations import MasterSitesControllerOperations +from ._server_sites_controller_operations import ServerSitesControllerOperations +from ._sites_controller_operations import SitesControllerOperations +from ._hyperv_sites_controller_operations import HypervSitesControllerOperations +from ._hyperv_dependency_map_controller_operations import HypervDependencyMapControllerOperations +from ._hyperv_cluster_controller_operations import HypervClusterControllerOperations +from ._hyperv_host_controller_operations import HypervHostControllerOperations +from ._hyperv_jobs_operations import HypervJobsOperations +from ._hyperv_machines_controller_operations import HypervMachinesControllerOperations +from ._hyperv_software_inventories_controller_operations import HypervSoftwareInventoriesControllerOperations +from ._hyperv_operations_status_controller_operations import HypervOperationsStatusControllerOperations +from ._hyperv_run_as_accounts_controller_operations import HypervRunAsAccountsControllerOperations +from ._import_jobs_controller_operations import ImportJobsControllerOperations +from ._import_machines_controller_operations import ImportMachinesControllerOperations +from ._master_sites_operations_status_controller_operations import MasterSitesOperationsStatusControllerOperations +from ._private_endpoint_connection_controller_operations import PrivateEndpointConnectionControllerOperations +from ._private_link_resources_controller_operations import PrivateLinkResourcesControllerOperations +from ._sql_sites_controller_operations import SqlSitesControllerOperations +from ._sql_discovery_site_data_source_controller_operations import SqlDiscoverySiteDataSourceControllerOperations +from ._sql_jobs_controller_operations import SqlJobsControllerOperations +from ._sql_operations_status_controller_operations import SqlOperationsStatusControllerOperations +from ._sql_run_as_accounts_controller_operations import SqlRunAsAccountsControllerOperations +from ._sql_availability_groups_controller_operations import SqlAvailabilityGroupsControllerOperations +from ._sql_databases_controller_operations import SqlDatabasesControllerOperations +from ._sql_servers_controller_operations import SqlServersControllerOperations +from ._web_app_sites_controller_operations import WebAppSitesControllerOperations +from ._web_app_discovery_site_data_sources_controller_operations import ( + WebAppDiscoverySiteDataSourcesControllerOperations, +) +from ._web_app_extended_machines_controller_operations import WebAppExtendedMachinesControllerOperations +from ._iis_web_applications_controller_operations import IisWebApplicationsControllerOperations +from ._iis_web_servers_controller_operations import IisWebServersControllerOperations +from ._web_app_run_as_accounts_controller_operations import WebAppRunAsAccountsControllerOperations +from ._tomcat_web_applications_controller_operations import TomcatWebApplicationsControllerOperations +from ._tomcat_web_servers_controller_operations import TomcatWebServersControllerOperations +from ._web_app_properties_controller_operations import WebAppPropertiesControllerOperations +from ._web_applications_controller_operations import WebApplicationsControllerOperations +from ._web_servers_controller_operations import WebServersControllerOperations +from ._server_dependency_map_controller_operations import ServerDependencyMapControllerOperations +from ._server_jobs_controller_operations import ServerJobsControllerOperations +from ._servers_controller_operations import ServersControllerOperations +from ._server_software_inventories_controller_operations import ServerSoftwareInventoriesControllerOperations +from ._server_operations_status_controller_operations import ServerOperationsStatusControllerOperations +from ._server_run_as_accounts_controller_operations import ServerRunAsAccountsControllerOperations +from ._dependency_map_controller_operations import DependencyMapControllerOperations +from ._vmware_host_controller_operations import VmwareHostControllerOperations +from ._hyperv_jobs_controller_operations import HypervJobsControllerOperations +from ._machines_controller_operations import MachinesControllerOperations +from ._vmware_software_inventories_controller_operations import VmwareSoftwareInventoriesControllerOperations +from ._vmware_operations_status_operations import VmwareOperationsStatusOperations +from ._run_as_accounts_controller_operations import RunAsAccountsControllerOperations +from ._vmware_properties_controller_operations import VmwarePropertiesControllerOperations +from ._vcenter_controller_operations import VcenterControllerOperations + +from ._patch import __all__ as _patch_all +from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import patch_sdk as _patch_sdk + +__all__ = [ + "Operations", + "HypervSitesOperations", + "ImportSitesControllerOperations", + "MasterSitesControllerOperations", + "ServerSitesControllerOperations", + "SitesControllerOperations", + "HypervSitesControllerOperations", + "HypervDependencyMapControllerOperations", + "HypervClusterControllerOperations", + "HypervHostControllerOperations", + "HypervJobsOperations", + "HypervMachinesControllerOperations", + "HypervSoftwareInventoriesControllerOperations", + "HypervOperationsStatusControllerOperations", + "HypervRunAsAccountsControllerOperations", + "ImportJobsControllerOperations", + "ImportMachinesControllerOperations", + "MasterSitesOperationsStatusControllerOperations", + "PrivateEndpointConnectionControllerOperations", + "PrivateLinkResourcesControllerOperations", + "SqlSitesControllerOperations", + "SqlDiscoverySiteDataSourceControllerOperations", + "SqlJobsControllerOperations", + "SqlOperationsStatusControllerOperations", + "SqlRunAsAccountsControllerOperations", + "SqlAvailabilityGroupsControllerOperations", + "SqlDatabasesControllerOperations", + "SqlServersControllerOperations", + "WebAppSitesControllerOperations", + "WebAppDiscoverySiteDataSourcesControllerOperations", + "WebAppExtendedMachinesControllerOperations", + "IisWebApplicationsControllerOperations", + "IisWebServersControllerOperations", + "WebAppRunAsAccountsControllerOperations", + "TomcatWebApplicationsControllerOperations", + "TomcatWebServersControllerOperations", + "WebAppPropertiesControllerOperations", + "WebApplicationsControllerOperations", + "WebServersControllerOperations", + "ServerDependencyMapControllerOperations", + "ServerJobsControllerOperations", + "ServersControllerOperations", + "ServerSoftwareInventoriesControllerOperations", + "ServerOperationsStatusControllerOperations", + "ServerRunAsAccountsControllerOperations", + "DependencyMapControllerOperations", + "VmwareHostControllerOperations", + "HypervJobsControllerOperations", + "MachinesControllerOperations", + "VmwareSoftwareInventoriesControllerOperations", + "VmwareOperationsStatusOperations", + "RunAsAccountsControllerOperations", + "VmwarePropertiesControllerOperations", + "VcenterControllerOperations", +] +__all__.extend([p for p in _patch_all if p not in __all__]) +_patch_sdk() diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_dependency_map_controller_operations.py new file mode 100644 index 000000000000..38d0ceefd987 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_dependency_map_controller_operations.py @@ -0,0 +1,1057 @@ +# 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._dependency_map_controller_operations import ( + build_client_group_members_request, + build_export_dependencies_request, + build_generate_coarse_map_request, + build_generate_detailed_map_request, + build_server_group_members_request, +) + +if sys.version_info >= (3, 9): + from collections.abc import MutableMapping +else: + from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class DependencyMapControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`dependency_map_controller` 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 _client_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsClientGroupMembersRequest") + + _request = build_client_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsClientGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsClientGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._client_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_dependencies_initial( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **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(request_body, (IOBase, bytes)): + _content = request_body + else: + _json = self._serialize.body(request_body, "DependencyMapServiceMapextensionsExportDependenciesRequest") + + _request = build_export_dependencies_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: _models.DependencyMapServiceMapextensionsExportDependenciesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Is either a + DependencyMapServiceMapextensionsExportDependenciesRequest type or a IO[bytes] type. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_dependencies_initial( + resource_group_name=resource_group_name, + site_name=site_name, + request_body=request_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _generate_coarse_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsScopeMapRequest") + + _request = build_generate_coarse_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsScopeMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsScopeMapRequest type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._generate_coarse_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _generate_detailed_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body( + map_request, "DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest" + ) + + _request = build_generate_detailed_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest type or a IO[bytes] type. + Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._generate_detailed_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _server_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsServerGroupMembersRequest") + + _request = build_server_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsServerGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsServerGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._server_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_cluster_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_cluster_controller_operations.py new file mode 100644 index 000000000000..983245367380 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_cluster_controller_operations.py @@ -0,0 +1,483 @@ +# 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._hyperv_cluster_controller_operations import ( + build_create_cluster_request, + build_delete_request, + build_get_cluster_request, + build_list_by_hyperv_site_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 HypervClusterControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_cluster_controller` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, filter: Optional[str] = None, **kwargs: Any + ) -> AsyncIterable["_models.HypervCluster"]: + """List HypervCluster resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :return: An iterator like instance of either HypervCluster or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervCluster] + :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.HypervClusterListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + 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("HypervClusterListResult", 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_cluster( + self, resource_group_name: str, site_name: str, cluster_name: str, **kwargs: Any + ) -> _models.HypervCluster: + """Method to get a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :return: HypervCluster or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervCluster + :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.HypervCluster] = kwargs.pop("cls", None) + + _request = build_get_cluster_request( + resource_group_name=resource_group_name, + site_name=site_name, + cluster_name=cluster_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("HypervCluster", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _create_cluster_initial( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: Union[_models.HypervCluster, IO[bytes]], + **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: + _json = self._serialize.body(body, "HypervCluster") + + _request = build_create_cluster_request( + resource_group_name=resource_group_name, + site_name=site_name, + cluster_name=cluster_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create_cluster( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: _models.HypervCluster, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.HypervCluster]: + """Method to create or update a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.HypervCluster + :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 HypervCluster or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.HypervCluster] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create_cluster( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.HypervCluster]: + """Method to create or update a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :param body: Resource create 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 either HypervCluster or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.HypervCluster] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create_cluster( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: Union[_models.HypervCluster, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.HypervCluster]: + """Method to create or update a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :param body: Resource create parameters. Is either a HypervCluster type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.HypervCluster or IO[bytes] + :return: An instance of AsyncLROPoller that returns either HypervCluster or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.HypervCluster] + :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.HypervCluster] = 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_cluster_initial( + resource_group_name=resource_group_name, + site_name=site_name, + cluster_name=cluster_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("HypervCluster", 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.HypervCluster].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.HypervCluster]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, cluster_name: str, **kwargs: Any + ) -> None: + """Delete a HypervCluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_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, + site_name=site_name, + cluster_name=cluster_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_dependency_map_controller_operations.py new file mode 100644 index 000000000000..b6fb9ef657a7 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_dependency_map_controller_operations.py @@ -0,0 +1,1256 @@ +# 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._hyperv_dependency_map_controller_operations import ( + build_client_group_members_request, + build_export_dependencies_request, + build_generate_coarse_map_request, + build_generate_detailed_map_request, + build_server_group_members_request, + build_update_dependency_map_status_request, +) + +if sys.version_info >= (3, 9): + from collections.abc import MutableMapping +else: + from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class HypervDependencyMapControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_dependency_map_controller` 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 _client_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsClientGroupMembersRequest") + + _request = build_client_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsClientGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsClientGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._client_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_dependencies_initial( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **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(request_body, (IOBase, bytes)): + _content = request_body + else: + _json = self._serialize.body(request_body, "DependencyMapServiceMapextensionsExportDependenciesRequest") + + _request = build_export_dependencies_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: _models.DependencyMapServiceMapextensionsExportDependenciesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Is either a + DependencyMapServiceMapextensionsExportDependenciesRequest type or a IO[bytes] type. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_dependencies_initial( + resource_group_name=resource_group_name, + site_name=site_name, + request_body=request_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _generate_coarse_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsScopeMapRequest") + + _request = build_generate_coarse_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsScopeMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsScopeMapRequest type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._generate_coarse_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _generate_detailed_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body( + map_request, "DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest" + ) + + _request = build_generate_detailed_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest type or a IO[bytes] type. + Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._generate_detailed_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _server_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsServerGroupMembersRequest") + + _request = build_server_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsServerGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsServerGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._server_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _update_dependency_map_status_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "UpdateMachineDepMapStatus") + + _request = build_update_dependency_map_status_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: _models.UpdateMachineDepMapStatus, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a UpdateMachineDepMapStatus + type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_dependency_map_status_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_host_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_host_controller_operations.py new file mode 100644 index 000000000000..4e6a0a15f903 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_host_controller_operations.py @@ -0,0 +1,541 @@ +# 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._hyperv_host_controller_operations import ( + build_create_request, + build_delete_request, + build_get_request, + build_list_by_hyperv_site_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 HypervHostControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_host_controller` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, filter: Optional[str] = None, **kwargs: Any + ) -> AsyncIterable["_models.HypervHost"]: + """List HypervHost resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :return: An iterator like instance of either HypervHost or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervHost] + :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.HypervHostListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + 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("HypervHostListResult", 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, site_name: str, host_name: str, **kwargs: Any) -> _models.HypervHost: + """Get a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :return: HypervHost or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervHost + :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.HypervHost] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + host_name=host_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("HypervHost", 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, + site_name: str, + host_name: str, + body: Union[_models.HypervHost, IO[bytes]], + **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: + _json = self._serialize.body(body, "HypervHost") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + host_name=host_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + host_name: str, + body: _models.HypervHost, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.HypervHost]: + """Create a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.HypervHost + :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 HypervHost or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.HypervHost] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + site_name: str, + host_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.HypervHost]: + """Create a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :param body: Resource create 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 either HypervHost or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.HypervHost] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + site_name: str, + host_name: str, + body: Union[_models.HypervHost, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.HypervHost]: + """Create a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :param body: Resource create parameters. Is either a HypervHost type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.HypervHost or IO[bytes] + :return: An instance of AsyncLROPoller that returns either HypervHost or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.HypervHost] + :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.HypervHost] = 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, + site_name=site_name, + host_name=host_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("HypervHost", 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.HypervHost].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.HypervHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _delete_initial( + self, resource_group_name: str, site_name: str, host_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, + site_name=site_name, + host_name=host_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, site_name: str, host_name: str, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Delete a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_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, + site_name=site_name, + host_name=host_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/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_controller_operations.py new file mode 100644 index 000000000000..43da69e9b774 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_controller_operations.py @@ -0,0 +1,200 @@ +# 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.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._hyperv_jobs_controller_operations import build_get_request, build_list_by_vmware_site_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 HypervJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_jobs_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.VmwareJob"]: + """List VmwareJob resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either VmwareJob or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.VmwareJob] + :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.VmwareJobListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.VmwareJob: + """Get a VmwareJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Jobs name. Required. + :type job_name: str + :return: VmwareJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareJob + :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.VmwareJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("VmwareJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_operations.py new file mode 100644 index 000000000000..512e6c2008a5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_jobs_operations.py @@ -0,0 +1,200 @@ +# 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.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._hyperv_jobs_operations import build_get_request, build_list_by_hyperv_site_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 HypervJobsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_jobs` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.HypervJob"]: + """List HypervJob resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either HypervJob or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervJob] + :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.HypervJobListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.HypervJob: + """Get a HypervJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job name. Required. + :type job_name: str + :return: HypervJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervJob + :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.HypervJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("HypervJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_machines_controller_operations.py new file mode 100644 index 000000000000..fc939931976e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_machines_controller_operations.py @@ -0,0 +1,572 @@ +# 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._hyperv_machines_controller_operations import ( + build_get_request, + build_list_by_hyperv_site_request, + build_update_properties_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class HypervMachinesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_machines_controller` 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_by_hyperv_site( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[int] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.HypervMachine"]: + """List HypervMachine resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: int + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either HypervMachine or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervMachine] + :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.HypervMachineListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("HypervMachineListResult", 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, site_name: str, machine_name: str, **kwargs: Any + ) -> _models.HypervMachine: + """Get a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :return: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :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.HypervMachine] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("HypervMachine", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: _models.HypervMachineUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervMachine: + """Update a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.HypervMachineUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervMachine: + """Update a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. 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: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: Union[_models.HypervMachineUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.HypervMachine: + """Update a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Is either a HypervMachineUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.HypervMachineUpdate or IO[bytes] + :return: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :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.HypervMachine] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "HypervMachineUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("HypervMachine", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _update_properties_initial( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **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(meta_data, (IOBase, bytes)): + _content = meta_data + else: + _json = self._serialize.body(meta_data, "MachineMetadataCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: _models.MachineMetadataCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update custom properties for HYPERV machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update custom properties for HYPERV machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update custom properties for HYPERV machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Is either a MachineMetadataCollection type + or a IO[bytes] type. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + meta_data=meta_data, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_operations_status_controller_operations.py new file mode 100644 index 000000000000..6b0e5d05ba66 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_operations_status_controller_operations.py @@ -0,0 +1,115 @@ +# 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._hyperv_operations_status_controller_operations import build_get_hyperv_operations_status_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 HypervOperationsStatusControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_operations_status_controller` 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_hyperv_operations_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """Method to get operation status. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_hyperv_operations_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..c30ddd2869cf --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_run_as_accounts_controller_operations.py @@ -0,0 +1,207 @@ +# 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.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._hyperv_run_as_accounts_controller_operations import ( + build_get_request, + build_list_by_hyperv_site_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 HypervRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_run_as_accounts_controller` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.HypervRunAsAccountResource"]: + """List HypervRunAsAccountResource resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either HypervRunAsAccountResource or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervRunAsAccountResource] + :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.HypervRunAsAccountResourceListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervRunAsAccountResourceListResult", 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, site_name: str, account_name: str, **kwargs: Any + ) -> _models.HypervRunAsAccountResource: + """Get a HypervRunAsAccountResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param account_name: RunAsAccounts name. Required. + :type account_name: str + :return: HypervRunAsAccountResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervRunAsAccountResource + :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.HypervRunAsAccountResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + account_name=account_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("HypervRunAsAccountResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_controller_operations.py new file mode 100644 index 000000000000..18b626a1a918 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_controller_operations.py @@ -0,0 +1,1007 @@ +# 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._hyperv_sites_controller_operations import ( + build_compute_error_summary_request, + build_computeusage_request, + build_create_request, + build_delete_request, + build_export_applications_request, + build_export_machine_errors_request, + build_get_request, + build_list_health_summary_request, + build_summary_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class HypervSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_sites_controller` 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, site_name: str, **kwargs: Any) -> _models.HypervSite: + """Get a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :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.HypervSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + body: _models.HypervSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Create a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.HypervSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Create a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create( + self, resource_group_name: str, site_name: str, body: Union[_models.HypervSite, IO[bytes]], **kwargs: Any + ) -> _models.HypervSite: + """Create a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a HypervSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.HypervSite or IO[bytes] + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :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.HypervSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "HypervSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("HypervSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: _models.HypervSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Update a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.HypervSiteUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Update a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, resource_group_name: str, site_name: str, body: Union[_models.HypervSiteUpdate, IO[bytes]], **kwargs: Any + ) -> _models.HypervSite: + """Update a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a HypervSiteUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.HypervSiteUpdate or IO[bytes] + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :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.HypervSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "HypervSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("HypervSite", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, 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 compute_error_summary( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get site error summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SiteErrorSummary] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_compute_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def computeusage( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.HypervSiteUsage: + """Get a hyperv site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: HypervSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSiteUsage + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.HypervSiteUsage] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_computeusage_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("HypervSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _export_applications_initial( + self, resource_group_name: str, site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_applications_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_applications( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_applications_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_machine_errors_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachineErrorsRequest") + + _request = build_export_machine_errors_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachineErrorsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachineErrorsRequest type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_machine_errors_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def list_health_summary( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> _models.SiteHealthSummaryCollection: + """Method to get site health summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: SiteHealthSummaryCollection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteHealthSummaryCollection + :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.SiteHealthSummaryCollection] = kwargs.pop("cls", None) + + _request = build_list_health_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("SiteHealthSummaryCollection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def summary(self, resource_group_name: str, site_name: str, **kwargs: Any) -> _models.HypervSiteUsage: + """Method to get site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: HypervSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSiteUsage + :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.HypervSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_operations.py new file mode 100644 index 000000000000..e213ea5f4fe1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_sites_operations.py @@ -0,0 +1,217 @@ +# 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._hyperv_sites_operations import ( + build_list_by_resource_group_request, + build_list_by_subscription_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 HypervSitesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_sites` 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_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.HypervSite"]: + """Get all hyperv sites. + + Get all the hyperv sites in the subscription. + + :return: An iterator like instance of either HypervSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervSite] + :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.HypervSiteListResult] = 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_subscription_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("HypervSiteListResult", 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 + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.HypervSite"]: + """Get all hyperv sites. + + Get all the hyperv sites in the subscription. + + :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 HypervSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervSite] + :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.HypervSiteListResult] = 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("HypervSiteListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_software_inventories_controller_operations.py new file mode 100644 index 000000000000..2dbc7f1aaf04 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_hyperv_software_inventories_controller_operations.py @@ -0,0 +1,218 @@ +# 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, Union +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._hyperv_software_inventories_controller_operations import ( + build_get_machine_software_inventory_request, + build_list_by_hyperv_machine_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 HypervSoftwareInventoriesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`hyperv_software_inventories_controller` 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_machine_software_inventory( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + **kwargs: Any + ) -> _models.HypervVmSoftwareInventory: + """Method to get a machines software inventory like applications and roles. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param default: Default value. "default" Required. + :type default: str or ~azure.mgmt.migrate.models.Default + :return: HypervVmSoftwareInventory or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervVmSoftwareInventory + :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.HypervVmSoftwareInventory] = kwargs.pop("cls", None) + + _request = build_get_machine_software_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_name, + default=default, + 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("HypervVmSoftwareInventory", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_hyperv_machine( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> AsyncIterable["_models.HypervVmSoftwareInventory"]: + """List HypervVmSoftwareInventory resources by HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :return: An iterator like instance of either HypervVmSoftwareInventory or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.HypervVmSoftwareInventory] + :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.HypervVmSoftwareInventoryListResult] = 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_hyperv_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("HypervVmSoftwareInventoryListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_applications_controller_operations.py new file mode 100644 index 000000000000..8675356ad70d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_applications_controller_operations.py @@ -0,0 +1,385 @@ +# 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._iis_web_applications_controller_operations import ( + build_get_request, + build_list_by_web_app_site_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 IisWebApplicationsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`iis_web_applications_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.IisWebApplications"]: + """Method to get all IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either IisWebApplications or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.IisWebApplications] + :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.IisWebApplicationsListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("IisWebApplicationsListResult", 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, site_name: str, web_app_site_name: str, web_application_name: str, **kwargs: Any + ) -> _models.IisWebApplications: + """Method to get an IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :return: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :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.IisWebApplications] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_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("IisWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: _models.IisWebApplicationsUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.IisWebApplications: + """Method to update tags on IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.IisWebApplicationsUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.IisWebApplications: + """Method to update tags on IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. 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: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: Union[_models.IisWebApplicationsUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.IisWebApplications: + """Method to update tags on IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. Is either a IisWebApplicationsUpdate type + or a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.IisWebApplicationsUpdate or IO[bytes] + :return: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :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.IisWebApplications] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "IisWebApplicationsUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("IisWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_servers_controller_operations.py new file mode 100644 index 000000000000..18bb5321f242 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_iis_web_servers_controller_operations.py @@ -0,0 +1,229 @@ +# 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.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._iis_web_servers_controller_operations import build_get_request, build_list_by_web_app_site_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 IisWebServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`iis_web_servers_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.IisWebServers"]: + """Method to get all IIS web servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either IisWebServers or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.IisWebServers] + :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.IisWebServersListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("IisWebServersListResult", 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, site_name: str, web_app_site_name: str, web_server_name: str, **kwargs: Any + ) -> _models.IisWebServers: + """Method to get an IIS web server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_server_name: Web server name. Required. + :type web_server_name: str + :return: IisWebServers or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebServers + :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.IisWebServers] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_server_name=web_server_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("IisWebServers", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_jobs_controller_operations.py new file mode 100644 index 000000000000..931d5db28c85 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_jobs_controller_operations.py @@ -0,0 +1,648 @@ +# 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.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._import_jobs_controller_operations import ( + build_get_deletejob_request, + build_get_exportjob_request, + build_get_importjob_request, + build_get_request, + build_list_by_import_site_request, + build_list_deletejobs_request, + build_list_exportjobs_request, + build_list_importjobs_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 ImportJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`import_jobs_controller` 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_deletejobs( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.DeleteImportMachinesJob"]: + """Method to get all delete import machines job for the given site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either DeleteImportMachinesJob or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.DeleteImportMachinesJob] + :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.PagedDeleteImportMachinesJob] = 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_deletejobs_request( + resource_group_name=resource_group_name, + site_name=site_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("PagedDeleteImportMachinesJob", 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_deletejob( + self, resource_group_name: str, site_name: str, job_name: str, **kwargs: Any + ) -> _models.DeleteImportMachinesJob: + """Gets the delete imported machines job with the given job name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job Arm Name. Required. + :type job_name: str + :return: DeleteImportMachinesJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.DeleteImportMachinesJob + :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.DeleteImportMachinesJob] = kwargs.pop("cls", None) + + _request = build_get_deletejob_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("DeleteImportMachinesJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_exportjobs( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ExportImportedMachinesJob"]: + """Method to get all export import machines job for the given site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ExportImportedMachinesJob or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ExportImportedMachinesJob] + :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.PagedExportImportedMachinesJob] = 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_exportjobs_request( + resource_group_name=resource_group_name, + site_name=site_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("PagedExportImportedMachinesJob", 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_exportjob( + self, resource_group_name: str, site_name: str, job_name: str, **kwargs: Any + ) -> _models.ExportImportedMachinesJob: + """Gets the export imported machines job with the given job name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job Arm Name. Required. + :type job_name: str + :return: ExportImportedMachinesJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ExportImportedMachinesJob + :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.ExportImportedMachinesJob] = kwargs.pop("cls", None) + + _request = build_get_exportjob_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ExportImportedMachinesJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_importjobs( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ImportMachinesJob"]: + """Method to get all import machines job for the given site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ImportMachinesJob or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ImportMachinesJob] + :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.PagedImportMachinesJob] = 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_importjobs_request( + resource_group_name=resource_group_name, + site_name=site_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("PagedImportMachinesJob", 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_importjob( + self, resource_group_name: str, site_name: str, job_name: str, **kwargs: Any + ) -> _models.ImportMachinesJob: + """Gets the import job with the given job name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job Arm Name. Required. + :type job_name: str + :return: ImportMachinesJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportMachinesJob + :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.ImportMachinesJob] = kwargs.pop("cls", None) + + _request = build_get_importjob_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ImportMachinesJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_import_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ImportJob"]: + """List ImportJob resources by ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ImportJob or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ImportJob] + :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.ImportJobListResult] = 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_import_site_request( + resource_group_name=resource_group_name, + site_name=site_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("ImportJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.ImportJob: + """Get a ImportJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Jobs name. Required. + :type job_name: str + :return: ImportJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportJob + :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.ImportJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ImportJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_machines_controller_operations.py new file mode 100644 index 000000000000..ba5ca154d803 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_machines_controller_operations.py @@ -0,0 +1,283 @@ +# 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.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._import_machines_controller_operations import ( + build_delete_request, + build_get_request, + build_list_by_import_site_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 ImportMachinesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`import_machines_controller` 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_by_import_site( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.ImportMachine"]: + """List ImportMachine resources by ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either ImportMachine or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ImportMachine] + :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.ImportMachineListResult] = 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_import_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("ImportMachineListResult", 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, site_name: str, machine_name: str, **kwargs: Any + ) -> _models.ImportMachine: + """Get a ImportMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: ImportMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportMachine + :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.ImportMachine] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("ImportMachine", 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, site_name: str, machine_name: str, **kwargs: Any + ) -> None: + """Delete a ImportMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_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, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_sites_controller_operations.py new file mode 100644 index 000000000000..ba1718c96464 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_import_sites_controller_operations.py @@ -0,0 +1,910 @@ +# 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._import_sites_controller_operations import ( + build_create_request, + build_delete_imported_machines_request, + build_delete_request, + build_export_uri_request, + build_get_request, + build_import_uri_request, + build_list_by_resource_group_request, + build_list_by_subscription_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class ImportSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`import_sites_controller` 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_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ImportSite"]: + """List ImportSite resources by subscription ID. + + :return: An iterator like instance of either ImportSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ImportSite] + :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.ImportSiteListResult] = 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_subscription_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("ImportSiteListResult", 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 + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.ImportSite"]: + """Get all import sites. + + :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 ImportSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ImportSite] + :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.ImportSiteListResult] = 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("ImportSiteListResult", 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, site_name: str, **kwargs: Any) -> _models.ImportSite: + """Get a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :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.ImportSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("ImportSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + body: _models.ImportSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Create a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.ImportSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Create a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create( + self, resource_group_name: str, site_name: str, body: Union[_models.ImportSite, IO[bytes]], **kwargs: Any + ) -> _models.ImportSite: + """Create a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a ImportSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.ImportSite or IO[bytes] + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :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.ImportSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ImportSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("ImportSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: _models.ImportSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Update a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.ImportSiteUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Update a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, resource_group_name: str, site_name: str, body: Union[_models.ImportSiteUpdate, IO[bytes]], **kwargs: Any + ) -> _models.ImportSite: + """Update a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a ImportSiteUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ImportSiteUpdate or IO[bytes] + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :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.ImportSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ImportSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("ImportSite", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, 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 delete_imported_machines( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SasUriResponse: + """Deletes the imported machines for site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SasUriResponse] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_delete_imported_machines_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SasUriResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def export_uri( + self, + resource_group_name: str, + site_name: str, + body: _models.SasUriResponse, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to export a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def export_uri( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to export a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def export_uri( + self, resource_group_name: str, site_name: str, body: Union[_models.SasUriResponse, IO[bytes]], **kwargs: Any + ) -> _models.SasUriResponse: + """Method to export a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a SasUriResponse type or a IO[bytes] + type. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse or IO[bytes] + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :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.SasUriResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SasUriResponse") + + _request = build_export_uri_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SasUriResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def import_uri( + self, + resource_group_name: str, + site_name: str, + body: _models.SasUriResponse, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to import a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def import_uri( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to import a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def import_uri( + self, resource_group_name: str, site_name: str, body: Union[_models.SasUriResponse, IO[bytes]], **kwargs: Any + ) -> _models.SasUriResponse: + """Method to import a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a SasUriResponse type or a IO[bytes] + type. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse or IO[bytes] + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :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.SasUriResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SasUriResponse") + + _request = build_import_uri_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SasUriResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_machines_controller_operations.py new file mode 100644 index 000000000000..101b42befb0f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_machines_controller_operations.py @@ -0,0 +1,613 @@ +# 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._machines_controller_operations import ( + build_get_request, + build_list_by_vmware_site_request, + build_start_request, + build_stop_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class MachinesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`machines_controller` 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_by_vmware_site( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[int] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.MachineResource"]: + """List MachineResource resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: int + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either MachineResource or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.MachineResource] + :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.MachineResourceListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("MachineResourceListResult", 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, site_name: str, machine_name: str, **kwargs: Any + ) -> _models.MachineResource: + """Get a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :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.MachineResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("MachineResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: _models.MachineResourceUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MachineResource: + """Update a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.MachineResourceUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MachineResource: + """Update a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. 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: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: Union[_models.MachineResourceUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.MachineResource: + """Update a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Is either a MachineResourceUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.MachineResourceUpdate or IO[bytes] + :return: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :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.MachineResource] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "MachineResourceUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("MachineResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _start_initial( + self, resource_group_name: str, site_name: str, machine_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_start_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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]: + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_start( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to start a machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._start_initial( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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): + deserialized = self._deserialize("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _stop_initial( + self, resource_group_name: str, site_name: str, machine_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_stop_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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]: + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_stop( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to stop a machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._stop_initial( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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): + deserialized = self._deserialize("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_controller_operations.py new file mode 100644 index 000000000000..dce1d61ff56c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_controller_operations.py @@ -0,0 +1,862 @@ +# 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._master_sites_controller_operations import ( + build_create_request, + build_delete_request, + build_error_summary_request, + build_get_request, + build_list_by_resource_group_request, + build_list_by_subscription_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 MasterSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`master_sites_controller` 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_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.MasterSite"]: + """Get all the sites in the subscription. + + List MasterSite resources by subscription ID. + + :return: An iterator like instance of either MasterSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSiteListResult] = 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_subscription_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("MasterSiteListResult", 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 + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.MasterSite"]: + """Get all sites. + + Get all the sites in the 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 MasterSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSiteListResult] = 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("MasterSiteListResult", 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, site_name: str, **kwargs: Any) -> _models.MasterSite: + """Method to get a master site. + + Get a MasterSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: MasterSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MasterSite + :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.MasterSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("MasterSite", 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, site_name: str, body: Union[_models.MasterSite, IO[bytes]], **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: + _json = self._serialize.body(body, "MasterSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + body: _models.MasterSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.MasterSite]: + """Method to create or update a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.MasterSite + :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 MasterSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.MasterSite]: + """Method to create or update a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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 either MasterSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, resource_group_name: str, site_name: str, body: Union[_models.MasterSite, IO[bytes]], **kwargs: Any + ) -> AsyncLROPoller[_models.MasterSite]: + """Method to create or update a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a MasterSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.MasterSite or IO[bytes] + :return: An instance of AsyncLROPoller that returns either MasterSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSite] = 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, + site_name=site_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("MasterSite", 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.MasterSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.MasterSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _update_initial( + self, + resource_group_name: str, + site_name: str, + properties: Union[_models.MasterSiteUpdate, IO[bytes]], + **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(properties, (IOBase, bytes)): + _content = properties + else: + _json = self._serialize.body(properties, "MasterSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + properties: _models.MasterSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.MasterSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: ~azure.mgmt.migrate.models.MasterSiteUpdate + :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 MasterSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + site_name: str, + properties: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.MasterSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: 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 MasterSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update( + self, + resource_group_name: str, + site_name: str, + properties: Union[_models.MasterSiteUpdate, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.MasterSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param properties: The resource properties to be updated. Is either a MasterSiteUpdate type or + a IO[bytes] type. Required. + :type properties: ~azure.mgmt.migrate.models.MasterSiteUpdate or IO[bytes] + :return: An instance of AsyncLROPoller that returns either MasterSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSite] = 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, + site_name=site_name, + properties=properties, + 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("MasterSite", 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.MasterSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.MasterSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> None: + """Method to delete a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @overload + async def error_summary( + self, + resource_group_name: str, + site_name: str, + body: _models.ErrorSummaryRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from master site for an appliance. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def error_summary( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from master site for an appliance. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def error_summary( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ErrorSummaryRequest, IO[bytes]], + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from master site for an appliance. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ErrorSummaryRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest or IO[bytes] + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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.SiteErrorSummary] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ErrorSummaryRequest") + + _request = build_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_operations_status_controller_operations.py new file mode 100644 index 000000000000..2201ab19e0aa --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_master_sites_operations_status_controller_operations.py @@ -0,0 +1,117 @@ +# 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._master_sites_operations_status_controller_operations import ( + build_get_vmware_operation_status_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 MasterSitesOperationsStatusControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`master_sites_operations_status_controller` 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_vmware_operation_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """A operation status resource belonging to a master site resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_vmware_operation_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_operations.py new file mode 100644 index 000000000000..46dc080dc1e3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_operations.py @@ -0,0 +1,130 @@ +# 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._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 Operations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'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") + + @distributed_trace + def list(self, **kwargs: Any) -> AsyncIterable["_models.Operation"]: + """List the operations for the provider. + + :return: An iterator like instance of either Operation or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.Operation] + :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.OperationListResult] = 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( + 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("OperationListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_patch.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_endpoint_connection_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_endpoint_connection_controller_operations.py new file mode 100644 index 000000000000..ba6769f60a72 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_endpoint_connection_controller_operations.py @@ -0,0 +1,412 @@ +# 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._private_endpoint_connection_controller_operations import ( + build_create_request, + build_delete_request, + build_get_request, + build_list_by_master_site_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 PrivateEndpointConnectionControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`private_endpoint_connection_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.PrivateEndpointConnection"]: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either PrivateEndpointConnection or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.PrivateEndpointConnection] + :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.PrivateEndpointConnectionListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("PrivateEndpointConnectionListResult", 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, site_name: str, pe_connection_name: str, **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :return: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :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.PrivateEndpointConnection] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + pe_connection_name=pe_connection_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("PrivateEndpointConnection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + pe_connection_name: str, + body: _models.PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + pe_connection_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :param body: Resource create 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: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create( + self, + resource_group_name: str, + site_name: str, + pe_connection_name: str, + body: Union[_models.PrivateEndpointConnection, IO[bytes]], + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :param body: Resource create parameters. Is either a PrivateEndpointConnection type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.PrivateEndpointConnection or IO[bytes] + :return: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :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.PrivateEndpointConnection] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "PrivateEndpointConnection") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + pe_connection_name=pe_connection_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, pe_connection_name: str, **kwargs: Any + ) -> None: + """Deletes the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_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, + site_name=site_name, + pe_connection_name=pe_connection_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_link_resources_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_link_resources_controller_operations.py new file mode 100644 index 000000000000..58891ef0971f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_private_link_resources_controller_operations.py @@ -0,0 +1,205 @@ +# 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.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._private_link_resources_controller_operations import ( + build_get_request, + build_list_by_master_site_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 PrivateLinkResourcesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`private_link_resources_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.PrivateLinkResource"]: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either PrivateLinkResource or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.PrivateLinkResource] + :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.PrivateLinkResourceListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("PrivateLinkResourceListResult", 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, site_name: str, private_link_resource_name: str, **kwargs: Any + ) -> _models.PrivateLinkResource: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param private_link_resource_name: Private link resource name. Required. + :type private_link_resource_name: str + :return: PrivateLinkResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateLinkResource + :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.PrivateLinkResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + private_link_resource_name=private_link_resource_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("PrivateLinkResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..59179dd1b537 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_run_as_accounts_controller_operations.py @@ -0,0 +1,204 @@ +# 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.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._run_as_accounts_controller_operations import build_get_request, build_list_by_vmware_site_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 RunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`run_as_accounts_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.VmwareRunAsAccountResource"]: + """List VmwareRunAsAccountResource resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either VmwareRunAsAccountResource or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.VmwareRunAsAccountResource] + :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.VmwareRunAsAccountResourceListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareRunAsAccountResourceListResult", 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, site_name: str, account_name: str, **kwargs: Any + ) -> _models.VmwareRunAsAccountResource: + """Get a VmwareRunAsAccountResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param account_name: RunAsAccounts name. Required. + :type account_name: str + :return: VmwareRunAsAccountResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareRunAsAccountResource + :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.VmwareRunAsAccountResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + account_name=account_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("VmwareRunAsAccountResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_dependency_map_controller_operations.py new file mode 100644 index 000000000000..2d455ebc81e5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_dependency_map_controller_operations.py @@ -0,0 +1,1057 @@ +# 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._server_dependency_map_controller_operations import ( + build_client_group_members_request, + build_export_dependencies_request, + build_generate_coarse_map_request, + build_generate_detailed_map_request, + build_server_group_members_request, +) + +if sys.version_info >= (3, 9): + from collections.abc import MutableMapping +else: + from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class ServerDependencyMapControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`server_dependency_map_controller` 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 _client_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsClientGroupMembersRequest") + + _request = build_client_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsClientGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsClientGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._client_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_dependencies_initial( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **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(request_body, (IOBase, bytes)): + _content = request_body + else: + _json = self._serialize.body(request_body, "DependencyMapServiceMapextensionsExportDependenciesRequest") + + _request = build_export_dependencies_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: _models.DependencyMapServiceMapextensionsExportDependenciesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Is either a + DependencyMapServiceMapextensionsExportDependenciesRequest type or a IO[bytes] type. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_dependencies_initial( + resource_group_name=resource_group_name, + site_name=site_name, + request_body=request_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _generate_coarse_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsScopeMapRequest") + + _request = build_generate_coarse_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsScopeMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsScopeMapRequest type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._generate_coarse_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _generate_detailed_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body( + map_request, "DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest" + ) + + _request = build_generate_detailed_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest type or a IO[bytes] type. + Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._generate_detailed_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _server_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsServerGroupMembersRequest") + + _request = build_server_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsServerGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsServerGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._server_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_jobs_controller_operations.py new file mode 100644 index 000000000000..18595c67f499 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_jobs_controller_operations.py @@ -0,0 +1,203 @@ +# 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.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._server_jobs_controller_operations import ( + build_get_request, + build_list_by_server_site_resource_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 ServerJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`server_jobs_controller` 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_by_server_site_resource( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ServerJob"]: + """List ServerJob resources by ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ServerJob or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ServerJob] + :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.ServerJobListResult] = 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_server_site_resource_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.ServerJob: + """Get a ServerJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Jobs name. Required. + :type job_name: str + :return: ServerJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerJob + :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.ServerJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ServerJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_operations_status_controller_operations.py new file mode 100644 index 000000000000..70b2372133fb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_operations_status_controller_operations.py @@ -0,0 +1,117 @@ +# 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._server_operations_status_controller_operations import ( + build_get_server_site_operations_status_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 ServerOperationsStatusControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`server_operations_status_controller` 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_server_site_operations_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """A operation status resource belonging to a site resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_server_site_operations_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..073ef3f85863 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_run_as_accounts_controller_operations.py @@ -0,0 +1,205 @@ +# 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.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._server_run_as_accounts_controller_operations import ( + build_get_request, + build_list_by_server_site_resource_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 ServerRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`server_run_as_accounts_controller` 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_by_server_site_resource( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ServerRunAsAccount"]: + """List ServerRunAsAccount resources by ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ServerRunAsAccount or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ServerRunAsAccount] + :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.ServerRunAsAccountListResult] = 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_server_site_resource_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerRunAsAccountListResult", 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, site_name: str, account_name: str, **kwargs: Any + ) -> _models.ServerRunAsAccount: + """Get a ServerRunAsAccount. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param account_name: RunAsAccounts name. Required. + :type account_name: str + :return: ServerRunAsAccount or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerRunAsAccount + :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.ServerRunAsAccount] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + account_name=account_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("ServerRunAsAccount", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_sites_controller_operations.py new file mode 100644 index 000000000000..d7cde71c7313 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_sites_controller_operations.py @@ -0,0 +1,1749 @@ +# 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._server_sites_controller_operations import ( + build_compute_error_summary_request, + build_computeusage_request, + build_create_request, + build_delete_request, + build_export_applications_request, + build_export_machine_errors_request, + build_get_request, + build_list_by_resource_group_request, + build_list_by_subscription_request, + build_list_health_summary_request, + build_refresh_site_request, + build_summary_request, + build_update_dependency_map_status_request, + build_update_properties_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class ServerSitesControllerOperations: # pylint: disable=too-many-public-methods + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`server_sites_controller` 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_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ServerSiteResource"]: + """List ServerSiteResource resources by subscription ID. + + :return: An iterator like instance of either ServerSiteResource or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ServerSiteResource] + :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.ServerSiteResourceListResult] = 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_subscription_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("ServerSiteResourceListResult", 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 + def list_by_resource_group( + self, resource_group_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ServerSiteResource"]: + """List ServerSiteResource resources by 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 ServerSiteResource or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ServerSiteResource] + :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.ServerSiteResourceListResult] = 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("ServerSiteResourceListResult", 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, site_name: str, **kwargs: Any) -> _models.ServerSiteResource: + """Get a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :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.ServerSiteResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerSiteResource", 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, + site_name: str, + body: Union[_models.ServerSiteResource, IO[bytes]], + **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: + _json = self._serialize.body(body, "ServerSiteResource") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + body: _models.ServerSiteResource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ServerSiteResource]: + """Create a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResource + :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 ServerSiteResource or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.ServerSiteResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.ServerSiteResource]: + """Create a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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 either ServerSiteResource or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.ServerSiteResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ServerSiteResource, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.ServerSiteResource]: + """Create a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a ServerSiteResource type or a IO[bytes] + type. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResource or IO[bytes] + :return: An instance of AsyncLROPoller that returns either ServerSiteResource or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.ServerSiteResource] + :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.ServerSiteResource] = 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, + site_name=site_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("ServerSiteResource", 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.ServerSiteResource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.ServerSiteResource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: _models.ServerSiteResourceUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ServerSiteResource: + """Update a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResourceUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ServerSiteResource: + """Update a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ServerSiteResourceUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.ServerSiteResource: + """Update a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a ServerSiteResourceUpdate type + or a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResourceUpdate or IO[bytes] + :return: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :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.ServerSiteResource] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ServerSiteResourceUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("ServerSiteResource", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, 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 compute_error_summary( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SiteErrorSummary: + """Get the error summary for a server site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SiteErrorSummary] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_compute_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def computeusage( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.ServerSiteUsageResponse: + """Get a serve site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: ServerSiteUsageResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteUsageResponse + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.ServerSiteUsageResponse] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_computeusage_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("ServerSiteUsageResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _export_applications_initial( + self, resource_group_name: str, site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_applications_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_applications( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_applications_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_machine_errors_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachineErrorsRequest") + + _request = build_export_machine_errors_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachineErrorsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachineErrorsRequest type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_machine_errors_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def list_health_summary( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> _models.SiteHealthSummaryCollection: + """Method to get site health summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: SiteHealthSummaryCollection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteHealthSummaryCollection + :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.SiteHealthSummaryCollection] = kwargs.pop("cls", None) + + _request = build_list_health_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("SiteHealthSummaryCollection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _refresh_site_initial( + self, resource_group_name: str, site_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_refresh_site_request( + resource_group_name=resource_group_name, + site_name=site_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]: + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_refresh_site(self, resource_group_name: str, site_name: str, **kwargs: Any) -> AsyncLROPoller[JSON]: + """Operation to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._refresh_site_initial( + resource_group_name=resource_group_name, + site_name=site_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): + deserialized = self._deserialize("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def summary(self, resource_group_name: str, site_name: str, **kwargs: Any) -> _models.ServerSiteUsage: + """Method to get site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: ServerSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteUsage + :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.ServerSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _update_dependency_map_status_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "UpdateMachineDepMapStatus") + + _request = build_update_dependency_map_status_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: _models.UpdateMachineDepMapStatus, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a UpdateMachineDepMapStatus + type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_dependency_map_status_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _update_properties_initial( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **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(meta_data, (IOBase, bytes)): + _content = meta_data + else: + _json = self._serialize.body(meta_data, "MachineMetadataCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: _models.MachineMetadataCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Operation to update custom properties for servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Operation to update custom properties for servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Operation to update custom properties for servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Is either a MachineMetadataCollection type + or a IO[bytes] type. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + meta_data=meta_data, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_software_inventories_controller_operations.py new file mode 100644 index 000000000000..829316620481 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_server_software_inventories_controller_operations.py @@ -0,0 +1,218 @@ +# 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, Union +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._server_software_inventories_controller_operations import ( + build_get_machine_software_inventory_request, + build_list_by_server_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 ServerSoftwareInventoriesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`server_software_inventories_controller` 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_machine_software_inventory( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + **kwargs: Any + ) -> _models.ServerSoftwareInventory: + """Method to get a machines software inventory like applications and roles. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param default: Default value. "default" Required. + :type default: str or ~azure.mgmt.migrate.models.Default + :return: ServerSoftwareInventory or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSoftwareInventory + :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.ServerSoftwareInventory] = kwargs.pop("cls", None) + + _request = build_get_machine_software_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_name, + default=default, + 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("ServerSoftwareInventory", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_server( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> AsyncIterable["_models.ServerSoftwareInventory"]: + """List ServerSoftwareInventory resources by Server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :return: An iterator like instance of either ServerSoftwareInventory or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.ServerSoftwareInventory] + :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.ServerSoftwareInventoryListResult] = 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_server_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("ServerSoftwareInventoryListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_servers_controller_operations.py new file mode 100644 index 000000000000..06e502cee896 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_servers_controller_operations.py @@ -0,0 +1,426 @@ +# 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._servers_controller_operations import ( + build_delete_machine_request, + build_get_machine_request, + build_list_by_server_site_resource_request, + build_update_machine_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 ServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`servers_controller` 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_by_server_site_resource( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.Server"]: + """Get all machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either Server or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.Server] + :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.ServerListResult] = 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_server_site_resource_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("ServerListResult", 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_machine( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> _models.Server: + """Get a Server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :return: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :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.Server] = kwargs.pop("cls", None) + + _request = build_get_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("Server", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def update_machine( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: _models.ServerUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Server: + """Update a Server machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.ServerUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update_machine( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Server: + """Update a Server machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. 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: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update_machine( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: Union[_models.ServerUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.Server: + """Update a Server machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Is either a ServerUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ServerUpdate or IO[bytes] + :return: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :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.Server] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ServerUpdate") + + _request = build_update_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("Server", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete_machine( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> None: + """Delete a Server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_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_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sites_controller_operations.py new file mode 100644 index 000000000000..6319e83cfb29 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sites_controller_operations.py @@ -0,0 +1,1369 @@ +# 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._sites_controller_operations import ( + build_compute_error_summary_request, + build_computeusage_request, + build_create_request, + build_delete_request, + build_export_applications_request, + build_export_machine_errors_request, + build_export_machines_request, + build_get_request, + build_list_by_resource_group_request, + build_list_by_subscription_request, + build_list_health_summary_request, + build_summary_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class SitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sites_controller` 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_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.VmwareSite"]: + """Get all vmware sites. + + Get all the vmware sites in the subscription. + + :return: An iterator like instance of either VmwareSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.VmwareSite] + :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.VmwareSiteListResult] = 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_subscription_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("VmwareSiteListResult", 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 + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VmwareSite"]: + """Get all vmware sites. + + Get all the vmware sites in the 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 VmwareSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.VmwareSite] + :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.VmwareSiteListResult] = 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("VmwareSiteListResult", 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, site_name: str, **kwargs: Any) -> _models.VmwareSite: + """Get a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :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.VmwareSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + body: _models.VmwareSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Create a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.VmwareSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Create a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create( + self, resource_group_name: str, site_name: str, body: Union[_models.VmwareSite, IO[bytes]], **kwargs: Any + ) -> _models.VmwareSite: + """Create a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a VmwareSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.VmwareSite or IO[bytes] + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :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.VmwareSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "VmwareSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("VmwareSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: _models.VmwareSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Update a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.VmwareSiteUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Update a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, resource_group_name: str, site_name: str, body: Union[_models.VmwareSiteUpdate, IO[bytes]], **kwargs: Any + ) -> _models.VmwareSite: + """Update a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a VmwareSiteUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.VmwareSiteUpdate or IO[bytes] + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :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.VmwareSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "VmwareSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("VmwareSite", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, 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 compute_error_summary( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get site error summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SiteErrorSummary] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_compute_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def computeusage( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.VmwareSiteUsage: + """Method to get site error summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: VmwareSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSiteUsage + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.VmwareSiteUsage] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_computeusage_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("VmwareSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _export_applications_initial( + self, resource_group_name: str, site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_applications_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_applications( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_applications_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_machine_errors_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachineErrorsRequest") + + _request = build_export_machine_errors_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachineErrorsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachineErrorsRequest type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_machine_errors_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_machines_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachinesRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachinesRequest") + + _request = build_export_machines_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machines( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachinesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachinesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_machines( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_machines( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachinesRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachinesRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachinesRequest or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_machines_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def list_health_summary( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> _models.SiteHealthSummaryCollection: + """Method to get site health summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: SiteHealthSummaryCollection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteHealthSummaryCollection + :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.SiteHealthSummaryCollection] = kwargs.pop("cls", None) + + _request = build_list_health_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("SiteHealthSummaryCollection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def summary(self, resource_group_name: str, site_name: str, **kwargs: Any) -> _models.VmwareSiteUsage: + """Method to get site usage/summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: VmwareSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSiteUsage + :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.VmwareSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_availability_groups_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_availability_groups_controller_operations.py new file mode 100644 index 000000000000..f9dda5ed05fe --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_availability_groups_controller_operations.py @@ -0,0 +1,239 @@ +# 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.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._sql_availability_groups_controller_operations import ( + build_get_request, + build_list_by_sql_site_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 SqlAvailabilityGroupsControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_availability_groups_controller` 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_by_sql_site( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.SqlAvailabilityGroup"]: + """Gets the sql availability groups. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either SqlAvailabilityGroup or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.SqlAvailabilityGroup] + :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.SqlAvailabilityGroupListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("SqlAvailabilityGroupListResult", 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, + site_name: str, + sql_site_name: str, + sql_availability_group_name: str, + **kwargs: Any + ) -> _models.SqlAvailabilityGroup: + """Gets the sql availability group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_availability_group_name: SQL availability group name. Required. + :type sql_availability_group_name: str + :return: SqlAvailabilityGroup or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlAvailabilityGroup + :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.SqlAvailabilityGroup] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_availability_group_name=sql_availability_group_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("SqlAvailabilityGroup", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_databases_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_databases_controller_operations.py new file mode 100644 index 000000000000..20a030bd7e1e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_databases_controller_operations.py @@ -0,0 +1,229 @@ +# 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.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._sql_databases_controller_operations import build_get_request, build_list_by_sql_site_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 SqlDatabasesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_databases_controller` 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_by_sql_site( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.SqlDatabaseV2"]: + """Gets the sql Databases. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either SqlDatabaseV2 or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.SqlDatabaseV2] + :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.SqlDatabaseV2ListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("SqlDatabaseV2ListResult", 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, site_name: str, sql_site_name: str, sql_database_name: str, **kwargs: Any + ) -> _models.SqlDatabaseV2: + """Gets the sql Database. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_database_name: SQL Database name. Required. + :type sql_database_name: str + :return: SqlDatabaseV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlDatabaseV2 + :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.SqlDatabaseV2] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_database_name=sql_database_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("SqlDatabaseV2", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_discovery_site_data_source_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_discovery_site_data_source_controller_operations.py new file mode 100644 index 000000000000..e986c080b38a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_discovery_site_data_source_controller_operations.py @@ -0,0 +1,516 @@ +# 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._sql_discovery_site_data_source_controller_operations import ( + build_create_request, + build_delete_request, + build_get_request, + build_list_by_sql_site_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 SqlDiscoverySiteDataSourceControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_discovery_site_data_source_controller` 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_by_sql_site( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.SqlDiscoverySiteDataSource"]: + """List SqlDiscoverySiteDataSource resources by SqlSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: An iterator like instance of either SqlDiscoverySiteDataSource or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :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.SqlDiscoverySiteDataSourceListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlDiscoverySiteDataSourceListResult", 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, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> _models.SqlDiscoverySiteDataSource: + """Get a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :return: SqlDiscoverySiteDataSource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource + :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.SqlDiscoverySiteDataSource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("SqlDiscoverySiteDataSource", 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, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.SqlDiscoverySiteDataSource, IO[bytes]], + **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: + _json = self._serialize.body(body, "SqlDiscoverySiteDataSource") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: _models.SqlDiscoverySiteDataSource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.SqlDiscoverySiteDataSource]: + """Create a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource + :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 SqlDiscoverySiteDataSource or the + result of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.SqlDiscoverySiteDataSource]: + """Create a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create 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 either SqlDiscoverySiteDataSource or the + result of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.SqlDiscoverySiteDataSource, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.SqlDiscoverySiteDataSource]: + """Create a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Is either a SqlDiscoverySiteDataSource type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource or IO[bytes] + :return: An instance of AsyncLROPoller that returns either SqlDiscoverySiteDataSource or the + result of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :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.SqlDiscoverySiteDataSource] = 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, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("SqlDiscoverySiteDataSource", 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.SqlDiscoverySiteDataSource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.SqlDiscoverySiteDataSource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> None: + """Delete a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_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, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_jobs_controller_operations.py new file mode 100644 index 000000000000..2bb800d4baa0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_jobs_controller_operations.py @@ -0,0 +1,212 @@ +# 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.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._sql_jobs_controller_operations import build_get_request, build_list_by_sql_site_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 SqlJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_jobs_controller` 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_by_sql_site( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.SqlJob"]: + """Gets the sql Jobs. + + List SqlJob resources by SqlSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: An iterator like instance of either SqlJob or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.SqlJob] + :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.SqlJobListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlJobListResult", 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, site_name: str, sql_site_name: str, job_name: str, **kwargs: Any + ) -> _models.SqlJob: + """Gets the sql Job. + + Get a SqlJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param job_name: SQL Job name. Required. + :type job_name: str + :return: SqlJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlJob + :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.SqlJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + job_name=job_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("SqlJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_operations_status_controller_operations.py new file mode 100644 index 000000000000..b40b8525f379 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_operations_status_controller_operations.py @@ -0,0 +1,118 @@ +# 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._sql_operations_status_controller_operations import build_get_sql_operation_status_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 SqlOperationsStatusControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_operations_status_controller` 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_sql_operation_status( + self, resource_group_name: str, site_name: str, sql_site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """Method to get operation status. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_sql_operation_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..eedcd6781810 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_run_as_accounts_controller_operations.py @@ -0,0 +1,208 @@ +# 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.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._sql_run_as_accounts_controller_operations import build_get_request, build_list_by_sql_site_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 SqlRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_run_as_accounts_controller` 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_by_sql_site( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.SqlRunAsAccount"]: + """List SqlRunAsAccount resources by SqlSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: An iterator like instance of either SqlRunAsAccount or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.SqlRunAsAccount] + :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.SqlRunAsAccountListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlRunAsAccountListResult", 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, site_name: str, sql_site_name: str, account_name: str, **kwargs: Any + ) -> _models.SqlRunAsAccount: + """Get a SqlRunAsAccount. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param account_name: SQL RunAsAccounts name. Required. + :type account_name: str + :return: SqlRunAsAccount or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlRunAsAccount + :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.SqlRunAsAccount] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + account_name=account_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("SqlRunAsAccount", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_servers_controller_operations.py new file mode 100644 index 000000000000..56f97201c143 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_servers_controller_operations.py @@ -0,0 +1,385 @@ +# 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._sql_servers_controller_operations import ( + build_get_request, + build_list_by_sql_site_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 SqlServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_servers_controller` 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_by_sql_site( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.SqlServerV2"]: + """Gets the sql servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either SqlServerV2 or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.SqlServerV2] + :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.SqlServerV2ListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("SqlServerV2ListResult", 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, site_name: str, sql_site_name: str, sql_server_name: str, **kwargs: Any + ) -> _models.SqlServerV2: + """Gets the sql server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :return: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :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.SqlServerV2] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_server_name=sql_server_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("SqlServerV2", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + sql_server_name: str, + body: _models.SqlServerV2Update, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlServerV2: + """Updates the sql server tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.SqlServerV2Update + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + sql_server_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlServerV2: + """Updates the sql server tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :param body: The resource properties to be updated. 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: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + sql_server_name: str, + body: Union[_models.SqlServerV2Update, IO[bytes]], + **kwargs: Any + ) -> _models.SqlServerV2: + """Updates the sql server tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :param body: The resource properties to be updated. Is either a SqlServerV2Update type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.SqlServerV2Update or IO[bytes] + :return: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :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.SqlServerV2] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SqlServerV2Update") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_server_name=sql_server_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SqlServerV2", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_sites_controller_operations.py new file mode 100644 index 000000000000..9d81d8fa1bce --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_sql_sites_controller_operations.py @@ -0,0 +1,1376 @@ +# 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._sql_sites_controller_operations import ( + build_create_request, + build_delete_request, + build_error_summary_request, + build_export_sql_server_errors_request, + build_export_sql_servers_request, + build_get_request, + build_list_by_master_site_request, + build_refresh_request, + build_summary_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class SqlSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`sql_sites_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.SqlSite"]: + """Method to get all sites. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either SqlSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.SqlSite] + :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.SqlSiteListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("SqlSiteListResult", 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, site_name: str, sql_site_name: str, **kwargs: Any) -> _models.SqlSite: + """Method to get a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :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.SqlSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.SqlSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlSite: + """Method to create a SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.SqlSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlSite: + """Method to create a SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: Resource create 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: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.SqlSite, IO[bytes]], + **kwargs: Any + ) -> _models.SqlSite: + """Method to create a SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: Resource create parameters. Is either a SqlSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.SqlSite or IO[bytes] + :return: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :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.SqlSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SqlSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("SqlSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + async def _update_initial( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + properties: Union[_models.SqlSiteUpdate, IO[bytes]], + **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(properties, (IOBase, bytes)): + _content = properties + else: + _json = self._serialize.body(properties, "SqlSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + sql_site_name: str, + properties: _models.SqlSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.SqlSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: ~azure.mgmt.migrate.models.SqlSiteUpdate + :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 SqlSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.SqlSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + properties: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.SqlSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: 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 SqlSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.SqlSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + properties: Union[_models.SqlSiteUpdate, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.SqlSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param properties: The resource properties to be updated. Is either a SqlSiteUpdate type or a + IO[bytes] type. Required. + :type properties: ~azure.mgmt.migrate.models.SqlSiteUpdate or IO[bytes] + :return: An instance of AsyncLROPoller that returns either SqlSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.SqlSite] + :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.SqlSite] = 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, + site_name=site_name, + sql_site_name=sql_site_name, + properties=properties, + 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("SqlSite", 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.SqlSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.SqlSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> None: + """Deletes the SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_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, + site_name=site_name, + sql_site_name=sql_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @overload + async def error_summary( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.ErrorSummaryRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def error_summary( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. 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: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def error_summary( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.ErrorSummaryRequest, IO[bytes]], + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Is either a ErrorSummaryRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest or IO[bytes] + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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.SiteErrorSummary] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ErrorSummaryRequest") + + _request = build_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _export_sql_server_errors_initial( + self, resource_group_name: str, site_name: str, sql_site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_sql_server_errors_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_sql_server_errors( + self, resource_group_name: str, site_name: str, sql_site_name: str, body: JSON, **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_sql_server_errors_initial( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _export_sql_servers_initial( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.ExportSqlServersRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportSqlServersRequest") + + _request = build_export_sql_servers_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_sql_servers( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.ExportSqlServersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportSqlServersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_sql_servers( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_sql_servers( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.ExportSqlServersRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Is either a ExportSqlServersRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportSqlServersRequest or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_sql_servers_initial( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _refresh_initial( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.SqlSiteRefreshBody, IO[bytes]], + **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: + _json = self._serialize.body(body, "SqlSiteRefreshBody") + + _request = build_refresh_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_refresh( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.SqlSiteRefreshBody, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.SqlSiteRefreshBody + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_refresh( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_refresh( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.SqlSiteRefreshBody, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Is either a SqlSiteRefreshBody type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.SqlSiteRefreshBody or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._refresh_initial( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def summary( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> _models.SqlSiteUsage: + """Method to get site usage/summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: SqlSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSiteUsage + :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.SqlSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_applications_controller_operations.py new file mode 100644 index 000000000000..6680ecf654ad --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_applications_controller_operations.py @@ -0,0 +1,313 @@ +# 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.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._tomcat_web_applications_controller_operations import ( + build_get_request, + build_list_by_web_app_site_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class TomcatWebApplicationsControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`tomcat_web_applications_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.TomcatWebApplications"]: + """Method to get all Tomcat web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either TomcatWebApplications or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.TomcatWebApplications] + :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.TomcatWebApplicationsListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("TomcatWebApplicationsListResult", 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, site_name: str, web_app_site_name: str, web_application_name: str, **kwargs: Any + ) -> _models.TomcatWebApplications: + """Method to get an Tomcat web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :return: TomcatWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.TomcatWebApplications + :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.TomcatWebApplications] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_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("TomcatWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: JSON, + **kwargs: Any + ) -> _models.TomcatWebApplications: + """Updates the Tomcat web application tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. Required. + :type body: JSON + :return: TomcatWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.TomcatWebApplications + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.TomcatWebApplications] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("TomcatWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_servers_controller_operations.py new file mode 100644 index 000000000000..8bcc0cb9e1e2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_tomcat_web_servers_controller_operations.py @@ -0,0 +1,232 @@ +# 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.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._tomcat_web_servers_controller_operations import ( + build_get_request, + build_list_by_web_app_site_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 TomcatWebServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`tomcat_web_servers_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.TomcatWebServers"]: + """Method to get all Tomcat web servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either TomcatWebServers or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.TomcatWebServers] + :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.TomcatWebServersListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("TomcatWebServersListResult", 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, site_name: str, web_app_site_name: str, web_server_name: str, **kwargs: Any + ) -> _models.TomcatWebServers: + """Method to get an Tomcat web server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_server_name: Web server name. Required. + :type web_server_name: str + :return: TomcatWebServers or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.TomcatWebServers + :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.TomcatWebServers] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_server_name=web_server_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("TomcatWebServers", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vcenter_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vcenter_controller_operations.py new file mode 100644 index 000000000000..77e3c7fb04e4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vcenter_controller_operations.py @@ -0,0 +1,481 @@ +# 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._vcenter_controller_operations import ( + build_create_request, + build_delete_request, + build_get_request, + build_list_by_vmware_site_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 VcenterControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`vcenter_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, filter: Optional[str] = None, **kwargs: Any + ) -> AsyncIterable["_models.Vcenter"]: + """List Vcenter resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :return: An iterator like instance of either Vcenter or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.Vcenter] + :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.VcenterListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + 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("VcenterListResult", 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, site_name: str, vcenter_name: str, **kwargs: Any) -> _models.Vcenter: + """Get a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :return: Vcenter or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Vcenter + :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.Vcenter] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + vcenter_name=vcenter_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("Vcenter", 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, + site_name: str, + vcenter_name: str, + body: Union[_models.Vcenter, IO[bytes]], + **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: + _json = self._serialize.body(body, "Vcenter") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + vcenter_name=vcenter_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + vcenter_name: str, + body: _models.Vcenter, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.Vcenter]: + """Create a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.Vcenter + :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 Vcenter or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.Vcenter] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + site_name: str, + vcenter_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.Vcenter]: + """Create a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :param body: Resource create 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 either Vcenter or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.Vcenter] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + site_name: str, + vcenter_name: str, + body: Union[_models.Vcenter, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.Vcenter]: + """Create a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :param body: Resource create parameters. Is either a Vcenter type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.Vcenter or IO[bytes] + :return: An instance of AsyncLROPoller that returns either Vcenter or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.Vcenter] + :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.Vcenter] = 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, + site_name=site_name, + vcenter_name=vcenter_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("Vcenter", 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.Vcenter].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.Vcenter]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, vcenter_name: str, **kwargs: Any + ) -> None: + """Delete a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_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, + site_name=site_name, + vcenter_name=vcenter_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_host_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_host_controller_operations.py new file mode 100644 index 000000000000..627602ea6c8a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_host_controller_operations.py @@ -0,0 +1,200 @@ +# 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.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._vmware_host_controller_operations import build_get_request, build_list_by_vmware_site_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 VmwareHostControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`vmware_host_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.VmwareHost"]: + """List VmwareHost resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either VmwareHost or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.VmwareHost] + :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.VmwareHostListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareHostListResult", 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, site_name: str, host_name: str, **kwargs: Any) -> _models.VmwareHost: + """Get a VmwareHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Hosts name. Required. + :type host_name: str + :return: VmwareHost or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareHost + :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.VmwareHost] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + host_name=host_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("VmwareHost", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_operations_status_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_operations_status_operations.py new file mode 100644 index 000000000000..fa4da255097f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_operations_status_operations.py @@ -0,0 +1,115 @@ +# 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._vmware_operations_status_operations import build_get_vmware_operation_status_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 VmwareOperationsStatusOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`vmware_operations_status` 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_vmware_operation_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """A operation status resource belonging to a site resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_vmware_operation_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_properties_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_properties_controller_operations.py new file mode 100644 index 000000000000..ab9b691f3244 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_properties_controller_operations.py @@ -0,0 +1,859 @@ +# 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._vmware_properties_controller_operations import ( + build_update_dependency_map_status_request, + build_update_properties_request, + build_update_run_as_account_request, + build_update_tags_request, +) + +if sys.version_info >= (3, 9): + from collections.abc import MutableMapping +else: + from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class VmwarePropertiesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`vmware_properties_controller` 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_dependency_map_status_initial( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **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(update_machine_dep_map_status, (IOBase, bytes)): + _content = update_machine_dep_map_status + else: + _json = self._serialize.body(update_machine_dep_map_status, "UpdateMachineDepMapStatus") + + _request = build_update_dependency_map_status_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: _models.UpdateMachineDepMapStatus, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a + site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_dep_map_status: The content of the action request. Required. + :type update_machine_dep_map_status: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a + site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_dep_map_status: The content of the action request. Required. + :type update_machine_dep_map_status: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a + site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_dep_map_status: The content of the action request. Is either a + UpdateMachineDepMapStatus type or a IO[bytes] type. Required. + :type update_machine_dep_map_status: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_dependency_map_status_initial( + resource_group_name=resource_group_name, + site_name=site_name, + update_machine_dep_map_status=update_machine_dep_map_status, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _update_properties_initial( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **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(meta_data, (IOBase, bytes)): + _content = meta_data + else: + _json = self._serialize.body(meta_data, "MachineMetadataCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: _models.MachineMetadataCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update properties for machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update properties for machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update properties for machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Is either a MachineMetadataCollection type + or a IO[bytes] type. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + meta_data=meta_data, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _update_run_as_account_initial( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: Union[_models.UpdateMachineRunAsAccount, IO[bytes]], + **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(update_machine_run_as_account, (IOBase, bytes)): + _content = update_machine_run_as_account + else: + _json = self._serialize.body(update_machine_run_as_account, "UpdateMachineRunAsAccount") + + _request = build_update_run_as_account_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_run_as_account( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: _models.UpdateMachineRunAsAccount, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_run_as_account: The content of the action request. Required. + :type update_machine_run_as_account: ~azure.mgmt.migrate.models.UpdateMachineRunAsAccount + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_run_as_account( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_run_as_account: The content of the action request. Required. + :type update_machine_run_as_account: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_run_as_account( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: Union[_models.UpdateMachineRunAsAccount, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_run_as_account: The content of the action request. Is either a + UpdateMachineRunAsAccount type or a IO[bytes] type. Required. + :type update_machine_run_as_account: ~azure.mgmt.migrate.models.UpdateMachineRunAsAccount or + IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_run_as_account_initial( + resource_group_name=resource_group_name, + site_name=site_name, + update_machine_run_as_account=update_machine_run_as_account, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _update_tags_initial( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: Union[_models.UpdateMachineTags, IO[bytes]], + **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(update_machine_tags, (IOBase, bytes)): + _content = update_machine_tags + else: + _json = self._serialize.body(update_machine_tags, "UpdateMachineTags") + + _request = build_update_tags_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_tags( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: _models.UpdateMachineTags, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_tags: The content of the action request. Required. + :type update_machine_tags: ~azure.mgmt.migrate.models.UpdateMachineTags + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_tags( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_tags: The content of the action request. Required. + :type update_machine_tags: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_tags( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: Union[_models.UpdateMachineTags, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_tags: The content of the action request. Is either a UpdateMachineTags + type or a IO[bytes] type. Required. + :type update_machine_tags: ~azure.mgmt.migrate.models.UpdateMachineTags or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_tags_initial( + resource_group_name=resource_group_name, + site_name=site_name, + update_machine_tags=update_machine_tags, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_software_inventories_controller_operations.py new file mode 100644 index 000000000000..097e2030b869 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_vmware_software_inventories_controller_operations.py @@ -0,0 +1,218 @@ +# 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, Union +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._vmware_software_inventories_controller_operations import ( + build_get_machine_software_inventory_request, + build_list_by_machine_resource_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 VmwareSoftwareInventoriesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`vmware_software_inventories_controller` 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_machine_software_inventory( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + **kwargs: Any + ) -> _models.VmwareMachineSoftwareInventory: + """Method to get a machines software inventory like applications and roles. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param default: Default value. "default" Required. + :type default: str or ~azure.mgmt.migrate.models.Default + :return: VmwareMachineSoftwareInventory or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareMachineSoftwareInventory + :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.VmwareMachineSoftwareInventory] = kwargs.pop("cls", None) + + _request = build_get_machine_software_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_name, + default=default, + 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("VmwareMachineSoftwareInventory", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_machine_resource( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> AsyncIterable["_models.VmwareMachineSoftwareInventory"]: + """List VmwareMachineSoftwareInventory resources by MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: An iterator like instance of either VmwareMachineSoftwareInventory or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.VmwareMachineSoftwareInventory] + :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.VmwareMachineSoftwareInventoryListResult] = 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_machine_resource_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("VmwareMachineSoftwareInventoryListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_discovery_site_data_sources_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_discovery_site_data_sources_controller_operations.py new file mode 100644 index 000000000000..452aefbdb46c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_discovery_site_data_sources_controller_operations.py @@ -0,0 +1,579 @@ +# 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._web_app_discovery_site_data_sources_controller_operations import ( + build_create_request, + build_delete_request, + build_get_request, + build_list_by_web_app_site_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 WebAppDiscoverySiteDataSourcesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`web_app_discovery_site_data_sources_controller` 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_by_web_app_site( + self, resource_group_name: str, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.DiscoverySiteDataSource"]: + """Method to get all Web app data sources in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: An iterator like instance of either DiscoverySiteDataSource or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :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.DiscoverySiteDataSourceListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("DiscoverySiteDataSourceListResult", 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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> _models.DiscoverySiteDataSource: + """Method to get a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :return: DiscoverySiteDataSource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.DiscoverySiteDataSource + :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.DiscoverySiteDataSource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("DiscoverySiteDataSource", 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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.DiscoverySiteDataSource, IO[bytes]], + **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: + _json = self._serialize.body(body, "DiscoverySiteDataSource") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: _models.DiscoverySiteDataSource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.DiscoverySiteDataSource]: + """Method to create or update a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.DiscoverySiteDataSource + :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 DiscoverySiteDataSource or the + result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.DiscoverySiteDataSource]: + """Method to create or update a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create 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 either DiscoverySiteDataSource or the + result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.DiscoverySiteDataSource, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.DiscoverySiteDataSource]: + """Method to create or update a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Is either a DiscoverySiteDataSource type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.DiscoverySiteDataSource or IO[bytes] + :return: An instance of AsyncLROPoller that returns either DiscoverySiteDataSource or the + result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :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.DiscoverySiteDataSource] = 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, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("DiscoverySiteDataSource", 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.DiscoverySiteDataSource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.DiscoverySiteDataSource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _delete_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_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, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Method to delete a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_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, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_extended_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_extended_machines_controller_operations.py new file mode 100644 index 000000000000..76bd79c51f22 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_extended_machines_controller_operations.py @@ -0,0 +1,239 @@ +# 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.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._web_app_extended_machines_controller_operations import ( + build_get_request, + build_list_by_web_app_site_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 WebAppExtendedMachinesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`web_app_extended_machines_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.WebAppExtendedMachine"]: + """Method to get all extended machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either WebAppExtendedMachine or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.WebAppExtendedMachine] + :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.WebAppExtendedMachineListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("WebAppExtendedMachineListResult", 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, + site_name: str, + web_app_site_name: str, + extended_machine_name: str, + **kwargs: Any + ) -> _models.WebAppExtendedMachine: + """Method to get a extended machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param extended_machine_name: Extended machine name. Required. + :type extended_machine_name: str + :return: WebAppExtendedMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppExtendedMachine + :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.WebAppExtendedMachine] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + extended_machine_name=extended_machine_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("WebAppExtendedMachine", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_properties_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_properties_controller_operations.py new file mode 100644 index 000000000000..59015bd08ae0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_properties_controller_operations.py @@ -0,0 +1,270 @@ +# 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._web_app_properties_controller_operations import build_update_properties_request + +if sys.version_info >= (3, 9): + from collections.abc import MutableMapping +else: + from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class WebAppPropertiesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`web_app_properties_controller` 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_properties_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.WebAppPropertiesCollection, IO[bytes]], + **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: + _json = self._serialize.body(body, "WebAppPropertiesCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.WebAppPropertiesCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update properties for web applications. + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.WebAppPropertiesCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update properties for web applications. + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.WebAppPropertiesCollection, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to update properties for web applications. + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a WebAppPropertiesCollection type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.WebAppPropertiesCollection or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..468a0542a378 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_run_as_accounts_controller_operations.py @@ -0,0 +1,211 @@ +# 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.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._web_app_run_as_accounts_controller_operations import ( + build_get_request, + build_list_by_web_app_site_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 WebAppRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`web_app_run_as_accounts_controller` 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_by_web_app_site( + self, resource_group_name: str, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.WebAppRunAsAccount"]: + """Method to get all run as accounts. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: An iterator like instance of either WebAppRunAsAccount or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.WebAppRunAsAccount] + :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.WebAppRunAsAccountListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("WebAppRunAsAccountListResult", 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, site_name: str, web_app_site_name: str, account_name: str, **kwargs: Any + ) -> _models.WebAppRunAsAccount: + """Method to get run as account. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param account_name: Run as account ARM name. Required. + :type account_name: str + :return: WebAppRunAsAccount or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppRunAsAccount + :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.WebAppRunAsAccount] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + account_name=account_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("WebAppRunAsAccount", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_sites_controller_operations.py new file mode 100644 index 000000000000..b03e60b371e1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_app_sites_controller_operations.py @@ -0,0 +1,1308 @@ +# 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._web_app_sites_controller_operations import ( + build_create_request, + build_delete_request, + build_error_summary_request, + build_export_inventory_request, + build_get_request, + build_list_by_master_site_request, + build_refresh_request, + build_summary_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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + + +class WebAppSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`web_app_sites_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> AsyncIterable["_models.WebAppSite"]: + """Method to get all sites. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either WebAppSite or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.WebAppSite] + :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.WebAppSiteListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("WebAppSiteListResult", 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, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> _models.WebAppSite: + """Method to get a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :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.WebAppSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("WebAppSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.WebAppSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WebAppSite: + """Method to create a WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.WebAppSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WebAppSite: + """Method to create a WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: Resource create 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: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.WebAppSite, IO[bytes]], + **kwargs: Any + ) -> _models.WebAppSite: + """Method to create a WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: Resource create parameters. Is either a WebAppSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.WebAppSite or IO[bytes] + :return: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :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.WebAppSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "WebAppSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("WebAppSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + async def _update_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + properties: Union[_models.WebAppSiteUpdate, IO[bytes]], + **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(properties, (IOBase, bytes)): + _content = properties + else: + _json = self._serialize.body(properties, "WebAppSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + web_app_site_name: str, + properties: _models.WebAppSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.WebAppSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: ~azure.mgmt.migrate.models.WebAppSiteUpdate + :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 WebAppSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.WebAppSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + properties: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.WebAppSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: 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 WebAppSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.WebAppSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + properties: Union[_models.WebAppSiteUpdate, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.WebAppSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param properties: The resource properties to be updated. Is either a WebAppSiteUpdate type or + a IO[bytes] type. Required. + :type properties: ~azure.mgmt.migrate.models.WebAppSiteUpdate or IO[bytes] + :return: An instance of AsyncLROPoller that returns either WebAppSite or the result of + cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.migrate.models.WebAppSite] + :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.WebAppSite] = 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, + site_name=site_name, + web_app_site_name=web_app_site_name, + properties=properties, + 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("WebAppSite", 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.WebAppSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.WebAppSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _delete_initial( + self, resource_group_name: str, site_name: str, web_app_site_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, + site_name=site_name, + web_app_site_name=web_app_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Deletes the WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_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, + site_name=site_name, + web_app_site_name=web_app_site_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 + + @overload + async def error_summary( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.ErrorSummaryRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """MMethod to get error summary from web app site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def error_summary( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """MMethod to get error summary from web app site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def error_summary( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ErrorSummaryRequest, IO[bytes]], + **kwargs: Any + ) -> _models.SiteErrorSummary: + """MMethod to get error summary from web app site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a ErrorSummaryRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest or IO[bytes] + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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.SiteErrorSummary] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ErrorSummaryRequest") + + _request = build_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _export_inventory_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ExportWebAppsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportWebAppsRequest") + + _request = build_export_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_inventory( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.ExportWebAppsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing web app inventory. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportWebAppsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_export_inventory( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing web app inventory. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_export_inventory( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ExportWebAppsRequest, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to generate report containing web app inventory. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a ExportWebAppsRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportWebAppsRequest or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._export_inventory_initial( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _refresh_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ProxySiteRefreshBody, IO[bytes]], + **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: + _json = self._serialize.body(body, "ProxySiteRefreshBody") + + _request = build_refresh_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_refresh( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.ProxySiteRefreshBody, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ProxySiteRefreshBody + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_refresh( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_refresh( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ProxySiteRefreshBody, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a ProxySiteRefreshBody type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ProxySiteRefreshBody or IO[bytes] + :return: An instance of AsyncLROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.AsyncLROPoller[JSON] + :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[JSON] = 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._refresh_initial( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace_async + async def summary( + self, resource_group_name: str, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> _models.WebAppSiteUsage: + """Method to get site usage/summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: WebAppSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSiteUsage + :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.WebAppSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("WebAppSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_applications_controller_operations.py new file mode 100644 index 000000000000..e1b114f5595c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_applications_controller_operations.py @@ -0,0 +1,164 @@ +# 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._web_applications_controller_operations import build_list_by_web_app_site_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 WebApplicationsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`web_applications_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.WebApplication"]: + """Method to get all IIS web applications. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either WebApplication or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.WebApplication] + :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.WebApplicationListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("WebApplicationListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_servers_controller_operations.py new file mode 100644 index 000000000000..151d4c553ad2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/aio/operations/_web_servers_controller_operations.py @@ -0,0 +1,164 @@ +# 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._web_servers_controller_operations import build_list_by_web_app_site_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 WebServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.aio.AzureMigrateDiscoveryService`'s + :attr:`web_servers_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> AsyncIterable["_models.WebServer"]: + """Method to get all web servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either WebServer or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.migrate.models.WebServer] + :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.WebServerListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("WebServerListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/__init__.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/__init__.py new file mode 100644 index 000000000000..a7c2eabffb61 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/__init__.py @@ -0,0 +1,627 @@ +# 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 ._models_py3 import Application +from ._models_py3 import ApplicationDiscovery +from ._models_py3 import AppsAndRoles +from ._models_py3 import BizTalkServer +from ._models_py3 import ConnectorUnit +from ._models_py3 import DeleteImportMachinesJob +from ._models_py3 import DeleteImportMachinesJobCollection +from ._models_py3 import DeleteImportedMachinesJobProperties +from ._models_py3 import DependencyMapDiscovery +from ._models_py3 import DependencyMapMachineInput +from ._models_py3 import DependencyMapServiceMapextensionsClientGroupMembersRequest +from ._models_py3 import DependencyMapServiceMapextensionsDependencyMapRequestFilters +from ._models_py3 import DependencyMapServiceMapextensionsExportDependenciesRequest +from ._models_py3 import DependencyMapServiceMapextensionsScopeMapRequest +from ._models_py3 import DependencyMapServiceMapextensionsServerGroupMembersRequest +from ._models_py3 import DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest +from ._models_py3 import DirectoryPath +from ._models_py3 import DiscoveryScopeErrorSummary +from ._models_py3 import DiscoverySiteDataSource +from ._models_py3 import DiscoverySiteDataSourceListResult +from ._models_py3 import DiscoverySiteDataSourceProperties +from ._models_py3 import ErrorAdditionalInfo +from ._models_py3 import ErrorDetail +from ._models_py3 import ErrorDetails +from ._models_py3 import ErrorResponse +from ._models_py3 import ErrorSummaryRequest +from ._models_py3 import Errors +from ._models_py3 import ExchangeServer +from ._models_py3 import ExportImportedMachinesJob +from ._models_py3 import ExportImportedMachinesJobEntityProperties +from ._models_py3 import ExportMachineErrorsRequest +from ._models_py3 import ExportMachinesRequest +from ._models_py3 import ExportSqlServersRequest +from ._models_py3 import ExportWebAppsRequest +from ._models_py3 import Feature +from ._models_py3 import FileMetaData +from ._models_py3 import FrontEndBinding +from ._models_py3 import GuestOsDetails +from ._models_py3 import HealthErrorDetails +from ._models_py3 import HypervCluster +from ._models_py3 import HypervClusterListResult +from ._models_py3 import HypervClusterProperties +from ._models_py3 import HypervDisk +from ._models_py3 import HypervHost +from ._models_py3 import HypervHostListResult +from ._models_py3 import HypervHostProperties +from ._models_py3 import HypervJob +from ._models_py3 import HypervJobListResult +from ._models_py3 import HypervMachine +from ._models_py3 import HypervMachineListResult +from ._models_py3 import HypervMachineProperties +from ._models_py3 import HypervMachineUpdate +from ._models_py3 import HypervMachineUpdateProperties +from ._models_py3 import HypervNetworkAdapter +from ._models_py3 import HypervOperationsStatusResource +from ._models_py3 import HypervRunAsAccountResource +from ._models_py3 import HypervRunAsAccountResourceListResult +from ._models_py3 import HypervSite +from ._models_py3 import HypervSiteListResult +from ._models_py3 import HypervSiteUpdate +from ._models_py3 import HypervSiteUpdateProperties +from ._models_py3 import HypervSiteUsage +from ._models_py3 import HypervVmSoftwareInventory +from ._models_py3 import HypervVmSoftwareInventoryListResult +from ._models_py3 import IisApplicationUnit +from ._models_py3 import IisVirtualApplicationUnit +from ._models_py3 import IisWebApplicationProperties +from ._models_py3 import IisWebApplications +from ._models_py3 import IisWebApplicationsListResult +from ._models_py3 import IisWebApplicationsUpdate +from ._models_py3 import IisWebApplicationsUpdateProperties +from ._models_py3 import IisWebServerProperties +from ._models_py3 import IisWebServers +from ._models_py3 import IisWebServersListResult +from ._models_py3 import ImportJob +from ._models_py3 import ImportJobListResult +from ._models_py3 import ImportMachine +from ._models_py3 import ImportMachineListResult +from ._models_py3 import ImportMachineProperties +from ._models_py3 import ImportMachinesJob +from ._models_py3 import ImportMachinesJobProperties +from ._models_py3 import ImportSite +from ._models_py3 import ImportSiteListResult +from ._models_py3 import ImportSiteProperties +from ._models_py3 import ImportSiteUpdate +from ._models_py3 import ImportSiteUpdateProperties +from ._models_py3 import JobErrorSummary +from ._models_py3 import JobProperties +from ._models_py3 import MachineMetadata +from ._models_py3 import MachineMetadataCollection +from ._models_py3 import MachineResource +from ._models_py3 import MachineResourceListResult +from ._models_py3 import MachineResourceUpdate +from ._models_py3 import MachineResourceUpdateProperties +from ._models_py3 import MachineSoftwareInventoryProperties +from ._models_py3 import MasterSite +from ._models_py3 import MasterSiteListResult +from ._models_py3 import MasterSiteProperties +from ._models_py3 import MasterSiteUpdate +from ._models_py3 import MasterSiteUpdateProperties +from ._models_py3 import OperatingSystem +from ._models_py3 import Operation +from ._models_py3 import OperationDisplay +from ._models_py3 import OperationListResult +from ._models_py3 import OperationStatus +from ._models_py3 import OperationStatusError +from ._models_py3 import OperationStatusProperties +from ._models_py3 import OracleDiscovery +from ._models_py3 import OtherDatabase +from ._models_py3 import PagedDeleteImportMachinesJob +from ._models_py3 import PagedExportImportedMachinesJob +from ._models_py3 import PagedImportMachinesJob +from ._models_py3 import PrivateEndpointConnection +from ._models_py3 import PrivateEndpointConnectionListResult +from ._models_py3 import PrivateEndpointConnectionPropertiesV2 +from ._models_py3 import PrivateLinkResource +from ._models_py3 import PrivateLinkResourceListResult +from ._models_py3 import PrivateLinkResourceProperties +from ._models_py3 import PrivateLinkServiceConnectionState +from ._models_py3 import ProcessorInfo +from ._models_py3 import ProductSupportStatus +from ._models_py3 import ProxyResource +from ._models_py3 import ProxySiteRefreshBody +from ._models_py3 import RequestExportMachineErrorsProperties +from ._models_py3 import Resource +from ._models_py3 import ResourceId +from ._models_py3 import RunAsAccountMachineInput +from ._models_py3 import RunAsAccountProperties +from ._models_py3 import SasUriResponse +from ._models_py3 import Server +from ._models_py3 import ServerDisk +from ._models_py3 import ServerJob +from ._models_py3 import ServerJobListResult +from ._models_py3 import ServerListResult +from ._models_py3 import ServerNetworkAdapter +from ._models_py3 import ServerProperties +from ._models_py3 import ServerRunAsAccount +from ._models_py3 import ServerRunAsAccountListResult +from ._models_py3 import ServerSiteResource +from ._models_py3 import ServerSiteResourceListResult +from ._models_py3 import ServerSiteResourceUpdate +from ._models_py3 import ServerSiteResourceUpdateProperties +from ._models_py3 import ServerSiteUsage +from ._models_py3 import ServerSiteUsageResponse +from ._models_py3 import ServerSoftwareInventory +from ._models_py3 import ServerSoftwareInventoryListResult +from ._models_py3 import ServerUpdate +from ._models_py3 import ServerUpdateProperties +from ._models_py3 import SharePointServer +from ._models_py3 import SiteAgentProperties +from ._models_py3 import SiteApplianceProperties +from ._models_py3 import SiteErrorSummary +from ._models_py3 import SiteHealthSummary +from ._models_py3 import SiteHealthSummaryCollection +from ._models_py3 import SiteProperties +from ._models_py3 import SiteSpnProperties +from ._models_py3 import SitesProperties +from ._models_py3 import SpringBootDiscovery +from ._models_py3 import SqlAvailabilityGroup +from ._models_py3 import SqlAvailabilityGroupListResult +from ._models_py3 import SqlAvailabilityGroupProperties +from ._models_py3 import SqlAvailabilityGroupReplicaInfo +from ._models_py3 import SqlAvailabilityReplicaOverview +from ._models_py3 import SqlAvailabilityReplicaProperties +from ._models_py3 import SqlDatabasePropertiesV2 +from ._models_py3 import SqlDatabaseReplicaInfo +from ._models_py3 import SqlDatabaseV2 +from ._models_py3 import SqlDatabaseV2ListResult +from ._models_py3 import SqlDiscovery +from ._models_py3 import SqlDiscoverySiteDataSource +from ._models_py3 import SqlDiscoverySiteDataSourceListResult +from ._models_py3 import SqlDiscoverySiteDataSourceProperties +from ._models_py3 import SqlFciProperties +from ._models_py3 import SqlJob +from ._models_py3 import SqlJobListResult +from ._models_py3 import SqlMachineOverview +from ._models_py3 import SqlRunAsAccount +from ._models_py3 import SqlRunAsAccountListResult +from ._models_py3 import SqlServerApplication +from ._models_py3 import SqlServerProperties +from ._models_py3 import SqlServerV2 +from ._models_py3 import SqlServerV2ListResult +from ._models_py3 import SqlServerV2Update +from ._models_py3 import SqlServerV2UpdateProperties +from ._models_py3 import SqlSite +from ._models_py3 import SqlSiteListResult +from ._models_py3 import SqlSiteProperties +from ._models_py3 import SqlSiteRefreshBody +from ._models_py3 import SqlSiteUpdate +from ._models_py3 import SqlSiteUpdateProperties +from ._models_py3 import SqlSiteUsage +from ._models_py3 import StaticDiscovery +from ._models_py3 import SystemCenter +from ._models_py3 import SystemData +from ._models_py3 import TagsMachineInput +from ._models_py3 import TomcatEngineUnit +from ._models_py3 import TomcatHostUnit +from ._models_py3 import TomcatServiceUnit +from ._models_py3 import TomcatWebApplicationProperties +from ._models_py3 import TomcatWebApplications +from ._models_py3 import TomcatWebApplicationsListResult +from ._models_py3 import TomcatWebServerProperties +from ._models_py3 import TomcatWebServers +from ._models_py3 import TomcatWebServersListResult +from ._models_py3 import TrackedResource +from ._models_py3 import UpdateMachineDepMapStatus +from ._models_py3 import UpdateMachineRunAsAccount +from ._models_py3 import UpdateMachineTags +from ._models_py3 import V20180501PreviewVmwareRunAsAccount +from ._models_py3 import V20180501PreviewVmwareRunAsAccountVmwareRunAsAccountCollection +from ._models_py3 import Vcenter +from ._models_py3 import VcenterListResult +from ._models_py3 import VcenterProperties +from ._models_py3 import VmwareDatastore +from ._models_py3 import VmwareDisk +from ._models_py3 import VmwareHost +from ._models_py3 import VmwareHostListResult +from ._models_py3 import VmwareHostProperties +from ._models_py3 import VmwareJob +from ._models_py3 import VmwareJobListResult +from ._models_py3 import VmwareMachineProperties +from ._models_py3 import VmwareMachineSoftwareInventory +from ._models_py3 import VmwareMachineSoftwareInventoryListResult +from ._models_py3 import VmwareNetworkAdapter +from ._models_py3 import VmwareRunAsAccountResource +from ._models_py3 import VmwareRunAsAccountResourceListResult +from ._models_py3 import VmwareSite +from ._models_py3 import VmwareSiteListResult +from ._models_py3 import VmwareSiteUpdate +from ._models_py3 import VmwareSiteUpdateProperties +from ._models_py3 import VmwareSiteUsage +from ._models_py3 import WebAppDiscovery +from ._models_py3 import WebAppExtendedMachine +from ._models_py3 import WebAppExtendedMachineListResult +from ._models_py3 import WebAppExtendedMachineProperties +from ._models_py3 import WebAppProperties +from ._models_py3 import WebAppPropertiesCollection +from ._models_py3 import WebAppRunAsAccount +from ._models_py3 import WebAppRunAsAccountListResult +from ._models_py3 import WebAppSite +from ._models_py3 import WebAppSiteListResult +from ._models_py3 import WebAppSiteProperties +from ._models_py3 import WebAppSiteUpdate +from ._models_py3 import WebAppSiteUpdateProperties +from ._models_py3 import WebAppSiteUsage +from ._models_py3 import WebApplication +from ._models_py3 import WebApplicationAppsAndRolesModel +from ._models_py3 import WebApplicationConfigurationUnit +from ._models_py3 import WebApplicationDirectoryUnit +from ._models_py3 import WebApplicationFramework +from ._models_py3 import WebApplicationListResult +from ._models_py3 import WebApplicationProperties +from ._models_py3 import WebRoleImportDisk +from ._models_py3 import WebRoleOperatingSystem +from ._models_py3 import WebServer +from ._models_py3 import WebServerListResult +from ._models_py3 import WebServerProperties + +from ._azure_migrate_discovery_service_enums import ActionType +from ._azure_migrate_discovery_service_enums import ApplicationDiscoveryScopeStatus +from ._azure_migrate_discovery_service_enums import CreatedByType +from ._azure_migrate_discovery_service_enums import Default +from ._azure_migrate_discovery_service_enums import DefaultAutoGenerated +from ._azure_migrate_discovery_service_enums import DefaultValues +from ._azure_migrate_discovery_service_enums import DeleteImportedMachinesJobPropertiesJobState +from ._azure_migrate_discovery_service_enums import DependencyMapDiscoveryScopeStatus +from ._azure_migrate_discovery_service_enums import DiscoveryScopeStatus +from ._azure_migrate_discovery_service_enums import DiscoveryScopes +from ._azure_migrate_discovery_service_enums import EsuStatus +from ._azure_migrate_discovery_service_enums import EsuYear +from ._azure_migrate_discovery_service_enums import ExportMachineErrorsProperties +from ._azure_migrate_discovery_service_enums import FCIInstanceState +from ._azure_migrate_discovery_service_enums import FileType +from ._azure_migrate_discovery_service_enums import HealthErrorDetailsDiscoveryScope +from ._azure_migrate_discovery_service_enums import HealthErrorDetailsSource +from ._azure_migrate_discovery_service_enums import HighAvailability +from ._azure_migrate_discovery_service_enums import HighAvailabilityValues +from ._azure_migrate_discovery_service_enums import ImportTypeValues +from ._azure_migrate_discovery_service_enums import JobResult +from ._azure_migrate_discovery_service_enums import MasterSitePropertiesPublicNetworkAccess +from ._azure_migrate_discovery_service_enums import MicrosoftAzureFDSWebRoleHealthErrorDetailsSource +from ._azure_migrate_discovery_service_enums import Origin +from ._azure_migrate_discovery_service_enums import PrivateLinkServiceConnectionStateStatus +from ._azure_migrate_discovery_service_enums import ProvisioningState +from ._azure_migrate_discovery_service_enums import SQLDiscoveryScopeStatus +from ._azure_migrate_discovery_service_enums import ShallowDiscoveryStatus +from ._azure_migrate_discovery_service_enums import SiteHealthSummaryFabricLayoutUpdateSourcesItem +from ._azure_migrate_discovery_service_enums import SqlAvailabilityGroupPropertiesAvailabilityGroupType +from ._azure_migrate_discovery_service_enums import ( + SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaCommitMode, +) +from ._azure_migrate_discovery_service_enums import SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaReadMode +from ._azure_migrate_discovery_service_enums import SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSeedMode +from ._azure_migrate_discovery_service_enums import SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaState +from ._azure_migrate_discovery_service_enums import ( + SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSyncStatus, +) +from ._azure_migrate_discovery_service_enums import SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaType +from ._azure_migrate_discovery_service_enums import SqlAvailabilityReplicaOverviewReplicaState +from ._azure_migrate_discovery_service_enums import SqlMachineOverviewFciRole +from ._azure_migrate_discovery_service_enums import SqlMetadataDiscoveryPipe +from ._azure_migrate_discovery_service_enums import SqlServerStatus +from ._azure_migrate_discovery_service_enums import SqlSitePropertiesDiscoveryScenario +from ._azure_migrate_discovery_service_enums import StaticDiscoveryScopeStatus +from ._azure_migrate_discovery_service_enums import SupportStatus +from ._azure_migrate_discovery_service_enums import VMwareDatastoreType +from ._azure_migrate_discovery_service_enums import Versions +from ._azure_migrate_discovery_service_enums import WebAppSitePropertiesDiscoveryScenario +from ._patch import __all__ as _patch_all +from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import patch_sdk as _patch_sdk + +__all__ = [ + "Application", + "ApplicationDiscovery", + "AppsAndRoles", + "BizTalkServer", + "ConnectorUnit", + "DeleteImportMachinesJob", + "DeleteImportMachinesJobCollection", + "DeleteImportedMachinesJobProperties", + "DependencyMapDiscovery", + "DependencyMapMachineInput", + "DependencyMapServiceMapextensionsClientGroupMembersRequest", + "DependencyMapServiceMapextensionsDependencyMapRequestFilters", + "DependencyMapServiceMapextensionsExportDependenciesRequest", + "DependencyMapServiceMapextensionsScopeMapRequest", + "DependencyMapServiceMapextensionsServerGroupMembersRequest", + "DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest", + "DirectoryPath", + "DiscoveryScopeErrorSummary", + "DiscoverySiteDataSource", + "DiscoverySiteDataSourceListResult", + "DiscoverySiteDataSourceProperties", + "ErrorAdditionalInfo", + "ErrorDetail", + "ErrorDetails", + "ErrorResponse", + "ErrorSummaryRequest", + "Errors", + "ExchangeServer", + "ExportImportedMachinesJob", + "ExportImportedMachinesJobEntityProperties", + "ExportMachineErrorsRequest", + "ExportMachinesRequest", + "ExportSqlServersRequest", + "ExportWebAppsRequest", + "Feature", + "FileMetaData", + "FrontEndBinding", + "GuestOsDetails", + "HealthErrorDetails", + "HypervCluster", + "HypervClusterListResult", + "HypervClusterProperties", + "HypervDisk", + "HypervHost", + "HypervHostListResult", + "HypervHostProperties", + "HypervJob", + "HypervJobListResult", + "HypervMachine", + "HypervMachineListResult", + "HypervMachineProperties", + "HypervMachineUpdate", + "HypervMachineUpdateProperties", + "HypervNetworkAdapter", + "HypervOperationsStatusResource", + "HypervRunAsAccountResource", + "HypervRunAsAccountResourceListResult", + "HypervSite", + "HypervSiteListResult", + "HypervSiteUpdate", + "HypervSiteUpdateProperties", + "HypervSiteUsage", + "HypervVmSoftwareInventory", + "HypervVmSoftwareInventoryListResult", + "IisApplicationUnit", + "IisVirtualApplicationUnit", + "IisWebApplicationProperties", + "IisWebApplications", + "IisWebApplicationsListResult", + "IisWebApplicationsUpdate", + "IisWebApplicationsUpdateProperties", + "IisWebServerProperties", + "IisWebServers", + "IisWebServersListResult", + "ImportJob", + "ImportJobListResult", + "ImportMachine", + "ImportMachineListResult", + "ImportMachineProperties", + "ImportMachinesJob", + "ImportMachinesJobProperties", + "ImportSite", + "ImportSiteListResult", + "ImportSiteProperties", + "ImportSiteUpdate", + "ImportSiteUpdateProperties", + "JobErrorSummary", + "JobProperties", + "MachineMetadata", + "MachineMetadataCollection", + "MachineResource", + "MachineResourceListResult", + "MachineResourceUpdate", + "MachineResourceUpdateProperties", + "MachineSoftwareInventoryProperties", + "MasterSite", + "MasterSiteListResult", + "MasterSiteProperties", + "MasterSiteUpdate", + "MasterSiteUpdateProperties", + "OperatingSystem", + "Operation", + "OperationDisplay", + "OperationListResult", + "OperationStatus", + "OperationStatusError", + "OperationStatusProperties", + "OracleDiscovery", + "OtherDatabase", + "PagedDeleteImportMachinesJob", + "PagedExportImportedMachinesJob", + "PagedImportMachinesJob", + "PrivateEndpointConnection", + "PrivateEndpointConnectionListResult", + "PrivateEndpointConnectionPropertiesV2", + "PrivateLinkResource", + "PrivateLinkResourceListResult", + "PrivateLinkResourceProperties", + "PrivateLinkServiceConnectionState", + "ProcessorInfo", + "ProductSupportStatus", + "ProxyResource", + "ProxySiteRefreshBody", + "RequestExportMachineErrorsProperties", + "Resource", + "ResourceId", + "RunAsAccountMachineInput", + "RunAsAccountProperties", + "SasUriResponse", + "Server", + "ServerDisk", + "ServerJob", + "ServerJobListResult", + "ServerListResult", + "ServerNetworkAdapter", + "ServerProperties", + "ServerRunAsAccount", + "ServerRunAsAccountListResult", + "ServerSiteResource", + "ServerSiteResourceListResult", + "ServerSiteResourceUpdate", + "ServerSiteResourceUpdateProperties", + "ServerSiteUsage", + "ServerSiteUsageResponse", + "ServerSoftwareInventory", + "ServerSoftwareInventoryListResult", + "ServerUpdate", + "ServerUpdateProperties", + "SharePointServer", + "SiteAgentProperties", + "SiteApplianceProperties", + "SiteErrorSummary", + "SiteHealthSummary", + "SiteHealthSummaryCollection", + "SiteProperties", + "SiteSpnProperties", + "SitesProperties", + "SpringBootDiscovery", + "SqlAvailabilityGroup", + "SqlAvailabilityGroupListResult", + "SqlAvailabilityGroupProperties", + "SqlAvailabilityGroupReplicaInfo", + "SqlAvailabilityReplicaOverview", + "SqlAvailabilityReplicaProperties", + "SqlDatabasePropertiesV2", + "SqlDatabaseReplicaInfo", + "SqlDatabaseV2", + "SqlDatabaseV2ListResult", + "SqlDiscovery", + "SqlDiscoverySiteDataSource", + "SqlDiscoverySiteDataSourceListResult", + "SqlDiscoverySiteDataSourceProperties", + "SqlFciProperties", + "SqlJob", + "SqlJobListResult", + "SqlMachineOverview", + "SqlRunAsAccount", + "SqlRunAsAccountListResult", + "SqlServerApplication", + "SqlServerProperties", + "SqlServerV2", + "SqlServerV2ListResult", + "SqlServerV2Update", + "SqlServerV2UpdateProperties", + "SqlSite", + "SqlSiteListResult", + "SqlSiteProperties", + "SqlSiteRefreshBody", + "SqlSiteUpdate", + "SqlSiteUpdateProperties", + "SqlSiteUsage", + "StaticDiscovery", + "SystemCenter", + "SystemData", + "TagsMachineInput", + "TomcatEngineUnit", + "TomcatHostUnit", + "TomcatServiceUnit", + "TomcatWebApplicationProperties", + "TomcatWebApplications", + "TomcatWebApplicationsListResult", + "TomcatWebServerProperties", + "TomcatWebServers", + "TomcatWebServersListResult", + "TrackedResource", + "UpdateMachineDepMapStatus", + "UpdateMachineRunAsAccount", + "UpdateMachineTags", + "V20180501PreviewVmwareRunAsAccount", + "V20180501PreviewVmwareRunAsAccountVmwareRunAsAccountCollection", + "Vcenter", + "VcenterListResult", + "VcenterProperties", + "VmwareDatastore", + "VmwareDisk", + "VmwareHost", + "VmwareHostListResult", + "VmwareHostProperties", + "VmwareJob", + "VmwareJobListResult", + "VmwareMachineProperties", + "VmwareMachineSoftwareInventory", + "VmwareMachineSoftwareInventoryListResult", + "VmwareNetworkAdapter", + "VmwareRunAsAccountResource", + "VmwareRunAsAccountResourceListResult", + "VmwareSite", + "VmwareSiteListResult", + "VmwareSiteUpdate", + "VmwareSiteUpdateProperties", + "VmwareSiteUsage", + "WebAppDiscovery", + "WebAppExtendedMachine", + "WebAppExtendedMachineListResult", + "WebAppExtendedMachineProperties", + "WebAppProperties", + "WebAppPropertiesCollection", + "WebAppRunAsAccount", + "WebAppRunAsAccountListResult", + "WebAppSite", + "WebAppSiteListResult", + "WebAppSiteProperties", + "WebAppSiteUpdate", + "WebAppSiteUpdateProperties", + "WebAppSiteUsage", + "WebApplication", + "WebApplicationAppsAndRolesModel", + "WebApplicationConfigurationUnit", + "WebApplicationDirectoryUnit", + "WebApplicationFramework", + "WebApplicationListResult", + "WebApplicationProperties", + "WebRoleImportDisk", + "WebRoleOperatingSystem", + "WebServer", + "WebServerListResult", + "WebServerProperties", + "ActionType", + "ApplicationDiscoveryScopeStatus", + "CreatedByType", + "Default", + "DefaultAutoGenerated", + "DefaultValues", + "DeleteImportedMachinesJobPropertiesJobState", + "DependencyMapDiscoveryScopeStatus", + "DiscoveryScopeStatus", + "DiscoveryScopes", + "EsuStatus", + "EsuYear", + "ExportMachineErrorsProperties", + "FCIInstanceState", + "FileType", + "HealthErrorDetailsDiscoveryScope", + "HealthErrorDetailsSource", + "HighAvailability", + "HighAvailabilityValues", + "ImportTypeValues", + "JobResult", + "MasterSitePropertiesPublicNetworkAccess", + "MicrosoftAzureFDSWebRoleHealthErrorDetailsSource", + "Origin", + "PrivateLinkServiceConnectionStateStatus", + "ProvisioningState", + "SQLDiscoveryScopeStatus", + "ShallowDiscoveryStatus", + "SiteHealthSummaryFabricLayoutUpdateSourcesItem", + "SqlAvailabilityGroupPropertiesAvailabilityGroupType", + "SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaCommitMode", + "SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaReadMode", + "SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSeedMode", + "SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaState", + "SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSyncStatus", + "SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaType", + "SqlAvailabilityReplicaOverviewReplicaState", + "SqlMachineOverviewFciRole", + "SqlMetadataDiscoveryPipe", + "SqlServerStatus", + "SqlSitePropertiesDiscoveryScenario", + "StaticDiscoveryScopeStatus", + "SupportStatus", + "VMwareDatastoreType", + "Versions", + "WebAppSitePropertiesDiscoveryScenario", +] +__all__.extend([p for p in _patch_all if p not in __all__]) +_patch_sdk() diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_azure_migrate_discovery_service_enums.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_azure_migrate_discovery_service_enums.py new file mode 100644 index 000000000000..b480fa0953ea --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_azure_migrate_discovery_service_enums.py @@ -0,0 +1,654 @@ +# 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 enum import Enum +from azure.core import CaseInsensitiveEnumMeta + + +class ActionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Enum. Indicates the action type. "Internal" refers to actions that are for internal only APIs.""" + + INTERNAL = "Internal" + + +class ApplicationDiscoveryScopeStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Application Discovery Scope Status.""" + + DISCOVERY_SUCCEEDED_ATLEAST_ONCE = "DiscoverySucceededAtleastOnce" + """DiscoverySucceededAtleastOnce value.""" + DISCOVERY_FAILED = "DiscoveryFailed" + """DiscoveryFailed value.""" + RUN_AS_ACCOUNT_NOT_ASSOCIATED = "RunAsAccountNotAssociated" + """RunAsAccountNotAssociated value.""" + DISCOVERY_NOT_STARTED = "DiscoveryNotStarted" + """DiscoveryNotStarted value.""" + DISCOVERY_IN_PROGRESS = "DiscoveryInProgress" + """DiscoveryInProgress value.""" + DISABLED = "Disabled" + """Disabled value.""" + DISCOVERY_PARTIALLY_SUCCEDED = "DiscoveryPartiallySucceded" + """DiscoveryPartiallySucceded value.""" + DISCOVERY_SUCCEEDED = "DiscoverySucceeded" + """DiscoverySucceeded value.""" + + +class CreatedByType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The type of identity that created the resource.""" + + USER = "User" + APPLICATION = "Application" + MANAGED_IDENTITY = "ManagedIdentity" + KEY = "Key" + + +class Default(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Default.""" + + DEFAULT = "default" + """default value.""" + + +class DefaultAutoGenerated(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """HealthError Details Source.""" + + DEFAULT = "default" + """default value.""" + + +class DefaultValues(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """HealthError Details Source.""" + + DEFAULT = "default" + """default value.""" + + +class DeleteImportedMachinesJobPropertiesJobState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Cosmos db Imported Machines JobEntity.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + VERIFIED = "Verified" + """Verified value.""" + VERIFIED_WITH_ERRORS = "VerifiedWithErrors" + """VerifiedWithErrors value.""" + COMPLETED = "Completed" + """Completed value.""" + FAILED = "Failed" + """Failed value.""" + + +class DependencyMapDiscoveryScopeStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """DependencyMap DiscoveryScope Status.""" + + DISCOVERY_SUCCEEDED_ATLEAST_ONCE = "DiscoverySucceededAtleastOnce" + """DiscoverySucceededAtleastOnce value.""" + DISCOVERY_FAILED = "DiscoveryFailed" + """DiscoveryFailed value.""" + RUN_AS_ACCOUNT_NOT_ASSOCIATED = "RunAsAccountNotAssociated" + """RunAsAccountNotAssociated value.""" + DISCOVERY_NOT_STARTED = "DiscoveryNotStarted" + """DiscoveryNotStarted value.""" + DISCOVERY_IN_PROGRESS = "DiscoveryInProgress" + """DiscoveryInProgress value.""" + DISABLED = "Disabled" + """Disabled value.""" + DISCOVERY_PARTIALLY_SUCCEDED = "DiscoveryPartiallySucceded" + """DiscoveryPartiallySucceded value.""" + DISCOVERY_SUCCEEDED = "DiscoverySucceeded" + """DiscoverySucceeded value.""" + + +class DiscoveryScopes(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Discovery scopes.""" + + APPS_AND_ROLES = "AppsAndRoles" + """AppsAndRoles value.""" + DEPENDENCY_MAP = "DependencyMap" + """DependencyMap value.""" + STATIC_DATA = "StaticData" + """StaticData value.""" + SQL_SERVER_CONNECTION_INFO = "SQLServerConnectionInfo" + """SQLServerConnectionInfo value.""" + + +class DiscoveryScopeStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Discovery Scope.""" + + DISCOVERY_SUCCEEDED_ATLEAST_ONCE = "DiscoverySucceededAtleastOnce" + """DiscoverySucceededAtleastOnce value.""" + DISCOVERY_FAILED = "DiscoveryFailed" + """DiscoveryFailed value.""" + RUN_AS_ACCOUNT_NOT_ASSOCIATED = "RunAsAccountNotAssociated" + """RunAsAccountNotAssociated value.""" + DISCOVERY_NOT_STARTED = "DiscoveryNotStarted" + """DiscoveryNotStarted value.""" + DISCOVERY_IN_PROGRESS = "DiscoveryInProgress" + """DiscoveryInProgress value.""" + DISABLED = "Disabled" + """Disabled value.""" + DISCOVERY_PARTIALLY_SUCCEDED = "DiscoveryPartiallySucceded" + """DiscoveryPartiallySucceded value.""" + DISCOVERY_SUCCEEDED = "DiscoverySucceeded" + """DiscoverySucceeded value.""" + + +class EsuStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """esu Status.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + ACTIVE = "Active" + """Active value.""" + IN_ACTIVE = "InActive" + """InActive value.""" + + +class EsuYear(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """esu year.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + FIRST_YEAR = "FirstYear" + """FirstYear value.""" + SECOND_YEAR = "SecondYear" + """SecondYear value.""" + THIRD_YEAR = "ThirdYear" + """ThirdYear value.""" + UPGRADE_YEAR = "UpgradeYear" + """UpgradeYear value.""" + + +class ExportMachineErrorsProperties(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Export Machine Errors Properties.""" + + APPS_AND_ROLES = "AppsAndRoles" + """AppsAndRoles value.""" + DEPENDENCY_MAP = "DependencyMap" + """DependencyMap value.""" + STATIC_DATA = "StaticData" + """StaticData value.""" + SQL_SERVER_CONNECTION_INFO = "SQLServerConnectionInfo" + """SQLServerConnectionInfo value.""" + + +class FCIInstanceState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """fci instance state.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + INHERITED = "Inherited" + """Inherited value.""" + INITIALIZING = "Initializing" + """Initializing value.""" + ONLINE = "Online" + """Online value.""" + OFFLINE = "Offline" + """Offline value.""" + FAILED = "Failed" + """Failed value.""" + PENDING = "Pending" + """Pending value.""" + ONLINE_PENDING = "OnlinePending" + """OnlinePending value.""" + OFFLINE_PENDING = "OfflinePending" + """OfflinePending value.""" + + +class FileType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """file type.""" + + ROWS = "Rows" + """Rows value.""" + LOG = "Log" + """Log value.""" + FILESTREAM = "Filestream" + """Filestream value.""" + NOT_SUPPORTED = "NotSupported" + """NotSpecified value.""" + FULLTEXT = "Fulltext" + """Fulltext value.""" + + +class HealthErrorDetailsDiscoveryScope(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Gets discovery scope for which error is encountered.""" + + APPS_AND_ROLES = "AppsAndRoles" + """AppsAndRoles value.""" + DEPENDENCY_MAP = "DependencyMap" + """DependencyMap value.""" + STATIC_DATA = "StaticData" + """StaticData value.""" + SQL_SERVER_CONNECTION_INFO = "SQLServerConnectionInfo" + """SQLServerConnectionInfo value.""" + DISCOVERY_TARGETS = "DiscoveryTargets" + """DiscoveryTargets value.""" + + +class HealthErrorDetailsSource(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """HealthError Details Source.""" + + REFRESH_FABRIC_LAYOUT = "RefreshFabricLayout" + """RefreshFabricLayout value.""" + REFRESH_FABRIC_LAYOUT_GUEST = "RefreshFabricLayoutGuest" + """RefreshFabricLayoutGuest value.""" + REFRESH_FABRIC_LAYOUT_DEPENDENCY_MAP = "RefreshFabricLayoutDependencyMap" + """RefreshFabricLayoutDependencyMap value.""" + + +class HighAvailability(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Value indicating whether the VM is highly available.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + NO = "No" + """No value.""" + YES = "Yes" + """Yes value.""" + + +class HighAvailabilityValues(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Value indicating whether the VM is highly available.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + NO = "No" + """No value.""" + YES = "Yes" + """Yes value.""" + + +class ImportTypeValues(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """ImportTypeValues enum.""" + + AZURE_MIGRATE_CSV = "AzureMigrateCSV" + """AzureMigrateCSV value.""" + RV_TOOLS_XLSX = "RVToolsXlsx" + """RVToolsXlsx value.""" + + +class JobResult(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """JobResultValues.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + COMPLETED = "Completed" + """Completed value.""" + COMPLETED_WITH_WARNINGS = "CompletedWithWarnings" + """CompletedWithWarnings value.""" + COMPLETED_WITH_ERRORS = "CompletedWithErrors" + """CompletedWithErrors value.""" + FAILED = "Failed" + """Failed value.""" + WAITING_FOR_BLOB_UPLOAD = "WaitingForBlobUpload" + """WaitingForBlobUpload value.""" + IN_PROGRESS = "InProgress" + """InProgress value.""" + + +class MasterSitePropertiesPublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """PublicNetworkAccess.""" + + NOT_SPECIFIED = "NotSpecified" + """NotSpecified value.""" + ENABLED = "Enabled" + """Enabled value.""" + DISABLED = "Disabled" + """Disabled value.""" + + +class MicrosoftAzureFDSWebRoleHealthErrorDetailsSource(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """HealthError Details Source.""" + + REFRESH_FABRIC_LAYOUT = "RefreshFabricLayout" + """RefreshFabricLayout value.""" + REFRESH_FABRIC_LAYOUT_GUEST = "RefreshFabricLayoutGuest" + """RefreshFabricLayoutGuest value.""" + REFRESH_FABRIC_LAYOUT_DEPENDENCY_MAP = "RefreshFabricLayoutDependencyMap" + """RefreshFabricLayoutDependencyMap value.""" + + +class Origin(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The intended executor of the operation; as in Resource Based Access Control (RBAC) and audit + logs UX. Default value is "user,system". + """ + + USER = "user" + SYSTEM = "system" + USER_SYSTEM = "user,system" + + +class PrivateLinkServiceConnectionStateStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """PrivateLinkServiceConnectionState.""" + + APPROVED = "Approved" + """Approved value.""" + PENDING = "Pending" + """Pending value.""" + REJECTED = "Rejected" + """Rejected value.""" + DISCONNECTED = "Disconnected" + """Disconnected value.""" + + +class ProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The status of the current operation.""" + + CREATED = "Created" + """Created value.""" + UPDATED = "Updated" + """Updated value.""" + RUNNING = "Running" + """Running value.""" + COMPLETED = "Completed" + """Completed value.""" + FAILED = "Failed" + """Failed value.""" + SUCCEEDED = "Succeeded" + """Succeeded value.""" + CANCELED = "Canceled" + """Canceled value.""" + + +class ShallowDiscoveryStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Shallow Discovery Status.""" + + DISCOVERY_SUCCEEDED_ATLEAST_ONCE = "DiscoverySucceededAtleastOnce" + """DiscoverySucceededAtleastOnce value.""" + DISCOVERY_FAILED = "DiscoveryFailed" + """DiscoveryFailed value.""" + RUN_AS_ACCOUNT_NOT_ASSOCIATED = "RunAsAccountNotAssociated" + """RunAsAccountNotAssociated value.""" + DISCOVERY_NOT_STARTED = "DiscoveryNotStarted" + """DiscoveryNotStarted value.""" + DISCOVERY_IN_PROGRESS = "DiscoveryInProgress" + """DiscoveryInProgress value.""" + DISABLED = "Disabled" + """Disabled value.""" + DISCOVERY_PARTIALLY_SUCCEDED = "DiscoveryPartiallySucceded" + """DiscoveryPartiallySucceded value.""" + DISCOVERY_SUCCEEDED = "DiscoverySucceeded" + """DiscoverySucceeded value.""" + + +class SiteHealthSummaryFabricLayoutUpdateSourcesItem(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """SiteHealthSummary FabricLayout UpdateSources.""" + + REFRESH_FABRIC_LAYOUT = "RefreshFabricLayout" + """RefreshFabricLayout value.""" + REFRESH_FABRIC_LAYOUT_GUEST = "RefreshFabricLayoutGuest" + """RefreshFabricLayoutGuest value.""" + REFRESH_FABRIC_LAYOUT_DEPENDENCY_MAP = "RefreshFabricLayoutDependencyMap" + """RefreshFabricLayoutDependencyMap value.""" + + +class SqlAvailabilityGroupPropertiesAvailabilityGroupType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """AvailabilityGroupType.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + TRADITIONAL = "Traditional" + """Traditional value.""" + DISTRIBUTED = "Distributed" + """Distributed value.""" + + +class SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaCommitMode( + str, Enum, metaclass=CaseInsensitiveEnumMeta +): + """ReplicaCommitMode.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + SYNCHRONOUS = "Synchronous" + """Synchronous value.""" + ASYNCHRONOUS = "Asynchronous" + """Asynchronous value.""" + + +class SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaReadMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """ReplicaReadMode.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + NONE = "None" + """None value.""" + READ_ONLY = "ReadOnly" + """ReadOnly value.""" + READ_WRITE = "ReadWrite" + """ReadWrite value.""" + + +class SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSeedMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """ReplicaSeedMode.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + MANUAL = "Manual" + """Manual value.""" + AUTOMATIC = "Automatic" + """Automatic value.""" + + +class SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """ReplicaState.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + PRIMARY = "Primary" + """Primary value.""" + SECONDARY = "Secondary" + """Secondary value.""" + + +class SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSyncStatus( + str, Enum, metaclass=CaseInsensitiveEnumMeta +): + """ReplicaSyncStatus.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + SYNCHRONIZED = "Synchronized" + """Synchronized value.""" + UNSYNCHRONIZED = "Unsynchronized" + """Unsynchronized value.""" + + +class SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """ReplicaType.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + DATABASE_REPLICA = "DatabaseReplica" + """DatabaseReplica value.""" + AVAILABILITY_GROUP_REPLICA = "AvailabilityGroupReplica" + """AvailabilityGroupReplica value.""" + + +class SqlAvailabilityReplicaOverviewReplicaState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Overview Replica State.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + PRIMARY = "Primary" + """Primary value.""" + SECONDARY = "Secondary" + """Secondary value.""" + + +class SQLDiscoveryScopeStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """SQLDiscoveryScope Status.""" + + DISCOVERY_SUCCEEDED_ATLEAST_ONCE = "DiscoverySucceededAtleastOnce" + """DiscoverySucceededAtleastOnce value.""" + DISCOVERY_FAILED = "DiscoveryFailed" + """DiscoveryFailed value.""" + RUN_AS_ACCOUNT_NOT_ASSOCIATED = "RunAsAccountNotAssociated" + """RunAsAccountNotAssociated value.""" + DISCOVERY_NOT_STARTED = "DiscoveryNotStarted" + """DiscoveryNotStarted value.""" + DISCOVERY_IN_PROGRESS = "DiscoveryInProgress" + """DiscoveryInProgress value.""" + DISABLED = "Disabled" + """Disabled value.""" + DISCOVERY_PARTIALLY_SUCCEDED = "DiscoveryPartiallySucceded" + """DiscoveryPartiallySucceded value.""" + DISCOVERY_SUCCEEDED = "DiscoverySucceeded" + """DiscoverySucceeded value.""" + + +class SqlMachineOverviewFciRole(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """sql fci role.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + NOT_APPLICABLE = "NotApplicable" + """NotApplicable value.""" + ACTIVE_NODE = "ActiveNode" + """ActiveNode value.""" + POSSIBLE_OWNER_NODE = "PossibleOwnerNode" + """PossibleOwnerNode value.""" + + +class SqlMetadataDiscoveryPipe(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """SQL metadata discovery pipe.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + V_MWARE = "VMware" + """VMware value.""" + POWER_SHELL = "PowerShell" + """PowerShell value.""" + SSH = "SSH" + """SSH value.""" + CIM = "CIM" + """CIM value.""" + OTHER = "Other" + """Other value.""" + + +class SqlServerStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """sql server status.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + CONTINUE_PENDING = "ContinuePending" + """ContinuePending value.""" + PAUSED = "Paused" + """Paused value.""" + PAUSE_PENDING = "PausePending" + """PausePending value.""" + RUNNING = "Running" + """Running value.""" + START_PENDING = "StartPending" + """StartPending value.""" + STOPPED = "Stopped" + """Stopped value.""" + STOP_PENDING = "StopPending" + """StopPending value.""" + + +class SqlSitePropertiesDiscoveryScenario(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """SqlSiteProperties DiscoveryScenario.""" + + MIGRATE = "Migrate" + """Migrate value.""" + DR = "DR" + """DR value.""" + + +class StaticDiscoveryScopeStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Static DiscoveryScopeStatus.""" + + DISCOVERY_SUCCEEDED_ATLEAST_ONCE = "DiscoverySucceededAtleastOnce" + """DiscoverySucceededAtleastOnce value.""" + DISCOVERY_FAILED = "DiscoveryFailed" + """DiscoveryFailed value.""" + RUN_AS_ACCOUNT_NOT_ASSOCIATED = "RunAsAccountNotAssociated" + """RunAsAccountNotAssociated value.""" + DISCOVERY_NOT_STARTED = "DiscoveryNotStarted" + """DiscoveryNotStarted value.""" + DISCOVERY_IN_PROGRESS = "DiscoveryInProgress" + """DiscoveryInProgress value.""" + DISABLED = "Disabled" + """Disabled value.""" + DISCOVERY_PARTIALLY_SUCCEDED = "DiscoveryPartiallySucceded" + """DiscoveryPartiallySucceded value.""" + DISCOVERY_SUCCEEDED = "DiscoverySucceeded" + """DiscoverySucceeded value.""" + + +class SupportStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """support status.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + MAINSTREAM = "Mainstream" + """Mainstream value.""" + EXTENDED = "Extended" + """Extended value.""" + + +class Versions(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """service api versions.""" + + V2018_05_01_PREVIEW = "2018-05-01-preview" + """Api version 2018-05-01-preview.""" + V2019_05_01_PREVIEW = "2019-05-01-preview" + """Api version 2019-05-01-preview.""" + V2020_01_01 = "2020-01-01" + """Api version 2020-01-01.""" + V2020_01_01_PREVIEW = "2020-01-01-preview" + """Api version 2020-01-01-preview.""" + V2020_02_01 = "2020-02-01" + """Api version 2020-02-01.""" + V2020_07_07 = "2020-07-07" + """Api version 2020-07-07.""" + V2020_07_10 = "2020-07-10" + """Api version 2020-07-10.""" + V2020_08_01_PREVIEW = "2020-08-01-preview" + """Api version 2020-08-01-preview.""" + V2020_11_11_PREVIEW = "2020-11-11-preview" + """Api version 2020-11-11-preview.""" + V2022_10_27 = "2022-10-27" + """Api version 2022-10-27.""" + V2023_06_06 = "2023-06-06" + """Api version 2023-06-06.""" + V2023_10_01_PREVIEW = "2023-10-01-preview" + """Api version 2023-10-01-preview.""" + + +class VMwareDatastoreType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """vmware datastore type.""" + + UNKNOWN = "Unknown" + """Unknown value.""" + CIFS = "CIFS" + """CIFS value.""" + NFS = "NFS" + """NFS value.""" + NFS41 = "NFS41" + """NFS41 value.""" + PMEM = "PMEM" + """PMEM value.""" + VFFS = "VFFS" + """VFFS value.""" + VMFS = "VMFS" + """VMFS value.""" + VSAN = "VSAN" + """VSAN value.""" + VVOL = "VVOL" + """VVOL value.""" + + +class WebAppSitePropertiesDiscoveryScenario(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Discovery Scenario.""" + + MIGRATE = "Migrate" + """Migrate value.""" + DR = "DR" + """DR value.""" diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_models_py3.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_models_py3.py new file mode 100644 index 000000000000..c9617509c2cb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_models_py3.py @@ -0,0 +1,13467 @@ +# 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 +import sys +from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union + +from .. import _serialization + +if sys.version_info >= (3, 9): + from collections.abc import MutableMapping +else: + from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports + +if TYPE_CHECKING: + # pylint: disable=unused-import,ungrouped-imports + from .. import models as _models +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object + + +class Application(_serialization.Model): + """Application in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar name: Gets or sets Name of the Application. + :vartype name: str + :ivar version: Gets or sets Version of the Application. + :vartype version: str + :ivar provider: Gets or sets Provider of the Application. + :vartype provider: str + """ + + _validation = { + "name": {"readonly": True}, + "version": {"readonly": True}, + "provider": {"readonly": True}, + } + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "provider": {"key": "provider", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.name = None + self.version = None + self.provider = None + + +class ApplicationDiscovery(_serialization.Model): + """Contains data related application and roles discovery scope. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar discovery_scope_status: Gets discovery scope status. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype discovery_scope_status: str or + ~azure.mgmt.migrate.models.ApplicationDiscoveryScopeStatus + :ivar errors: Gets errors for discovery scope. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar hydrated_run_as_account_id: Gets the run as account ID with which feature worked + successfully. + + It is discovered by the agent from the list of credentials. + :vartype hydrated_run_as_account_id: str + """ + + _validation = { + "discovery_scope_status": {"readonly": True}, + "errors": {"readonly": True}, + "hydrated_run_as_account_id": {"readonly": True}, + } + + _attribute_map = { + "discovery_scope_status": {"key": "discoveryScopeStatus", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "hydrated_run_as_account_id": {"key": "hydratedRunAsAccountId", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.discovery_scope_status = None + self.errors = None + self.hydrated_run_as_account_id = None + + +class AppsAndRoles(_serialization.Model): + """AppsAndRoles in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar applications: Gets or sets Applications of the AppsAndRoles. + :vartype applications: list[~azure.mgmt.migrate.models.Application] + :ivar web_applications: Gets or sets WebApplications of the AppsAndRoles. + :vartype web_applications: list[~azure.mgmt.migrate.models.WebApplicationAppsAndRolesModel] + :ivar features: Gets or sets Features of the AppsAndRoles. + :vartype features: list[~azure.mgmt.migrate.models.Feature] + :ivar sql_servers: Gets or sets SQLServers of the AppsAndRoles. + :vartype sql_servers: list[~azure.mgmt.migrate.models.SqlServerApplication] + :ivar share_point_servers: Gets or sets SharePointServers of the AppsAndRoles. + :vartype share_point_servers: list[~azure.mgmt.migrate.models.SharePointServer] + :ivar system_centers: Gets or sets SystemCenters of the AppsAndRoles. + :vartype system_centers: list[~azure.mgmt.migrate.models.SystemCenter] + :ivar biz_talk_servers: Gets or sets BizTalkServers of the AppsAndRoles. + :vartype biz_talk_servers: list[~azure.mgmt.migrate.models.BizTalkServer] + :ivar exchange_servers: Gets or sets ExchangeServers of the AppsAndRoles. + :vartype exchange_servers: list[~azure.mgmt.migrate.models.ExchangeServer] + :ivar other_databases: Gets or sets OtherDatabaseServers of the AppsAndRoles. + :vartype other_databases: list[~azure.mgmt.migrate.models.OtherDatabase] + """ + + _validation = { + "applications": {"readonly": True}, + "web_applications": {"readonly": True}, + "features": {"readonly": True}, + "sql_servers": {"readonly": True}, + "share_point_servers": {"readonly": True}, + "system_centers": {"readonly": True}, + "biz_talk_servers": {"readonly": True}, + "exchange_servers": {"readonly": True}, + "other_databases": {"readonly": True}, + } + + _attribute_map = { + "applications": {"key": "applications", "type": "[Application]"}, + "web_applications": {"key": "webApplications", "type": "[WebApplicationAppsAndRolesModel]"}, + "features": {"key": "features", "type": "[Feature]"}, + "sql_servers": {"key": "sqlServers", "type": "[SqlServerApplication]"}, + "share_point_servers": {"key": "sharePointServers", "type": "[SharePointServer]"}, + "system_centers": {"key": "systemCenters", "type": "[SystemCenter]"}, + "biz_talk_servers": {"key": "bizTalkServers", "type": "[BizTalkServer]"}, + "exchange_servers": {"key": "exchangeServers", "type": "[ExchangeServer]"}, + "other_databases": {"key": "otherDatabases", "type": "[OtherDatabase]"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.applications = None + self.web_applications = None + self.features = None + self.sql_servers = None + self.share_point_servers = None + self.system_centers = None + self.biz_talk_servers = None + self.exchange_servers = None + self.other_databases = None + + +class BizTalkServer(_serialization.Model): + """BizTalkServer in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar product_name: Gets or sets ProductName of the BizTalkServer. + :vartype product_name: str + :ivar status: Gets or sets Status of the BizTalkServer. + :vartype status: str + """ + + _validation = { + "product_name": {"readonly": True}, + "status": {"readonly": True}, + } + + _attribute_map = { + "product_name": {"key": "productName", "type": "str"}, + "status": {"key": "status", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.product_name = None + self.status = None + + +class ConnectorUnit(_serialization.Model): + """The connectors of the web server. + + :ivar bindings: Gets or sets the bindings for the connector. + :vartype bindings: list[~azure.mgmt.migrate.models.FrontEndBinding] + """ + + _attribute_map = { + "bindings": {"key": "bindings", "type": "[FrontEndBinding]"}, + } + + def __init__(self, *, bindings: Optional[List["_models.FrontEndBinding"]] = None, **kwargs: Any) -> None: + """ + :keyword bindings: Gets or sets the bindings for the connector. + :paramtype bindings: list[~azure.mgmt.migrate.models.FrontEndBinding] + """ + super().__init__(**kwargs) + self.bindings = bindings + + +class DeleteImportedMachinesJobProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Delete Imported Machines Job Properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar blob_name: blob name. + :vartype blob_name: str + :ivar error_sas_uri: error Sas Uri. + :vartype error_sas_uri: str + :ivar job_state: job State. Known values are: "Unknown", "Verified", "VerifiedWithErrors", + "Completed", and "Failed". + :vartype job_state: str or + ~azure.mgmt.migrate.models.DeleteImportedMachinesJobPropertiesJobState + :ivar number_of_machines_deleted: number Of Machines Deleted. + :vartype number_of_machines_deleted: int + :ivar deletion_confirmation: deletion Confirmation. + :vartype deletion_confirmation: bool + :ivar errors: errors list. + :vartype errors: list[str] + :ivar status: Gets or sets the Job status. + :vartype status: str + :ivar start_time: Gets or sets the Job start time. + :vartype start_time: str + :ivar end_time: Gets or sets the Job end time. + :vartype end_time: str + :ivar display_name: Gets or sets the Display name. + :vartype display_name: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "errors": {"readonly": True}, + } + + _attribute_map = { + "blob_name": {"key": "blobName", "type": "str"}, + "error_sas_uri": {"key": "errorSasUri", "type": "str"}, + "job_state": {"key": "jobState", "type": "str"}, + "number_of_machines_deleted": {"key": "numberOfMachinesDeleted", "type": "int"}, + "deletion_confirmation": {"key": "deletionConfirmation", "type": "bool"}, + "errors": {"key": "errors", "type": "[str]"}, + "status": {"key": "status", "type": "str"}, + "start_time": {"key": "startTime", "type": "str"}, + "end_time": {"key": "endTime", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + blob_name: Optional[str] = None, + error_sas_uri: Optional[str] = None, + job_state: Optional[Union[str, "_models.DeleteImportedMachinesJobPropertiesJobState"]] = None, + number_of_machines_deleted: Optional[int] = None, + deletion_confirmation: Optional[bool] = None, + status: Optional[str] = None, + start_time: Optional[str] = None, + end_time: Optional[str] = None, + display_name: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword blob_name: blob name. + :paramtype blob_name: str + :keyword error_sas_uri: error Sas Uri. + :paramtype error_sas_uri: str + :keyword job_state: job State. Known values are: "Unknown", "Verified", "VerifiedWithErrors", + "Completed", and "Failed". + :paramtype job_state: str or + ~azure.mgmt.migrate.models.DeleteImportedMachinesJobPropertiesJobState + :keyword number_of_machines_deleted: number Of Machines Deleted. + :paramtype number_of_machines_deleted: int + :keyword deletion_confirmation: deletion Confirmation. + :paramtype deletion_confirmation: bool + :keyword status: Gets or sets the Job status. + :paramtype status: str + :keyword start_time: Gets or sets the Job start time. + :paramtype start_time: str + :keyword end_time: Gets or sets the Job end time. + :paramtype end_time: str + :keyword display_name: Gets or sets the Display name. + :paramtype display_name: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.blob_name = blob_name + self.error_sas_uri = error_sas_uri + self.job_state = job_state + self.number_of_machines_deleted = number_of_machines_deleted + self.deletion_confirmation = deletion_confirmation + self.errors = None + self.status = status + self.start_time = start_time + self.end_time = end_time + self.display_name = display_name + self.provisioning_state = provisioning_state + + +class Resource(_serialization.Model): + """Common fields that are returned in the response for all Azure Resource Manager resources. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.id = None + self.name = None + self.type = None + self.system_data = None + + +class ProxyResource(Resource): + """The resource model definition for a Azure Resource Manager proxy resource. It will not have + tags and a location. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + """ + + +class DeleteImportMachinesJob(ProxyResource): + """A host resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.DeleteImportedMachinesJobProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "DeleteImportedMachinesJobProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.DeleteImportedMachinesJobProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.DeleteImportedMachinesJobProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class DeleteImportMachinesJobCollection(_serialization.Model): + """Delete import machines job collection. + + 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 value: Gets the list of jobs. Required. + :vartype value: list[~azure.mgmt.migrate.models.DeleteImportMachinesJob] + :ivar next_link: Gets the value of next link. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + "next_link": {"readonly": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[DeleteImportMachinesJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.DeleteImportMachinesJob"], **kwargs: Any) -> None: + """ + :keyword value: Gets the list of jobs. Required. + :paramtype value: list[~azure.mgmt.migrate.models.DeleteImportMachinesJob] + """ + super().__init__(**kwargs) + self.value = value + self.next_link = None + + +class DependencyMapDiscovery(_serialization.Model): + """Contains data related dependency map discovery scope. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar discovery_scope_status: Gets discovery scope status. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype discovery_scope_status: str or + ~azure.mgmt.migrate.models.DependencyMapDiscoveryScopeStatus + :ivar errors: Gets errors for discovery scope. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar hydrated_run_as_account_id: Gets the run as account ID with which feature worked + successfully. + + It is discovered by the agent from the list of credentials. + :vartype hydrated_run_as_account_id: str + """ + + _validation = { + "discovery_scope_status": {"readonly": True}, + "errors": {"readonly": True}, + "hydrated_run_as_account_id": {"readonly": True}, + } + + _attribute_map = { + "discovery_scope_status": {"key": "discoveryScopeStatus", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "hydrated_run_as_account_id": {"key": "hydratedRunAsAccountId", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.discovery_scope_status = None + self.errors = None + self.hydrated_run_as_account_id = None + + +class DependencyMapMachineInput(_serialization.Model): + """Machine class. + + :ivar machine_id: Gets or sets the ARM id of the machine. + :vartype machine_id: str + :ivar is_dependency_map_to_be_enabled: Gets or sets a value indicating whether + dependency mapping is to + be enabled or not. + :vartype is_dependency_map_to_be_enabled: bool + """ + + _attribute_map = { + "machine_id": {"key": "machineId", "type": "str"}, + "is_dependency_map_to_be_enabled": {"key": "isDependencyMapToBeEnabled", "type": "bool"}, + } + + def __init__( + self, *, machine_id: Optional[str] = None, is_dependency_map_to_be_enabled: Optional[bool] = None, **kwargs: Any + ) -> None: + """ + :keyword machine_id: Gets or sets the ARM id of the machine. + :paramtype machine_id: str + :keyword is_dependency_map_to_be_enabled: Gets or sets a value indicating whether + dependency mapping is to + be enabled or not. + :paramtype is_dependency_map_to_be_enabled: bool + """ + super().__init__(**kwargs) + self.machine_id = machine_id + self.is_dependency_map_to_be_enabled = is_dependency_map_to_be_enabled + + +class DependencyMapServiceMapextensionsClientGroupMembersRequest(_serialization.Model): # pylint: disable=name-too-long + """DependencyMap ServiceMap extensions Client GroupMembers Request. + + :ivar machine_id: id of machine. + :vartype machine_id: str + :ivar process_group_name: process group name. + :vartype process_group_name: str + :ivar process_name: name of process. + :vartype process_name: str + :ivar start_time: start time. + :vartype start_time: ~datetime.datetime + :ivar end_time: end time. + :vartype end_time: ~datetime.datetime + :ivar filters: optional filters. + :vartype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + + _attribute_map = { + "machine_id": {"key": "machineId", "type": "str"}, + "process_group_name": {"key": "processGroupName", "type": "str"}, + "process_name": {"key": "processName", "type": "str"}, + "start_time": {"key": "startTime", "type": "iso-8601"}, + "end_time": {"key": "endTime", "type": "iso-8601"}, + "filters": {"key": "filters", "type": "DependencyMapServiceMapextensionsDependencyMapRequestFilters"}, + } + + def __init__( + self, + *, + machine_id: Optional[str] = None, + process_group_name: Optional[str] = None, + process_name: Optional[str] = None, + start_time: Optional[datetime.datetime] = None, + end_time: Optional[datetime.datetime] = None, + filters: Optional["_models.DependencyMapServiceMapextensionsDependencyMapRequestFilters"] = None, + **kwargs: Any + ) -> None: + """ + :keyword machine_id: id of machine. + :paramtype machine_id: str + :keyword process_group_name: process group name. + :paramtype process_group_name: str + :keyword process_name: name of process. + :paramtype process_name: str + :keyword start_time: start time. + :paramtype start_time: ~datetime.datetime + :keyword end_time: end time. + :paramtype end_time: ~datetime.datetime + :keyword filters: optional filters. + :paramtype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + super().__init__(**kwargs) + self.machine_id = machine_id + self.process_group_name = process_group_name + self.process_name = process_name + self.start_time = start_time + self.end_time = end_time + self.filters = filters + + +class DependencyMapServiceMapextensionsDependencyMapRequestFilters( + _serialization.Model +): # pylint: disable=name-too-long + """DependencyMap Service Map extensions Dependency Map Request Filters. + + :ivar machine_ids: array of machine ids. + :vartype machine_ids: list[str] + :ivar process_ids: array of process Ids. + :vartype process_ids: list[str] + """ + + _attribute_map = { + "machine_ids": {"key": "machineIds", "type": "[str]"}, + "process_ids": {"key": "processIds", "type": "[str]"}, + } + + def __init__( + self, *, machine_ids: Optional[List[str]] = None, process_ids: Optional[List[str]] = None, **kwargs: Any + ) -> None: + """ + :keyword machine_ids: array of machine ids. + :paramtype machine_ids: list[str] + :keyword process_ids: array of process Ids. + :paramtype process_ids: list[str] + """ + super().__init__(**kwargs) + self.machine_ids = machine_ids + self.process_ids = process_ids + + +class DependencyMapServiceMapextensionsExportDependenciesRequest(_serialization.Model): # pylint: disable=name-too-long + """DependencyMap ServiceMap extensions ExportDependencies Request. + + :ivar start_time: start time. + :vartype start_time: ~datetime.datetime + :ivar end_time: end time. + :vartype end_time: ~datetime.datetime + """ + + _attribute_map = { + "start_time": {"key": "startTime", "type": "iso-8601"}, + "end_time": {"key": "endTime", "type": "iso-8601"}, + } + + def __init__( + self, + *, + start_time: Optional[datetime.datetime] = None, + end_time: Optional[datetime.datetime] = None, + **kwargs: Any + ) -> None: + """ + :keyword start_time: start time. + :paramtype start_time: ~datetime.datetime + :keyword end_time: end time. + :paramtype end_time: ~datetime.datetime + """ + super().__init__(**kwargs) + self.start_time = start_time + self.end_time = end_time + + +class DependencyMapServiceMapextensionsScopeMapRequest(_serialization.Model): # pylint: disable=name-too-long + """DependencyMap ServiceMapextensions Scope MapRequest. + + :ivar start_time: start time. + :vartype start_time: ~datetime.datetime + :ivar end_time: end time. + :vartype end_time: ~datetime.datetime + :ivar filters: optional filters. + :vartype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + + _attribute_map = { + "start_time": {"key": "startTime", "type": "iso-8601"}, + "end_time": {"key": "endTime", "type": "iso-8601"}, + "filters": {"key": "filters", "type": "DependencyMapServiceMapextensionsDependencyMapRequestFilters"}, + } + + def __init__( + self, + *, + start_time: Optional[datetime.datetime] = None, + end_time: Optional[datetime.datetime] = None, + filters: Optional["_models.DependencyMapServiceMapextensionsDependencyMapRequestFilters"] = None, + **kwargs: Any + ) -> None: + """ + :keyword start_time: start time. + :paramtype start_time: ~datetime.datetime + :keyword end_time: end time. + :paramtype end_time: ~datetime.datetime + :keyword filters: optional filters. + :paramtype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + super().__init__(**kwargs) + self.start_time = start_time + self.end_time = end_time + self.filters = filters + + +class DependencyMapServiceMapextensionsServerGroupMembersRequest(_serialization.Model): # pylint: disable=name-too-long + """DependencyMap ServiceMap extensions ServerGroup Members Request. + + :ivar server_port: port of server. + :vartype server_port: int + :ivar start_time: start time. + :vartype start_time: ~datetime.datetime + :ivar end_time: end time. + :vartype end_time: ~datetime.datetime + :ivar filters: optional filters. + :vartype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + + _attribute_map = { + "server_port": {"key": "serverPort", "type": "int"}, + "start_time": {"key": "startTime", "type": "iso-8601"}, + "end_time": {"key": "endTime", "type": "iso-8601"}, + "filters": {"key": "filters", "type": "DependencyMapServiceMapextensionsDependencyMapRequestFilters"}, + } + + def __init__( + self, + *, + server_port: Optional[int] = None, + start_time: Optional[datetime.datetime] = None, + end_time: Optional[datetime.datetime] = None, + filters: Optional["_models.DependencyMapServiceMapextensionsDependencyMapRequestFilters"] = None, + **kwargs: Any + ) -> None: + """ + :keyword server_port: port of server. + :paramtype server_port: int + :keyword start_time: start time. + :paramtype start_time: ~datetime.datetime + :keyword end_time: end time. + :paramtype end_time: ~datetime.datetime + :keyword filters: optional filters. + :paramtype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + super().__init__(**kwargs) + self.server_port = server_port + self.start_time = start_time + self.end_time = end_time + self.filters = filters + + +class DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest( + _serialization.Model +): # pylint: disable=name-too-long + """DependencyMap ServiceMap extensions SingleMachine DetailedMap Request. + + :ivar machine_id: id of machine. + :vartype machine_id: str + :ivar start_time: start time. + :vartype start_time: ~datetime.datetime + :ivar end_time: end time. + :vartype end_time: ~datetime.datetime + :ivar filters: optional filters. + :vartype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + + _attribute_map = { + "machine_id": {"key": "machineId", "type": "str"}, + "start_time": {"key": "startTime", "type": "iso-8601"}, + "end_time": {"key": "endTime", "type": "iso-8601"}, + "filters": {"key": "filters", "type": "DependencyMapServiceMapextensionsDependencyMapRequestFilters"}, + } + + def __init__( + self, + *, + machine_id: Optional[str] = None, + start_time: Optional[datetime.datetime] = None, + end_time: Optional[datetime.datetime] = None, + filters: Optional["_models.DependencyMapServiceMapextensionsDependencyMapRequestFilters"] = None, + **kwargs: Any + ) -> None: + """ + :keyword machine_id: id of machine. + :paramtype machine_id: str + :keyword start_time: start time. + :paramtype start_time: ~datetime.datetime + :keyword end_time: end time. + :paramtype end_time: ~datetime.datetime + :keyword filters: optional filters. + :paramtype filters: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsDependencyMapRequestFilters + """ + super().__init__(**kwargs) + self.machine_id = machine_id + self.start_time = start_time + self.end_time = end_time + self.filters = filters + + +class DirectoryPath(_serialization.Model): + """Second level entity for virtual directories. + + :ivar virtual: Gets or sets the virtual path for the directory. + :vartype virtual: str + :ivar physical: Gets or sets the physical path of the directory on the web server. + :vartype physical: str + """ + + _attribute_map = { + "virtual": {"key": "virtual", "type": "str"}, + "physical": {"key": "physical", "type": "str"}, + } + + def __init__(self, *, virtual: Optional[str] = None, physical: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword virtual: Gets or sets the virtual path for the directory. + :paramtype virtual: str + :keyword physical: Gets or sets the physical path of the directory on the web server. + :paramtype physical: str + """ + super().__init__(**kwargs) + self.virtual = virtual + self.physical = physical + + +class DiscoveryScopeErrorSummary(_serialization.Model): + """The site error summary model. + + All required parameters must be populated in order to send to server. + + :ivar affected_resource_type: Gets or sets the affected resource type. Required. + :vartype affected_resource_type: str + :ivar affected_objects_count: Count of affected objects. Required. + :vartype affected_objects_count: int + :ivar discovery_scope: Gets or sets the count of affected objects. Required. Known values are: + "AppsAndRoles", "DependencyMap", "StaticData", and "SQLServerConnectionInfo". + :vartype discovery_scope: str or ~azure.mgmt.migrate.models.DiscoveryScopes + """ + + _validation = { + "affected_resource_type": {"required": True}, + "affected_objects_count": {"required": True}, + "discovery_scope": {"required": True}, + } + + _attribute_map = { + "affected_resource_type": {"key": "affectedResourceType", "type": "str"}, + "affected_objects_count": {"key": "affectedObjectsCount", "type": "int"}, + "discovery_scope": {"key": "discoveryScope", "type": "str"}, + } + + def __init__( + self, + *, + affected_resource_type: str, + affected_objects_count: int, + discovery_scope: Union[str, "_models.DiscoveryScopes"], + **kwargs: Any + ) -> None: + """ + :keyword affected_resource_type: Gets or sets the affected resource type. Required. + :paramtype affected_resource_type: str + :keyword affected_objects_count: Count of affected objects. Required. + :paramtype affected_objects_count: int + :keyword discovery_scope: Gets or sets the count of affected objects. Required. Known values + are: "AppsAndRoles", "DependencyMap", "StaticData", and "SQLServerConnectionInfo". + :paramtype discovery_scope: str or ~azure.mgmt.migrate.models.DiscoveryScopes + """ + super().__init__(**kwargs) + self.affected_resource_type = affected_resource_type + self.affected_objects_count = affected_objects_count + self.discovery_scope = discovery_scope + + +class DiscoverySiteDataSource(ProxyResource): + """Web app data source web model. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.DiscoverySiteDataSourceProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "DiscoverySiteDataSourceProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.DiscoverySiteDataSourceProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.DiscoverySiteDataSourceProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class DiscoverySiteDataSourceListResult(_serialization.Model): + """The response of a DiscoverySiteDataSource list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The DiscoverySiteDataSource items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[DiscoverySiteDataSource]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.DiscoverySiteDataSource"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The DiscoverySiteDataSource items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :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 DiscoverySiteDataSourceProperties(_serialization.Model): + """Discovery site data source properties class. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar discovery_site_id: Gets or sets the discovery site Id. + :vartype discovery_site_id: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "discovery_site_id": {"key": "discoverySiteId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, *, discovery_site_id: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword discovery_site_id: Gets or sets the discovery site Id. + :paramtype discovery_site_id: str + """ + super().__init__(**kwargs) + self.discovery_site_id = discovery_site_id + self.provisioning_state = None + + +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.migrate.models.ErrorDetail] + :ivar additional_info: The error additional info. + :vartype additional_info: list[~azure.mgmt.migrate.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 ErrorDetails(_serialization.Model): + """Error contract returned when some exception occurs in Rest API. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar code: Gets the error code. + :vartype code: str + :ivar message: Gets the error message. + :vartype message: str + :ivar possible_causes: Gets the possible causes of error. + :vartype possible_causes: str + :ivar recommended_action: Gets the recommended action to resolve error. + :vartype recommended_action: str + :ivar severity: Gets the error severity. + :vartype severity: str + :ivar is_agent_reported_error: Gets a value indicating whether the error originated from a + agent or not. + :vartype is_agent_reported_error: bool + :ivar agent_error_code: Gets the agent error code. + :vartype agent_error_code: str + :ivar agent_error_message: Gets the error message from the agent. + :vartype agent_error_message: str + :ivar agent_error_possible_causes: Gets possible causes for the agent error. + :vartype agent_error_possible_causes: str + :ivar agent_error_recommended_action: Gets the recommended action for the agent error. + :vartype agent_error_recommended_action: str + """ + + _validation = { + "code": {"readonly": True}, + "message": {"readonly": True}, + "possible_causes": {"readonly": True}, + "recommended_action": {"readonly": True}, + "severity": {"readonly": True}, + "is_agent_reported_error": {"readonly": True}, + "agent_error_code": {"readonly": True}, + "agent_error_message": {"readonly": True}, + "agent_error_possible_causes": {"readonly": True}, + "agent_error_recommended_action": {"readonly": True}, + } + + _attribute_map = { + "code": {"key": "code", "type": "str"}, + "message": {"key": "message", "type": "str"}, + "possible_causes": {"key": "possibleCauses", "type": "str"}, + "recommended_action": {"key": "recommendedAction", "type": "str"}, + "severity": {"key": "severity", "type": "str"}, + "is_agent_reported_error": {"key": "isAgentReportedError", "type": "bool"}, + "agent_error_code": {"key": "agentErrorCode", "type": "str"}, + "agent_error_message": {"key": "agentErrorMessage", "type": "str"}, + "agent_error_possible_causes": {"key": "agentErrorPossibleCauses", "type": "str"}, + "agent_error_recommended_action": {"key": "agentErrorRecommendedAction", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.code = None + self.message = None + self.possible_causes = None + self.recommended_action = None + self.severity = None + self.is_agent_reported_error = None + self.agent_error_code = None + self.agent_error_message = None + self.agent_error_possible_causes = None + self.agent_error_recommended_action = 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.migrate.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.migrate.models.ErrorDetail + """ + super().__init__(**kwargs) + self.error = error + + +class Errors(_serialization.Model): # pylint: disable=too-many-instance-attributes + """The properties of a server machine tracked resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar message: Gets the error message. + :vartype message: str + :ivar message_parameters: Gets the message parameters. + :vartype message_parameters: dict[str, str] + :ivar appliance_name: Gets the appliance name. + :vartype appliance_name: str + :ivar id: Gets the error ID. + :vartype id: int + :ivar code: Gets the error name. + :vartype code: str + :ivar possible_causes: Gets the possible causes of error. + :vartype possible_causes: str + :ivar recommended_action: Gets the recommended action to resolve error. + :vartype recommended_action: str + :ivar severity: Gets the error severity. + :vartype severity: str + :ivar summary_message: Gets the error summary message. + :vartype summary_message: str + :ivar source: Gets the error source. Known values are: "RefreshFabricLayout", + "RefreshFabricLayoutGuest", and "RefreshFabricLayoutDependencyMap". + :vartype source: str or + ~azure.mgmt.migrate.models.MicrosoftAzureFDSWebRoleHealthErrorDetailsSource + :ivar updated_time_stamp: Gets the time stamp when the error was updated. + :vartype updated_time_stamp: ~datetime.datetime + :ivar run_as_account_id: Gets run as account id used while performing discovery of + entity. + :vartype run_as_account_id: str + :ivar discovery_scope: Gets discovery scope for which error is encountered. Known + values are: "AppsAndRoles", "DependencyMap", "StaticData", "SQLServerConnectionInfo", and + "DiscoveryTargets". + :vartype discovery_scope: str or ~azure.mgmt.migrate.models.HealthErrorDetailsDiscoveryScope + """ + + _validation = { + "message": {"readonly": True}, + "message_parameters": {"readonly": True}, + "appliance_name": {"readonly": True}, + "id": {"readonly": True}, + "code": {"readonly": True}, + "possible_causes": {"readonly": True}, + "recommended_action": {"readonly": True}, + "severity": {"readonly": True}, + "summary_message": {"readonly": True}, + "source": {"readonly": True}, + "updated_time_stamp": {"readonly": True}, + "run_as_account_id": {"readonly": True}, + "discovery_scope": {"readonly": True}, + } + + _attribute_map = { + "message": {"key": "message", "type": "str"}, + "message_parameters": {"key": "messageParameters", "type": "{str}"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "id": {"key": "id", "type": "int"}, + "code": {"key": "code", "type": "str"}, + "possible_causes": {"key": "possibleCauses", "type": "str"}, + "recommended_action": {"key": "recommendedAction", "type": "str"}, + "severity": {"key": "severity", "type": "str"}, + "summary_message": {"key": "summaryMessage", "type": "str"}, + "source": {"key": "source", "type": "str"}, + "updated_time_stamp": {"key": "updatedTimeStamp", "type": "iso-8601"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "discovery_scope": {"key": "discoveryScope", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.message = None + self.message_parameters = None + self.appliance_name = None + self.id = None + self.code = None + self.possible_causes = None + self.recommended_action = None + self.severity = None + self.summary_message = None + self.source = None + self.updated_time_stamp = None + self.run_as_account_id = None + self.discovery_scope = None + + +class ErrorSummaryRequest(_serialization.Model): + """ErrorSummaryRequest body. + + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + """ + + _attribute_map = { + "appliance_name": {"key": "applianceName", "type": "str"}, + } + + def __init__(self, *, appliance_name: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + """ + super().__init__(**kwargs) + self.appliance_name = appliance_name + + +class ExchangeServer(_serialization.Model): + """ExchangeServer in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar product_name: Gets or sets ProductName of the ExchangeServer. + :vartype product_name: str + :ivar edition: Gets or sets Edition of the ExchangeServer. + :vartype edition: str + :ivar roles: Gets or sets Roles of the ExchangeServer. + :vartype roles: str + :ivar service_pack: Gets or sets ServicePack of the ExchangeServer. + :vartype service_pack: str + :ivar version: Gets or sets Version of the ExchangeServer. + :vartype version: str + """ + + _validation = { + "product_name": {"readonly": True}, + "edition": {"readonly": True}, + "roles": {"readonly": True}, + "service_pack": {"readonly": True}, + "version": {"readonly": True}, + } + + _attribute_map = { + "product_name": {"key": "productName", "type": "str"}, + "edition": {"key": "edition", "type": "str"}, + "roles": {"key": "roles", "type": "str"}, + "service_pack": {"key": "servicePack", "type": "str"}, + "version": {"key": "version", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.product_name = None + self.edition = None + self.roles = None + self.service_pack = None + self.version = None + + +class ExportImportedMachinesJob(_serialization.Model): + """Export machines job REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar type: Type name for export job. + :vartype type: str + :ivar properties: Gets or sets the export machines job properties. + :vartype properties: ~azure.mgmt.migrate.models.ExportImportedMachinesJobEntityProperties + :ivar id: Gets or sets the relative ARM name to get job. + :vartype id: str + :ivar name: Gets or sets the Job ID. + :vartype name: str + :ivar status: Gets or sets the Job status. + :vartype status: str + :ivar start_time: Gets or sets the Job start time. + :vartype start_time: str + :ivar end_time: Gets or sets the Job end time. + :vartype end_time: str + :ivar display_name: Gets or sets the Display name. + :vartype display_name: str + """ + + _validation = { + "type": {"readonly": True}, + } + + _attribute_map = { + "type": {"key": "type", "type": "str"}, + "properties": {"key": "properties", "type": "ExportImportedMachinesJobEntityProperties"}, + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "start_time": {"key": "startTime", "type": "str"}, + "end_time": {"key": "endTime", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + } + + def __init__( + self, + *, + properties: Optional["_models.ExportImportedMachinesJobEntityProperties"] = None, + id: Optional[str] = None, # pylint: disable=redefined-builtin + name: Optional[str] = None, + status: Optional[str] = None, + start_time: Optional[str] = None, + end_time: Optional[str] = None, + display_name: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword properties: Gets or sets the export machines job properties. + :paramtype properties: ~azure.mgmt.migrate.models.ExportImportedMachinesJobEntityProperties + :keyword id: Gets or sets the relative ARM name to get job. + :paramtype id: str + :keyword name: Gets or sets the Job ID. + :paramtype name: str + :keyword status: Gets or sets the Job status. + :paramtype status: str + :keyword start_time: Gets or sets the Job start time. + :paramtype start_time: str + :keyword end_time: Gets or sets the Job end time. + :paramtype end_time: str + :keyword display_name: Gets or sets the Display name. + :paramtype display_name: str + """ + super().__init__(**kwargs) + self.type = None + self.properties = properties + self.id = id + self.name = name + self.status = status + self.start_time = start_time + self.end_time = end_time + self.display_name = display_name + + +class ExportImportedMachinesJobEntityProperties(_serialization.Model): # pylint: disable=name-too-long + """Export Imported Machines JobEntity Properties. + + :ivar blob_name: blob name. + :vartype blob_name: str + :ivar sas_uri: sas uri. + :vartype sas_uri: str + """ + + _attribute_map = { + "blob_name": {"key": "blobName", "type": "str"}, + "sas_uri": {"key": "sasUri", "type": "str"}, + } + + def __init__(self, *, blob_name: Optional[str] = None, sas_uri: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword blob_name: blob name. + :paramtype blob_name: str + :keyword sas_uri: sas uri. + :paramtype sas_uri: str + """ + super().__init__(**kwargs) + self.blob_name = blob_name + self.sas_uri = sas_uri + + +class ExportMachineErrorsRequest(_serialization.Model): + """The body of export machine errors request. + + :ivar properties: Gets or sets the properties. + :vartype properties: ~azure.mgmt.migrate.models.RequestExportMachineErrorsProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "RequestExportMachineErrorsProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.RequestExportMachineErrorsProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: Gets or sets the properties. + :paramtype properties: ~azure.mgmt.migrate.models.RequestExportMachineErrorsProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ExportMachinesRequest(_serialization.Model): + """ExportMachinesRequest body. + + :ivar filter: filter options. + :vartype filter: str + """ + + _attribute_map = { + "filter": {"key": "filter", "type": "str"}, + } + + def __init__(self, *, filter: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + """ + :keyword filter: filter options. + :paramtype filter: str + """ + super().__init__(**kwargs) + self.filter = filter + + +class ExportSqlServersRequest(_serialization.Model): + """ExportSqlServerRequest body. + + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + :ivar filter: filter options. + :vartype filter: str + """ + + _attribute_map = { + "appliance_name": {"key": "applianceName", "type": "str"}, + "filter": {"key": "filter", "type": "str"}, + } + + def __init__( + self, + *, + appliance_name: Optional[str] = None, + filter: Optional[str] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: + """ + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + :keyword filter: filter options. + :paramtype filter: str + """ + super().__init__(**kwargs) + self.appliance_name = appliance_name + self.filter = filter + + +class ExportWebAppsRequest(_serialization.Model): + """ExportWebAppsRequest body. + + :ivar filter: filter options. + :vartype filter: str + """ + + _attribute_map = { + "filter": {"key": "filter", "type": "str"}, + } + + def __init__(self, *, filter: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + """ + :keyword filter: filter options. + :paramtype filter: str + """ + super().__init__(**kwargs) + self.filter = filter + + +class Feature(_serialization.Model): + """Feature in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar name: Gets or sets Name of the Feature. + :vartype name: str + :ivar feature_type: Gets or sets FeatureType of the Feature. + :vartype feature_type: str + :ivar parent: Gets or sets Parent of the Feature. + :vartype parent: str + :ivar status: Gets or sets Status of the Feature. + :vartype status: str + """ + + _validation = { + "name": {"readonly": True}, + "feature_type": {"readonly": True}, + "parent": {"readonly": True}, + "status": {"readonly": True}, + } + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "feature_type": {"key": "featureType", "type": "str"}, + "parent": {"key": "parent", "type": "str"}, + "status": {"key": "status", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.name = None + self.feature_type = None + self.parent = None + self.status = None + + +class FileMetaData(_serialization.Model): + """File Metadata web model. + + :ivar logical_name: Gets or sets the logical name of the file. + :vartype logical_name: str + :ivar physical_full_name: Gets or sets the operating-system full path of the file. + :vartype physical_full_name: str + :ivar file_type: Gets or sets the database file type. Known values are: "Rows", "Log", + "Filestream", "NotSupported", and "Fulltext". + :vartype file_type: str or ~azure.mgmt.migrate.models.FileType + :ivar size_in_mb: Gets or sets the size of the file in MB. + :vartype size_in_mb: float + :ivar is_memory_optimized_data_option_enabled: Gets or sets a value indicating whether memory + optimized data option is enabled. + :vartype is_memory_optimized_data_option_enabled: bool + """ + + _attribute_map = { + "logical_name": {"key": "logicalName", "type": "str"}, + "physical_full_name": {"key": "physicalFullName", "type": "str"}, + "file_type": {"key": "fileType", "type": "str"}, + "size_in_mb": {"key": "sizeInMb", "type": "float"}, + "is_memory_optimized_data_option_enabled": {"key": "isMemoryOptimizedDataOptionEnabled", "type": "bool"}, + } + + def __init__( + self, + *, + logical_name: Optional[str] = None, + physical_full_name: Optional[str] = None, + file_type: Optional[Union[str, "_models.FileType"]] = None, + size_in_mb: Optional[float] = None, + is_memory_optimized_data_option_enabled: Optional[bool] = None, + **kwargs: Any + ) -> None: + """ + :keyword logical_name: Gets or sets the logical name of the file. + :paramtype logical_name: str + :keyword physical_full_name: Gets or sets the operating-system full path of the file. + :paramtype physical_full_name: str + :keyword file_type: Gets or sets the database file type. Known values are: "Rows", "Log", + "Filestream", "NotSupported", and "Fulltext". + :paramtype file_type: str or ~azure.mgmt.migrate.models.FileType + :keyword size_in_mb: Gets or sets the size of the file in MB. + :paramtype size_in_mb: float + :keyword is_memory_optimized_data_option_enabled: Gets or sets a value indicating whether + memory optimized data option is enabled. + :paramtype is_memory_optimized_data_option_enabled: bool + """ + super().__init__(**kwargs) + self.logical_name = logical_name + self.physical_full_name = physical_full_name + self.file_type = file_type + self.size_in_mb = size_in_mb + self.is_memory_optimized_data_option_enabled = is_memory_optimized_data_option_enabled + + +class FrontEndBinding(_serialization.Model): + """Frontend bindings for a web application. + + :ivar protocol: Gets or sets the Binding protocol. + :vartype protocol: str + :ivar host_name: Gets or sets the Host name. + :vartype host_name: str + :ivar port: Gets or sets the Port number. + :vartype port: str + :ivar ip_address: Gets or sets the IP Address. + :vartype ip_address: str + """ + + _attribute_map = { + "protocol": {"key": "protocol", "type": "str"}, + "host_name": {"key": "hostName", "type": "str"}, + "port": {"key": "port", "type": "str"}, + "ip_address": {"key": "ipAddress", "type": "str"}, + } + + def __init__( + self, + *, + protocol: Optional[str] = None, + host_name: Optional[str] = None, + port: Optional[str] = None, + ip_address: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword protocol: Gets or sets the Binding protocol. + :paramtype protocol: str + :keyword host_name: Gets or sets the Host name. + :paramtype host_name: str + :keyword port: Gets or sets the Port number. + :paramtype port: str + :keyword ip_address: Gets or sets the IP Address. + :paramtype ip_address: str + """ + super().__init__(**kwargs) + self.protocol = protocol + self.host_name = host_name + self.port = port + self.ip_address = ip_address + + +class GuestOsDetails(_serialization.Model): + """Second level object. Data related to a machine's operating system. Serialized and + stored as part of Machine Rest object. + + :ivar os_type: Gets or sets the type of the operating system. + :vartype os_type: str + :ivar os_name: Gets or sets the Name of the operating system. + :vartype os_name: str + :ivar os_version: Gets or sets the Version of the operating system. + :vartype os_version: str + :ivar os_architecture: Gets or sets the Architecture of the operating system. + :vartype os_architecture: str + """ + + _attribute_map = { + "os_type": {"key": "osType", "type": "str"}, + "os_name": {"key": "osName", "type": "str"}, + "os_version": {"key": "osVersion", "type": "str"}, + "os_architecture": {"key": "osArchitecture", "type": "str"}, + } + + def __init__( + self, + *, + os_type: Optional[str] = None, + os_name: Optional[str] = None, + os_version: Optional[str] = None, + os_architecture: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword os_type: Gets or sets the type of the operating system. + :paramtype os_type: str + :keyword os_name: Gets or sets the Name of the operating system. + :paramtype os_name: str + :keyword os_version: Gets or sets the Version of the operating system. + :paramtype os_version: str + :keyword os_architecture: Gets or sets the Architecture of the operating system. + :paramtype os_architecture: str + """ + super().__init__(**kwargs) + self.os_type = os_type + self.os_name = os_name + self.os_version = os_version + self.os_architecture = os_architecture + + +class HealthErrorDetails(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Error contract returned when some exception occurs in Rest API. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar message: Gets the error message. + :vartype message: str + :ivar message_parameters: Gets the message parameters. + :vartype message_parameters: dict[str, str] + :ivar appliance_name: Gets the appliance name. + :vartype appliance_name: str + :ivar id: Gets the error ID. + :vartype id: int + :ivar code: Gets the error name. + :vartype code: str + :ivar possible_causes: Gets the possible causes of error. + :vartype possible_causes: str + :ivar recommended_action: Gets the recommended action to resolve error. + :vartype recommended_action: str + :ivar severity: Gets the error severity. + :vartype severity: str + :ivar summary_message: Gets the error summary message. + :vartype summary_message: str + :ivar source: Gets the error source. Known values are: "RefreshFabricLayout", + "RefreshFabricLayoutGuest", and "RefreshFabricLayoutDependencyMap". + :vartype source: str or ~azure.mgmt.migrate.models.HealthErrorDetailsSource + :ivar updated_time_stamp: Gets the time stamp when the error was updated. + :vartype updated_time_stamp: ~datetime.datetime + :ivar run_as_account_id: Gets run as account id used while performing discovery of + entity. + :vartype run_as_account_id: str + :ivar discovery_scope: Gets discovery scope for which error is encountered. Known + values are: "AppsAndRoles", "DependencyMap", "StaticData", "SQLServerConnectionInfo", and + "DiscoveryTargets". + :vartype discovery_scope: str or ~azure.mgmt.migrate.models.HealthErrorDetailsDiscoveryScope + """ + + _validation = { + "message": {"readonly": True}, + "message_parameters": {"readonly": True}, + "appliance_name": {"readonly": True}, + "id": {"readonly": True}, + "code": {"readonly": True}, + "possible_causes": {"readonly": True}, + "recommended_action": {"readonly": True}, + "severity": {"readonly": True}, + "summary_message": {"readonly": True}, + "source": {"readonly": True}, + "updated_time_stamp": {"readonly": True}, + "run_as_account_id": {"readonly": True}, + "discovery_scope": {"readonly": True}, + } + + _attribute_map = { + "message": {"key": "message", "type": "str"}, + "message_parameters": {"key": "messageParameters", "type": "{str}"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "id": {"key": "id", "type": "int"}, + "code": {"key": "code", "type": "str"}, + "possible_causes": {"key": "possibleCauses", "type": "str"}, + "recommended_action": {"key": "recommendedAction", "type": "str"}, + "severity": {"key": "severity", "type": "str"}, + "summary_message": {"key": "summaryMessage", "type": "str"}, + "source": {"key": "source", "type": "str"}, + "updated_time_stamp": {"key": "updatedTimeStamp", "type": "iso-8601"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "discovery_scope": {"key": "discoveryScope", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.message = None + self.message_parameters = None + self.appliance_name = None + self.id = None + self.code = None + self.possible_causes = None + self.recommended_action = None + self.severity = None + self.summary_message = None + self.source = None + self.updated_time_stamp = None + self.run_as_account_id = None + self.discovery_scope = None + + +class HypervCluster(ProxyResource): + """A cluster resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.HypervClusterProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "HypervClusterProperties"}, + } + + def __init__(self, *, properties: Optional["_models.HypervClusterProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.HypervClusterProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervClusterListResult(_serialization.Model): + """The response of a HypervCluster list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The HypervCluster items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.HypervCluster] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[HypervCluster]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.HypervCluster"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The HypervCluster items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.HypervCluster] + :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 HypervClusterProperties(_serialization.Model): + """The properties of Hyperv Cluster. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar created_timestamp: Gets the timestamp marking Hyper-V cluster creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last updated on the Hyper-V cluster. + :vartype updated_timestamp: str + :ivar fqdn: Gets or sets the FQDN/IPAddress of the Hyper-V cluster. + :vartype fqdn: str + :ivar functional_level: Gets the functional level of the Hyper-V cluster. + :vartype functional_level: int + :ivar status: Gets the status of the Hyper-V cluster. + :vartype status: str + :ivar run_as_account_id: Gets or sets Run as account ID of the Hyper-V cluster. + :vartype run_as_account_id: str + :ivar host_fqdn_list: Gets or sets list of hosts (FQDN) currently being tracked by the cluster. + :vartype host_fqdn_list: list[str] + :ivar errors: Gets the errors. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "functional_level": {"readonly": True}, + "status": {"readonly": True}, + "errors": {"readonly": True}, + } + + _attribute_map = { + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "fqdn": {"key": "fqdn", "type": "str"}, + "functional_level": {"key": "functionalLevel", "type": "int"}, + "status": {"key": "status", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "host_fqdn_list": {"key": "hostFqdnList", "type": "[str]"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + fqdn: Optional[str] = None, + run_as_account_id: Optional[str] = None, + host_fqdn_list: Optional[List[str]] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword fqdn: Gets or sets the FQDN/IPAddress of the Hyper-V cluster. + :paramtype fqdn: str + :keyword run_as_account_id: Gets or sets Run as account ID of the Hyper-V cluster. + :paramtype run_as_account_id: str + :keyword host_fqdn_list: Gets or sets list of hosts (FQDN) currently being tracked by the + cluster. + :paramtype host_fqdn_list: list[str] + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.created_timestamp = None + self.updated_timestamp = None + self.fqdn = fqdn + self.functional_level = None + self.status = None + self.run_as_account_id = run_as_account_id + self.host_fqdn_list = host_fqdn_list + self.errors = None + self.provisioning_state = provisioning_state + + +class HypervDisk(_serialization.Model): + """Second level object returned as part of Machine REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar instance_id: Id of the disk. + :vartype instance_id: str + :ivar vhd_id: VHD Id of the disk. + :vartype vhd_id: str + :ivar max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :vartype max_size_in_bytes: int + :ivar name: Gets or sets Name of the disk. + :vartype name: str + :ivar disk_type: Gets or sets Type of the disk. + :vartype disk_type: str + :ivar lun: Gets or sets LUN of the disk. + :vartype lun: int + :ivar path: Gets or sets Path of the disk. + :vartype path: str + """ + + _validation = { + "instance_id": {"readonly": True}, + "vhd_id": {"readonly": True}, + } + + _attribute_map = { + "instance_id": {"key": "instanceId", "type": "str"}, + "vhd_id": {"key": "vhdId", "type": "str"}, + "max_size_in_bytes": {"key": "maxSizeInBytes", "type": "int"}, + "name": {"key": "name", "type": "str"}, + "disk_type": {"key": "diskType", "type": "str"}, + "lun": {"key": "lun", "type": "int"}, + "path": {"key": "path", "type": "str"}, + } + + def __init__( + self, + *, + max_size_in_bytes: Optional[int] = None, + name: Optional[str] = None, + disk_type: Optional[str] = None, + lun: Optional[int] = None, + path: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :paramtype max_size_in_bytes: int + :keyword name: Gets or sets Name of the disk. + :paramtype name: str + :keyword disk_type: Gets or sets Type of the disk. + :paramtype disk_type: str + :keyword lun: Gets or sets LUN of the disk. + :paramtype lun: int + :keyword path: Gets or sets Path of the disk. + :paramtype path: str + """ + super().__init__(**kwargs) + self.instance_id = None + self.vhd_id = None + self.max_size_in_bytes = max_size_in_bytes + self.name = name + self.disk_type = disk_type + self.lun = lun + self.path = path + + +class HypervHost(ProxyResource): + """A host resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.HypervHostProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "HypervHostProperties"}, + } + + def __init__(self, *, properties: Optional["_models.HypervHostProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.HypervHostProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervHostListResult(_serialization.Model): + """The response of a HypervHost list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The HypervHost items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.HypervHost] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[HypervHost]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.HypervHost"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The HypervHost items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.HypervHost] + :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 HypervHostProperties(_serialization.Model): + """The properties of Hyperv Host. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar created_timestamp: Gets the timestamp marking Hyper-V host creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last updated on the Hyper-V host. + :vartype updated_timestamp: str + :ivar fqdn: Gets or sets the FQDN/IPAddress of the Hyper-V host. + :vartype fqdn: str + :ivar run_as_account_id: Gets or sets the run as account ID of the Hyper-V host. + :vartype run_as_account_id: str + :ivar version: Gets the version of the Hyper-V host. + :vartype version: str + :ivar errors: Gets the errors. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "version": {"readonly": True}, + "errors": {"readonly": True}, + } + + _attribute_map = { + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "fqdn": {"key": "fqdn", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + fqdn: Optional[str] = None, + run_as_account_id: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword fqdn: Gets or sets the FQDN/IPAddress of the Hyper-V host. + :paramtype fqdn: str + :keyword run_as_account_id: Gets or sets the run as account ID of the Hyper-V host. + :paramtype run_as_account_id: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.created_timestamp = None + self.updated_timestamp = None + self.fqdn = fqdn + self.run_as_account_id = run_as_account_id + self.version = None + self.errors = None + self.provisioning_state = provisioning_state + + +class HypervJob(ProxyResource): + """A job resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.JobProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "JobProperties"}, + } + + def __init__(self, *, properties: Optional["_models.JobProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.JobProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervJobListResult(_serialization.Model): + """The response of a HypervJob list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The HypervJob items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.HypervJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[HypervJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.HypervJob"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The HypervJob items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.HypervJob] + :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 HypervMachine(ProxyResource): + """A machine resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.HypervMachineProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "HypervMachineProperties"}, + } + + def __init__(self, *, properties: Optional["_models.HypervMachineProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.HypervMachineProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervMachineListResult(_serialization.Model): + """The response of a HypervMachine list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The HypervMachine items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.HypervMachine] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[HypervMachine]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.HypervMachine"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The HypervMachine items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.HypervMachine] + :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 HypervMachineProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """The properties of AddressResource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar instance_uuid: On-premise Instance UUID of the machine. + :vartype instance_uuid: str + :ivar host_fqdn: Host FQDN/IPAddress. + :vartype host_fqdn: str + :ivar host_id: Host ARM ID. + :vartype host_id: str + :ivar generation: Generation of the virtual machine. + :vartype generation: int + :ivar version: VM version. + :vartype version: str + :ivar high_availability: Value indicating whether the VM is highly available. Known values are: + "Unknown", "No", and "Yes". + :vartype high_availability: str or ~azure.mgmt.migrate.models.HighAvailability + :ivar cluster_fqdn: Cluster FQDN/IPAddress. + :vartype cluster_fqdn: str + :ivar cluster_id: Cluster ARM ID. + :vartype cluster_id: str + :ivar max_memory_mb: Max memory of the virtual machine in MB. + :vartype max_memory_mb: int + :ivar is_dynamic_memory_enabled: Value indicating whether dynamic memory is enabled for the VM. + :vartype is_dynamic_memory_enabled: bool + :ivar disks: Disks attached to the machine. + :vartype disks: list[~azure.mgmt.migrate.models.HypervDisk] + :ivar network_adapters: Network adapters attached to the machine. + :vartype network_adapters: list[~azure.mgmt.migrate.models.HypervNetworkAdapter] + :ivar management_server_type: Management server type captured as a string representation of the + + {Microsoft.Azure.FDS.WebRole.HyperVMachineBase.HyperVMachineBaseProperties.ManagementServerType} + enumeration. + :vartype management_server_type: str + :ivar secure_boot_template_id: Gets or sets the SecureBootTemplateId setting of the VM. + :vartype secure_boot_template_id: str + :ivar secure_boot_enabled: Gets or sets a value indicating whether Secure boot is enabled for + the VM. + :vartype secure_boot_enabled: bool + :ivar secure_boot_template: Gets or sets the SecureBootTemplateId setting of the VM. + :vartype secure_boot_template: str + :ivar tpm_enabled: Gets or sets a value indicating whether trusted platform module is enabled + on + the VM. + :vartype tpm_enabled: bool + :ivar ksd_enabled: Gets or sets a value indicating whether key storage device is enabled on the + VM. + :vartype ksd_enabled: bool + :ivar shielding_requested: Gets or sets a value indicating whether shielding is enabled for the + VM. + :vartype shielding_requested: bool + :ivar data_protection_requested: Gets or sets a value indicating whether data protection is + requested for a VM. + :vartype data_protection_requested: bool + :ivar encrypt_state_and_vm_migration_traffic: Gets or sets a value indicating whether + encryption of state and migration + traffic is enabled for the VM. + :vartype encrypt_state_and_vm_migration_traffic: bool + :ivar virtualization_based_security_opt_out: Gets or sets a value indicating whether VM + virtualization based security is + enabled for the VM. + :vartype virtualization_based_security_opt_out: bool + :ivar power_status: Gets the Machine power status. + :vartype power_status: str + :ivar vm_fqdn: Gets the VM FQDN. + :vartype vm_fqdn: str + :ivar vm_configuration_file_location: Gets the Root location of the VM configuration file. + :vartype vm_configuration_file_location: str + :ivar firmware: Gets or sets the firmware. + :vartype firmware: str + :ivar guest_os_details: Operating System Details extracted from the guest bu + executing script inside the guest VM. + :vartype guest_os_details: ~azure.mgmt.migrate.models.GuestOsDetails + :ivar number_of_applications: Number of applications installed in the guest VM. + :vartype number_of_applications: int + :ivar guest_details_discovery_timestamp: The last time at which the Guest Details was + discovered + or the + error while discovering guest details based discovery + of the + machine. + :vartype guest_details_discovery_timestamp: ~datetime.datetime + :ivar is_guest_details_discovery_in_progress: Whether Refresh Fabric Layout Guest Details has + been completed once. + + Portal will show discovery in progress, if this value is true. + :vartype is_guest_details_discovery_in_progress: bool + :ivar dependency_mapping: Gets or sets if dependency mapping feature is enabled or not + for + the VM. + :vartype dependency_mapping: str + :ivar dependency_mapping_start_time: Gets or sets when dependency mapping collection is last + started. + :vartype dependency_mapping_start_time: ~datetime.datetime + :ivar dependency_mapping_end_time: Gets or sets when dependency mapping collection was last + disabled. + :vartype dependency_mapping_end_time: str + :ivar run_as_account_id: Gets or sets the run as account ID of the machine. + :vartype run_as_account_id: str + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar errors: Errors for machine. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar application_discovery: Gets the data related to application discovery. + :vartype application_discovery: ~azure.mgmt.migrate.models.ApplicationDiscovery + :ivar dependency_map_discovery: Gets the data related to dependency map discovery. + :vartype dependency_map_discovery: ~azure.mgmt.migrate.models.DependencyMapDiscovery + :ivar static_discovery: Gets the data related to static discovery. + :vartype static_discovery: ~azure.mgmt.migrate.models.StaticDiscovery + :ivar sql_discovery: Gets the data related to SQL discovery. + :vartype sql_discovery: ~azure.mgmt.migrate.models.SqlDiscovery + :ivar web_app_discovery: Gets the data related to Web application discovery. + :vartype web_app_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar oracle_discovery: Gets the data related to Oracle discovery. + :vartype oracle_discovery: ~azure.mgmt.migrate.models.OracleDiscovery + :ivar spring_boot_discovery: Gets the data related to SpringBoot discovery. + :vartype spring_boot_discovery: ~azure.mgmt.migrate.models.SpringBootDiscovery + :ivar iis_discovery: Gets the data related to iis discovery. + :vartype iis_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar tomcat_discovery: Gets the data related to tomcat discovery. + :vartype tomcat_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar apps_and_roles: Apps And Roles of the Server. + :vartype apps_and_roles: ~azure.mgmt.migrate.models.AppsAndRoles + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :vartype number_of_processor_core: int + :ivar allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :vartype allocated_memory_in_mb: float + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :ivar bios_serial_number: Gets or sets the Machine BIOS serial number. + :vartype bios_serial_number: str + :ivar bios_guid: Gets or sets the BIOS GUID. + :vartype bios_guid: str + :ivar display_name: Gets the Display name of the machine. + :vartype display_name: str + :ivar is_deleted: Value indicating whether VM is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the marking machine creation. + :vartype created_timestamp: str + :ivar tags: Gets or sets tags on the resource. + :vartype tags: dict[str, str] + :ivar updated_timestamp: Gets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "instance_uuid": {"readonly": True}, + "host_fqdn": {"readonly": True}, + "host_id": {"readonly": True}, + "generation": {"readonly": True}, + "version": {"readonly": True}, + "high_availability": {"readonly": True}, + "cluster_fqdn": {"readonly": True}, + "cluster_id": {"readonly": True}, + "max_memory_mb": {"readonly": True}, + "is_dynamic_memory_enabled": {"readonly": True}, + "disks": {"readonly": True}, + "network_adapters": {"readonly": True}, + "management_server_type": {"readonly": True}, + "secure_boot_template_id": {"readonly": True}, + "secure_boot_enabled": {"readonly": True}, + "secure_boot_template": {"readonly": True}, + "tpm_enabled": {"readonly": True}, + "ksd_enabled": {"readonly": True}, + "shielding_requested": {"readonly": True}, + "data_protection_requested": {"readonly": True}, + "encrypt_state_and_vm_migration_traffic": {"readonly": True}, + "virtualization_based_security_opt_out": {"readonly": True}, + "power_status": {"readonly": True}, + "vm_fqdn": {"readonly": True}, + "vm_configuration_file_location": {"readonly": True}, + "guest_os_details": {"readonly": True}, + "number_of_applications": {"readonly": True}, + "guest_details_discovery_timestamp": {"readonly": True}, + "is_guest_details_discovery_in_progress": {"readonly": True}, + "dependency_mapping": {"readonly": True}, + "dependency_mapping_start_time": {"readonly": True}, + "dependency_mapping_end_time": {"readonly": True}, + "appliance_names": {"readonly": True}, + "errors": {"readonly": True}, + "application_discovery": {"readonly": True}, + "dependency_map_discovery": {"readonly": True}, + "static_discovery": {"readonly": True}, + "sql_discovery": {"readonly": True}, + "web_app_discovery": {"readonly": True}, + "oracle_discovery": {"readonly": True}, + "spring_boot_discovery": {"readonly": True}, + "iis_discovery": {"readonly": True}, + "tomcat_discovery": {"readonly": True}, + "apps_and_roles": {"readonly": True}, + "display_name": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "instance_uuid": {"key": "instanceUuid", "type": "str"}, + "host_fqdn": {"key": "hostFqdn", "type": "str"}, + "host_id": {"key": "hostId", "type": "str"}, + "generation": {"key": "generation", "type": "int"}, + "version": {"key": "version", "type": "str"}, + "high_availability": {"key": "highAvailability", "type": "str"}, + "cluster_fqdn": {"key": "clusterFqdn", "type": "str"}, + "cluster_id": {"key": "clusterId", "type": "str"}, + "max_memory_mb": {"key": "maxMemoryMb", "type": "int"}, + "is_dynamic_memory_enabled": {"key": "isDynamicMemoryEnabled", "type": "bool"}, + "disks": {"key": "disks", "type": "[HypervDisk]"}, + "network_adapters": {"key": "networkAdapters", "type": "[HypervNetworkAdapter]"}, + "management_server_type": {"key": "managementServerType", "type": "str"}, + "secure_boot_template_id": {"key": "secureBootTemplateId", "type": "str"}, + "secure_boot_enabled": {"key": "secureBootEnabled", "type": "bool"}, + "secure_boot_template": {"key": "secureBootTemplate", "type": "str"}, + "tpm_enabled": {"key": "tpmEnabled", "type": "bool"}, + "ksd_enabled": {"key": "ksdEnabled", "type": "bool"}, + "shielding_requested": {"key": "shieldingRequested", "type": "bool"}, + "data_protection_requested": {"key": "dataProtectionRequested", "type": "bool"}, + "encrypt_state_and_vm_migration_traffic": {"key": "encryptStateAndVmMigrationTraffic", "type": "bool"}, + "virtualization_based_security_opt_out": {"key": "virtualizationBasedSecurityOptOut", "type": "bool"}, + "power_status": {"key": "powerStatus", "type": "str"}, + "vm_fqdn": {"key": "vmFqdn", "type": "str"}, + "vm_configuration_file_location": {"key": "vmConfigurationFileLocation", "type": "str"}, + "firmware": {"key": "firmware", "type": "str"}, + "guest_os_details": {"key": "guestOsDetails", "type": "GuestOsDetails"}, + "number_of_applications": {"key": "numberOfApplications", "type": "int"}, + "guest_details_discovery_timestamp": {"key": "guestDetailsDiscoveryTimestamp", "type": "iso-8601"}, + "is_guest_details_discovery_in_progress": {"key": "isGuestDetailsDiscoveryInProgress", "type": "bool"}, + "dependency_mapping": {"key": "dependencyMapping", "type": "str"}, + "dependency_mapping_start_time": {"key": "dependencyMappingStartTime", "type": "iso-8601"}, + "dependency_mapping_end_time": {"key": "dependencyMappingEndTime", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "application_discovery": {"key": "applicationDiscovery", "type": "ApplicationDiscovery"}, + "dependency_map_discovery": {"key": "dependencyMapDiscovery", "type": "DependencyMapDiscovery"}, + "static_discovery": {"key": "staticDiscovery", "type": "StaticDiscovery"}, + "sql_discovery": {"key": "sqlDiscovery", "type": "SqlDiscovery"}, + "web_app_discovery": {"key": "webAppDiscovery", "type": "WebAppDiscovery"}, + "oracle_discovery": {"key": "oracleDiscovery", "type": "OracleDiscovery"}, + "spring_boot_discovery": {"key": "springBootDiscovery", "type": "SpringBootDiscovery"}, + "iis_discovery": {"key": "iisDiscovery", "type": "WebAppDiscovery"}, + "tomcat_discovery": {"key": "tomcatDiscovery", "type": "WebAppDiscovery"}, + "apps_and_roles": {"key": "appsAndRoles", "type": "AppsAndRoles"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "number_of_processor_core": {"key": "numberOfProcessorCore", "type": "int"}, + "allocated_memory_in_mb": {"key": "allocatedMemoryInMb", "type": "float"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "OperatingSystem"}, + "bios_serial_number": {"key": "biosSerialNumber", "type": "str"}, + "bios_guid": {"key": "biosGuid", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( # pylint: disable=too-many-locals + self, + *, + firmware: Optional[str] = None, + run_as_account_id: Optional[str] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + number_of_processor_core: Optional[int] = None, + allocated_memory_in_mb: Optional[float] = None, + operating_system_details: Optional["_models.OperatingSystem"] = None, + bios_serial_number: Optional[str] = None, + bios_guid: Optional[str] = None, + tags: Optional[Dict[str, str]] = None, + **kwargs: Any + ) -> None: + """ + :keyword firmware: Gets or sets the firmware. + :paramtype firmware: str + :keyword run_as_account_id: Gets or sets the run as account ID of the machine. + :paramtype run_as_account_id: str + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :paramtype number_of_processor_core: int + :keyword allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :paramtype allocated_memory_in_mb: float + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :keyword bios_serial_number: Gets or sets the Machine BIOS serial number. + :paramtype bios_serial_number: str + :keyword bios_guid: Gets or sets the BIOS GUID. + :paramtype bios_guid: str + :keyword tags: Gets or sets tags on the resource. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.instance_uuid = None + self.host_fqdn = None + self.host_id = None + self.generation = None + self.version = None + self.high_availability = None + self.cluster_fqdn = None + self.cluster_id = None + self.max_memory_mb = None + self.is_dynamic_memory_enabled = None + self.disks = None + self.network_adapters = None + self.management_server_type = None + self.secure_boot_template_id = None + self.secure_boot_enabled = None + self.secure_boot_template = None + self.tpm_enabled = None + self.ksd_enabled = None + self.shielding_requested = None + self.data_protection_requested = None + self.encrypt_state_and_vm_migration_traffic = None + self.virtualization_based_security_opt_out = None + self.power_status = None + self.vm_fqdn = None + self.vm_configuration_file_location = None + self.firmware = firmware + self.guest_os_details = None + self.number_of_applications = None + self.guest_details_discovery_timestamp = None + self.is_guest_details_discovery_in_progress = None + self.dependency_mapping = None + self.dependency_mapping_start_time = None + self.dependency_mapping_end_time = None + self.run_as_account_id = run_as_account_id + self.appliance_names = None + self.errors = None + self.application_discovery = None + self.dependency_map_discovery = None + self.static_discovery = None + self.sql_discovery = None + self.web_app_discovery = None + self.oracle_discovery = None + self.spring_boot_discovery = None + self.iis_discovery = None + self.tomcat_discovery = None + self.apps_and_roles = None + self.product_support_status = product_support_status + self.number_of_processor_core = number_of_processor_core + self.allocated_memory_in_mb = allocated_memory_in_mb + self.operating_system_details = operating_system_details + self.bios_serial_number = bios_serial_number + self.bios_guid = bios_guid + self.display_name = None + self.is_deleted = None + self.created_timestamp = None + self.tags = tags + self.updated_timestamp = None + self.provisioning_state = None + + +class HypervMachineUpdate(_serialization.Model): + """The type used for update operations of the HypervMachine. + + :ivar properties: The updatable properties of the HypervMachine. + :vartype properties: ~azure.mgmt.migrate.models.HypervMachineUpdateProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "HypervMachineUpdateProperties"}, + } + + def __init__(self, *, properties: Optional["_models.HypervMachineUpdateProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The updatable properties of the HypervMachine. + :paramtype properties: ~azure.mgmt.migrate.models.HypervMachineUpdateProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervMachineUpdateProperties(_serialization.Model): + """The updatable properties of the HypervMachine. + + :ivar firmware: Gets or sets the firmware. + :vartype firmware: str + :ivar run_as_account_id: Gets or sets the run as account ID of the machine. + :vartype run_as_account_id: str + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :vartype number_of_processor_core: int + :ivar allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :vartype allocated_memory_in_mb: float + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :ivar bios_serial_number: Gets or sets the Machine BIOS serial number. + :vartype bios_serial_number: str + :ivar bios_guid: Gets or sets the BIOS GUID. + :vartype bios_guid: str + :ivar tags: Gets or sets tags on the resource. + :vartype tags: dict[str, str] + """ + + _attribute_map = { + "firmware": {"key": "firmware", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "number_of_processor_core": {"key": "numberOfProcessorCore", "type": "int"}, + "allocated_memory_in_mb": {"key": "allocatedMemoryInMb", "type": "float"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "OperatingSystem"}, + "bios_serial_number": {"key": "biosSerialNumber", "type": "str"}, + "bios_guid": {"key": "biosGuid", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + } + + def __init__( + self, + *, + firmware: Optional[str] = None, + run_as_account_id: Optional[str] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + number_of_processor_core: Optional[int] = None, + allocated_memory_in_mb: Optional[float] = None, + operating_system_details: Optional["_models.OperatingSystem"] = None, + bios_serial_number: Optional[str] = None, + bios_guid: Optional[str] = None, + tags: Optional[Dict[str, str]] = None, + **kwargs: Any + ) -> None: + """ + :keyword firmware: Gets or sets the firmware. + :paramtype firmware: str + :keyword run_as_account_id: Gets or sets the run as account ID of the machine. + :paramtype run_as_account_id: str + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :paramtype number_of_processor_core: int + :keyword allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :paramtype allocated_memory_in_mb: float + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :keyword bios_serial_number: Gets or sets the Machine BIOS serial number. + :paramtype bios_serial_number: str + :keyword bios_guid: Gets or sets the BIOS GUID. + :paramtype bios_guid: str + :keyword tags: Gets or sets tags on the resource. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.firmware = firmware + self.run_as_account_id = run_as_account_id + self.product_support_status = product_support_status + self.number_of_processor_core = number_of_processor_core + self.allocated_memory_in_mb = allocated_memory_in_mb + self.operating_system_details = operating_system_details + self.bios_serial_number = bios_serial_number + self.bios_guid = bios_guid + self.tags = tags + + +class HypervNetworkAdapter(_serialization.Model): + """Second level object represented in responses as part of Machine REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar network_id: Network Id. + :vartype network_id: str + :ivar subnet_name: Name of the VM subnet within the virtual network the NIC is attached to. + :vartype subnet_name: str + :ivar static_ip_address: Static IP address. + :vartype static_ip_address: str + :ivar nic_type: Mac address of the NIC. + :vartype nic_type: str + :ivar nic_id: Gets or sets the NIC Id. + :vartype nic_id: str + :ivar mac_address: Gets or sets Mac address of the NIC. + :vartype mac_address: str + :ivar ip_address_list: Gets IP addresses for the machine. + :vartype ip_address_list: list[str] + :ivar network_name: Gets or sets Network Name. + :vartype network_name: str + :ivar ip_address_type: Gets or sets Type of the IP address. + :vartype ip_address_type: str + """ + + _validation = { + "network_id": {"readonly": True}, + "subnet_name": {"readonly": True}, + "static_ip_address": {"readonly": True}, + "nic_type": {"readonly": True}, + "ip_address_list": {"readonly": True}, + } + + _attribute_map = { + "network_id": {"key": "networkId", "type": "str"}, + "subnet_name": {"key": "subnetName", "type": "str"}, + "static_ip_address": {"key": "staticIpAddress", "type": "str"}, + "nic_type": {"key": "nicType", "type": "str"}, + "nic_id": {"key": "nicId", "type": "str"}, + "mac_address": {"key": "macAddress", "type": "str"}, + "ip_address_list": {"key": "ipAddressList", "type": "[str]"}, + "network_name": {"key": "networkName", "type": "str"}, + "ip_address_type": {"key": "ipAddressType", "type": "str"}, + } + + def __init__( + self, + *, + nic_id: Optional[str] = None, + mac_address: Optional[str] = None, + network_name: Optional[str] = None, + ip_address_type: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword nic_id: Gets or sets the NIC Id. + :paramtype nic_id: str + :keyword mac_address: Gets or sets Mac address of the NIC. + :paramtype mac_address: str + :keyword network_name: Gets or sets Network Name. + :paramtype network_name: str + :keyword ip_address_type: Gets or sets Type of the IP address. + :paramtype ip_address_type: str + """ + super().__init__(**kwargs) + self.network_id = None + self.subnet_name = None + self.static_ip_address = None + self.nic_type = None + self.nic_id = nic_id + self.mac_address = mac_address + self.ip_address_list = None + self.network_name = network_name + self.ip_address_type = ip_address_type + + +class HypervOperationsStatusResource(ProxyResource): + """A cluster resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.HypervClusterProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "HypervClusterProperties"}, + } + + def __init__(self, *, properties: Optional["_models.HypervClusterProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.HypervClusterProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervRunAsAccountResource(ProxyResource): + """A machine resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "RunAsAccountProperties"}, + } + + def __init__(self, *, properties: Optional["_models.RunAsAccountProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervRunAsAccountResourceListResult(_serialization.Model): + """The response of a HypervRunAsAccountResource list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The HypervRunAsAccountResource items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.HypervRunAsAccountResource] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[HypervRunAsAccountResource]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.HypervRunAsAccountResource"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The HypervRunAsAccountResource items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.HypervRunAsAccountResource] + :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 TrackedResource(Resource): + """The resource model definition for an Azure Resource Manager tracked top level resource which + has 'tags' and a 'location'. + + 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: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + "location": {"required": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "tags": {"key": "tags", "type": "{str}"}, + "location": {"key": "location", "type": "str"}, + } + + def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kwargs: Any) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword location: The geo-location where the resource lives. Required. + :paramtype location: str + """ + super().__init__(**kwargs) + self.tags = tags + self.location = location + + +class HypervSite(TrackedResource): + """A HyperV SiteResource. + + 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: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.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: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SiteProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + "location": {"required": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "tags": {"key": "tags", "type": "{str}"}, + "location": {"key": "location", "type": "str"}, + "properties": {"key": "properties", "type": "SiteProperties"}, + } + + def __init__( + self, + *, + location: str, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.SiteProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword location: The geo-location where the resource lives. Required. + :paramtype location: str + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SiteProperties + """ + super().__init__(tags=tags, location=location, **kwargs) + self.properties = properties + + +class HypervSiteListResult(_serialization.Model): + """The response of a HypervSite list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The HypervSite items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.HypervSite] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[HypervSite]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.HypervSite"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The HypervSite items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.HypervSite] + :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 HypervSiteUpdate(_serialization.Model): + """The type used for update operations of the HypervSite. + + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar properties: The updatable properties of the HypervSite. + :vartype properties: ~azure.mgmt.migrate.models.HypervSiteUpdateProperties + """ + + _attribute_map = { + "tags": {"key": "tags", "type": "{str}"}, + "properties": {"key": "properties", "type": "HypervSiteUpdateProperties"}, + } + + def __init__( + self, + *, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.HypervSiteUpdateProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword properties: The updatable properties of the HypervSite. + :paramtype properties: ~azure.mgmt.migrate.models.HypervSiteUpdateProperties + """ + super().__init__(**kwargs) + self.tags = tags + self.properties = properties + + +class HypervSiteUpdateProperties(_serialization.Model): + """The updatable properties of the HypervSite. + + :ivar service_principal_identity_details: Gets or sets the service principal identity details + used by agent for + communication + to the service. + :vartype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :ivar agent_details: Gets or sets the on-premises agent details. + :vartype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + :ivar discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :vartype discovery_solution_id: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _attribute_map = { + "service_principal_identity_details": {"key": "servicePrincipalIdentityDetails", "type": "SiteSpnProperties"}, + "agent_details": {"key": "agentDetails", "type": "SiteAgentProperties"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "discovery_solution_id": {"key": "discoverySolutionId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + service_principal_identity_details: Optional["_models.SiteSpnProperties"] = None, + agent_details: Optional["_models.SiteAgentProperties"] = None, + appliance_name: Optional[str] = None, + discovery_solution_id: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword service_principal_identity_details: Gets or sets the service principal identity + details used by agent for + communication + to the service. + :paramtype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :keyword agent_details: Gets or sets the on-premises agent details. + :paramtype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + :keyword discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :paramtype discovery_solution_id: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.service_principal_identity_details = service_principal_identity_details + self.agent_details = agent_details + self.appliance_name = appliance_name + self.discovery_solution_id = discovery_solution_id + self.provisioning_state = provisioning_state + + +class HypervSiteUsage(_serialization.Model): + """Hyper-V site usage. + + :ivar machine_count: Gets or sets the number of machines discovered in the site. + :vartype machine_count: int + :ivar run_as_account_count: Gets or sets the number of run as accounts in the site. + :vartype run_as_account_count: int + :ivar host_count: Gets or sets the number of hosts part of the site. + :vartype host_count: int + :ivar cluster_count: Gets or sets the number of clusters part of the site. + :vartype cluster_count: int + """ + + _attribute_map = { + "machine_count": {"key": "machineCount", "type": "int"}, + "run_as_account_count": {"key": "runAsAccountCount", "type": "int"}, + "host_count": {"key": "hostCount", "type": "int"}, + "cluster_count": {"key": "clusterCount", "type": "int"}, + } + + def __init__( + self, + *, + machine_count: Optional[int] = None, + run_as_account_count: Optional[int] = None, + host_count: Optional[int] = None, + cluster_count: Optional[int] = None, + **kwargs: Any + ) -> None: + """ + :keyword machine_count: Gets or sets the number of machines discovered in the site. + :paramtype machine_count: int + :keyword run_as_account_count: Gets or sets the number of run as accounts in the site. + :paramtype run_as_account_count: int + :keyword host_count: Gets or sets the number of hosts part of the site. + :paramtype host_count: int + :keyword cluster_count: Gets or sets the number of clusters part of the site. + :paramtype cluster_count: int + """ + super().__init__(**kwargs) + self.machine_count = machine_count + self.run_as_account_count = run_as_account_count + self.host_count = host_count + self.cluster_count = cluster_count + + +class HypervVmSoftwareInventory(ProxyResource): + """HyperV VM software inventory REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.MachineSoftwareInventoryProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "MachineSoftwareInventoryProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.MachineSoftwareInventoryProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.MachineSoftwareInventoryProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class HypervVmSoftwareInventoryListResult(_serialization.Model): + """The response of a HypervVmSoftwareInventory list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The HypervVmSoftwareInventory items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.HypervVmSoftwareInventory] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[HypervVmSoftwareInventory]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.HypervVmSoftwareInventory"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The HypervVmSoftwareInventory items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.HypervVmSoftwareInventory] + :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 IisApplicationUnit(_serialization.Model): + """Second-level object for identification of application units in a web site + hosted on IIS. + + :ivar application_pool_name: Gets or sets the path of the directory. + :vartype application_pool_name: str + :ivar managed_pipeline_mode: Gets or sets the managed pipeline mode. + :vartype managed_pipeline_mode: str + :ivar runtime_version: Gets or sets the runtime version. + :vartype runtime_version: str + :ivar enable32_bit_api_on_win64: Gets or sets a value indicating whether 32 bit applications + are allowed to run + on 64 bit. + :vartype enable32_bit_api_on_win64: bool + :ivar path: Gets or sets the path of the directory. + :vartype path: ~azure.mgmt.migrate.models.DirectoryPath + :ivar directories: Gets or sets the list of directories. + :vartype directories: list[~azure.mgmt.migrate.models.DirectoryPath] + """ + + _attribute_map = { + "application_pool_name": {"key": "applicationPoolName", "type": "str"}, + "managed_pipeline_mode": {"key": "managedPipelineMode", "type": "str"}, + "runtime_version": {"key": "runtimeVersion", "type": "str"}, + "enable32_bit_api_on_win64": {"key": "enable32BitApiOnWin64", "type": "bool"}, + "path": {"key": "path", "type": "DirectoryPath"}, + "directories": {"key": "directories", "type": "[DirectoryPath]"}, + } + + def __init__( + self, + *, + application_pool_name: Optional[str] = None, + managed_pipeline_mode: Optional[str] = None, + runtime_version: Optional[str] = None, + enable32_bit_api_on_win64: Optional[bool] = None, + path: Optional["_models.DirectoryPath"] = None, + directories: Optional[List["_models.DirectoryPath"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword application_pool_name: Gets or sets the path of the directory. + :paramtype application_pool_name: str + :keyword managed_pipeline_mode: Gets or sets the managed pipeline mode. + :paramtype managed_pipeline_mode: str + :keyword runtime_version: Gets or sets the runtime version. + :paramtype runtime_version: str + :keyword enable32_bit_api_on_win64: Gets or sets a value indicating whether 32 bit applications + are allowed to run + on 64 bit. + :paramtype enable32_bit_api_on_win64: bool + :keyword path: Gets or sets the path of the directory. + :paramtype path: ~azure.mgmt.migrate.models.DirectoryPath + :keyword directories: Gets or sets the list of directories. + :paramtype directories: list[~azure.mgmt.migrate.models.DirectoryPath] + """ + super().__init__(**kwargs) + self.application_pool_name = application_pool_name + self.managed_pipeline_mode = managed_pipeline_mode + self.runtime_version = runtime_version + self.enable32_bit_api_on_win64 = enable32_bit_api_on_win64 + self.path = path + self.directories = directories + + +class IisVirtualApplicationUnit(_serialization.Model): + """Second-level object for identification of virtual applications in a web site + hosted on IIS. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar is_virtual_directory: Gets a value indicating whether the application corresponds to a + directory. + :vartype is_virtual_directory: bool + :ivar path: Gets or sets the path of the directory. + :vartype path: ~azure.mgmt.migrate.models.DirectoryPath + :ivar directories: Gets or sets the list of directories. + :vartype directories: list[~azure.mgmt.migrate.models.DirectoryPath] + """ + + _validation = { + "is_virtual_directory": {"readonly": True}, + } + + _attribute_map = { + "is_virtual_directory": {"key": "isVirtualDirectory", "type": "bool"}, + "path": {"key": "path", "type": "DirectoryPath"}, + "directories": {"key": "directories", "type": "[DirectoryPath]"}, + } + + def __init__( + self, + *, + path: Optional["_models.DirectoryPath"] = None, + directories: Optional[List["_models.DirectoryPath"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword path: Gets or sets the path of the directory. + :paramtype path: ~azure.mgmt.migrate.models.DirectoryPath + :keyword directories: Gets or sets the list of directories. + :paramtype directories: list[~azure.mgmt.migrate.models.DirectoryPath] + """ + super().__init__(**kwargs) + self.is_virtual_directory = None + self.path = path + self.directories = directories + + +class IisWebApplicationProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for web application properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar applications: Gets the list of application units for the web site. + :vartype applications: list[~azure.mgmt.migrate.models.IisApplicationUnit] + :ivar virtual_applications: Gets the list of application units for the web site. + :vartype virtual_applications: list[~azure.mgmt.migrate.models.IisVirtualApplicationUnit] + :ivar web_server_id: Gets the web server id. + :vartype web_server_id: str + :ivar web_server_name: Gets the web server name. + :vartype web_server_name: str + :ivar machine_arm_ids: Gets the list of machine ARM Ids on which the SQL server is deployed. + :vartype machine_arm_ids: list[str] + :ivar virtual_path: Gets the virtual path of the application. + :vartype virtual_path: str + :ivar physical_path: Gets the physical path of the application. + :vartype physical_path: str + :ivar bindings: Gets the front end bindings for the application. + :vartype bindings: list[~azure.mgmt.migrate.models.FrontEndBinding] + :ivar frameworks: Gets the Error details. + :vartype frameworks: list[~azure.mgmt.migrate.models.WebApplicationFramework] + :ivar configurations: Gets the configuration. + :vartype configurations: list[~azure.mgmt.migrate.models.WebApplicationConfigurationUnit] + :ivar directories: Gets the directories. + :vartype directories: list[~azure.mgmt.migrate.models.WebApplicationDirectoryUnit] + :ivar display_name: Gets the display name. + :vartype display_name: str + :ivar tags: Gets or sets tags on the resource. + :vartype tags: dict[str, str] + :ivar static_folders: Static folders. + :vartype static_folders: list[str] + :ivar machine_display_name: Machine display name. + :vartype machine_display_name: str + :ivar is_deleted: Gets a value indicating whether application is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the timestamp marking creation of the entity. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last update operation. + :vartype updated_timestamp: str + :ivar server_type: Gets the server type. + :vartype server_type: str + :ivar errors: Gets the Error details. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar has_errors: Gets a value indicating whether the WebApp has errors or not. + :vartype has_errors: bool + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "applications": {"readonly": True}, + "virtual_applications": {"readonly": True}, + "web_server_id": {"readonly": True}, + "web_server_name": {"readonly": True}, + "machine_arm_ids": {"readonly": True}, + "virtual_path": {"readonly": True}, + "physical_path": {"readonly": True}, + "bindings": {"readonly": True}, + "frameworks": {"readonly": True}, + "configurations": {"readonly": True}, + "directories": {"readonly": True}, + "display_name": {"readonly": True}, + "static_folders": {"readonly": True}, + "machine_display_name": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "server_type": {"readonly": True}, + "errors": {"readonly": True}, + "appliance_names": {"readonly": True}, + "has_errors": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "applications": {"key": "applications", "type": "[IisApplicationUnit]"}, + "virtual_applications": {"key": "virtualApplications", "type": "[IisVirtualApplicationUnit]"}, + "web_server_id": {"key": "webServerId", "type": "str"}, + "web_server_name": {"key": "webServerName", "type": "str"}, + "machine_arm_ids": {"key": "machineArmIds", "type": "[str]"}, + "virtual_path": {"key": "virtualPath", "type": "str"}, + "physical_path": {"key": "physicalPath", "type": "str"}, + "bindings": {"key": "bindings", "type": "[FrontEndBinding]"}, + "frameworks": {"key": "frameworks", "type": "[WebApplicationFramework]"}, + "configurations": {"key": "configurations", "type": "[WebApplicationConfigurationUnit]"}, + "directories": {"key": "directories", "type": "[WebApplicationDirectoryUnit]"}, + "display_name": {"key": "displayName", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "static_folders": {"key": "staticFolders", "type": "[str]"}, + "machine_display_name": {"key": "machineDisplayName", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "server_type": {"key": "serverType", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "has_errors": {"key": "hasErrors", "type": "bool"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, *, tags: Optional[Dict[str, str]] = None, **kwargs: Any) -> None: + """ + :keyword tags: Gets or sets tags on the resource. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.applications = None + self.virtual_applications = None + self.web_server_id = None + self.web_server_name = None + self.machine_arm_ids = None + self.virtual_path = None + self.physical_path = None + self.bindings = None + self.frameworks = None + self.configurations = None + self.directories = None + self.display_name = None + self.tags = tags + self.static_folders = None + self.machine_display_name = None + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.server_type = None + self.errors = None + self.appliance_names = None + self.has_errors = None + self.provisioning_state = None + + +class IisWebApplications(ProxyResource): + """Web application REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.IisWebApplicationProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "IisWebApplicationProperties"}, + } + + def __init__(self, *, properties: Optional["_models.IisWebApplicationProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.IisWebApplicationProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class IisWebApplicationsListResult(_serialization.Model): + """The response of a IisWebApplications list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The IisWebApplications items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.IisWebApplications] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[IisWebApplications]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.IisWebApplications"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The IisWebApplications items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.IisWebApplications] + :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 IisWebApplicationsUpdate(_serialization.Model): + """The type used for update operations of the IisWebApplications. + + :ivar properties: The updatable properties of the IisWebApplications. + :vartype properties: ~azure.mgmt.migrate.models.IisWebApplicationsUpdateProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "IisWebApplicationsUpdateProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.IisWebApplicationsUpdateProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The updatable properties of the IisWebApplications. + :paramtype properties: ~azure.mgmt.migrate.models.IisWebApplicationsUpdateProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class IisWebApplicationsUpdateProperties(_serialization.Model): + """The updatable properties of the IisWebApplications. + + :ivar tags: Gets or sets tags on the resource. + :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: Gets or sets tags on the resource. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.tags = tags + + +class IisWebServerProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for web server properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar configuration_location: Gets the configuration location. + :vartype configuration_location: str + :ivar version: Gets the configuration location. + :vartype version: str + :ivar machine_ids: Gets the list of machines. + :vartype machine_ids: list[str] + :ivar web_applications: Gets the list of web applications. + :vartype web_applications: list[str] + :ivar display_name: Gets the display name. + :vartype display_name: str + :ivar server_fqdn: Gets the server FQDN. + :vartype server_fqdn: str + :ivar run_as_account_id: Gets the run as account id. + :vartype run_as_account_id: str + :ivar is_deleted: Gets a value indicating whether application is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the timestamp marking creation of the entity. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last update operation. + :vartype updated_timestamp: str + :ivar server_type: Gets the server type. + :vartype server_type: str + :ivar errors: Gets the Error details. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar has_errors: Gets a value indicating whether the WebApp has errors or not. + :vartype has_errors: bool + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "configuration_location": {"readonly": True}, + "version": {"readonly": True}, + "machine_ids": {"readonly": True}, + "web_applications": {"readonly": True}, + "display_name": {"readonly": True}, + "server_fqdn": {"readonly": True}, + "run_as_account_id": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "server_type": {"readonly": True}, + "errors": {"readonly": True}, + "appliance_names": {"readonly": True}, + "has_errors": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "configuration_location": {"key": "configurationLocation", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "machine_ids": {"key": "machineIds", "type": "[str]"}, + "web_applications": {"key": "webApplications", "type": "[str]"}, + "display_name": {"key": "displayName", "type": "str"}, + "server_fqdn": {"key": "serverFqdn", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "server_type": {"key": "serverType", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "has_errors": {"key": "hasErrors", "type": "bool"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.configuration_location = None + self.version = None + self.machine_ids = None + self.web_applications = None + self.display_name = None + self.server_fqdn = None + self.run_as_account_id = None + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.server_type = None + self.errors = None + self.appliance_names = None + self.has_errors = None + self.provisioning_state = None + + +class IisWebServers(ProxyResource): + """Web server REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.IisWebServerProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "IisWebServerProperties"}, + } + + def __init__(self, *, properties: Optional["_models.IisWebServerProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.IisWebServerProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class IisWebServersListResult(_serialization.Model): + """The response of a IisWebServers list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The IisWebServers items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.IisWebServers] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[IisWebServers]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.IisWebServers"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The IisWebServers items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.IisWebServers] + :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 ImportJob(ProxyResource): + """A host resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.JobProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "JobProperties"}, + } + + def __init__(self, *, properties: Optional["_models.JobProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.JobProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ImportJobListResult(_serialization.Model): + """The response of a ImportJob list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The ImportJob items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.ImportJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ImportJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.ImportJob"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The ImportJob items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.ImportJob] + :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 ImportMachine(ProxyResource): + """An machine resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.ImportMachineProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "ImportMachineProperties"}, + } + + def __init__(self, *, properties: Optional["_models.ImportMachineProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.ImportMachineProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ImportMachineListResult(_serialization.Model): + """The response of a ImportMachine list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The ImportMachine items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.ImportMachine] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ImportMachine]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.ImportMachine"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The ImportMachine items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.ImportMachine] + :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 ImportMachineProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for machine properties. + + 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 firmware: Firmware of the machine. + :vartype firmware: str + :ivar percentage_cpu_utilization: CPU utilization. + :vartype percentage_cpu_utilization: float + :ivar percentage_memory_utilization: Memory utilization. + :vartype percentage_memory_utilization: float + :ivar number_of_disks: Number of disks. + :vartype number_of_disks: int + :ivar total_disk_read_operations_per_second: Total disk read operations per second. + :vartype total_disk_read_operations_per_second: float + :ivar total_disk_write_operations_per_second: Total disk write operations per second. + :vartype total_disk_write_operations_per_second: float + :ivar total_disk_write_throughput: Total disk write throughput. + :vartype total_disk_write_throughput: float + :ivar total_disk_read_throughput: Total disk read throughput. + :vartype total_disk_read_throughput: float + :ivar mac_address: MAC Address. + :vartype mac_address: str + :ivar ip_addresses: IP Addresses. + :vartype ip_addresses: list[str] + :ivar machine_id: Machine ID. + :vartype machine_id: str + :ivar machine_manager_id: Machine manager ID. + :vartype machine_manager_id: str + :ivar number_of_network_adapters: Number of network adapters. + :vartype number_of_network_adapters: int + :ivar network_in_throughput: Network in throughput. + :vartype network_in_throughput: float + :ivar network_out_throughput: Network out throughput. + :vartype network_out_throughput: float + :ivar server_type: Server type. + :vartype server_type: str + :ivar hypervisor: Fabric type. + :vartype hypervisor: str + :ivar hypervisor_version_number: Hypervisor version number. + :vartype hypervisor_version_number: str + :ivar disks: Disks attached to the machine. + :vartype disks: list[~azure.mgmt.migrate.models.WebRoleImportDisk] + :ivar vm_fqdn: Server FQDN. + :vartype vm_fqdn: str + :ivar storage_in_use_gb: Total storage in use. + :vartype storage_in_use_gb: float + :ivar number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :vartype number_of_processor_core: int + :ivar allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :vartype allocated_memory_in_mb: float + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.WebRoleOperatingSystem + :ivar bios_serial_number: Gets or sets the Machine BIOS serial number. + :vartype bios_serial_number: str + :ivar bios_guid: Gets or sets the BIOS GUID. + :vartype bios_guid: str + :ivar display_name: Gets the Display name of the machine. + :vartype display_name: str + :ivar is_deleted: Value indicating whether VM is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the marking machine creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar tags: machine tags. Required. + :vartype tags: dict[str, str] + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "firmware": {"readonly": True}, + "percentage_cpu_utilization": {"readonly": True}, + "percentage_memory_utilization": {"readonly": True}, + "number_of_disks": {"readonly": True}, + "total_disk_read_operations_per_second": {"readonly": True}, + "total_disk_write_operations_per_second": {"readonly": True}, + "total_disk_write_throughput": {"readonly": True}, + "total_disk_read_throughput": {"readonly": True}, + "mac_address": {"readonly": True}, + "ip_addresses": {"readonly": True}, + "machine_id": {"readonly": True}, + "machine_manager_id": {"readonly": True}, + "number_of_network_adapters": {"readonly": True}, + "network_in_throughput": {"readonly": True}, + "network_out_throughput": {"readonly": True}, + "server_type": {"readonly": True}, + "hypervisor": {"readonly": True}, + "hypervisor_version_number": {"readonly": True}, + "disks": {"readonly": True}, + "vm_fqdn": {"readonly": True}, + "storage_in_use_gb": {"readonly": True}, + "display_name": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "tags": {"required": True}, + } + + _attribute_map = { + "firmware": {"key": "firmware", "type": "str"}, + "percentage_cpu_utilization": {"key": "percentageCpuUtilization", "type": "float"}, + "percentage_memory_utilization": {"key": "percentageMemoryUtilization", "type": "float"}, + "number_of_disks": {"key": "numberOfDisks", "type": "int"}, + "total_disk_read_operations_per_second": {"key": "totalDiskReadOperationsPerSecond", "type": "float"}, + "total_disk_write_operations_per_second": {"key": "totalDiskWriteOperationsPerSecond", "type": "float"}, + "total_disk_write_throughput": {"key": "totalDiskWriteThroughput", "type": "float"}, + "total_disk_read_throughput": {"key": "totalDiskReadThroughput", "type": "float"}, + "mac_address": {"key": "macAddress", "type": "str"}, + "ip_addresses": {"key": "ipAddresses", "type": "[str]"}, + "machine_id": {"key": "machineId", "type": "str"}, + "machine_manager_id": {"key": "machineManagerId", "type": "str"}, + "number_of_network_adapters": {"key": "numberOfNetworkAdapters", "type": "int"}, + "network_in_throughput": {"key": "networkInThroughput", "type": "float"}, + "network_out_throughput": {"key": "networkOutThroughput", "type": "float"}, + "server_type": {"key": "serverType", "type": "str"}, + "hypervisor": {"key": "hypervisor", "type": "str"}, + "hypervisor_version_number": {"key": "hypervisorVersionNumber", "type": "str"}, + "disks": {"key": "disks", "type": "[WebRoleImportDisk]"}, + "vm_fqdn": {"key": "vmFqdn", "type": "str"}, + "storage_in_use_gb": {"key": "storageInUseGb", "type": "float"}, + "number_of_processor_core": {"key": "numberOfProcessorCore", "type": "int"}, + "allocated_memory_in_mb": {"key": "allocatedMemoryInMb", "type": "float"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "WebRoleOperatingSystem"}, + "bios_serial_number": {"key": "biosSerialNumber", "type": "str"}, + "bios_guid": {"key": "biosGuid", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( # pylint: disable=too-many-locals + self, + *, + tags: Dict[str, str], + number_of_processor_core: Optional[int] = None, + allocated_memory_in_mb: Optional[float] = None, + operating_system_details: Optional["_models.WebRoleOperatingSystem"] = None, + bios_serial_number: Optional[str] = None, + bios_guid: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :paramtype number_of_processor_core: int + :keyword allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :paramtype allocated_memory_in_mb: float + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.WebRoleOperatingSystem + :keyword bios_serial_number: Gets or sets the Machine BIOS serial number. + :paramtype bios_serial_number: str + :keyword bios_guid: Gets or sets the BIOS GUID. + :paramtype bios_guid: str + :keyword tags: machine tags. Required. + :paramtype tags: dict[str, str] + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.firmware = None + self.percentage_cpu_utilization = None + self.percentage_memory_utilization = None + self.number_of_disks = None + self.total_disk_read_operations_per_second = None + self.total_disk_write_operations_per_second = None + self.total_disk_write_throughput = None + self.total_disk_read_throughput = None + self.mac_address = None + self.ip_addresses = None + self.machine_id = None + self.machine_manager_id = None + self.number_of_network_adapters = None + self.network_in_throughput = None + self.network_out_throughput = None + self.server_type = None + self.hypervisor = None + self.hypervisor_version_number = None + self.disks = None + self.vm_fqdn = None + self.storage_in_use_gb = None + self.number_of_processor_core = number_of_processor_core + self.allocated_memory_in_mb = allocated_memory_in_mb + self.operating_system_details = operating_system_details + self.bios_serial_number = bios_serial_number + self.bios_guid = bios_guid + self.display_name = None + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.tags = tags + self.provisioning_state = provisioning_state + + +class ImportMachinesJob(_serialization.Model): + """Import machines Job REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar type: Handled by resource provider. Type = + Microsoft.OffAzure/ImportSites/jobs/importJobs. + :vartype type: str + :ivar properties: Gets or sets the import job properties. + :vartype properties: ~azure.mgmt.migrate.models.ImportMachinesJobProperties + :ivar id: Gets or sets the relative ARM name to get job. + :vartype id: str + :ivar name: Gets or sets the Job ID. + :vartype name: str + :ivar status: Gets or sets the Job status. + :vartype status: str + :ivar start_time: Gets or sets the Job start time. + :vartype start_time: str + :ivar end_time: Gets or sets the Job end time. + :vartype end_time: str + :ivar display_name: Gets or sets the Display name. + :vartype display_name: str + """ + + _validation = { + "type": {"readonly": True}, + } + + _attribute_map = { + "type": {"key": "type", "type": "str"}, + "properties": {"key": "properties", "type": "ImportMachinesJobProperties"}, + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "start_time": {"key": "startTime", "type": "str"}, + "end_time": {"key": "endTime", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + } + + def __init__( + self, + *, + properties: Optional["_models.ImportMachinesJobProperties"] = None, + id: Optional[str] = None, # pylint: disable=redefined-builtin + name: Optional[str] = None, + status: Optional[str] = None, + start_time: Optional[str] = None, + end_time: Optional[str] = None, + display_name: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword properties: Gets or sets the import job properties. + :paramtype properties: ~azure.mgmt.migrate.models.ImportMachinesJobProperties + :keyword id: Gets or sets the relative ARM name to get job. + :paramtype id: str + :keyword name: Gets or sets the Job ID. + :paramtype name: str + :keyword status: Gets or sets the Job status. + :paramtype status: str + :keyword start_time: Gets or sets the Job start time. + :paramtype start_time: str + :keyword end_time: Gets or sets the Job end time. + :paramtype end_time: str + :keyword display_name: Gets or sets the Display name. + :paramtype display_name: str + """ + super().__init__(**kwargs) + self.type = None + self.properties = properties + self.id = id + self.name = name + self.status = status + self.start_time = start_time + self.end_time = end_time + self.display_name = display_name + + +class ImportMachinesJobProperties(_serialization.Model): + """ImportMachines JobProperties. + + :ivar blob_name: blob name. + :vartype blob_name: str + :ivar blob_sas_uri: blob sasUri. + :vartype blob_sas_uri: str + :ivar job_result: job Result. Known values are: "Unknown", "Completed", + "CompletedWithWarnings", "CompletedWithErrors", "Failed", "WaitingForBlobUpload", and + "InProgress". + :vartype job_result: str or ~azure.mgmt.migrate.models.JobResult + :ivar number_of_machines_imported: number Of Machines Imported. + :vartype number_of_machines_imported: int + :ivar blob_creation_time_stamp: blob Creation TimeStamp. + :vartype blob_creation_time_stamp: ~datetime.datetime + :ivar error_summary: error summary. + :vartype error_summary: ~azure.mgmt.migrate.models.JobErrorSummary + """ + + _attribute_map = { + "blob_name": {"key": "blobName", "type": "str"}, + "blob_sas_uri": {"key": "blobSasUri", "type": "str"}, + "job_result": {"key": "jobResult", "type": "str"}, + "number_of_machines_imported": {"key": "numberOfMachinesImported", "type": "int"}, + "blob_creation_time_stamp": {"key": "blobCreationTimeStamp", "type": "iso-8601"}, + "error_summary": {"key": "errorSummary", "type": "JobErrorSummary"}, + } + + def __init__( + self, + *, + blob_name: Optional[str] = None, + blob_sas_uri: Optional[str] = None, + job_result: Optional[Union[str, "_models.JobResult"]] = None, + number_of_machines_imported: Optional[int] = None, + blob_creation_time_stamp: Optional[datetime.datetime] = None, + error_summary: Optional["_models.JobErrorSummary"] = None, + **kwargs: Any + ) -> None: + """ + :keyword blob_name: blob name. + :paramtype blob_name: str + :keyword blob_sas_uri: blob sasUri. + :paramtype blob_sas_uri: str + :keyword job_result: job Result. Known values are: "Unknown", "Completed", + "CompletedWithWarnings", "CompletedWithErrors", "Failed", "WaitingForBlobUpload", and + "InProgress". + :paramtype job_result: str or ~azure.mgmt.migrate.models.JobResult + :keyword number_of_machines_imported: number Of Machines Imported. + :paramtype number_of_machines_imported: int + :keyword blob_creation_time_stamp: blob Creation TimeStamp. + :paramtype blob_creation_time_stamp: ~datetime.datetime + :keyword error_summary: error summary. + :paramtype error_summary: ~azure.mgmt.migrate.models.JobErrorSummary + """ + super().__init__(**kwargs) + self.blob_name = blob_name + self.blob_sas_uri = blob_sas_uri + self.job_result = job_result + self.number_of_machines_imported = number_of_machines_imported + self.blob_creation_time_stamp = blob_creation_time_stamp + self.error_summary = error_summary + + +class ImportSite(TrackedResource): + """A ImportSite. + + 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: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.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: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.ImportSiteProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + "location": {"required": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "tags": {"key": "tags", "type": "{str}"}, + "location": {"key": "location", "type": "str"}, + "properties": {"key": "properties", "type": "ImportSiteProperties"}, + } + + def __init__( + self, + *, + location: str, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.ImportSiteProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword location: The geo-location where the resource lives. Required. + :paramtype location: str + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.ImportSiteProperties + """ + super().__init__(tags=tags, location=location, **kwargs) + self.properties = properties + + +class ImportSiteListResult(_serialization.Model): + """The response of a ImportSite list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The ImportSite items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.ImportSite] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ImportSite]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.ImportSite"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The ImportSite items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.ImportSite] + :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 ImportSiteProperties(_serialization.Model): + """The properties of ImportSiteResource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :vartype discovery_solution_id: str + :ivar master_site_id: Gets the Master Site this site is linked to. + :vartype master_site_id: str + :ivar service_endpoint: Gets the service endpoint. + :vartype service_endpoint: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "master_site_id": {"readonly": True}, + "service_endpoint": {"readonly": True}, + } + + _attribute_map = { + "discovery_solution_id": {"key": "discoverySolutionId", "type": "str"}, + "master_site_id": {"key": "masterSiteId", "type": "str"}, + "service_endpoint": {"key": "serviceEndpoint", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + discovery_solution_id: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :paramtype discovery_solution_id: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.discovery_solution_id = discovery_solution_id + self.master_site_id = None + self.service_endpoint = None + self.provisioning_state = provisioning_state + + +class ImportSiteUpdate(_serialization.Model): + """The type used for update operations of the ImportSite. + + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar properties: The updatable properties of the ImportSite. + :vartype properties: ~azure.mgmt.migrate.models.ImportSiteUpdateProperties + """ + + _attribute_map = { + "tags": {"key": "tags", "type": "{str}"}, + "properties": {"key": "properties", "type": "ImportSiteUpdateProperties"}, + } + + def __init__( + self, + *, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.ImportSiteUpdateProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword properties: The updatable properties of the ImportSite. + :paramtype properties: ~azure.mgmt.migrate.models.ImportSiteUpdateProperties + """ + super().__init__(**kwargs) + self.tags = tags + self.properties = properties + + +class ImportSiteUpdateProperties(_serialization.Model): + """The updatable properties of the ImportSite. + + :ivar discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :vartype discovery_solution_id: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _attribute_map = { + "discovery_solution_id": {"key": "discoverySolutionId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + discovery_solution_id: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :paramtype discovery_solution_id: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.discovery_solution_id = discovery_solution_id + self.provisioning_state = provisioning_state + + +class JobErrorSummary(_serialization.Model): + """Job Error Summary. + + :ivar errors: errors list. + :vartype errors: list[str] + :ivar error_count: error count. + :vartype error_count: int + :ivar warning_count: warning Count. + :vartype warning_count: int + """ + + _attribute_map = { + "errors": {"key": "errors", "type": "[str]"}, + "error_count": {"key": "errorCount", "type": "int"}, + "warning_count": {"key": "warningCount", "type": "int"}, + } + + def __init__( + self, + *, + errors: Optional[List[str]] = None, + error_count: Optional[int] = None, + warning_count: Optional[int] = None, + **kwargs: Any + ) -> None: + """ + :keyword errors: errors list. + :paramtype errors: list[str] + :keyword error_count: error count. + :paramtype error_count: int + :keyword warning_count: warning Count. + :paramtype warning_count: int + """ + super().__init__(**kwargs) + self.errors = errors + self.error_count = error_count + self.warning_count = warning_count + + +class JobProperties(_serialization.Model): + """Job REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar status: Gets operation status. + :vartype status: str + :ivar start_time: Gets operation start time. + :vartype start_time: str + :ivar end_time: Gets operation end time. + :vartype end_time: str + :ivar display_name: Gets or sets the display name of the Job. + :vartype display_name: str + :ivar client_request_id: Gets the client request Id used in the operation execution context. + :vartype client_request_id: str + :ivar activity_id: Gets the activity Id used in the operation execution context. + :vartype activity_id: str + :ivar errors: Gets the errors. + :vartype errors: list[~azure.mgmt.migrate.models.ErrorDetails] + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "status": {"readonly": True}, + "start_time": {"readonly": True}, + "end_time": {"readonly": True}, + "display_name": {"readonly": True}, + "client_request_id": {"readonly": True}, + "activity_id": {"readonly": True}, + "errors": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "status": {"key": "status", "type": "str"}, + "start_time": {"key": "startTime", "type": "str"}, + "end_time": {"key": "endTime", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + "client_request_id": {"key": "clientRequestId", "type": "str"}, + "activity_id": {"key": "activityId", "type": "str"}, + "errors": {"key": "errors", "type": "[ErrorDetails]"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.status = None + self.start_time = None + self.end_time = None + self.display_name = None + self.client_request_id = None + self.activity_id = None + self.errors = None + self.provisioning_state = None + + +class MachineMetadata(_serialization.Model): + """The properties of Machine MetaData. + + All required parameters must be populated in order to send to server. + + :ivar machine_arm_id: arm id of the machine. Required. + :vartype machine_arm_id: str + :ivar dependency_mapping: value representing state of dependency mapping (enabled/disabled). + Required. + :vartype dependency_mapping: str + :ivar tags: machine tags. Required. + :vartype tags: dict[str, str] + """ + + _validation = { + "machine_arm_id": {"required": True}, + "dependency_mapping": {"required": True}, + "tags": {"required": True}, + } + + _attribute_map = { + "machine_arm_id": {"key": "machineArmId", "type": "str"}, + "dependency_mapping": {"key": "dependencyMapping", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + } + + def __init__(self, *, machine_arm_id: str, dependency_mapping: str, tags: Dict[str, str], **kwargs: Any) -> None: + """ + :keyword machine_arm_id: arm id of the machine. Required. + :paramtype machine_arm_id: str + :keyword dependency_mapping: value representing state of dependency mapping (enabled/disabled). + Required. + :paramtype dependency_mapping: str + :keyword tags: machine tags. Required. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.machine_arm_id = machine_arm_id + self.dependency_mapping = dependency_mapping + self.tags = tags + + +class MachineMetadataCollection(_serialization.Model): + """The list of Machine MetaData. + + All required parameters must be populated in order to send to server. + + :ivar value: The list of Machine MetaData. Required. + :vartype value: list[~azure.mgmt.migrate.models.MachineMetadata] + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[MachineMetadata]"}, + } + + def __init__(self, *, value: List["_models.MachineMetadata"], **kwargs: Any) -> None: + """ + :keyword value: The list of Machine MetaData. Required. + :paramtype value: list[~azure.mgmt.migrate.models.MachineMetadata] + """ + super().__init__(**kwargs) + self.value = value + + +class MachineResource(ProxyResource): + """An machine resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.VmwareMachineProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "VmwareMachineProperties"}, + } + + def __init__(self, *, properties: Optional["_models.VmwareMachineProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.VmwareMachineProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class MachineResourceListResult(_serialization.Model): + """The response of a MachineResource list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The MachineResource items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.MachineResource] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[MachineResource]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.MachineResource"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The MachineResource items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.MachineResource] + :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 MachineResourceUpdate(_serialization.Model): + """The type used for update operations of the MachineResource. + + :ivar properties: The updatable properties of the MachineResource. + :vartype properties: ~azure.mgmt.migrate.models.MachineResourceUpdateProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "MachineResourceUpdateProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.MachineResourceUpdateProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The updatable properties of the MachineResource. + :paramtype properties: ~azure.mgmt.migrate.models.MachineResourceUpdateProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class MachineResourceUpdateProperties(_serialization.Model): + """The updatable properties of the MachineResource. + + :ivar firmware: Gets or sets the firmware. + :vartype firmware: str + :ivar run_as_account_id: Gets or sets the run as account ID of the machine. + :vartype run_as_account_id: str + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :vartype number_of_processor_core: int + :ivar allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :vartype allocated_memory_in_mb: float + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :ivar bios_serial_number: Gets or sets the Machine BIOS serial number. + :vartype bios_serial_number: str + :ivar bios_guid: Gets or sets the BIOS GUID. + :vartype bios_guid: str + :ivar tags: Gets or sets tags on the VMware machine. + :vartype tags: dict[str, str] + """ + + _attribute_map = { + "firmware": {"key": "firmware", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "number_of_processor_core": {"key": "numberOfProcessorCore", "type": "int"}, + "allocated_memory_in_mb": {"key": "allocatedMemoryInMb", "type": "float"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "OperatingSystem"}, + "bios_serial_number": {"key": "biosSerialNumber", "type": "str"}, + "bios_guid": {"key": "biosGuid", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + } + + def __init__( + self, + *, + firmware: Optional[str] = None, + run_as_account_id: Optional[str] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + number_of_processor_core: Optional[int] = None, + allocated_memory_in_mb: Optional[float] = None, + operating_system_details: Optional["_models.OperatingSystem"] = None, + bios_serial_number: Optional[str] = None, + bios_guid: Optional[str] = None, + tags: Optional[Dict[str, str]] = None, + **kwargs: Any + ) -> None: + """ + :keyword firmware: Gets or sets the firmware. + :paramtype firmware: str + :keyword run_as_account_id: Gets or sets the run as account ID of the machine. + :paramtype run_as_account_id: str + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :paramtype number_of_processor_core: int + :keyword allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :paramtype allocated_memory_in_mb: float + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :keyword bios_serial_number: Gets or sets the Machine BIOS serial number. + :paramtype bios_serial_number: str + :keyword bios_guid: Gets or sets the BIOS GUID. + :paramtype bios_guid: str + :keyword tags: Gets or sets tags on the VMware machine. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.firmware = firmware + self.run_as_account_id = run_as_account_id + self.product_support_status = product_support_status + self.number_of_processor_core = number_of_processor_core + self.allocated_memory_in_mb = allocated_memory_in_mb + self.operating_system_details = operating_system_details + self.bios_serial_number = bios_serial_number + self.bios_guid = bios_guid + self.tags = tags + + +class MachineSoftwareInventoryProperties(_serialization.Model): + """Class for machine software inventory properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar apps_and_roles: Apps And Roles of the machine entity. + :vartype apps_and_roles: ~azure.mgmt.migrate.models.AppsAndRoles + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "apps_and_roles": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "apps_and_roles": {"key": "appsAndRoles", "type": "AppsAndRoles"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.apps_and_roles = None + self.provisioning_state = None + + +class MasterSite(TrackedResource): + """A MasterSite. + + 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: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.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: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.MasterSiteProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + "location": {"required": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "tags": {"key": "tags", "type": "{str}"}, + "location": {"key": "location", "type": "str"}, + "properties": {"key": "properties", "type": "MasterSiteProperties"}, + } + + def __init__( + self, + *, + location: str, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.MasterSiteProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword location: The geo-location where the resource lives. Required. + :paramtype location: str + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.MasterSiteProperties + """ + super().__init__(tags=tags, location=location, **kwargs) + self.properties = properties + + +class MasterSiteListResult(_serialization.Model): + """The response of a MasterSite list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The MasterSite items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.MasterSite] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[MasterSite]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.MasterSite"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The MasterSite items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.MasterSite] + :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 MasterSiteProperties(_serialization.Model): + """Class for site properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar public_network_access: Gets or sets the state of public network access. Known values are: + "NotSpecified", "Enabled", and "Disabled". + :vartype public_network_access: str or + ~azure.mgmt.migrate.models.MasterSitePropertiesPublicNetworkAccess + :ivar allow_multiple_sites: Gets or sets a value indicating whether multiple sites per site + type are + allowed. + :vartype allow_multiple_sites: bool + :ivar sites: Gets or sets the sites that are a part of Master Site. + The key + should contain the Site ARM name. + :vartype sites: list[str] + :ivar customer_storage_account_arm_id: Gets or sets a value for customer storage account ARM + id. + :vartype customer_storage_account_arm_id: str + :ivar private_endpoint_connections: Gets the private endpoint connections. + :vartype private_endpoint_connections: + list[~azure.mgmt.migrate.models.PrivateEndpointConnection] + :ivar nested_sites: Gets the nested sites under Master Site. + :vartype nested_sites: list[str] + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "private_endpoint_connections": {"readonly": True}, + "nested_sites": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "public_network_access": {"key": "publicNetworkAccess", "type": "str"}, + "allow_multiple_sites": {"key": "allowMultipleSites", "type": "bool"}, + "sites": {"key": "sites", "type": "[str]"}, + "customer_storage_account_arm_id": {"key": "customerStorageAccountArmId", "type": "str"}, + "private_endpoint_connections": {"key": "privateEndpointConnections", "type": "[PrivateEndpointConnection]"}, + "nested_sites": {"key": "nestedSites", "type": "[str]"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + public_network_access: Optional[Union[str, "_models.MasterSitePropertiesPublicNetworkAccess"]] = None, + allow_multiple_sites: Optional[bool] = None, + sites: Optional[List[str]] = None, + customer_storage_account_arm_id: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword public_network_access: Gets or sets the state of public network access. Known values + are: "NotSpecified", "Enabled", and "Disabled". + :paramtype public_network_access: str or + ~azure.mgmt.migrate.models.MasterSitePropertiesPublicNetworkAccess + :keyword allow_multiple_sites: Gets or sets a value indicating whether multiple sites per site + type are + allowed. + :paramtype allow_multiple_sites: bool + :keyword sites: Gets or sets the sites that are a part of Master Site. + The key + should contain the Site ARM name. + :paramtype sites: list[str] + :keyword customer_storage_account_arm_id: Gets or sets a value for customer storage account ARM + id. + :paramtype customer_storage_account_arm_id: str + """ + super().__init__(**kwargs) + self.public_network_access = public_network_access + self.allow_multiple_sites = allow_multiple_sites + self.sites = sites + self.customer_storage_account_arm_id = customer_storage_account_arm_id + self.private_endpoint_connections = None + self.nested_sites = None + self.provisioning_state = None + + +class MasterSiteUpdate(_serialization.Model): + """The type used for update operations of the MasterSite. + + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar properties: The updatable properties of the MasterSite. + :vartype properties: ~azure.mgmt.migrate.models.MasterSiteUpdateProperties + """ + + _attribute_map = { + "tags": {"key": "tags", "type": "{str}"}, + "properties": {"key": "properties", "type": "MasterSiteUpdateProperties"}, + } + + def __init__( + self, + *, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.MasterSiteUpdateProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword properties: The updatable properties of the MasterSite. + :paramtype properties: ~azure.mgmt.migrate.models.MasterSiteUpdateProperties + """ + super().__init__(**kwargs) + self.tags = tags + self.properties = properties + + +class MasterSiteUpdateProperties(_serialization.Model): + """The updatable properties of the MasterSite. + + :ivar public_network_access: Gets or sets the state of public network access. Known values are: + "NotSpecified", "Enabled", and "Disabled". + :vartype public_network_access: str or + ~azure.mgmt.migrate.models.MasterSitePropertiesPublicNetworkAccess + :ivar allow_multiple_sites: Gets or sets a value indicating whether multiple sites per site + type are + allowed. + :vartype allow_multiple_sites: bool + :ivar sites: Gets or sets the sites that are a part of Master Site. + The key + should contain the Site ARM name. + :vartype sites: list[str] + :ivar customer_storage_account_arm_id: Gets or sets a value for customer storage account ARM + id. + :vartype customer_storage_account_arm_id: str + """ + + _attribute_map = { + "public_network_access": {"key": "publicNetworkAccess", "type": "str"}, + "allow_multiple_sites": {"key": "allowMultipleSites", "type": "bool"}, + "sites": {"key": "sites", "type": "[str]"}, + "customer_storage_account_arm_id": {"key": "customerStorageAccountArmId", "type": "str"}, + } + + def __init__( + self, + *, + public_network_access: Optional[Union[str, "_models.MasterSitePropertiesPublicNetworkAccess"]] = None, + allow_multiple_sites: Optional[bool] = None, + sites: Optional[List[str]] = None, + customer_storage_account_arm_id: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword public_network_access: Gets or sets the state of public network access. Known values + are: "NotSpecified", "Enabled", and "Disabled". + :paramtype public_network_access: str or + ~azure.mgmt.migrate.models.MasterSitePropertiesPublicNetworkAccess + :keyword allow_multiple_sites: Gets or sets a value indicating whether multiple sites per site + type are + allowed. + :paramtype allow_multiple_sites: bool + :keyword sites: Gets or sets the sites that are a part of Master Site. + The key + should contain the Site ARM name. + :paramtype sites: list[str] + :keyword customer_storage_account_arm_id: Gets or sets a value for customer storage account ARM + id. + :paramtype customer_storage_account_arm_id: str + """ + super().__init__(**kwargs) + self.public_network_access = public_network_access + self.allow_multiple_sites = allow_multiple_sites + self.sites = sites + self.customer_storage_account_arm_id = customer_storage_account_arm_id + + +class OperatingSystem(_serialization.Model): + """Second level object returned as part of Machine REST resource. + + :ivar os_type: Gets or sets the type of the operating system. + :vartype os_type: str + :ivar os_name: Gets or sets the Name of the operating system. + :vartype os_name: str + :ivar os_version: Gets or sets the Version of the operating system. + :vartype os_version: str + :ivar os_architecture: Gets or sets the Architecture of the operating system. + :vartype os_architecture: str + """ + + _attribute_map = { + "os_type": {"key": "osType", "type": "str"}, + "os_name": {"key": "osName", "type": "str"}, + "os_version": {"key": "osVersion", "type": "str"}, + "os_architecture": {"key": "osArchitecture", "type": "str"}, + } + + def __init__( + self, + *, + os_type: Optional[str] = None, + os_name: Optional[str] = None, + os_version: Optional[str] = None, + os_architecture: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword os_type: Gets or sets the type of the operating system. + :paramtype os_type: str + :keyword os_name: Gets or sets the Name of the operating system. + :paramtype os_name: str + :keyword os_version: Gets or sets the Version of the operating system. + :paramtype os_version: str + :keyword os_architecture: Gets or sets the Architecture of the operating system. + :paramtype os_architecture: str + """ + super().__init__(**kwargs) + self.os_type = os_type + self.os_name = os_name + self.os_version = os_version + self.os_architecture = os_architecture + + +class Operation(_serialization.Model): + """Details of a REST API operation, returned from the Resource Provider Operations API. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar name: The name of the operation, as per Resource-Based Access Control (RBAC). Examples: + "Microsoft.Compute/virtualMachines/write", "Microsoft.Compute/virtualMachines/capture/action". + :vartype name: str + :ivar is_data_action: Whether the operation applies to data-plane. This is "true" for + data-plane operations and "false" for ARM/control-plane operations. + :vartype is_data_action: bool + :ivar display: Localized display information for this particular operation. + :vartype display: ~azure.mgmt.migrate.models.OperationDisplay + :ivar origin: The intended executor of the operation; as in Resource Based Access Control + (RBAC) and audit logs UX. Default value is "user,system". Known values are: "user", "system", + and "user,system". + :vartype origin: str or ~azure.mgmt.migrate.models.Origin + :ivar action_type: Enum. Indicates the action type. "Internal" refers to actions that are for + internal only APIs. "Internal" + :vartype action_type: str or ~azure.mgmt.migrate.models.ActionType + """ + + _validation = { + "name": {"readonly": True}, + "is_data_action": {"readonly": True}, + "origin": {"readonly": True}, + "action_type": {"readonly": True}, + } + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "is_data_action": {"key": "isDataAction", "type": "bool"}, + "display": {"key": "display", "type": "OperationDisplay"}, + "origin": {"key": "origin", "type": "str"}, + "action_type": {"key": "actionType", "type": "str"}, + } + + def __init__(self, *, display: Optional["_models.OperationDisplay"] = None, **kwargs: Any) -> None: + """ + :keyword display: Localized display information for this particular operation. + :paramtype display: ~azure.mgmt.migrate.models.OperationDisplay + """ + super().__init__(**kwargs) + self.name = None + self.is_data_action = None + self.display = display + self.origin = None + self.action_type = None + + +class OperationDisplay(_serialization.Model): + """Localized display information for this particular operation. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar provider: The localized friendly form of the resource provider name, e.g. "Microsoft + Monitoring Insights" or "Microsoft Compute". + :vartype provider: str + :ivar resource: The localized friendly name of the resource type related to this operation. + E.g. "Virtual Machines" or "Job Schedule Collections". + :vartype resource: str + :ivar operation: The concise, localized friendly name for the operation; suitable for + dropdowns. E.g. "Create or Update Virtual Machine", "Restart Virtual Machine". + :vartype operation: str + :ivar description: The short, localized friendly description of the operation; suitable for + tool tips and detailed views. + :vartype description: str + """ + + _validation = { + "provider": {"readonly": True}, + "resource": {"readonly": True}, + "operation": {"readonly": True}, + "description": {"readonly": True}, + } + + _attribute_map = { + "provider": {"key": "provider", "type": "str"}, + "resource": {"key": "resource", "type": "str"}, + "operation": {"key": "operation", "type": "str"}, + "description": {"key": "description", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.provider = None + self.resource = None + self.operation = None + self.description = None + + +class OperationListResult(_serialization.Model): + """A list of REST API operations supported by an Azure Resource Provider. It contains an URL link + to get the next set of results. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: List of operations supported by the resource provider. + :vartype value: list[~azure.mgmt.migrate.models.Operation] + :ivar next_link: URL to get the next set of operation list results (if there are any). + :vartype next_link: str + """ + + _validation = { + "value": {"readonly": True}, + "next_link": {"readonly": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[Operation]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.value = None + self.next_link = None + + +class OperationStatus(_serialization.Model): + """Operation status REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Gets the Id. + :vartype id: str + :ivar name: Gets the operation name. + :vartype name: str + :ivar status: Gets the status of the operation. ARM expects the terminal status to be one + of + Succeeded/ Failed/ Canceled. All other values imply that the + operation is still running. + :vartype status: str + :ivar start_time: Gets the start time. + :vartype start_time: str + :ivar end_time: Gets the start time. + :vartype end_time: str + :ivar error: Gets the error. + :vartype error: ~azure.mgmt.migrate.models.OperationStatusError + :ivar properties: Operation status properties. + :vartype properties: ~azure.mgmt.migrate.models.OperationStatusProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "status": {"readonly": True}, + "start_time": {"readonly": True}, + "end_time": {"readonly": True}, + "error": {"readonly": True}, + "properties": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "start_time": {"key": "startTime", "type": "str"}, + "end_time": {"key": "endTime", "type": "str"}, + "error": {"key": "error", "type": "OperationStatusError"}, + "properties": {"key": "properties", "type": "OperationStatusProperties"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.id = None + self.name = None + self.status = None + self.start_time = None + self.end_time = None + self.error = None + self.properties = None + + +class OperationStatusError(_serialization.Model): + """Class for operation status errors. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar code: Gets the error code. + :vartype code: str + :ivar message: Gets the error message. + :vartype message: str + """ + + _validation = { + "code": {"readonly": True}, + "message": {"readonly": True}, + } + + _attribute_map = { + "code": {"key": "code", "type": "str"}, + "message": {"key": "message", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.code = None + self.message = None + + +class OperationStatusProperties(_serialization.Model): + """Class for operation result properties. + + :ivar result: Gets or sets the result or output of the workflow. + :vartype result: str + """ + + _attribute_map = { + "result": {"key": "result", "type": "str"}, + } + + def __init__(self, *, result: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword result: Gets or sets the result or output of the workflow. + :paramtype result: str + """ + super().__init__(**kwargs) + self.result = result + + +class OracleDiscovery(_serialization.Model): + """Data related to a machine's Oracle discovery. + + :ivar total_instance_count: Gets or sets number of successfully discovered instances. + :vartype total_instance_count: int + :ivar total_database_count: Gets or sets number of successfully discovered databases. + :vartype total_database_count: int + :ivar shallow_discovery_status: Gets or sets status of partner shallow discovery. Known values + are: "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype shallow_discovery_status: str or ~azure.mgmt.migrate.models.ShallowDiscoveryStatus + :ivar discovery_scope_status: Gets or sets status of partner discovery. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype discovery_scope_status: str or ~azure.mgmt.migrate.models.DiscoveryScopeStatus + """ + + _attribute_map = { + "total_instance_count": {"key": "totalInstanceCount", "type": "int"}, + "total_database_count": {"key": "totalDatabaseCount", "type": "int"}, + "shallow_discovery_status": {"key": "shallowDiscoveryStatus", "type": "str"}, + "discovery_scope_status": {"key": "discoveryScopeStatus", "type": "str"}, + } + + def __init__( + self, + *, + total_instance_count: Optional[int] = None, + total_database_count: Optional[int] = None, + shallow_discovery_status: Optional[Union[str, "_models.ShallowDiscoveryStatus"]] = None, + discovery_scope_status: Optional[Union[str, "_models.DiscoveryScopeStatus"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword total_instance_count: Gets or sets number of successfully discovered instances. + :paramtype total_instance_count: int + :keyword total_database_count: Gets or sets number of successfully discovered databases. + :paramtype total_database_count: int + :keyword shallow_discovery_status: Gets or sets status of partner shallow discovery. Known + values are: "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :paramtype shallow_discovery_status: str or ~azure.mgmt.migrate.models.ShallowDiscoveryStatus + :keyword discovery_scope_status: Gets or sets status of partner discovery. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :paramtype discovery_scope_status: str or ~azure.mgmt.migrate.models.DiscoveryScopeStatus + """ + super().__init__(**kwargs) + self.total_instance_count = total_instance_count + self.total_database_count = total_database_count + self.shallow_discovery_status = shallow_discovery_status + self.discovery_scope_status = discovery_scope_status + + +class OtherDatabase(_serialization.Model): + """OtherDatabase in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar database_type: Gets or sets DatabaseType of the OtherDatabase. + :vartype database_type: str + :ivar instance: Gets or sets Instance of the OtherDatabase. + :vartype instance: str + :ivar version: Gets or sets Version of the OtherDatabase. + :vartype version: str + """ + + _validation = { + "database_type": {"readonly": True}, + "instance": {"readonly": True}, + "version": {"readonly": True}, + } + + _attribute_map = { + "database_type": {"key": "databaseType", "type": "str"}, + "instance": {"key": "instance", "type": "str"}, + "version": {"key": "version", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.database_type = None + self.instance = None + self.version = None + + +class PagedDeleteImportMachinesJob(_serialization.Model): + """Paged collection of DeleteImportMachinesJob items. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: The DeleteImportMachinesJob items on this page. + :vartype value: list[~azure.mgmt.migrate.models.DeleteImportMachinesJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"readonly": True}, + "next_link": {"readonly": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[DeleteImportMachinesJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.value = None + self.next_link = None + + +class PagedExportImportedMachinesJob(_serialization.Model): + """Paged collection of ExportImportedMachinesJob items. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: The ExportImportedMachinesJob items on this page. + :vartype value: list[~azure.mgmt.migrate.models.ExportImportedMachinesJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"readonly": True}, + "next_link": {"readonly": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ExportImportedMachinesJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.value = None + self.next_link = None + + +class PagedImportMachinesJob(_serialization.Model): + """Paged collection of ImportMachinesJob items. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: The ImportMachinesJob items on this page. + :vartype value: list[~azure.mgmt.migrate.models.ImportMachinesJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"readonly": True}, + "next_link": {"readonly": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ImportMachinesJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.value = None + self.next_link = None + + +class PrivateEndpointConnection(ProxyResource): + """REST model used to encapsulate Private Link properties for tracked resources. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.PrivateEndpointConnectionPropertiesV2 + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "PrivateEndpointConnectionPropertiesV2"}, + } + + def __init__( + self, *, properties: Optional["_models.PrivateEndpointConnectionPropertiesV2"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.PrivateEndpointConnectionPropertiesV2 + """ + super().__init__(**kwargs) + self.properties = properties + + +class PrivateEndpointConnectionListResult(_serialization.Model): + """The response of a PrivateEndpointConnection list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The PrivateEndpointConnection items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.PrivateEndpointConnection] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[PrivateEndpointConnection]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.PrivateEndpointConnection"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The PrivateEndpointConnection items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.PrivateEndpointConnection] + :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 PrivateEndpointConnectionPropertiesV2(_serialization.Model): + """PrivateEndpointConnectionProperties V2. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar group_ids: array of group ids. + :vartype group_ids: list[str] + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + :ivar private_endpoint: private endpoints. + :vartype private_endpoint: ~azure.mgmt.migrate.models.ResourceId + :ivar private_link_service_connection_state: private endpoints connection state. + :vartype private_link_service_connection_state: + ~azure.mgmt.migrate.models.PrivateLinkServiceConnectionState + """ + + _validation = { + "group_ids": {"readonly": True}, + "provisioning_state": {"readonly": True}, + "private_endpoint": {"readonly": True}, + } + + _attribute_map = { + "group_ids": {"key": "groupIds", "type": "[str]"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + "private_endpoint": {"key": "privateEndpoint", "type": "ResourceId"}, + "private_link_service_connection_state": { + "key": "privateLinkServiceConnectionState", + "type": "PrivateLinkServiceConnectionState", + }, + } + + def __init__( + self, + *, + private_link_service_connection_state: Optional["_models.PrivateLinkServiceConnectionState"] = None, + **kwargs: Any + ) -> None: + """ + :keyword private_link_service_connection_state: private endpoints connection state. + :paramtype private_link_service_connection_state: + ~azure.mgmt.migrate.models.PrivateLinkServiceConnectionState + """ + super().__init__(**kwargs) + self.group_ids = None + self.provisioning_state = None + self.private_endpoint = None + self.private_link_service_connection_state = private_link_service_connection_state + + +class PrivateLinkResource(ProxyResource): + """REST model used to encapsulate Private Link properties for tracked resources. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.PrivateLinkResourceProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "PrivateLinkResourceProperties"}, + } + + def __init__(self, *, properties: Optional["_models.PrivateLinkResourceProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.PrivateLinkResourceProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class PrivateLinkResourceListResult(_serialization.Model): + """The response of a PrivateLinkResource list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The PrivateLinkResource items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.PrivateLinkResource] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[PrivateLinkResource]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.PrivateLinkResource"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The PrivateLinkResource items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.PrivateLinkResource] + :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 PrivateLinkResourceProperties(_serialization.Model): + """private link resource properties model. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar required_members: required members. + :vartype required_members: list[str] + :ivar required_zone_names: required zone names. + :vartype required_zone_names: list[str] + :ivar group_id: group id. + :vartype group_id: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "required_members": {"key": "requiredMembers", "type": "[str]"}, + "required_zone_names": {"key": "requiredZoneNames", "type": "[str]"}, + "group_id": {"key": "groupId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + required_members: Optional[List[str]] = None, + required_zone_names: Optional[List[str]] = None, + group_id: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword required_members: required members. + :paramtype required_members: list[str] + :keyword required_zone_names: required zone names. + :paramtype required_zone_names: list[str] + :keyword group_id: group id. + :paramtype group_id: str + """ + super().__init__(**kwargs) + self.required_members = required_members + self.required_zone_names = required_zone_names + self.group_id = group_id + self.provisioning_state = None + + +class PrivateLinkServiceConnectionState(_serialization.Model): + """Service Connection State. + + :ivar status: state status. Known values are: "Approved", "Pending", "Rejected", and + "Disconnected". + :vartype status: str or ~azure.mgmt.migrate.models.PrivateLinkServiceConnectionStateStatus + :ivar description: description string. + :vartype description: str + :ivar actions_required: actions required. + :vartype actions_required: str + """ + + _attribute_map = { + "status": {"key": "status", "type": "str"}, + "description": {"key": "description", "type": "str"}, + "actions_required": {"key": "actionsRequired", "type": "str"}, + } + + def __init__( + self, + *, + status: Optional[Union[str, "_models.PrivateLinkServiceConnectionStateStatus"]] = None, + description: Optional[str] = None, + actions_required: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword status: state status. Known values are: "Approved", "Pending", "Rejected", and + "Disconnected". + :paramtype status: str or ~azure.mgmt.migrate.models.PrivateLinkServiceConnectionStateStatus + :keyword description: description string. + :paramtype description: str + :keyword actions_required: actions required. + :paramtype actions_required: str + """ + super().__init__(**kwargs) + self.status = status + self.description = description + self.actions_required = actions_required + + +class ProcessorInfo(_serialization.Model): + """Processor Information. + + :ivar name: Gets or sets the name\\model of a processor. + :vartype name: str + :ivar number_of_sockets: Gets or sets the number of sockets. + :vartype number_of_sockets: int + :ivar number_of_cores_per_socket: Gets or sets the total number of cores in a socket. + :vartype number_of_cores_per_socket: int + """ + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "number_of_sockets": {"key": "numberOfSockets", "type": "int"}, + "number_of_cores_per_socket": {"key": "numberOfCoresPerSocket", "type": "int"}, + } + + def __init__( + self, + *, + name: Optional[str] = None, + number_of_sockets: Optional[int] = None, + number_of_cores_per_socket: Optional[int] = None, + **kwargs: Any + ) -> None: + """ + :keyword name: Gets or sets the name\\model of a processor. + :paramtype name: str + :keyword number_of_sockets: Gets or sets the number of sockets. + :paramtype number_of_sockets: int + :keyword number_of_cores_per_socket: Gets or sets the total number of cores in a socket. + :paramtype number_of_cores_per_socket: int + """ + super().__init__(**kwargs) + self.name = name + self.number_of_sockets = number_of_sockets + self.number_of_cores_per_socket = number_of_cores_per_socket + + +class ProductSupportStatus(_serialization.Model): + """product support status. + + :ivar current_version: current version. + :vartype current_version: str + :ivar esu_status: esu status. Known values are: "Unknown", "Active", and "InActive". + :vartype esu_status: str or ~azure.mgmt.migrate.models.EsuStatus + :ivar support_status: support status. Known values are: "Unknown", "Mainstream", and + "Extended". + :vartype support_status: str or ~azure.mgmt.migrate.models.SupportStatus + :ivar support_end_date: support end date. + :vartype support_end_date: ~datetime.datetime + :ivar esu_year: esu year. Known values are: "Unknown", "FirstYear", "SecondYear", "ThirdYear", + and "UpgradeYear". + :vartype esu_year: str or ~azure.mgmt.migrate.models.EsuYear + """ + + _attribute_map = { + "current_version": {"key": "currentVersion", "type": "str"}, + "esu_status": {"key": "esuStatus", "type": "str"}, + "support_status": {"key": "supportStatus", "type": "str"}, + "support_end_date": {"key": "supportEndDate", "type": "iso-8601"}, + "esu_year": {"key": "esuYear", "type": "str"}, + } + + def __init__( + self, + *, + current_version: Optional[str] = None, + esu_status: Optional[Union[str, "_models.EsuStatus"]] = None, + support_status: Optional[Union[str, "_models.SupportStatus"]] = None, + support_end_date: Optional[datetime.datetime] = None, + esu_year: Optional[Union[str, "_models.EsuYear"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword current_version: current version. + :paramtype current_version: str + :keyword esu_status: esu status. Known values are: "Unknown", "Active", and "InActive". + :paramtype esu_status: str or ~azure.mgmt.migrate.models.EsuStatus + :keyword support_status: support status. Known values are: "Unknown", "Mainstream", and + "Extended". + :paramtype support_status: str or ~azure.mgmt.migrate.models.SupportStatus + :keyword support_end_date: support end date. + :paramtype support_end_date: ~datetime.datetime + :keyword esu_year: esu year. Known values are: "Unknown", "FirstYear", "SecondYear", + "ThirdYear", and "UpgradeYear". + :paramtype esu_year: str or ~azure.mgmt.migrate.models.EsuYear + """ + super().__init__(**kwargs) + self.current_version = current_version + self.esu_status = esu_status + self.support_status = support_status + self.support_end_date = support_end_date + self.esu_year = esu_year + + +class ProxySiteRefreshBody(_serialization.Model): + """Request body for Proxy site refresh action. + + :ivar appliance_name: Gets or sets the appliance name of the agent in the site. + :vartype appliance_name: str + """ + + _attribute_map = { + "appliance_name": {"key": "applianceName", "type": "str"}, + } + + def __init__(self, *, appliance_name: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword appliance_name: Gets or sets the appliance name of the agent in the site. + :paramtype appliance_name: str + """ + super().__init__(**kwargs) + self.appliance_name = appliance_name + + +class RequestExportMachineErrorsProperties(_serialization.Model): + """The Properties class for export machine errors request body. + + :ivar discovery_scope: Gets or sets the discovery scope. Known values are: "AppsAndRoles", + "DependencyMap", "StaticData", and "SQLServerConnectionInfo". + :vartype discovery_scope: str or ~azure.mgmt.migrate.models.ExportMachineErrorsProperties + """ + + _attribute_map = { + "discovery_scope": {"key": "discoveryScope", "type": "str"}, + } + + def __init__( + self, *, discovery_scope: Optional[Union[str, "_models.ExportMachineErrorsProperties"]] = None, **kwargs: Any + ) -> None: + """ + :keyword discovery_scope: Gets or sets the discovery scope. Known values are: "AppsAndRoles", + "DependencyMap", "StaticData", and "SQLServerConnectionInfo". + :paramtype discovery_scope: str or ~azure.mgmt.migrate.models.ExportMachineErrorsProperties + """ + super().__init__(**kwargs) + self.discovery_scope = discovery_scope + + +class ResourceId(_serialization.Model): + """PrivateEndpointModelsResourceId. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: id name. + :vartype id: str + """ + + _validation = { + "id": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.id = None + + +class RunAsAccountMachineInput(_serialization.Model): + """Machine class. + + :ivar machine_id: Gets or sets the ARM id of the machine. + :vartype machine_id: str + :ivar run_as_account_id: run as AccountId. + :vartype run_as_account_id: str + """ + + _attribute_map = { + "machine_id": {"key": "machineId", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + } + + def __init__( + self, *, machine_id: Optional[str] = None, run_as_account_id: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword machine_id: Gets or sets the ARM id of the machine. + :paramtype machine_id: str + :keyword run_as_account_id: run as AccountId. + :paramtype run_as_account_id: str + """ + super().__init__(**kwargs) + self.machine_id = machine_id + self.run_as_account_id = run_as_account_id + + +class RunAsAccountProperties(_serialization.Model): + """Class for run as account properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar display_name: Display name of the run as account. + :vartype display_name: str + :ivar credential_type: Gets the credential type of the run as account. + :vartype credential_type: str + :ivar created_timestamp: Timestamp marking run as account creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Timestamp marking last updated on the run as account. + :vartype updated_timestamp: str + :ivar appliance_name: Gets the appliance name of the run as account. + :vartype appliance_name: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "display_name": {"readonly": True}, + "credential_type": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "appliance_name": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "display_name": {"key": "displayName", "type": "str"}, + "credential_type": {"key": "credentialType", "type": "str"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.display_name = None + self.credential_type = None + self.created_timestamp = None + self.updated_timestamp = None + self.appliance_name = None + self.provisioning_state = None + + +class SasUriResponse(_serialization.Model): + """Import URI response class. + + :ivar job_arm_id: Gets or sets the job ARM ID. + :vartype job_arm_id: str + :ivar uri: Gets or sets the SAS URI. + :vartype uri: str + :ivar import_type: Gets or sets the import type. Known values are: "AzureMigrateCSV" and + "RVToolsXlsx". + :vartype import_type: str or ~azure.mgmt.migrate.models.ImportTypeValues + """ + + _attribute_map = { + "job_arm_id": {"key": "jobArmId", "type": "str"}, + "uri": {"key": "uri", "type": "str"}, + "import_type": {"key": "importType", "type": "str"}, + } + + def __init__( + self, + *, + job_arm_id: Optional[str] = None, + uri: Optional[str] = None, + import_type: Optional[Union[str, "_models.ImportTypeValues"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword job_arm_id: Gets or sets the job ARM ID. + :paramtype job_arm_id: str + :keyword uri: Gets or sets the SAS URI. + :paramtype uri: str + :keyword import_type: Gets or sets the import type. Known values are: "AzureMigrateCSV" and + "RVToolsXlsx". + :paramtype import_type: str or ~azure.mgmt.migrate.models.ImportTypeValues + """ + super().__init__(**kwargs) + self.job_arm_id = job_arm_id + self.uri = uri + self.import_type = import_type + + +class Server(ProxyResource): + """A machine resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.ServerProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "ServerProperties"}, + } + + def __init__(self, *, properties: Optional["_models.ServerProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.ServerProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ServerDisk(_serialization.Model): + """Server disk details. + + :ivar id: Gets or sets Id of the disk. + :vartype id: str + :ivar generated_id: Gets or sets generated Id of the disk. + :vartype generated_id: str + :ivar max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :vartype max_size_in_bytes: int + :ivar name: Gets or sets Name of the disk. + :vartype name: str + :ivar disk_type: Gets or sets Type of the disk. + :vartype disk_type: str + :ivar lun: Gets or sets LUN of the disk. + :vartype lun: int + :ivar path: Gets or sets Path of the disk. + :vartype path: str + """ + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "generated_id": {"key": "generatedId", "type": "str"}, + "max_size_in_bytes": {"key": "maxSizeInBytes", "type": "int"}, + "name": {"key": "name", "type": "str"}, + "disk_type": {"key": "diskType", "type": "str"}, + "lun": {"key": "lun", "type": "int"}, + "path": {"key": "path", "type": "str"}, + } + + def __init__( + self, + *, + id: Optional[str] = None, # pylint: disable=redefined-builtin + generated_id: Optional[str] = None, + max_size_in_bytes: Optional[int] = None, + name: Optional[str] = None, + disk_type: Optional[str] = None, + lun: Optional[int] = None, + path: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword id: Gets or sets Id of the disk. + :paramtype id: str + :keyword generated_id: Gets or sets generated Id of the disk. + :paramtype generated_id: str + :keyword max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :paramtype max_size_in_bytes: int + :keyword name: Gets or sets Name of the disk. + :paramtype name: str + :keyword disk_type: Gets or sets Type of the disk. + :paramtype disk_type: str + :keyword lun: Gets or sets LUN of the disk. + :paramtype lun: int + :keyword path: Gets or sets Path of the disk. + :paramtype path: str + """ + super().__init__(**kwargs) + self.id = id + self.generated_id = generated_id + self.max_size_in_bytes = max_size_in_bytes + self.name = name + self.disk_type = disk_type + self.lun = lun + self.path = path + + +class ServerJob(ProxyResource): + """A host resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.JobProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "JobProperties"}, + } + + def __init__(self, *, properties: Optional["_models.JobProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.JobProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ServerJobListResult(_serialization.Model): + """The response of a ServerJob list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The ServerJob items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.ServerJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ServerJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.ServerJob"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The ServerJob items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.ServerJob] + :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 ServerListResult(_serialization.Model): + """The response of a Server list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The Server items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.Server] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[Server]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.Server"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The Server items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.Server] + :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 ServerNetworkAdapter(_serialization.Model): + """Second level object represented in responses as part of Machine REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar nic_id: Gets or sets the NIC Id. + :vartype nic_id: str + :ivar mac_address: Gets or sets Mac address of the NIC. + :vartype mac_address: str + :ivar ip_address_list: Gets IP addresses for the machine. + :vartype ip_address_list: list[str] + :ivar network_name: Gets or sets Network Name. + :vartype network_name: str + :ivar ip_address_type: Gets or sets Type of the IP address. + :vartype ip_address_type: str + """ + + _validation = { + "ip_address_list": {"readonly": True}, + } + + _attribute_map = { + "nic_id": {"key": "nicId", "type": "str"}, + "mac_address": {"key": "macAddress", "type": "str"}, + "ip_address_list": {"key": "ipAddressList", "type": "[str]"}, + "network_name": {"key": "networkName", "type": "str"}, + "ip_address_type": {"key": "ipAddressType", "type": "str"}, + } + + def __init__( + self, + *, + nic_id: Optional[str] = None, + mac_address: Optional[str] = None, + network_name: Optional[str] = None, + ip_address_type: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword nic_id: Gets or sets the NIC Id. + :paramtype nic_id: str + :keyword mac_address: Gets or sets Mac address of the NIC. + :paramtype mac_address: str + :keyword network_name: Gets or sets Network Name. + :paramtype network_name: str + :keyword ip_address_type: Gets or sets Type of the IP address. + :paramtype ip_address_type: str + """ + super().__init__(**kwargs) + self.nic_id = nic_id + self.mac_address = mac_address + self.ip_address_list = None + self.network_name = network_name + self.ip_address_type = ip_address_type + + +class ServerProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Server REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar display_name: Gets the Display name of the machine. + For server entity hydrated + FQDN is set as display name + as the server id and server name are + same. + :vartype display_name: str + :ivar fqdn: Gets or sets the FQDN/IPAddress of the server. + :vartype fqdn: str + :ivar network_adapters: Gets or sets the network adapters of the server. + :vartype network_adapters: list[~azure.mgmt.migrate.models.ServerNetworkAdapter] + :ivar hydrated_fqdn: Gets or sets the FQDN of machine which can be changed. + :vartype hydrated_fqdn: str + :ivar disks: Gets or sets the disk details of server. + :vartype disks: list[~azure.mgmt.migrate.models.ServerDisk] + :ivar validation_required: Gets or sets the validation required for the machine. + :vartype validation_required: str + :ivar processor_info: Processor Info of the Server machine. + :vartype processor_info: ~azure.mgmt.migrate.models.ProcessorInfo + :ivar firmware: Gets or sets the firmware. + :vartype firmware: str + :ivar guest_os_details: Operating System Details extracted from the guest + bu executing + script inside the guest VM. + :vartype guest_os_details: ~azure.mgmt.migrate.models.GuestOsDetails + :ivar number_of_applications: Number of applications installed in the guest VM. + :vartype number_of_applications: int + :ivar guest_details_discovery_timestamp: The last time at which the Guest Details was + discovered + or the + error while discovering guest details based discovery + of the + machine. + :vartype guest_details_discovery_timestamp: ~datetime.datetime + :ivar is_guest_details_discovery_in_progress: Whether Refresh Fabric Layout Guest Details has + been completed once. + + Portal will show discovery in progress, if this value is true. + :vartype is_guest_details_discovery_in_progress: bool + :ivar dependency_mapping: Gets or sets if dependency mapping feature is enabled or not + for + the VM. + :vartype dependency_mapping: str + :ivar dependency_mapping_start_time: Gets or sets when dependency mapping collection is last + started. + :vartype dependency_mapping_start_time: ~datetime.datetime + :ivar dependency_mapping_end_time: Gets or sets when dependency mapping collection was last + disabled. + :vartype dependency_mapping_end_time: str + :ivar run_as_account_id: Gets or sets the run as account ID of the machine. + :vartype run_as_account_id: str + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar errors: Errors for machine. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar application_discovery: Gets the data related to application discovery. + :vartype application_discovery: ~azure.mgmt.migrate.models.ApplicationDiscovery + :ivar dependency_map_discovery: Gets the data related to dependency map discovery. + :vartype dependency_map_discovery: ~azure.mgmt.migrate.models.DependencyMapDiscovery + :ivar static_discovery: Gets the data related to static discovery. + :vartype static_discovery: ~azure.mgmt.migrate.models.StaticDiscovery + :ivar sql_discovery: Gets the data related to SQL discovery. + :vartype sql_discovery: ~azure.mgmt.migrate.models.SqlDiscovery + :ivar web_app_discovery: Gets the data related to Web application discovery. + :vartype web_app_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar oracle_discovery: Gets the data related to Oracle discovery. + :vartype oracle_discovery: ~azure.mgmt.migrate.models.OracleDiscovery + :ivar spring_boot_discovery: Gets the data related to SpringBoot discovery. + :vartype spring_boot_discovery: ~azure.mgmt.migrate.models.SpringBootDiscovery + :ivar iis_discovery: Gets the data related to iis discovery. + :vartype iis_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar tomcat_discovery: Gets the data related to tomcat discovery. + :vartype tomcat_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar apps_and_roles: Apps And Roles of the Server. + :vartype apps_and_roles: ~azure.mgmt.migrate.models.AppsAndRoles + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :vartype number_of_processor_core: int + :ivar allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :vartype allocated_memory_in_mb: float + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :ivar bios_serial_number: Gets or sets the Machine BIOS serial number. + :vartype bios_serial_number: str + :ivar bios_guid: Gets or sets the BIOS GUID. + :vartype bios_guid: str + :ivar is_deleted: Value indicating whether VM is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the marking machine creation. + :vartype created_timestamp: str + :ivar tags: Gets or sets tags on the Server machine. + :vartype tags: dict[str, JSON] + :ivar updated_timestamp: Gets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar provisioning_state: Provisioning state of the server resource. Known values are: + "Created", "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "display_name": {"readonly": True}, + "processor_info": {"readonly": True}, + "guest_os_details": {"readonly": True}, + "number_of_applications": {"readonly": True}, + "guest_details_discovery_timestamp": {"readonly": True}, + "is_guest_details_discovery_in_progress": {"readonly": True}, + "dependency_mapping": {"readonly": True}, + "dependency_mapping_start_time": {"readonly": True}, + "dependency_mapping_end_time": {"readonly": True}, + "appliance_names": {"readonly": True}, + "errors": {"readonly": True}, + "application_discovery": {"readonly": True}, + "dependency_map_discovery": {"readonly": True}, + "static_discovery": {"readonly": True}, + "sql_discovery": {"readonly": True}, + "web_app_discovery": {"readonly": True}, + "oracle_discovery": {"readonly": True}, + "spring_boot_discovery": {"readonly": True}, + "iis_discovery": {"readonly": True}, + "tomcat_discovery": {"readonly": True}, + "apps_and_roles": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + } + + _attribute_map = { + "display_name": {"key": "displayName", "type": "str"}, + "fqdn": {"key": "fqdn", "type": "str"}, + "network_adapters": {"key": "networkAdapters", "type": "[ServerNetworkAdapter]"}, + "hydrated_fqdn": {"key": "hydratedFqdn", "type": "str"}, + "disks": {"key": "disks", "type": "[ServerDisk]"}, + "validation_required": {"key": "validationRequired", "type": "str"}, + "processor_info": {"key": "processorInfo", "type": "ProcessorInfo"}, + "firmware": {"key": "firmware", "type": "str"}, + "guest_os_details": {"key": "guestOsDetails", "type": "GuestOsDetails"}, + "number_of_applications": {"key": "numberOfApplications", "type": "int"}, + "guest_details_discovery_timestamp": {"key": "guestDetailsDiscoveryTimestamp", "type": "iso-8601"}, + "is_guest_details_discovery_in_progress": {"key": "isGuestDetailsDiscoveryInProgress", "type": "bool"}, + "dependency_mapping": {"key": "dependencyMapping", "type": "str"}, + "dependency_mapping_start_time": {"key": "dependencyMappingStartTime", "type": "iso-8601"}, + "dependency_mapping_end_time": {"key": "dependencyMappingEndTime", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "application_discovery": {"key": "applicationDiscovery", "type": "ApplicationDiscovery"}, + "dependency_map_discovery": {"key": "dependencyMapDiscovery", "type": "DependencyMapDiscovery"}, + "static_discovery": {"key": "staticDiscovery", "type": "StaticDiscovery"}, + "sql_discovery": {"key": "sqlDiscovery", "type": "SqlDiscovery"}, + "web_app_discovery": {"key": "webAppDiscovery", "type": "WebAppDiscovery"}, + "oracle_discovery": {"key": "oracleDiscovery", "type": "OracleDiscovery"}, + "spring_boot_discovery": {"key": "springBootDiscovery", "type": "SpringBootDiscovery"}, + "iis_discovery": {"key": "iisDiscovery", "type": "WebAppDiscovery"}, + "tomcat_discovery": {"key": "tomcatDiscovery", "type": "WebAppDiscovery"}, + "apps_and_roles": {"key": "appsAndRoles", "type": "AppsAndRoles"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "number_of_processor_core": {"key": "numberOfProcessorCore", "type": "int"}, + "allocated_memory_in_mb": {"key": "allocatedMemoryInMb", "type": "float"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "OperatingSystem"}, + "bios_serial_number": {"key": "biosSerialNumber", "type": "str"}, + "bios_guid": {"key": "biosGuid", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "tags": {"key": "tags", "type": "{object}"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( # pylint: disable=too-many-locals + self, + *, + fqdn: Optional[str] = None, + network_adapters: Optional[List["_models.ServerNetworkAdapter"]] = None, + hydrated_fqdn: Optional[str] = None, + disks: Optional[List["_models.ServerDisk"]] = None, + validation_required: Optional[str] = None, + firmware: Optional[str] = None, + run_as_account_id: Optional[str] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + number_of_processor_core: Optional[int] = None, + allocated_memory_in_mb: Optional[float] = None, + operating_system_details: Optional["_models.OperatingSystem"] = None, + bios_serial_number: Optional[str] = None, + bios_guid: Optional[str] = None, + tags: Optional[Dict[str, JSON]] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword fqdn: Gets or sets the FQDN/IPAddress of the server. + :paramtype fqdn: str + :keyword network_adapters: Gets or sets the network adapters of the server. + :paramtype network_adapters: list[~azure.mgmt.migrate.models.ServerNetworkAdapter] + :keyword hydrated_fqdn: Gets or sets the FQDN of machine which can be changed. + :paramtype hydrated_fqdn: str + :keyword disks: Gets or sets the disk details of server. + :paramtype disks: list[~azure.mgmt.migrate.models.ServerDisk] + :keyword validation_required: Gets or sets the validation required for the machine. + :paramtype validation_required: str + :keyword firmware: Gets or sets the firmware. + :paramtype firmware: str + :keyword run_as_account_id: Gets or sets the run as account ID of the machine. + :paramtype run_as_account_id: str + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :paramtype number_of_processor_core: int + :keyword allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :paramtype allocated_memory_in_mb: float + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :keyword bios_serial_number: Gets or sets the Machine BIOS serial number. + :paramtype bios_serial_number: str + :keyword bios_guid: Gets or sets the BIOS GUID. + :paramtype bios_guid: str + :keyword tags: Gets or sets tags on the Server machine. + :paramtype tags: dict[str, JSON] + :keyword provisioning_state: Provisioning state of the server resource. Known values are: + "Created", "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.display_name = None + self.fqdn = fqdn + self.network_adapters = network_adapters + self.hydrated_fqdn = hydrated_fqdn + self.disks = disks + self.validation_required = validation_required + self.processor_info = None + self.firmware = firmware + self.guest_os_details = None + self.number_of_applications = None + self.guest_details_discovery_timestamp = None + self.is_guest_details_discovery_in_progress = None + self.dependency_mapping = None + self.dependency_mapping_start_time = None + self.dependency_mapping_end_time = None + self.run_as_account_id = run_as_account_id + self.appliance_names = None + self.errors = None + self.application_discovery = None + self.dependency_map_discovery = None + self.static_discovery = None + self.sql_discovery = None + self.web_app_discovery = None + self.oracle_discovery = None + self.spring_boot_discovery = None + self.iis_discovery = None + self.tomcat_discovery = None + self.apps_and_roles = None + self.product_support_status = product_support_status + self.number_of_processor_core = number_of_processor_core + self.allocated_memory_in_mb = allocated_memory_in_mb + self.operating_system_details = operating_system_details + self.bios_serial_number = bios_serial_number + self.bios_guid = bios_guid + self.is_deleted = None + self.created_timestamp = None + self.tags = tags + self.updated_timestamp = None + self.provisioning_state = provisioning_state + + +class ServerRunAsAccount(ProxyResource): + """A run as account resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "RunAsAccountProperties"}, + } + + def __init__(self, *, properties: Optional["_models.RunAsAccountProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ServerRunAsAccountListResult(_serialization.Model): + """The response of a ServerRunAsAccount list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The ServerRunAsAccount items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.ServerRunAsAccount] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ServerRunAsAccount]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.ServerRunAsAccount"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The ServerRunAsAccount items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.ServerRunAsAccount] + :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 ServerSiteResource(TrackedResource): + """A ServerSiteResource. + + 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: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.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: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SitesProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + "location": {"required": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "tags": {"key": "tags", "type": "{str}"}, + "location": {"key": "location", "type": "str"}, + "properties": {"key": "properties", "type": "SitesProperties"}, + } + + def __init__( + self, + *, + location: str, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.SitesProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword location: The geo-location where the resource lives. Required. + :paramtype location: str + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SitesProperties + """ + super().__init__(tags=tags, location=location, **kwargs) + self.properties = properties + + +class ServerSiteResourceListResult(_serialization.Model): + """The response of a ServerSiteResource list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The ServerSiteResource items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.ServerSiteResource] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ServerSiteResource]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.ServerSiteResource"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The ServerSiteResource items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.ServerSiteResource] + :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 ServerSiteResourceUpdate(_serialization.Model): + """The type used for update operations of the ServerSiteResource. + + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar properties: The updatable properties of the ServerSiteResource. + :vartype properties: ~azure.mgmt.migrate.models.ServerSiteResourceUpdateProperties + """ + + _attribute_map = { + "tags": {"key": "tags", "type": "{str}"}, + "properties": {"key": "properties", "type": "ServerSiteResourceUpdateProperties"}, + } + + def __init__( + self, + *, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.ServerSiteResourceUpdateProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword properties: The updatable properties of the ServerSiteResource. + :paramtype properties: ~azure.mgmt.migrate.models.ServerSiteResourceUpdateProperties + """ + super().__init__(**kwargs) + self.tags = tags + self.properties = properties + + +class ServerSiteResourceUpdateProperties(_serialization.Model): + """The updatable properties of the ServerSiteResource. + + :ivar service_principal_identity_details: Gets or sets the service principal identity details + used by agent for + communication + to the service. + :vartype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :ivar agent_details: Gets or sets the on-premises agent details. + :vartype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + :ivar discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :vartype discovery_solution_id: str + """ + + _attribute_map = { + "service_principal_identity_details": {"key": "servicePrincipalIdentityDetails", "type": "SiteSpnProperties"}, + "agent_details": {"key": "agentDetails", "type": "SiteAgentProperties"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "discovery_solution_id": {"key": "discoverySolutionId", "type": "str"}, + } + + def __init__( + self, + *, + service_principal_identity_details: Optional["_models.SiteSpnProperties"] = None, + agent_details: Optional["_models.SiteAgentProperties"] = None, + appliance_name: Optional[str] = None, + discovery_solution_id: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword service_principal_identity_details: Gets or sets the service principal identity + details used by agent for + communication + to the service. + :paramtype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :keyword agent_details: Gets or sets the on-premises agent details. + :paramtype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + :keyword discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :paramtype discovery_solution_id: str + """ + super().__init__(**kwargs) + self.service_principal_identity_details = service_principal_identity_details + self.agent_details = agent_details + self.appliance_name = appliance_name + self.discovery_solution_id = discovery_solution_id + + +class ServerSiteUsage(_serialization.Model): + """Server site usage. + + :ivar run_as_account_count: Gets or sets the number of run as accounts in the site. + :vartype run_as_account_count: int + :ivar server_count: Gets or sets the number of servers part of the site. + :vartype server_count: int + """ + + _attribute_map = { + "run_as_account_count": {"key": "runAsAccountCount", "type": "int"}, + "server_count": {"key": "serverCount", "type": "int"}, + } + + def __init__( + self, *, run_as_account_count: Optional[int] = None, server_count: Optional[int] = None, **kwargs: Any + ) -> None: + """ + :keyword run_as_account_count: Gets or sets the number of run as accounts in the site. + :paramtype run_as_account_count: int + :keyword server_count: Gets or sets the number of servers part of the site. + :paramtype server_count: int + """ + super().__init__(**kwargs) + self.run_as_account_count = run_as_account_count + self.server_count = server_count + + +class ServerSiteUsageResponse(_serialization.Model): + """The properties of ServerSiteUsageResponse. + + All required parameters must be populated in order to send to server. + + :ivar run_as_accounts: Gets or sets the number of run as accounts in the site. Required. + :vartype run_as_accounts: int + :ivar server_count: Gets or sets the number of servers part of the site. Required. + :vartype server_count: int + """ + + _validation = { + "run_as_accounts": {"required": True}, + "server_count": {"required": True}, + } + + _attribute_map = { + "run_as_accounts": {"key": "runAsAccounts", "type": "int"}, + "server_count": {"key": "serverCount", "type": "int"}, + } + + def __init__(self, *, run_as_accounts: int, server_count: int, **kwargs: Any) -> None: + """ + :keyword run_as_accounts: Gets or sets the number of run as accounts in the site. Required. + :paramtype run_as_accounts: int + :keyword server_count: Gets or sets the number of servers part of the site. Required. + :paramtype server_count: int + """ + super().__init__(**kwargs) + self.run_as_accounts = run_as_accounts + self.server_count = server_count + + +class ServerSoftwareInventory(ProxyResource): + """An software inventory resource belonging to a server resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.MachineSoftwareInventoryProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "MachineSoftwareInventoryProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.MachineSoftwareInventoryProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.MachineSoftwareInventoryProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ServerSoftwareInventoryListResult(_serialization.Model): + """The response of a ServerSoftwareInventory list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The ServerSoftwareInventory items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.ServerSoftwareInventory] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[ServerSoftwareInventory]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.ServerSoftwareInventory"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The ServerSoftwareInventory items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.ServerSoftwareInventory] + :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 ServerUpdate(_serialization.Model): + """The type used for update operations of the Server. + + :ivar properties: The updatable properties of the Server. + :vartype properties: ~azure.mgmt.migrate.models.ServerUpdateProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "ServerUpdateProperties"}, + } + + def __init__(self, *, properties: Optional["_models.ServerUpdateProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The updatable properties of the Server. + :paramtype properties: ~azure.mgmt.migrate.models.ServerUpdateProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class ServerUpdateProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """The updatable properties of the Server. + + :ivar fqdn: Gets or sets the FQDN/IPAddress of the server. + :vartype fqdn: str + :ivar network_adapters: Gets or sets the network adapters of the server. + :vartype network_adapters: list[~azure.mgmt.migrate.models.ServerNetworkAdapter] + :ivar hydrated_fqdn: Gets or sets the FQDN of machine which can be changed. + :vartype hydrated_fqdn: str + :ivar disks: Gets or sets the disk details of server. + :vartype disks: list[~azure.mgmt.migrate.models.ServerDisk] + :ivar validation_required: Gets or sets the validation required for the machine. + :vartype validation_required: str + :ivar firmware: Gets or sets the firmware. + :vartype firmware: str + :ivar run_as_account_id: Gets or sets the run as account ID of the machine. + :vartype run_as_account_id: str + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :vartype number_of_processor_core: int + :ivar allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :vartype allocated_memory_in_mb: float + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :ivar bios_serial_number: Gets or sets the Machine BIOS serial number. + :vartype bios_serial_number: str + :ivar bios_guid: Gets or sets the BIOS GUID. + :vartype bios_guid: str + :ivar tags: Gets or sets tags on the Server machine. + :vartype tags: dict[str, JSON] + :ivar provisioning_state: Provisioning state of the server resource. Known values are: + "Created", "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _attribute_map = { + "fqdn": {"key": "fqdn", "type": "str"}, + "network_adapters": {"key": "networkAdapters", "type": "[ServerNetworkAdapter]"}, + "hydrated_fqdn": {"key": "hydratedFqdn", "type": "str"}, + "disks": {"key": "disks", "type": "[ServerDisk]"}, + "validation_required": {"key": "validationRequired", "type": "str"}, + "firmware": {"key": "firmware", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "number_of_processor_core": {"key": "numberOfProcessorCore", "type": "int"}, + "allocated_memory_in_mb": {"key": "allocatedMemoryInMb", "type": "float"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "OperatingSystem"}, + "bios_serial_number": {"key": "biosSerialNumber", "type": "str"}, + "bios_guid": {"key": "biosGuid", "type": "str"}, + "tags": {"key": "tags", "type": "{object}"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + fqdn: Optional[str] = None, + network_adapters: Optional[List["_models.ServerNetworkAdapter"]] = None, + hydrated_fqdn: Optional[str] = None, + disks: Optional[List["_models.ServerDisk"]] = None, + validation_required: Optional[str] = None, + firmware: Optional[str] = None, + run_as_account_id: Optional[str] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + number_of_processor_core: Optional[int] = None, + allocated_memory_in_mb: Optional[float] = None, + operating_system_details: Optional["_models.OperatingSystem"] = None, + bios_serial_number: Optional[str] = None, + bios_guid: Optional[str] = None, + tags: Optional[Dict[str, JSON]] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword fqdn: Gets or sets the FQDN/IPAddress of the server. + :paramtype fqdn: str + :keyword network_adapters: Gets or sets the network adapters of the server. + :paramtype network_adapters: list[~azure.mgmt.migrate.models.ServerNetworkAdapter] + :keyword hydrated_fqdn: Gets or sets the FQDN of machine which can be changed. + :paramtype hydrated_fqdn: str + :keyword disks: Gets or sets the disk details of server. + :paramtype disks: list[~azure.mgmt.migrate.models.ServerDisk] + :keyword validation_required: Gets or sets the validation required for the machine. + :paramtype validation_required: str + :keyword firmware: Gets or sets the firmware. + :paramtype firmware: str + :keyword run_as_account_id: Gets or sets the run as account ID of the machine. + :paramtype run_as_account_id: str + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :paramtype number_of_processor_core: int + :keyword allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :paramtype allocated_memory_in_mb: float + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :keyword bios_serial_number: Gets or sets the Machine BIOS serial number. + :paramtype bios_serial_number: str + :keyword bios_guid: Gets or sets the BIOS GUID. + :paramtype bios_guid: str + :keyword tags: Gets or sets tags on the Server machine. + :paramtype tags: dict[str, JSON] + :keyword provisioning_state: Provisioning state of the server resource. Known values are: + "Created", "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.fqdn = fqdn + self.network_adapters = network_adapters + self.hydrated_fqdn = hydrated_fqdn + self.disks = disks + self.validation_required = validation_required + self.firmware = firmware + self.run_as_account_id = run_as_account_id + self.product_support_status = product_support_status + self.number_of_processor_core = number_of_processor_core + self.allocated_memory_in_mb = allocated_memory_in_mb + self.operating_system_details = operating_system_details + self.bios_serial_number = bios_serial_number + self.bios_guid = bios_guid + self.tags = tags + self.provisioning_state = provisioning_state + + +class SharePointServer(_serialization.Model): + """SharePointServer in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar product_name: Gets or sets ProductName of the SharePointServer. + :vartype product_name: str + :ivar is_enterprise: Gets or sets a value indicating whether the SharePointServer is + Enterprise. + :vartype is_enterprise: bool + :ivar status: Gets or sets Status of the SharePointServer. + :vartype status: str + :ivar version: Gets or sets Version of the SharePointServer. + :vartype version: str + """ + + _validation = { + "product_name": {"readonly": True}, + "is_enterprise": {"readonly": True}, + "status": {"readonly": True}, + "version": {"readonly": True}, + } + + _attribute_map = { + "product_name": {"key": "productName", "type": "str"}, + "is_enterprise": {"key": "isEnterprise", "type": "bool"}, + "status": {"key": "status", "type": "str"}, + "version": {"key": "version", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.product_name = None + self.is_enterprise = None + self.status = None + self.version = None + + +class SiteAgentProperties(_serialization.Model): + """Class for site agent properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Gets the ID of the agent. + :vartype id: str + :ivar version: Gets the version of the agent. + :vartype version: str + :ivar last_heart_beat_utc: Gets the last heartbeat time of the agent in UTC. + :vartype last_heart_beat_utc: ~datetime.datetime + :ivar key_vault_uri: Gets or sets the key vault URI. + :vartype key_vault_uri: str + :ivar key_vault_id: Gets or sets the key vault ARM Id. + :vartype key_vault_id: str + """ + + _validation = { + "id": {"readonly": True}, + "version": {"readonly": True}, + "last_heart_beat_utc": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "last_heart_beat_utc": {"key": "lastHeartBeatUtc", "type": "iso-8601"}, + "key_vault_uri": {"key": "keyVaultUri", "type": "str"}, + "key_vault_id": {"key": "keyVaultId", "type": "str"}, + } + + def __init__( + self, *, key_vault_uri: Optional[str] = None, key_vault_id: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword key_vault_uri: Gets or sets the key vault URI. + :paramtype key_vault_uri: str + :keyword key_vault_id: Gets or sets the key vault ARM Id. + :paramtype key_vault_id: str + """ + super().__init__(**kwargs) + self.id = None + self.version = None + self.last_heart_beat_utc = None + self.key_vault_uri = key_vault_uri + self.key_vault_id = key_vault_id + + +class SiteApplianceProperties(_serialization.Model): + """Class for site appliance properties. + + :ivar service_principal_identity_details: Gets or sets the service principal identity details + used by agent for communication to the service. + :vartype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :ivar agent_details: Gets or sets the on-premises agent details. + :vartype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + """ + + _attribute_map = { + "service_principal_identity_details": {"key": "servicePrincipalIdentityDetails", "type": "SiteSpnProperties"}, + "agent_details": {"key": "agentDetails", "type": "SiteAgentProperties"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + } + + def __init__( + self, + *, + service_principal_identity_details: Optional["_models.SiteSpnProperties"] = None, + agent_details: Optional["_models.SiteAgentProperties"] = None, + appliance_name: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword service_principal_identity_details: Gets or sets the service principal identity + details used by agent for communication to the service. + :paramtype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :keyword agent_details: Gets or sets the on-premises agent details. + :paramtype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + """ + super().__init__(**kwargs) + self.service_principal_identity_details = service_principal_identity_details + self.agent_details = agent_details + self.appliance_name = appliance_name + + +class SiteErrorSummary(_serialization.Model): + """Site error summary. + + All required parameters must be populated in order to send to server. + + :ivar appliance_name: Appliance Name. Required. + :vartype appliance_name: str + :ivar discovery_scope_error_summaries: Gets the error message. Required. + :vartype discovery_scope_error_summaries: ~azure.mgmt.migrate.models.DiscoveryScopeErrorSummary + :ivar next_link: The link to fetch more models. + :vartype next_link: str + """ + + _validation = { + "appliance_name": {"required": True}, + "discovery_scope_error_summaries": {"required": True}, + } + + _attribute_map = { + "appliance_name": {"key": "applianceName", "type": "str"}, + "discovery_scope_error_summaries": { + "key": "discoveryScopeErrorSummaries", + "type": "DiscoveryScopeErrorSummary", + }, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, + *, + appliance_name: str, + discovery_scope_error_summaries: "_models.DiscoveryScopeErrorSummary", + next_link: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword appliance_name: Appliance Name. Required. + :paramtype appliance_name: str + :keyword discovery_scope_error_summaries: Gets the error message. Required. + :paramtype discovery_scope_error_summaries: + ~azure.mgmt.migrate.models.DiscoveryScopeErrorSummary + :keyword next_link: The link to fetch more models. + :paramtype next_link: str + """ + super().__init__(**kwargs) + self.appliance_name = appliance_name + self.discovery_scope_error_summaries = discovery_scope_error_summaries + self.next_link = next_link + + +class SiteHealthSummary(_serialization.Model): # pylint: disable=too-many-instance-attributes + """The properties of ServerSiteResource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar appliance_name: Gets the appliance name. + :vartype appliance_name: str + :ivar error_message: Gets the error message. + :vartype error_message: str + :ivar summary_message: Gets the summary message. + :vartype summary_message: str + :ivar error_id: Gets the error Id. + :vartype error_id: int + :ivar error_code: Gets the error code. + :vartype error_code: str + :ivar affected_objects_count: Gets or sets the count of affected objects. + :vartype affected_objects_count: int + :ivar hit_count: Gets or sets the hit count of the error. + :vartype hit_count: int + :ivar severity: Gets the severity of error. + :vartype severity: str + :ivar remediation_guidance: Gets the remediation guidance. + :vartype remediation_guidance: str + :ivar affected_resource_type: Gets the affected resource type. + :vartype affected_resource_type: str + :ivar affected_resources: Gets or sets the affected resources. + :vartype affected_resources: list[str] + :ivar fabric_layout_update_sources: Gets or sets sources of the exception. + :vartype fabric_layout_update_sources: list[str or + ~azure.mgmt.migrate.models.SiteHealthSummaryFabricLayoutUpdateSourcesItem] + """ + + _validation = { + "appliance_name": {"readonly": True}, + "error_message": {"readonly": True}, + "summary_message": {"readonly": True}, + "error_id": {"readonly": True}, + "error_code": {"readonly": True}, + "severity": {"readonly": True}, + "remediation_guidance": {"readonly": True}, + "affected_resource_type": {"readonly": True}, + } + + _attribute_map = { + "appliance_name": {"key": "applianceName", "type": "str"}, + "error_message": {"key": "errorMessage", "type": "str"}, + "summary_message": {"key": "summaryMessage", "type": "str"}, + "error_id": {"key": "errorId", "type": "int"}, + "error_code": {"key": "errorCode", "type": "str"}, + "affected_objects_count": {"key": "affectedObjectsCount", "type": "int"}, + "hit_count": {"key": "hitCount", "type": "int"}, + "severity": {"key": "severity", "type": "str"}, + "remediation_guidance": {"key": "remediationGuidance", "type": "str"}, + "affected_resource_type": {"key": "affectedResourceType", "type": "str"}, + "affected_resources": {"key": "affectedResources", "type": "[str]"}, + "fabric_layout_update_sources": {"key": "fabricLayoutUpdateSources", "type": "[str]"}, + } + + def __init__( + self, + *, + affected_objects_count: Optional[int] = None, + hit_count: Optional[int] = None, + affected_resources: Optional[List[str]] = None, + fabric_layout_update_sources: Optional[ + List[Union[str, "_models.SiteHealthSummaryFabricLayoutUpdateSourcesItem"]] + ] = None, + **kwargs: Any + ) -> None: + """ + :keyword affected_objects_count: Gets or sets the count of affected objects. + :paramtype affected_objects_count: int + :keyword hit_count: Gets or sets the hit count of the error. + :paramtype hit_count: int + :keyword affected_resources: Gets or sets the affected resources. + :paramtype affected_resources: list[str] + :keyword fabric_layout_update_sources: Gets or sets sources of the exception. + :paramtype fabric_layout_update_sources: list[str or + ~azure.mgmt.migrate.models.SiteHealthSummaryFabricLayoutUpdateSourcesItem] + """ + super().__init__(**kwargs) + self.appliance_name = None + self.error_message = None + self.summary_message = None + self.error_id = None + self.error_code = None + self.affected_objects_count = affected_objects_count + self.hit_count = hit_count + self.severity = None + self.remediation_guidance = None + self.affected_resource_type = None + self.affected_resources = affected_resources + self.fabric_layout_update_sources = fabric_layout_update_sources + + +class SiteHealthSummaryCollection(_serialization.Model): + """Collection of SiteHealthSummary. + + 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 value: Gets the list of SiteHealthSummary. Required. + :vartype value: list[~azure.mgmt.migrate.models.SiteHealthSummary] + :ivar next_link: Gets the value of next link. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + "next_link": {"readonly": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SiteHealthSummary]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.SiteHealthSummary"], **kwargs: Any) -> None: + """ + :keyword value: Gets the list of SiteHealthSummary. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SiteHealthSummary] + """ + super().__init__(**kwargs) + self.value = value + self.next_link = None + + +class SiteProperties(_serialization.Model): + """The properties of VMwareSiteResource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar master_site_id: Gets the Master Site this site is linked to. + :vartype master_site_id: str + :ivar service_principal_identity_details: Gets or sets the service principal identity details + used by agent for + communication + to the service. + :vartype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :ivar agent_details: Gets or sets the on-premises agent details. + :vartype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + :ivar discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :vartype discovery_solution_id: str + :ivar service_endpoint: Gets the service endpoint. + :vartype service_endpoint: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "master_site_id": {"readonly": True}, + "service_endpoint": {"readonly": True}, + } + + _attribute_map = { + "master_site_id": {"key": "masterSiteId", "type": "str"}, + "service_principal_identity_details": {"key": "servicePrincipalIdentityDetails", "type": "SiteSpnProperties"}, + "agent_details": {"key": "agentDetails", "type": "SiteAgentProperties"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "discovery_solution_id": {"key": "discoverySolutionId", "type": "str"}, + "service_endpoint": {"key": "serviceEndpoint", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + service_principal_identity_details: Optional["_models.SiteSpnProperties"] = None, + agent_details: Optional["_models.SiteAgentProperties"] = None, + appliance_name: Optional[str] = None, + discovery_solution_id: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword service_principal_identity_details: Gets or sets the service principal identity + details used by agent for + communication + to the service. + :paramtype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :keyword agent_details: Gets or sets the on-premises agent details. + :paramtype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + :keyword discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :paramtype discovery_solution_id: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.master_site_id = None + self.service_principal_identity_details = service_principal_identity_details + self.agent_details = agent_details + self.appliance_name = appliance_name + self.discovery_solution_id = discovery_solution_id + self.service_endpoint = None + self.provisioning_state = provisioning_state + + +class SiteSpnProperties(_serialization.Model): + """Class for site properties. + + :ivar tenant_id: Gets or sets the tenant Id for the service principal with which the + on-premise + management/data plane components would communicate with + our Azure services. + :vartype tenant_id: str + :ivar application_id: Gets or sets the application/client Id for the service principal with + which + the + on-premise management/data plane components would communicate + with our Azure + services. + :vartype application_id: str + :ivar object_id: Gets or sets the object Id of the service principal with which the on-premise + + .. code-block:: + + management/data plane components would communicate with our Azure + + services. + :vartype object_id: str + :ivar audience: Gets or sets the intended audience for the service principal. + :vartype audience: str + :ivar aad_authority: Gets or sets the AAD Authority URL which was used to request the token for + the + service principal. + :vartype aad_authority: str + :ivar raw_cert_data: Gets or sets the raw certificate data for building certificate expiry + flows. + :vartype raw_cert_data: str + """ + + _attribute_map = { + "tenant_id": {"key": "tenantId", "type": "str"}, + "application_id": {"key": "applicationId", "type": "str"}, + "object_id": {"key": "objectId", "type": "str"}, + "audience": {"key": "audience", "type": "str"}, + "aad_authority": {"key": "aadAuthority", "type": "str"}, + "raw_cert_data": {"key": "rawCertData", "type": "str"}, + } + + def __init__( + self, + *, + tenant_id: Optional[str] = None, + application_id: Optional[str] = None, + object_id: Optional[str] = None, + audience: Optional[str] = None, + aad_authority: Optional[str] = None, + raw_cert_data: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword tenant_id: Gets or sets the tenant Id for the service principal with which the + on-premise + management/data plane components would communicate with + our Azure services. + :paramtype tenant_id: str + :keyword application_id: Gets or sets the application/client Id for the service principal with + which + the + on-premise management/data plane components would communicate + with our Azure + services. + :paramtype application_id: str + :keyword object_id: Gets or sets the object Id of the service principal with which the + on-premise + + .. code-block:: + + management/data plane components would communicate with our Azure + + services. + :paramtype object_id: str + :keyword audience: Gets or sets the intended audience for the service principal. + :paramtype audience: str + :keyword aad_authority: Gets or sets the AAD Authority URL which was used to request the token + for + the + service principal. + :paramtype aad_authority: str + :keyword raw_cert_data: Gets or sets the raw certificate data for building certificate expiry + flows. + :paramtype raw_cert_data: str + """ + super().__init__(**kwargs) + self.tenant_id = tenant_id + self.application_id = application_id + self.object_id = object_id + self.audience = audience + self.aad_authority = aad_authority + self.raw_cert_data = raw_cert_data + + +class SitesProperties(_serialization.Model): + """The properties of SiteResource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar master_site_id: Gets the Master Site this site is linked to. + :vartype master_site_id: str + :ivar service_principal_identity_details: Gets or sets the service principal identity details + used by agent for + communication + to the service. + :vartype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :ivar agent_details: Gets or sets the on-premises agent details. + :vartype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + :ivar discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :vartype discovery_solution_id: str + :ivar service_endpoint: Gets the service endpoint. + :vartype service_endpoint: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "master_site_id": {"readonly": True}, + "service_endpoint": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "master_site_id": {"key": "masterSiteId", "type": "str"}, + "service_principal_identity_details": {"key": "servicePrincipalIdentityDetails", "type": "SiteSpnProperties"}, + "agent_details": {"key": "agentDetails", "type": "SiteAgentProperties"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "discovery_solution_id": {"key": "discoverySolutionId", "type": "str"}, + "service_endpoint": {"key": "serviceEndpoint", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + service_principal_identity_details: Optional["_models.SiteSpnProperties"] = None, + agent_details: Optional["_models.SiteAgentProperties"] = None, + appliance_name: Optional[str] = None, + discovery_solution_id: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword service_principal_identity_details: Gets or sets the service principal identity + details used by agent for + communication + to the service. + :paramtype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :keyword agent_details: Gets or sets the on-premises agent details. + :paramtype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + :keyword discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :paramtype discovery_solution_id: str + """ + super().__init__(**kwargs) + self.master_site_id = None + self.service_principal_identity_details = service_principal_identity_details + self.agent_details = agent_details + self.appliance_name = appliance_name + self.discovery_solution_id = discovery_solution_id + self.service_endpoint = None + self.provisioning_state = None + + +class SpringBootDiscovery(_serialization.Model): + """Data related to a machine's spring boot discovery. + + :ivar total_instance_count: Gets or sets number of successfully discovered instances. + :vartype total_instance_count: int + :ivar total_application_count: Gets or sets number of successfully discovered applications. + :vartype total_application_count: int + :ivar shallow_discovery_status: Gets or sets status of partner shallow discovery. Known values + are: "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype shallow_discovery_status: str or ~azure.mgmt.migrate.models.ShallowDiscoveryStatus + :ivar discovery_scope_status: Gets or sets status of partner discovery. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype discovery_scope_status: str or ~azure.mgmt.migrate.models.DiscoveryScopeStatus + """ + + _attribute_map = { + "total_instance_count": {"key": "totalInstanceCount", "type": "int"}, + "total_application_count": {"key": "totalApplicationCount", "type": "int"}, + "shallow_discovery_status": {"key": "shallowDiscoveryStatus", "type": "str"}, + "discovery_scope_status": {"key": "discoveryScopeStatus", "type": "str"}, + } + + def __init__( + self, + *, + total_instance_count: Optional[int] = None, + total_application_count: Optional[int] = None, + shallow_discovery_status: Optional[Union[str, "_models.ShallowDiscoveryStatus"]] = None, + discovery_scope_status: Optional[Union[str, "_models.DiscoveryScopeStatus"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword total_instance_count: Gets or sets number of successfully discovered instances. + :paramtype total_instance_count: int + :keyword total_application_count: Gets or sets number of successfully discovered applications. + :paramtype total_application_count: int + :keyword shallow_discovery_status: Gets or sets status of partner shallow discovery. Known + values are: "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :paramtype shallow_discovery_status: str or ~azure.mgmt.migrate.models.ShallowDiscoveryStatus + :keyword discovery_scope_status: Gets or sets status of partner discovery. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :paramtype discovery_scope_status: str or ~azure.mgmt.migrate.models.DiscoveryScopeStatus + """ + super().__init__(**kwargs) + self.total_instance_count = total_instance_count + self.total_application_count = total_application_count + self.shallow_discovery_status = shallow_discovery_status + self.discovery_scope_status = discovery_scope_status + + +class SqlAvailabilityGroup(ProxyResource): + """Class representing the web model of SQL Availability Group. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SqlAvailabilityGroupProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "SqlAvailabilityGroupProperties"}, + } + + def __init__(self, *, properties: Optional["_models.SqlAvailabilityGroupProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SqlAvailabilityGroupProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlAvailabilityGroupListResult(_serialization.Model): + """The response of a SqlAvailabilityGroup list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The SqlAvailabilityGroup items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.SqlAvailabilityGroup] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SqlAvailabilityGroup]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.SqlAvailabilityGroup"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The SqlAvailabilityGroup items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SqlAvailabilityGroup] + :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 SqlAvailabilityGroupProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for SQL Server availability group properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar availability_group_name: Gets or sets the SQL Availability Group Name. + :vartype availability_group_name: str + :ivar availability_group_type: Gets or sets the type of availability group. + Currently there are two types of AGs Traditional and Distributed. Known values are: "Unknown", + "Traditional", and "Distributed". + :vartype availability_group_type: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupPropertiesAvailabilityGroupType + :ivar is_multi_sub_net: Gets or sets a value indicating whether AG is multi subnet or not. + :vartype is_multi_sub_net: bool + :ivar cluster_name: Gets or sets the Cluster name where AG is hosted. + :vartype cluster_name: str + :ivar availability_replicas: Gets the list of availability replica properties which together + forms this availability group. + :vartype availability_replicas: + list[~azure.mgmt.migrate.models.SqlAvailabilityReplicaProperties] + :ivar parent_replica_overview_list: Gets the parent availability replica overview if any. + This would be + set with details of parent AG and AR for cases where this availability group is + a part of a distributed AG. + Currently, we do not populate this + since discovery and linking of DAG(Distributed Availability Group) is not + implemented. + :vartype parent_replica_overview_list: + list[~azure.mgmt.migrate.models.SqlAvailabilityReplicaOverview] + :ivar is_part_of_distributed_availability_group: Gets or sets a value indicating whether this + Availability group is part of a + distributed AG. + :vartype is_part_of_distributed_availability_group: bool + :ivar is_deleted: Gets or sets a value indicating whether the entity is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets or sets the marking machine creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "availability_replicas": {"readonly": True}, + "parent_replica_overview_list": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "availability_group_name": {"key": "availabilityGroupName", "type": "str"}, + "availability_group_type": {"key": "availabilityGroupType", "type": "str"}, + "is_multi_sub_net": {"key": "isMultiSubNet", "type": "bool"}, + "cluster_name": {"key": "clusterName", "type": "str"}, + "availability_replicas": {"key": "availabilityReplicas", "type": "[SqlAvailabilityReplicaProperties]"}, + "parent_replica_overview_list": { + "key": "parentReplicaOverviewList", + "type": "[SqlAvailabilityReplicaOverview]", + }, + "is_part_of_distributed_availability_group": {"key": "isPartOfDistributedAvailabilityGroup", "type": "bool"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + availability_group_name: Optional[str] = None, + availability_group_type: Optional[ + Union[str, "_models.SqlAvailabilityGroupPropertiesAvailabilityGroupType"] + ] = None, + is_multi_sub_net: Optional[bool] = None, + cluster_name: Optional[str] = None, + is_part_of_distributed_availability_group: Optional[bool] = None, + is_deleted: Optional[bool] = None, + created_timestamp: Optional[str] = None, + updated_timestamp: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword availability_group_name: Gets or sets the SQL Availability Group Name. + :paramtype availability_group_name: str + :keyword availability_group_type: Gets or sets the type of availability group. + Currently there are two types of AGs Traditional and Distributed. Known values are: "Unknown", + "Traditional", and "Distributed". + :paramtype availability_group_type: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupPropertiesAvailabilityGroupType + :keyword is_multi_sub_net: Gets or sets a value indicating whether AG is multi subnet or not. + :paramtype is_multi_sub_net: bool + :keyword cluster_name: Gets or sets the Cluster name where AG is hosted. + :paramtype cluster_name: str + :keyword is_part_of_distributed_availability_group: Gets or sets a value indicating whether + this Availability group is part of a + distributed AG. + :paramtype is_part_of_distributed_availability_group: bool + :keyword is_deleted: Gets or sets a value indicating whether the entity is deleted. + :paramtype is_deleted: bool + :keyword created_timestamp: Gets or sets the marking machine creation. + :paramtype created_timestamp: str + :keyword updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :paramtype updated_timestamp: str + """ + super().__init__(**kwargs) + self.availability_group_name = availability_group_name + self.availability_group_type = availability_group_type + self.is_multi_sub_net = is_multi_sub_net + self.cluster_name = cluster_name + self.availability_replicas = None + self.parent_replica_overview_list = None + self.is_part_of_distributed_availability_group = is_part_of_distributed_availability_group + self.is_deleted = is_deleted + self.created_timestamp = created_timestamp + self.updated_timestamp = updated_timestamp + self.provisioning_state = None + + +class SqlAvailabilityGroupReplicaInfo(_serialization.Model): + """Class which represents the SQL availability replica properties of type AG. + + :ivar cluster_name: Gets or sets the name of the cluster on which this replica is hosted. + :vartype cluster_name: str + :ivar availability_group_name: Gets or sets the name of the availability replica. + :vartype availability_group_name: str + :ivar availability_group_arm_id: Gets or sets the AG ARM ID which are part of this Replica. + :vartype availability_group_arm_id: str + """ + + _attribute_map = { + "cluster_name": {"key": "clusterName", "type": "str"}, + "availability_group_name": {"key": "availabilityGroupName", "type": "str"}, + "availability_group_arm_id": {"key": "availabilityGroupArmId", "type": "str"}, + } + + def __init__( + self, + *, + cluster_name: Optional[str] = None, + availability_group_name: Optional[str] = None, + availability_group_arm_id: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword cluster_name: Gets or sets the name of the cluster on which this replica is hosted. + :paramtype cluster_name: str + :keyword availability_group_name: Gets or sets the name of the availability replica. + :paramtype availability_group_name: str + :keyword availability_group_arm_id: Gets or sets the AG ARM ID which are part of this Replica. + :paramtype availability_group_arm_id: str + """ + super().__init__(**kwargs) + self.cluster_name = cluster_name + self.availability_group_name = availability_group_name + self.availability_group_arm_id = availability_group_arm_id + + +class SqlAvailabilityReplicaOverview(_serialization.Model): + """Web model for SQL replica overview. + + :ivar replica_state: Gets or sets the state of the availability replica. Known values are: + "Unknown", "Primary", and "Secondary". + :vartype replica_state: str or + ~azure.mgmt.migrate.models.SqlAvailabilityReplicaOverviewReplicaState + :ivar availability_replica_id: Gets or sets the SQL Availability Replica Id within the + Availability Group this + database is a part of. + :vartype availability_replica_id: str + :ivar availability_group_arm_id: Gets or sets the Availability Group Id in which this database + participates if + any. It would be set if the database has + {Microsoft.Azure.FDS.CosmosDB.SqlDatabaseEntity.IsDatabaseHighlyAvailable} has + a value "true". + :vartype availability_group_arm_id: str + :ivar availability_group_name: Gets or sets the SQL Availability group name. + :vartype availability_group_name: str + """ + + _attribute_map = { + "replica_state": {"key": "replicaState", "type": "str"}, + "availability_replica_id": {"key": "availabilityReplicaId", "type": "str"}, + "availability_group_arm_id": {"key": "availabilityGroupArmId", "type": "str"}, + "availability_group_name": {"key": "availabilityGroupName", "type": "str"}, + } + + def __init__( + self, + *, + replica_state: Optional[Union[str, "_models.SqlAvailabilityReplicaOverviewReplicaState"]] = None, + availability_replica_id: Optional[str] = None, + availability_group_arm_id: Optional[str] = None, + availability_group_name: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword replica_state: Gets or sets the state of the availability replica. Known values are: + "Unknown", "Primary", and "Secondary". + :paramtype replica_state: str or + ~azure.mgmt.migrate.models.SqlAvailabilityReplicaOverviewReplicaState + :keyword availability_replica_id: Gets or sets the SQL Availability Replica Id within the + Availability Group this + database is a part of. + :paramtype availability_replica_id: str + :keyword availability_group_arm_id: Gets or sets the Availability Group Id in which this + database participates if + any. It would be set if the database has + {Microsoft.Azure.FDS.CosmosDB.SqlDatabaseEntity.IsDatabaseHighlyAvailable} has + a value "true". + :paramtype availability_group_arm_id: str + :keyword availability_group_name: Gets or sets the SQL Availability group name. + :paramtype availability_group_name: str + """ + super().__init__(**kwargs) + self.replica_state = replica_state + self.availability_replica_id = availability_replica_id + self.availability_group_arm_id = availability_group_arm_id + self.availability_group_name = availability_group_name + + +class SqlAvailabilityReplicaProperties(_serialization.Model): + """Class for SQL Server availability replica properties. + + :ivar availability_replica_name: Gets or sets the SQL Availability Replica Name. + :vartype availability_replica_name: str + :ivar availability_replica_id: Gets or sets the SQL Availability Replica Id. + :vartype availability_replica_id: str + :ivar replica_type: Gets or sets the replica type of the availability group. Known values are: + "Unknown", "DatabaseReplica", and "AvailabilityGroupReplica". + :vartype replica_type: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaType + :ivar replica_state: Gets or sets the state of the availability replica. Known values are: + "Unknown", "Primary", and "Secondary". + :vartype replica_state: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaState + :ivar replica_sync_status: Gets or sets the synchronization status of the replica. Known values + are: "Unknown", "Synchronized", and "Unsynchronized". + :vartype replica_sync_status: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSyncStatus + :ivar replica_commit_mode: Gets or sets the commit mode of the replica. Known values are: + "Unknown", "Synchronous", and "Asynchronous". + :vartype replica_commit_mode: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaCommitMode + :ivar replica_read_mode: Gets or sets the read mode of the replica. Known values are: + "Unknown", "None", "ReadOnly", and "ReadWrite". + :vartype replica_read_mode: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaReadMode + :ivar replica_seed_mode: Gets or sets the seed mode of the replica. Known values are: + "Unknown", "Manual", and "Automatic". + :vartype replica_seed_mode: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSeedMode + :ivar sql_database_replica_info: Gets or sets the Availability Replica object if Replica is of + type database. + This is null in case Replica is + of type AG. This is to be populated + in case of normal AG. + :vartype sql_database_replica_info: ~azure.mgmt.migrate.models.SqlDatabaseReplicaInfo + :ivar sql_availability_group_replica_info: Gets or sets the Availability Replica object if + Replica is of type AG. This is + null in case Replica is + of type database. This is to be populated + in case AG is of type DAG. + :vartype sql_availability_group_replica_info: + ~azure.mgmt.migrate.models.SqlAvailabilityGroupReplicaInfo + """ + + _attribute_map = { + "availability_replica_name": {"key": "availabilityReplicaName", "type": "str"}, + "availability_replica_id": {"key": "availabilityReplicaId", "type": "str"}, + "replica_type": {"key": "replicaType", "type": "str"}, + "replica_state": {"key": "replicaState", "type": "str"}, + "replica_sync_status": {"key": "replicaSyncStatus", "type": "str"}, + "replica_commit_mode": {"key": "replicaCommitMode", "type": "str"}, + "replica_read_mode": {"key": "replicaReadMode", "type": "str"}, + "replica_seed_mode": {"key": "replicaSeedMode", "type": "str"}, + "sql_database_replica_info": {"key": "sqlDatabaseReplicaInfo", "type": "SqlDatabaseReplicaInfo"}, + "sql_availability_group_replica_info": { + "key": "sqlAvailabilityGroupReplicaInfo", + "type": "SqlAvailabilityGroupReplicaInfo", + }, + } + + def __init__( + self, + *, + availability_replica_name: Optional[str] = None, + availability_replica_id: Optional[str] = None, + replica_type: Optional[ + Union[str, "_models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaType"] + ] = None, + replica_state: Optional[ + Union[str, "_models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaState"] + ] = None, + replica_sync_status: Optional[ + Union[str, "_models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSyncStatus"] + ] = None, + replica_commit_mode: Optional[ + Union[str, "_models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaCommitMode"] + ] = None, + replica_read_mode: Optional[ + Union[str, "_models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaReadMode"] + ] = None, + replica_seed_mode: Optional[ + Union[str, "_models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSeedMode"] + ] = None, + sql_database_replica_info: Optional["_models.SqlDatabaseReplicaInfo"] = None, + sql_availability_group_replica_info: Optional["_models.SqlAvailabilityGroupReplicaInfo"] = None, + **kwargs: Any + ) -> None: + """ + :keyword availability_replica_name: Gets or sets the SQL Availability Replica Name. + :paramtype availability_replica_name: str + :keyword availability_replica_id: Gets or sets the SQL Availability Replica Id. + :paramtype availability_replica_id: str + :keyword replica_type: Gets or sets the replica type of the availability group. Known values + are: "Unknown", "DatabaseReplica", and "AvailabilityGroupReplica". + :paramtype replica_type: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaType + :keyword replica_state: Gets or sets the state of the availability replica. Known values are: + "Unknown", "Primary", and "Secondary". + :paramtype replica_state: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaState + :keyword replica_sync_status: Gets or sets the synchronization status of the replica. Known + values are: "Unknown", "Synchronized", and "Unsynchronized". + :paramtype replica_sync_status: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSyncStatus + :keyword replica_commit_mode: Gets or sets the commit mode of the replica. Known values are: + "Unknown", "Synchronous", and "Asynchronous". + :paramtype replica_commit_mode: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaCommitMode + :keyword replica_read_mode: Gets or sets the read mode of the replica. Known values are: + "Unknown", "None", "ReadOnly", and "ReadWrite". + :paramtype replica_read_mode: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaReadMode + :keyword replica_seed_mode: Gets or sets the seed mode of the replica. Known values are: + "Unknown", "Manual", and "Automatic". + :paramtype replica_seed_mode: str or + ~azure.mgmt.migrate.models.SqlAvailabilityGroupSqlAvailabilityReplicaPropertiesReplicaSeedMode + :keyword sql_database_replica_info: Gets or sets the Availability Replica object if Replica is + of type database. + This is null in case Replica is + of type AG. This is to be populated + in case of normal AG. + :paramtype sql_database_replica_info: ~azure.mgmt.migrate.models.SqlDatabaseReplicaInfo + :keyword sql_availability_group_replica_info: Gets or sets the Availability Replica object if + Replica is of type AG. This is + null in case Replica is + of type database. This is to be populated + in case AG is of type DAG. + :paramtype sql_availability_group_replica_info: + ~azure.mgmt.migrate.models.SqlAvailabilityGroupReplicaInfo + """ + super().__init__(**kwargs) + self.availability_replica_name = availability_replica_name + self.availability_replica_id = availability_replica_id + self.replica_type = replica_type + self.replica_state = replica_state + self.replica_sync_status = replica_sync_status + self.replica_commit_mode = replica_commit_mode + self.replica_read_mode = replica_read_mode + self.replica_seed_mode = replica_seed_mode + self.sql_database_replica_info = sql_database_replica_info + self.sql_availability_group_replica_info = sql_availability_group_replica_info + + +class SqlDatabasePropertiesV2(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for SQL Server database properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar parent_replica_overview: Gets or sets the Replica overview. + :vartype parent_replica_overview: ~azure.mgmt.migrate.models.SqlAvailabilityReplicaOverview + :ivar is_database_highly_available: Gets or sets a value indicating whether this database is a + part of an HA setup. + :vartype is_database_highly_available: bool + :ivar file_metadata_list: Gets the file metadata list. + :vartype file_metadata_list: list[~azure.mgmt.migrate.models.FileMetaData] + :ivar hostname: Gets or sets the hostname. + :vartype hostname: str + :ivar sql_server_name: Gets or sets the SQL Server Id in which the database resides. + :vartype sql_server_name: str + :ivar status: Gets or sets the status. + :vartype status: str + :ivar size_mb: Gets or sets the size in Mb. + :vartype size_mb: float + :ivar database_name: Gets or sets the SQL database name. + :vartype database_name: str + :ivar sql_server_arm_id: Gets or sets the SQL Server ARM Id in which the database resides. + :vartype sql_server_arm_id: str + :ivar compatibility_level: Gets or sets the database compatibility level. + :vartype compatibility_level: str + :ivar is_deleted: Gets or sets a value indicating whether the entity is deleted. + :vartype is_deleted: bool + :ivar errors: Gets the errors. + :vartype errors: list[~azure.mgmt.migrate.models.Errors] + :ivar created_timestamp: Gets or sets the marking machine creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "file_metadata_list": {"readonly": True}, + "errors": {"readonly": True}, + } + + _attribute_map = { + "parent_replica_overview": {"key": "parentReplicaOverview", "type": "SqlAvailabilityReplicaOverview"}, + "is_database_highly_available": {"key": "isDatabaseHighlyAvailable", "type": "bool"}, + "file_metadata_list": {"key": "fileMetadataList", "type": "[FileMetaData]"}, + "hostname": {"key": "hostname", "type": "str"}, + "sql_server_name": {"key": "sqlServerName", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "size_mb": {"key": "sizeMb", "type": "float"}, + "database_name": {"key": "databaseName", "type": "str"}, + "sql_server_arm_id": {"key": "sqlServerArmId", "type": "str"}, + "compatibility_level": {"key": "compatibilityLevel", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "errors": {"key": "errors", "type": "[Errors]"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + parent_replica_overview: Optional["_models.SqlAvailabilityReplicaOverview"] = None, + is_database_highly_available: Optional[bool] = None, + hostname: Optional[str] = None, + sql_server_name: Optional[str] = None, + status: Optional[str] = None, + size_mb: Optional[float] = None, + database_name: Optional[str] = None, + sql_server_arm_id: Optional[str] = None, + compatibility_level: Optional[str] = None, + is_deleted: Optional[bool] = None, + created_timestamp: Optional[str] = None, + updated_timestamp: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword parent_replica_overview: Gets or sets the Replica overview. + :paramtype parent_replica_overview: ~azure.mgmt.migrate.models.SqlAvailabilityReplicaOverview + :keyword is_database_highly_available: Gets or sets a value indicating whether this database is + a part of an HA setup. + :paramtype is_database_highly_available: bool + :keyword hostname: Gets or sets the hostname. + :paramtype hostname: str + :keyword sql_server_name: Gets or sets the SQL Server Id in which the database resides. + :paramtype sql_server_name: str + :keyword status: Gets or sets the status. + :paramtype status: str + :keyword size_mb: Gets or sets the size in Mb. + :paramtype size_mb: float + :keyword database_name: Gets or sets the SQL database name. + :paramtype database_name: str + :keyword sql_server_arm_id: Gets or sets the SQL Server ARM Id in which the database resides. + :paramtype sql_server_arm_id: str + :keyword compatibility_level: Gets or sets the database compatibility level. + :paramtype compatibility_level: str + :keyword is_deleted: Gets or sets a value indicating whether the entity is deleted. + :paramtype is_deleted: bool + :keyword created_timestamp: Gets or sets the marking machine creation. + :paramtype created_timestamp: str + :keyword updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :paramtype updated_timestamp: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.parent_replica_overview = parent_replica_overview + self.is_database_highly_available = is_database_highly_available + self.file_metadata_list = None + self.hostname = hostname + self.sql_server_name = sql_server_name + self.status = status + self.size_mb = size_mb + self.database_name = database_name + self.sql_server_arm_id = sql_server_arm_id + self.compatibility_level = compatibility_level + self.is_deleted = is_deleted + self.errors = None + self.created_timestamp = created_timestamp + self.updated_timestamp = updated_timestamp + self.provisioning_state = provisioning_state + + +class SqlDatabaseReplicaInfo(_serialization.Model): + """Class which represents the SQL availability replica properties of type + databases. + + :ivar host_name: Gets or sets the host name of the availability replica. + :vartype host_name: str + :ivar sql_server_name: Gets or sets the SQL Server name of the availability replica. + :vartype sql_server_name: str + :ivar sql_server_arm_id: Gets or sets the SQL Server name of the availability replica. + :vartype sql_server_arm_id: str + """ + + _attribute_map = { + "host_name": {"key": "hostName", "type": "str"}, + "sql_server_name": {"key": "sqlServerName", "type": "str"}, + "sql_server_arm_id": {"key": "sqlServerArmId", "type": "str"}, + } + + def __init__( + self, + *, + host_name: Optional[str] = None, + sql_server_name: Optional[str] = None, + sql_server_arm_id: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword host_name: Gets or sets the host name of the availability replica. + :paramtype host_name: str + :keyword sql_server_name: Gets or sets the SQL Server name of the availability replica. + :paramtype sql_server_name: str + :keyword sql_server_arm_id: Gets or sets the SQL Server name of the availability replica. + :paramtype sql_server_arm_id: str + """ + super().__init__(**kwargs) + self.host_name = host_name + self.sql_server_name = sql_server_name + self.sql_server_arm_id = sql_server_arm_id + + +class SqlDatabaseV2(ProxyResource): + """Class representing the web model of SQL Database. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SqlDatabasePropertiesV2 + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "SqlDatabasePropertiesV2"}, + } + + def __init__(self, *, properties: Optional["_models.SqlDatabasePropertiesV2"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SqlDatabasePropertiesV2 + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlDatabaseV2ListResult(_serialization.Model): + """The response of a SqlDatabaseV2 list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The SqlDatabaseV2 items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.SqlDatabaseV2] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SqlDatabaseV2]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.SqlDatabaseV2"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The SqlDatabaseV2 items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SqlDatabaseV2] + :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 SqlDiscovery(_serialization.Model): + """Contains data related SQL discovery. + + :ivar successfully_discovered_server_count: successfully Discovered ServerCount. + :vartype successfully_discovered_server_count: int + :ivar total_server_count: total Server Count. + :vartype total_server_count: int + :ivar sql_metadata_hydrated_run_as_account_id: sql Metadata Hydrated RunAsAccountId. + :vartype sql_metadata_hydrated_run_as_account_id: str + :ivar sql_metadata_discovery_pipe: sql Metadata DiscoveryPipe. Known values are: "Unknown", + "VMware", "PowerShell", "SSH", "CIM", and "Other". + :vartype sql_metadata_discovery_pipe: str or + ~azure.mgmt.migrate.models.SqlMetadataDiscoveryPipe + :ivar discovery_scope_status: discovery Scope Status. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype discovery_scope_status: str or ~azure.mgmt.migrate.models.SQLDiscoveryScopeStatus + """ + + _attribute_map = { + "successfully_discovered_server_count": {"key": "successfullyDiscoveredServerCount", "type": "int"}, + "total_server_count": {"key": "totalServerCount", "type": "int"}, + "sql_metadata_hydrated_run_as_account_id": {"key": "sqlMetadataHydratedRunAsAccountId", "type": "str"}, + "sql_metadata_discovery_pipe": {"key": "sqlMetadataDiscoveryPipe", "type": "str"}, + "discovery_scope_status": {"key": "discoveryScopeStatus", "type": "str"}, + } + + def __init__( + self, + *, + successfully_discovered_server_count: Optional[int] = None, + total_server_count: Optional[int] = None, + sql_metadata_hydrated_run_as_account_id: Optional[str] = None, + sql_metadata_discovery_pipe: Optional[Union[str, "_models.SqlMetadataDiscoveryPipe"]] = None, + discovery_scope_status: Optional[Union[str, "_models.SQLDiscoveryScopeStatus"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword successfully_discovered_server_count: successfully Discovered ServerCount. + :paramtype successfully_discovered_server_count: int + :keyword total_server_count: total Server Count. + :paramtype total_server_count: int + :keyword sql_metadata_hydrated_run_as_account_id: sql Metadata Hydrated RunAsAccountId. + :paramtype sql_metadata_hydrated_run_as_account_id: str + :keyword sql_metadata_discovery_pipe: sql Metadata DiscoveryPipe. Known values are: "Unknown", + "VMware", "PowerShell", "SSH", "CIM", and "Other". + :paramtype sql_metadata_discovery_pipe: str or + ~azure.mgmt.migrate.models.SqlMetadataDiscoveryPipe + :keyword discovery_scope_status: discovery Scope Status. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :paramtype discovery_scope_status: str or ~azure.mgmt.migrate.models.SQLDiscoveryScopeStatus + """ + super().__init__(**kwargs) + self.successfully_discovered_server_count = successfully_discovered_server_count + self.total_server_count = total_server_count + self.sql_metadata_hydrated_run_as_account_id = sql_metadata_hydrated_run_as_account_id + self.sql_metadata_discovery_pipe = sql_metadata_discovery_pipe + self.discovery_scope_status = discovery_scope_status + + +class SqlDiscoverySiteDataSource(ProxyResource): + """A SQL discovery site data source resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSourceProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "SqlDiscoverySiteDataSourceProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.SqlDiscoverySiteDataSourceProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSourceProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlDiscoverySiteDataSourceListResult(_serialization.Model): + """The response of a SqlDiscoverySiteDataSource list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The SqlDiscoverySiteDataSource items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SqlDiscoverySiteDataSource]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.SqlDiscoverySiteDataSource"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The SqlDiscoverySiteDataSource items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :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 SqlDiscoverySiteDataSourceProperties(_serialization.Model): + """Discovery site data source properties class. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar discovery_site_id: Gets or sets the discovery site Id. + :vartype discovery_site_id: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "discovery_site_id": {"key": "discoverySiteId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, *, discovery_site_id: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword discovery_site_id: Gets or sets the discovery site Id. + :paramtype discovery_site_id: str + """ + super().__init__(**kwargs) + self.discovery_site_id = discovery_site_id + self.provisioning_state = None + + +class SqlFciProperties(_serialization.Model): + """The class representing the properties for an FCI instance. + + :ivar state: Gets or sets the state of the FCI instance. Known values are: "Unknown", + "Inherited", "Initializing", "Online", "Offline", "Failed", "Pending", "OnlinePending", and + "OfflinePending". + :vartype state: str or ~azure.mgmt.migrate.models.FCIInstanceState + :ivar network_name: Gets or sets the FCI Network Name used to connect to this FCI instance. + :vartype network_name: str + :ivar is_multi_subnet: Gets or sets a value indicating whether the FCI is hosted on possible + owners + which are in multiple subnets. + :vartype is_multi_subnet: bool + :ivar shared_disk_count: Gets or sets the count of Shared Disks for SQL FCI. + :vartype shared_disk_count: int + """ + + _attribute_map = { + "state": {"key": "state", "type": "str"}, + "network_name": {"key": "networkName", "type": "str"}, + "is_multi_subnet": {"key": "isMultiSubnet", "type": "bool"}, + "shared_disk_count": {"key": "sharedDiskCount", "type": "int"}, + } + + def __init__( + self, + *, + state: Optional[Union[str, "_models.FCIInstanceState"]] = None, + network_name: Optional[str] = None, + is_multi_subnet: Optional[bool] = None, + shared_disk_count: Optional[int] = None, + **kwargs: Any + ) -> None: + """ + :keyword state: Gets or sets the state of the FCI instance. Known values are: "Unknown", + "Inherited", "Initializing", "Online", "Offline", "Failed", "Pending", "OnlinePending", and + "OfflinePending". + :paramtype state: str or ~azure.mgmt.migrate.models.FCIInstanceState + :keyword network_name: Gets or sets the FCI Network Name used to connect to this FCI instance. + :paramtype network_name: str + :keyword is_multi_subnet: Gets or sets a value indicating whether the FCI is hosted on possible + owners + which are in multiple subnets. + :paramtype is_multi_subnet: bool + :keyword shared_disk_count: Gets or sets the count of Shared Disks for SQL FCI. + :paramtype shared_disk_count: int + """ + super().__init__(**kwargs) + self.state = state + self.network_name = network_name + self.is_multi_subnet = is_multi_subnet + self.shared_disk_count = shared_disk_count + + +class SqlJob(ProxyResource): + """Class representing the web model of SQL Database. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.JobProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "JobProperties"}, + } + + def __init__(self, *, properties: Optional["_models.JobProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.JobProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlJobListResult(_serialization.Model): + """The response of a SqlJob list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The SqlJob items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.SqlJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SqlJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.SqlJob"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The SqlJob items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SqlJob] + :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 SqlMachineOverview(_serialization.Model): + """Gets or sets SQL machine overview data. + + :ivar machine_arm_id: Gets or sets SQL machine ARM ID. + :vartype machine_arm_id: str + :ivar display_name: Gets or sets SQL machine display name. + :vartype display_name: str + :ivar fci_role: Gets or sets SQL machine FCI role. Known values are: "Unknown", + "NotApplicable", "ActiveNode", and "PossibleOwnerNode". + :vartype fci_role: str or ~azure.mgmt.migrate.models.SqlMachineOverviewFciRole + """ + + _attribute_map = { + "machine_arm_id": {"key": "machineArmId", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + "fci_role": {"key": "fciRole", "type": "str"}, + } + + def __init__( + self, + *, + machine_arm_id: Optional[str] = None, + display_name: Optional[str] = None, + fci_role: Optional[Union[str, "_models.SqlMachineOverviewFciRole"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword machine_arm_id: Gets or sets SQL machine ARM ID. + :paramtype machine_arm_id: str + :keyword display_name: Gets or sets SQL machine display name. + :paramtype display_name: str + :keyword fci_role: Gets or sets SQL machine FCI role. Known values are: "Unknown", + "NotApplicable", "ActiveNode", and "PossibleOwnerNode". + :paramtype fci_role: str or ~azure.mgmt.migrate.models.SqlMachineOverviewFciRole + """ + super().__init__(**kwargs) + self.machine_arm_id = machine_arm_id + self.display_name = display_name + self.fci_role = fci_role + + +class SqlRunAsAccount(ProxyResource): + """A runasaccount resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "RunAsAccountProperties"}, + } + + def __init__(self, *, properties: Optional["_models.RunAsAccountProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlRunAsAccountListResult(_serialization.Model): + """The response of a SqlRunAsAccount list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The SqlRunAsAccount items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.SqlRunAsAccount] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SqlRunAsAccount]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.SqlRunAsAccount"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The SqlRunAsAccount items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SqlRunAsAccount] + :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 SqlServerApplication(_serialization.Model): # pylint: disable=too-many-instance-attributes + """SQLServer in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar name: Gets or sets Name of the SQLServer. + :vartype name: str + :ivar edition: Gets or sets Edition of the SQLServer. + :vartype edition: str + :ivar service_pack: Gets or sets ServicePack of the SQLServer. + :vartype service_pack: str + :ivar version: Gets or sets Version of the SQLServer. + :vartype version: str + :ivar clustered: Gets or sets Clustered of the SQLServer. + :vartype clustered: str + :ivar cluster_name: Gets or sets ClusterName of the SQLServer. + :vartype cluster_name: str + :ivar dns_host_name: Gets or sets the DNS host name of SQLServer. + :vartype dns_host_name: str + :ivar port: Gets or sets the port of the SQLServer. + :vartype port: str + :ivar comma_separated_ips: Gets or sets the comma separated IPs of the SQLServer. + :vartype comma_separated_ips: str + :ivar status: Gets the status. + :vartype status: str + :ivar is_named_pipe_enabled: Gets or sets the value which reflects if Named Pipe is enabled or + not. + :vartype is_named_pipe_enabled: bool + :ivar is_tcp_ip_enabled: Gets or sets the value which reflects if Named Pipe is enabled or not. + :vartype is_tcp_ip_enabled: bool + :ivar named_pipe_name: Gets the status. + :vartype named_pipe_name: str + """ + + _validation = { + "name": {"readonly": True}, + "edition": {"readonly": True}, + "service_pack": {"readonly": True}, + "version": {"readonly": True}, + "clustered": {"readonly": True}, + "cluster_name": {"readonly": True}, + "dns_host_name": {"readonly": True}, + "port": {"readonly": True}, + "comma_separated_ips": {"readonly": True}, + "status": {"readonly": True}, + "is_named_pipe_enabled": {"readonly": True}, + "is_tcp_ip_enabled": {"readonly": True}, + "named_pipe_name": {"readonly": True}, + } + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "edition": {"key": "edition", "type": "str"}, + "service_pack": {"key": "servicePack", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "clustered": {"key": "clustered", "type": "str"}, + "cluster_name": {"key": "clusterName", "type": "str"}, + "dns_host_name": {"key": "dnsHostName", "type": "str"}, + "port": {"key": "port", "type": "str"}, + "comma_separated_ips": {"key": "commaSeparatedIps", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "is_named_pipe_enabled": {"key": "isNamedPipeEnabled", "type": "bool"}, + "is_tcp_ip_enabled": {"key": "isTcpIpEnabled", "type": "bool"}, + "named_pipe_name": {"key": "namedPipeName", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.name = None + self.edition = None + self.service_pack = None + self.version = None + self.clustered = None + self.cluster_name = None + self.dns_host_name = None + self.port = None + self.comma_separated_ips = None + self.status = None + self.is_named_pipe_enabled = None + self.is_tcp_ip_enabled = None + self.named_pipe_name = None + + +class SqlServerProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for SQL Server properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar machine_overview_list: Gets the Machine Overview properties of all machines on which the + SQL server is deployed. + :vartype machine_overview_list: list[~azure.mgmt.migrate.models.SqlMachineOverview] + :ivar number_of_ag_databases: Gets or sets the number of databases part of availability group + on this SQL instance. + :vartype number_of_ag_databases: int + :ivar sql_fci_properties: Gets or sets the FCI properties if the instance is an FCI instance. + This would be populated if IsClustered is true. + :vartype sql_fci_properties: ~azure.mgmt.migrate.models.SqlFciProperties + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar version: Gets or sets the SQL server version. + :vartype version: str + :ivar number_of_user_databases: Gets or sets total number of users databases. + :vartype number_of_user_databases: int + :ivar sum_of_user_databases_size_in_mb: Gets or sets total size of all user databases. + :vartype sum_of_user_databases_size_in_mb: float + :ivar temp_db_size_in_mb: Gets or sets size of temp database. + :vartype temp_db_size_in_mb: float + :ivar max_server_memory_in_use_in_mb: Gets or sets maximum server memory in use. + :vartype max_server_memory_in_use_in_mb: float + :ivar visible_online_core_count: Gets or sets the SQL server number of cores that have visible + online status. + :vartype visible_online_core_count: int + :ivar num_of_logins: Gets or sets number of logins. + :vartype num_of_logins: int + :ivar physical_cpu_count: Gets or sets physical CPU count. + :vartype physical_cpu_count: float + :ivar logical_cpu_count: Gets or sets logical CPU count. + :vartype logical_cpu_count: int + :ivar engine_edition: Gets or sets the SQL server engine edition. + :vartype engine_edition: str + :ivar edition: Gets or sets the SQL server edition. + :vartype edition: str + :ivar is_high_availability_enabled: Gets or sets a value indicating whether High Availability + is enabled or not. + :vartype is_high_availability_enabled: bool + :ivar is_clustered: Gets or sets a value indicating whether failover cluster is configured or + not. + :vartype is_clustered: bool + :ivar hyperthread_ratio: Gets or sets the Hyper thread ratio. + :vartype hyperthread_ratio: int + :ivar sql_start_time: Gets or sets the SQL start time. + :vartype sql_start_time: ~datetime.datetime + :ivar machine_arm_ids: Gets the list of machine ARM Ids on which the SQL server is deployed. + :vartype machine_arm_ids: list[str] + :ivar run_as_account_id: Gets or sets the run as account ID of the SQL server. + :vartype run_as_account_id: str + :ivar hydrated_run_as_account_id: Gets or sets the hydrated run as account ID of the SQL + server. + :vartype hydrated_run_as_account_id: str + :ivar host_name: Gets or sets hostname. + :vartype host_name: str + :ivar sql_server_name: Gets or sets the SQL server instance name. + :vartype sql_server_name: str + :ivar port_number: Gets or sets the Preferred Port Number. + :vartype port_number: int + :ivar errors: Gets the errors. + :vartype errors: list[~azure.mgmt.migrate.models.Errors] + :ivar tags: Gets or sets tags on the resource. + :vartype tags: dict[str, JSON] + :ivar is_deleted: Gets or sets a value indicating whether the entity is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets or sets the marking machine creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar status: Gets or sets the powered on status. Known values are: "Unknown", + "ContinuePending", "Paused", "PausePending", "Running", "StartPending", "Stopped", and + "StopPending". + :vartype status: str or ~azure.mgmt.migrate.models.SqlServerStatus + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "machine_overview_list": {"readonly": True}, + "machine_arm_ids": {"readonly": True}, + "errors": {"readonly": True}, + } + + _attribute_map = { + "machine_overview_list": {"key": "machineOverviewList", "type": "[SqlMachineOverview]"}, + "number_of_ag_databases": {"key": "numberOfAgDatabases", "type": "int"}, + "sql_fci_properties": {"key": "sqlFciProperties", "type": "SqlFciProperties"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "version": {"key": "version", "type": "str"}, + "number_of_user_databases": {"key": "numberOfUserDatabases", "type": "int"}, + "sum_of_user_databases_size_in_mb": {"key": "sumOfUserDatabasesSizeInMb", "type": "float"}, + "temp_db_size_in_mb": {"key": "tempDbSizeInMb", "type": "float"}, + "max_server_memory_in_use_in_mb": {"key": "maxServerMemoryInUseInMb", "type": "float"}, + "visible_online_core_count": {"key": "visibleOnlineCoreCount", "type": "int"}, + "num_of_logins": {"key": "numOfLogins", "type": "int"}, + "physical_cpu_count": {"key": "physicalCpuCount", "type": "float"}, + "logical_cpu_count": {"key": "logicalCpuCount", "type": "int"}, + "engine_edition": {"key": "engineEdition", "type": "str"}, + "edition": {"key": "edition", "type": "str"}, + "is_high_availability_enabled": {"key": "isHighAvailabilityEnabled", "type": "bool"}, + "is_clustered": {"key": "isClustered", "type": "bool"}, + "hyperthread_ratio": {"key": "hyperthreadRatio", "type": "int"}, + "sql_start_time": {"key": "sqlStartTime", "type": "iso-8601"}, + "machine_arm_ids": {"key": "machineArmIds", "type": "[str]"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "hydrated_run_as_account_id": {"key": "hydratedRunAsAccountId", "type": "str"}, + "host_name": {"key": "hostName", "type": "str"}, + "sql_server_name": {"key": "sqlServerName", "type": "str"}, + "port_number": {"key": "portNumber", "type": "int"}, + "errors": {"key": "errors", "type": "[Errors]"}, + "tags": {"key": "tags", "type": "{object}"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( # pylint: disable=too-many-locals + self, + *, + number_of_ag_databases: Optional[int] = None, + sql_fci_properties: Optional["_models.SqlFciProperties"] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + version: Optional[str] = None, + number_of_user_databases: Optional[int] = None, + sum_of_user_databases_size_in_mb: Optional[float] = None, + temp_db_size_in_mb: Optional[float] = None, + max_server_memory_in_use_in_mb: Optional[float] = None, + visible_online_core_count: Optional[int] = None, + num_of_logins: Optional[int] = None, + physical_cpu_count: Optional[float] = None, + logical_cpu_count: Optional[int] = None, + engine_edition: Optional[str] = None, + edition: Optional[str] = None, + is_high_availability_enabled: Optional[bool] = None, + is_clustered: Optional[bool] = None, + hyperthread_ratio: Optional[int] = None, + sql_start_time: Optional[datetime.datetime] = None, + run_as_account_id: Optional[str] = None, + hydrated_run_as_account_id: Optional[str] = None, + host_name: Optional[str] = None, + sql_server_name: Optional[str] = None, + port_number: Optional[int] = None, + tags: Optional[Dict[str, JSON]] = None, + is_deleted: Optional[bool] = None, + created_timestamp: Optional[str] = None, + updated_timestamp: Optional[str] = None, + status: Optional[Union[str, "_models.SqlServerStatus"]] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword number_of_ag_databases: Gets or sets the number of databases part of availability + group on this SQL instance. + :paramtype number_of_ag_databases: int + :keyword sql_fci_properties: Gets or sets the FCI properties if the instance is an FCI + instance. This would be populated if IsClustered is true. + :paramtype sql_fci_properties: ~azure.mgmt.migrate.models.SqlFciProperties + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword version: Gets or sets the SQL server version. + :paramtype version: str + :keyword number_of_user_databases: Gets or sets total number of users databases. + :paramtype number_of_user_databases: int + :keyword sum_of_user_databases_size_in_mb: Gets or sets total size of all user databases. + :paramtype sum_of_user_databases_size_in_mb: float + :keyword temp_db_size_in_mb: Gets or sets size of temp database. + :paramtype temp_db_size_in_mb: float + :keyword max_server_memory_in_use_in_mb: Gets or sets maximum server memory in use. + :paramtype max_server_memory_in_use_in_mb: float + :keyword visible_online_core_count: Gets or sets the SQL server number of cores that have + visible online status. + :paramtype visible_online_core_count: int + :keyword num_of_logins: Gets or sets number of logins. + :paramtype num_of_logins: int + :keyword physical_cpu_count: Gets or sets physical CPU count. + :paramtype physical_cpu_count: float + :keyword logical_cpu_count: Gets or sets logical CPU count. + :paramtype logical_cpu_count: int + :keyword engine_edition: Gets or sets the SQL server engine edition. + :paramtype engine_edition: str + :keyword edition: Gets or sets the SQL server edition. + :paramtype edition: str + :keyword is_high_availability_enabled: Gets or sets a value indicating whether High + Availability is enabled or not. + :paramtype is_high_availability_enabled: bool + :keyword is_clustered: Gets or sets a value indicating whether failover cluster is configured + or not. + :paramtype is_clustered: bool + :keyword hyperthread_ratio: Gets or sets the Hyper thread ratio. + :paramtype hyperthread_ratio: int + :keyword sql_start_time: Gets or sets the SQL start time. + :paramtype sql_start_time: ~datetime.datetime + :keyword run_as_account_id: Gets or sets the run as account ID of the SQL server. + :paramtype run_as_account_id: str + :keyword hydrated_run_as_account_id: Gets or sets the hydrated run as account ID of the SQL + server. + :paramtype hydrated_run_as_account_id: str + :keyword host_name: Gets or sets hostname. + :paramtype host_name: str + :keyword sql_server_name: Gets or sets the SQL server instance name. + :paramtype sql_server_name: str + :keyword port_number: Gets or sets the Preferred Port Number. + :paramtype port_number: int + :keyword tags: Gets or sets tags on the resource. + :paramtype tags: dict[str, JSON] + :keyword is_deleted: Gets or sets a value indicating whether the entity is deleted. + :paramtype is_deleted: bool + :keyword created_timestamp: Gets or sets the marking machine creation. + :paramtype created_timestamp: str + :keyword updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :paramtype updated_timestamp: str + :keyword status: Gets or sets the powered on status. Known values are: "Unknown", + "ContinuePending", "Paused", "PausePending", "Running", "StartPending", "Stopped", and + "StopPending". + :paramtype status: str or ~azure.mgmt.migrate.models.SqlServerStatus + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.machine_overview_list = None + self.number_of_ag_databases = number_of_ag_databases + self.sql_fci_properties = sql_fci_properties + self.product_support_status = product_support_status + self.version = version + self.number_of_user_databases = number_of_user_databases + self.sum_of_user_databases_size_in_mb = sum_of_user_databases_size_in_mb + self.temp_db_size_in_mb = temp_db_size_in_mb + self.max_server_memory_in_use_in_mb = max_server_memory_in_use_in_mb + self.visible_online_core_count = visible_online_core_count + self.num_of_logins = num_of_logins + self.physical_cpu_count = physical_cpu_count + self.logical_cpu_count = logical_cpu_count + self.engine_edition = engine_edition + self.edition = edition + self.is_high_availability_enabled = is_high_availability_enabled + self.is_clustered = is_clustered + self.hyperthread_ratio = hyperthread_ratio + self.sql_start_time = sql_start_time + self.machine_arm_ids = None + self.run_as_account_id = run_as_account_id + self.hydrated_run_as_account_id = hydrated_run_as_account_id + self.host_name = host_name + self.sql_server_name = sql_server_name + self.port_number = port_number + self.errors = None + self.tags = tags + self.is_deleted = is_deleted + self.created_timestamp = created_timestamp + self.updated_timestamp = updated_timestamp + self.status = status + self.provisioning_state = provisioning_state + + +class SqlServerV2(ProxyResource): + """Class representing the web model of SQL Server. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SqlServerProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "SqlServerProperties"}, + } + + def __init__(self, *, properties: Optional["_models.SqlServerProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SqlServerProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlServerV2ListResult(_serialization.Model): + """The response of a SqlServerV2 list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The SqlServerV2 items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.SqlServerV2] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SqlServerV2]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.SqlServerV2"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The SqlServerV2 items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SqlServerV2] + :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 SqlServerV2Update(_serialization.Model): + """The type used for update operations of the SqlServerV2. + + :ivar properties: The updatable properties of the SqlServerV2. + :vartype properties: ~azure.mgmt.migrate.models.SqlServerV2UpdateProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "SqlServerV2UpdateProperties"}, + } + + def __init__(self, *, properties: Optional["_models.SqlServerV2UpdateProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The updatable properties of the SqlServerV2. + :paramtype properties: ~azure.mgmt.migrate.models.SqlServerV2UpdateProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlServerV2UpdateProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """The updatable properties of the SqlServerV2. + + :ivar number_of_ag_databases: Gets or sets the number of databases part of availability group + on this SQL instance. + :vartype number_of_ag_databases: int + :ivar sql_fci_properties: Gets or sets the FCI properties if the instance is an FCI instance. + This would be populated if IsClustered is true. + :vartype sql_fci_properties: ~azure.mgmt.migrate.models.SqlFciProperties + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar version: Gets or sets the SQL server version. + :vartype version: str + :ivar number_of_user_databases: Gets or sets total number of users databases. + :vartype number_of_user_databases: int + :ivar sum_of_user_databases_size_in_mb: Gets or sets total size of all user databases. + :vartype sum_of_user_databases_size_in_mb: float + :ivar temp_db_size_in_mb: Gets or sets size of temp database. + :vartype temp_db_size_in_mb: float + :ivar max_server_memory_in_use_in_mb: Gets or sets maximum server memory in use. + :vartype max_server_memory_in_use_in_mb: float + :ivar visible_online_core_count: Gets or sets the SQL server number of cores that have visible + online status. + :vartype visible_online_core_count: int + :ivar num_of_logins: Gets or sets number of logins. + :vartype num_of_logins: int + :ivar physical_cpu_count: Gets or sets physical CPU count. + :vartype physical_cpu_count: float + :ivar logical_cpu_count: Gets or sets logical CPU count. + :vartype logical_cpu_count: int + :ivar engine_edition: Gets or sets the SQL server engine edition. + :vartype engine_edition: str + :ivar edition: Gets or sets the SQL server edition. + :vartype edition: str + :ivar is_high_availability_enabled: Gets or sets a value indicating whether High Availability + is enabled or not. + :vartype is_high_availability_enabled: bool + :ivar is_clustered: Gets or sets a value indicating whether failover cluster is configured or + not. + :vartype is_clustered: bool + :ivar hyperthread_ratio: Gets or sets the Hyper thread ratio. + :vartype hyperthread_ratio: int + :ivar sql_start_time: Gets or sets the SQL start time. + :vartype sql_start_time: ~datetime.datetime + :ivar run_as_account_id: Gets or sets the run as account ID of the SQL server. + :vartype run_as_account_id: str + :ivar hydrated_run_as_account_id: Gets or sets the hydrated run as account ID of the SQL + server. + :vartype hydrated_run_as_account_id: str + :ivar host_name: Gets or sets hostname. + :vartype host_name: str + :ivar sql_server_name: Gets or sets the SQL server instance name. + :vartype sql_server_name: str + :ivar port_number: Gets or sets the Preferred Port Number. + :vartype port_number: int + :ivar tags: Gets or sets tags on the resource. + :vartype tags: dict[str, JSON] + :ivar is_deleted: Gets or sets a value indicating whether the entity is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets or sets the marking machine creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar status: Gets or sets the powered on status. Known values are: "Unknown", + "ContinuePending", "Paused", "PausePending", "Running", "StartPending", "Stopped", and + "StopPending". + :vartype status: str or ~azure.mgmt.migrate.models.SqlServerStatus + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _attribute_map = { + "number_of_ag_databases": {"key": "numberOfAgDatabases", "type": "int"}, + "sql_fci_properties": {"key": "sqlFciProperties", "type": "SqlFciProperties"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "version": {"key": "version", "type": "str"}, + "number_of_user_databases": {"key": "numberOfUserDatabases", "type": "int"}, + "sum_of_user_databases_size_in_mb": {"key": "sumOfUserDatabasesSizeInMb", "type": "float"}, + "temp_db_size_in_mb": {"key": "tempDbSizeInMb", "type": "float"}, + "max_server_memory_in_use_in_mb": {"key": "maxServerMemoryInUseInMb", "type": "float"}, + "visible_online_core_count": {"key": "visibleOnlineCoreCount", "type": "int"}, + "num_of_logins": {"key": "numOfLogins", "type": "int"}, + "physical_cpu_count": {"key": "physicalCpuCount", "type": "float"}, + "logical_cpu_count": {"key": "logicalCpuCount", "type": "int"}, + "engine_edition": {"key": "engineEdition", "type": "str"}, + "edition": {"key": "edition", "type": "str"}, + "is_high_availability_enabled": {"key": "isHighAvailabilityEnabled", "type": "bool"}, + "is_clustered": {"key": "isClustered", "type": "bool"}, + "hyperthread_ratio": {"key": "hyperthreadRatio", "type": "int"}, + "sql_start_time": {"key": "sqlStartTime", "type": "iso-8601"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "hydrated_run_as_account_id": {"key": "hydratedRunAsAccountId", "type": "str"}, + "host_name": {"key": "hostName", "type": "str"}, + "sql_server_name": {"key": "sqlServerName", "type": "str"}, + "port_number": {"key": "portNumber", "type": "int"}, + "tags": {"key": "tags", "type": "{object}"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( # pylint: disable=too-many-locals + self, + *, + number_of_ag_databases: Optional[int] = None, + sql_fci_properties: Optional["_models.SqlFciProperties"] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + version: Optional[str] = None, + number_of_user_databases: Optional[int] = None, + sum_of_user_databases_size_in_mb: Optional[float] = None, + temp_db_size_in_mb: Optional[float] = None, + max_server_memory_in_use_in_mb: Optional[float] = None, + visible_online_core_count: Optional[int] = None, + num_of_logins: Optional[int] = None, + physical_cpu_count: Optional[float] = None, + logical_cpu_count: Optional[int] = None, + engine_edition: Optional[str] = None, + edition: Optional[str] = None, + is_high_availability_enabled: Optional[bool] = None, + is_clustered: Optional[bool] = None, + hyperthread_ratio: Optional[int] = None, + sql_start_time: Optional[datetime.datetime] = None, + run_as_account_id: Optional[str] = None, + hydrated_run_as_account_id: Optional[str] = None, + host_name: Optional[str] = None, + sql_server_name: Optional[str] = None, + port_number: Optional[int] = None, + tags: Optional[Dict[str, JSON]] = None, + is_deleted: Optional[bool] = None, + created_timestamp: Optional[str] = None, + updated_timestamp: Optional[str] = None, + status: Optional[Union[str, "_models.SqlServerStatus"]] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword number_of_ag_databases: Gets or sets the number of databases part of availability + group on this SQL instance. + :paramtype number_of_ag_databases: int + :keyword sql_fci_properties: Gets or sets the FCI properties if the instance is an FCI + instance. This would be populated if IsClustered is true. + :paramtype sql_fci_properties: ~azure.mgmt.migrate.models.SqlFciProperties + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword version: Gets or sets the SQL server version. + :paramtype version: str + :keyword number_of_user_databases: Gets or sets total number of users databases. + :paramtype number_of_user_databases: int + :keyword sum_of_user_databases_size_in_mb: Gets or sets total size of all user databases. + :paramtype sum_of_user_databases_size_in_mb: float + :keyword temp_db_size_in_mb: Gets or sets size of temp database. + :paramtype temp_db_size_in_mb: float + :keyword max_server_memory_in_use_in_mb: Gets or sets maximum server memory in use. + :paramtype max_server_memory_in_use_in_mb: float + :keyword visible_online_core_count: Gets or sets the SQL server number of cores that have + visible online status. + :paramtype visible_online_core_count: int + :keyword num_of_logins: Gets or sets number of logins. + :paramtype num_of_logins: int + :keyword physical_cpu_count: Gets or sets physical CPU count. + :paramtype physical_cpu_count: float + :keyword logical_cpu_count: Gets or sets logical CPU count. + :paramtype logical_cpu_count: int + :keyword engine_edition: Gets or sets the SQL server engine edition. + :paramtype engine_edition: str + :keyword edition: Gets or sets the SQL server edition. + :paramtype edition: str + :keyword is_high_availability_enabled: Gets or sets a value indicating whether High + Availability is enabled or not. + :paramtype is_high_availability_enabled: bool + :keyword is_clustered: Gets or sets a value indicating whether failover cluster is configured + or not. + :paramtype is_clustered: bool + :keyword hyperthread_ratio: Gets or sets the Hyper thread ratio. + :paramtype hyperthread_ratio: int + :keyword sql_start_time: Gets or sets the SQL start time. + :paramtype sql_start_time: ~datetime.datetime + :keyword run_as_account_id: Gets or sets the run as account ID of the SQL server. + :paramtype run_as_account_id: str + :keyword hydrated_run_as_account_id: Gets or sets the hydrated run as account ID of the SQL + server. + :paramtype hydrated_run_as_account_id: str + :keyword host_name: Gets or sets hostname. + :paramtype host_name: str + :keyword sql_server_name: Gets or sets the SQL server instance name. + :paramtype sql_server_name: str + :keyword port_number: Gets or sets the Preferred Port Number. + :paramtype port_number: int + :keyword tags: Gets or sets tags on the resource. + :paramtype tags: dict[str, JSON] + :keyword is_deleted: Gets or sets a value indicating whether the entity is deleted. + :paramtype is_deleted: bool + :keyword created_timestamp: Gets or sets the marking machine creation. + :paramtype created_timestamp: str + :keyword updated_timestamp: Gets or sets the timestamp marking last updated on the machine. + :paramtype updated_timestamp: str + :keyword status: Gets or sets the powered on status. Known values are: "Unknown", + "ContinuePending", "Paused", "PausePending", "Running", "StartPending", "Stopped", and + "StopPending". + :paramtype status: str or ~azure.mgmt.migrate.models.SqlServerStatus + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.number_of_ag_databases = number_of_ag_databases + self.sql_fci_properties = sql_fci_properties + self.product_support_status = product_support_status + self.version = version + self.number_of_user_databases = number_of_user_databases + self.sum_of_user_databases_size_in_mb = sum_of_user_databases_size_in_mb + self.temp_db_size_in_mb = temp_db_size_in_mb + self.max_server_memory_in_use_in_mb = max_server_memory_in_use_in_mb + self.visible_online_core_count = visible_online_core_count + self.num_of_logins = num_of_logins + self.physical_cpu_count = physical_cpu_count + self.logical_cpu_count = logical_cpu_count + self.engine_edition = engine_edition + self.edition = edition + self.is_high_availability_enabled = is_high_availability_enabled + self.is_clustered = is_clustered + self.hyperthread_ratio = hyperthread_ratio + self.sql_start_time = sql_start_time + self.run_as_account_id = run_as_account_id + self.hydrated_run_as_account_id = hydrated_run_as_account_id + self.host_name = host_name + self.sql_server_name = sql_server_name + self.port_number = port_number + self.tags = tags + self.is_deleted = is_deleted + self.created_timestamp = created_timestamp + self.updated_timestamp = updated_timestamp + self.status = status + self.provisioning_state = provisioning_state + + +class SqlSite(ProxyResource): + """SQL site web model. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SqlSiteProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "SqlSiteProperties"}, + } + + def __init__(self, *, properties: Optional["_models.SqlSiteProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SqlSiteProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlSiteListResult(_serialization.Model): + """The response of a SqlSite list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The SqlSite items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.SqlSite] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[SqlSite]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.SqlSite"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The SqlSite items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.SqlSite] + :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 SqlSiteProperties(_serialization.Model): + """Class for SQL site properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar site_appliance_properties_collection: Gets or sets the appliance details used by service + to communicate + + to the appliance. + :vartype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :ivar discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" and + "DR". + :vartype discovery_scenario: str or + ~azure.mgmt.migrate.models.SqlSitePropertiesDiscoveryScenario + :ivar service_endpoint: Gets the service endpoint. + :vartype service_endpoint: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "service_endpoint": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "site_appliance_properties_collection": { + "key": "siteAppliancePropertiesCollection", + "type": "[SiteApplianceProperties]", + }, + "discovery_scenario": {"key": "discoveryScenario", "type": "str"}, + "service_endpoint": {"key": "serviceEndpoint", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + site_appliance_properties_collection: Optional[List["_models.SiteApplianceProperties"]] = None, + discovery_scenario: Optional[Union[str, "_models.SqlSitePropertiesDiscoveryScenario"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword site_appliance_properties_collection: Gets or sets the appliance details used by + service to communicate + + to the appliance. + :paramtype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :keyword discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" + and "DR". + :paramtype discovery_scenario: str or + ~azure.mgmt.migrate.models.SqlSitePropertiesDiscoveryScenario + """ + super().__init__(**kwargs) + self.site_appliance_properties_collection = site_appliance_properties_collection + self.discovery_scenario = discovery_scenario + self.service_endpoint = None + self.provisioning_state = None + + +class SqlSiteRefreshBody(_serialization.Model): + """SQL site refresh. + + :ivar appliance_name: Gets or sets the appliance name of the agent in the site. + :vartype appliance_name: str + """ + + _attribute_map = { + "appliance_name": {"key": "applianceName", "type": "str"}, + } + + def __init__(self, *, appliance_name: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword appliance_name: Gets or sets the appliance name of the agent in the site. + :paramtype appliance_name: str + """ + super().__init__(**kwargs) + self.appliance_name = appliance_name + + +class SqlSiteUpdate(_serialization.Model): + """The type used for update operations of the SqlSite. + + :ivar properties: The updatable properties of the SqlSite. + :vartype properties: ~azure.mgmt.migrate.models.SqlSiteUpdateProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "SqlSiteUpdateProperties"}, + } + + def __init__(self, *, properties: Optional["_models.SqlSiteUpdateProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The updatable properties of the SqlSite. + :paramtype properties: ~azure.mgmt.migrate.models.SqlSiteUpdateProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class SqlSiteUpdateProperties(_serialization.Model): + """The updatable properties of the SqlSite. + + :ivar site_appliance_properties_collection: Gets or sets the appliance details used by service + to communicate + + to the appliance. + :vartype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :ivar discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" and + "DR". + :vartype discovery_scenario: str or + ~azure.mgmt.migrate.models.SqlSitePropertiesDiscoveryScenario + """ + + _attribute_map = { + "site_appliance_properties_collection": { + "key": "siteAppliancePropertiesCollection", + "type": "[SiteApplianceProperties]", + }, + "discovery_scenario": {"key": "discoveryScenario", "type": "str"}, + } + + def __init__( + self, + *, + site_appliance_properties_collection: Optional[List["_models.SiteApplianceProperties"]] = None, + discovery_scenario: Optional[Union[str, "_models.SqlSitePropertiesDiscoveryScenario"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword site_appliance_properties_collection: Gets or sets the appliance details used by + service to communicate + + to the appliance. + :paramtype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :keyword discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" + and "DR". + :paramtype discovery_scenario: str or + ~azure.mgmt.migrate.models.SqlSitePropertiesDiscoveryScenario + """ + super().__init__(**kwargs) + self.site_appliance_properties_collection = site_appliance_properties_collection + self.discovery_scenario = discovery_scenario + + +class SqlSiteUsage(_serialization.Model): + """SQL site usage. + + :ivar server_count: Gets or sets the number of servers discovered in the site. + :vartype server_count: int + :ivar database_count: Gets or sets the number of databases discovered in the site. + :vartype database_count: int + :ivar run_as_account_count: Gets or sets the number of run as accounts in the site. + :vartype run_as_account_count: int + """ + + _attribute_map = { + "server_count": {"key": "serverCount", "type": "int"}, + "database_count": {"key": "databaseCount", "type": "int"}, + "run_as_account_count": {"key": "runAsAccountCount", "type": "int"}, + } + + def __init__( + self, + *, + server_count: Optional[int] = None, + database_count: Optional[int] = None, + run_as_account_count: Optional[int] = None, + **kwargs: Any + ) -> None: + """ + :keyword server_count: Gets or sets the number of servers discovered in the site. + :paramtype server_count: int + :keyword database_count: Gets or sets the number of databases discovered in the site. + :paramtype database_count: int + :keyword run_as_account_count: Gets or sets the number of run as accounts in the site. + :paramtype run_as_account_count: int + """ + super().__init__(**kwargs) + self.server_count = server_count + self.database_count = database_count + self.run_as_account_count = run_as_account_count + + +class StaticDiscovery(_serialization.Model): + """Contains data related static data discovery scope. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar discovery_scope_status: Gets discovery scope status. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype discovery_scope_status: str or ~azure.mgmt.migrate.models.StaticDiscoveryScopeStatus + :ivar errors: Gets errors for discovery scope. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar hydrated_run_as_account_id: Gets the run as account ID with which feature worked + successfully. + + It is discovered by the agent from the list of credentials. + :vartype hydrated_run_as_account_id: str + """ + + _validation = { + "discovery_scope_status": {"readonly": True}, + "errors": {"readonly": True}, + "hydrated_run_as_account_id": {"readonly": True}, + } + + _attribute_map = { + "discovery_scope_status": {"key": "discoveryScopeStatus", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "hydrated_run_as_account_id": {"key": "hydratedRunAsAccountId", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.discovery_scope_status = None + self.errors = None + self.hydrated_run_as_account_id = None + + +class SystemCenter(_serialization.Model): + """SystemCenter in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar product_name: Gets or sets ProductName of the SystemCenter. + :vartype product_name: str + :ivar status: Gets or sets Status of the SystemCenter. + :vartype status: str + :ivar version: Gets or sets Version of the SystemCenter. + :vartype version: str + """ + + _validation = { + "product_name": {"readonly": True}, + "status": {"readonly": True}, + "version": {"readonly": True}, + } + + _attribute_map = { + "product_name": {"key": "productName", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "version": {"key": "version", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.product_name = None + self.status = None + self.version = 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.migrate.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.migrate.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.migrate.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.migrate.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 TagsMachineInput(_serialization.Model): + """Machine class. + + :ivar machine_id: Gets or sets the ARM id of the machine. + :vartype machine_id: str + :ivar tags: run as AccountId. + :vartype tags: dict[str, str] + """ + + _attribute_map = { + "machine_id": {"key": "machineId", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + } + + def __init__( + self, *, machine_id: Optional[str] = None, tags: Optional[Dict[str, str]] = None, **kwargs: Any + ) -> None: + """ + :keyword machine_id: Gets or sets the ARM id of the machine. + :paramtype machine_id: str + :keyword tags: run as AccountId. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.machine_id = machine_id + self.tags = tags + + +class TomcatEngineUnit(_serialization.Model): + """Tomcat engine data. + + :ivar name: Gets or sets the name of the engine. + :vartype name: str + :ivar hosts: Gets or sets the hosts defined for the engine. + :vartype hosts: list[~azure.mgmt.migrate.models.TomcatHostUnit] + """ + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "hosts": {"key": "hosts", "type": "[TomcatHostUnit]"}, + } + + def __init__( + self, *, name: Optional[str] = None, hosts: Optional[List["_models.TomcatHostUnit"]] = None, **kwargs: Any + ) -> None: + """ + :keyword name: Gets or sets the name of the engine. + :paramtype name: str + :keyword hosts: Gets or sets the hosts defined for the engine. + :paramtype hosts: list[~azure.mgmt.migrate.models.TomcatHostUnit] + """ + super().__init__(**kwargs) + self.name = name + self.hosts = hosts + + +class TomcatHostUnit(_serialization.Model): + """Tomcat host data. + + :ivar name: Gets or sets the name of the host. + :vartype name: str + :ivar app_base_path: Gets or sets the app base path for the host. + :vartype app_base_path: str + :ivar app_base: Gets or sets the app base value configured for the host. + :vartype app_base: str + """ + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "app_base_path": {"key": "appBasePath", "type": "str"}, + "app_base": {"key": "appBase", "type": "str"}, + } + + def __init__( + self, + *, + name: Optional[str] = None, + app_base_path: Optional[str] = None, + app_base: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword name: Gets or sets the name of the host. + :paramtype name: str + :keyword app_base_path: Gets or sets the app base path for the host. + :paramtype app_base_path: str + :keyword app_base: Gets or sets the app base value configured for the host. + :paramtype app_base: str + """ + super().__init__(**kwargs) + self.name = name + self.app_base_path = app_base_path + self.app_base = app_base + + +class TomcatServiceUnit(_serialization.Model): + """Tomcat service data. + + :ivar name: Gets or sets the name of the service. + :vartype name: str + :ivar connectors: Gets or sets the connectors defined for the service component. + :vartype connectors: list[~azure.mgmt.migrate.models.ConnectorUnit] + :ivar engine: Gets or sets the engine defined for the service component. + :vartype engine: ~azure.mgmt.migrate.models.TomcatEngineUnit + """ + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "connectors": {"key": "connectors", "type": "[ConnectorUnit]"}, + "engine": {"key": "engine", "type": "TomcatEngineUnit"}, + } + + def __init__( + self, + *, + name: Optional[str] = None, + connectors: Optional[List["_models.ConnectorUnit"]] = None, + engine: Optional["_models.TomcatEngineUnit"] = None, + **kwargs: Any + ) -> None: + """ + :keyword name: Gets or sets the name of the service. + :paramtype name: str + :keyword connectors: Gets or sets the connectors defined for the service component. + :paramtype connectors: list[~azure.mgmt.migrate.models.ConnectorUnit] + :keyword engine: Gets or sets the engine defined for the service component. + :paramtype engine: ~azure.mgmt.migrate.models.TomcatEngineUnit + """ + super().__init__(**kwargs) + self.name = name + self.connectors = connectors + self.engine = engine + + +class TomcatWebApplicationProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for web application properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar has_database_dependency: Boolean value having true if the application has database + dependency. + :vartype has_database_dependency: bool + :ivar is_external_logging_configured: Boolean value having true if the application has file + dependency. + :vartype is_external_logging_configured: bool + :ivar web_server_id: Gets the web server id. + :vartype web_server_id: str + :ivar web_server_name: Gets the web server name. + :vartype web_server_name: str + :ivar machine_arm_ids: Gets the list of machine ARM Ids on which the web application is + deployed. + :vartype machine_arm_ids: list[str] + :ivar virtual_path: Gets the virtual path of the application. + :vartype virtual_path: str + :ivar physical_path: Gets the physical path of the application. + :vartype physical_path: str + :ivar bindings: Gets the front end bindings for the application. + :vartype bindings: list[~azure.mgmt.migrate.models.FrontEndBinding] + :ivar frameworks: Gets the Error details. + :vartype frameworks: list[~azure.mgmt.migrate.models.WebApplicationFramework] + :ivar configurations: Gets the configuration. + :vartype configurations: list[~azure.mgmt.migrate.models.WebApplicationConfigurationUnit] + :ivar directories: Gets the directories. + :vartype directories: list[~azure.mgmt.migrate.models.WebApplicationDirectoryUnit] + :ivar display_name: Gets the display name. + :vartype display_name: str + :ivar tags: Gets tags that can be used with ODATA. + :vartype tags: dict[str, str] + :ivar static_folders: Static folders. + :vartype static_folders: list[str] + :ivar machine_display_name: Machine display name. + :vartype machine_display_name: str + :ivar is_deleted: Gets a value indicating whether application is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the timestamp marking creation of the entity. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last update operation. + :vartype updated_timestamp: str + :ivar server_type: Gets the server type. + :vartype server_type: str + :ivar errors: Gets the Error details. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar has_errors: Gets a value indicating whether the WebApp has errors or not. + :vartype has_errors: bool + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "has_database_dependency": {"readonly": True}, + "is_external_logging_configured": {"readonly": True}, + "web_server_id": {"readonly": True}, + "web_server_name": {"readonly": True}, + "machine_arm_ids": {"readonly": True}, + "virtual_path": {"readonly": True}, + "physical_path": {"readonly": True}, + "bindings": {"readonly": True}, + "frameworks": {"readonly": True}, + "configurations": {"readonly": True}, + "directories": {"readonly": True}, + "display_name": {"readonly": True}, + "tags": {"readonly": True}, + "static_folders": {"readonly": True}, + "machine_display_name": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "server_type": {"readonly": True}, + "errors": {"readonly": True}, + "appliance_names": {"readonly": True}, + "has_errors": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "has_database_dependency": {"key": "hasDatabaseDependency", "type": "bool"}, + "is_external_logging_configured": {"key": "isExternalLoggingConfigured", "type": "bool"}, + "web_server_id": {"key": "webServerId", "type": "str"}, + "web_server_name": {"key": "webServerName", "type": "str"}, + "machine_arm_ids": {"key": "machineArmIds", "type": "[str]"}, + "virtual_path": {"key": "virtualPath", "type": "str"}, + "physical_path": {"key": "physicalPath", "type": "str"}, + "bindings": {"key": "bindings", "type": "[FrontEndBinding]"}, + "frameworks": {"key": "frameworks", "type": "[WebApplicationFramework]"}, + "configurations": {"key": "configurations", "type": "[WebApplicationConfigurationUnit]"}, + "directories": {"key": "directories", "type": "[WebApplicationDirectoryUnit]"}, + "display_name": {"key": "displayName", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "static_folders": {"key": "staticFolders", "type": "[str]"}, + "machine_display_name": {"key": "machineDisplayName", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "server_type": {"key": "serverType", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "has_errors": {"key": "hasErrors", "type": "bool"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.has_database_dependency = None + self.is_external_logging_configured = None + self.web_server_id = None + self.web_server_name = None + self.machine_arm_ids = None + self.virtual_path = None + self.physical_path = None + self.bindings = None + self.frameworks = None + self.configurations = None + self.directories = None + self.display_name = None + self.tags = None + self.static_folders = None + self.machine_display_name = None + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.server_type = None + self.errors = None + self.appliance_names = None + self.has_errors = None + self.provisioning_state = None + + +class TomcatWebApplications(ProxyResource): + """Web application REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.TomcatWebApplicationProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "TomcatWebApplicationProperties"}, + } + + def __init__(self, *, properties: Optional["_models.TomcatWebApplicationProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.TomcatWebApplicationProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class TomcatWebApplicationsListResult(_serialization.Model): + """The response of a TomcatWebApplications list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The TomcatWebApplications items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.TomcatWebApplications] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[TomcatWebApplications]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.TomcatWebApplications"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The TomcatWebApplications items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.TomcatWebApplications] + :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 TomcatWebServerProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for web server properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar services: Gets or sets the services defined in the server. + :vartype services: list[~azure.mgmt.migrate.models.TomcatServiceUnit] + :ivar catalina_home: Catalina Home. + :vartype catalina_home: str + :ivar jvm_version: Version of the JVM. + :vartype jvm_version: str + :ivar session_persistence_mechanism: session persistence mechanism. + :vartype session_persistence_mechanism: str + :ivar is_clustering_present: Gets a value indicating whether application is deleted. + :vartype is_clustering_present: bool + :ivar is_memory_realm_present: Gets a value indicating whether application is deleted. + :vartype is_memory_realm_present: bool + :ivar is_session_tracking_present: is session tracking present. + :vartype is_session_tracking_present: bool + :ivar is_access_log_valve_present: is access log valve present. + :vartype is_access_log_valve_present: bool + :ivar max_memory_usage_in_mb: max memory usage in mb. + :vartype max_memory_usage_in_mb: str + :ivar configuration_location: Gets the configuration location. + :vartype configuration_location: str + :ivar version: Gets the configuration location. + :vartype version: str + :ivar machine_ids: Gets the list of machines. + :vartype machine_ids: list[str] + :ivar web_applications: Gets the list of web applications. + :vartype web_applications: list[str] + :ivar display_name: Gets the display name. + :vartype display_name: str + :ivar server_fqdn: Gets the server FQDN. + :vartype server_fqdn: str + :ivar run_as_account_id: Gets the run as account id. + :vartype run_as_account_id: str + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :ivar is_deleted: Gets a value indicating whether application is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the timestamp marking creation of the entity. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last update operation. + :vartype updated_timestamp: str + :ivar server_type: Gets the server type. + :vartype server_type: str + :ivar errors: Gets the Error details. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar has_errors: Gets a value indicating whether the WebApp has errors or not. + :vartype has_errors: bool + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "catalina_home": {"readonly": True}, + "jvm_version": {"readonly": True}, + "session_persistence_mechanism": {"readonly": True}, + "is_memory_realm_present": {"readonly": True}, + "is_session_tracking_present": {"readonly": True}, + "is_access_log_valve_present": {"readonly": True}, + "max_memory_usage_in_mb": {"readonly": True}, + "configuration_location": {"readonly": True}, + "version": {"readonly": True}, + "machine_ids": {"readonly": True}, + "web_applications": {"readonly": True}, + "display_name": {"readonly": True}, + "server_fqdn": {"readonly": True}, + "run_as_account_id": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "server_type": {"readonly": True}, + "errors": {"readonly": True}, + "appliance_names": {"readonly": True}, + "has_errors": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "services": {"key": "services", "type": "[TomcatServiceUnit]"}, + "catalina_home": {"key": "catalinaHome", "type": "str"}, + "jvm_version": {"key": "jvmVersion", "type": "str"}, + "session_persistence_mechanism": {"key": "sessionPersistenceMechanism", "type": "str"}, + "is_clustering_present": {"key": "isClusteringPresent", "type": "bool"}, + "is_memory_realm_present": {"key": "isMemoryRealmPresent", "type": "bool"}, + "is_session_tracking_present": {"key": "isSessionTrackingPresent", "type": "bool"}, + "is_access_log_valve_present": {"key": "isAccessLogValvePresent", "type": "bool"}, + "max_memory_usage_in_mb": {"key": "maxMemoryUsageInMb", "type": "str"}, + "configuration_location": {"key": "configurationLocation", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "machine_ids": {"key": "machineIds", "type": "[str]"}, + "web_applications": {"key": "webApplications", "type": "[str]"}, + "display_name": {"key": "displayName", "type": "str"}, + "server_fqdn": {"key": "serverFqdn", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "OperatingSystem"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "server_type": {"key": "serverType", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "has_errors": {"key": "hasErrors", "type": "bool"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( # pylint: disable=too-many-locals + self, + *, + services: Optional[List["_models.TomcatServiceUnit"]] = None, + is_clustering_present: Optional[bool] = None, + operating_system_details: Optional["_models.OperatingSystem"] = None, + **kwargs: Any + ) -> None: + """ + :keyword services: Gets or sets the services defined in the server. + :paramtype services: list[~azure.mgmt.migrate.models.TomcatServiceUnit] + :keyword is_clustering_present: Gets a value indicating whether application is deleted. + :paramtype is_clustering_present: bool + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + """ + super().__init__(**kwargs) + self.services = services + self.catalina_home = None + self.jvm_version = None + self.session_persistence_mechanism = None + self.is_clustering_present = is_clustering_present + self.is_memory_realm_present = None + self.is_session_tracking_present = None + self.is_access_log_valve_present = None + self.max_memory_usage_in_mb = None + self.configuration_location = None + self.version = None + self.machine_ids = None + self.web_applications = None + self.display_name = None + self.server_fqdn = None + self.run_as_account_id = None + self.operating_system_details = operating_system_details + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.server_type = None + self.errors = None + self.appliance_names = None + self.has_errors = None + self.provisioning_state = None + + +class TomcatWebServers(ProxyResource): + """Web server REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.TomcatWebServerProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "TomcatWebServerProperties"}, + } + + def __init__(self, *, properties: Optional["_models.TomcatWebServerProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.TomcatWebServerProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class TomcatWebServersListResult(_serialization.Model): + """The response of a TomcatWebServers list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The TomcatWebServers items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.TomcatWebServers] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[TomcatWebServers]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.TomcatWebServers"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The TomcatWebServers items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.TomcatWebServers] + :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 UpdateMachineDepMapStatus(_serialization.Model): + """Defines class that represents input to enable or disable DMP on machine + + .. code-block:: + + for cosmos entity operation. + + :ivar machines: Gets or sets the machine collection. + :vartype machines: list[~azure.mgmt.migrate.models.DependencyMapMachineInput] + """ + + _attribute_map = { + "machines": {"key": "machines", "type": "[DependencyMapMachineInput]"}, + } + + def __init__(self, *, machines: Optional[List["_models.DependencyMapMachineInput"]] = None, **kwargs: Any) -> None: + """ + :keyword machines: Gets or sets the machine collection. + :paramtype machines: list[~azure.mgmt.migrate.models.DependencyMapMachineInput] + """ + super().__init__(**kwargs) + self.machines = machines + + +class UpdateMachineRunAsAccount(_serialization.Model): + """Defines class that represents input to update RunAsAccount on machine + + .. code-block:: + + for cosmos entity operation. + + :ivar machines: Gets or sets the machine collection. + :vartype machines: list[~azure.mgmt.migrate.models.RunAsAccountMachineInput] + """ + + _attribute_map = { + "machines": {"key": "machines", "type": "[RunAsAccountMachineInput]"}, + } + + def __init__(self, *, machines: Optional[List["_models.RunAsAccountMachineInput"]] = None, **kwargs: Any) -> None: + """ + :keyword machines: Gets or sets the machine collection. + :paramtype machines: list[~azure.mgmt.migrate.models.RunAsAccountMachineInput] + """ + super().__init__(**kwargs) + self.machines = machines + + +class UpdateMachineTags(_serialization.Model): + """Defines class that represents input to update Tags on machine + + .. code-block:: + + for cosmos entity operation. + + :ivar machines: Gets or sets the machine collection. + :vartype machines: list[~azure.mgmt.migrate.models.TagsMachineInput] + """ + + _attribute_map = { + "machines": {"key": "machines", "type": "[TagsMachineInput]"}, + } + + def __init__(self, *, machines: Optional[List["_models.TagsMachineInput"]] = None, **kwargs: Any) -> None: + """ + :keyword machines: Gets or sets the machine collection. + :paramtype machines: list[~azure.mgmt.migrate.models.TagsMachineInput] + """ + super().__init__(**kwargs) + self.machines = machines + + +class V20180501PreviewVmwareRunAsAccount(_serialization.Model): + """Run as account REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Relative URL to get this run as account. + :vartype id: str + :ivar name: Gets the Name of the Run as account. + :vartype name: str + :ivar type: Handled by resource provider. Type = + Microsoft.OffAzure/VMWareSites/RunAsAccounts. + :vartype type: str + :ivar properties: Gets nested properties. + :vartype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "properties": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "properties": {"key": "properties", "type": "RunAsAccountProperties"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.id = None + self.name = None + self.type = None + self.properties = None + + +class V20180501PreviewVmwareRunAsAccountVmwareRunAsAccountCollection( + _serialization.Model +): # pylint: disable=name-too-long + """Collection of VMware run as accounts. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar value: Gets the list of run as accounts. + :vartype value: list[~azure.mgmt.migrate.models.V20180501PreviewVmwareRunAsAccount] + :ivar next_link: Gets the value of next link. + :vartype next_link: str + """ + + _validation = { + "next_link": {"readonly": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[V20180501PreviewVmwareRunAsAccount]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: Optional[List["_models.V20180501PreviewVmwareRunAsAccount"]] = None, **kwargs: Any + ) -> None: + """ + :keyword value: Gets the list of run as accounts. + :paramtype value: list[~azure.mgmt.migrate.models.V20180501PreviewVmwareRunAsAccount] + """ + super().__init__(**kwargs) + self.value = value + self.next_link = None + + +class Vcenter(ProxyResource): + """A vcenter resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.VcenterProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "VcenterProperties"}, + } + + def __init__(self, *, properties: Optional["_models.VcenterProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.VcenterProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class VcenterListResult(_serialization.Model): + """The response of a Vcenter list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The Vcenter items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.Vcenter] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[Vcenter]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.Vcenter"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The Vcenter items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.Vcenter] + :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 VcenterProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """The properties of VMwareSiteResource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar run_as_account_id: Gets or sets the run as account ID of the vCenter. + :vartype run_as_account_id: str + :ivar errors: Gets the errors. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar created_timestamp: Gets the timestamp marking vCenter creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last updated on the vCenter. + :vartype updated_timestamp: str + :ivar fqdn: Gets or sets the FQDN/IPAddress of the vCenter. + :vartype fqdn: str + :ivar port: Gets or sets the port of the vCenter. + :vartype port: str + :ivar version: Gets the version of the vCenter. + :vartype version: str + :ivar perf_statistics_level: Gets the performance statistics enabled on the vCenter. + :vartype perf_statistics_level: str + :ivar instance_uuid: Gets the instance UUID of the vCenter. + :vartype instance_uuid: str + :ivar friendly_name: Gets or sets the friendly name of the vCenter. + :vartype friendly_name: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "errors": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "version": {"readonly": True}, + "perf_statistics_level": {"readonly": True}, + "instance_uuid": {"readonly": True}, + } + + _attribute_map = { + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "fqdn": {"key": "fqdn", "type": "str"}, + "port": {"key": "port", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "perf_statistics_level": {"key": "perfStatisticsLevel", "type": "str"}, + "instance_uuid": {"key": "instanceUuid", "type": "str"}, + "friendly_name": {"key": "friendlyName", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + run_as_account_id: Optional[str] = None, + fqdn: Optional[str] = None, + port: Optional[str] = None, + friendly_name: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword run_as_account_id: Gets or sets the run as account ID of the vCenter. + :paramtype run_as_account_id: str + :keyword fqdn: Gets or sets the FQDN/IPAddress of the vCenter. + :paramtype fqdn: str + :keyword port: Gets or sets the port of the vCenter. + :paramtype port: str + :keyword friendly_name: Gets or sets the friendly name of the vCenter. + :paramtype friendly_name: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.run_as_account_id = run_as_account_id + self.errors = None + self.created_timestamp = None + self.updated_timestamp = None + self.fqdn = fqdn + self.port = port + self.version = None + self.perf_statistics_level = None + self.instance_uuid = None + self.friendly_name = friendly_name + self.provisioning_state = provisioning_state + + +class VmwareDatastore(_serialization.Model): + """Second level object returned as part of VMware host REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar uuid: Data store UUID. + :vartype uuid: str + :ivar capacity_in_gb: Capacity of the data store. + :vartype capacity_in_gb: float + :ivar free_space_in_gb: Free space of the data store. + :vartype free_space_in_gb: float + :ivar type: Type of the data store. Known values are: "Unknown", "CIFS", "NFS", "NFS41", + "PMEM", "VFFS", "VMFS", "VSAN", and "VVOL". + :vartype type: str or ~azure.mgmt.migrate.models.VMwareDatastoreType + :ivar symbolic_name: Symbolic name of the data store. + :vartype symbolic_name: str + """ + + _validation = { + "uuid": {"readonly": True}, + "capacity_in_gb": {"readonly": True}, + "free_space_in_gb": {"readonly": True}, + "type": {"readonly": True}, + "symbolic_name": {"readonly": True}, + } + + _attribute_map = { + "uuid": {"key": "uuid", "type": "str"}, + "capacity_in_gb": {"key": "capacityInGb", "type": "float"}, + "free_space_in_gb": {"key": "freeSpaceInGb", "type": "float"}, + "type": {"key": "type", "type": "str"}, + "symbolic_name": {"key": "symbolicName", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.uuid = None + self.capacity_in_gb = None + self.free_space_in_gb = None + self.type = None + self.symbolic_name = None + + +class VmwareDisk(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Second level object returned as part of Machine REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar uuid: Disk UUID. + :vartype uuid: str + :ivar label: Label of the disk. + :vartype label: str + :ivar disk_provisioning_policy: The provisioning policy of the disk. + It is Thin or Thick or + Unknown for the VMWare VMDK. + :vartype disk_provisioning_policy: str + :ivar disk_scrubbing_policy: The scrubbing policy of disks which can be + eagerly zeroed or + lazily zeroed. + :vartype disk_scrubbing_policy: str + :ivar disk_mode: Disk mode property used for identifying independent disks. + :vartype disk_mode: str + :ivar controller_type: Gets or sets a value indicating the type of the disk controller type. + :vartype controller_type: str + :ivar max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :vartype max_size_in_bytes: int + :ivar name: Gets or sets Name of the disk. + :vartype name: str + :ivar disk_type: Gets or sets Type of the disk. + :vartype disk_type: str + :ivar lun: Gets or sets LUN of the disk. + :vartype lun: int + :ivar path: Gets or sets Path of the disk. + :vartype path: str + """ + + _validation = { + "uuid": {"readonly": True}, + "label": {"readonly": True}, + "disk_provisioning_policy": {"readonly": True}, + "disk_scrubbing_policy": {"readonly": True}, + "disk_mode": {"readonly": True}, + "controller_type": {"readonly": True}, + } + + _attribute_map = { + "uuid": {"key": "uuid", "type": "str"}, + "label": {"key": "label", "type": "str"}, + "disk_provisioning_policy": {"key": "diskProvisioningPolicy", "type": "str"}, + "disk_scrubbing_policy": {"key": "diskScrubbingPolicy", "type": "str"}, + "disk_mode": {"key": "diskMode", "type": "str"}, + "controller_type": {"key": "controllerType", "type": "str"}, + "max_size_in_bytes": {"key": "maxSizeInBytes", "type": "int"}, + "name": {"key": "name", "type": "str"}, + "disk_type": {"key": "diskType", "type": "str"}, + "lun": {"key": "lun", "type": "int"}, + "path": {"key": "path", "type": "str"}, + } + + def __init__( + self, + *, + max_size_in_bytes: Optional[int] = None, + name: Optional[str] = None, + disk_type: Optional[str] = None, + lun: Optional[int] = None, + path: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :paramtype max_size_in_bytes: int + :keyword name: Gets or sets Name of the disk. + :paramtype name: str + :keyword disk_type: Gets or sets Type of the disk. + :paramtype disk_type: str + :keyword lun: Gets or sets LUN of the disk. + :paramtype lun: int + :keyword path: Gets or sets Path of the disk. + :paramtype path: str + """ + super().__init__(**kwargs) + self.uuid = None + self.label = None + self.disk_provisioning_policy = None + self.disk_scrubbing_policy = None + self.disk_mode = None + self.controller_type = None + self.max_size_in_bytes = max_size_in_bytes + self.name = name + self.disk_type = disk_type + self.lun = lun + self.path = path + + +class VmwareHost(ProxyResource): + """A host resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.VmwareHostProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "VmwareHostProperties"}, + } + + def __init__(self, *, properties: Optional["_models.VmwareHostProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.VmwareHostProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class VmwareHostListResult(_serialization.Model): + """The response of a VmwareHost list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The VmwareHost items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.VmwareHost] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[VmwareHost]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.VmwareHost"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The VmwareHost items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.VmwareHost] + :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 VmwareHostProperties(_serialization.Model): + """Class for host properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar created_timestamp: Gets the timestamp marking VMware host creation. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last updated on the VMware host. + :vartype updated_timestamp: str + :ivar datastores: Gets the data stores. + :vartype datastores: list[~azure.mgmt.migrate.models.VmwareDatastore] + :ivar vcenter_id: Gets the V-center ID. + :vartype vcenter_id: str + :ivar uuid: Gets the instance UUID of the vmware host. + :vartype uuid: str + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "datastores": {"readonly": True}, + "vcenter_id": {"readonly": True}, + "uuid": {"readonly": True}, + "appliance_names": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "datastores": {"key": "datastores", "type": "[VmwareDatastore]"}, + "vcenter_id": {"key": "vcenterId", "type": "str"}, + "uuid": {"key": "uuid", "type": "str"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.created_timestamp = None + self.updated_timestamp = None + self.datastores = None + self.vcenter_id = None + self.uuid = None + self.appliance_names = None + self.provisioning_state = None + + +class VmwareJob(ProxyResource): + """A host resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.JobProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "JobProperties"}, + } + + def __init__(self, *, properties: Optional["_models.JobProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.JobProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class VmwareJobListResult(_serialization.Model): + """The response of a VmwareJob list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The VmwareJob items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.VmwareJob] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[VmwareJob]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.VmwareJob"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The VmwareJob items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.VmwareJob] + :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 VmwareMachineProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for machine properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar v_center_fqdn: VCenter FQDN/IPAddress. + :vartype v_center_fqdn: str + :ivar v_center_id: VCenter ARM ID. + :vartype v_center_id: str + :ivar instance_uuid: On-premise Instance UUID of the machine. + :vartype instance_uuid: str + :ivar data_center_scope: Scope of the data center. + :vartype data_center_scope: str + :ivar description: User description of the machine. + :vartype description: str + :ivar disks: Disks attached to the machine. + :vartype disks: list[~azure.mgmt.migrate.models.VmwareDisk] + :ivar host_in_maintenance_mode: Indicates whether the host is in maintenance mode. + :vartype host_in_maintenance_mode: bool + :ivar host_name: The host name. + :vartype host_name: str + :ivar host_power_state: The host power state. + :vartype host_power_state: str + :ivar host_version: The host version. + :vartype host_version: str + :ivar network_adapters: Network adapters attached to the machine. + :vartype network_adapters: list[~azure.mgmt.migrate.models.VmwareNetworkAdapter] + :ivar v_mware_tools_status: VMware tools status. + :vartype v_mware_tools_status: str + :ivar v_mware_tools_version: VMware tools version. + :vartype v_mware_tools_version: str + :ivar change_tracking_supported: Value indicating whether change tracking is supported. + :vartype change_tracking_supported: bool + :ivar change_tracking_enabled: Value indicating whether change tracking is enabled. + :vartype change_tracking_enabled: bool + :ivar max_snapshots: Maximum number of snapshots for the VM. Default value is -1. + :vartype max_snapshots: int + :ivar disk_enabled_uuid: Gets or sets whether Disk Enabled UUID is set or not. + :vartype disk_enabled_uuid: str + :ivar number_of_snapshots: Number of snapshots for the VM. Default value is -1. + :vartype number_of_snapshots: int + :ivar alt_guest_name: Gets or sets the machine alt guest name. + :vartype alt_guest_name: str + :ivar power_status: Gets the Machine power status. + :vartype power_status: str + :ivar vm_fqdn: Gets the VM FQDN. + :vartype vm_fqdn: str + :ivar vm_configuration_file_location: Gets the Root location of the VM configuration file. + :vartype vm_configuration_file_location: str + :ivar firmware: Gets or sets the firmware. + :vartype firmware: str + :ivar guest_os_details: Operating System Details extracted from the guest + bu executing + script inside the guest VM. + :vartype guest_os_details: ~azure.mgmt.migrate.models.GuestOsDetails + :ivar number_of_applications: Number of applications installed in the guest VM. + :vartype number_of_applications: int + :ivar guest_details_discovery_timestamp: The last time at which the Guest Details was + discovered + or the + error while discovering guest details based discovery + of the + machine. + :vartype guest_details_discovery_timestamp: ~datetime.datetime + :ivar is_guest_details_discovery_in_progress: Whether Refresh Fabric Layout Guest Details has + been completed once. + + Portal will show discovery in progress, if this value is true. + :vartype is_guest_details_discovery_in_progress: bool + :ivar dependency_mapping: Gets or sets if dependency mapping feature is enabled or not + for + the VM. + :vartype dependency_mapping: str + :ivar dependency_mapping_start_time: Gets or sets when dependency mapping collection is last + started. + :vartype dependency_mapping_start_time: ~datetime.datetime + :ivar dependency_mapping_end_time: Gets or sets when dependency mapping collection was last + disabled. + :vartype dependency_mapping_end_time: str + :ivar run_as_account_id: Gets or sets the run as account ID of the machine. + :vartype run_as_account_id: str + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar errors: Errors for machine. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar application_discovery: Gets the data related to application discovery. + :vartype application_discovery: ~azure.mgmt.migrate.models.ApplicationDiscovery + :ivar dependency_map_discovery: Gets the data related to dependency map discovery. + :vartype dependency_map_discovery: ~azure.mgmt.migrate.models.DependencyMapDiscovery + :ivar static_discovery: Gets the data related to static discovery. + :vartype static_discovery: ~azure.mgmt.migrate.models.StaticDiscovery + :ivar sql_discovery: Gets the data related to SQL discovery. + :vartype sql_discovery: ~azure.mgmt.migrate.models.SqlDiscovery + :ivar web_app_discovery: Gets the data related to Web application discovery. + :vartype web_app_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar oracle_discovery: Gets the data related to Oracle discovery. + :vartype oracle_discovery: ~azure.mgmt.migrate.models.OracleDiscovery + :ivar spring_boot_discovery: Gets the data related to SpringBoot discovery. + :vartype spring_boot_discovery: ~azure.mgmt.migrate.models.SpringBootDiscovery + :ivar iis_discovery: Gets the data related to iis discovery. + :vartype iis_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar tomcat_discovery: Gets the data related to tomcat discovery. + :vartype tomcat_discovery: ~azure.mgmt.migrate.models.WebAppDiscovery + :ivar apps_and_roles: Apps And Roles of the Server. + :vartype apps_and_roles: ~azure.mgmt.migrate.models.AppsAndRoles + :ivar product_support_status: product support status. + :vartype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :ivar number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :vartype number_of_processor_core: int + :ivar allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :vartype allocated_memory_in_mb: float + :ivar operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :vartype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :ivar bios_serial_number: Gets or sets the Machine BIOS serial number. + :vartype bios_serial_number: str + :ivar bios_guid: Gets or sets the BIOS GUID. + :vartype bios_guid: str + :ivar display_name: Gets the Display name of the machine. + :vartype display_name: str + :ivar is_deleted: Value indicating whether VM is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the marking machine creation. + :vartype created_timestamp: str + :ivar tags: Gets or sets tags on the VMware machine. + :vartype tags: dict[str, str] + :ivar updated_timestamp: Gets the timestamp marking last updated on the machine. + :vartype updated_timestamp: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "v_center_fqdn": {"readonly": True}, + "v_center_id": {"readonly": True}, + "instance_uuid": {"readonly": True}, + "data_center_scope": {"readonly": True}, + "description": {"readonly": True}, + "disks": {"readonly": True}, + "host_in_maintenance_mode": {"readonly": True}, + "host_name": {"readonly": True}, + "host_power_state": {"readonly": True}, + "host_version": {"readonly": True}, + "network_adapters": {"readonly": True}, + "v_mware_tools_status": {"readonly": True}, + "v_mware_tools_version": {"readonly": True}, + "change_tracking_supported": {"readonly": True}, + "change_tracking_enabled": {"readonly": True}, + "max_snapshots": {"readonly": True}, + "disk_enabled_uuid": {"readonly": True}, + "number_of_snapshots": {"readonly": True}, + "alt_guest_name": {"readonly": True}, + "power_status": {"readonly": True}, + "vm_fqdn": {"readonly": True}, + "vm_configuration_file_location": {"readonly": True}, + "guest_os_details": {"readonly": True}, + "number_of_applications": {"readonly": True}, + "guest_details_discovery_timestamp": {"readonly": True}, + "is_guest_details_discovery_in_progress": {"readonly": True}, + "dependency_mapping": {"readonly": True}, + "dependency_mapping_start_time": {"readonly": True}, + "dependency_mapping_end_time": {"readonly": True}, + "appliance_names": {"readonly": True}, + "errors": {"readonly": True}, + "application_discovery": {"readonly": True}, + "dependency_map_discovery": {"readonly": True}, + "static_discovery": {"readonly": True}, + "sql_discovery": {"readonly": True}, + "web_app_discovery": {"readonly": True}, + "oracle_discovery": {"readonly": True}, + "spring_boot_discovery": {"readonly": True}, + "iis_discovery": {"readonly": True}, + "tomcat_discovery": {"readonly": True}, + "apps_and_roles": {"readonly": True}, + "display_name": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "v_center_fqdn": {"key": "vCenterFqdn", "type": "str"}, + "v_center_id": {"key": "vCenterId", "type": "str"}, + "instance_uuid": {"key": "instanceUuid", "type": "str"}, + "data_center_scope": {"key": "dataCenterScope", "type": "str"}, + "description": {"key": "description", "type": "str"}, + "disks": {"key": "disks", "type": "[VmwareDisk]"}, + "host_in_maintenance_mode": {"key": "hostInMaintenanceMode", "type": "bool"}, + "host_name": {"key": "hostName", "type": "str"}, + "host_power_state": {"key": "hostPowerState", "type": "str"}, + "host_version": {"key": "hostVersion", "type": "str"}, + "network_adapters": {"key": "networkAdapters", "type": "[VmwareNetworkAdapter]"}, + "v_mware_tools_status": {"key": "vMwareToolsStatus", "type": "str"}, + "v_mware_tools_version": {"key": "vMwareToolsVersion", "type": "str"}, + "change_tracking_supported": {"key": "changeTrackingSupported", "type": "bool"}, + "change_tracking_enabled": {"key": "changeTrackingEnabled", "type": "bool"}, + "max_snapshots": {"key": "maxSnapshots", "type": "int"}, + "disk_enabled_uuid": {"key": "diskEnabledUuid", "type": "str"}, + "number_of_snapshots": {"key": "numberOfSnapshots", "type": "int"}, + "alt_guest_name": {"key": "altGuestName", "type": "str"}, + "power_status": {"key": "powerStatus", "type": "str"}, + "vm_fqdn": {"key": "vmFqdn", "type": "str"}, + "vm_configuration_file_location": {"key": "vmConfigurationFileLocation", "type": "str"}, + "firmware": {"key": "firmware", "type": "str"}, + "guest_os_details": {"key": "guestOsDetails", "type": "GuestOsDetails"}, + "number_of_applications": {"key": "numberOfApplications", "type": "int"}, + "guest_details_discovery_timestamp": {"key": "guestDetailsDiscoveryTimestamp", "type": "iso-8601"}, + "is_guest_details_discovery_in_progress": {"key": "isGuestDetailsDiscoveryInProgress", "type": "bool"}, + "dependency_mapping": {"key": "dependencyMapping", "type": "str"}, + "dependency_mapping_start_time": {"key": "dependencyMappingStartTime", "type": "iso-8601"}, + "dependency_mapping_end_time": {"key": "dependencyMappingEndTime", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "application_discovery": {"key": "applicationDiscovery", "type": "ApplicationDiscovery"}, + "dependency_map_discovery": {"key": "dependencyMapDiscovery", "type": "DependencyMapDiscovery"}, + "static_discovery": {"key": "staticDiscovery", "type": "StaticDiscovery"}, + "sql_discovery": {"key": "sqlDiscovery", "type": "SqlDiscovery"}, + "web_app_discovery": {"key": "webAppDiscovery", "type": "WebAppDiscovery"}, + "oracle_discovery": {"key": "oracleDiscovery", "type": "OracleDiscovery"}, + "spring_boot_discovery": {"key": "springBootDiscovery", "type": "SpringBootDiscovery"}, + "iis_discovery": {"key": "iisDiscovery", "type": "WebAppDiscovery"}, + "tomcat_discovery": {"key": "tomcatDiscovery", "type": "WebAppDiscovery"}, + "apps_and_roles": {"key": "appsAndRoles", "type": "AppsAndRoles"}, + "product_support_status": {"key": "productSupportStatus", "type": "ProductSupportStatus"}, + "number_of_processor_core": {"key": "numberOfProcessorCore", "type": "int"}, + "allocated_memory_in_mb": {"key": "allocatedMemoryInMb", "type": "float"}, + "operating_system_details": {"key": "operatingSystemDetails", "type": "OperatingSystem"}, + "bios_serial_number": {"key": "biosSerialNumber", "type": "str"}, + "bios_guid": {"key": "biosGuid", "type": "str"}, + "display_name": {"key": "displayName", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( # pylint: disable=too-many-locals + self, + *, + firmware: Optional[str] = None, + run_as_account_id: Optional[str] = None, + product_support_status: Optional["_models.ProductSupportStatus"] = None, + number_of_processor_core: Optional[int] = None, + allocated_memory_in_mb: Optional[float] = None, + operating_system_details: Optional["_models.OperatingSystem"] = None, + bios_serial_number: Optional[str] = None, + bios_guid: Optional[str] = None, + tags: Optional[Dict[str, str]] = None, + **kwargs: Any + ) -> None: + """ + :keyword firmware: Gets or sets the firmware. + :paramtype firmware: str + :keyword run_as_account_id: Gets or sets the run as account ID of the machine. + :paramtype run_as_account_id: str + :keyword product_support_status: product support status. + :paramtype product_support_status: ~azure.mgmt.migrate.models.ProductSupportStatus + :keyword number_of_processor_core: Gets or sets the Number of Processor Cores + allocated for the + machine. + :paramtype number_of_processor_core: int + :keyword allocated_memory_in_mb: Gets or sets the allocated Memory in MB. + :paramtype allocated_memory_in_mb: float + :keyword operating_system_details: Gets or sets the Operating System Details installed on the + machine. + :paramtype operating_system_details: ~azure.mgmt.migrate.models.OperatingSystem + :keyword bios_serial_number: Gets or sets the Machine BIOS serial number. + :paramtype bios_serial_number: str + :keyword bios_guid: Gets or sets the BIOS GUID. + :paramtype bios_guid: str + :keyword tags: Gets or sets tags on the VMware machine. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.v_center_fqdn = None + self.v_center_id = None + self.instance_uuid = None + self.data_center_scope = None + self.description = None + self.disks = None + self.host_in_maintenance_mode = None + self.host_name = None + self.host_power_state = None + self.host_version = None + self.network_adapters = None + self.v_mware_tools_status = None + self.v_mware_tools_version = None + self.change_tracking_supported = None + self.change_tracking_enabled = None + self.max_snapshots = None + self.disk_enabled_uuid = None + self.number_of_snapshots = None + self.alt_guest_name = None + self.power_status = None + self.vm_fqdn = None + self.vm_configuration_file_location = None + self.firmware = firmware + self.guest_os_details = None + self.number_of_applications = None + self.guest_details_discovery_timestamp = None + self.is_guest_details_discovery_in_progress = None + self.dependency_mapping = None + self.dependency_mapping_start_time = None + self.dependency_mapping_end_time = None + self.run_as_account_id = run_as_account_id + self.appliance_names = None + self.errors = None + self.application_discovery = None + self.dependency_map_discovery = None + self.static_discovery = None + self.sql_discovery = None + self.web_app_discovery = None + self.oracle_discovery = None + self.spring_boot_discovery = None + self.iis_discovery = None + self.tomcat_discovery = None + self.apps_and_roles = None + self.product_support_status = product_support_status + self.number_of_processor_core = number_of_processor_core + self.allocated_memory_in_mb = allocated_memory_in_mb + self.operating_system_details = operating_system_details + self.bios_serial_number = bios_serial_number + self.bios_guid = bios_guid + self.display_name = None + self.is_deleted = None + self.created_timestamp = None + self.tags = tags + self.updated_timestamp = None + self.provisioning_state = None + + +class VmwareMachineSoftwareInventory(ProxyResource): + """An software inventory resource belonging to a machine resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.MachineSoftwareInventoryProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "MachineSoftwareInventoryProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.MachineSoftwareInventoryProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.MachineSoftwareInventoryProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class VmwareMachineSoftwareInventoryListResult(_serialization.Model): + """The response of a VmwareMachineSoftwareInventory list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The VmwareMachineSoftwareInventory items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.VmwareMachineSoftwareInventory] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[VmwareMachineSoftwareInventory]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.VmwareMachineSoftwareInventory"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The VmwareMachineSoftwareInventory items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.VmwareMachineSoftwareInventory] + :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 VmwareNetworkAdapter(_serialization.Model): + """Second level object represented in responses as part of Machine REST resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar label: Label of the NIC. + :vartype label: str + :ivar adapter_type: Gets or sets the adapter type. + :vartype adapter_type: str + :ivar nic_id: Gets or sets the NIC Id. + :vartype nic_id: str + :ivar mac_address: Gets or sets Mac address of the NIC. + :vartype mac_address: str + :ivar ip_address_list: Gets IP addresses for the machine. + :vartype ip_address_list: list[str] + :ivar network_name: Gets or sets Network Name. + :vartype network_name: str + :ivar ip_address_type: Gets or sets Type of the IP address. + :vartype ip_address_type: str + """ + + _validation = { + "label": {"readonly": True}, + "adapter_type": {"readonly": True}, + "ip_address_list": {"readonly": True}, + } + + _attribute_map = { + "label": {"key": "label", "type": "str"}, + "adapter_type": {"key": "adapterType", "type": "str"}, + "nic_id": {"key": "nicId", "type": "str"}, + "mac_address": {"key": "macAddress", "type": "str"}, + "ip_address_list": {"key": "ipAddressList", "type": "[str]"}, + "network_name": {"key": "networkName", "type": "str"}, + "ip_address_type": {"key": "ipAddressType", "type": "str"}, + } + + def __init__( + self, + *, + nic_id: Optional[str] = None, + mac_address: Optional[str] = None, + network_name: Optional[str] = None, + ip_address_type: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword nic_id: Gets or sets the NIC Id. + :paramtype nic_id: str + :keyword mac_address: Gets or sets Mac address of the NIC. + :paramtype mac_address: str + :keyword network_name: Gets or sets Network Name. + :paramtype network_name: str + :keyword ip_address_type: Gets or sets Type of the IP address. + :paramtype ip_address_type: str + """ + super().__init__(**kwargs) + self.label = None + self.adapter_type = None + self.nic_id = nic_id + self.mac_address = mac_address + self.ip_address_list = None + self.network_name = network_name + self.ip_address_type = ip_address_type + + +class VmwareRunAsAccountResource(ProxyResource): + """A runasaccount resource belonging to a site resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "RunAsAccountProperties"}, + } + + def __init__(self, *, properties: Optional["_models.RunAsAccountProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class VmwareRunAsAccountResourceListResult(_serialization.Model): + """The response of a VmwareRunAsAccountResource list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The VmwareRunAsAccountResource items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.VmwareRunAsAccountResource] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[VmwareRunAsAccountResource]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.VmwareRunAsAccountResource"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The VmwareRunAsAccountResource items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.VmwareRunAsAccountResource] + :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 VmwareSite(TrackedResource): + """A VmwareSite. + + 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: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.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: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.SitesProperties + :ivar e_tag: If eTag is provided in the response body, it may also be provided as a header per + the normal etag convention. Entity tags are used for comparing two or more entities from the + same requested resource. HTTP/1.1 uses entity tags in the etag (section 14.19), If-Match + (section 14.24), If-None-Match (section 14.26), and If-Range (section 14.27) header fields. + :vartype e_tag: str + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + "location": {"required": True}, + "e_tag": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "tags": {"key": "tags", "type": "{str}"}, + "location": {"key": "location", "type": "str"}, + "properties": {"key": "properties", "type": "SitesProperties"}, + "e_tag": {"key": "eTag", "type": "str"}, + } + + def __init__( + self, + *, + location: str, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.SitesProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword location: The geo-location where the resource lives. Required. + :paramtype location: str + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.SitesProperties + """ + super().__init__(tags=tags, location=location, **kwargs) + self.properties = properties + self.e_tag = None + + +class VmwareSiteListResult(_serialization.Model): + """The response of a VmwareSite list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The VmwareSite items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.VmwareSite] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[VmwareSite]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.VmwareSite"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The VmwareSite items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.VmwareSite] + :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 VmwareSiteUpdate(_serialization.Model): + """The type used for update operations of the VmwareSite. + + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar properties: The updatable properties of the VmwareSite. + :vartype properties: ~azure.mgmt.migrate.models.VmwareSiteUpdateProperties + """ + + _attribute_map = { + "tags": {"key": "tags", "type": "{str}"}, + "properties": {"key": "properties", "type": "VmwareSiteUpdateProperties"}, + } + + def __init__( + self, + *, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.VmwareSiteUpdateProperties"] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword properties: The updatable properties of the VmwareSite. + :paramtype properties: ~azure.mgmt.migrate.models.VmwareSiteUpdateProperties + """ + super().__init__(**kwargs) + self.tags = tags + self.properties = properties + + +class VmwareSiteUpdateProperties(_serialization.Model): + """The updatable properties of the VmwareSite. + + :ivar service_principal_identity_details: Gets or sets the service principal identity details + used by agent for + communication + to the service. + :vartype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :ivar agent_details: Gets or sets the on-premises agent details. + :vartype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :ivar appliance_name: Gets or sets the Appliance Name. + :vartype appliance_name: str + :ivar discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :vartype discovery_solution_id: str + :ivar provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _attribute_map = { + "service_principal_identity_details": {"key": "servicePrincipalIdentityDetails", "type": "SiteSpnProperties"}, + "agent_details": {"key": "agentDetails", "type": "SiteAgentProperties"}, + "appliance_name": {"key": "applianceName", "type": "str"}, + "discovery_solution_id": {"key": "discoverySolutionId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + service_principal_identity_details: Optional["_models.SiteSpnProperties"] = None, + agent_details: Optional["_models.SiteAgentProperties"] = None, + appliance_name: Optional[str] = None, + discovery_solution_id: Optional[str] = None, + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword service_principal_identity_details: Gets or sets the service principal identity + details used by agent for + communication + to the service. + :paramtype service_principal_identity_details: ~azure.mgmt.migrate.models.SiteSpnProperties + :keyword agent_details: Gets or sets the on-premises agent details. + :paramtype agent_details: ~azure.mgmt.migrate.models.SiteAgentProperties + :keyword appliance_name: Gets or sets the Appliance Name. + :paramtype appliance_name: str + :keyword discovery_solution_id: Gets or sets the ARM ID of migration hub solution for SDS. + :paramtype discovery_solution_id: str + :keyword provisioning_state: The status of the last operation. Known values are: "Created", + "Updated", "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :paramtype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + super().__init__(**kwargs) + self.service_principal_identity_details = service_principal_identity_details + self.agent_details = agent_details + self.appliance_name = appliance_name + self.discovery_solution_id = discovery_solution_id + self.provisioning_state = provisioning_state + + +class VmwareSiteUsage(_serialization.Model): + """VMware site usage. + + :ivar machine_count: Gets or sets the number of machines discovered in the site. + :vartype machine_count: int + :ivar run_as_account_count: Gets or sets the number of run as accounts in the site. + :vartype run_as_account_count: int + :ivar v_center_count: Gets or sets the number of vCenters part of the site. + :vartype v_center_count: int + """ + + _attribute_map = { + "machine_count": {"key": "machineCount", "type": "int"}, + "run_as_account_count": {"key": "runAsAccountCount", "type": "int"}, + "v_center_count": {"key": "vCenterCount", "type": "int"}, + } + + def __init__( + self, + *, + machine_count: Optional[int] = None, + run_as_account_count: Optional[int] = None, + v_center_count: Optional[int] = None, + **kwargs: Any + ) -> None: + """ + :keyword machine_count: Gets or sets the number of machines discovered in the site. + :paramtype machine_count: int + :keyword run_as_account_count: Gets or sets the number of run as accounts in the site. + :paramtype run_as_account_count: int + :keyword v_center_count: Gets or sets the number of vCenters part of the site. + :paramtype v_center_count: int + """ + super().__init__(**kwargs) + self.machine_count = machine_count + self.run_as_account_count = run_as_account_count + self.v_center_count = v_center_count + + +class WebAppDiscovery(_serialization.Model): + """Data related to a machine's WebApps discovery. + + :ivar total_web_server_count: Gets or sets number of successfully discovered web servers.. + :vartype total_web_server_count: int + :ivar total_web_application_count: Gets or sets number of successfully discovered web + applications. + :vartype total_web_application_count: int + :ivar discovery_scope_status: Gets or sets status of partner discovery. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :vartype discovery_scope_status: str or ~azure.mgmt.migrate.models.DiscoveryScopeStatus + """ + + _attribute_map = { + "total_web_server_count": {"key": "totalWebServerCount", "type": "int"}, + "total_web_application_count": {"key": "totalWebApplicationCount", "type": "int"}, + "discovery_scope_status": {"key": "discoveryScopeStatus", "type": "str"}, + } + + def __init__( + self, + *, + total_web_server_count: Optional[int] = None, + total_web_application_count: Optional[int] = None, + discovery_scope_status: Optional[Union[str, "_models.DiscoveryScopeStatus"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword total_web_server_count: Gets or sets number of successfully discovered web servers.. + :paramtype total_web_server_count: int + :keyword total_web_application_count: Gets or sets number of successfully discovered web + applications. + :paramtype total_web_application_count: int + :keyword discovery_scope_status: Gets or sets status of partner discovery. Known values are: + "DiscoverySucceededAtleastOnce", "DiscoveryFailed", "RunAsAccountNotAssociated", + "DiscoveryNotStarted", "DiscoveryInProgress", "Disabled", "DiscoveryPartiallySucceded", and + "DiscoverySucceeded". + :paramtype discovery_scope_status: str or ~azure.mgmt.migrate.models.DiscoveryScopeStatus + """ + super().__init__(**kwargs) + self.total_web_server_count = total_web_server_count + self.total_web_application_count = total_web_application_count + self.discovery_scope_status = discovery_scope_status + + +class WebAppExtendedMachine(ProxyResource): + """Web app extended machine REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.WebAppExtendedMachineProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "WebAppExtendedMachineProperties"}, + } + + def __init__( + self, *, properties: Optional["_models.WebAppExtendedMachineProperties"] = None, **kwargs: Any + ) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.WebAppExtendedMachineProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class WebAppExtendedMachineListResult(_serialization.Model): + """The response of a WebAppExtendedMachine list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The WebAppExtendedMachine items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.WebAppExtendedMachine] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[WebAppExtendedMachine]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.WebAppExtendedMachine"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The WebAppExtendedMachine items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.WebAppExtendedMachine] + :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 WebAppExtendedMachineProperties(_serialization.Model): + """Class for web extended machine properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar is_deleted: Gets a value indicating whether application is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the timestamp marking creation of the entity. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last update operation. + :vartype updated_timestamp: str + :ivar errors: Gets the Error details. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar host_name: Gets the hydrated host name. + :vartype host_name: str + :ivar machine_display_name: Gets the Name of the extended machine. + :vartype machine_display_name: str + :ivar machine_id: Gets the machine ARM id. + :vartype machine_id: str + :ivar run_as_account_id: Gets the run as account ID of the machine used for web app discovery. + :vartype run_as_account_id: str + :ivar hydrated_run_as_account_id: Gets the agent hydrated run as account. + :vartype hydrated_run_as_account_id: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "errors": {"readonly": True}, + "host_name": {"readonly": True}, + "machine_display_name": {"readonly": True}, + "machine_id": {"readonly": True}, + "run_as_account_id": {"readonly": True}, + "hydrated_run_as_account_id": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "host_name": {"key": "hostName", "type": "str"}, + "machine_display_name": {"key": "machineDisplayName", "type": "str"}, + "machine_id": {"key": "machineId", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "hydrated_run_as_account_id": {"key": "hydratedRunAsAccountId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.errors = None + self.host_name = None + self.machine_display_name = None + self.machine_id = None + self.run_as_account_id = None + self.hydrated_run_as_account_id = None + self.provisioning_state = None + + +class WebApplication(ProxyResource): + """Web application REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.WebApplicationProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "WebApplicationProperties"}, + } + + def __init__(self, *, properties: Optional["_models.WebApplicationProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.WebApplicationProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class WebApplicationAppsAndRolesModel(_serialization.Model): + """WebApplication in the guest virtual machine. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar name: Gets or sets Name of the WebApplication. + :vartype name: str + :ivar status: Gets or sets Status of the WebApplication. + :vartype status: str + :ivar platform: Gets or sets Platform of the WebApplication. + :vartype platform: str + :ivar group_name: Gets or sets GroupName of the WebApplication. + :vartype group_name: str + :ivar web_server: Gets or sets WebServer of the WebApplication. + :vartype web_server: str + :ivar application_pool: Gets or sets ApplicationPool of the WebApplication. + :vartype application_pool: str + """ + + _validation = { + "name": {"readonly": True}, + "status": {"readonly": True}, + "platform": {"readonly": True}, + "group_name": {"readonly": True}, + "web_server": {"readonly": True}, + "application_pool": {"readonly": True}, + } + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "status": {"key": "status", "type": "str"}, + "platform": {"key": "platform", "type": "str"}, + "group_name": {"key": "groupName", "type": "str"}, + "web_server": {"key": "webServer", "type": "str"}, + "application_pool": {"key": "applicationPool", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.name = None + self.status = None + self.platform = None + self.group_name = None + self.web_server = None + self.application_pool = None + + +class WebApplicationConfigurationUnit(_serialization.Model): + """Web Application configuration unit. + + :ivar name: Gets or sets the configuration name. + :vartype name: str + :ivar file_path: Gets or sets the configuration file path. + :vartype file_path: str + :ivar local_file_path: Gets or sets the configuration local file path. + :vartype local_file_path: str + :ivar target_file_path: Gets or sets the configuration target file path. + :vartype target_file_path: str + :ivar section: Gets or sets the configuration section in the file. + :vartype section: str + :ivar type: Gets or sets the configuration type. + :vartype type: str + :ivar is_deployment_time_editable: Gets or sets a value indicating whether the configuration is + edited or not by + the user. + :vartype is_deployment_time_editable: bool + :ivar identifier: Gets or sets the identifier for the configuration. + :vartype identifier: str + """ + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "file_path": {"key": "filePath", "type": "str"}, + "local_file_path": {"key": "localFilePath", "type": "str"}, + "target_file_path": {"key": "targetFilePath", "type": "str"}, + "section": {"key": "section", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "is_deployment_time_editable": {"key": "isDeploymentTimeEditable", "type": "bool"}, + "identifier": {"key": "identifier", "type": "str"}, + } + + def __init__( + self, + *, + name: Optional[str] = None, + file_path: Optional[str] = None, + local_file_path: Optional[str] = None, + target_file_path: Optional[str] = None, + section: Optional[str] = None, + type: Optional[str] = None, + is_deployment_time_editable: Optional[bool] = None, + identifier: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword name: Gets or sets the configuration name. + :paramtype name: str + :keyword file_path: Gets or sets the configuration file path. + :paramtype file_path: str + :keyword local_file_path: Gets or sets the configuration local file path. + :paramtype local_file_path: str + :keyword target_file_path: Gets or sets the configuration target file path. + :paramtype target_file_path: str + :keyword section: Gets or sets the configuration section in the file. + :paramtype section: str + :keyword type: Gets or sets the configuration type. + :paramtype type: str + :keyword is_deployment_time_editable: Gets or sets a value indicating whether the configuration + is edited or not by + the user. + :paramtype is_deployment_time_editable: bool + :keyword identifier: Gets or sets the identifier for the configuration. + :paramtype identifier: str + """ + super().__init__(**kwargs) + self.name = name + self.file_path = file_path + self.local_file_path = local_file_path + self.target_file_path = target_file_path + self.section = section + self.type = type + self.is_deployment_time_editable = is_deployment_time_editable + self.identifier = identifier + + +class WebApplicationDirectoryUnit(_serialization.Model): + """Web Application directory unit. + + :ivar id: Gets or sets the unique id corresponding to the application directory. + :vartype id: str + :ivar is_editable: Gets or sets a value indicating whether the directory object is editable. + + .. code-block:: + + True when the directory is added as an optional directory, false when + + discovery is done + manually. + :vartype is_editable: bool + :ivar source_paths: Gets or sets the paths of the directory on the source machine. + :vartype source_paths: list[str] + :ivar local_scratch_path: Gets or sets the local scratch path at which the directories has been + copied. + :vartype local_scratch_path: str + :ivar mount_path: Gets or sets the mount path of the application directory. + :vartype mount_path: str + :ivar source_size: Gets or sets the size of the directory on the source machine. + :vartype source_size: str + """ + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "is_editable": {"key": "isEditable", "type": "bool"}, + "source_paths": {"key": "sourcePaths", "type": "[str]"}, + "local_scratch_path": {"key": "localScratchPath", "type": "str"}, + "mount_path": {"key": "mountPath", "type": "str"}, + "source_size": {"key": "sourceSize", "type": "str"}, + } + + def __init__( + self, + *, + id: Optional[str] = None, # pylint: disable=redefined-builtin + is_editable: Optional[bool] = None, + source_paths: Optional[List[str]] = None, + local_scratch_path: Optional[str] = None, + mount_path: Optional[str] = None, + source_size: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword id: Gets or sets the unique id corresponding to the application directory. + :paramtype id: str + :keyword is_editable: Gets or sets a value indicating whether the directory object is editable. + + .. code-block:: + + True when the directory is added as an optional directory, false when + + discovery is done + manually. + :paramtype is_editable: bool + :keyword source_paths: Gets or sets the paths of the directory on the source machine. + :paramtype source_paths: list[str] + :keyword local_scratch_path: Gets or sets the local scratch path at which the directories has + been copied. + :paramtype local_scratch_path: str + :keyword mount_path: Gets or sets the mount path of the application directory. + :paramtype mount_path: str + :keyword source_size: Gets or sets the size of the directory on the source machine. + :paramtype source_size: str + """ + super().__init__(**kwargs) + self.id = id + self.is_editable = is_editable + self.source_paths = source_paths + self.local_scratch_path = local_scratch_path + self.mount_path = mount_path + self.source_size = source_size + + +class WebApplicationFramework(_serialization.Model): + """Framework specific data for a web application. + + :ivar name: Gets or sets Name of the framework. + :vartype name: str + :ivar version: Gets or sets Version of the framework. + :vartype version: str + """ + + _attribute_map = { + "name": {"key": "name", "type": "str"}, + "version": {"key": "version", "type": "str"}, + } + + def __init__(self, *, name: Optional[str] = None, version: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword name: Gets or sets Name of the framework. + :paramtype name: str + :keyword version: Gets or sets Version of the framework. + :paramtype version: str + """ + super().__init__(**kwargs) + self.name = name + self.version = version + + +class WebApplicationListResult(_serialization.Model): + """The response of a WebApplication list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The WebApplication items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.WebApplication] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[WebApplication]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.WebApplication"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The WebApplication items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.WebApplication] + :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 WebApplicationProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for web application properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar web_server_id: Gets the web server id. + :vartype web_server_id: str + :ivar web_server_name: Gets the web server name. + :vartype web_server_name: str + :ivar machine_arm_ids: Gets the list of machine ARM Ids on which the SQL server is deployed. + :vartype machine_arm_ids: list[str] + :ivar virtual_path: Gets the virtual path of the application. + :vartype virtual_path: str + :ivar physical_path: Gets the physical path of the application. + :vartype physical_path: str + :ivar bindings: Gets the front end bindings for the application. + :vartype bindings: list[~azure.mgmt.migrate.models.FrontEndBinding] + :ivar frameworks: Gets the Error details. + :vartype frameworks: list[~azure.mgmt.migrate.models.WebApplicationFramework] + :ivar configurations: Gets the configuration. + :vartype configurations: list[~azure.mgmt.migrate.models.WebApplicationConfigurationUnit] + :ivar directories: Gets the directories. + :vartype directories: list[~azure.mgmt.migrate.models.WebApplicationDirectoryUnit] + :ivar display_name: Gets the display name. + :vartype display_name: str + :ivar tags: Gets tags that can be used with ODATA. + :vartype tags: dict[str, str] + :ivar static_folders: Static folders. + :vartype static_folders: list[str] + :ivar machine_display_name: Machine display name. + :vartype machine_display_name: str + :ivar is_deleted: Gets a value indicating whether application is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the timestamp marking creation of the entity. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last update operation. + :vartype updated_timestamp: str + :ivar server_type: Gets the server type. + :vartype server_type: str + :ivar errors: Gets the Error details. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar has_errors: Gets a value indicating whether the WebApp has errors or not. + :vartype has_errors: bool + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "web_server_id": {"readonly": True}, + "web_server_name": {"readonly": True}, + "machine_arm_ids": {"readonly": True}, + "virtual_path": {"readonly": True}, + "physical_path": {"readonly": True}, + "bindings": {"readonly": True}, + "frameworks": {"readonly": True}, + "configurations": {"readonly": True}, + "directories": {"readonly": True}, + "display_name": {"readonly": True}, + "tags": {"readonly": True}, + "static_folders": {"readonly": True}, + "machine_display_name": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "server_type": {"readonly": True}, + "errors": {"readonly": True}, + "appliance_names": {"readonly": True}, + "has_errors": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "web_server_id": {"key": "webServerId", "type": "str"}, + "web_server_name": {"key": "webServerName", "type": "str"}, + "machine_arm_ids": {"key": "machineArmIds", "type": "[str]"}, + "virtual_path": {"key": "virtualPath", "type": "str"}, + "physical_path": {"key": "physicalPath", "type": "str"}, + "bindings": {"key": "bindings", "type": "[FrontEndBinding]"}, + "frameworks": {"key": "frameworks", "type": "[WebApplicationFramework]"}, + "configurations": {"key": "configurations", "type": "[WebApplicationConfigurationUnit]"}, + "directories": {"key": "directories", "type": "[WebApplicationDirectoryUnit]"}, + "display_name": {"key": "displayName", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "static_folders": {"key": "staticFolders", "type": "[str]"}, + "machine_display_name": {"key": "machineDisplayName", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "server_type": {"key": "serverType", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "has_errors": {"key": "hasErrors", "type": "bool"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.web_server_id = None + self.web_server_name = None + self.machine_arm_ids = None + self.virtual_path = None + self.physical_path = None + self.bindings = None + self.frameworks = None + self.configurations = None + self.directories = None + self.display_name = None + self.tags = None + self.static_folders = None + self.machine_display_name = None + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.server_type = None + self.errors = None + self.appliance_names = None + self.has_errors = None + self.provisioning_state = None + + +class WebAppProperties(_serialization.Model): + """Class representing web app properties web model. + + :ivar web_app_arm_id: Gets or sets the arm id of the web app. + :vartype web_app_arm_id: str + :ivar tags: Gets or sets Tags. + :vartype tags: dict[str, str] + """ + + _attribute_map = { + "web_app_arm_id": {"key": "webAppArmId", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + } + + def __init__( + self, *, web_app_arm_id: Optional[str] = None, tags: Optional[Dict[str, str]] = None, **kwargs: Any + ) -> None: + """ + :keyword web_app_arm_id: Gets or sets the arm id of the web app. + :paramtype web_app_arm_id: str + :keyword tags: Gets or sets Tags. + :paramtype tags: dict[str, str] + """ + super().__init__(**kwargs) + self.web_app_arm_id = web_app_arm_id + self.tags = tags + + +class WebAppPropertiesCollection(_serialization.Model): + """Collection of web app properties. + + :ivar web_apps: Gets or sets the list of web app properties. + :vartype web_apps: list[~azure.mgmt.migrate.models.WebAppProperties] + """ + + _attribute_map = { + "web_apps": {"key": "webApps", "type": "[WebAppProperties]"}, + } + + def __init__(self, *, web_apps: Optional[List["_models.WebAppProperties"]] = None, **kwargs: Any) -> None: + """ + :keyword web_apps: Gets or sets the list of web app properties. + :paramtype web_apps: list[~azure.mgmt.migrate.models.WebAppProperties] + """ + super().__init__(**kwargs) + self.web_apps = web_apps + + +class WebAppRunAsAccount(ProxyResource): + """Run as account REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "RunAsAccountProperties"}, + } + + def __init__(self, *, properties: Optional["_models.RunAsAccountProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.RunAsAccountProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class WebAppRunAsAccountListResult(_serialization.Model): + """The response of a WebAppRunAsAccount list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The WebAppRunAsAccount items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.WebAppRunAsAccount] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[WebAppRunAsAccount]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__( + self, *, value: List["_models.WebAppRunAsAccount"], next_link: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword value: The WebAppRunAsAccount items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.WebAppRunAsAccount] + :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 WebAppSite(ProxyResource): + """WebApp site web model. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.WebAppSiteProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "WebAppSiteProperties"}, + } + + def __init__(self, *, properties: Optional["_models.WebAppSiteProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.WebAppSiteProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class WebAppSiteListResult(_serialization.Model): + """The response of a WebAppSite list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The WebAppSite items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.WebAppSite] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[WebAppSite]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.WebAppSite"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The WebAppSite items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.WebAppSite] + :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 WebAppSiteProperties(_serialization.Model): + """Class for Web app site properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar site_appliance_properties_collection: Gets or sets the appliance details used by service + to communicate + + to the appliance. + :vartype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :ivar discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" and + "DR". + :vartype discovery_scenario: str or + ~azure.mgmt.migrate.models.WebAppSitePropertiesDiscoveryScenario + :ivar service_endpoint: Gets the service endpoint. + :vartype service_endpoint: str + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "service_endpoint": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "site_appliance_properties_collection": { + "key": "siteAppliancePropertiesCollection", + "type": "[SiteApplianceProperties]", + }, + "discovery_scenario": {"key": "discoveryScenario", "type": "str"}, + "service_endpoint": {"key": "serviceEndpoint", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__( + self, + *, + site_appliance_properties_collection: Optional[List["_models.SiteApplianceProperties"]] = None, + discovery_scenario: Optional[Union[str, "_models.WebAppSitePropertiesDiscoveryScenario"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword site_appliance_properties_collection: Gets or sets the appliance details used by + service to communicate + + to the appliance. + :paramtype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :keyword discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" + and "DR". + :paramtype discovery_scenario: str or + ~azure.mgmt.migrate.models.WebAppSitePropertiesDiscoveryScenario + """ + super().__init__(**kwargs) + self.site_appliance_properties_collection = site_appliance_properties_collection + self.discovery_scenario = discovery_scenario + self.service_endpoint = None + self.provisioning_state = None + + +class WebAppSiteUpdate(_serialization.Model): + """The type used for update operations of the WebAppSite. + + :ivar properties: The updatable properties of the WebAppSite. + :vartype properties: ~azure.mgmt.migrate.models.WebAppSiteUpdateProperties + """ + + _attribute_map = { + "properties": {"key": "properties", "type": "WebAppSiteUpdateProperties"}, + } + + def __init__(self, *, properties: Optional["_models.WebAppSiteUpdateProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The updatable properties of the WebAppSite. + :paramtype properties: ~azure.mgmt.migrate.models.WebAppSiteUpdateProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class WebAppSiteUpdateProperties(_serialization.Model): + """The updatable properties of the WebAppSite. + + :ivar site_appliance_properties_collection: Gets or sets the appliance details used by service + to communicate + + to the appliance. + :vartype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :ivar discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" and + "DR". + :vartype discovery_scenario: str or + ~azure.mgmt.migrate.models.WebAppSitePropertiesDiscoveryScenario + """ + + _attribute_map = { + "site_appliance_properties_collection": { + "key": "siteAppliancePropertiesCollection", + "type": "[SiteApplianceProperties]", + }, + "discovery_scenario": {"key": "discoveryScenario", "type": "str"}, + } + + def __init__( + self, + *, + site_appliance_properties_collection: Optional[List["_models.SiteApplianceProperties"]] = None, + discovery_scenario: Optional[Union[str, "_models.WebAppSitePropertiesDiscoveryScenario"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword site_appliance_properties_collection: Gets or sets the appliance details used by + service to communicate + + to the appliance. + :paramtype site_appliance_properties_collection: + list[~azure.mgmt.migrate.models.SiteApplianceProperties] + :keyword discovery_scenario: Gets or sets the discovery scenario. Known values are: "Migrate" + and "DR". + :paramtype discovery_scenario: str or + ~azure.mgmt.migrate.models.WebAppSitePropertiesDiscoveryScenario + """ + super().__init__(**kwargs) + self.site_appliance_properties_collection = site_appliance_properties_collection + self.discovery_scenario = discovery_scenario + + +class WebAppSiteUsage(_serialization.Model): + """Web app site usage. + + :ivar web_server_count: Gets or sets the number of web servers discovered in the site. + :vartype web_server_count: int + :ivar web_application_count: Gets or sets the number of web applications discovered in the + site. + :vartype web_application_count: int + :ivar run_as_account_count: Gets or sets the number of run as accounts in the site. + :vartype run_as_account_count: int + """ + + _attribute_map = { + "web_server_count": {"key": "webServerCount", "type": "int"}, + "web_application_count": {"key": "webApplicationCount", "type": "int"}, + "run_as_account_count": {"key": "runAsAccountCount", "type": "int"}, + } + + def __init__( + self, + *, + web_server_count: Optional[int] = None, + web_application_count: Optional[int] = None, + run_as_account_count: Optional[int] = None, + **kwargs: Any + ) -> None: + """ + :keyword web_server_count: Gets or sets the number of web servers discovered in the site. + :paramtype web_server_count: int + :keyword web_application_count: Gets or sets the number of web applications discovered in the + site. + :paramtype web_application_count: int + :keyword run_as_account_count: Gets or sets the number of run as accounts in the site. + :paramtype run_as_account_count: int + """ + super().__init__(**kwargs) + self.web_server_count = web_server_count + self.web_application_count = web_application_count + self.run_as_account_count = run_as_account_count + + +class WebRoleImportDisk(_serialization.Model): + """Class representing the imported machine web model. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar megabytes_per_second_of_read: Disk read throughput. + :vartype megabytes_per_second_of_read: float + :ivar megabytes_per_second_of_write: Disk write throughput. + :vartype megabytes_per_second_of_write: float + :ivar number_of_read_operations_per_second: Disk read IOPS. + :vartype number_of_read_operations_per_second: float + :ivar number_of_write_operations_per_second: Disk write IOPS. + :vartype number_of_write_operations_per_second: float + :ivar max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :vartype max_size_in_bytes: int + :ivar name: Gets or sets Name of the disk. + :vartype name: str + :ivar disk_type: Gets or sets Type of the disk. + :vartype disk_type: str + :ivar lun: Gets or sets LUN of the disk. + :vartype lun: int + :ivar path: Gets or sets Path of the disk. + :vartype path: str + """ + + _validation = { + "megabytes_per_second_of_read": {"readonly": True}, + "megabytes_per_second_of_write": {"readonly": True}, + "number_of_read_operations_per_second": {"readonly": True}, + "number_of_write_operations_per_second": {"readonly": True}, + } + + _attribute_map = { + "megabytes_per_second_of_read": {"key": "megabytesPerSecondOfRead", "type": "float"}, + "megabytes_per_second_of_write": {"key": "megabytesPerSecondOfWrite", "type": "float"}, + "number_of_read_operations_per_second": {"key": "numberOfReadOperationsPerSecond", "type": "float"}, + "number_of_write_operations_per_second": {"key": "numberOfWriteOperationsPerSecond", "type": "float"}, + "max_size_in_bytes": {"key": "maxSizeInBytes", "type": "int"}, + "name": {"key": "name", "type": "str"}, + "disk_type": {"key": "diskType", "type": "str"}, + "lun": {"key": "lun", "type": "int"}, + "path": {"key": "path", "type": "str"}, + } + + def __init__( + self, + *, + max_size_in_bytes: Optional[int] = None, + name: Optional[str] = None, + disk_type: Optional[str] = None, + lun: Optional[int] = None, + path: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword max_size_in_bytes: Gets or sets Bytes allocated for the disk. + :paramtype max_size_in_bytes: int + :keyword name: Gets or sets Name of the disk. + :paramtype name: str + :keyword disk_type: Gets or sets Type of the disk. + :paramtype disk_type: str + :keyword lun: Gets or sets LUN of the disk. + :paramtype lun: int + :keyword path: Gets or sets Path of the disk. + :paramtype path: str + """ + super().__init__(**kwargs) + self.megabytes_per_second_of_read = None + self.megabytes_per_second_of_write = None + self.number_of_read_operations_per_second = None + self.number_of_write_operations_per_second = None + self.max_size_in_bytes = max_size_in_bytes + self.name = name + self.disk_type = disk_type + self.lun = lun + self.path = path + + +class WebRoleOperatingSystem(_serialization.Model): + """Second level object returned as part of Machine REST resource. + + :ivar os_type: Gets or sets the type of the operating system. + :vartype os_type: str + :ivar os_name: Gets or sets the Name of the operating system. + :vartype os_name: str + :ivar os_version: Gets or sets the Version of the operating system. + :vartype os_version: str + :ivar os_architecture: Gets or sets the Architecture of the operating system. + :vartype os_architecture: str + """ + + _attribute_map = { + "os_type": {"key": "osType", "type": "str"}, + "os_name": {"key": "osName", "type": "str"}, + "os_version": {"key": "osVersion", "type": "str"}, + "os_architecture": {"key": "osArchitecture", "type": "str"}, + } + + def __init__( + self, + *, + os_type: Optional[str] = None, + os_name: Optional[str] = None, + os_version: Optional[str] = None, + os_architecture: Optional[str] = None, + **kwargs: Any + ) -> None: + """ + :keyword os_type: Gets or sets the type of the operating system. + :paramtype os_type: str + :keyword os_name: Gets or sets the Name of the operating system. + :paramtype os_name: str + :keyword os_version: Gets or sets the Version of the operating system. + :paramtype os_version: str + :keyword os_architecture: Gets or sets the Architecture of the operating system. + :paramtype os_architecture: str + """ + super().__init__(**kwargs) + self.os_type = os_type + self.os_name = os_name + self.os_version = os_version + self.os_architecture = os_architecture + + +class WebServer(ProxyResource): + """Web application REST Resource. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. # pylint: disable=line-too-long + :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.migrate.models.SystemData + :ivar properties: The resource-specific properties for this resource. + :vartype properties: ~azure.mgmt.migrate.models.WebServerProperties + """ + + _validation = { + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "system_data": {"readonly": True}, + } + + _attribute_map = { + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "system_data": {"key": "systemData", "type": "SystemData"}, + "properties": {"key": "properties", "type": "WebServerProperties"}, + } + + def __init__(self, *, properties: Optional["_models.WebServerProperties"] = None, **kwargs: Any) -> None: + """ + :keyword properties: The resource-specific properties for this resource. + :paramtype properties: ~azure.mgmt.migrate.models.WebServerProperties + """ + super().__init__(**kwargs) + self.properties = properties + + +class WebServerListResult(_serialization.Model): + """The response of a WebServer list operation. + + All required parameters must be populated in order to send to server. + + :ivar value: The WebServer items on this page. Required. + :vartype value: list[~azure.mgmt.migrate.models.WebServer] + :ivar next_link: The link to the next page of items. + :vartype next_link: str + """ + + _validation = { + "value": {"required": True}, + } + + _attribute_map = { + "value": {"key": "value", "type": "[WebServer]"}, + "next_link": {"key": "nextLink", "type": "str"}, + } + + def __init__(self, *, value: List["_models.WebServer"], next_link: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword value: The WebServer items on this page. Required. + :paramtype value: list[~azure.mgmt.migrate.models.WebServer] + :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 WebServerProperties(_serialization.Model): # pylint: disable=too-many-instance-attributes + """Class for web server properties. + + Variables are only populated by the server, and will be ignored when sending a request. + + :ivar configuration_location: Gets the configuration location. + :vartype configuration_location: str + :ivar version: Gets the configuration location. + :vartype version: str + :ivar machine_ids: Gets the list of machines. + :vartype machine_ids: list[str] + :ivar web_applications: Gets the list of web applications. + :vartype web_applications: list[str] + :ivar display_name: Gets the display name. + :vartype display_name: str + :ivar server_fqdn: Gets the server FQDN. + :vartype server_fqdn: str + :ivar run_as_account_id: Gets the run as account id. + :vartype run_as_account_id: str + :ivar is_deleted: Gets a value indicating whether application is deleted. + :vartype is_deleted: bool + :ivar created_timestamp: Gets the timestamp marking creation of the entity. + :vartype created_timestamp: str + :ivar updated_timestamp: Gets the timestamp marking last update operation. + :vartype updated_timestamp: str + :ivar server_type: Gets the server type. + :vartype server_type: str + :ivar errors: Gets the Error details. + :vartype errors: list[~azure.mgmt.migrate.models.HealthErrorDetails] + :ivar appliance_names: Gets the appliance names. + :vartype appliance_names: list[str] + :ivar has_errors: Gets a value indicating whether the WebApp has errors or not. + :vartype has_errors: bool + :ivar provisioning_state: provisioning state enum. Known values are: "Created", "Updated", + "Running", "Completed", "Failed", "Succeeded", and "Canceled". + :vartype provisioning_state: str or ~azure.mgmt.migrate.models.ProvisioningState + """ + + _validation = { + "configuration_location": {"readonly": True}, + "version": {"readonly": True}, + "machine_ids": {"readonly": True}, + "web_applications": {"readonly": True}, + "display_name": {"readonly": True}, + "server_fqdn": {"readonly": True}, + "run_as_account_id": {"readonly": True}, + "is_deleted": {"readonly": True}, + "created_timestamp": {"readonly": True}, + "updated_timestamp": {"readonly": True}, + "server_type": {"readonly": True}, + "errors": {"readonly": True}, + "appliance_names": {"readonly": True}, + "has_errors": {"readonly": True}, + "provisioning_state": {"readonly": True}, + } + + _attribute_map = { + "configuration_location": {"key": "configurationLocation", "type": "str"}, + "version": {"key": "version", "type": "str"}, + "machine_ids": {"key": "machineIds", "type": "[str]"}, + "web_applications": {"key": "webApplications", "type": "[str]"}, + "display_name": {"key": "displayName", "type": "str"}, + "server_fqdn": {"key": "serverFqdn", "type": "str"}, + "run_as_account_id": {"key": "runAsAccountId", "type": "str"}, + "is_deleted": {"key": "isDeleted", "type": "bool"}, + "created_timestamp": {"key": "createdTimestamp", "type": "str"}, + "updated_timestamp": {"key": "updatedTimestamp", "type": "str"}, + "server_type": {"key": "serverType", "type": "str"}, + "errors": {"key": "errors", "type": "[HealthErrorDetails]"}, + "appliance_names": {"key": "applianceNames", "type": "[str]"}, + "has_errors": {"key": "hasErrors", "type": "bool"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) + self.configuration_location = None + self.version = None + self.machine_ids = None + self.web_applications = None + self.display_name = None + self.server_fqdn = None + self.run_as_account_id = None + self.is_deleted = None + self.created_timestamp = None + self.updated_timestamp = None + self.server_type = None + self.errors = None + self.appliance_names = None + self.has_errors = None + self.provisioning_state = None diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_patch.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/models/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/__init__.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/__init__.py new file mode 100644 index 000000000000..4f913fc50d15 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/__init__.py @@ -0,0 +1,127 @@ +# 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 ._operations import Operations +from ._hyperv_sites_operations import HypervSitesOperations +from ._import_sites_controller_operations import ImportSitesControllerOperations +from ._master_sites_controller_operations import MasterSitesControllerOperations +from ._server_sites_controller_operations import ServerSitesControllerOperations +from ._sites_controller_operations import SitesControllerOperations +from ._hyperv_sites_controller_operations import HypervSitesControllerOperations +from ._hyperv_dependency_map_controller_operations import HypervDependencyMapControllerOperations +from ._hyperv_cluster_controller_operations import HypervClusterControllerOperations +from ._hyperv_host_controller_operations import HypervHostControllerOperations +from ._hyperv_jobs_operations import HypervJobsOperations +from ._hyperv_machines_controller_operations import HypervMachinesControllerOperations +from ._hyperv_software_inventories_controller_operations import HypervSoftwareInventoriesControllerOperations +from ._hyperv_operations_status_controller_operations import HypervOperationsStatusControllerOperations +from ._hyperv_run_as_accounts_controller_operations import HypervRunAsAccountsControllerOperations +from ._import_jobs_controller_operations import ImportJobsControllerOperations +from ._import_machines_controller_operations import ImportMachinesControllerOperations +from ._master_sites_operations_status_controller_operations import MasterSitesOperationsStatusControllerOperations +from ._private_endpoint_connection_controller_operations import PrivateEndpointConnectionControllerOperations +from ._private_link_resources_controller_operations import PrivateLinkResourcesControllerOperations +from ._sql_sites_controller_operations import SqlSitesControllerOperations +from ._sql_discovery_site_data_source_controller_operations import SqlDiscoverySiteDataSourceControllerOperations +from ._sql_jobs_controller_operations import SqlJobsControllerOperations +from ._sql_operations_status_controller_operations import SqlOperationsStatusControllerOperations +from ._sql_run_as_accounts_controller_operations import SqlRunAsAccountsControllerOperations +from ._sql_availability_groups_controller_operations import SqlAvailabilityGroupsControllerOperations +from ._sql_databases_controller_operations import SqlDatabasesControllerOperations +from ._sql_servers_controller_operations import SqlServersControllerOperations +from ._web_app_sites_controller_operations import WebAppSitesControllerOperations +from ._web_app_discovery_site_data_sources_controller_operations import ( + WebAppDiscoverySiteDataSourcesControllerOperations, +) +from ._web_app_extended_machines_controller_operations import WebAppExtendedMachinesControllerOperations +from ._iis_web_applications_controller_operations import IisWebApplicationsControllerOperations +from ._iis_web_servers_controller_operations import IisWebServersControllerOperations +from ._web_app_run_as_accounts_controller_operations import WebAppRunAsAccountsControllerOperations +from ._tomcat_web_applications_controller_operations import TomcatWebApplicationsControllerOperations +from ._tomcat_web_servers_controller_operations import TomcatWebServersControllerOperations +from ._web_app_properties_controller_operations import WebAppPropertiesControllerOperations +from ._web_applications_controller_operations import WebApplicationsControllerOperations +from ._web_servers_controller_operations import WebServersControllerOperations +from ._server_dependency_map_controller_operations import ServerDependencyMapControllerOperations +from ._server_jobs_controller_operations import ServerJobsControllerOperations +from ._servers_controller_operations import ServersControllerOperations +from ._server_software_inventories_controller_operations import ServerSoftwareInventoriesControllerOperations +from ._server_operations_status_controller_operations import ServerOperationsStatusControllerOperations +from ._server_run_as_accounts_controller_operations import ServerRunAsAccountsControllerOperations +from ._dependency_map_controller_operations import DependencyMapControllerOperations +from ._vmware_host_controller_operations import VmwareHostControllerOperations +from ._hyperv_jobs_controller_operations import HypervJobsControllerOperations +from ._machines_controller_operations import MachinesControllerOperations +from ._vmware_software_inventories_controller_operations import VmwareSoftwareInventoriesControllerOperations +from ._vmware_operations_status_operations import VmwareOperationsStatusOperations +from ._run_as_accounts_controller_operations import RunAsAccountsControllerOperations +from ._vmware_properties_controller_operations import VmwarePropertiesControllerOperations +from ._vcenter_controller_operations import VcenterControllerOperations + +from ._patch import __all__ as _patch_all +from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import patch_sdk as _patch_sdk + +__all__ = [ + "Operations", + "HypervSitesOperations", + "ImportSitesControllerOperations", + "MasterSitesControllerOperations", + "ServerSitesControllerOperations", + "SitesControllerOperations", + "HypervSitesControllerOperations", + "HypervDependencyMapControllerOperations", + "HypervClusterControllerOperations", + "HypervHostControllerOperations", + "HypervJobsOperations", + "HypervMachinesControllerOperations", + "HypervSoftwareInventoriesControllerOperations", + "HypervOperationsStatusControllerOperations", + "HypervRunAsAccountsControllerOperations", + "ImportJobsControllerOperations", + "ImportMachinesControllerOperations", + "MasterSitesOperationsStatusControllerOperations", + "PrivateEndpointConnectionControllerOperations", + "PrivateLinkResourcesControllerOperations", + "SqlSitesControllerOperations", + "SqlDiscoverySiteDataSourceControllerOperations", + "SqlJobsControllerOperations", + "SqlOperationsStatusControllerOperations", + "SqlRunAsAccountsControllerOperations", + "SqlAvailabilityGroupsControllerOperations", + "SqlDatabasesControllerOperations", + "SqlServersControllerOperations", + "WebAppSitesControllerOperations", + "WebAppDiscoverySiteDataSourcesControllerOperations", + "WebAppExtendedMachinesControllerOperations", + "IisWebApplicationsControllerOperations", + "IisWebServersControllerOperations", + "WebAppRunAsAccountsControllerOperations", + "TomcatWebApplicationsControllerOperations", + "TomcatWebServersControllerOperations", + "WebAppPropertiesControllerOperations", + "WebApplicationsControllerOperations", + "WebServersControllerOperations", + "ServerDependencyMapControllerOperations", + "ServerJobsControllerOperations", + "ServersControllerOperations", + "ServerSoftwareInventoriesControllerOperations", + "ServerOperationsStatusControllerOperations", + "ServerRunAsAccountsControllerOperations", + "DependencyMapControllerOperations", + "VmwareHostControllerOperations", + "HypervJobsControllerOperations", + "MachinesControllerOperations", + "VmwareSoftwareInventoriesControllerOperations", + "VmwareOperationsStatusOperations", + "RunAsAccountsControllerOperations", + "VmwarePropertiesControllerOperations", + "VcenterControllerOperations", +] +__all__.extend([p for p in _patch_all if p not in __all__]) +_patch_sdk() diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_dependency_map_controller_operations.py new file mode 100644 index 000000000000..ed470ed8284d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_dependency_map_controller_operations.py @@ -0,0 +1,1234 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_client_group_members_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/clientGroupMembers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_export_dependencies_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/exportDependencies", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_generate_coarse_map_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/generateCoarseMap", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_generate_detailed_map_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/generateDetailedMap", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_server_group_members_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/serverGroupMembers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 DependencyMapControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`dependency_map_controller` 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 _client_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsClientGroupMembersRequest") + + _request = build_client_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsClientGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsClientGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._client_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_dependencies_initial( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **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(request_body, (IOBase, bytes)): + _content = request_body + else: + _json = self._serialize.body(request_body, "DependencyMapServiceMapextensionsExportDependenciesRequest") + + _request = build_export_dependencies_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: _models.DependencyMapServiceMapextensionsExportDependenciesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Is either a + DependencyMapServiceMapextensionsExportDependenciesRequest type or a IO[bytes] type. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_dependencies_initial( + resource_group_name=resource_group_name, + site_name=site_name, + request_body=request_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _generate_coarse_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsScopeMapRequest") + + _request = build_generate_coarse_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsScopeMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsScopeMapRequest type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._generate_coarse_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _generate_detailed_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body( + map_request, "DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest" + ) + + _request = build_generate_detailed_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest type or a IO[bytes] type. + Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._generate_detailed_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _server_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsServerGroupMembersRequest") + + _request = build_server_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsServerGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsServerGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._server_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_cluster_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_cluster_controller_operations.py new file mode 100644 index 000000000000..bb34909352b2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_cluster_controller_operations.py @@ -0,0 +1,620 @@ +# 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_by_hyperv_site_request( + resource_group_name: str, site_name: str, subscription_id: str, *, filter: 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/clusters", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_get_cluster_request( + resource_group_name: str, site_name: str, cluster_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/clusters/{clusterName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_cluster_request( + resource_group_name: str, site_name: str, cluster_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/clusters/{clusterName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, cluster_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/clusters/{clusterName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 HypervClusterControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_cluster_controller` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, filter: Optional[str] = None, **kwargs: Any + ) -> Iterable["_models.HypervCluster"]: + """List HypervCluster resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :return: An iterator like instance of either HypervCluster or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervCluster] + :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.HypervClusterListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + 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("HypervClusterListResult", 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_cluster( + self, resource_group_name: str, site_name: str, cluster_name: str, **kwargs: Any + ) -> _models.HypervCluster: + """Method to get a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :return: HypervCluster or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervCluster + :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.HypervCluster] = kwargs.pop("cls", None) + + _request = build_get_cluster_request( + resource_group_name=resource_group_name, + site_name=site_name, + cluster_name=cluster_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("HypervCluster", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _create_cluster_initial( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: Union[_models.HypervCluster, IO[bytes]], + **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: + _json = self._serialize.body(body, "HypervCluster") + + _request = build_create_cluster_request( + resource_group_name=resource_group_name, + site_name=site_name, + cluster_name=cluster_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create_cluster( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: _models.HypervCluster, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.HypervCluster]: + """Method to create or update a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.HypervCluster + :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 HypervCluster or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.HypervCluster] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create_cluster( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.HypervCluster]: + """Method to create or update a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :param body: Resource create 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 either HypervCluster or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.HypervCluster] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create_cluster( + self, + resource_group_name: str, + site_name: str, + cluster_name: str, + body: Union[_models.HypervCluster, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.HypervCluster]: + """Method to create or update a Hyper-V cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_name: str + :param body: Resource create parameters. Is either a HypervCluster type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.HypervCluster or IO[bytes] + :return: An instance of LROPoller that returns either HypervCluster or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.HypervCluster] + :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.HypervCluster] = 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_cluster_initial( + resource_group_name=resource_group_name, + site_name=site_name, + cluster_name=cluster_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("HypervCluster", 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.HypervCluster].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.HypervCluster]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, cluster_name: str, **kwargs: Any + ) -> None: + """Delete a HypervCluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param cluster_name: Cluster ARM name. Required. + :type cluster_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, + site_name=site_name, + cluster_name=cluster_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_dependency_map_controller_operations.py new file mode 100644 index 000000000000..a7634388f5af --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_dependency_map_controller_operations.py @@ -0,0 +1,1468 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_client_group_members_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/clientGroupMembers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_export_dependencies_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/exportDependencies", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_generate_coarse_map_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/generateCoarseMap", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_generate_detailed_map_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/generateDetailedMap", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_server_group_members_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/serverGroupMembers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_dependency_map_status_request( # pylint: disable=name-too-long + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/updateDependencyMapStatus", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 HypervDependencyMapControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_dependency_map_controller` 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 _client_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsClientGroupMembersRequest") + + _request = build_client_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsClientGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsClientGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._client_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_dependencies_initial( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **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(request_body, (IOBase, bytes)): + _content = request_body + else: + _json = self._serialize.body(request_body, "DependencyMapServiceMapextensionsExportDependenciesRequest") + + _request = build_export_dependencies_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: _models.DependencyMapServiceMapextensionsExportDependenciesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Is either a + DependencyMapServiceMapextensionsExportDependenciesRequest type or a IO[bytes] type. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_dependencies_initial( + resource_group_name=resource_group_name, + site_name=site_name, + request_body=request_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _generate_coarse_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsScopeMapRequest") + + _request = build_generate_coarse_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsScopeMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsScopeMapRequest type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._generate_coarse_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _generate_detailed_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body( + map_request, "DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest" + ) + + _request = build_generate_detailed_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest type or a IO[bytes] type. + Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._generate_detailed_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _server_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsServerGroupMembersRequest") + + _request = build_server_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsServerGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsServerGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._server_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _update_dependency_map_status_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "UpdateMachineDepMapStatus") + + _request = build_update_dependency_map_status_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: _models.UpdateMachineDepMapStatus, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a UpdateMachineDepMapStatus + type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_dependency_map_status_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_host_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_host_controller_operations.py new file mode 100644 index 000000000000..4144d7e8165c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_host_controller_operations.py @@ -0,0 +1,673 @@ +# 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_by_hyperv_site_request( + resource_group_name: str, site_name: str, subscription_id: str, *, filter: 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/hosts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "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, site_name: str, host_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/hosts/{hostName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "hostName": _SERIALIZER.url("host_name", host_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, host_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/hosts/{hostName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "hostName": _SERIALIZER.url("host_name", host_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, host_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/hosts/{hostName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "hostName": _SERIALIZER.url("host_name", host_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 HypervHostControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_host_controller` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, filter: Optional[str] = None, **kwargs: Any + ) -> Iterable["_models.HypervHost"]: + """List HypervHost resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :return: An iterator like instance of either HypervHost or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervHost] + :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.HypervHostListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + 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("HypervHostListResult", 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, site_name: str, host_name: str, **kwargs: Any) -> _models.HypervHost: + """Get a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :return: HypervHost or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervHost + :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.HypervHost] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + host_name=host_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("HypervHost", 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, + site_name: str, + host_name: str, + body: Union[_models.HypervHost, IO[bytes]], + **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: + _json = self._serialize.body(body, "HypervHost") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + host_name=host_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + host_name: str, + body: _models.HypervHost, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.HypervHost]: + """Create a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.HypervHost + :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 HypervHost or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.HypervHost] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + site_name: str, + host_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.HypervHost]: + """Create a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :param body: Resource create 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 either HypervHost or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.HypervHost] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + site_name: str, + host_name: str, + body: Union[_models.HypervHost, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.HypervHost]: + """Create a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_name: str + :param body: Resource create parameters. Is either a HypervHost type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.HypervHost or IO[bytes] + :return: An instance of LROPoller that returns either HypervHost or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.HypervHost] + :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.HypervHost] = 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, + site_name=site_name, + host_name=host_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("HypervHost", 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.HypervHost].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.HypervHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _delete_initial( + self, resource_group_name: str, site_name: str, host_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, + site_name=site_name, + host_name=host_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, site_name: str, host_name: str, **kwargs: Any) -> LROPoller[None]: + """Delete a HypervHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Host name. Required. + :type host_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, + site_name=site_name, + host_name=host_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/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_controller_operations.py new file mode 100644 index 000000000000..1ef9a3029676 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_controller_operations.py @@ -0,0 +1,269 @@ +# 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_by_vmware_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/jobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/jobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class HypervJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_jobs_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.VmwareJob"]: + """List VmwareJob resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either VmwareJob or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.VmwareJob] + :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.VmwareJobListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.VmwareJob: + """Get a VmwareJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Jobs name. Required. + :type job_name: str + :return: VmwareJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareJob + :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.VmwareJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("VmwareJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_operations.py new file mode 100644 index 000000000000..77dda4597dc9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_jobs_operations.py @@ -0,0 +1,269 @@ +# 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_by_hyperv_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/jobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/jobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class HypervJobsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_jobs` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.HypervJob"]: + """List HypervJob resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either HypervJob or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervJob] + :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.HypervJobListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.HypervJob: + """Get a HypervJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job name. Required. + :type job_name: str + :return: HypervJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervJob + :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.HypervJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("HypervJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_machines_controller_operations.py new file mode 100644 index 000000000000..00ef2b4d16d4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_machines_controller_operations.py @@ -0,0 +1,725 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_hyperv_site_request( + resource_group_name: str, + site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[int] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/machines", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "int") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_properties_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/updateProperties", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 HypervMachinesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_machines_controller` 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_by_hyperv_site( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[int] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.HypervMachine"]: + """List HypervMachine resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: int + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either HypervMachine or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervMachine] + :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.HypervMachineListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("HypervMachineListResult", 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, site_name: str, machine_name: str, **kwargs: Any) -> _models.HypervMachine: + """Get a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :return: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :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.HypervMachine] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("HypervMachine", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: _models.HypervMachineUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervMachine: + """Update a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.HypervMachineUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervMachine: + """Update a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. 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: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: Union[_models.HypervMachineUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.HypervMachine: + """Update a HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Is either a HypervMachineUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.HypervMachineUpdate or IO[bytes] + :return: HypervMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervMachine + :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.HypervMachine] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "HypervMachineUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("HypervMachine", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _update_properties_initial( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **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(meta_data, (IOBase, bytes)): + _content = meta_data + else: + _json = self._serialize.body(meta_data, "MachineMetadataCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: _models.MachineMetadataCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update custom properties for HYPERV machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update custom properties for HYPERV machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update custom properties for HYPERV machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Is either a MachineMetadataCollection type + or a IO[bytes] type. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + meta_data=meta_data, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_operations_status_controller_operations.py new file mode 100644 index 000000000000..4a5bbf87d1b0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_operations_status_controller_operations.py @@ -0,0 +1,154 @@ +# 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_hyperv_operations_status_request( # pylint: disable=name-too-long + resource_group_name: str, site_name: str, operation_status_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/operationsStatus/{operationStatusName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "operationStatusName": _SERIALIZER.url( + "operation_status_name", operation_status_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class HypervOperationsStatusControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_operations_status_controller` 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_hyperv_operations_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """Method to get operation status. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_hyperv_operations_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..8ba40b84dfe9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_run_as_accounts_controller_operations.py @@ -0,0 +1,272 @@ +# 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_by_hyperv_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/runAsAccounts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, account_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/runAsAccounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "accountName": _SERIALIZER.url("account_name", account_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class HypervRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_run_as_accounts_controller` 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_by_hyperv_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.HypervRunAsAccountResource"]: + """List HypervRunAsAccountResource resources by HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either HypervRunAsAccountResource or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervRunAsAccountResource] + :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.HypervRunAsAccountResourceListResult] = 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_hyperv_site_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervRunAsAccountResourceListResult", 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, site_name: str, account_name: str, **kwargs: Any + ) -> _models.HypervRunAsAccountResource: + """Get a HypervRunAsAccountResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param account_name: RunAsAccounts name. Required. + :type account_name: str + :return: HypervRunAsAccountResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervRunAsAccountResource + :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.HypervRunAsAccountResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + account_name=account_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("HypervRunAsAccountResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_controller_operations.py new file mode 100644 index 000000000000..9e173b430fe0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_controller_operations.py @@ -0,0 +1,1337 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_compute_error_summary_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/computeErrorSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_computeusage_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/computeusage", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_applications_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/exportApplications", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_machine_errors_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/hypervSites/{siteName}/exportMachineErrors", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_list_health_summary_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/listHealthSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_summary_request(resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/summary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 HypervSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_sites_controller` 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, site_name: str, **kwargs: Any) -> _models.HypervSite: + """Get a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :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.HypervSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + body: _models.HypervSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Create a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.HypervSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Create a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create( + self, resource_group_name: str, site_name: str, body: Union[_models.HypervSite, IO[bytes]], **kwargs: Any + ) -> _models.HypervSite: + """Create a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a HypervSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.HypervSite or IO[bytes] + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :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.HypervSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "HypervSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("HypervSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: _models.HypervSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Update a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.HypervSiteUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.HypervSite: + """Update a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, resource_group_name: str, site_name: str, body: Union[_models.HypervSiteUpdate, IO[bytes]], **kwargs: Any + ) -> _models.HypervSite: + """Update a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a HypervSiteUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.HypervSiteUpdate or IO[bytes] + :return: HypervSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSite + :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.HypervSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "HypervSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("HypervSite", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a HypervSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def compute_error_summary( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get site error summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SiteErrorSummary] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_compute_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def computeusage( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.HypervSiteUsage: + """Get a hyperv site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: HypervSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSiteUsage + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.HypervSiteUsage] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_computeusage_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("HypervSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _export_applications_initial( + self, resource_group_name: str, site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_applications_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_applications( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_applications_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_machine_errors_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachineErrorsRequest") + + _request = build_export_machine_errors_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachineErrorsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachineErrorsRequest type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_machine_errors_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list_health_summary( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> _models.SiteHealthSummaryCollection: + """Method to get site health summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: SiteHealthSummaryCollection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteHealthSummaryCollection + :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.SiteHealthSummaryCollection] = kwargs.pop("cls", None) + + _request = build_list_health_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("SiteHealthSummaryCollection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def summary(self, resource_group_name: str, site_name: str, **kwargs: Any) -> _models.HypervSiteUsage: + """Method to get site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: HypervSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervSiteUsage + :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.HypervSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("HypervSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_operations.py new file mode 100644 index 000000000000..cbeb8a258193 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_sites_operations.py @@ -0,0 +1,271 @@ +# 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_by_subscription_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.OffAzure/hypervSites") + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", 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_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "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) + + +class HypervSitesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_sites` 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_by_subscription(self, **kwargs: Any) -> Iterable["_models.HypervSite"]: + """Get all hyperv sites. + + Get all the hyperv sites in the subscription. + + :return: An iterator like instance of either HypervSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervSite] + :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.HypervSiteListResult] = 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_subscription_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("HypervSiteListResult", 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 list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.HypervSite"]: + """Get all hyperv sites. + + Get all the hyperv sites in the subscription. + + :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 HypervSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervSite] + :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.HypervSiteListResult] = 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("HypervSiteListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_software_inventories_controller_operations.py new file mode 100644 index 000000000000..f583db7a423c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_hyperv_software_inventories_controller_operations.py @@ -0,0 +1,290 @@ +# 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, Union +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_get_machine_software_inventory_request( # pylint: disable=name-too-long + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/machines/{machineName}/softwareInventories/{default}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "default": _SERIALIZER.url("default", default, "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_hyperv_machine_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/hypervSites/{siteName}/machines/{machineName}/softwareinventories", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class HypervSoftwareInventoriesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`hyperv_software_inventories_controller` 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_machine_software_inventory( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + **kwargs: Any + ) -> _models.HypervVmSoftwareInventory: + """Method to get a machines software inventory like applications and roles. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :param default: Default value. "default" Required. + :type default: str or ~azure.mgmt.migrate.models.Default + :return: HypervVmSoftwareInventory or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.HypervVmSoftwareInventory + :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.HypervVmSoftwareInventory] = kwargs.pop("cls", None) + + _request = build_get_machine_software_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_name, + default=default, + 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("HypervVmSoftwareInventory", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_hyperv_machine( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> Iterable["_models.HypervVmSoftwareInventory"]: + """List HypervVmSoftwareInventory resources by HypervMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: HypervMachine name. Required. + :type machine_name: str + :return: An iterator like instance of either HypervVmSoftwareInventory or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.HypervVmSoftwareInventory] + :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.HypervVmSoftwareInventoryListResult] = 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_hyperv_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("HypervVmSoftwareInventoryListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_applications_controller_operations.py new file mode 100644 index 000000000000..0f94bb9eda76 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_applications_controller_operations.py @@ -0,0 +1,529 @@ +# 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_by_web_app_site_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/iisWebApplications", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + web_app_site_name: str, + web_application_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/iisWebApplications/{webApplicationName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "webApplicationName": _SERIALIZER.url( + "web_application_name", web_application_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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_update_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/iisWebApplications/{webApplicationName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "webApplicationName": _SERIALIZER.url( + "web_application_name", web_application_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class IisWebApplicationsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`iis_web_applications_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.IisWebApplications"]: + """Method to get all IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either IisWebApplications or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.IisWebApplications] + :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.IisWebApplicationsListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("IisWebApplicationsListResult", 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, site_name: str, web_app_site_name: str, web_application_name: str, **kwargs: Any + ) -> _models.IisWebApplications: + """Method to get an IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :return: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :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.IisWebApplications] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_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("IisWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: _models.IisWebApplicationsUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.IisWebApplications: + """Method to update tags on IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.IisWebApplicationsUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.IisWebApplications: + """Method to update tags on IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. 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: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: Union[_models.IisWebApplicationsUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.IisWebApplications: + """Method to update tags on IIS web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. Is either a IisWebApplicationsUpdate type + or a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.IisWebApplicationsUpdate or IO[bytes] + :return: IisWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebApplications + :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.IisWebApplications] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "IisWebApplicationsUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("IisWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_servers_controller_operations.py new file mode 100644 index 000000000000..f4427c26ea44 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_iis_web_servers_controller_operations.py @@ -0,0 +1,328 @@ +# 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_by_web_app_site_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/iisWebServers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + web_app_site_name: str, + web_server_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/iisWebServers/{webServerName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "webServerName": _SERIALIZER.url("web_server_name", web_server_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class IisWebServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`iis_web_servers_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.IisWebServers"]: + """Method to get all IIS web servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either IisWebServers or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.IisWebServers] + :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.IisWebServersListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("IisWebServersListResult", 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, site_name: str, web_app_site_name: str, web_server_name: str, **kwargs: Any + ) -> _models.IisWebServers: + """Method to get an IIS web server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_server_name: Web server name. Required. + :type web_server_name: str + :return: IisWebServers or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.IisWebServers + :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.IisWebServers] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_server_name=web_server_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("IisWebServers", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_jobs_controller_operations.py new file mode 100644 index 000000000000..a679b3169793 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_jobs_controller_operations.py @@ -0,0 +1,907 @@ +# 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_deletejobs_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/deleteJobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_deletejob_request( + resource_group_name: str, site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/deleteJobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_exportjobs_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/exportJobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_exportjob_request( + resource_group_name: str, site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/exportJobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_importjobs_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/importJobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_importjob_request( + resource_group_name: str, site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/importJobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_import_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/jobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/jobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class ImportJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`import_jobs_controller` 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_deletejobs( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.DeleteImportMachinesJob"]: + """Method to get all delete import machines job for the given site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either DeleteImportMachinesJob or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.DeleteImportMachinesJob] + :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.PagedDeleteImportMachinesJob] = 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_deletejobs_request( + resource_group_name=resource_group_name, + site_name=site_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("PagedDeleteImportMachinesJob", 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_deletejob( + self, resource_group_name: str, site_name: str, job_name: str, **kwargs: Any + ) -> _models.DeleteImportMachinesJob: + """Gets the delete imported machines job with the given job name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job Arm Name. Required. + :type job_name: str + :return: DeleteImportMachinesJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.DeleteImportMachinesJob + :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.DeleteImportMachinesJob] = kwargs.pop("cls", None) + + _request = build_get_deletejob_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("DeleteImportMachinesJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_exportjobs( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.ExportImportedMachinesJob"]: + """Method to get all export import machines job for the given site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ExportImportedMachinesJob or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ExportImportedMachinesJob] + :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.PagedExportImportedMachinesJob] = 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_exportjobs_request( + resource_group_name=resource_group_name, + site_name=site_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("PagedExportImportedMachinesJob", 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_exportjob( + self, resource_group_name: str, site_name: str, job_name: str, **kwargs: Any + ) -> _models.ExportImportedMachinesJob: + """Gets the export imported machines job with the given job name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job Arm Name. Required. + :type job_name: str + :return: ExportImportedMachinesJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ExportImportedMachinesJob + :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.ExportImportedMachinesJob] = kwargs.pop("cls", None) + + _request = build_get_exportjob_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ExportImportedMachinesJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_importjobs( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.ImportMachinesJob"]: + """Method to get all import machines job for the given site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ImportMachinesJob or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ImportMachinesJob] + :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.PagedImportMachinesJob] = 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_importjobs_request( + resource_group_name=resource_group_name, + site_name=site_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("PagedImportMachinesJob", 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_importjob( + self, resource_group_name: str, site_name: str, job_name: str, **kwargs: Any + ) -> _models.ImportMachinesJob: + """Gets the import job with the given job name. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Job Arm Name. Required. + :type job_name: str + :return: ImportMachinesJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportMachinesJob + :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.ImportMachinesJob] = kwargs.pop("cls", None) + + _request = build_get_importjob_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ImportMachinesJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_import_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.ImportJob"]: + """List ImportJob resources by ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ImportJob or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ImportJob] + :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.ImportJobListResult] = 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_import_site_request( + resource_group_name=resource_group_name, + site_name=site_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("ImportJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.ImportJob: + """Get a ImportJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Jobs name. Required. + :type job_name: str + :return: ImportJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportJob + :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.ImportJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ImportJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_machines_controller_operations.py new file mode 100644 index 000000000000..8aee67f3452c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_machines_controller_operations.py @@ -0,0 +1,398 @@ +# 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_by_import_site_request( + resource_group_name: str, + site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/machines", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 ImportMachinesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`import_machines_controller` 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_by_import_site( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.ImportMachine"]: + """List ImportMachine resources by ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either ImportMachine or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ImportMachine] + :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.ImportMachineListResult] = 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_import_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("ImportMachineListResult", 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, site_name: str, machine_name: str, **kwargs: Any) -> _models.ImportMachine: + """Get a ImportMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: ImportMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportMachine + :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.ImportMachine] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("ImportMachine", 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, site_name: str, machine_name: str, **kwargs: Any + ) -> None: + """Delete a ImportMachine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_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, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_sites_controller_operations.py new file mode 100644 index 000000000000..60f419b0947d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_import_sites_controller_operations.py @@ -0,0 +1,1194 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_subscription_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.OffAzure/importSites") + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", 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_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/importSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/importSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/importSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_delete_imported_machines_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/importSites/{siteName}/deleteImportedMachines", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_uri_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/importSites/{siteName}/exportUri", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_import_uri_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/importSites/{siteName}/importUri", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 ImportSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`import_sites_controller` 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_by_subscription(self, **kwargs: Any) -> Iterable["_models.ImportSite"]: + """List ImportSite resources by subscription ID. + + :return: An iterator like instance of either ImportSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ImportSite] + :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.ImportSiteListResult] = 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_subscription_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("ImportSiteListResult", 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 list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.ImportSite"]: + """Get all import sites. + + :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 ImportSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ImportSite] + :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.ImportSiteListResult] = 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("ImportSiteListResult", 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, site_name: str, **kwargs: Any) -> _models.ImportSite: + """Get a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :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.ImportSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("ImportSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + body: _models.ImportSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Create a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.ImportSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Create a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create( + self, resource_group_name: str, site_name: str, body: Union[_models.ImportSite, IO[bytes]], **kwargs: Any + ) -> _models.ImportSite: + """Create a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a ImportSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.ImportSite or IO[bytes] + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :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.ImportSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ImportSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("ImportSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: _models.ImportSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Update a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.ImportSiteUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ImportSite: + """Update a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, resource_group_name: str, site_name: str, body: Union[_models.ImportSiteUpdate, IO[bytes]], **kwargs: Any + ) -> _models.ImportSite: + """Update a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a ImportSiteUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ImportSiteUpdate or IO[bytes] + :return: ImportSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ImportSite + :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.ImportSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ImportSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("ImportSite", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a ImportSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def delete_imported_machines( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SasUriResponse: + """Deletes the imported machines for site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SasUriResponse] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_delete_imported_machines_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SasUriResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def export_uri( + self, + resource_group_name: str, + site_name: str, + body: _models.SasUriResponse, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to export a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def export_uri( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to export a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def export_uri( + self, resource_group_name: str, site_name: str, body: Union[_models.SasUriResponse, IO[bytes]], **kwargs: Any + ) -> _models.SasUriResponse: + """Method to export a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a SasUriResponse type or a IO[bytes] + type. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse or IO[bytes] + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :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.SasUriResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SasUriResponse") + + _request = build_export_uri_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SasUriResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def import_uri( + self, + resource_group_name: str, + site_name: str, + body: _models.SasUriResponse, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to import a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def import_uri( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SasUriResponse: + """Method to import a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def import_uri( + self, resource_group_name: str, site_name: str, body: Union[_models.SasUriResponse, IO[bytes]], **kwargs: Any + ) -> _models.SasUriResponse: + """Method to import a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a SasUriResponse type or a IO[bytes] + type. Required. + :type body: ~azure.mgmt.migrate.models.SasUriResponse or IO[bytes] + :return: SasUriResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SasUriResponse + :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.SasUriResponse] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SasUriResponse") + + _request = build_import_uri_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SasUriResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_machines_controller_operations.py new file mode 100644 index 000000000000..2ad1889e7897 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_machines_controller_operations.py @@ -0,0 +1,797 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_vmware_site_request( + resource_group_name: str, + site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[int] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/machines", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "int") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_start_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/machines/{machineName}/start", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_stop_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/machines/{machineName}/stop", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 MachinesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`machines_controller` 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_by_vmware_site( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[int] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.MachineResource"]: + """List MachineResource resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: int + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either MachineResource or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.MachineResource] + :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.MachineResourceListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("MachineResourceListResult", 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, site_name: str, machine_name: str, **kwargs: Any + ) -> _models.MachineResource: + """Get a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :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.MachineResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("MachineResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: _models.MachineResourceUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MachineResource: + """Update a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.MachineResourceUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.MachineResource: + """Update a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. 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: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: Union[_models.MachineResourceUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.MachineResource: + """Update a MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Is either a MachineResourceUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.MachineResourceUpdate or IO[bytes] + :return: MachineResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MachineResource + :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.MachineResource] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "MachineResourceUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("MachineResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _start_initial( + self, resource_group_name: str, site_name: str, machine_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_start_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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]: + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_start( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> LROPoller[JSON]: + """Method to start a machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._start_initial( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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): + deserialized = self._deserialize("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _stop_initial( + self, resource_group_name: str, site_name: str, machine_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_stop_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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]: + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_stop(self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any) -> LROPoller[JSON]: + """Method to stop a machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._stop_initial( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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): + deserialized = self._deserialize("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_controller_operations.py new file mode 100644 index 000000000000..51260ea2effe --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_controller_operations.py @@ -0,0 +1,1070 @@ +# 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_by_subscription_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.OffAzure/masterSites") + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", 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_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_error_summary_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/errorSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 MasterSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`master_sites_controller` 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_by_subscription(self, **kwargs: Any) -> Iterable["_models.MasterSite"]: + """Get all the sites in the subscription. + + List MasterSite resources by subscription ID. + + :return: An iterator like instance of either MasterSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSiteListResult] = 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_subscription_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("MasterSiteListResult", 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 list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.MasterSite"]: + """Get all sites. + + Get all the sites in the 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 MasterSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSiteListResult] = 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("MasterSiteListResult", 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, site_name: str, **kwargs: Any) -> _models.MasterSite: + """Method to get a master site. + + Get a MasterSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: MasterSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.MasterSite + :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.MasterSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("MasterSite", 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, site_name: str, body: Union[_models.MasterSite, IO[bytes]], **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: + _json = self._serialize.body(body, "MasterSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + body: _models.MasterSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.MasterSite]: + """Method to create or update a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.MasterSite + :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 MasterSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.MasterSite]: + """Method to create or update a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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 either MasterSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, resource_group_name: str, site_name: str, body: Union[_models.MasterSite, IO[bytes]], **kwargs: Any + ) -> LROPoller[_models.MasterSite]: + """Method to create or update a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a MasterSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.MasterSite or IO[bytes] + :return: An instance of LROPoller that returns either MasterSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSite] = 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, + site_name=site_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("MasterSite", 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.MasterSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.MasterSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _update_initial( + self, + resource_group_name: str, + site_name: str, + properties: Union[_models.MasterSiteUpdate, IO[bytes]], + **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(properties, (IOBase, bytes)): + _content = properties + else: + _json = self._serialize.body(properties, "MasterSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + properties: _models.MasterSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.MasterSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: ~azure.mgmt.migrate.models.MasterSiteUpdate + :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 MasterSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + site_name: str, + properties: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.MasterSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: 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 MasterSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.MasterSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update( + self, + resource_group_name: str, + site_name: str, + properties: Union[_models.MasterSiteUpdate, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.MasterSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param properties: The resource properties to be updated. Is either a MasterSiteUpdate type or + a IO[bytes] type. Required. + :type properties: ~azure.mgmt.migrate.models.MasterSiteUpdate or IO[bytes] + :return: An instance of LROPoller that returns either MasterSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.MasterSite] + :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.MasterSite] = 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, + site_name=site_name, + properties=properties, + 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("MasterSite", 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.MasterSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.MasterSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> None: + """Method to delete a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @overload + def error_summary( + self, + resource_group_name: str, + site_name: str, + body: _models.ErrorSummaryRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from master site for an appliance. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def error_summary( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from master site for an appliance. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def error_summary( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ErrorSummaryRequest, IO[bytes]], + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from master site for an appliance. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ErrorSummaryRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest or IO[bytes] + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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.SiteErrorSummary] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ErrorSummaryRequest") + + _request = build_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_operations_status_controller_operations.py new file mode 100644 index 000000000000..421c4e9a59b4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_master_sites_operations_status_controller_operations.py @@ -0,0 +1,154 @@ +# 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_vmware_operation_status_request( # pylint: disable=name-too-long + resource_group_name: str, site_name: str, operation_status_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/operationsStatus/{operationStatusName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "operationStatusName": _SERIALIZER.url( + "operation_status_name", operation_status_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class MasterSitesOperationsStatusControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`master_sites_operations_status_controller` 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_vmware_operation_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """A operation status resource belonging to a master site resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_vmware_operation_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_operations.py new file mode 100644 index 000000000000..f1d1f7a960a6 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_operations.py @@ -0,0 +1,152 @@ +# 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(**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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop("template_url", "/providers/Microsoft.OffAzure/operations") + + # 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) + + +class Operations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`operations` 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.Operation"]: + """List the operations for the provider. + + :return: An iterator like instance of either Operation or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.Operation] + :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.OperationListResult] = 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( + 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("OperationListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_patch.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_endpoint_connection_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_endpoint_connection_controller_operations.py new file mode 100644 index 000000000000..26ee4e295b7b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_endpoint_connection_controller_operations.py @@ -0,0 +1,552 @@ +# 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_by_master_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/privateEndpointConnections", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, pe_connection_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/privateEndpointConnections/{peConnectionName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "peConnectionName": _SERIALIZER.url( + "pe_connection_name", pe_connection_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, site_name: str, pe_connection_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/privateEndpointConnections/{peConnectionName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "peConnectionName": _SERIALIZER.url( + "pe_connection_name", pe_connection_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, site_name: str, pe_connection_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/privateEndpointConnections/{peConnectionName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "peConnectionName": _SERIALIZER.url( + "pe_connection_name", pe_connection_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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 PrivateEndpointConnectionControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`private_endpoint_connection_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.PrivateEndpointConnection"]: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either PrivateEndpointConnection or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.PrivateEndpointConnection] + :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.PrivateEndpointConnectionListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("PrivateEndpointConnectionListResult", 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, site_name: str, pe_connection_name: str, **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :return: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :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.PrivateEndpointConnection] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + pe_connection_name=pe_connection_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("PrivateEndpointConnection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + pe_connection_name: str, + body: _models.PrivateEndpointConnection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + pe_connection_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :param body: Resource create 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: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create( + self, + resource_group_name: str, + site_name: str, + pe_connection_name: str, + body: Union[_models.PrivateEndpointConnection, IO[bytes]], + **kwargs: Any + ) -> _models.PrivateEndpointConnection: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_name: str + :param body: Resource create parameters. Is either a PrivateEndpointConnection type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.PrivateEndpointConnection or IO[bytes] + :return: PrivateEndpointConnection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateEndpointConnection + :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.PrivateEndpointConnection] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "PrivateEndpointConnection") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + pe_connection_name=pe_connection_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, pe_connection_name: str, **kwargs: Any + ) -> None: + """Deletes the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param pe_connection_name: Private link resource name. Required. + :type pe_connection_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, + site_name=site_name, + pe_connection_name=pe_connection_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_link_resources_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_link_resources_controller_operations.py new file mode 100644 index 000000000000..b940ab08aeaa --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_private_link_resources_controller_operations.py @@ -0,0 +1,273 @@ +# 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_by_master_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/privateLinkResources", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, private_link_resource_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/privateLinkResources/{privateLinkResourceName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "privateLinkResourceName": _SERIALIZER.url( + "private_link_resource_name", private_link_resource_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class PrivateLinkResourcesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`private_link_resources_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.PrivateLinkResource"]: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either PrivateLinkResource or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.PrivateLinkResource] + :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.PrivateLinkResourceListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("PrivateLinkResourceListResult", 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, site_name: str, private_link_resource_name: str, **kwargs: Any + ) -> _models.PrivateLinkResource: + """Gets the private link resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param private_link_resource_name: Private link resource name. Required. + :type private_link_resource_name: str + :return: PrivateLinkResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.PrivateLinkResource + :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.PrivateLinkResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + private_link_resource_name=private_link_resource_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("PrivateLinkResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..ce6743e69602 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_run_as_accounts_controller_operations.py @@ -0,0 +1,272 @@ +# 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_by_vmware_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/runAsAccounts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, account_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/runAsAccounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "accountName": _SERIALIZER.url("account_name", account_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class RunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`run_as_accounts_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.VmwareRunAsAccountResource"]: + """List VmwareRunAsAccountResource resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either VmwareRunAsAccountResource or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.VmwareRunAsAccountResource] + :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.VmwareRunAsAccountResourceListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareRunAsAccountResourceListResult", 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, site_name: str, account_name: str, **kwargs: Any + ) -> _models.VmwareRunAsAccountResource: + """Get a VmwareRunAsAccountResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param account_name: RunAsAccounts name. Required. + :type account_name: str + :return: VmwareRunAsAccountResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareRunAsAccountResource + :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.VmwareRunAsAccountResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + account_name=account_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("VmwareRunAsAccountResource", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_dependency_map_controller_operations.py new file mode 100644 index 000000000000..e72c47e84083 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_dependency_map_controller_operations.py @@ -0,0 +1,1234 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_client_group_members_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/clientGroupMembers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_export_dependencies_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/exportDependencies", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_generate_coarse_map_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/generateCoarseMap", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_generate_detailed_map_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/generateDetailedMap", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_server_group_members_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/serverGroupMembers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 ServerDependencyMapControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`server_dependency_map_controller` 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 _client_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsClientGroupMembersRequest") + + _request = build_client_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsClientGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_client_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsClientGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list client group members for the selected client group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsClientGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsClientGroupMembersRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._client_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_dependencies_initial( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **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(request_body, (IOBase, bytes)): + _content = request_body + else: + _json = self._serialize.body(request_body, "DependencyMapServiceMapextensionsExportDependenciesRequest") + + _request = build_export_dependencies_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: _models.DependencyMapServiceMapextensionsExportDependenciesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Required. + :type request_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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_dependencies( + self, + resource_group_name: str, + site_name: str, + request_body: Union[_models.DependencyMapServiceMapextensionsExportDependenciesRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate report containing agentless dependencies. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param request_body: The content of the action request. Is either a + DependencyMapServiceMapextensionsExportDependenciesRequest type or a IO[bytes] type. Required. + :type request_body: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsExportDependenciesRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_dependencies_initial( + resource_group_name=resource_group_name, + site_name=site_name, + request_body=request_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _generate_coarse_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsScopeMapRequest") + + _request = build_generate_coarse_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsScopeMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_generate_coarse_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsScopeMapRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate coarse map for the list of machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsScopeMapRequest type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsScopeMapRequest + or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._generate_coarse_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _generate_detailed_map_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body( + map_request, "DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest" + ) + + _request = build_generate_detailed_map_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_generate_detailed_map( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to generate detailed map for a selected machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest type or a IO[bytes] type. + Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsSingleMachineDetailedMapRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._generate_detailed_map_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _server_group_members_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "DependencyMapServiceMapextensionsServerGroupMembersRequest") + + _request = build_server_group_members_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: _models.DependencyMapServiceMapextensionsServerGroupMembersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_server_group_members( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.DependencyMapServiceMapextensionsServerGroupMembersRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """API to list server group members for the selected server group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a + DependencyMapServiceMapextensionsServerGroupMembersRequest type or a IO[bytes] type. Required. + :type map_request: + ~azure.mgmt.migrate.models.DependencyMapServiceMapextensionsServerGroupMembersRequest or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._server_group_members_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_jobs_controller_operations.py new file mode 100644 index 000000000000..8116700b565e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_jobs_controller_operations.py @@ -0,0 +1,269 @@ +# 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_by_server_site_resource_request( # pylint: disable=name-too-long + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/jobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/jobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class ServerJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`server_jobs_controller` 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_by_server_site_resource( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.ServerJob"]: + """List ServerJob resources by ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ServerJob or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ServerJob] + :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.ServerJobListResult] = 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_server_site_resource_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerJobListResult", 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, site_name: str, job_name: str, **kwargs: Any) -> _models.ServerJob: + """Get a ServerJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param job_name: Jobs name. Required. + :type job_name: str + :return: ServerJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerJob + :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.ServerJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + job_name=job_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("ServerJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_operations_status_controller_operations.py new file mode 100644 index 000000000000..2d7ca4f5d2c2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_operations_status_controller_operations.py @@ -0,0 +1,154 @@ +# 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_server_site_operations_status_request( # pylint: disable=name-too-long + resource_group_name: str, site_name: str, operation_status_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/operationsStatus/{operationStatusName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "operationStatusName": _SERIALIZER.url( + "operation_status_name", operation_status_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class ServerOperationsStatusControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`server_operations_status_controller` 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_server_site_operations_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """A operation status resource belonging to a site resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_server_site_operations_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..16adbc4a826c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_run_as_accounts_controller_operations.py @@ -0,0 +1,271 @@ +# 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_by_server_site_resource_request( # pylint: disable=name-too-long + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/runAsAccounts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, account_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/runAsAccounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "accountName": _SERIALIZER.url("account_name", account_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class ServerRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`server_run_as_accounts_controller` 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_by_server_site_resource( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.ServerRunAsAccount"]: + """List ServerRunAsAccount resources by ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either ServerRunAsAccount or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ServerRunAsAccount] + :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.ServerRunAsAccountListResult] = 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_server_site_resource_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerRunAsAccountListResult", 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, site_name: str, account_name: str, **kwargs: Any + ) -> _models.ServerRunAsAccount: + """Get a ServerRunAsAccount. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param account_name: RunAsAccounts name. Required. + :type account_name: str + :return: ServerRunAsAccount or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerRunAsAccount + :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.ServerRunAsAccount] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + account_name=account_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("ServerRunAsAccount", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_sites_controller_operations.py new file mode 100644 index 000000000000..7351c545712a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_sites_controller_operations.py @@ -0,0 +1,2228 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_subscription_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.OffAzure/serverSites") + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", 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_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_compute_error_summary_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/serverSites/{siteName}/computeErrorSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_computeusage_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/serverSites/{siteName}/computeusage", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_applications_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/serverSites/{siteName}/exportApplications", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_machine_errors_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/exportMachineErrors", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_list_health_summary_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/listHealthSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_refresh_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/refreshSite", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_summary_request(resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/summary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_dependency_map_status_request( # pylint: disable=name-too-long + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/updateDependencyMapStatus", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_properties_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/updateProperties", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 ServerSitesControllerOperations: # pylint: disable=too-many-public-methods + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`server_sites_controller` 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_by_subscription(self, **kwargs: Any) -> Iterable["_models.ServerSiteResource"]: + """List ServerSiteResource resources by subscription ID. + + :return: An iterator like instance of either ServerSiteResource or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ServerSiteResource] + :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.ServerSiteResourceListResult] = 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_subscription_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("ServerSiteResourceListResult", 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 list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.ServerSiteResource"]: + """List ServerSiteResource resources by 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 ServerSiteResource or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ServerSiteResource] + :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.ServerSiteResourceListResult] = 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("ServerSiteResourceListResult", 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, site_name: str, **kwargs: Any) -> _models.ServerSiteResource: + """Get a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :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.ServerSiteResource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerSiteResource", 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, + site_name: str, + body: Union[_models.ServerSiteResource, IO[bytes]], + **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: + _json = self._serialize.body(body, "ServerSiteResource") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + body: _models.ServerSiteResource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ServerSiteResource]: + """Create a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResource + :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 ServerSiteResource or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.ServerSiteResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.ServerSiteResource]: + """Create a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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 either ServerSiteResource or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.ServerSiteResource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ServerSiteResource, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.ServerSiteResource]: + """Create a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a ServerSiteResource type or a IO[bytes] + type. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResource or IO[bytes] + :return: An instance of LROPoller that returns either ServerSiteResource or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.ServerSiteResource] + :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.ServerSiteResource] = 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, + site_name=site_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("ServerSiteResource", 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.ServerSiteResource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.ServerSiteResource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: _models.ServerSiteResourceUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ServerSiteResource: + """Update a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResourceUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.ServerSiteResource: + """Update a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ServerSiteResourceUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.ServerSiteResource: + """Update a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a ServerSiteResourceUpdate type + or a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ServerSiteResourceUpdate or IO[bytes] + :return: ServerSiteResource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteResource + :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.ServerSiteResource] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ServerSiteResourceUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("ServerSiteResource", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a ServerSiteResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def compute_error_summary( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SiteErrorSummary: + """Get the error summary for a server site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SiteErrorSummary] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_compute_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def computeusage( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.ServerSiteUsageResponse: + """Get a serve site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: ServerSiteUsageResponse or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteUsageResponse + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.ServerSiteUsageResponse] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_computeusage_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("ServerSiteUsageResponse", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _export_applications_initial( + self, resource_group_name: str, site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_applications_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_applications( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_applications_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_machine_errors_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachineErrorsRequest") + + _request = build_export_machine_errors_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachineErrorsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachineErrorsRequest type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_machine_errors_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list_health_summary( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> _models.SiteHealthSummaryCollection: + """Method to get site health summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: SiteHealthSummaryCollection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteHealthSummaryCollection + :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.SiteHealthSummaryCollection] = kwargs.pop("cls", None) + + _request = build_list_health_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("SiteHealthSummaryCollection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _refresh_site_initial(self, resource_group_name: str, site_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_refresh_site_request( + resource_group_name=resource_group_name, + site_name=site_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]: + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_refresh_site(self, resource_group_name: str, site_name: str, **kwargs: Any) -> LROPoller[JSON]: + """Operation to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._refresh_site_initial( + resource_group_name=resource_group_name, + site_name=site_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): + deserialized = self._deserialize("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def summary(self, resource_group_name: str, site_name: str, **kwargs: Any) -> _models.ServerSiteUsage: + """Method to get site usage. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: ServerSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSiteUsage + :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.ServerSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("ServerSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _update_dependency_map_status_initial( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **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(map_request, (IOBase, bytes)): + _content = map_request + else: + _json = self._serialize.body(map_request, "UpdateMachineDepMapStatus") + + _request = build_update_dependency_map_status_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: _models.UpdateMachineDepMapStatus, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Required. + :type map_request: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + map_request: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param map_request: The content of the action request. Is either a UpdateMachineDepMapStatus + type or a IO[bytes] type. Required. + :type map_request: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_dependency_map_status_initial( + resource_group_name=resource_group_name, + site_name=site_name, + map_request=map_request, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _update_properties_initial( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **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(meta_data, (IOBase, bytes)): + _content = meta_data + else: + _json = self._serialize.body(meta_data, "MachineMetadataCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: _models.MachineMetadataCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Operation to update custom properties for servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Operation to update custom properties for servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Operation to update custom properties for servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Is either a MachineMetadataCollection type + or a IO[bytes] type. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + meta_data=meta_data, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_software_inventories_controller_operations.py new file mode 100644 index 000000000000..e477990249aa --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_server_software_inventories_controller_operations.py @@ -0,0 +1,290 @@ +# 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, Union +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_get_machine_software_inventory_request( # pylint: disable=name-too-long + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/machines/{machineName}/softwareInventories/{default}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "default": _SERIALIZER.url("default", default, "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_server_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/machines/{machineName}/softwareinventories", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class ServerSoftwareInventoriesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`server_software_inventories_controller` 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_machine_software_inventory( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + **kwargs: Any + ) -> _models.ServerSoftwareInventory: + """Method to get a machines software inventory like applications and roles. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param default: Default value. "default" Required. + :type default: str or ~azure.mgmt.migrate.models.Default + :return: ServerSoftwareInventory or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.ServerSoftwareInventory + :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.ServerSoftwareInventory] = kwargs.pop("cls", None) + + _request = build_get_machine_software_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_name, + default=default, + 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("ServerSoftwareInventory", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_server( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> Iterable["_models.ServerSoftwareInventory"]: + """List ServerSoftwareInventory resources by Server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :return: An iterator like instance of either ServerSoftwareInventory or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.ServerSoftwareInventory] + :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.ServerSoftwareInventoryListResult] = 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_server_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("ServerSoftwareInventoryListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_servers_controller_operations.py new file mode 100644 index 000000000000..498b68c8cf53 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_servers_controller_operations.py @@ -0,0 +1,577 @@ +# 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_by_server_site_resource_request( # pylint: disable=name-too-long + resource_group_name: str, + site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/machines", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_get_machine_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_machine_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-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.OffAzure/serverSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_machine_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/serverSites/{siteName}/machines/{machineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 ServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`servers_controller` 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_by_server_site_resource( + self, + resource_group_name: str, + site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.Server"]: + """Get all machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either Server or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.Server] + :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.ServerListResult] = 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_server_site_resource_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("ServerListResult", 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_machine(self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any) -> _models.Server: + """Get a Server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :return: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :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.Server] = kwargs.pop("cls", None) + + _request = build_get_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("Server", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def update_machine( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: _models.ServerUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Server: + """Update a Server machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.ServerUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update_machine( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Server: + """Update a Server machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. 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: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update_machine( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + body: Union[_models.ServerUpdate, IO[bytes]], + **kwargs: Any + ) -> _models.Server: + """Update a Server machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_name: str + :param body: The resource properties to be updated. Is either a ServerUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ServerUpdate or IO[bytes] + :return: Server or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Server + :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.Server] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ServerUpdate") + + _request = build_update_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("Server", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete_machine( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> None: + """Delete a Server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: A server machine name. Required. + :type machine_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_machine_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sites_controller_operations.py new file mode 100644 index 000000000000..8aa0d3a45118 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sites_controller_operations.py @@ -0,0 +1,1785 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_subscription_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.OffAzure/vmwareSites") + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", 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_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_compute_error_summary_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/computeErrorSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_computeusage_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/computeusage", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_applications_request( + resource_group_name: str, site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/exportApplications", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_machine_errors_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/exportMachineErrors", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_export_machines_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/exportMachines", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_list_health_summary_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/listHealthSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_summary_request(resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/summary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 SitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sites_controller` 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_by_subscription(self, **kwargs: Any) -> Iterable["_models.VmwareSite"]: + """Get all vmware sites. + + Get all the vmware sites in the subscription. + + :return: An iterator like instance of either VmwareSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.VmwareSite] + :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.VmwareSiteListResult] = 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_subscription_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("VmwareSiteListResult", 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 list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VmwareSite"]: + """Get all vmware sites. + + Get all the vmware sites in the 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 VmwareSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.VmwareSite] + :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.VmwareSiteListResult] = 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("VmwareSiteListResult", 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, site_name: str, **kwargs: Any) -> _models.VmwareSite: + """Get a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :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.VmwareSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + body: _models.VmwareSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Create a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.VmwareSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Create a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create 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: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create( + self, resource_group_name: str, site_name: str, body: Union[_models.VmwareSite, IO[bytes]], **kwargs: Any + ) -> _models.VmwareSite: + """Create a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: Resource create parameters. Is either a VmwareSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.VmwareSite or IO[bytes] + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :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.VmwareSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "VmwareSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("VmwareSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: _models.VmwareSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Update a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.VmwareSiteUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VmwareSite: + """Update a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. 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: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, resource_group_name: str, site_name: str, body: Union[_models.VmwareSiteUpdate, IO[bytes]], **kwargs: Any + ) -> _models.VmwareSite: + """Update a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The resource properties to be updated. Is either a VmwareSiteUpdate type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.VmwareSiteUpdate or IO[bytes] + :return: VmwareSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSite + :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.VmwareSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "VmwareSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("VmwareSite", 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, site_name: str, **kwargs: Any + ) -> None: + """Delete a VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_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, + site_name=site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def compute_error_summary( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get site error summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.SiteErrorSummary] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_compute_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def computeusage( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> _models.VmwareSiteUsage: + """Method to get site error summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: VmwareSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSiteUsage + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.VmwareSiteUsage] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_computeusage_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("VmwareSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _export_applications_initial( + self, resource_group_name: str, site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_applications_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_applications( + self, resource_group_name: str, site_name: str, body: JSON, **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_applications_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_machine_errors_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachineErrorsRequest") + + _request = build_export_machine_errors_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachineErrorsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_machine_errors( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachineErrorsRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the errors encountered during guest discovery of the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachineErrorsRequest type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachineErrorsRequest or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_machine_errors_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_machines_initial( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachinesRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportMachinesRequest") + + _request = build_export_machines_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_machines( + self, + resource_group_name: str, + site_name: str, + body: _models.ExportMachinesRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachinesRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_machines( + self, + resource_group_name: str, + site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_machines( + self, + resource_group_name: str, + site_name: str, + body: Union[_models.ExportMachinesRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing + machine and the deep discovery of the application installed in the machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param body: The content of the action request. Is either a ExportMachinesRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportMachinesRequest or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_machines_initial( + resource_group_name=resource_group_name, + site_name=site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list_health_summary( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> _models.SiteHealthSummaryCollection: + """Method to get site health summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: SiteHealthSummaryCollection or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteHealthSummaryCollection + :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.SiteHealthSummaryCollection] = kwargs.pop("cls", None) + + _request = build_list_health_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("SiteHealthSummaryCollection", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def summary(self, resource_group_name: str, site_name: str, **kwargs: Any) -> _models.VmwareSiteUsage: + """Method to get site usage/summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: VmwareSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareSiteUsage + :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.VmwareSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_availability_groups_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_availability_groups_controller_operations.py new file mode 100644 index 000000000000..94faea34a6fc --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_availability_groups_controller_operations.py @@ -0,0 +1,332 @@ +# 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_by_sql_site_request( + resource_group_name: str, + site_name: str, + sql_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/sqlAvailabilityGroups", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + sql_site_name: str, + sql_availability_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/sqlAvailabilityGroups/{sqlAvailabilityGroupName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlAvailabilityGroupName": _SERIALIZER.url( + "sql_availability_group_name", sql_availability_group_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class SqlAvailabilityGroupsControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_availability_groups_controller` 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_by_sql_site( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.SqlAvailabilityGroup"]: + """Gets the sql availability groups. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either SqlAvailabilityGroup or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.SqlAvailabilityGroup] + :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.SqlAvailabilityGroupListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("SqlAvailabilityGroupListResult", 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, + site_name: str, + sql_site_name: str, + sql_availability_group_name: str, + **kwargs: Any + ) -> _models.SqlAvailabilityGroup: + """Gets the sql availability group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_availability_group_name: SQL availability group name. Required. + :type sql_availability_group_name: str + :return: SqlAvailabilityGroup or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlAvailabilityGroup + :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.SqlAvailabilityGroup] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_availability_group_name=sql_availability_group_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("SqlAvailabilityGroup", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_databases_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_databases_controller_operations.py new file mode 100644 index 000000000000..29e1d629b631 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_databases_controller_operations.py @@ -0,0 +1,326 @@ +# 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_by_sql_site_request( + resource_group_name: str, + site_name: str, + sql_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/sqlDatabases", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + sql_site_name: str, + sql_database_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/sqlDatabases/{sqlDatabaseName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlDatabaseName": _SERIALIZER.url( + "sql_database_name", sql_database_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class SqlDatabasesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_databases_controller` 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_by_sql_site( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.SqlDatabaseV2"]: + """Gets the sql Databases. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either SqlDatabaseV2 or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.SqlDatabaseV2] + :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.SqlDatabaseV2ListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("SqlDatabaseV2ListResult", 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, site_name: str, sql_site_name: str, sql_database_name: str, **kwargs: Any + ) -> _models.SqlDatabaseV2: + """Gets the sql Database. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_database_name: SQL Database name. Required. + :type sql_database_name: str + :return: SqlDatabaseV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlDatabaseV2 + :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.SqlDatabaseV2] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_database_name=sql_database_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("SqlDatabaseV2", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_discovery_site_data_source_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_discovery_site_data_source_controller_operations.py new file mode 100644 index 000000000000..3fc00d73bdf5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_discovery_site_data_source_controller_operations.py @@ -0,0 +1,672 @@ +# 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_by_sql_site_request( + resource_group_name: str, site_name: str, sql_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/discoverySiteDataSources", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, + site_name: str, + sql_site_name: str, + discovery_site_data_source_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/discoverySiteDataSources/{discoverySiteDataSourceName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "discoverySiteDataSourceName": _SERIALIZER.url( + "discovery_site_data_source_name", discovery_site_data_source_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, + site_name: str, + sql_site_name: str, + discovery_site_data_source_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/discoverySiteDataSources/{discoverySiteDataSourceName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "discoverySiteDataSourceName": _SERIALIZER.url( + "discovery_site_data_source_name", discovery_site_data_source_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, + site_name: str, + sql_site_name: str, + discovery_site_data_source_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/discoverySiteDataSources/{discoverySiteDataSourceName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "discoverySiteDataSourceName": _SERIALIZER.url( + "discovery_site_data_source_name", discovery_site_data_source_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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 SqlDiscoverySiteDataSourceControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_discovery_site_data_source_controller` 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_by_sql_site( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> Iterable["_models.SqlDiscoverySiteDataSource"]: + """List SqlDiscoverySiteDataSource resources by SqlSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: An iterator like instance of either SqlDiscoverySiteDataSource or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :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.SqlDiscoverySiteDataSourceListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlDiscoverySiteDataSourceListResult", 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, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> _models.SqlDiscoverySiteDataSource: + """Get a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :return: SqlDiscoverySiteDataSource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource + :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.SqlDiscoverySiteDataSource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("SqlDiscoverySiteDataSource", 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, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.SqlDiscoverySiteDataSource, IO[bytes]], + **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: + _json = self._serialize.body(body, "SqlDiscoverySiteDataSource") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: _models.SqlDiscoverySiteDataSource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.SqlDiscoverySiteDataSource]: + """Create a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource + :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 SqlDiscoverySiteDataSource or the result + of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.SqlDiscoverySiteDataSource]: + """Create a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create 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 either SqlDiscoverySiteDataSource or the result + of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.SqlDiscoverySiteDataSource, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.SqlDiscoverySiteDataSource]: + """Create a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Is either a SqlDiscoverySiteDataSource type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource or IO[bytes] + :return: An instance of LROPoller that returns either SqlDiscoverySiteDataSource or the result + of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.SqlDiscoverySiteDataSource] + :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.SqlDiscoverySiteDataSource] = 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, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("SqlDiscoverySiteDataSource", 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.SqlDiscoverySiteDataSource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.SqlDiscoverySiteDataSource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> None: + """Delete a SqlDiscoverySiteDataSource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param discovery_site_data_source_name: SQL Discovery site data source name. Required. + :type discovery_site_data_source_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, + site_name=site_name, + sql_site_name=sql_site_name, + discovery_site_data_source_name=discovery_site_data_source_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_jobs_controller_operations.py new file mode 100644 index 000000000000..a49e81dccc5e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_jobs_controller_operations.py @@ -0,0 +1,283 @@ +# 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_by_sql_site_request( + resource_group_name: str, site_name: str, sql_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/jobs", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, sql_site_name: str, job_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/jobs/{jobName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "jobName": _SERIALIZER.url("job_name", job_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class SqlJobsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_jobs_controller` 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_by_sql_site( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> Iterable["_models.SqlJob"]: + """Gets the sql Jobs. + + List SqlJob resources by SqlSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: An iterator like instance of either SqlJob or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.SqlJob] + :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.SqlJobListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlJobListResult", 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, site_name: str, sql_site_name: str, job_name: str, **kwargs: Any + ) -> _models.SqlJob: + """Gets the sql Job. + + Get a SqlJob. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param job_name: SQL Job name. Required. + :type job_name: str + :return: SqlJob or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlJob + :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.SqlJob] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + job_name=job_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("SqlJob", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_operations_status_controller_operations.py new file mode 100644 index 000000000000..c2879ebc0c5c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_operations_status_controller_operations.py @@ -0,0 +1,163 @@ +# 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_sql_operation_status_request( + resource_group_name: str, + site_name: str, + sql_site_name: str, + operation_status_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/operationsStatus/{operationStatusName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "operationStatusName": _SERIALIZER.url( + "operation_status_name", operation_status_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class SqlOperationsStatusControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_operations_status_controller` 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_sql_operation_status( + self, resource_group_name: str, site_name: str, sql_site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """Method to get operation status. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_sql_operation_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..8a15bcc5a855 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_run_as_accounts_controller_operations.py @@ -0,0 +1,279 @@ +# 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_by_sql_site_request( + resource_group_name: str, site_name: str, sql_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/runAsAccounts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, sql_site_name: str, account_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/runAsAccounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "accountName": _SERIALIZER.url("account_name", account_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class SqlRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_run_as_accounts_controller` 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_by_sql_site( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> Iterable["_models.SqlRunAsAccount"]: + """List SqlRunAsAccount resources by SqlSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: An iterator like instance of either SqlRunAsAccount or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.SqlRunAsAccount] + :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.SqlRunAsAccountListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlRunAsAccountListResult", 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, site_name: str, sql_site_name: str, account_name: str, **kwargs: Any + ) -> _models.SqlRunAsAccount: + """Get a SqlRunAsAccount. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param account_name: SQL RunAsAccounts name. Required. + :type account_name: str + :return: SqlRunAsAccount or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlRunAsAccount + :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.SqlRunAsAccount] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + account_name=account_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("SqlRunAsAccount", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_servers_controller_operations.py new file mode 100644 index 000000000000..68d97207000f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_servers_controller_operations.py @@ -0,0 +1,519 @@ +# 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_by_sql_site_request( + resource_group_name: str, + site_name: str, + sql_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/sqlServers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + sql_site_name: str, + sql_server_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/sqlServers/{sqlServerName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlServerName": _SERIALIZER.url("sql_server_name", sql_server_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_request( + resource_group_name: str, + site_name: str, + sql_site_name: str, + sql_server_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/sqlServers/{sqlServerName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlServerName": _SERIALIZER.url("sql_server_name", sql_server_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class SqlServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_servers_controller` 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_by_sql_site( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.SqlServerV2"]: + """Gets the sql servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either SqlServerV2 or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.SqlServerV2] + :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.SqlServerV2ListResult] = 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_sql_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("SqlServerV2ListResult", 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, site_name: str, sql_site_name: str, sql_server_name: str, **kwargs: Any + ) -> _models.SqlServerV2: + """Gets the sql server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :return: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :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.SqlServerV2] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_server_name=sql_server_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("SqlServerV2", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + sql_server_name: str, + body: _models.SqlServerV2Update, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlServerV2: + """Updates the sql server tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :param body: The resource properties to be updated. Required. + :type body: ~azure.mgmt.migrate.models.SqlServerV2Update + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + sql_server_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlServerV2: + """Updates the sql server tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :param body: The resource properties to be updated. 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: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + sql_server_name: str, + body: Union[_models.SqlServerV2Update, IO[bytes]], + **kwargs: Any + ) -> _models.SqlServerV2: + """Updates the sql server tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param sql_server_name: SQL server name. Required. + :type sql_server_name: str + :param body: The resource properties to be updated. Is either a SqlServerV2Update type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.SqlServerV2Update or IO[bytes] + :return: SqlServerV2 or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlServerV2 + :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.SqlServerV2] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SqlServerV2Update") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + sql_server_name=sql_server_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SqlServerV2", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_sites_controller_operations.py new file mode 100644 index 000000000000..a2e620c38e16 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_sql_sites_controller_operations.py @@ -0,0 +1,1721 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_master_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, sql_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, sql_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, sql_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, sql_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_error_summary_request( + resource_group_name: str, site_name: str, sql_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/errorSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_export_sql_server_errors_request( + resource_group_name: str, site_name: str, sql_site_name: str, subscription_id: str, *, json: JSON, **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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/exportSqlServerErrors", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, json=json, **kwargs) + + +def build_export_sql_servers_request( + resource_group_name: str, site_name: str, sql_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/exportSqlServers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_refresh_request( + resource_group_name: str, site_name: str, sql_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/refresh", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_summary_request( + resource_group_name: str, site_name: str, sql_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/sqlSites/{sqlSiteName}/summary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "sqlSiteName": _SERIALIZER.url("sql_site_name", sql_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 SqlSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`sql_sites_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.SqlSite"]: + """Method to get all sites. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either SqlSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.SqlSite] + :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.SqlSiteListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("SqlSiteListResult", 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, site_name: str, sql_site_name: str, **kwargs: Any) -> _models.SqlSite: + """Method to get a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :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.SqlSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.SqlSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlSite: + """Method to create a SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.SqlSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SqlSite: + """Method to create a SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: Resource create 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: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.SqlSite, IO[bytes]], + **kwargs: Any + ) -> _models.SqlSite: + """Method to create a SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: Resource create parameters. Is either a SqlSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.SqlSite or IO[bytes] + :return: SqlSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSite + :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.SqlSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "SqlSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("SqlSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + def _update_initial( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + properties: Union[_models.SqlSiteUpdate, IO[bytes]], + **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(properties, (IOBase, bytes)): + _content = properties + else: + _json = self._serialize.body(properties, "SqlSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + sql_site_name: str, + properties: _models.SqlSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.SqlSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: ~azure.mgmt.migrate.models.SqlSiteUpdate + :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 SqlSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.SqlSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + properties: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.SqlSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: 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 SqlSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.SqlSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + properties: Union[_models.SqlSiteUpdate, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.SqlSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param properties: The resource properties to be updated. Is either a SqlSiteUpdate type or a + IO[bytes] type. Required. + :type properties: ~azure.mgmt.migrate.models.SqlSiteUpdate or IO[bytes] + :return: An instance of LROPoller that returns either SqlSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.SqlSite] + :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.SqlSite] = 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, + site_name=site_name, + sql_site_name=sql_site_name, + properties=properties, + 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("SqlSite", 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.SqlSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.SqlSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> None: + """Deletes the SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_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, + site_name=site_name, + sql_site_name=sql_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @overload + def error_summary( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.ErrorSummaryRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def error_summary( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. 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: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def error_summary( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.ErrorSummaryRequest, IO[bytes]], + **kwargs: Any + ) -> _models.SiteErrorSummary: + """Method to get error summary from SQL site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Is either a ErrorSummaryRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest or IO[bytes] + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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.SiteErrorSummary] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ErrorSummaryRequest") + + _request = build_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _export_sql_server_errors_initial( + self, resource_group_name: str, site_name: str, sql_site_name: str, body: JSON, **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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_export_sql_server_errors_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_sql_server_errors( + self, resource_group_name: str, site_name: str, sql_site_name: str, body: JSON, **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: JSON + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[JSON] = 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._export_sql_server_errors_initial( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _export_sql_servers_initial( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.ExportSqlServersRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportSqlServersRequest") + + _request = build_export_sql_servers_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_sql_servers( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.ExportSqlServersRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportSqlServersRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_sql_servers( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_sql_servers( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.ExportSqlServersRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing SQL servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Is either a ExportSqlServersRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportSqlServersRequest or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_sql_servers_initial( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _refresh_initial( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.SqlSiteRefreshBody, IO[bytes]], + **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: + _json = self._serialize.body(body, "SqlSiteRefreshBody") + + _request = build_refresh_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_refresh( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: _models.SqlSiteRefreshBody, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.SqlSiteRefreshBody + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_refresh( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_refresh( + self, + resource_group_name: str, + site_name: str, + sql_site_name: str, + body: Union[_models.SqlSiteRefreshBody, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :param body: The content of the action request. Is either a SqlSiteRefreshBody type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.SqlSiteRefreshBody or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._refresh_initial( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def summary( + self, resource_group_name: str, site_name: str, sql_site_name: str, **kwargs: Any + ) -> _models.SqlSiteUsage: + """Method to get site usage/summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param sql_site_name: SQL site name. Required. + :type sql_site_name: str + :return: SqlSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SqlSiteUsage + :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.SqlSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + sql_site_name=sql_site_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("SqlSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_applications_controller_operations.py new file mode 100644 index 000000000000..72422268d82b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_applications_controller_operations.py @@ -0,0 +1,458 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_web_app_site_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/tomcatWebApplications", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + web_app_site_name: str, + web_application_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/tomcatWebApplications/{webApplicationName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "webApplicationName": _SERIALIZER.url( + "web_application_name", web_application_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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_update_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + subscription_id: str, + *, + json: JSON, + **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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/tomcatWebApplications/{webApplicationName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "webApplicationName": _SERIALIZER.url( + "web_application_name", web_application_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, json=json, **kwargs) + + +class TomcatWebApplicationsControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`tomcat_web_applications_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.TomcatWebApplications"]: + """Method to get all Tomcat web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either TomcatWebApplications or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.TomcatWebApplications] + :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.TomcatWebApplicationsListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("TomcatWebApplicationsListResult", 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, site_name: str, web_app_site_name: str, web_application_name: str, **kwargs: Any + ) -> _models.TomcatWebApplications: + """Method to get an Tomcat web application. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :return: TomcatWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.TomcatWebApplications + :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.TomcatWebApplications] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_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("TomcatWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + web_application_name: str, + body: JSON, + **kwargs: Any + ) -> _models.TomcatWebApplications: + """Updates the Tomcat web application tags. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_application_name: Web application name. Required. + :type web_application_name: str + :param body: The resource properties to be updated. Required. + :type body: JSON + :return: TomcatWebApplications or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.TomcatWebApplications + :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: str = kwargs.pop("content_type", _headers.pop("Content-Type", "application/json")) + cls: ClsType[_models.TomcatWebApplications] = kwargs.pop("cls", None) + + _json = self._serialize.body(body, "object") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_application_name=web_application_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + 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("TomcatWebApplications", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_servers_controller_operations.py new file mode 100644 index 000000000000..a0e7b81579e7 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_tomcat_web_servers_controller_operations.py @@ -0,0 +1,328 @@ +# 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_by_web_app_site_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/tomcatWebServers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + web_app_site_name: str, + web_server_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/tomcatWebServers/{webServerName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "webServerName": _SERIALIZER.url("web_server_name", web_server_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class TomcatWebServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`tomcat_web_servers_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.TomcatWebServers"]: + """Method to get all Tomcat web servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either TomcatWebServers or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.TomcatWebServers] + :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.TomcatWebServersListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("TomcatWebServersListResult", 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, site_name: str, web_app_site_name: str, web_server_name: str, **kwargs: Any + ) -> _models.TomcatWebServers: + """Method to get an Tomcat web server. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param web_server_name: Web server name. Required. + :type web_server_name: str + :return: TomcatWebServers or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.TomcatWebServers + :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.TomcatWebServers] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + web_server_name=web_server_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("TomcatWebServers", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vcenter_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vcenter_controller_operations.py new file mode 100644 index 000000000000..a9cc9d5792ff --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vcenter_controller_operations.py @@ -0,0 +1,615 @@ +# 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_by_vmware_site_request( + resource_group_name: str, site_name: str, subscription_id: str, *, filter: 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/vcenters", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "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, site_name: str, vcenter_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/vcenters/{vcenterName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "vcenterName": _SERIALIZER.url("vcenter_name", vcenter_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, vcenter_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/vcenters/{vcenterName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "vcenterName": _SERIALIZER.url("vcenter_name", vcenter_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, vcenter_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/vcenters/{vcenterName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "vcenterName": _SERIALIZER.url("vcenter_name", vcenter_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 VcenterControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`vcenter_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, filter: Optional[str] = None, **kwargs: Any + ) -> Iterable["_models.Vcenter"]: + """List Vcenter resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :return: An iterator like instance of either Vcenter or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.Vcenter] + :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.VcenterListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + subscription_id=self._config.subscription_id, + filter=filter, + 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("VcenterListResult", 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, site_name: str, vcenter_name: str, **kwargs: Any) -> _models.Vcenter: + """Get a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :return: Vcenter or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.Vcenter + :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.Vcenter] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + vcenter_name=vcenter_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("Vcenter", 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, + site_name: str, + vcenter_name: str, + body: Union[_models.Vcenter, IO[bytes]], + **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: + _json = self._serialize.body(body, "Vcenter") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + vcenter_name=vcenter_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + vcenter_name: str, + body: _models.Vcenter, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.Vcenter]: + """Create a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.Vcenter + :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 Vcenter or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.Vcenter] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + site_name: str, + vcenter_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.Vcenter]: + """Create a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :param body: Resource create 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 either Vcenter or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.Vcenter] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + site_name: str, + vcenter_name: str, + body: Union[_models.Vcenter, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.Vcenter]: + """Create a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_name: str + :param body: Resource create parameters. Is either a Vcenter type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.Vcenter or IO[bytes] + :return: An instance of LROPoller that returns either Vcenter or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.Vcenter] + :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.Vcenter] = 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, + site_name=site_name, + vcenter_name=vcenter_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("Vcenter", 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.Vcenter].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.Vcenter]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, site_name: str, vcenter_name: str, **kwargs: Any + ) -> None: + """Delete a Vcenter. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param vcenter_name: VCenters name. Required. + :type vcenter_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, + site_name=site_name, + vcenter_name=vcenter_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_host_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_host_controller_operations.py new file mode 100644 index 000000000000..e35f4ec209cc --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_host_controller_operations.py @@ -0,0 +1,269 @@ +# 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_by_vmware_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/hosts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, host_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/hosts/{hostName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "hostName": _SERIALIZER.url("host_name", host_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class VmwareHostControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`vmware_host_controller` 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_by_vmware_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.VmwareHost"]: + """List VmwareHost resources by VmwareSite. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either VmwareHost or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.VmwareHost] + :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.VmwareHostListResult] = 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_vmware_site_request( + resource_group_name=resource_group_name, + site_name=site_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("VmwareHostListResult", 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, site_name: str, host_name: str, **kwargs: Any) -> _models.VmwareHost: + """Get a VmwareHost. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param host_name: Hosts name. Required. + :type host_name: str + :return: VmwareHost or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareHost + :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.VmwareHost] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + host_name=host_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("VmwareHost", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_operations_status_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_operations_status_operations.py new file mode 100644 index 000000000000..285de6fe8a57 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_operations_status_operations.py @@ -0,0 +1,154 @@ +# 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_vmware_operation_status_request( # pylint: disable=name-too-long + resource_group_name: str, site_name: str, operation_status_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/operationsStatus/{operationStatusName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "operationStatusName": _SERIALIZER.url( + "operation_status_name", operation_status_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class VmwareOperationsStatusOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`vmware_operations_status` 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_vmware_operation_status( + self, resource_group_name: str, site_name: str, operation_status_name: str, **kwargs: Any + ) -> _models.OperationStatus: + """A operation status resource belonging to a site resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param operation_status_name: Operation status Arm Name. Required. + :type operation_status_name: str + :return: OperationStatus or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.OperationStatus + :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.OperationStatus] = kwargs.pop("cls", None) + + _request = build_get_vmware_operation_status_request( + resource_group_name=resource_group_name, + site_name=site_name, + operation_status_name=operation_status_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("OperationStatus", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_properties_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_properties_controller_operations.py new file mode 100644 index 000000000000..a41e08997e01 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_properties_controller_operations.py @@ -0,0 +1,1001 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_update_dependency_map_status_request( # pylint: disable=name-too-long + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/updateDependencyMapStatus", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_properties_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/updateProperties", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_run_as_account_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/updateRunAsAccount", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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_update_tags_request( + resource_group_name: str, site_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", "2023-10-01-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.OffAzure/vmwareSites/{siteName}/updateTags", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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 VmwarePropertiesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`vmware_properties_controller` 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_dependency_map_status_initial( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **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(update_machine_dep_map_status, (IOBase, bytes)): + _content = update_machine_dep_map_status + else: + _json = self._serialize.body(update_machine_dep_map_status, "UpdateMachineDepMapStatus") + + _request = build_update_dependency_map_status_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: _models.UpdateMachineDepMapStatus, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a + site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_dep_map_status: The content of the action request. Required. + :type update_machine_dep_map_status: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a + site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_dep_map_status: The content of the action request. Required. + :type update_machine_dep_map_status: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_dependency_map_status( + self, + resource_group_name: str, + site_name: str, + update_machine_dep_map_status: Union[_models.UpdateMachineDepMapStatus, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to enable disable dependency map status for machines + in a + site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_dep_map_status: The content of the action request. Is either a + UpdateMachineDepMapStatus type or a IO[bytes] type. Required. + :type update_machine_dep_map_status: ~azure.mgmt.migrate.models.UpdateMachineDepMapStatus or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_dependency_map_status_initial( + resource_group_name=resource_group_name, + site_name=site_name, + update_machine_dep_map_status=update_machine_dep_map_status, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _update_properties_initial( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **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(meta_data, (IOBase, bytes)): + _content = meta_data + else: + _json = self._serialize.body(meta_data, "MachineMetadataCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: _models.MachineMetadataCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update properties for machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update properties for machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Required. + :type meta_data: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + meta_data: Union[_models.MachineMetadataCollection, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update properties for machines in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param meta_data: The content of the action request. Is either a MachineMetadataCollection type + or a IO[bytes] type. Required. + :type meta_data: ~azure.mgmt.migrate.models.MachineMetadataCollection or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + meta_data=meta_data, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _update_run_as_account_initial( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: Union[_models.UpdateMachineRunAsAccount, IO[bytes]], + **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(update_machine_run_as_account, (IOBase, bytes)): + _content = update_machine_run_as_account + else: + _json = self._serialize.body(update_machine_run_as_account, "UpdateMachineRunAsAccount") + + _request = build_update_run_as_account_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_run_as_account( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: _models.UpdateMachineRunAsAccount, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_run_as_account: The content of the action request. Required. + :type update_machine_run_as_account: ~azure.mgmt.migrate.models.UpdateMachineRunAsAccount + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_run_as_account( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_run_as_account: The content of the action request. Required. + :type update_machine_run_as_account: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_run_as_account( + self, + resource_group_name: str, + site_name: str, + update_machine_run_as_account: Union[_models.UpdateMachineRunAsAccount, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_run_as_account: The content of the action request. Is either a + UpdateMachineRunAsAccount type or a IO[bytes] type. Required. + :type update_machine_run_as_account: ~azure.mgmt.migrate.models.UpdateMachineRunAsAccount or + IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_run_as_account_initial( + resource_group_name=resource_group_name, + site_name=site_name, + update_machine_run_as_account=update_machine_run_as_account, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _update_tags_initial( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: Union[_models.UpdateMachineTags, IO[bytes]], + **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(update_machine_tags, (IOBase, bytes)): + _content = update_machine_tags + else: + _json = self._serialize.body(update_machine_tags, "UpdateMachineTags") + + _request = build_update_tags_request( + resource_group_name=resource_group_name, + site_name=site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_tags( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: _models.UpdateMachineTags, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_tags: The content of the action request. Required. + :type update_machine_tags: ~azure.mgmt.migrate.models.UpdateMachineTags + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_tags( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_tags: The content of the action request. Required. + :type update_machine_tags: 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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_tags( + self, + resource_group_name: str, + site_name: str, + update_machine_tags: Union[_models.UpdateMachineTags, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to associate Run as account to machine + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param update_machine_tags: The content of the action request. Is either a UpdateMachineTags + type or a IO[bytes] type. Required. + :type update_machine_tags: ~azure.mgmt.migrate.models.UpdateMachineTags or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_tags_initial( + resource_group_name=resource_group_name, + site_name=site_name, + update_machine_tags=update_machine_tags, + 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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_software_inventories_controller_operations.py new file mode 100644 index 000000000000..bd51edffdb2d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_vmware_software_inventories_controller_operations.py @@ -0,0 +1,290 @@ +# 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, Union +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_get_machine_software_inventory_request( # pylint: disable=name-too-long + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/machines/{machineName}/softwareInventories/{default}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "default": _SERIALIZER.url("default", default, "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_machine_resource_request( + resource_group_name: str, site_name: str, machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/vmwareSites/{siteName}/machines/{machineName}/softwareinventories", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "machineName": _SERIALIZER.url("machine_name", machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class VmwareSoftwareInventoriesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`vmware_software_inventories_controller` 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_machine_software_inventory( + self, + resource_group_name: str, + site_name: str, + machine_name: str, + default: Union[str, _models.Default], + **kwargs: Any + ) -> _models.VmwareMachineSoftwareInventory: + """Method to get a machines software inventory like applications and roles. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :param default: Default value. "default" Required. + :type default: str or ~azure.mgmt.migrate.models.Default + :return: VmwareMachineSoftwareInventory or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.VmwareMachineSoftwareInventory + :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.VmwareMachineSoftwareInventory] = kwargs.pop("cls", None) + + _request = build_get_machine_software_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_name, + default=default, + 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("VmwareMachineSoftwareInventory", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def list_by_machine_resource( + self, resource_group_name: str, site_name: str, machine_name: str, **kwargs: Any + ) -> Iterable["_models.VmwareMachineSoftwareInventory"]: + """List VmwareMachineSoftwareInventory resources by MachineResource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param machine_name: Machine name. Required. + :type machine_name: str + :return: An iterator like instance of either VmwareMachineSoftwareInventory or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.VmwareMachineSoftwareInventory] + :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.VmwareMachineSoftwareInventoryListResult] = 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_machine_resource_request( + resource_group_name=resource_group_name, + site_name=site_name, + machine_name=machine_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("VmwareMachineSoftwareInventoryListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_discovery_site_data_sources_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_discovery_site_data_sources_controller_operations.py new file mode 100644 index 000000000000..2f8bd88a99c8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_discovery_site_data_sources_controller_operations.py @@ -0,0 +1,746 @@ +# 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_by_web_app_site_request( + resource_group_name: str, site_name: str, web_app_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/discoverySiteDataSources", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/discoverySiteDataSources/{discoverySiteDataSourceName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "discoverySiteDataSourceName": _SERIALIZER.url( + "discovery_site_data_source_name", discovery_site_data_source_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/discoverySiteDataSources/{discoverySiteDataSourceName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "discoverySiteDataSourceName": _SERIALIZER.url( + "discovery_site_data_source_name", discovery_site_data_source_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/discoverySiteDataSources/{discoverySiteDataSourceName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "discoverySiteDataSourceName": _SERIALIZER.url( + "discovery_site_data_source_name", discovery_site_data_source_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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 WebAppDiscoverySiteDataSourcesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`web_app_discovery_site_data_sources_controller` 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_by_web_app_site( + self, resource_group_name: str, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> Iterable["_models.DiscoverySiteDataSource"]: + """Method to get all Web app data sources in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: An iterator like instance of either DiscoverySiteDataSource or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :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.DiscoverySiteDataSourceListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("DiscoverySiteDataSourceListResult", 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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> _models.DiscoverySiteDataSource: + """Method to get a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :return: DiscoverySiteDataSource or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.DiscoverySiteDataSource + :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.DiscoverySiteDataSource] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("DiscoverySiteDataSource", 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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.DiscoverySiteDataSource, IO[bytes]], + **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: + _json = self._serialize.body(body, "DiscoverySiteDataSource") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + 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_create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: _models.DiscoverySiteDataSource, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.DiscoverySiteDataSource]: + """Method to create or update a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.DiscoverySiteDataSource + :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 DiscoverySiteDataSource or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.DiscoverySiteDataSource]: + """Method to create or update a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create 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 either DiscoverySiteDataSource or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + body: Union[_models.DiscoverySiteDataSource, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.DiscoverySiteDataSource]: + """Method to create or update a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_name: str + :param body: Resource create parameters. Is either a DiscoverySiteDataSource type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.DiscoverySiteDataSource or IO[bytes] + :return: An instance of LROPoller that returns either DiscoverySiteDataSource or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.DiscoverySiteDataSource] + :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.DiscoverySiteDataSource] = 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, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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("DiscoverySiteDataSource", 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.DiscoverySiteDataSource].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.DiscoverySiteDataSource]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _delete_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_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, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + web_app_site_name: str, + discovery_site_data_source_name: str, + **kwargs: Any + ) -> LROPoller[None]: + """Method to delete a Web app data source in site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param discovery_site_data_source_name: Data Source ARM name. Required. + :type discovery_site_data_source_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, + site_name=site_name, + web_app_site_name=web_app_site_name, + discovery_site_data_source_name=discovery_site_data_source_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/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_extended_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_extended_machines_controller_operations.py new file mode 100644 index 000000000000..ac3f7c0911e5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_extended_machines_controller_operations.py @@ -0,0 +1,336 @@ +# 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_by_web_app_site_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/extendedMachines", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # 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, + site_name: str, + web_app_site_name: str, + extended_machine_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/extendedMachines/{extendedMachineName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "extendedMachineName": _SERIALIZER.url( + "extended_machine_name", extended_machine_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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) + + +class WebAppExtendedMachinesControllerOperations: # pylint: disable=name-too-long + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`web_app_extended_machines_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.WebAppExtendedMachine"]: + """Method to get all extended machines. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either WebAppExtendedMachine or the result of + cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.WebAppExtendedMachine] + :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.WebAppExtendedMachineListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("WebAppExtendedMachineListResult", 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, + site_name: str, + web_app_site_name: str, + extended_machine_name: str, + **kwargs: Any + ) -> _models.WebAppExtendedMachine: + """Method to get a extended machine. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param extended_machine_name: Extended machine name. Required. + :type extended_machine_name: str + :return: WebAppExtendedMachine or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppExtendedMachine + :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.WebAppExtendedMachine] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + extended_machine_name=extended_machine_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("WebAppExtendedMachine", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_properties_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_properties_controller_operations.py new file mode 100644 index 000000000000..f958010acb8c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_properties_controller_operations.py @@ -0,0 +1,312 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_update_properties_request( + resource_group_name: str, site_name: str, web_app_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/updateProperties", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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 WebAppPropertiesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`web_app_properties_controller` 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_properties_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.WebAppPropertiesCollection, IO[bytes]], + **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: + _json = self._serialize.body(body, "WebAppPropertiesCollection") + + _request = build_update_properties_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_properties( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.WebAppPropertiesCollection, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update properties for web applications. + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.WebAppPropertiesCollection + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update properties for web applications. + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update_properties( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.WebAppPropertiesCollection, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to update properties for web applications. + in a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a WebAppPropertiesCollection type or + a IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.WebAppPropertiesCollection or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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_properties_initial( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..ca7a536ee0d6 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_run_as_accounts_controller_operations.py @@ -0,0 +1,288 @@ +# 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_by_web_app_site_request( + resource_group_name: str, site_name: str, web_app_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/runasaccounts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, + site_name: str, + web_app_site_name: str, + account_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/runasaccounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + "accountName": _SERIALIZER.url("account_name", account_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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) + + +class WebAppRunAsAccountsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`web_app_run_as_accounts_controller` 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_by_web_app_site( + self, resource_group_name: str, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> Iterable["_models.WebAppRunAsAccount"]: + """Method to get all run as accounts. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: An iterator like instance of either WebAppRunAsAccount or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.WebAppRunAsAccount] + :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.WebAppRunAsAccountListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("WebAppRunAsAccountListResult", 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, site_name: str, web_app_site_name: str, account_name: str, **kwargs: Any + ) -> _models.WebAppRunAsAccount: + """Method to get run as account. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param account_name: Run as account ARM name. Required. + :type account_name: str + :return: WebAppRunAsAccount or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppRunAsAccount + :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.WebAppRunAsAccount] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + account_name=account_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("WebAppRunAsAccount", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_sites_controller_operations.py new file mode 100644 index 000000000000..8e6f0dd5c61a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_app_sites_controller_operations.py @@ -0,0 +1,1633 @@ +# 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 +JSON = MutableMapping[str, Any] # pylint: disable=unsubscriptable-object +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_master_site_request( + resource_group_name: str, site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + } + + _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, site_name: str, web_app_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, site_name: str, web_app_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, site_name: str, web_app_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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, site_name: str, web_app_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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_error_summary_request( + resource_group_name: str, site_name: str, web_app_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/errorSummary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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_export_inventory_request( + resource_group_name: str, site_name: str, web_app_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/exportInventory", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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_refresh_request( + resource_group_name: str, site_name: str, web_app_site_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", "2023-10-01-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.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/refresh", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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_summary_request( + resource_group_name: str, site_name: str, web_app_site_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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/summary", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _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 WebAppSitesControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`web_app_sites_controller` 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_by_master_site( + self, resource_group_name: str, site_name: str, **kwargs: Any + ) -> Iterable["_models.WebAppSite"]: + """Method to get all sites. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :return: An iterator like instance of either WebAppSite or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.WebAppSite] + :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.WebAppSiteListResult] = 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_master_site_request( + resource_group_name=resource_group_name, + site_name=site_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("WebAppSiteListResult", 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, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> _models.WebAppSite: + """Method to get a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :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.WebAppSite] = kwargs.pop("cls", None) + + _request = build_get_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("WebAppSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.WebAppSite, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WebAppSite: + """Method to create a WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: Resource create parameters. Required. + :type body: ~azure.mgmt.migrate.models.WebAppSite + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.WebAppSite: + """Method to create a WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: Resource create 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: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.WebAppSite, IO[bytes]], + **kwargs: Any + ) -> _models.WebAppSite: + """Method to create a WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: Resource create parameters. Is either a WebAppSite type or a IO[bytes] type. + Required. + :type body: ~azure.mgmt.migrate.models.WebAppSite or IO[bytes] + :return: WebAppSite or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSite + :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.WebAppSite] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "WebAppSite") + + _request = build_create_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = self._deserialize("WebAppSite", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + def _update_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + properties: Union[_models.WebAppSiteUpdate, IO[bytes]], + **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(properties, (IOBase, bytes)): + _content = properties + else: + _json = self._serialize.body(properties, "WebAppSiteUpdate") + + _request = build_update_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, + site_name: str, + web_app_site_name: str, + properties: _models.WebAppSiteUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.WebAppSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: ~azure.mgmt.migrate.models.WebAppSiteUpdate + :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 WebAppSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.WebAppSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + properties: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.WebAppSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param properties: The resource properties to be updated. Required. + :type properties: 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 WebAppSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.WebAppSite] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + properties: Union[_models.WebAppSiteUpdate, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.WebAppSite]: + """Method to update an existing site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param properties: The resource properties to be updated. Is either a WebAppSiteUpdate type or + a IO[bytes] type. Required. + :type properties: ~azure.mgmt.migrate.models.WebAppSiteUpdate or IO[bytes] + :return: An instance of LROPoller that returns either WebAppSite or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.migrate.models.WebAppSite] + :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.WebAppSite] = 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, + site_name=site_name, + web_app_site_name=web_app_site_name, + properties=properties, + 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("WebAppSite", 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.WebAppSite].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.WebAppSite]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _delete_initial( + self, resource_group_name: str, site_name: str, web_app_site_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, + site_name=site_name, + web_app_site_name=web_app_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> LROPoller[None]: + """Deletes the WebApp site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_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, + site_name=site_name, + web_app_site_name=web_app_site_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 + + @overload + def error_summary( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.ErrorSummaryRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """MMethod to get error summary from web app site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def error_summary( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.SiteErrorSummary: + """MMethod to get error summary from web app site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def error_summary( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ErrorSummaryRequest, IO[bytes]], + **kwargs: Any + ) -> _models.SiteErrorSummary: + """MMethod to get error summary from web app site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a ErrorSummaryRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ErrorSummaryRequest or IO[bytes] + :return: SiteErrorSummary or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.SiteErrorSummary + :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.SiteErrorSummary] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(body, (IOBase, bytes)): + _content = body + else: + _json = self._serialize.body(body, "ErrorSummaryRequest") + + _request = build_error_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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.ErrorResponse, pipeline_response) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + deserialized = self._deserialize("SiteErrorSummary", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _export_inventory_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ExportWebAppsRequest, IO[bytes]], + **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: + _json = self._serialize.body(body, "ExportWebAppsRequest") + + _request = build_export_inventory_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_export_inventory( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.ExportWebAppsRequest, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing web app inventory. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ExportWebAppsRequest + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_export_inventory( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing web app inventory. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_export_inventory( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ExportWebAppsRequest, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to generate report containing web app inventory. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a ExportWebAppsRequest type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ExportWebAppsRequest or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._export_inventory_initial( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _refresh_initial( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ProxySiteRefreshBody, IO[bytes]], + **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: + _json = self._serialize.body(body, "ProxySiteRefreshBody") + + _request = build_refresh_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + 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_refresh( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: _models.ProxySiteRefreshBody, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Required. + :type body: ~azure.mgmt.migrate.models.ProxySiteRefreshBody + :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 JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_refresh( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. 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 either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_refresh( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + body: Union[_models.ProxySiteRefreshBody, IO[bytes]], + **kwargs: Any + ) -> LROPoller[JSON]: + """Method to refresh a site. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param body: The content of the action request. Is either a ProxySiteRefreshBody type or a + IO[bytes] type. Required. + :type body: ~azure.mgmt.migrate.models.ProxySiteRefreshBody or IO[bytes] + :return: An instance of LROPoller that returns either JSON or the result of cls(response) + :rtype: ~azure.core.polling.LROPoller[JSON] + :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[JSON] = 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._refresh_initial( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("object", 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[JSON].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[JSON](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def summary( + self, resource_group_name: str, site_name: str, web_app_site_name: str, **kwargs: Any + ) -> _models.WebAppSiteUsage: + """Method to get site usage/summary. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :return: WebAppSiteUsage or the result of cls(response) + :rtype: ~azure.mgmt.migrate.models.WebAppSiteUsage + :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.WebAppSiteUsage] = kwargs.pop("cls", None) + + _request = build_summary_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_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("WebAppSiteUsage", pipeline_response.http_response) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_applications_controller_operations.py new file mode 100644 index 000000000000..8ef515aab344 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_applications_controller_operations.py @@ -0,0 +1,222 @@ +# 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_by_web_app_site_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/webApplications", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class WebApplicationsControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`web_applications_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.WebApplication"]: + """Method to get all IIS web applications. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either WebApplication or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.WebApplication] + :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.WebApplicationListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("WebApplicationListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_servers_controller_operations.py new file mode 100644 index 000000000000..2833a06b6925 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/operations/_web_servers_controller_operations.py @@ -0,0 +1,222 @@ +# 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_by_web_app_site_request( + resource_group_name: str, + site_name: str, + web_app_site_name: str, + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = 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", "2023-10-01-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.OffAzure/masterSites/{siteName}/webAppSites/{webAppSiteName}/webServers", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), + "resourceGroupName": _SERIALIZER.url( + "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 + ), + "siteName": _SERIALIZER.url("site_name", site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$"), + "webAppSiteName": _SERIALIZER.url( + "web_app_site_name", web_app_site_name, "str", pattern=r"^[a-zA-Z0-9-]{3,24}$" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if filter is not None: + _params["filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["top"] = _SERIALIZER.query("top", top, "str") + if continuation_token_parameter is not None: + _params["continuationToken"] = _SERIALIZER.query( + "continuation_token_parameter", continuation_token_parameter, "str" + ) + if total_record_count is not None: + _params["totalRecordCount"] = _SERIALIZER.query("total_record_count", total_record_count, "int") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class WebServersControllerOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.migrate.AzureMigrateDiscoveryService`'s + :attr:`web_servers_controller` 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_by_web_app_site( + self, + resource_group_name: str, + site_name: str, + web_app_site_name: str, + filter: Optional[str] = None, + top: Optional[str] = None, + continuation_token_parameter: Optional[str] = None, + total_record_count: Optional[int] = None, + **kwargs: Any + ) -> Iterable["_models.WebServer"]: + """Method to get all web servers. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param site_name: Site name. Required. + :type site_name: str + :param web_app_site_name: Web app site name. Required. + :type web_app_site_name: str + :param filter: filter query. Default value is None. + :type filter: str + :param top: page size query. Default value is None. + :type top: str + :param continuation_token_parameter: Optional parameter for continuation token. Default value + is None. + :type continuation_token_parameter: str + :param total_record_count: Total count of machines in the given site. Default value is None. + :type total_record_count: int + :return: An iterator like instance of either WebServer or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.migrate.models.WebServer] + :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.WebServerListResult] = 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_web_app_site_request( + resource_group_name=resource_group_name, + site_name=site_name, + web_app_site_name=web_app_site_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + continuation_token_parameter=continuation_token_parameter, + total_record_count=total_record_count, + 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("WebServerListResult", 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) diff --git a/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/py.typed b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/py.typed new file mode 100644 index 000000000000..e5aff4f83af8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/azure/mgmt/migrate/py.typed @@ -0,0 +1 @@ +# Marker file for PEP 561. \ No newline at end of file diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_client_group_members_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_client_group_members_maximum_set_gen.py new file mode 100644 index 000000000000..b4676577eff2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_client_group_members_maximum_set_gen.py @@ -0,0 +1,50 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python dependency_map_controller_client_group_members_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.dependency_map_controller.begin_client_group_members( + resource_group_name="rgmigrate", + site_name="1332kQXJXIUI1E", + map_request={ + "endTime": "2024-01-18T11:33:08.082Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "machineId": "zbsqaglovhnxlroparbdzfs", + "processGroupName": "lawalpbrzupvjeglgvluyjtvijnqcj", + "processName": "ygqxfpuvfsiiotyxveq", + "startTime": "2024-01-18T11:33:08.082Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/DependencyMapController_ClientGroupMembers_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_export_dependencies_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_export_dependencies_maximum_set_gen.py new file mode 100644 index 000000000000..9d2ff516f798 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_export_dependencies_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python dependency_map_controller_export_dependencies_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.dependency_map_controller.begin_export_dependencies( + resource_group_name="rgmigrate", + site_name="5Ue8NYF35i9k", + request_body={"endTime": "2024-01-18T11:33:13.789Z", "startTime": "2024-01-18T11:33:13.789Z"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/DependencyMapController_ExportDependencies_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_coarse_map_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_coarse_map_maximum_set_gen.py new file mode 100644 index 000000000000..b7efc9457243 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_coarse_map_maximum_set_gen.py @@ -0,0 +1,47 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python dependency_map_controller_generate_coarse_map_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.dependency_map_controller.begin_generate_coarse_map( + resource_group_name="rgmigrate", + site_name="R875", + map_request={ + "endTime": "2024-01-18T11:33:15.220Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "startTime": "2024-01-18T11:33:15.220Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/DependencyMapController_GenerateCoarseMap_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_detailed_map_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_detailed_map_maximum_set_gen.py new file mode 100644 index 000000000000..d5727cdab36f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_generate_detailed_map_maximum_set_gen.py @@ -0,0 +1,48 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python dependency_map_controller_generate_detailed_map_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.dependency_map_controller.begin_generate_detailed_map( + resource_group_name="rgmigrate", + site_name="Q-Q-Cu-C-YO-9U6V", + map_request={ + "endTime": "2024-01-18T11:33:15.874Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "machineId": "mqiazddot", + "startTime": "2024-01-18T11:33:15.874Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/DependencyMapController_GenerateDetailedMap_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_server_group_members_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_server_group_members_maximum_set_gen.py new file mode 100644 index 000000000000..8ddfcdc2cc49 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/dependency_map_controller_server_group_members_maximum_set_gen.py @@ -0,0 +1,48 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python dependency_map_controller_server_group_members_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.dependency_map_controller.begin_server_group_members( + resource_group_name="rgmigrate", + site_name="2128Na75J6T7D-RY5-", + map_request={ + "endTime": "2024-01-18T11:33:31.988Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "serverPort": 27, + "startTime": "2024-01-18T11:33:31.988Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/DependencyMapController_ServerGroupMembers_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_create_cluster_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_create_cluster_maximum_set_gen.py new file mode 100644 index 000000000000..5bcbdb28b012 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_create_cluster_maximum_set_gen.py @@ -0,0 +1,51 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_cluster_controller_create_cluster_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_cluster_controller.begin_create_cluster( + resource_group_name="rgmigrate", + site_name="7-ZE7-O1HU2-s-EyMu6", + cluster_name="67b-v01-nC3--6tCUl", + body={ + "properties": { + "fqdn": "rpfphmlbzericn", + "hostFqdnList": ["ozzrrnscesqywisszgtrnqkxp"], + "provisioningState": "Created", + "runAsAccountId": "qgqrfeczloubuaitiudd", + } + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervClusterController_CreateCluster_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..edf7e53511f3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_delete_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_cluster_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.hyperv_cluster_controller.delete( + resource_group_name="rgmigrate", + site_name="THF16H-1PV3v6C4Jc4W", + cluster_name="-8KDr--3f3-83o7G914d66E", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervClusterController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_get_cluster_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_get_cluster_maximum_set_gen.py new file mode 100644 index 000000000000..cdc0781aef86 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_get_cluster_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_cluster_controller_get_cluster_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_cluster_controller.get_cluster( + resource_group_name="rgmigrate", + site_name="l4-RY9H60-H3d6Rp-2-k3", + cluster_name="u0-28M--", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervClusterController_GetCluster_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_list_by_hyperv_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_list_by_hyperv_site_maximum_set_gen.py new file mode 100644 index 000000000000..3a184d2cd884 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_cluster_controller_list_by_hyperv_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_cluster_controller_list_by_hyperv_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_cluster_controller.list_by_hyperv_site( + resource_group_name="rgmigrate", + site_name="d-h4D-H-yG82", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervClusterController_ListByHypervSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_client_group_members_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_client_group_members_maximum_set_gen.py new file mode 100644 index 000000000000..6aabf58bda6d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_client_group_members_maximum_set_gen.py @@ -0,0 +1,50 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_dependency_map_controller_client_group_members_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_dependency_map_controller.begin_client_group_members( + resource_group_name="rgmigrate", + site_name="K3NSZP6wU26Dj8-55", + map_request={ + "endTime": "2024-01-18T11:33:08.082Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "machineId": "zbsqaglovhnxlroparbdzfs", + "processGroupName": "lawalpbrzupvjeglgvluyjtvijnqcj", + "processName": "ygqxfpuvfsiiotyxveq", + "startTime": "2024-01-18T11:33:08.082Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervDependencyMapController_ClientGroupMembers_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_export_dependencies_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_export_dependencies_maximum_set_gen.py new file mode 100644 index 000000000000..d62a9a61d840 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_export_dependencies_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_dependency_map_controller_export_dependencies_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_dependency_map_controller.begin_export_dependencies( + resource_group_name="rgmigrate", + site_name="G--42YNE78-MW", + request_body={"endTime": "2024-01-18T11:33:13.789Z", "startTime": "2024-01-18T11:33:13.789Z"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervDependencyMapController_ExportDependencies_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_coarse_map_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_coarse_map_maximum_set_gen.py new file mode 100644 index 000000000000..14982872af06 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_coarse_map_maximum_set_gen.py @@ -0,0 +1,47 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_dependency_map_controller_generate_coarse_map_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_dependency_map_controller.begin_generate_coarse_map( + resource_group_name="rgmigrate", + site_name="-66145C-Ov8i-l", + map_request={ + "endTime": "2024-01-18T11:33:15.220Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "startTime": "2024-01-18T11:33:15.220Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervDependencyMapController_GenerateCoarseMap_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_detailed_map_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_detailed_map_maximum_set_gen.py new file mode 100644 index 000000000000..64a19cb0183a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_generate_detailed_map_maximum_set_gen.py @@ -0,0 +1,48 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_dependency_map_controller_generate_detailed_map_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_dependency_map_controller.begin_generate_detailed_map( + resource_group_name="rgmigrate", + site_name="84g-119S-X8dCJm6", + map_request={ + "endTime": "2024-01-18T11:33:15.874Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "machineId": "mqiazddot", + "startTime": "2024-01-18T11:33:15.874Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervDependencyMapController_GenerateDetailedMap_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_server_group_members_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_server_group_members_maximum_set_gen.py new file mode 100644 index 000000000000..825eb7435c17 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_server_group_members_maximum_set_gen.py @@ -0,0 +1,48 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_dependency_map_controller_server_group_members_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_dependency_map_controller.begin_server_group_members( + resource_group_name="rgmigrate", + site_name="--6U-bZ67-5XIt91T", + map_request={ + "endTime": "2024-01-18T11:33:31.988Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "serverPort": 27, + "startTime": "2024-01-18T11:33:31.988Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervDependencyMapController_ServerGroupMembers_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_update_dependency_map_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_update_dependency_map_status_maximum_set_gen.py new file mode 100644 index 000000000000..23f8600daa74 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_dependency_map_controller_update_dependency_map_status_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_dependency_map_controller_update_dependency_map_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_dependency_map_controller.begin_update_dependency_map_status( + resource_group_name="rgmigrate", + site_name="BXINc225D6-1Hyr7", + map_request={"machines": [{"isDependencyMapToBeEnabled": True, "machineId": "ysqizqtyaskpprxkqaeyyopln"}]}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervDependencyMapController_UpdateDependencyMapStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..a7cf2c28ec82 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_create_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_host_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_host_controller.begin_create( + resource_group_name="rgmigrate", + site_name="f3m9-Tf--3-n7-40USK7XX", + host_name="1FXhE45296-eJUF1Y-", + body={"properties": {"fqdn": "lk", "provisioningState": "Created", "runAsAccountId": "epkvjlsbaigwfxwgzo"}}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervHostController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..c74b1680e7ba --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_delete_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_host_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.hyperv_host_controller.begin_delete( + resource_group_name="rgmigrate", + site_name="EL2-eW4-c784t2-", + host_name="HURB-TsA", + ).result() + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervHostController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..87a0ccbc37f1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_host_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_host_controller.get( + resource_group_name="rgmigrate", + site_name="-3zTU--SvlGY-FoUh", + host_name="25JSQ7RhWxXUIW-ELi", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervHostController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_list_by_hyperv_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_list_by_hyperv_site_maximum_set_gen.py new file mode 100644 index 000000000000..a5df0ad38091 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_host_controller_list_by_hyperv_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_host_controller_list_by_hyperv_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_host_controller.list_by_hyperv_site( + resource_group_name="rgmigrate", + site_name="537erp8U864VI5F7", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervHostController_ListByHypervSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..62b921ab5e0e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_jobs_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_jobs_controller.get( + resource_group_name="rgmigrate", + site_name="66xjB52-9--e", + job_name="V-258-4xS165l95", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervJobsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_list_by_vmware_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_list_by_vmware_site_maximum_set_gen.py new file mode 100644 index 000000000000..156d4cb2beef --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_controller_list_by_vmware_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_jobs_controller_list_by_vmware_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_jobs_controller.list_by_vmware_site( + resource_group_name="rgmigrate", + site_name="13y-18qK-", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervJobsController_ListByVmwareSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_get_maximum_set_gen.py new file mode 100644 index 000000000000..cbba16a61ac5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_jobs_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_jobs.get( + resource_group_name="rgmigrate", + site_name="f-WFME---c", + job_name="5B-2L-3sMO-H-n-F", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervJobs_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_list_by_hyperv_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_list_by_hyperv_site_maximum_set_gen.py new file mode 100644 index 000000000000..6884a760080a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_jobs_list_by_hyperv_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_jobs_list_by_hyperv_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_jobs.list_by_hyperv_site( + resource_group_name="rgmigrate", + site_name="71BmgvX-WJF9H5N7-5-cR", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervJobs_ListByHypervSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..4c0bc2fd982d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_machines_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_machines_controller.get( + resource_group_name="rgmigrate", + site_name="55--o3Q", + machine_name="90-HQO", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervMachinesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_list_by_hyperv_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_list_by_hyperv_site_maximum_set_gen.py new file mode 100644 index 000000000000..2941db07f7da --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_list_by_hyperv_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_machines_controller_list_by_hyperv_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_machines_controller.list_by_hyperv_site( + resource_group_name="rgmigrate", + site_name="V-15KAwn1-23Z3J52B-cQC", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervMachinesController_ListByHypervSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..0a8b18d42654 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_maximum_set_gen.py @@ -0,0 +1,67 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_machines_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_machines_controller.update( + resource_group_name="rgmigrate", + site_name="1-DCv-XE---Nq-03", + machine_name="LgE8-B1C--X", + body={ + "properties": { + "allocatedMemoryInMb": 23, + "biosGuid": "dmxswgeyvspvafusl", + "biosSerialNumber": "iyvikycmaekqufzkmdijssqekbrah", + "firmware": "z", + "numberOfProcessorCore": 4, + "operatingSystemDetails": { + "osArchitecture": "lxeotrwmodham", + "osName": "udcjzawhhylqauidhrau", + "osType": "s", + "osVersion": "jzblxvpzhmqdpmot", + }, + "productSupportStatus": { + "currentVersion": "dlrvdqalihttfwvnnbskpzikdx", + "esuStatus": "Unknown", + "esuYear": "Unknown", + "supportEndDate": "2024-01-18T11:33:22.395Z", + "supportStatus": "Unknown", + }, + "runAsAccountId": "lpvwkktxwqlgdbnzg", + "tags": {"key9256": "fyje"}, + } + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervMachinesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_properties_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_properties_maximum_set_gen.py new file mode 100644 index 000000000000..cfa3fbc7a5c1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_machines_controller_update_properties_maximum_set_gen.py @@ -0,0 +1,51 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_machines_controller_update_properties_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_machines_controller.begin_update_properties( + resource_group_name="rgmigrate", + site_name="ewO-81-l-", + meta_data={ + "value": [ + { + "dependencyMapping": "mehkbjgdktlveubatdamkaawrfmkwc", + "machineArmId": "cbfxbnpaesafshmuirygemyx", + "tags": {"key925": "kvcnqygwaiw"}, + } + ] + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervMachinesController_UpdateProperties_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_operations_status_controller_get_hyperv_operations_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_operations_status_controller_get_hyperv_operations_status_maximum_set_gen.py new file mode 100644 index 000000000000..1a85e4b01484 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_operations_status_controller_get_hyperv_operations_status_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_operations_status_controller_get_hyperv_operations_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_operations_status_controller.get_hyperv_operations_status( + resource_group_name="rgmigrate", + site_name="-br7-R", + operation_status_name="74UH-", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervOperationsStatusController_GetHypervOperationsStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..534568e28988 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_run_as_accounts_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_run_as_accounts_controller.get( + resource_group_name="rgmigrate", + site_name="-v1e8lWSoqI2j-FQC-2-Q", + account_name="Qr--WF-B4MFWM6", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervRunAsAccountsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_list_by_hyperv_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_list_by_hyperv_site_maximum_set_gen.py new file mode 100644 index 000000000000..be3f9978e890 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_run_as_accounts_controller_list_by_hyperv_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_run_as_accounts_controller_list_by_hyperv_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_run_as_accounts_controller.list_by_hyperv_site( + resource_group_name="rgmigrate", + site_name="-7QDJ-LoJ82kg-P3YS0-C", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervRunAsAccountsController_ListByHypervSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..68f9eaa93ce2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_create_maximum_set_gen.py @@ -0,0 +1,60 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites_controller.create( + resource_group_name="rgmigrate", + site_name="B9be-JfY48Vc6SDN8d1CCp-", + body={ + "location": "jzfqhihimugilssu", + "properties": { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "bplbumeu", + "discoverySolutionId": "sqgvbumhsdeunojunhypdunvcutlev", + "provisioningState": "Created", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + }, + "tags": {"key9210": "yitjyobwfkbprpkfcmogsywoot"}, + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSitesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..556906d5898f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_delete_maximum_set_gen.py @@ -0,0 +1,41 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.hyperv_sites_controller.delete( + resource_group_name="rgmigrate", + site_name="516m8O9-75LL18U", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSitesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_export_machine_errors_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_export_machine_errors_maximum_set_gen.py new file mode 100644 index 000000000000..ab74b5cad8f1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_export_machine_errors_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_controller_export_machine_errors_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites_controller.begin_export_machine_errors( + resource_group_name="rgmigrate", + site_name="74Ds26", + body={"properties": {"discoveryScope": "AppsAndRoles"}}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSitesController_ExportMachineErrors_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..aca52253807f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_get_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites_controller.get( + resource_group_name="rgmigrate", + site_name="04p-YXAMxEA994743LG35bK-", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSitesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_list_health_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_list_health_summary_maximum_set_gen.py new file mode 100644 index 000000000000..bb04d9b83cfb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_list_health_summary_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_controller_list_health_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites_controller.list_health_summary( + resource_group_name="rgmigrate", + site_name="5CT5D", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSitesController_ListHealthSummary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_summary_maximum_set_gen.py new file mode 100644 index 000000000000..8494b8d8adef --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_summary_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_controller_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites_controller.summary( + resource_group_name="rgmigrate", + site_name="-0-6Byn", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSitesController_Summary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..8b92807655f0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_controller_update_maximum_set_gen.py @@ -0,0 +1,59 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites_controller.update( + resource_group_name="rgmigrate", + site_name="X2UJ-6-3p64TUj", + body={ + "properties": { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "jtphpskximt", + "discoverySolutionId": "woihspbrxqmel", + "provisioningState": "Created", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + }, + "tags": {"key8611": "ifdzfafhrcras"}, + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSitesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_resource_group_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_resource_group_maximum_set_gen.py new file mode 100644 index 000000000000..0307f22cfe62 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_resource_group_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_list_by_resource_group_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites.list_by_resource_group( + resource_group_name="rgmigrate", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSites_ListByResourceGroup_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_subscription_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_subscription_maximum_set_gen.py new file mode 100644 index 000000000000..00e762e07f5b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_sites_list_by_subscription_maximum_set_gen.py @@ -0,0 +1,40 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_sites_list_by_subscription_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_sites.list_by_subscription() + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSites_ListBySubscription_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py new file mode 100644 index 000000000000..0d50a786e591 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_software_inventories_controller.get_machine_software_inventory( + resource_group_name="rgmigrate", + site_name="--rW-Hc63Rn--54", + machine_name="36-DV9", + default="default", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSoftwareInventoriesController_GetMachineSoftwareInventory_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_list_by_hyperv_machine_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_list_by_hyperv_machine_maximum_set_gen.py new file mode 100644 index 000000000000..d672c622ca10 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/hyperv_software_inventories_controller_list_by_hyperv_machine_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python hyperv_software_inventories_controller_list_by_hyperv_machine_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.hyperv_software_inventories_controller.list_by_hyperv_machine( + resource_group_name="rgmigrate", + site_name="1UbP6YWKj-Nf", + machine_name="VQ-HROe1jmHEI17-B2-sSVY", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/HypervSoftwareInventoriesController_ListByHypervMachine_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..bcca10526f43 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python iis_web_applications_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.iis_web_applications_controller.get( + resource_group_name="rgmigrate", + site_name="3Q4HX-9-eV7E", + web_app_site_name="5-18J68j1C5D-R1---p0", + web_application_name="72AESeVSD-N2388VI-3K", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/IisWebApplicationsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..ea2dfb2d4b6d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python iis_web_applications_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.iis_web_applications_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="Ybs6T7iJLIc", + web_app_site_name="-UVtIW--VKZ-HtCj-43", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/IisWebApplicationsController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..5db57a7dd7fa --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_applications_controller_update_maximum_set_gen.py @@ -0,0 +1,45 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python iis_web_applications_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.iis_web_applications_controller.update( + resource_group_name="rgmigrate", + site_name="66C6e5--JT", + web_app_site_name="r4NNJ96p4Y2l-3F456A", + web_application_name="53XW34L-5", + body={"properties": {"tags": {"key7171": "bgjjlyxsczvasauqmxjeeuzltv"}}}, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/IisWebApplicationsController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..3de56e48f5cb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python iis_web_servers_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.iis_web_servers_controller.get( + resource_group_name="rgmigrate", + site_name="SR-96OCM--XT36JL", + web_app_site_name="P7--4VIF8-i23", + web_server_name="c-7kNY--G2LK5xF1S1-S--", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/IisWebServersController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..99b4134e51ca --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/iis_web_servers_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python iis_web_servers_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.iis_web_servers_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="--YE-25jFZ15-fcNL8HPAr1", + web_app_site_name="yko5f8", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/IisWebServersController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_deletejob_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_deletejob_maximum_set_gen.py new file mode 100644 index 000000000000..50db33ac04a8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_deletejob_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_get_deletejob_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.get_deletejob( + resource_group_name="rgmigrate", + site_name="AL7-H5T6-0", + job_name="D5X-4-K18-78Y2u", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_GetDeletejob_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_exportjob_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_exportjob_maximum_set_gen.py new file mode 100644 index 000000000000..c407eb40ac5e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_exportjob_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_get_exportjob_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.get_exportjob( + resource_group_name="rgmigrate", + site_name="--j3-58---6JK-5I", + job_name="E-KH", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_GetExportjob_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_importjob_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_importjob_maximum_set_gen.py new file mode 100644 index 000000000000..393936f29eff --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_importjob_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_get_importjob_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.get_importjob( + resource_group_name="rgmigrate", + site_name="hR53N130Y3C87-0Y7", + job_name="o4r5i3iJ3mKCUm7-P-8SZ2P", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_GetImportjob_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..ccd258b51ce6 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.get( + resource_group_name="rgmigrate", + site_name="1bYz4-666S63-PQ08fH", + job_name="6PK797l34W-5q6M", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_by_import_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_by_import_site_maximum_set_gen.py new file mode 100644 index 000000000000..2a87ea627c70 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_by_import_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_list_by_import_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.list_by_import_site( + resource_group_name="rgmigrate", + site_name="-1EV3z1-q8yW35t--KCgU", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_ListByImportSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_deletejobs_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_deletejobs_maximum_set_gen.py new file mode 100644 index 000000000000..4cf3a8dc46cd --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_deletejobs_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_list_deletejobs_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.list_deletejobs( + resource_group_name="rgmigrate", + site_name="A2LE", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_ListDeletejobs_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_exportjobs_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_exportjobs_maximum_set_gen.py new file mode 100644 index 000000000000..35a0c5a5cc59 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_exportjobs_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_list_exportjobs_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.list_exportjobs( + resource_group_name="rgmigrate", + site_name="V-zR1-J3068BzG8", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_ListExportjobs_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_importjobs_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_importjobs_maximum_set_gen.py new file mode 100644 index 000000000000..ba54b9e1667a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_jobs_controller_list_importjobs_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_jobs_controller_list_importjobs_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_jobs_controller.list_importjobs( + resource_group_name="rgmigrate", + site_name="V--Y", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportJobsController_ListImportjobs_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..5b24aa5b0aa7 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_delete_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_machines_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.import_machines_controller.delete( + resource_group_name="rgmigrate", + site_name="f3lKWv8R8", + machine_name="-AKBb-", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportMachinesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..0fc739a25316 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_machines_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_machines_controller.get( + resource_group_name="rgmigrate", + site_name="SZ1-jnNO41CNL-", + machine_name="M6Lh9M6", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportMachinesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_list_by_import_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_list_by_import_site_maximum_set_gen.py new file mode 100644 index 000000000000..f2dbc914c27e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_machines_controller_list_by_import_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_machines_controller_list_by_import_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_machines_controller.list_by_import_site( + resource_group_name="rgmigrate", + site_name="6FM4S4-C-766rL995aG-d", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportMachinesController_ListByImportSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..ddae8e274aff --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_create_maximum_set_gen.py @@ -0,0 +1,47 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_sites_controller.create( + resource_group_name="rgmigrate", + site_name="44GV2SnONP-cPFOQ18Zmn", + body={ + "location": "adfizitxdfwwrkgywpqqupuhkvwie", + "properties": {"discoverySolutionId": "cgwjekybxrlaunxf", "provisioningState": "Created"}, + "tags": {"key592": "pj"}, + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..5b1cdea1c080 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_delete_maximum_set_gen.py @@ -0,0 +1,41 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.import_sites_controller.delete( + resource_group_name="rgmigrate", + site_name="Y-W30297U--37", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_export_uri_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_export_uri_maximum_set_gen.py new file mode 100644 index 000000000000..0baeaf4a7213 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_export_uri_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_export_uri_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_sites_controller.export_uri( + resource_group_name="rgmigrate", + site_name="D8DY8-JKtM11-2", + body={"importType": "AzureMigrateCSV", "jobArmId": "jmzeoqwbofyyeqpnkuhylh", "uri": "xfvwpw"}, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_ExportUri_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..d5a05e1e4ae4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_get_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_sites_controller.get( + resource_group_name="rgmigrate", + site_name="W6Dm3H6", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_import_uri_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_import_uri_maximum_set_gen.py new file mode 100644 index 000000000000..9cf4667304be --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_import_uri_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_import_uri_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_sites_controller.import_uri( + resource_group_name="rgmigrate", + site_name="H8vtJ0EZ8J1-i", + body={"importType": "AzureMigrateCSV", "jobArmId": "jmzeoqwbofyyeqpnkuhylh", "uri": "xfvwpw"}, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_ImportUri_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_resource_group_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_resource_group_maximum_set_gen.py new file mode 100644 index 000000000000..acb268235ac1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_resource_group_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_list_by_resource_group_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_sites_controller.list_by_resource_group( + resource_group_name="rgmigrate", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_ListByResourceGroup_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_subscription_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_subscription_maximum_set_gen.py new file mode 100644 index 000000000000..d3f9f74978b5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_list_by_subscription_maximum_set_gen.py @@ -0,0 +1,40 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_list_by_subscription_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_sites_controller.list_by_subscription() + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_ListBySubscription_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..e6927a9e6b57 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/import_sites_controller_update_maximum_set_gen.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python import_sites_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.import_sites_controller.update( + resource_group_name="rgmigrate", + site_name="873L8-xW5-2k3q3b-", + body={ + "properties": {"discoverySolutionId": "pwqffaxl", "provisioningState": "Created"}, + "tags": {"key6528": "i"}, + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ImportSitesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..6624043f7a37 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python machines_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.machines_controller.get( + resource_group_name="rgmigrate", + site_name="Ihi6-KBnYM7", + machine_name="86TJ63B5Vgd3M26-65QoE", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MachinesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_list_by_vmware_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_list_by_vmware_site_maximum_set_gen.py new file mode 100644 index 000000000000..5bb3f4176ce0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_list_by_vmware_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python machines_controller_list_by_vmware_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.machines_controller.list_by_vmware_site( + resource_group_name="rgmigrate", + site_name="5P-X15WfEPa08", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MachinesController_ListByVmwareSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_start_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_start_maximum_set_gen.py new file mode 100644 index 000000000000..de006c39c8f3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_start_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python machines_controller_start_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.machines_controller.begin_start( + resource_group_name="rgmigrate", + site_name="T85s-9-eBX", + machine_name="OjcCY-5KoYP--uRg8J-NIW", + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MachinesController_Start_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_stop_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_stop_maximum_set_gen.py new file mode 100644 index 000000000000..5ef66471e81d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_stop_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python machines_controller_stop_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.machines_controller.begin_stop( + resource_group_name="rgmigrate", + site_name="-55D03vr8O-I-Q5", + machine_name="t200-F5G-3V6", + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MachinesController_Stop_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..8a017b0101f3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/machines_controller_update_maximum_set_gen.py @@ -0,0 +1,67 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python machines_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.machines_controller.update( + resource_group_name="rgmigrate", + site_name="3N24-", + machine_name="a6-8UYCOk-", + body={ + "properties": { + "allocatedMemoryInMb": 25, + "biosGuid": "kwkxophhlurfasscw", + "biosSerialNumber": "ldjfzv", + "firmware": "raqixesvqmegpansongqje", + "numberOfProcessorCore": 19, + "operatingSystemDetails": { + "osArchitecture": "lxeotrwmodham", + "osName": "udcjzawhhylqauidhrau", + "osType": "s", + "osVersion": "jzblxvpzhmqdpmot", + }, + "productSupportStatus": { + "currentVersion": "dlrvdqalihttfwvnnbskpzikdx", + "esuStatus": "Unknown", + "esuYear": "Unknown", + "supportEndDate": "2024-01-18T11:33:22.395Z", + "supportStatus": "Unknown", + }, + "runAsAccountId": "rxvwltgjjzoqvmrojv", + "tags": {"key8241": "buqgbixvhiqpxs"}, + } + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MachinesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..2aa4e17437a3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_create_maximum_set_gen.py @@ -0,0 +1,53 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.master_sites_controller.begin_create( + resource_group_name="rgmigrate", + site_name="4Y-BY-44e-2fDcKpH-E5", + body={ + "location": "g", + "properties": { + "allowMultipleSites": True, + "customerStorageAccountArmId": "tndgdhvguaaa", + "provisioningState": "Created", + "publicNetworkAccess": "NotSpecified", + "sites": ["lzbawsokqsbzpsocpiwbsjxf"], + }, + "tags": {"key5556": "htomzlbxaiqgcmaaq"}, + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..dc4ec73e7f3d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_delete_maximum_set_gen.py @@ -0,0 +1,41 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.master_sites_controller.delete( + resource_group_name="rgmigrate", + site_name="ICJR0F6JRI", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_error_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_error_summary_maximum_set_gen.py new file mode 100644 index 000000000000..41f20144d7de --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_error_summary_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_controller_error_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.master_sites_controller.error_summary( + resource_group_name="rgmigrate", + site_name="PHm3EiH368dq324-j7N", + body={"applianceName": "gzmhffsahpaizswhjd"}, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesController_ErrorSummary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..4319aace8f2a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_get_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.master_sites_controller.get( + resource_group_name="rgmigrate", + site_name="ATsMK6XNqy-R4h9O8V1T8Jh", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_resource_group_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_resource_group_maximum_set_gen.py new file mode 100644 index 000000000000..d6a8547050d9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_resource_group_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_controller_list_by_resource_group_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.master_sites_controller.list_by_resource_group( + resource_group_name="rgmigrate", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesController_ListByResourceGroup_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_subscription_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_subscription_maximum_set_gen.py new file mode 100644 index 000000000000..3e841c656400 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_list_by_subscription_maximum_set_gen.py @@ -0,0 +1,40 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_controller_list_by_subscription_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.master_sites_controller.list_by_subscription() + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesController_ListBySubscription_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..e9fc8cedea63 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_controller_update_maximum_set_gen.py @@ -0,0 +1,51 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.master_sites_controller.begin_update( + resource_group_name="rgmigrate", + site_name="06OFZ8T18n-8W31Iw-2c46x", + properties={ + "properties": { + "allowMultipleSites": True, + "customerStorageAccountArmId": "opgwuzsibguavfkuxeism", + "publicNetworkAccess": "NotSpecified", + "sites": ["w"], + }, + "tags": {"key1664": "tibpubkzsw"}, + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_operations_status_controller_get_vmware_operation_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_operations_status_controller_get_vmware_operation_status_maximum_set_gen.py new file mode 100644 index 000000000000..0118d047defa --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/master_sites_operations_status_controller_get_vmware_operation_status_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python master_sites_operations_status_controller_get_vmware_operation_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.master_sites_operations_status_controller.get_vmware_operation_status( + resource_group_name="rgmigrate", + site_name="vW2-X", + operation_status_name="t8D5-5X1b-T-p39S", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/MasterSitesOperationsStatusController_GetVmwareOperationStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/operations_list_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/operations_list_maximum_set_gen.py new file mode 100644 index 000000000000..e4f6d4878c58 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/operations_list_maximum_set_gen.py @@ -0,0 +1,40 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python operations_list_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.operations.list() + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/Operations_List_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..7f1e006c225e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_create_maximum_set_gen.py @@ -0,0 +1,54 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python private_endpoint_connection_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.private_endpoint_connection_controller.create( + resource_group_name="rgmigrate", + site_name="-4k3gsPqU5Y3", + pe_connection_name="Q3Y7-B-E80QLei7BHb337-P", + body={ + "properties": { + "privateEndpoint": {}, + "privateLinkServiceConnectionState": { + "actionsRequired": "easeoi", + "description": "zieisbmqzndfwsnt", + "status": "Approved", + }, + "provisioningState": "Created", + } + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/PrivateEndpointConnectionController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..7504639221d5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_delete_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python private_endpoint_connection_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.private_endpoint_connection_controller.delete( + resource_group_name="rgmigrate", + site_name="EX1T446-w-1zPD", + pe_connection_name="M6--1rGB2", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/PrivateEndpointConnectionController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..38552f21dff6 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python private_endpoint_connection_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.private_endpoint_connection_controller.get( + resource_group_name="rgmigrate", + site_name="L8PW6X1249-d-J-POh1N", + pe_connection_name="O21gr1J6pN", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/PrivateEndpointConnectionController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_list_by_master_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_list_by_master_site_maximum_set_gen.py new file mode 100644 index 000000000000..53a5841533e3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_endpoint_connection_controller_list_by_master_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python private_endpoint_connection_controller_list_by_master_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.private_endpoint_connection_controller.list_by_master_site( + resource_group_name="rgmigrate", + site_name="-K7D3lyd0", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/PrivateEndpointConnectionController_ListByMasterSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..6a714830bdeb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python private_link_resources_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.private_link_resources_controller.get( + resource_group_name="rgmigrate", + site_name="8tU8Q--c4-1sP3Bxi", + private_link_resource_name="-G-B5--Y714-SZ2", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/PrivateLinkResourcesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_list_by_master_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_list_by_master_site_maximum_set_gen.py new file mode 100644 index 000000000000..b9e88635ed74 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/private_link_resources_controller_list_by_master_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python private_link_resources_controller_list_by_master_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.private_link_resources_controller.list_by_master_site( + resource_group_name="rgmigrate", + site_name="07575kPYDYD7--", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/PrivateLinkResourcesController_ListByMasterSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..d4a5e9e15248 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python run_as_accounts_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.run_as_accounts_controller.get( + resource_group_name="rgmigrate", + site_name="Oe7P-4-", + account_name="R9sMJ-CKk54Jl15Q", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/RunAsAccountsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_list_by_vmware_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_list_by_vmware_site_maximum_set_gen.py new file mode 100644 index 000000000000..8a3a882b6f96 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/run_as_accounts_controller_list_by_vmware_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python run_as_accounts_controller_list_by_vmware_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.run_as_accounts_controller.list_by_vmware_site( + resource_group_name="rgmigrate", + site_name="1UyI-LBP-8-HB-X0pi-IK2", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/RunAsAccountsController_ListByVmwareSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_client_group_members_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_client_group_members_maximum_set_gen.py new file mode 100644 index 000000000000..b895017ea78c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_client_group_members_maximum_set_gen.py @@ -0,0 +1,50 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_dependency_map_controller_client_group_members_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_dependency_map_controller.begin_client_group_members( + resource_group_name="rgmigrate", + site_name="70h8--WL95QGnCS8H", + map_request={ + "endTime": "2024-01-18T11:33:08.082Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "machineId": "zbsqaglovhnxlroparbdzfs", + "processGroupName": "lawalpbrzupvjeglgvluyjtvijnqcj", + "processName": "ygqxfpuvfsiiotyxveq", + "startTime": "2024-01-18T11:33:08.082Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerDependencyMapController_ClientGroupMembers_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_export_dependencies_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_export_dependencies_maximum_set_gen.py new file mode 100644 index 000000000000..595ecc16b910 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_export_dependencies_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_dependency_map_controller_export_dependencies_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_dependency_map_controller.begin_export_dependencies( + resource_group_name="rgmigrate", + site_name="5Q7SK-Uu5-87-vRN21", + request_body={"endTime": "2024-01-18T11:33:13.789Z", "startTime": "2024-01-18T11:33:13.789Z"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerDependencyMapController_ExportDependencies_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_coarse_map_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_coarse_map_maximum_set_gen.py new file mode 100644 index 000000000000..39351a8f0f6a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_coarse_map_maximum_set_gen.py @@ -0,0 +1,47 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_dependency_map_controller_generate_coarse_map_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_dependency_map_controller.begin_generate_coarse_map( + resource_group_name="rgmigrate", + site_name="--0-0Qy95bgI-457", + map_request={ + "endTime": "2024-01-18T11:33:15.220Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "startTime": "2024-01-18T11:33:15.220Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerDependencyMapController_GenerateCoarseMap_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_detailed_map_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_detailed_map_maximum_set_gen.py new file mode 100644 index 000000000000..39e1b2edc300 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_generate_detailed_map_maximum_set_gen.py @@ -0,0 +1,48 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_dependency_map_controller_generate_detailed_map_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_dependency_map_controller.begin_generate_detailed_map( + resource_group_name="rgmigrate", + site_name="k--P3", + map_request={ + "endTime": "2024-01-18T11:33:15.874Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "machineId": "mqiazddot", + "startTime": "2024-01-18T11:33:15.874Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerDependencyMapController_GenerateDetailedMap_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_server_group_members_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_server_group_members_maximum_set_gen.py new file mode 100644 index 000000000000..31d1bee6b74f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_dependency_map_controller_server_group_members_maximum_set_gen.py @@ -0,0 +1,48 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_dependency_map_controller_server_group_members_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_dependency_map_controller.begin_server_group_members( + resource_group_name="rgmigrate", + site_name="iM05uh-m22r3012Jj7L1", + map_request={ + "endTime": "2024-01-18T11:33:31.988Z", + "filters": {"machineIds": ["nokzgytxl"], "processIds": ["pxbqhupqqlwgta"]}, + "serverPort": 27, + "startTime": "2024-01-18T11:33:31.988Z", + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerDependencyMapController_ServerGroupMembers_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..8c5bf3cba81e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_jobs_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_jobs_controller.get( + resource_group_name="rgmigrate", + site_name="6qN92n-h-3W-13-6-X4", + job_name="aS9X-4S8B41vbY79H5-", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerJobsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_list_by_server_site_resource_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_list_by_server_site_resource_maximum_set_gen.py new file mode 100644 index 000000000000..6dd579631026 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_jobs_controller_list_by_server_site_resource_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_jobs_controller_list_by_server_site_resource_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_jobs_controller.list_by_server_site_resource( + resource_group_name="rgmigrate", + site_name="56-7-", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerJobsController_ListByServerSiteResource_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_operations_status_controller_get_server_site_operations_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_operations_status_controller_get_server_site_operations_status_maximum_set_gen.py new file mode 100644 index 000000000000..b69bf03885c4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_operations_status_controller_get_server_site_operations_status_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_operations_status_controller_get_server_site_operations_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_operations_status_controller.get_server_site_operations_status( + resource_group_name="rgmigrate", + site_name="ZDxb5o", + operation_status_name="N-e--8-", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerOperationsStatusController_GetServerSiteOperationsStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..bbcd8cd9b5f7 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_run_as_accounts_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_run_as_accounts_controller.get( + resource_group_name="rgmigrate", + site_name="-JS187-", + account_name="ER-0o2", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerRunAsAccountsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_list_by_server_site_resource_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_list_by_server_site_resource_maximum_set_gen.py new file mode 100644 index 000000000000..b5388ac867e0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_run_as_accounts_controller_list_by_server_site_resource_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_run_as_accounts_controller_list_by_server_site_resource_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_run_as_accounts_controller.list_by_server_site_resource( + resource_group_name="rgmigrate", + site_name="7-57Q14-4r1q4LZk-FIGMC9Q", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerRunAsAccountsController_ListByServerSiteResource_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..a9a0facc8416 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_create_maximum_set_gen.py @@ -0,0 +1,60 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.begin_create( + resource_group_name="rgmigrate", + site_name="W28q433h1-", + body={ + "location": "rutllgppwqgtiklewhuesohgqb", + "properties": { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "nhhwywytwjbhjtpwxkcxqhqmcrowg", + "discoverySolutionId": "dlsekq", + "provisioningState": "Created", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + }, + "tags": {"key8004": "abecvbhselathbrca"}, + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..125ced2843e2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_delete_maximum_set_gen.py @@ -0,0 +1,41 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.server_sites_controller.delete( + resource_group_name="rgmigrate", + site_name="-1415-x7L-i-iVWkP8", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_export_machine_errors_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_export_machine_errors_maximum_set_gen.py new file mode 100644 index 000000000000..1411fceb994a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_export_machine_errors_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_export_machine_errors_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.begin_export_machine_errors( + resource_group_name="rgmigrate", + site_name="JZTs--1-D64-T8-URG", + body={"properties": {"discoveryScope": "AppsAndRoles"}}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_ExportMachineErrors_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..f5d7edbc1fcc --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_get_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.get( + resource_group_name="rgmigrate", + site_name="9-3-Svn1P2", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_resource_group_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_resource_group_maximum_set_gen.py new file mode 100644 index 000000000000..658b4fc514fa --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_resource_group_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_list_by_resource_group_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.list_by_resource_group( + resource_group_name="rgmigrate", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_ListByResourceGroup_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_subscription_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_subscription_maximum_set_gen.py new file mode 100644 index 000000000000..6df89f5e2bf9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_by_subscription_maximum_set_gen.py @@ -0,0 +1,40 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_list_by_subscription_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.list_by_subscription() + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_ListBySubscription_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_health_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_health_summary_maximum_set_gen.py new file mode 100644 index 000000000000..b2069b74d228 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_list_health_summary_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_list_health_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.list_health_summary( + resource_group_name="rgmigrate", + site_name="l-n--8Q", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_ListHealthSummary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_refresh_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_refresh_site_maximum_set_gen.py new file mode 100644 index 000000000000..fcd889d1f00e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_refresh_site_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_refresh_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.begin_refresh_site( + resource_group_name="rgmigrate", + site_name="SyO---TQ7Es667pB1f0N5V", + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_RefreshSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_summary_maximum_set_gen.py new file mode 100644 index 000000000000..2d13134a7979 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_summary_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.summary( + resource_group_name="rgmigrate", + site_name="1C7", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_Summary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_dependency_map_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_dependency_map_status_maximum_set_gen.py new file mode 100644 index 000000000000..4f74fa484845 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_dependency_map_status_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_update_dependency_map_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.begin_update_dependency_map_status( + resource_group_name="rgmigrate", + site_name="-P1F17KB-Gw2423-tT741U0", + map_request={"machines": [{"isDependencyMapToBeEnabled": True, "machineId": "ysqizqtyaskpprxkqaeyyopln"}]}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_UpdateDependencyMapStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..457c38fa03e8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_maximum_set_gen.py @@ -0,0 +1,58 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.update( + resource_group_name="rgmigrate", + site_name="DlD858-", + body={ + "properties": { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "jhmejalr", + "discoverySolutionId": "glsibqdimzlmgiyvjwxrhuasrptyve", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + }, + "tags": {"key4258": "oxytdpswqvow"}, + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_properties_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_properties_maximum_set_gen.py new file mode 100644 index 000000000000..17c49f088547 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_sites_controller_update_properties_maximum_set_gen.py @@ -0,0 +1,51 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_sites_controller_update_properties_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_sites_controller.begin_update_properties( + resource_group_name="rgmigrate", + site_name="--Y-2W-", + meta_data={ + "value": [ + { + "dependencyMapping": "mehkbjgdktlveubatdamkaawrfmkwc", + "machineArmId": "cbfxbnpaesafshmuirygemyx", + "tags": {"key925": "kvcnqygwaiw"}, + } + ] + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSitesController_UpdateProperties_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py new file mode 100644 index 000000000000..98b707e28b69 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_software_inventories_controller.get_machine_software_inventory( + resource_group_name="rgmigrate", + site_name="HF6V2Ej4V6Z", + machine_name="NL6-267-Pk-5B95", + default="default", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSoftwareInventoriesController_GetMachineSoftwareInventory_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_list_by_server_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_list_by_server_maximum_set_gen.py new file mode 100644 index 000000000000..c4497b3ec28a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/server_software_inventories_controller_list_by_server_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python server_software_inventories_controller_list_by_server_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.server_software_inventories_controller.list_by_server( + resource_group_name="rgmigrate", + site_name="AwZNt4yc", + machine_name="62-T", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServerSoftwareInventoriesController_ListByServer_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_delete_machine_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_delete_machine_maximum_set_gen.py new file mode 100644 index 000000000000..5bc5146c6930 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_delete_machine_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python servers_controller_delete_machine_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.servers_controller.delete_machine( + resource_group_name="rgmigrate", + site_name="-n197127644D28qW744", + machine_name="7-66-qTKM-q73---Q7O", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServersController_DeleteMachine_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_get_machine_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_get_machine_maximum_set_gen.py new file mode 100644 index 000000000000..2f4609f81523 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_get_machine_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python servers_controller_get_machine_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.servers_controller.get_machine( + resource_group_name="rgmigrate", + site_name="1BK6L5S-6F57M7C21-", + machine_name="FIQ6MR5IxK6oV86-hRW--j", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServersController_GetMachine_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_list_by_server_site_resource_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_list_by_server_site_resource_maximum_set_gen.py new file mode 100644 index 000000000000..8c7e45d562d4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_list_by_server_site_resource_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python servers_controller_list_by_server_site_resource_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.servers_controller.list_by_server_site_resource( + resource_group_name="rgmigrate", + site_name="6Jex-VW-8D-", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServersController_ListByServerSiteResource_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_update_machine_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_update_machine_maximum_set_gen.py new file mode 100644 index 000000000000..0d0925b9c32b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/servers_controller_update_machine_maximum_set_gen.py @@ -0,0 +1,90 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python servers_controller_update_machine_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.servers_controller.update_machine( + resource_group_name="rgmigrate", + site_name="2HJb4-f8H1C-J", + machine_name="82xP-5", + body={ + "properties": { + "allocatedMemoryInMb": 8, + "biosGuid": "sgkhcqrumsjuwlnfhctnwus", + "biosSerialNumber": "sbphszczqgmpgcfk", + "disks": [ + { + "diskType": "yvplqohvladrponmtgxpsk", + "generatedId": "qdotapzwauwybroybkfs", + "id": "ttwabuszbdxurwxpqth", + "lun": 17, + "maxSizeInBytes": 24, + "name": "ewdnvthdyaqxjyurxmzndxdllt", + "path": "wjuritjsfzgnxekoiaxhlbrpjvixi", + } + ], + "firmware": "ajvjivcsmfyuajoep", + "fqdn": "enzxkpgnaemrgkovstac", + "hydratedFqdn": "ghu", + "networkAdapters": [ + { + "ipAddressType": "qjdgdeosrivmkeqahzlnwyxara", + "macAddress": "hiuyciojrlzaddb", + "networkName": "bgnmxymooxnsanrzibls", + "nicId": "oh", + } + ], + "numberOfProcessorCore": 20, + "operatingSystemDetails": { + "osArchitecture": "lxeotrwmodham", + "osName": "udcjzawhhylqauidhrau", + "osType": "s", + "osVersion": "jzblxvpzhmqdpmot", + }, + "productSupportStatus": { + "currentVersion": "dlrvdqalihttfwvnnbskpzikdx", + "esuStatus": "Unknown", + "esuYear": "Unknown", + "supportEndDate": "2024-01-18T11:33:22.395Z", + "supportStatus": "Unknown", + }, + "provisioningState": "Created", + "runAsAccountId": "oqfsldxayft", + "tags": {"key36": {}}, + "validationRequired": "dgqelhhko", + } + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/ServersController_UpdateMachine_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..aec74ae5685b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_create_maximum_set_gen.py @@ -0,0 +1,60 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.create( + resource_group_name="rgmigrate", + site_name="3w-E-i5L", + body={ + "location": "wcccoelallobfu", + "properties": { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "nhhwywytwjbhjtpwxkcxqhqmcrowg", + "discoverySolutionId": "dlsekq", + "provisioningState": "Created", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + }, + "tags": {"key2178": "gfjjytrnctgvmi"}, + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..1ba1e69217c3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_delete_maximum_set_gen.py @@ -0,0 +1,41 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.sites_controller.delete( + resource_group_name="rgmigrate", + site_name="PiW11F6QAg-", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machine_errors_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machine_errors_maximum_set_gen.py new file mode 100644 index 000000000000..b209f521a6fd --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machine_errors_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_export_machine_errors_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.begin_export_machine_errors( + resource_group_name="rgmigrate", + site_name="9QD-VAV18O1Sb-unBV30X", + body={"properties": {"discoveryScope": "AppsAndRoles"}}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_ExportMachineErrors_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machines_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machines_maximum_set_gen.py new file mode 100644 index 000000000000..55afc741d14b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_export_machines_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_export_machines_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.begin_export_machines( + resource_group_name="rgmigrate", + site_name="O45c6F82W4U6-0-bO", + body={"filter": "mgtewmbxropivmgndxrbufgxpmo"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_ExportMachines_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..cc9ac2a1c147 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_get_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.get( + resource_group_name="rgmigrate", + site_name="3-64--12P2Q-7P9-2-F-", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_resource_group_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_resource_group_maximum_set_gen.py new file mode 100644 index 000000000000..ff143d148f74 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_resource_group_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_list_by_resource_group_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.list_by_resource_group( + resource_group_name="rgmigrate", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_ListByResourceGroup_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_subscription_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_subscription_maximum_set_gen.py new file mode 100644 index 000000000000..a608c308c757 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_by_subscription_maximum_set_gen.py @@ -0,0 +1,40 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_list_by_subscription_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.list_by_subscription() + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_ListBySubscription_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_health_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_health_summary_maximum_set_gen.py new file mode 100644 index 000000000000..7abac80d081e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_list_health_summary_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_list_health_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.list_health_summary( + resource_group_name="rgmigrate", + site_name="-qBx", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_ListHealthSummary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_summary_maximum_set_gen.py new file mode 100644 index 000000000000..19bff812955a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_summary_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.summary( + resource_group_name="rgmigrate", + site_name="9dW7AR2-W98852IN2---GRU", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_Summary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..7041601673c9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sites_controller_update_maximum_set_gen.py @@ -0,0 +1,59 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sites_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sites_controller.update( + resource_group_name="rgmigrate", + site_name="6-5N8-b81", + body={ + "properties": { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "lsfkebcliutqeevmn", + "discoverySolutionId": "czzqa", + "provisioningState": "Created", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + }, + "tags": {"key1253": "vtfixy"}, + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SitesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..6823ee556ff5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_availability_groups_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_availability_groups_controller.get( + resource_group_name="rgmigrate", + site_name="5-7W6U24O04E9oKW-Uu3", + sql_site_name="51HR84W---1", + sql_availability_group_name="QH9N-7YGCT160", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlAvailabilityGroupsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_list_by_sql_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_list_by_sql_site_maximum_set_gen.py new file mode 100644 index 000000000000..7a5cf899483d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_availability_groups_controller_list_by_sql_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_availability_groups_controller_list_by_sql_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_availability_groups_controller.list_by_sql_site( + resource_group_name="rgmigrate", + site_name="A6JL", + sql_site_name="Iy--O", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlAvailabilityGroupsController_ListBySqlSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..54382d4d5de4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_databases_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_databases_controller.get( + resource_group_name="rgmigrate", + site_name="Vg681-46pU7-1", + sql_site_name="-vO6EW-2UD", + sql_database_name="1k-11553Ir-7gV6-3T-76", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlDatabasesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_list_by_sql_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_list_by_sql_site_maximum_set_gen.py new file mode 100644 index 000000000000..8df28a1589d4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_databases_controller_list_by_sql_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_databases_controller_list_by_sql_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_databases_controller.list_by_sql_site( + resource_group_name="rgmigrate", + site_name="-X36w63R3356-8q1HI07", + sql_site_name="4----63-4472-12N-", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlDatabasesController_ListBySqlSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..5d443c64a33c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_create_maximum_set_gen.py @@ -0,0 +1,45 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_discovery_site_data_source_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_discovery_site_data_source_controller.begin_create( + resource_group_name="rgmigrate", + site_name="v-MwBCpvV-", + sql_site_name="NO-8-W75XK5-4Xfs54I-n", + discovery_site_data_source_name="jN--823UiQ", + body={"properties": {"discoverySiteId": "ndmdh", "provisioningState": "Created"}}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlDiscoverySiteDataSourceController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..310b1ab6ac0f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_delete_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_discovery_site_data_source_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.sql_discovery_site_data_source_controller.delete( + resource_group_name="rgmigrate", + site_name="5-6-SmU52A-w616N1uV", + sql_site_name="t1-s8R--f6T34L", + discovery_site_data_source_name="Vr0-1E23-O4-IV1w6", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlDiscoverySiteDataSourceController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..c0564a672860 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_discovery_site_data_source_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_discovery_site_data_source_controller.get( + resource_group_name="rgmigrate", + site_name="34K-r--7e443uLdyG-S-M3-Y", + sql_site_name="3-Q4-J7-", + discovery_site_data_source_name="2hyjS68CSU-Se4hiPNk2", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlDiscoverySiteDataSourceController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_list_by_sql_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_list_by_sql_site_maximum_set_gen.py new file mode 100644 index 000000000000..2084728e0650 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_discovery_site_data_source_controller_list_by_sql_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_discovery_site_data_source_controller_list_by_sql_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_discovery_site_data_source_controller.list_by_sql_site( + resource_group_name="rgmigrate", + site_name="1L3-06Ug", + sql_site_name="cvn6667m", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlDiscoverySiteDataSourceController_ListBySqlSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..111a59f8850f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_jobs_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_jobs_controller.get( + resource_group_name="rgmigrate", + site_name="-8F1-jW27Q-4", + sql_site_name="c36-J78odt-h69b3", + job_name="-cf5aU", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlJobsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_list_by_sql_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_list_by_sql_site_maximum_set_gen.py new file mode 100644 index 000000000000..962d5a7ef747 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_jobs_controller_list_by_sql_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_jobs_controller_list_by_sql_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_jobs_controller.list_by_sql_site( + resource_group_name="rgmigrate", + site_name="H44", + sql_site_name="646B--g", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlJobsController_ListBySqlSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_operations_status_controller_get_sql_operation_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_operations_status_controller_get_sql_operation_status_maximum_set_gen.py new file mode 100644 index 000000000000..f2889d14fbc9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_operations_status_controller_get_sql_operation_status_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_operations_status_controller_get_sql_operation_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_operations_status_controller.get_sql_operation_status( + resource_group_name="rgmigrate", + site_name="0uJYYIVL-NX23x33I31", + sql_site_name="Y4l-bwV847fX5--6---VP5D", + operation_status_name="-2z43", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlOperationsStatusController_GetSqlOperationStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..d0bc6214e3c4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_run_as_accounts_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_run_as_accounts_controller.get( + resource_group_name="rgmigrate", + site_name="7s32E4yPhN5jCPW-", + sql_site_name="1-b4--1--5JH61Q-t94991W", + account_name="Cu--68", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlRunAsAccountsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_list_by_sql_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_list_by_sql_site_maximum_set_gen.py new file mode 100644 index 000000000000..848682b889af --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_run_as_accounts_controller_list_by_sql_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_run_as_accounts_controller_list_by_sql_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_run_as_accounts_controller.list_by_sql_site( + resource_group_name="rgmigrate", + site_name="3I4yO63MT--06G33--OJjO7", + sql_site_name="-gLI1---DX1J-V1j8L4P", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlRunAsAccountsController_ListBySqlSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..cc82e37480d0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_servers_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_servers_controller.get( + resource_group_name="rgmigrate", + site_name="d1-Q3-oD-Z", + sql_site_name="r436C6-04Rz7Cz-P6f", + sql_server_name="Q314T-9n3E45", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlServersController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_list_by_sql_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_list_by_sql_site_maximum_set_gen.py new file mode 100644 index 000000000000..6202e1962114 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_list_by_sql_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_servers_controller_list_by_sql_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_servers_controller.list_by_sql_site( + resource_group_name="rgmigrate", + site_name="-l-RL-C", + sql_site_name="443s9KUxB35Eod363NW8qiZB", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlServersController_ListBySqlSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..ead17e12a0a1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_servers_controller_update_maximum_set_gen.py @@ -0,0 +1,88 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_servers_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_servers_controller.update( + resource_group_name="rgmigrate", + site_name="1Jj-SVw-9c-I67KdU", + sql_site_name="4-WOG1O-RF0Lm22-k61", + sql_server_name="8wB3t-37", + body={ + "properties": { + "createdTimestamp": "zxzgzldsoouqabmjraprijr", + "edition": "xzyqnupboa", + "engineEdition": "bddqdimixvuo", + "hostName": "wjyokwswayhcfvcvgcxzsq", + "hydratedRunAsAccountId": "dhfgeqy", + "hyperthreadRatio": 29, + "isClustered": True, + "isDeleted": True, + "isHighAvailabilityEnabled": True, + "logicalCpuCount": 15, + "maxServerMemoryInUseInMb": 8, + "numOfLogins": 14, + "numberOfAgDatabases": 26, + "numberOfUserDatabases": 24, + "physicalCpuCount": 22, + "portNumber": 8, + "productSupportStatus": { + "currentVersion": "dlrvdqalihttfwvnnbskpzikdx", + "esuStatus": "Unknown", + "esuYear": "Unknown", + "supportEndDate": "2024-01-18T11:33:22.395Z", + "supportStatus": "Unknown", + }, + "provisioningState": "Created", + "runAsAccountId": "wo", + "sqlFciProperties": { + "isMultiSubnet": True, + "networkName": "rmtviuhjzclchzlbxmilstiyyeiog", + "sharedDiskCount": 20, + "state": "Unknown", + }, + "sqlServerName": "ixefej", + "sqlStartTime": "2024-01-18T11:34:43.726Z", + "status": "Unknown", + "sumOfUserDatabasesSizeInMb": 20, + "tags": {"key9912": {}}, + "tempDbSizeInMb": 28, + "updatedTimestamp": "aytchtagfrol", + "version": "laneupphflqwaoeoqnjt", + "visibleOnlineCoreCount": 14, + } + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlServersController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..ac3f2affa842 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_create_maximum_set_gen.py @@ -0,0 +1,63 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.create( + resource_group_name="rgmigrate", + site_name="H9S6I-U9UI5-J1", + sql_site_name="6G-FTRBDS-tN1-C6", + body={ + "properties": { + "discoveryScenario": "Migrate", + "provisioningState": "Created", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "jqv", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + } + ], + } + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..6af54b55a931 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_delete_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.sql_sites_controller.delete( + resource_group_name="rgmigrate", + site_name="31sL315xu-r-I--57", + sql_site_name="B1L--Oc88B-", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_error_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_error_summary_maximum_set_gen.py new file mode 100644 index 000000000000..0a76eeafad77 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_error_summary_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_error_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.error_summary( + resource_group_name="rgmigrate", + site_name="-UuOu2u3503N4u-", + sql_site_name="d9I452ZS--16", + body={"applianceName": "gzmhffsahpaizswhjd"}, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_ErrorSummary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_export_sql_servers_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_export_sql_servers_maximum_set_gen.py new file mode 100644 index 000000000000..840f8ad8ccb6 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_export_sql_servers_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_export_sql_servers_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.begin_export_sql_servers( + resource_group_name="rgmigrate", + site_name="-N5360H5--qBr8Is-w", + sql_site_name="3Wl-Ci0bT2WL2fV-W0w", + body={"applianceName": "tkp", "filter": "nqbzryzcxcedlytkm"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_ExportSqlServers_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..e11fdca92a07 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.get( + resource_group_name="rgmigrate", + site_name="23-JjGjG2BVC362-8Nxd1", + sql_site_name="sU5", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_list_by_master_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_list_by_master_site_maximum_set_gen.py new file mode 100644 index 000000000000..7372e49cdd78 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_list_by_master_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_list_by_master_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.list_by_master_site( + resource_group_name="rgmigrate", + site_name="Xl-7X131NVGP1-12IT6", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_ListByMasterSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_refresh_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_refresh_maximum_set_gen.py new file mode 100644 index 000000000000..99e5876b2913 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_refresh_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_refresh_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.begin_refresh( + resource_group_name="rgmigrate", + site_name="581DI4-", + sql_site_name="-O0q2016L0-", + body={"applianceName": "juzzwzjhcxdcpjypwshdccowqhrlw"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_Refresh_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_summary_maximum_set_gen.py new file mode 100644 index 000000000000..a40420728a03 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_summary_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.summary( + resource_group_name="rgmigrate", + site_name="--5---", + sql_site_name="P-S-pYLZ-", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_Summary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..1c82d966f866 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/sql_sites_controller_update_maximum_set_gen.py @@ -0,0 +1,62 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python sql_sites_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.sql_sites_controller.begin_update( + resource_group_name="rgmigrate", + site_name="3d2XGeA6-15zBUX2QT-", + sql_site_name="CK78r-15Q8DjE5-Q--RDC2N", + properties={ + "properties": { + "discoveryScenario": "Migrate", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "jqv", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + } + ], + } + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/SqlSitesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..9ea1691a6275 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python tomcat_web_applications_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.tomcat_web_applications_controller.get( + resource_group_name="rgmigrate", + site_name="-I-Se69--5YB51-7N1uuad-", + web_app_site_name="7EFP5KH-27TE", + web_application_name="272Y4n4m1LE7-VI7T6e9", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/TomcatWebApplicationsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..cbf047702468 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_applications_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python tomcat_web_applications_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.tomcat_web_applications_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="O24L2C-110-07", + web_app_site_name="Q42tG-m", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/TomcatWebApplicationsController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..025defd62117 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python tomcat_web_servers_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.tomcat_web_servers_controller.get( + resource_group_name="rgmigrate", + site_name="k-6Va3b", + web_app_site_name="K-G3Z----hB5iCI-Vrk2-1e", + web_server_name="XTG25CRK3", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/TomcatWebServersController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..0177254364c1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/tomcat_web_servers_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python tomcat_web_servers_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.tomcat_web_servers_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="CiWt8C-g2L6fUZ3K3-h", + web_app_site_name="8l-C33-rp-V-4-PS", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/TomcatWebServersController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..4822255b973d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_create_maximum_set_gen.py @@ -0,0 +1,52 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vcenter_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vcenter_controller.begin_create( + resource_group_name="rgmigrate", + site_name="6T42r-99CFg-44-gFP", + vcenter_name="23LDJ0XB7UT-L", + body={ + "properties": { + "fqdn": "fuclodcnofphkcrnex", + "friendlyName": "qvycnab", + "port": "bdospcacndrchuqluozmwrvetxul", + "provisioningState": "Created", + "runAsAccountId": "pqbieohxxmfri", + } + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VcenterController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..e9ccf3c050b7 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_delete_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vcenter_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.vcenter_controller.delete( + resource_group_name="rgmigrate", + site_name="--7S0K36vH3", + vcenter_name="-A-J8R7E-g0", + ) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VcenterController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..a523e9d51b2d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vcenter_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vcenter_controller.get( + resource_group_name="rgmigrate", + site_name="-C6Jvk-M633YBJ-7", + vcenter_name="LN54", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VcenterController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_list_by_vmware_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_list_by_vmware_site_maximum_set_gen.py new file mode 100644 index 000000000000..0d24ad4d305e --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vcenter_controller_list_by_vmware_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vcenter_controller_list_by_vmware_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vcenter_controller.list_by_vmware_site( + resource_group_name="rgmigrate", + site_name="OCJ-c8qUW", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VcenterController_ListByVmwareSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..49ee8fa11c8f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_host_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_host_controller.get( + resource_group_name="rgmigrate", + site_name="--nML9b2DN", + host_name="22DhjSY-hN--KES1", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwareHostController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_list_by_vmware_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_list_by_vmware_site_maximum_set_gen.py new file mode 100644 index 000000000000..c7125497686f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_host_controller_list_by_vmware_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_host_controller_list_by_vmware_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_host_controller.list_by_vmware_site( + resource_group_name="rgmigrate", + site_name="xK6Lq", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwareHostController_ListByVmwareSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_operations_status_get_vmware_operation_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_operations_status_get_vmware_operation_status_maximum_set_gen.py new file mode 100644 index 000000000000..0be1909ebdd4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_operations_status_get_vmware_operation_status_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_operations_status_get_vmware_operation_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_operations_status.get_vmware_operation_status( + resource_group_name="rgmigrate", + site_name="KQc-", + operation_status_name="-OFD3s3B7", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwareOperationsStatus_GetVmwareOperationStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_dependency_map_status_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_dependency_map_status_maximum_set_gen.py new file mode 100644 index 000000000000..43e0259e2010 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_dependency_map_status_maximum_set_gen.py @@ -0,0 +1,45 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_properties_controller_update_dependency_map_status_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_properties_controller.begin_update_dependency_map_status( + resource_group_name="rgmigrate", + site_name="28-HV9tQM1", + update_machine_dep_map_status={ + "machines": [{"isDependencyMapToBeEnabled": True, "machineId": "ysqizqtyaskpprxkqaeyyopln"}] + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwarePropertiesController_UpdateDependencyMapStatus_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_properties_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_properties_maximum_set_gen.py new file mode 100644 index 000000000000..9cbe6b9fa1a5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_properties_maximum_set_gen.py @@ -0,0 +1,51 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_properties_controller_update_properties_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_properties_controller.begin_update_properties( + resource_group_name="rgmigrate", + site_name="6KX3-k6-DI5Q8F4-DL", + meta_data={ + "value": [ + { + "dependencyMapping": "mehkbjgdktlveubatdamkaawrfmkwc", + "machineArmId": "cbfxbnpaesafshmuirygemyx", + "tags": {"key925": "kvcnqygwaiw"}, + } + ] + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwarePropertiesController_UpdateProperties_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_tags_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_tags_maximum_set_gen.py new file mode 100644 index 000000000000..f2baa325e979 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_properties_controller_update_tags_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_properties_controller_update_tags_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_properties_controller.begin_update_tags( + resource_group_name="rgmigrate", + site_name="m4-x-6-0n5qM33WWM5", + update_machine_tags={"machines": [{"machineId": "qunrgqsyhzqwugvgr", "tags": {"key7333": "horsaf"}}]}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwarePropertiesController_UpdateTags_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py new file mode 100644 index 000000000000..875a6c980256 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_software_inventories_controller_get_machine_software_inventory_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_software_inventories_controller.get_machine_software_inventory( + resource_group_name="rgmigrate", + site_name="-F6ViTRve-A7173v3", + machine_name="62Ly", + default="default", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwareSoftwareInventoriesController_GetMachineSoftwareInventory_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_list_by_machine_resource_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_list_by_machine_resource_maximum_set_gen.py new file mode 100644 index 000000000000..c5e1c89bde3a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/vmware_software_inventories_controller_list_by_machine_resource_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python vmware_software_inventories_controller_list_by_machine_resource_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.vmware_software_inventories_controller.list_by_machine_resource( + resource_group_name="rgmigrate", + site_name="F-piD", + machine_name="4pY-V1i-S-Q-tJ-CW", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/VmwareSoftwareInventoriesController_ListByMachineResource_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..20e3391c0f84 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_create_maximum_set_gen.py @@ -0,0 +1,45 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_discovery_site_data_sources_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_discovery_site_data_sources_controller.begin_create( + resource_group_name="rgmigrate", + site_name="3363G4K-7-XWJ7384S", + web_app_site_name="42883RRtL4-CE155--u65v-", + discovery_site_data_source_name="XJ-V535YCO5L3PIWdB", + body={"properties": {"discoverySiteId": "gsqqojvnjtbdabo", "provisioningState": "Created"}}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppDiscoverySiteDataSourcesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..5d48a653389b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_delete_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_discovery_site_data_sources_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.web_app_discovery_site_data_sources_controller.begin_delete( + resource_group_name="rgmigrate", + site_name="-1B---62T", + web_app_site_name="o23W", + discovery_site_data_source_name="3DZE3-3m7E3-3Mk-", + ).result() + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppDiscoverySiteDataSourcesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..cad08c373f7b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_discovery_site_data_sources_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_discovery_site_data_sources_controller.get( + resource_group_name="rgmigrate", + site_name="-24n-4KK8rOJOa0", + web_app_site_name="RS01YYEI-KQ6t-pke", + discovery_site_data_source_name="bS4N-Ve77I3", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppDiscoverySiteDataSourcesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..79bc42eb819b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_discovery_site_data_sources_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_discovery_site_data_sources_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_discovery_site_data_sources_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="5-ECR-5-2", + web_app_site_name="8F7p-t-SVxV", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppDiscoverySiteDataSourcesController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..541ce881dbe5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_extended_machines_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_extended_machines_controller.get( + resource_group_name="rgmigrate", + site_name="6vhV3", + web_app_site_name="3Q4fch0Jg-ZDO4Q-6Fr5A683", + extended_machine_name="76-vIS678Y3GPD2-GW6X86", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppExtendedMachinesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..b651b4dcad29 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_extended_machines_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_extended_machines_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_extended_machines_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="4UInM", + web_app_site_name="g6I16C15-2", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppExtendedMachinesController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_properties_controller_update_properties_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_properties_controller_update_properties_maximum_set_gen.py new file mode 100644 index 000000000000..9a8693c7ae71 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_properties_controller_update_properties_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_properties_controller_update_properties_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_properties_controller.begin_update_properties( + resource_group_name="rgmigrate", + site_name="09u", + web_app_site_name="-K-X85Go-QY1INX-H34", + body={"webApps": [{"tags": {"key7358": "wfxubzp"}, "webAppArmId": "ytmkdtigfuip"}]}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppPropertiesController_UpdateProperties_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..a7b8adcabee5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_get_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_run_as_accounts_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_run_as_accounts_controller.get( + resource_group_name="rgmigrate", + site_name="5V-AedeyXMe23Fd5-s97VP", + web_app_site_name="S28O-LQJ", + account_name="YGHg41E44", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppRunAsAccountsController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..8d1f576b55c2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_run_as_accounts_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_run_as_accounts_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_run_as_accounts_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="5C7-95863tPMCn", + web_app_site_name="2QX1-0-1442ND-a-6", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppRunAsAccountsController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_create_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_create_maximum_set_gen.py new file mode 100644 index 000000000000..1ea4c8c41d0f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_create_maximum_set_gen.py @@ -0,0 +1,63 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_create_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.create( + resource_group_name="rgmigrate", + site_name="57NK0P6SR1J11IDI-JyJ9", + web_app_site_name="5-oF2-s1LNU0-NQ820gA4D4t", + body={ + "properties": { + "discoveryScenario": "Migrate", + "provisioningState": "Created", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "jqv", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + } + ], + } + }, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_Create_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_delete_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_delete_maximum_set_gen.py new file mode 100644 index 000000000000..090fc1f2592b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_delete_maximum_set_gen.py @@ -0,0 +1,42 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_delete_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + client.web_app_sites_controller.begin_delete( + resource_group_name="rgmigrate", + site_name="O27S7Y1p6--4y-8ZuX-Rm51", + web_app_site_name="R9K0M3-5DM25-LL--9", + ).result() + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_Delete_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_error_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_error_summary_maximum_set_gen.py new file mode 100644 index 000000000000..1e80ee7a4240 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_error_summary_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_error_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.error_summary( + resource_group_name="rgmigrate", + site_name="sJ7O7OpAw5-g-E-AB3", + web_app_site_name="X-7-3V9M6tG621bM", + body={"applianceName": "gzmhffsahpaizswhjd"}, + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_ErrorSummary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_export_inventory_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_export_inventory_maximum_set_gen.py new file mode 100644 index 000000000000..66771b105896 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_export_inventory_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_export_inventory_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.begin_export_inventory( + resource_group_name="rgmigrate", + site_name="z6IFGJv4C4pIJ-4", + web_app_site_name="V5UoS68X-X", + body={"filter": "ujgnemjnwfrgzmnwx"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_ExportInventory_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_get_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_get_maximum_set_gen.py new file mode 100644 index 000000000000..d5c6d5101e84 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_get_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_get_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.get( + resource_group_name="rgmigrate", + site_name="N82", + web_app_site_name="B0Mt17X-xB71D22F", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_Get_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_list_by_master_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_list_by_master_site_maximum_set_gen.py new file mode 100644 index 000000000000..0c768d523679 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_list_by_master_site_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_list_by_master_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.list_by_master_site( + resource_group_name="rgmigrate", + site_name="RNY-Gz6PP", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_ListByMasterSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_refresh_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_refresh_maximum_set_gen.py new file mode 100644 index 000000000000..b77c83509578 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_refresh_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_refresh_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.begin_refresh( + resource_group_name="rgmigrate", + site_name="8xJ911PM-8G7", + web_app_site_name="YQ5246", + body={"applianceName": "yovbznk"}, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_Refresh_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_summary_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_summary_maximum_set_gen.py new file mode 100644 index 000000000000..812b67539f55 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_summary_maximum_set_gen.py @@ -0,0 +1,43 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_summary_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.summary( + resource_group_name="rgmigrate", + site_name="---3nO", + web_app_site_name="CXDHwGq-M1BS", + ) + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_Summary_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_update_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_update_maximum_set_gen.py new file mode 100644 index 000000000000..fce7453ba0b5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_app_sites_controller_update_maximum_set_gen.py @@ -0,0 +1,62 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_app_sites_controller_update_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_app_sites_controller.begin_update( + resource_group_name="rgmigrate", + site_name="W03x2---1q9-poH3021XK-8", + web_app_site_name="8-8UX--E--B-0575gI55p", + properties={ + "properties": { + "discoveryScenario": "Migrate", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": {"keyVaultId": "chqtsrpahfayfuqrusmbakvgdqny", "keyVaultUri": "w"}, + "applianceName": "jqv", + "servicePrincipalIdentityDetails": { + "aadAuthority": "pykd", + "applicationId": "lg", + "audience": "emwqxwwqqzwk", + "objectId": "etkbysdzluxvpxkciooxjjmvl", + "rawCertData": "kewdmdjrt", + "tenantId": "aqvsxunpdlrxclnpudkykycsdf", + }, + } + ], + } + }, + ).result() + print(response) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebAppSitesController_Update_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_applications_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_applications_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..35fc745987ae --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_applications_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_applications_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_applications_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="-611-z", + web_app_site_name="R-5y3W52-2V", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebApplicationsController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_samples/web_servers_controller_list_by_web_app_site_maximum_set_gen.py b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_servers_controller_list_by_web_app_site_maximum_set_gen.py new file mode 100644 index 000000000000..8917eb71a397 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_samples/web_servers_controller_list_by_web_app_site_maximum_set_gen.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-migrate +# USAGE + python web_servers_controller_list_by_web_app_site_maximum_set_gen.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 = AzureMigrateDiscoveryService( + credential=DefaultAzureCredential(), + subscription_id="BCE48E0E-6240-491D-A4D4-4306580B1B82", + ) + + response = client.web_servers_controller.list_by_web_app_site( + resource_group_name="rgmigrate", + site_name="46I23G-V8AOSK3", + web_app_site_name="n137S-OG8F--2w", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/migrate/resource-manager/Microsoft.OffAzure/preview/2023-10-01-preview/examples/WebServersController_ListByWebAppSite_MaximumSet_Gen.json +if __name__ == "__main__": + main() diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/conftest.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/conftest.py new file mode 100644 index 000000000000..145faa956281 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/conftest.py @@ -0,0 +1,47 @@ +# 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 os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# aovid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + azuremigratediscoveryservice_subscription_id = os.environ.get( + "AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + azuremigratediscoveryservice_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + azuremigratediscoveryservice_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + azuremigratediscoveryservice_client_secret = os.environ.get( + "AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=azuremigratediscoveryservice_subscription_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=azuremigratediscoveryservice_tenant_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=azuremigratediscoveryservice_client_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=azuremigratediscoveryservice_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;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations.py new file mode 100644 index 000000000000..aef0bcacd8a8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations.py @@ -0,0 +1,105 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceDependencyMapControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_client_group_members(self, resource_group): + response = self.client.dependency_map_controller.begin_client_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "processGroupName": "str", + "processName": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_export_dependencies(self, resource_group): + response = self.client.dependency_map_controller.begin_export_dependencies( + resource_group_name=resource_group.name, + site_name="str", + request_body={"endTime": "2020-02-20 00:00:00", "startTime": "2020-02-20 00:00:00"}, + api_version="2023-10-01-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_generate_coarse_map(self, resource_group): + response = self.client.dependency_map_controller.begin_generate_coarse_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_generate_detailed_map(self, resource_group): + response = self.client.dependency_map_controller.begin_generate_detailed_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_server_group_members(self, resource_group): + response = self.client.dependency_map_controller.begin_server_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "serverPort": 0, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations_async.py new file mode 100644 index 000000000000..0e2a04f83d53 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_dependency_map_controller_operations_async.py @@ -0,0 +1,116 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceDependencyMapControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_client_group_members(self, resource_group): + response = await ( + await self.client.dependency_map_controller.begin_client_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "processGroupName": "str", + "processName": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_export_dependencies(self, resource_group): + response = await ( + await self.client.dependency_map_controller.begin_export_dependencies( + resource_group_name=resource_group.name, + site_name="str", + request_body={"endTime": "2020-02-20 00:00:00", "startTime": "2020-02-20 00:00:00"}, + api_version="2023-10-01-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_generate_coarse_map(self, resource_group): + response = await ( + await self.client.dependency_map_controller.begin_generate_coarse_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_generate_detailed_map(self, resource_group): + response = await ( + await self.client.dependency_map_controller.begin_generate_detailed_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_server_group_members(self, resource_group): + response = await ( + await self.client.dependency_map_controller.begin_server_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "serverPort": 0, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations.py new file mode 100644 index 000000000000..fa516eaf63bb --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations.py @@ -0,0 +1,110 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervClusterControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_cluster_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_cluster(self, resource_group): + response = self.client.hyperv_cluster_controller.get_cluster( + resource_group_name=resource_group.name, + site_name="str", + cluster_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_create_cluster(self, resource_group): + response = self.client.hyperv_cluster_controller.begin_create_cluster( + resource_group_name=resource_group.name, + site_name="str", + cluster_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "createdTimestamp": "str", + "errors": [ + { + "applianceName": "str", + "code": "str", + "discoveryScope": "str", + "id": 0, + "message": "str", + "messageParameters": {"str": "str"}, + "possibleCauses": "str", + "recommendedAction": "str", + "runAsAccountId": "str", + "severity": "str", + "source": "str", + "summaryMessage": "str", + "updatedTimeStamp": "2020-02-20 00:00:00", + } + ], + "fqdn": "str", + "functionalLevel": 0, + "hostFqdnList": ["str"], + "provisioningState": "str", + "runAsAccountId": "str", + "status": "str", + "updatedTimestamp": "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", + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = self.client.hyperv_cluster_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + cluster_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations_async.py new file mode 100644 index 000000000000..be635d32014b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_cluster_controller_operations_async.py @@ -0,0 +1,113 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervClusterControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_cluster_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_cluster(self, resource_group): + response = await self.client.hyperv_cluster_controller.get_cluster( + resource_group_name=resource_group.name, + site_name="str", + cluster_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_create_cluster(self, resource_group): + response = await ( + await self.client.hyperv_cluster_controller.begin_create_cluster( + resource_group_name=resource_group.name, + site_name="str", + cluster_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "createdTimestamp": "str", + "errors": [ + { + "applianceName": "str", + "code": "str", + "discoveryScope": "str", + "id": 0, + "message": "str", + "messageParameters": {"str": "str"}, + "possibleCauses": "str", + "recommendedAction": "str", + "runAsAccountId": "str", + "severity": "str", + "source": "str", + "summaryMessage": "str", + "updatedTimeStamp": "2020-02-20 00:00:00", + } + ], + "fqdn": "str", + "functionalLevel": 0, + "hostFqdnList": ["str"], + "provisioningState": "str", + "runAsAccountId": "str", + "status": "str", + "updatedTimestamp": "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", + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = await self.client.hyperv_cluster_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + cluster_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations.py new file mode 100644 index 000000000000..b984b6121d5b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations.py @@ -0,0 +1,118 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervDependencyMapControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_client_group_members(self, resource_group): + response = self.client.hyperv_dependency_map_controller.begin_client_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "processGroupName": "str", + "processName": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_export_dependencies(self, resource_group): + response = self.client.hyperv_dependency_map_controller.begin_export_dependencies( + resource_group_name=resource_group.name, + site_name="str", + request_body={"endTime": "2020-02-20 00:00:00", "startTime": "2020-02-20 00:00:00"}, + api_version="2023-10-01-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_generate_coarse_map(self, resource_group): + response = self.client.hyperv_dependency_map_controller.begin_generate_coarse_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_generate_detailed_map(self, resource_group): + response = self.client.hyperv_dependency_map_controller.begin_generate_detailed_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_server_group_members(self, resource_group): + response = self.client.hyperv_dependency_map_controller.begin_server_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "serverPort": 0, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_dependency_map_status(self, resource_group): + response = self.client.hyperv_dependency_map_controller.begin_update_dependency_map_status( + resource_group_name=resource_group.name, + site_name="str", + map_request={"machines": [{"isDependencyMapToBeEnabled": bool, "machineId": "str"}]}, + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations_async.py new file mode 100644 index 000000000000..2faa6c61c3df --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_dependency_map_controller_operations_async.py @@ -0,0 +1,131 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervDependencyMapControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_client_group_members(self, resource_group): + response = await ( + await self.client.hyperv_dependency_map_controller.begin_client_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "processGroupName": "str", + "processName": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_export_dependencies(self, resource_group): + response = await ( + await self.client.hyperv_dependency_map_controller.begin_export_dependencies( + resource_group_name=resource_group.name, + site_name="str", + request_body={"endTime": "2020-02-20 00:00:00", "startTime": "2020-02-20 00:00:00"}, + api_version="2023-10-01-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_generate_coarse_map(self, resource_group): + response = await ( + await self.client.hyperv_dependency_map_controller.begin_generate_coarse_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_generate_detailed_map(self, resource_group): + response = await ( + await self.client.hyperv_dependency_map_controller.begin_generate_detailed_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_server_group_members(self, resource_group): + response = await ( + await self.client.hyperv_dependency_map_controller.begin_server_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "serverPort": 0, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_dependency_map_status(self, resource_group): + response = await ( + await self.client.hyperv_dependency_map_controller.begin_update_dependency_map_status( + resource_group_name=resource_group.name, + site_name="str", + map_request={"machines": [{"isDependencyMapToBeEnabled": bool, "machineId": "str"}]}, + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations.py new file mode 100644 index 000000000000..2075668ea22a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations.py @@ -0,0 +1,108 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervHostControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_host_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_host_controller.get( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + api_version="2023-10-01-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.hyperv_host_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "createdTimestamp": "str", + "errors": [ + { + "applianceName": "str", + "code": "str", + "discoveryScope": "str", + "id": 0, + "message": "str", + "messageParameters": {"str": "str"}, + "possibleCauses": "str", + "recommendedAction": "str", + "runAsAccountId": "str", + "severity": "str", + "source": "str", + "summaryMessage": "str", + "updatedTimeStamp": "2020-02-20 00:00:00", + } + ], + "fqdn": "str", + "provisioningState": "str", + "runAsAccountId": "str", + "updatedTimestamp": "str", + "version": "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", + }, + api_version="2023-10-01-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.hyperv_host_controller.begin_delete( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations_async.py new file mode 100644 index 000000000000..2a5afa1e76b6 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_host_controller_operations_async.py @@ -0,0 +1,113 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervHostControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_host_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_host_controller.get( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + api_version="2023-10-01-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.hyperv_host_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "createdTimestamp": "str", + "errors": [ + { + "applianceName": "str", + "code": "str", + "discoveryScope": "str", + "id": 0, + "message": "str", + "messageParameters": {"str": "str"}, + "possibleCauses": "str", + "recommendedAction": "str", + "runAsAccountId": "str", + "severity": "str", + "source": "str", + "summaryMessage": "str", + "updatedTimeStamp": "2020-02-20 00:00:00", + } + ], + "fqdn": "str", + "provisioningState": "str", + "runAsAccountId": "str", + "updatedTimestamp": "str", + "version": "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", + }, + api_version="2023-10-01-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.hyperv_host_controller.begin_delete( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations.py new file mode 100644 index 000000000000..dc4e9e34373d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervJobsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_vmware_site(self, resource_group): + response = self.client.hyperv_jobs_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations_async.py new file mode 100644 index 000000000000..be3599cbf484 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_controller_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervJobsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_vmware_site(self, resource_group): + response = self.client.hyperv_jobs_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations.py new file mode 100644 index 000000000000..987ed2670632 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervJobsOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_jobs.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_jobs.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations_async.py new file mode 100644 index 000000000000..57d85c4814e2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_jobs_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervJobsOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_jobs.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_jobs.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations.py new file mode 100644 index 000000000000..a5530a7aa4e3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations.py @@ -0,0 +1,94 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervMachinesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_machines_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.hyperv_machines_controller.update( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + body={ + "properties": { + "allocatedMemoryInMb": 0.0, + "biosGuid": "str", + "biosSerialNumber": "str", + "firmware": "str", + "numberOfProcessorCore": 0, + "operatingSystemDetails": { + "osArchitecture": "str", + "osName": "str", + "osType": "str", + "osVersion": "str", + }, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "runAsAccountId": "str", + "tags": {"str": "str"}, + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_update_properties(self, resource_group): + response = self.client.hyperv_machines_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + meta_data={"value": [{"dependencyMapping": "str", "machineArmId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations_async.py new file mode 100644 index 000000000000..fedab140a08b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_machines_controller_operations_async.py @@ -0,0 +1,97 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervMachinesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_machines_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.hyperv_machines_controller.update( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + body={ + "properties": { + "allocatedMemoryInMb": 0.0, + "biosGuid": "str", + "biosSerialNumber": "str", + "firmware": "str", + "numberOfProcessorCore": 0, + "operatingSystemDetails": { + "osArchitecture": "str", + "osName": "str", + "osType": "str", + "osVersion": "str", + }, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "runAsAccountId": "str", + "tags": {"str": "str"}, + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_update_properties(self, resource_group): + response = await ( + await self.client.hyperv_machines_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + meta_data={"value": [{"dependencyMapping": "str", "machineArmId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations.py new file mode 100644 index 000000000000..5618a0285df1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations.py @@ -0,0 +1,32 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervOperationsStatusControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_hyperv_operations_status(self, resource_group): + response = self.client.hyperv_operations_status_controller.get_hyperv_operations_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations_async.py new file mode 100644 index 000000000000..8096418c52e2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_operations_status_controller_operations_async.py @@ -0,0 +1,33 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervOperationsStatusControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_hyperv_operations_status(self, resource_group): + response = await self.client.hyperv_operations_status_controller.get_hyperv_operations_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..203284f41a6c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervRunAsAccountsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_run_as_accounts_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations_async.py new file mode 100644 index 000000000000..6f568a4398c9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_run_as_accounts_controller_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervRunAsAccountsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_hyperv_site(self, resource_group): + response = self.client.hyperv_run_as_accounts_controller.list_by_hyperv_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.hyperv_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations.py new file mode 100644 index 000000000000..ac84820f4d21 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations.py @@ -0,0 +1,203 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervSitesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get(self, resource_group): + response = self.client.hyperv_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_create(self, resource_group): + response = self.client.hyperv_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.hyperv_sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={ + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "provisioningState": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete(self, resource_group): + response = self.client.hyperv_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_compute_error_summary(self, resource_group): + response = self.client.hyperv_sites_controller.compute_error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_computeusage(self, resource_group): + response = self.client.hyperv_sites_controller.computeusage( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_export_applications(self, resource_group): + response = self.client.hyperv_sites_controller.begin_export_applications( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-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_export_machine_errors(self, resource_group): + response = self.client.hyperv_sites_controller.begin_export_machine_errors( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoveryScope": "str"}}, + api_version="2023-10-01-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_list_health_summary(self, resource_group): + response = self.client.hyperv_sites_controller.list_health_summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_summary(self, resource_group): + response = self.client.hyperv_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations_async.py new file mode 100644 index 000000000000..ed50013bf061 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_controller_operations_async.py @@ -0,0 +1,208 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervSitesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get(self, resource_group): + response = await self.client.hyperv_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_create(self, resource_group): + response = await self.client.hyperv_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.hyperv_sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={ + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "provisioningState": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete(self, resource_group): + response = await self.client.hyperv_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_compute_error_summary(self, resource_group): + response = await self.client.hyperv_sites_controller.compute_error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_computeusage(self, resource_group): + response = await self.client.hyperv_sites_controller.computeusage( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_export_applications(self, resource_group): + response = await ( + await self.client.hyperv_sites_controller.begin_export_applications( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-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_export_machine_errors(self, resource_group): + response = await ( + await self.client.hyperv_sites_controller.begin_export_machine_errors( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoveryScope": "str"}}, + api_version="2023-10-01-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_list_health_summary(self, resource_group): + response = await self.client.hyperv_sites_controller.list_health_summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_summary(self, resource_group): + response = await self.client.hyperv_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations.py new file mode 100644 index 000000000000..137a309ea8e9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations.py @@ -0,0 +1,40 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervSitesOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_subscription(self, resource_group): + response = self.client.hyperv_sites.list_by_subscription( + api_version="2023-10-01-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.hyperv_sites.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-preview", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations_async.py new file mode 100644 index 000000000000..e0bba4a22fab --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_sites_operations_async.py @@ -0,0 +1,41 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervSitesOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_subscription(self, resource_group): + response = self.client.hyperv_sites.list_by_subscription( + api_version="2023-10-01-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.hyperv_sites.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-preview", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations.py new file mode 100644 index 000000000000..1837076fdf18 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervSoftwareInventoriesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_machine_software_inventory(self, resource_group): + response = self.client.hyperv_software_inventories_controller.get_machine_software_inventory( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + default="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_hyperv_machine(self, resource_group): + response = self.client.hyperv_software_inventories_controller.list_by_hyperv_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations_async.py new file mode 100644 index 000000000000..d199c1f4e1be --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_hyperv_software_inventories_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceHypervSoftwareInventoriesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_machine_software_inventory(self, resource_group): + response = await self.client.hyperv_software_inventories_controller.get_machine_software_inventory( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + default="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_hyperv_machine(self, resource_group): + response = self.client.hyperv_software_inventories_controller.list_by_hyperv_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations.py new file mode 100644 index 000000000000..79a181d9af41 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations.py @@ -0,0 +1,61 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceIisWebApplicationsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.iis_web_applications_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.iis_web_applications_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.iis_web_applications_controller.update( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + body={"properties": {"tags": {"str": "str"}}}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations_async.py new file mode 100644 index 000000000000..cc1a340ea357 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_applications_controller_operations_async.py @@ -0,0 +1,62 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceIisWebApplicationsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.iis_web_applications_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.iis_web_applications_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.iis_web_applications_controller.update( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + body={"properties": {"tags": {"str": "str"}}}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations.py new file mode 100644 index 000000000000..de318dd4cb96 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceIisWebServersControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.iis_web_servers_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.iis_web_servers_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_server_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations_async.py new file mode 100644 index 000000000000..5a45fa6ff4a3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_iis_web_servers_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceIisWebServersControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.iis_web_servers_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.iis_web_servers_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_server_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations.py new file mode 100644 index 000000000000..cb517f41bf38 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations.py @@ -0,0 +1,119 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceImportJobsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_deletejobs(self, resource_group): + response = self.client.import_jobs_controller.list_deletejobs( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_deletejob(self, resource_group): + response = self.client.import_jobs_controller.get_deletejob( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_exportjobs(self, resource_group): + response = self.client.import_jobs_controller.list_exportjobs( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_exportjob(self, resource_group): + response = self.client.import_jobs_controller.get_exportjob( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_importjobs(self, resource_group): + response = self.client.import_jobs_controller.list_importjobs( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_importjob(self, resource_group): + response = self.client.import_jobs_controller.get_importjob( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_import_site(self, resource_group): + response = self.client.import_jobs_controller.list_by_import_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.import_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations_async.py new file mode 100644 index 000000000000..4a05d08438c3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_jobs_controller_operations_async.py @@ -0,0 +1,120 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceImportJobsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_deletejobs(self, resource_group): + response = self.client.import_jobs_controller.list_deletejobs( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_deletejob(self, resource_group): + response = await self.client.import_jobs_controller.get_deletejob( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_exportjobs(self, resource_group): + response = self.client.import_jobs_controller.list_exportjobs( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_exportjob(self, resource_group): + response = await self.client.import_jobs_controller.get_exportjob( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_importjobs(self, resource_group): + response = self.client.import_jobs_controller.list_importjobs( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_importjob(self, resource_group): + response = await self.client.import_jobs_controller.get_importjob( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_import_site(self, resource_group): + response = self.client.import_jobs_controller.list_by_import_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.import_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations.py new file mode 100644 index 000000000000..063c34305339 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations.py @@ -0,0 +1,57 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceImportMachinesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_import_site(self, resource_group): + response = self.client.import_machines_controller.list_by_import_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.import_machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete(self, resource_group): + response = self.client.import_machines_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations_async.py new file mode 100644 index 000000000000..7515002f6a7d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_machines_controller_operations_async.py @@ -0,0 +1,58 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceImportMachinesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_import_site(self, resource_group): + response = self.client.import_machines_controller.list_by_import_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.import_machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete(self, resource_group): + response = await self.client.import_machines_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations.py new file mode 100644 index 000000000000..67b5376947b3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations.py @@ -0,0 +1,149 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceImportSitesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_subscription(self, resource_group): + response = self.client.import_sites_controller.list_by_subscription( + api_version="2023-10-01-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.import_sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.import_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_create(self, resource_group): + response = self.client.import_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.import_sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoverySolutionId": "str", "provisioningState": "str"}, "tags": {"str": "str"}}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete(self, resource_group): + response = self.client.import_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete_imported_machines(self, resource_group): + response = self.client.import_sites_controller.delete_imported_machines( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_export_uri(self, resource_group): + response = self.client.import_sites_controller.export_uri( + resource_group_name=resource_group.name, + site_name="str", + body={"importType": "str", "jobArmId": "str", "uri": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_import_uri(self, resource_group): + response = self.client.import_sites_controller.import_uri( + resource_group_name=resource_group.name, + site_name="str", + body={"importType": "str", "jobArmId": "str", "uri": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations_async.py new file mode 100644 index 000000000000..1e12469c5871 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_import_sites_controller_operations_async.py @@ -0,0 +1,150 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceImportSitesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_subscription(self, resource_group): + response = self.client.import_sites_controller.list_by_subscription( + api_version="2023-10-01-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.import_sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.import_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_create(self, resource_group): + response = await self.client.import_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.import_sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoverySolutionId": "str", "provisioningState": "str"}, "tags": {"str": "str"}}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete(self, resource_group): + response = await self.client.import_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete_imported_machines(self, resource_group): + response = await self.client.import_sites_controller.delete_imported_machines( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_export_uri(self, resource_group): + response = await self.client.import_sites_controller.export_uri( + resource_group_name=resource_group.name, + site_name="str", + body={"importType": "str", "jobArmId": "str", "uri": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_import_uri(self, resource_group): + response = await self.client.import_sites_controller.import_uri( + resource_group_name=resource_group.name, + site_name="str", + body={"importType": "str", "jobArmId": "str", "uri": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations.py new file mode 100644 index 000000000000..98bbb894759a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations.py @@ -0,0 +1,107 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceMachinesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_vmware_site(self, resource_group): + response = self.client.machines_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.machines_controller.update( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + body={ + "properties": { + "allocatedMemoryInMb": 0.0, + "biosGuid": "str", + "biosSerialNumber": "str", + "firmware": "str", + "numberOfProcessorCore": 0, + "operatingSystemDetails": { + "osArchitecture": "str", + "osName": "str", + "osType": "str", + "osVersion": "str", + }, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "runAsAccountId": "str", + "tags": {"str": "str"}, + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_start(self, resource_group): + response = self.client.machines_controller.begin_start( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-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_stop(self, resource_group): + response = self.client.machines_controller.begin_stop( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations_async.py new file mode 100644 index 000000000000..3e983f8f109b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_machines_controller_operations_async.py @@ -0,0 +1,112 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceMachinesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_vmware_site(self, resource_group): + response = self.client.machines_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.machines_controller.update( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + body={ + "properties": { + "allocatedMemoryInMb": 0.0, + "biosGuid": "str", + "biosSerialNumber": "str", + "firmware": "str", + "numberOfProcessorCore": 0, + "operatingSystemDetails": { + "osArchitecture": "str", + "osName": "str", + "osType": "str", + "osVersion": "str", + }, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "runAsAccountId": "str", + "tags": {"str": "str"}, + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_start(self, resource_group): + response = await ( + await self.client.machines_controller.begin_start( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-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_stop(self, resource_group): + response = await ( + await self.client.machines_controller.begin_stop( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations.py new file mode 100644 index 000000000000..01cde48a3a0c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations.py @@ -0,0 +1,158 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceMasterSitesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_subscription(self, resource_group): + response = self.client.master_sites_controller.list_by_subscription( + api_version="2023-10-01-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.master_sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.master_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.master_sites_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "allowMultipleSites": bool, + "customerStorageAccountArmId": "str", + "nestedSites": ["str"], + "privateEndpointConnections": [ + { + "id": "str", + "name": "str", + "properties": { + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "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", + } + ], + "provisioningState": "str", + "publicNetworkAccess": "str", + "sites": ["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", + }, + api_version="2023-10-01-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.master_sites_controller.begin_update( + resource_group_name=resource_group.name, + site_name="str", + properties={ + "properties": { + "allowMultipleSites": bool, + "customerStorageAccountArmId": "str", + "publicNetworkAccess": "str", + "sites": ["str"], + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = self.client.master_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_error_summary(self, resource_group): + response = self.client.master_sites_controller.error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations_async.py new file mode 100644 index 000000000000..af72c8c5fd04 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_controller_operations_async.py @@ -0,0 +1,163 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceMasterSitesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_subscription(self, resource_group): + response = self.client.master_sites_controller.list_by_subscription( + api_version="2023-10-01-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.master_sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.master_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.master_sites_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "allowMultipleSites": bool, + "customerStorageAccountArmId": "str", + "nestedSites": ["str"], + "privateEndpointConnections": [ + { + "id": "str", + "name": "str", + "properties": { + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "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", + } + ], + "provisioningState": "str", + "publicNetworkAccess": "str", + "sites": ["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", + }, + api_version="2023-10-01-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.master_sites_controller.begin_update( + resource_group_name=resource_group.name, + site_name="str", + properties={ + "properties": { + "allowMultipleSites": bool, + "customerStorageAccountArmId": "str", + "publicNetworkAccess": "str", + "sites": ["str"], + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = await self.client.master_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_error_summary(self, resource_group): + response = await self.client.master_sites_controller.error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations.py new file mode 100644 index 000000000000..32c80cf48a79 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations.py @@ -0,0 +1,32 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceMasterSitesOperationsStatusControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_vmware_operation_status(self, resource_group): + response = self.client.master_sites_operations_status_controller.get_vmware_operation_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations_async.py new file mode 100644 index 000000000000..1cc47846ea42 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_master_sites_operations_status_controller_operations_async.py @@ -0,0 +1,33 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceMasterSitesOperationsStatusControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_vmware_operation_status(self, resource_group): + response = await self.client.master_sites_operations_status_controller.get_vmware_operation_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations.py new file mode 100644 index 000000000000..12e9638468a2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations.py @@ -0,0 +1,29 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list(self, resource_group): + response = self.client.operations.list( + api_version="2023-10-01-preview", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations_async.py new file mode 100644 index 000000000000..86d1631c64a8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_operations_async.py @@ -0,0 +1,30 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, 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="2023-10-01-preview", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations.py new file mode 100644 index 000000000000..fb88d0f1300b --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations.py @@ -0,0 +1,93 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServicePrivateEndpointConnectionControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_master_site(self, resource_group): + response = self.client.private_endpoint_connection_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.private_endpoint_connection_controller.get( + resource_group_name=resource_group.name, + site_name="str", + pe_connection_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_create(self, resource_group): + response = self.client.private_endpoint_connection_controller.create( + resource_group_name=resource_group.name, + site_name="str", + pe_connection_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete(self, resource_group): + response = self.client.private_endpoint_connection_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + pe_connection_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations_async.py new file mode 100644 index 000000000000..2812f28497ac --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_endpoint_connection_controller_operations_async.py @@ -0,0 +1,94 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServicePrivateEndpointConnectionControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_master_site(self, resource_group): + response = self.client.private_endpoint_connection_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.private_endpoint_connection_controller.get( + resource_group_name=resource_group.name, + site_name="str", + pe_connection_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_create(self, resource_group): + response = await self.client.private_endpoint_connection_controller.create( + resource_group_name=resource_group.name, + site_name="str", + pe_connection_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "groupIds": ["str"], + "privateEndpoint": {"id": "str"}, + "privateLinkServiceConnectionState": { + "actionsRequired": "str", + "description": "str", + "status": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete(self, resource_group): + response = await self.client.private_endpoint_connection_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + pe_connection_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations.py new file mode 100644 index 000000000000..e65b70eba28d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServicePrivateLinkResourcesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_master_site(self, resource_group): + response = self.client.private_link_resources_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.private_link_resources_controller.get( + resource_group_name=resource_group.name, + site_name="str", + private_link_resource_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations_async.py new file mode 100644 index 000000000000..c3d412e1a00c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_private_link_resources_controller_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServicePrivateLinkResourcesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_master_site(self, resource_group): + response = self.client.private_link_resources_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.private_link_resources_controller.get( + resource_group_name=resource_group.name, + site_name="str", + private_link_resource_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..ba2dd059bca0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceRunAsAccountsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_vmware_site(self, resource_group): + response = self.client.run_as_accounts_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations_async.py new file mode 100644 index 000000000000..911cc99a19e6 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_run_as_accounts_controller_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceRunAsAccountsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_vmware_site(self, resource_group): + response = self.client.run_as_accounts_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations.py new file mode 100644 index 000000000000..44629ace7257 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations.py @@ -0,0 +1,105 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerDependencyMapControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_client_group_members(self, resource_group): + response = self.client.server_dependency_map_controller.begin_client_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "processGroupName": "str", + "processName": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_export_dependencies(self, resource_group): + response = self.client.server_dependency_map_controller.begin_export_dependencies( + resource_group_name=resource_group.name, + site_name="str", + request_body={"endTime": "2020-02-20 00:00:00", "startTime": "2020-02-20 00:00:00"}, + api_version="2023-10-01-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_generate_coarse_map(self, resource_group): + response = self.client.server_dependency_map_controller.begin_generate_coarse_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_generate_detailed_map(self, resource_group): + response = self.client.server_dependency_map_controller.begin_generate_detailed_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_server_group_members(self, resource_group): + response = self.client.server_dependency_map_controller.begin_server_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "serverPort": 0, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations_async.py new file mode 100644 index 000000000000..cbb3ddd1f98d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_dependency_map_controller_operations_async.py @@ -0,0 +1,116 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerDependencyMapControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_client_group_members(self, resource_group): + response = await ( + await self.client.server_dependency_map_controller.begin_client_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "processGroupName": "str", + "processName": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_export_dependencies(self, resource_group): + response = await ( + await self.client.server_dependency_map_controller.begin_export_dependencies( + resource_group_name=resource_group.name, + site_name="str", + request_body={"endTime": "2020-02-20 00:00:00", "startTime": "2020-02-20 00:00:00"}, + api_version="2023-10-01-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_generate_coarse_map(self, resource_group): + response = await ( + await self.client.server_dependency_map_controller.begin_generate_coarse_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_generate_detailed_map(self, resource_group): + response = await ( + await self.client.server_dependency_map_controller.begin_generate_detailed_map( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "machineId": "str", + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-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_server_group_members(self, resource_group): + response = await ( + await self.client.server_dependency_map_controller.begin_server_group_members( + resource_group_name=resource_group.name, + site_name="str", + map_request={ + "endTime": "2020-02-20 00:00:00", + "filters": {"machineIds": ["str"], "processIds": ["str"]}, + "serverPort": 0, + "startTime": "2020-02-20 00:00:00", + }, + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations.py new file mode 100644 index 000000000000..86124d3bf820 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerJobsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_server_site_resource(self, resource_group): + response = self.client.server_jobs_controller.list_by_server_site_resource( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.server_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations_async.py new file mode 100644 index 000000000000..b56dc2e6ef7a --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_jobs_controller_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerJobsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_server_site_resource(self, resource_group): + response = self.client.server_jobs_controller.list_by_server_site_resource( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.server_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations.py new file mode 100644 index 000000000000..ab7075b0765c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations.py @@ -0,0 +1,32 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerOperationsStatusControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_server_site_operations_status(self, resource_group): + response = self.client.server_operations_status_controller.get_server_site_operations_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations_async.py new file mode 100644 index 000000000000..ab6f4df52f08 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_operations_status_controller_operations_async.py @@ -0,0 +1,33 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerOperationsStatusControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_server_site_operations_status(self, resource_group): + response = await self.client.server_operations_status_controller.get_server_site_operations_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..71c62d8a562c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerRunAsAccountsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_server_site_resource(self, resource_group): + response = self.client.server_run_as_accounts_controller.list_by_server_site_resource( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.server_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations_async.py new file mode 100644 index 000000000000..8de37277c968 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_run_as_accounts_controller_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerRunAsAccountsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_server_site_resource(self, resource_group): + response = self.client.server_run_as_accounts_controller.list_by_server_site_resource( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.server_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations.py new file mode 100644 index 000000000000..441e11194ca8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations.py @@ -0,0 +1,261 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerSitesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_subscription(self, resource_group): + response = self.client.server_sites_controller.list_by_subscription( + api_version="2023-10-01-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.server_sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.server_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.server_sites_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-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_update(self, resource_group): + response = self.client.server_sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={ + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete(self, resource_group): + response = self.client.server_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_compute_error_summary(self, resource_group): + response = self.client.server_sites_controller.compute_error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_computeusage(self, resource_group): + response = self.client.server_sites_controller.computeusage( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_export_applications(self, resource_group): + response = self.client.server_sites_controller.begin_export_applications( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-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_export_machine_errors(self, resource_group): + response = self.client.server_sites_controller.begin_export_machine_errors( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoveryScope": "str"}}, + api_version="2023-10-01-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_list_health_summary(self, resource_group): + response = self.client.server_sites_controller.list_health_summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_refresh_site(self, resource_group): + response = self.client.server_sites_controller.begin_refresh_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_summary(self, resource_group): + response = self.client.server_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_update_dependency_map_status(self, resource_group): + response = self.client.server_sites_controller.begin_update_dependency_map_status( + resource_group_name=resource_group.name, + site_name="str", + map_request={"machines": [{"isDependencyMapToBeEnabled": bool, "machineId": "str"}]}, + api_version="2023-10-01-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_properties(self, resource_group): + response = self.client.server_sites_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + meta_data={"value": [{"dependencyMapping": "str", "machineArmId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations_async.py new file mode 100644 index 000000000000..176eec878487 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_sites_controller_operations_async.py @@ -0,0 +1,274 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerSitesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_subscription(self, resource_group): + response = self.client.server_sites_controller.list_by_subscription( + api_version="2023-10-01-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.server_sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.server_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.server_sites_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "id": "str", + "name": "str", + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-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_update(self, resource_group): + response = await self.client.server_sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={ + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete(self, resource_group): + response = await self.client.server_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_compute_error_summary(self, resource_group): + response = await self.client.server_sites_controller.compute_error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_computeusage(self, resource_group): + response = await self.client.server_sites_controller.computeusage( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_export_applications(self, resource_group): + response = await ( + await self.client.server_sites_controller.begin_export_applications( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-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_export_machine_errors(self, resource_group): + response = await ( + await self.client.server_sites_controller.begin_export_machine_errors( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoveryScope": "str"}}, + api_version="2023-10-01-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_list_health_summary(self, resource_group): + response = await self.client.server_sites_controller.list_health_summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_refresh_site(self, resource_group): + response = await ( + await self.client.server_sites_controller.begin_refresh_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_summary(self, resource_group): + response = await self.client.server_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_update_dependency_map_status(self, resource_group): + response = await ( + await self.client.server_sites_controller.begin_update_dependency_map_status( + resource_group_name=resource_group.name, + site_name="str", + map_request={"machines": [{"isDependencyMapToBeEnabled": bool, "machineId": "str"}]}, + api_version="2023-10-01-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_properties(self, resource_group): + response = await ( + await self.client.server_sites_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + meta_data={"value": [{"dependencyMapping": "str", "machineArmId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations.py new file mode 100644 index 000000000000..7e68ea8d426c --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerSoftwareInventoriesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_machine_software_inventory(self, resource_group): + response = self.client.server_software_inventories_controller.get_machine_software_inventory( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + default="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_server(self, resource_group): + response = self.client.server_software_inventories_controller.list_by_server( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations_async.py new file mode 100644 index 000000000000..d23821face8f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_server_software_inventories_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServerSoftwareInventoriesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_machine_software_inventory(self, resource_group): + response = await self.client.server_software_inventories_controller.get_machine_software_inventory( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + default="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_server(self, resource_group): + response = self.client.server_software_inventories_controller.list_by_server( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations.py new file mode 100644 index 000000000000..000e84ffd55f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations.py @@ -0,0 +1,118 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServersControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_server_site_resource(self, resource_group): + response = self.client.servers_controller.list_by_server_site_resource( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_machine(self, resource_group): + response = self.client.servers_controller.get_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update_machine(self, resource_group): + response = self.client.servers_controller.update_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + body={ + "properties": { + "allocatedMemoryInMb": 0.0, + "biosGuid": "str", + "biosSerialNumber": "str", + "disks": [ + { + "diskType": "str", + "generatedId": "str", + "id": "str", + "lun": 0, + "maxSizeInBytes": 0, + "name": "str", + "path": "str", + } + ], + "firmware": "str", + "fqdn": "str", + "hydratedFqdn": "str", + "networkAdapters": [ + { + "ipAddressList": ["str"], + "ipAddressType": "str", + "macAddress": "str", + "networkName": "str", + "nicId": "str", + } + ], + "numberOfProcessorCore": 0, + "operatingSystemDetails": { + "osArchitecture": "str", + "osName": "str", + "osType": "str", + "osVersion": "str", + }, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "provisioningState": "str", + "runAsAccountId": "str", + "tags": {"str": {}}, + "validationRequired": "str", + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete_machine(self, resource_group): + response = self.client.servers_controller.delete_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations_async.py new file mode 100644 index 000000000000..73f6c98ddc80 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_servers_controller_operations_async.py @@ -0,0 +1,119 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceServersControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_server_site_resource(self, resource_group): + response = self.client.servers_controller.list_by_server_site_resource( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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_machine(self, resource_group): + response = await self.client.servers_controller.get_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update_machine(self, resource_group): + response = await self.client.servers_controller.update_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + body={ + "properties": { + "allocatedMemoryInMb": 0.0, + "biosGuid": "str", + "biosSerialNumber": "str", + "disks": [ + { + "diskType": "str", + "generatedId": "str", + "id": "str", + "lun": 0, + "maxSizeInBytes": 0, + "name": "str", + "path": "str", + } + ], + "firmware": "str", + "fqdn": "str", + "hydratedFqdn": "str", + "networkAdapters": [ + { + "ipAddressList": ["str"], + "ipAddressType": "str", + "macAddress": "str", + "networkName": "str", + "nicId": "str", + } + ], + "numberOfProcessorCore": 0, + "operatingSystemDetails": { + "osArchitecture": "str", + "osName": "str", + "osType": "str", + "osVersion": "str", + }, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "provisioningState": "str", + "runAsAccountId": "str", + "tags": {"str": {}}, + "validationRequired": "str", + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete_machine(self, resource_group): + response = await self.client.servers_controller.delete_machine( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations.py new file mode 100644 index 000000000000..2a1713224c26 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations.py @@ -0,0 +1,238 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSitesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_subscription(self, resource_group): + response = self.client.sites_controller.list_by_subscription( + api_version="2023-10-01-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.sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_create(self, resource_group): + response = self.client.sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "eTag": "str", + "id": "str", + "name": "str", + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={ + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "provisioningState": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_delete(self, resource_group): + response = self.client.sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_compute_error_summary(self, resource_group): + response = self.client.sites_controller.compute_error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_computeusage(self, resource_group): + response = self.client.sites_controller.computeusage( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_export_applications(self, resource_group): + response = self.client.sites_controller.begin_export_applications( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-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_export_machine_errors(self, resource_group): + response = self.client.sites_controller.begin_export_machine_errors( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoveryScope": "str"}}, + api_version="2023-10-01-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_export_machines(self, resource_group): + response = self.client.sites_controller.begin_export_machines( + resource_group_name=resource_group.name, + site_name="str", + body={"filter": "str"}, + api_version="2023-10-01-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_list_health_summary(self, resource_group): + response = self.client.sites_controller.list_health_summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_summary(self, resource_group): + response = self.client.sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations_async.py new file mode 100644 index 000000000000..6d350b8b6d98 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sites_controller_operations_async.py @@ -0,0 +1,245 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSitesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_subscription(self, resource_group): + response = self.client.sites_controller.list_by_subscription( + api_version="2023-10-01-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.sites_controller.list_by_resource_group( + resource_group_name=resource_group.name, + api_version="2023-10-01-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.sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_create(self, resource_group): + response = await self.client.sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + body={ + "location": "str", + "eTag": "str", + "id": "str", + "name": "str", + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "masterSiteId": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.sites_controller.update( + resource_group_name=resource_group.name, + site_name="str", + body={ + "properties": { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "discoverySolutionId": "str", + "provisioningState": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + }, + "tags": {"str": "str"}, + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_delete(self, resource_group): + response = await self.client.sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_compute_error_summary(self, resource_group): + response = await self.client.sites_controller.compute_error_summary( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_computeusage(self, resource_group): + response = await self.client.sites_controller.computeusage( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_export_applications(self, resource_group): + response = await ( + await self.client.sites_controller.begin_export_applications( + resource_group_name=resource_group.name, + site_name="str", + body={}, + api_version="2023-10-01-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_export_machine_errors(self, resource_group): + response = await ( + await self.client.sites_controller.begin_export_machine_errors( + resource_group_name=resource_group.name, + site_name="str", + body={"properties": {"discoveryScope": "str"}}, + api_version="2023-10-01-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_export_machines(self, resource_group): + response = await ( + await self.client.sites_controller.begin_export_machines( + resource_group_name=resource_group.name, + site_name="str", + body={"filter": "str"}, + api_version="2023-10-01-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_list_health_summary(self, resource_group): + response = await self.client.sites_controller.list_health_summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_summary(self, resource_group): + response = await self.client.sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations.py new file mode 100644 index 000000000000..e4b9af53cbca --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlAvailabilityGroupsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_sql_site(self, resource_group): + response = self.client.sql_availability_groups_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_availability_groups_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_availability_group_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations_async.py new file mode 100644 index 000000000000..d3f6d5dbbcac --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_availability_groups_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlAvailabilityGroupsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_sql_site(self, resource_group): + response = self.client.sql_availability_groups_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_availability_groups_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_availability_group_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations.py new file mode 100644 index 000000000000..92b6f2d04178 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlDatabasesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_sql_site(self, resource_group): + response = self.client.sql_databases_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_databases_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_database_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations_async.py new file mode 100644 index 000000000000..aafb859896c0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_databases_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlDatabasesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_sql_site(self, resource_group): + response = self.client.sql_databases_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_databases_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_database_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations.py new file mode 100644 index 000000000000..537ec89faf6d --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations.py @@ -0,0 +1,88 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlDiscoverySiteDataSourceControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_sql_site(self, resource_group): + response = self.client.sql_discovery_site_data_source_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_discovery_site_data_source_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-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.sql_discovery_site_data_source_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + discovery_site_data_source_name="str", + body={ + "id": "str", + "name": "str", + "properties": {"discoverySiteId": "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", + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = self.client.sql_discovery_site_data_source_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations_async.py new file mode 100644 index 000000000000..ecbf3ec937ca --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_discovery_site_data_source_controller_operations_async.py @@ -0,0 +1,91 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlDiscoverySiteDataSourceControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_sql_site(self, resource_group): + response = self.client.sql_discovery_site_data_source_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_discovery_site_data_source_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-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.sql_discovery_site_data_source_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + discovery_site_data_source_name="str", + body={ + "id": "str", + "name": "str", + "properties": {"discoverySiteId": "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", + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = await self.client.sql_discovery_site_data_source_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations.py new file mode 100644 index 000000000000..2c277f18ac61 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlJobsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_sql_site(self, resource_group): + response = self.client.sql_jobs_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations_async.py new file mode 100644 index 000000000000..77a6bc140223 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_jobs_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlJobsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_sql_site(self, resource_group): + response = self.client.sql_jobs_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_jobs_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + job_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations.py new file mode 100644 index 000000000000..c7c43dc36848 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations.py @@ -0,0 +1,33 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlOperationsStatusControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_sql_operation_status(self, resource_group): + response = self.client.sql_operations_status_controller.get_sql_operation_status( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations_async.py new file mode 100644 index 000000000000..3727fb2da6ea --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_operations_status_controller_operations_async.py @@ -0,0 +1,34 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlOperationsStatusControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_sql_operation_status(self, resource_group): + response = await self.client.sql_operations_status_controller.get_sql_operation_status( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..5844b18373e4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlRunAsAccountsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_sql_site(self, resource_group): + response = self.client.sql_run_as_accounts_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations_async.py new file mode 100644 index 000000000000..ea83c81321b9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_run_as_accounts_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlRunAsAccountsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_sql_site(self, resource_group): + response = self.client.sql_run_as_accounts_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations.py new file mode 100644 index 000000000000..3b49d9316046 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations.py @@ -0,0 +1,104 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlServersControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_sql_site(self, resource_group): + response = self.client.sql_servers_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_servers_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_server_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.sql_servers_controller.update( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_server_name="str", + body={ + "properties": { + "createdTimestamp": "str", + "edition": "str", + "engineEdition": "str", + "hostName": "str", + "hydratedRunAsAccountId": "str", + "hyperthreadRatio": 0, + "isClustered": bool, + "isDeleted": bool, + "isHighAvailabilityEnabled": bool, + "logicalCpuCount": 0, + "maxServerMemoryInUseInMb": 0.0, + "numOfLogins": 0, + "numberOfAgDatabases": 0, + "numberOfUserDatabases": 0, + "physicalCpuCount": 0.0, + "portNumber": 0, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "provisioningState": "str", + "runAsAccountId": "str", + "sqlFciProperties": { + "isMultiSubnet": bool, + "networkName": "str", + "sharedDiskCount": 0, + "state": "str", + }, + "sqlServerName": "str", + "sqlStartTime": "2020-02-20 00:00:00", + "status": "str", + "sumOfUserDatabasesSizeInMb": 0.0, + "tags": {"str": {}}, + "tempDbSizeInMb": 0.0, + "updatedTimestamp": "str", + "version": "str", + "visibleOnlineCoreCount": 0, + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations_async.py new file mode 100644 index 000000000000..3e5502a6c013 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_servers_controller_operations_async.py @@ -0,0 +1,105 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlServersControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_sql_site(self, resource_group): + response = self.client.sql_servers_controller.list_by_sql_site( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-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.sql_servers_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_server_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.sql_servers_controller.update( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + sql_server_name="str", + body={ + "properties": { + "createdTimestamp": "str", + "edition": "str", + "engineEdition": "str", + "hostName": "str", + "hydratedRunAsAccountId": "str", + "hyperthreadRatio": 0, + "isClustered": bool, + "isDeleted": bool, + "isHighAvailabilityEnabled": bool, + "logicalCpuCount": 0, + "maxServerMemoryInUseInMb": 0.0, + "numOfLogins": 0, + "numberOfAgDatabases": 0, + "numberOfUserDatabases": 0, + "physicalCpuCount": 0.0, + "portNumber": 0, + "productSupportStatus": { + "currentVersion": "str", + "esuStatus": "str", + "esuYear": "str", + "supportEndDate": "2020-02-20 00:00:00", + "supportStatus": "str", + }, + "provisioningState": "str", + "runAsAccountId": "str", + "sqlFciProperties": { + "isMultiSubnet": bool, + "networkName": "str", + "sharedDiskCount": 0, + "state": "str", + }, + "sqlServerName": "str", + "sqlStartTime": "2020-02-20 00:00:00", + "status": "str", + "sumOfUserDatabasesSizeInMb": 0.0, + "tags": {"str": {}}, + "tempDbSizeInMb": 0.0, + "updatedTimestamp": "str", + "version": "str", + "visibleOnlineCoreCount": 0, + } + }, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations.py new file mode 100644 index 000000000000..1223adb33eaa --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations.py @@ -0,0 +1,215 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlSitesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_master_site(self, resource_group): + response = self.client.sql_sites_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.sql_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_create(self, resource_group): + response = self.client.sql_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "discoveryScenario": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # 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.sql_sites_controller.begin_update( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + properties={ + "properties": { + "discoveryScenario": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + } + ], + } + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = self.client.sql_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_error_summary(self, resource_group): + response = self.client.sql_sites_controller.error_summary( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_export_sql_server_errors(self, resource_group): + response = self.client.sql_sites_controller.begin_export_sql_server_errors( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={}, + api_version="2023-10-01-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_export_sql_servers(self, resource_group): + response = self.client.sql_sites_controller.begin_export_sql_servers( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={"applianceName": "str", "filter": "str"}, + api_version="2023-10-01-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_refresh(self, resource_group): + response = self.client.sql_sites_controller.begin_refresh( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-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_summary(self, resource_group): + response = self.client.sql_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations_async.py new file mode 100644 index 000000000000..f0d66f337bd4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_sql_sites_controller_operations_async.py @@ -0,0 +1,224 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceSqlSitesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_master_site(self, resource_group): + response = self.client.sql_sites_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.sql_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_create(self, resource_group): + response = await self.client.sql_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "discoveryScenario": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # 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.sql_sites_controller.begin_update( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + properties={ + "properties": { + "discoveryScenario": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + } + ], + } + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = await self.client.sql_sites_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_error_summary(self, resource_group): + response = await self.client.sql_sites_controller.error_summary( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_export_sql_server_errors(self, resource_group): + response = await ( + await self.client.sql_sites_controller.begin_export_sql_server_errors( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={}, + api_version="2023-10-01-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_export_sql_servers(self, resource_group): + response = await ( + await self.client.sql_sites_controller.begin_export_sql_servers( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={"applianceName": "str", "filter": "str"}, + api_version="2023-10-01-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_refresh(self, resource_group): + response = await ( + await self.client.sql_sites_controller.begin_refresh( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-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_summary(self, resource_group): + response = await self.client.sql_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + sql_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations.py new file mode 100644 index 000000000000..4a8884b463d4 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations.py @@ -0,0 +1,61 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceTomcatWebApplicationsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.tomcat_web_applications_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.tomcat_web_applications_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_update(self, resource_group): + response = self.client.tomcat_web_applications_controller.update( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations_async.py new file mode 100644 index 000000000000..4e9226cf0ef0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_applications_controller_operations_async.py @@ -0,0 +1,62 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceTomcatWebApplicationsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.tomcat_web_applications_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.tomcat_web_applications_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_update(self, resource_group): + response = await self.client.tomcat_web_applications_controller.update( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_application_name="str", + body={}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations.py new file mode 100644 index 000000000000..264d87643685 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceTomcatWebServersControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.tomcat_web_servers_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.tomcat_web_servers_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_server_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations_async.py new file mode 100644 index 000000000000..425207091855 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_tomcat_web_servers_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceTomcatWebServersControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.tomcat_web_servers_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.tomcat_web_servers_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + web_server_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations.py new file mode 100644 index 000000000000..ce9ec7f93bf2 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations.py @@ -0,0 +1,112 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVcenterControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_vmware_site(self, resource_group): + response = self.client.vcenter_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.vcenter_controller.get( + resource_group_name=resource_group.name, + site_name="str", + vcenter_name="str", + api_version="2023-10-01-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.vcenter_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + vcenter_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "createdTimestamp": "str", + "errors": [ + { + "applianceName": "str", + "code": "str", + "discoveryScope": "str", + "id": 0, + "message": "str", + "messageParameters": {"str": "str"}, + "possibleCauses": "str", + "recommendedAction": "str", + "runAsAccountId": "str", + "severity": "str", + "source": "str", + "summaryMessage": "str", + "updatedTimeStamp": "2020-02-20 00:00:00", + } + ], + "fqdn": "str", + "friendlyName": "str", + "instanceUuid": "str", + "perfStatisticsLevel": "str", + "port": "str", + "provisioningState": "str", + "runAsAccountId": "str", + "updatedTimestamp": "str", + "version": "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", + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = self.client.vcenter_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + vcenter_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations_async.py new file mode 100644 index 000000000000..11e88f8288a0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vcenter_controller_operations_async.py @@ -0,0 +1,115 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVcenterControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_vmware_site(self, resource_group): + response = self.client.vcenter_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.vcenter_controller.get( + resource_group_name=resource_group.name, + site_name="str", + vcenter_name="str", + api_version="2023-10-01-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.vcenter_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + vcenter_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "createdTimestamp": "str", + "errors": [ + { + "applianceName": "str", + "code": "str", + "discoveryScope": "str", + "id": 0, + "message": "str", + "messageParameters": {"str": "str"}, + "possibleCauses": "str", + "recommendedAction": "str", + "runAsAccountId": "str", + "severity": "str", + "source": "str", + "summaryMessage": "str", + "updatedTimeStamp": "2020-02-20 00:00:00", + } + ], + "fqdn": "str", + "friendlyName": "str", + "instanceUuid": "str", + "perfStatisticsLevel": "str", + "port": "str", + "provisioningState": "str", + "runAsAccountId": "str", + "updatedTimestamp": "str", + "version": "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", + }, + api_version="2023-10-01-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_delete(self, resource_group): + response = await self.client.vcenter_controller.delete( + resource_group_name=resource_group.name, + site_name="str", + vcenter_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations.py new file mode 100644 index 000000000000..99ad53e031b3 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations.py @@ -0,0 +1,44 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwareHostControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_vmware_site(self, resource_group): + response = self.client.vmware_host_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.vmware_host_controller.get( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations_async.py new file mode 100644 index 000000000000..53342de1444f --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_host_controller_operations_async.py @@ -0,0 +1,45 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwareHostControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_vmware_site(self, resource_group): + response = self.client.vmware_host_controller.list_by_vmware_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.vmware_host_controller.get( + resource_group_name=resource_group.name, + site_name="str", + host_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations.py new file mode 100644 index 000000000000..791743f17da8 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations.py @@ -0,0 +1,32 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwareOperationsStatusOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_vmware_operation_status(self, resource_group): + response = self.client.vmware_operations_status.get_vmware_operation_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations_async.py new file mode 100644 index 000000000000..92556ab2b0e9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_operations_status_operations_async.py @@ -0,0 +1,33 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwareOperationsStatusOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_vmware_operation_status(self, resource_group): + response = await self.client.vmware_operations_status.get_vmware_operation_status( + resource_group_name=resource_group.name, + site_name="str", + operation_status_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations.py new file mode 100644 index 000000000000..ffeceb682433 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations.py @@ -0,0 +1,71 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwarePropertiesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_update_dependency_map_status(self, resource_group): + response = self.client.vmware_properties_controller.begin_update_dependency_map_status( + resource_group_name=resource_group.name, + site_name="str", + update_machine_dep_map_status={"machines": [{"isDependencyMapToBeEnabled": bool, "machineId": "str"}]}, + api_version="2023-10-01-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_properties(self, resource_group): + response = self.client.vmware_properties_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + meta_data={"value": [{"dependencyMapping": "str", "machineArmId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-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_run_as_account(self, resource_group): + response = self.client.vmware_properties_controller.begin_update_run_as_account( + resource_group_name=resource_group.name, + site_name="str", + update_machine_run_as_account={"machines": [{"machineId": "str", "runAsAccountId": "str"}]}, + api_version="2023-10-01-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_tags(self, resource_group): + response = self.client.vmware_properties_controller.begin_update_tags( + resource_group_name=resource_group.name, + site_name="str", + update_machine_tags={"machines": [{"machineId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations_async.py new file mode 100644 index 000000000000..11e881b96406 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_properties_controller_operations_async.py @@ -0,0 +1,80 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwarePropertiesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_update_dependency_map_status(self, resource_group): + response = await ( + await self.client.vmware_properties_controller.begin_update_dependency_map_status( + resource_group_name=resource_group.name, + site_name="str", + update_machine_dep_map_status={"machines": [{"isDependencyMapToBeEnabled": bool, "machineId": "str"}]}, + api_version="2023-10-01-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_properties(self, resource_group): + response = await ( + await self.client.vmware_properties_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + meta_data={"value": [{"dependencyMapping": "str", "machineArmId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-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_run_as_account(self, resource_group): + response = await ( + await self.client.vmware_properties_controller.begin_update_run_as_account( + resource_group_name=resource_group.name, + site_name="str", + update_machine_run_as_account={"machines": [{"machineId": "str", "runAsAccountId": "str"}]}, + api_version="2023-10-01-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_tags(self, resource_group): + response = await ( + await self.client.vmware_properties_controller.begin_update_tags( + resource_group_name=resource_group.name, + site_name="str", + update_machine_tags={"machines": [{"machineId": "str", "tags": {"str": "str"}}]}, + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations.py new file mode 100644 index 000000000000..f7ea3cd6c468 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwareSoftwareInventoriesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_get_machine_software_inventory(self, resource_group): + response = self.client.vmware_software_inventories_controller.get_machine_software_inventory( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + default="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_machine_resource(self, resource_group): + response = self.client.vmware_software_inventories_controller.list_by_machine_resource( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations_async.py new file mode 100644 index 000000000000..9e80eb475b41 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_vmware_software_inventories_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceVmwareSoftwareInventoriesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_get_machine_software_inventory(self, resource_group): + response = await self.client.vmware_software_inventories_controller.get_machine_software_inventory( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + default="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_machine_resource(self, resource_group): + response = self.client.vmware_software_inventories_controller.list_by_machine_resource( + resource_group_name=resource_group.name, + site_name="str", + machine_name="str", + api_version="2023-10-01-preview", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations.py new file mode 100644 index 000000000000..2e7fb2ba8653 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations.py @@ -0,0 +1,88 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppDiscoverySiteDataSourcesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.web_app_discovery_site_data_sources_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.web_app_discovery_site_data_sources_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-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.web_app_discovery_site_data_sources_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + discovery_site_data_source_name="str", + body={ + "id": "str", + "name": "str", + "properties": {"discoverySiteId": "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", + }, + api_version="2023-10-01-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.web_app_discovery_site_data_sources_controller.begin_delete( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations_async.py new file mode 100644 index 000000000000..dc925e8beb55 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_discovery_site_data_sources_controller_operations_async.py @@ -0,0 +1,95 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppDiscoverySiteDataSourcesControllerOperationsAsync( + AzureMgmtRecordedTestCase +): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.web_app_discovery_site_data_sources_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.web_app_discovery_site_data_sources_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-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.web_app_discovery_site_data_sources_controller.begin_create( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + discovery_site_data_source_name="str", + body={ + "id": "str", + "name": "str", + "properties": {"discoverySiteId": "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", + }, + api_version="2023-10-01-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.web_app_discovery_site_data_sources_controller.begin_delete( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + discovery_site_data_source_name="str", + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations.py new file mode 100644 index 000000000000..349a30bd5ed9 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppExtendedMachinesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.web_app_extended_machines_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.web_app_extended_machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + extended_machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations_async.py new file mode 100644 index 000000000000..0f4c21c5d639 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_extended_machines_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppExtendedMachinesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.web_app_extended_machines_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.web_app_extended_machines_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + extended_machine_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations.py new file mode 100644 index 000000000000..e5d5e7b00690 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations.py @@ -0,0 +1,33 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppPropertiesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_update_properties(self, resource_group): + response = self.client.web_app_properties_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"webApps": [{"tags": {"str": "str"}, "webAppArmId": "str"}]}, + api_version="2023-10-01-preview", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations_async.py new file mode 100644 index 000000000000..cf84f7421706 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_properties_controller_operations_async.py @@ -0,0 +1,36 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppPropertiesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_update_properties(self, resource_group): + response = await ( + await self.client.web_app_properties_controller.begin_update_properties( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"webApps": [{"tags": {"str": "str"}, "webAppArmId": "str"}]}, + api_version="2023-10-01-preview", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations.py new file mode 100644 index 000000000000..807bd0b16d17 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations.py @@ -0,0 +1,46 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppRunAsAccountsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.web_app_run_as_accounts_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.web_app_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations_async.py new file mode 100644 index 000000000000..050cbce63afd --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_run_as_accounts_controller_operations_async.py @@ -0,0 +1,47 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppRunAsAccountsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.web_app_run_as_accounts_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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.web_app_run_as_accounts_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + account_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations.py new file mode 100644 index 000000000000..242dce072932 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations.py @@ -0,0 +1,201 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppSitesControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_master_site(self, resource_group): + response = self.client.web_app_sites_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.web_app_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_create(self, resource_group): + response = self.client.web_app_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "discoveryScenario": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # 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.web_app_sites_controller.begin_update( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + properties={ + "properties": { + "discoveryScenario": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + } + ], + } + }, + api_version="2023-10-01-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.web_app_sites_controller.begin_delete( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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_error_summary(self, resource_group): + response = self.client.web_app_sites_controller.error_summary( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_begin_export_inventory(self, resource_group): + response = self.client.web_app_sites_controller.begin_export_inventory( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"filter": "str"}, + api_version="2023-10-01-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_refresh(self, resource_group): + response = self.client.web_app_sites_controller.begin_refresh( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-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_summary(self, resource_group): + response = self.client.web_app_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations_async.py new file mode 100644 index 000000000000..9b06441928e5 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_app_sites_controller_operations_async.py @@ -0,0 +1,210 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebAppSitesControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_master_site(self, resource_group): + response = self.client.web_app_sites_controller.list_by_master_site( + resource_group_name=resource_group.name, + site_name="str", + api_version="2023-10-01-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.web_app_sites_controller.get( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_create(self, resource_group): + response = await self.client.web_app_sites_controller.create( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={ + "id": "str", + "name": "str", + "properties": { + "discoveryScenario": "str", + "provisioningState": "str", + "serviceEndpoint": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "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", + }, + api_version="2023-10-01-preview", + ) + + # 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.web_app_sites_controller.begin_update( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + properties={ + "properties": { + "discoveryScenario": "str", + "siteAppliancePropertiesCollection": [ + { + "agentDetails": { + "id": "str", + "keyVaultId": "str", + "keyVaultUri": "str", + "lastHeartBeatUtc": "2020-02-20 00:00:00", + "version": "str", + }, + "applianceName": "str", + "servicePrincipalIdentityDetails": { + "aadAuthority": "str", + "applicationId": "str", + "audience": "str", + "objectId": "str", + "rawCertData": "str", + "tenantId": "str", + }, + } + ], + } + }, + api_version="2023-10-01-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.web_app_sites_controller.begin_delete( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-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_error_summary(self, resource_group): + response = await self.client.web_app_sites_controller.error_summary( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_begin_export_inventory(self, resource_group): + response = await ( + await self.client.web_app_sites_controller.begin_export_inventory( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"filter": "str"}, + api_version="2023-10-01-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_refresh(self, resource_group): + response = await ( + await self.client.web_app_sites_controller.begin_refresh( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + body={"applianceName": "str"}, + api_version="2023-10-01-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_summary(self, resource_group): + response = await self.client.web_app_sites_controller.summary( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations.py new file mode 100644 index 000000000000..665d6e596455 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations.py @@ -0,0 +1,32 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebApplicationsControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.web_applications_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations_async.py new file mode 100644 index 000000000000..42532708d9c1 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_applications_controller_operations_async.py @@ -0,0 +1,33 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebApplicationsControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.web_applications_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations.py new file mode 100644 index 000000000000..31a8f6409a90 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations.py @@ -0,0 +1,32 @@ +# 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.migrate import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebServersControllerOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_list_by_web_app_site(self, resource_group): + response = self.client.web_servers_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations_async.py b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations_async.py new file mode 100644 index 000000000000..bcda6de9c9df --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/generated_tests/test_azure_migrate_discovery_service_web_servers_controller_operations_async.py @@ -0,0 +1,33 @@ +# 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.migrate.aio import AzureMigrateDiscoveryService + +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 TestAzureMigrateDiscoveryServiceWebServersControllerOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(AzureMigrateDiscoveryService, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_list_by_web_app_site(self, resource_group): + response = self.client.web_servers_controller.list_by_web_app_site( + resource_group_name=resource_group.name, + site_name="str", + web_app_site_name="str", + api_version="2023-10-01-preview", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/sdk/migrate/azure-mgmt-migrate/sdk_packaging.toml b/sdk/migrate/azure-mgmt-migrate/sdk_packaging.toml new file mode 100644 index 000000000000..678360b1afb0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/sdk_packaging.toml @@ -0,0 +1,11 @@ +[packaging] +package_name = "azure-mgmt-migrate" +package_nspkg = "azure-mgmt-nspkg" +package_pprint_name = "MyService Management" +package_doc_id = "" +is_stable = false +is_arm = true +need_msrestazure = false +need_azuremgmtcore = true +sample_link = "" +exclude_folders = "" diff --git a/sdk/migrate/azure-mgmt-migrate/setup.py b/sdk/migrate/azure-mgmt-migrate/setup.py new file mode 100644 index 000000000000..d1b8856f7dd0 --- /dev/null +++ b/sdk/migrate/azure-mgmt-migrate/setup.py @@ -0,0 +1,83 @@ +#!/usr/bin/env python + +# ------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for +# license information. +# -------------------------------------------------------------------------- + +import re +import os.path +from io import open +from setuptools import find_packages, setup + +# Change the PACKAGE_NAME only to change folder and different name +PACKAGE_NAME = "azure-mgmt-migrate" +PACKAGE_PPRINT_NAME = "MyService Management" + +# a-b-c => a/b/c +package_folder_path = PACKAGE_NAME.replace("-", "/") +# a-b-c => a.b.c +namespace_name = PACKAGE_NAME.replace("-", ".") + +# Version extraction inspired from 'requests' +with open( + os.path.join(package_folder_path, "version.py") + if os.path.exists(os.path.join(package_folder_path, "version.py")) + else os.path.join(package_folder_path, "_version.py"), + "r", +) as fd: + version = re.search(r'^VERSION\s*=\s*[\'"]([^\'"]*)[\'"]', fd.read(), re.MULTILINE).group(1) + +if not version: + raise RuntimeError("Cannot find version information") + +with open("README.md", encoding="utf-8") as f: + readme = f.read() +with open("CHANGELOG.md", encoding="utf-8") as f: + changelog = f.read() + +setup( + name=PACKAGE_NAME, + version=version, + description="Microsoft Azure {} Client Library for Python".format(PACKAGE_PPRINT_NAME), + long_description=readme + "\n\n" + changelog, + long_description_content_type="text/markdown", + license="MIT License", + author="Microsoft Corporation", + author_email="azpysdkhelp@microsoft.com", + url="https://github.com/Azure/azure-sdk-for-python", + keywords="azure, azure sdk", # update with search keywords relevant to the azure service / product + classifiers=[ + "Development Status :: 4 - Beta", + "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", + "Programming Language :: Python :: 3.12", + "License :: OSI Approved :: MIT License", + ], + zip_safe=False, + packages=find_packages( + exclude=[ + "tests", + # Exclude packages that will be covered by PEP420 or nspkg + "azure", + "azure.mgmt", + ] + ), + include_package_data=True, + package_data={ + "pytyped": ["py.typed"], + }, + install_requires=[ + "isodate>=0.6.1", + "typing-extensions>=4.6.0", + "azure-common>=1.1", + "azure-mgmt-core>=1.3.2", + ], + python_requires=">=3.8", +) diff --git a/sdk/migrate/ci.yml b/sdk/migrate/ci.yml new file mode 100644 index 000000000000..004996a805e6 --- /dev/null +++ b/sdk/migrate/ci.yml @@ -0,0 +1,34 @@ +# DO NOT EDIT THIS FILE +# This file is generated automatically and any changes will be lost. + +trigger: + branches: + include: + - main + - hotfix/* + - release/* + - restapi* + paths: + include: + - sdk/migrate/ + +pr: + branches: + include: + - main + - feature/* + - hotfix/* + - release/* + - restapi* + paths: + include: + - sdk/migrate/ + +extends: + template: ../../eng/pipelines/templates/stages/archetype-sdk-client.yml + parameters: + ServiceDirectory: migrate + TestProxy: true + Artifacts: + - name: azure-mgmt-migrate + safeName: azuremgmtmigrate